Update get_particle_id/cpu

As implemented in pyAMReX with
AMReX-Codes/pyamrex#165

Python/pywarpx/_libwarpx.py

@@ -94,6 +94,9 @@ def load_library(self):
         except ImportError:
             raise Exception(f"Dimensionality '{self.geometry_dim}' was not compiled in this Python install. Please recompile with -DWarpX_DIMS={_dims}")
 
+        # matching space-dim of pyAMReX module
+        self.amr = self.libwarpx_so.amr
+
     def getNProcs(self):
         '''
 

Python/pywarpx/particle_containers.py

@@ -248,14 +248,8 @@ def get_particle_id(self, level=0):
         numbers on each tile.
 
         '''
-        raise NotImplementedError()
-        ids = []
         structs = self.get_particle_structs(level)
-        for ptile_of_structs in structs:
-            arr = np.empty(ptile_of_structs.shape, np.int64)
-            self.libwarpx_so.warpx_convert_id_to_long(arr, ptile_of_structs, arr.size)
-            ids.append(arr)
-        return ids
+        return [libwarpx.amr.unpack_ids(struct['cpuid']) for struct in structs]
 
     def get_particle_cpu(self, level=0):
         '''
@@ -264,14 +258,8 @@ def get_particle_cpu(self, level=0):
         numbers on each tile.
 
         '''
-        raise NotImplementedError()
-        cpus = []
         structs = self.get_particle_structs(level)
-        for ptile_of_structs in structs:
-            arr = np.empty(ptile_of_structs.shape, np.int32)
-            self.libwarpx_so.warpx_convert_cpu_to_int(arr, ptile_of_structs, arr.size)
-            cpus.append(arr)
-        return cpus
+        return [libwarpx.amr.unpack_cpus(struct['cpuid']) for struct in structs]
 
     def get_particle_x(self, level=0):
         '''