Changeset 986 for codes/icosagcm/devel/Python/dynamico/partition.pyx

Timestamp:

10/24/19 18:27:01 (5 years ago)

Author:

dubos

Message:

devel/Python : update reorder.py and fix C syntax of partition.h

File:

• codes/icosagcm/devel/Python/dynamico/partition.pyx (modified) (4 diffs)

codes/icosagcm/devel/Python/dynamico/partition.pyx

@@ -1 @@
-
 from ctypes import c_void_p, c_int, c_double, c_float, c_bool
 from numpy cimport ndarray
@@ -6 +5 @@
 
 ctypedef int idx_t
+ctypedef unsigned long uint64_t
 
 cdef extern from "partition.h":
@@ -11 +11 @@
                      idx_t vtxdist[], idx_t xadj_loc[], idx_t adjncy_loc[],
                      idx_t nparts, idx_t part[])
+    void dynamico_morton_encode(int n, const int *x, const int *y, const int *z, long *m)
 
 
@@ -51 +52 @@
     dynamico_partition_graph(mpi_rank, mpi_size, &vtxdist[0], &xadj_loc[0], &adjncy_loc[0], nparts, &owner[0])
     return owner
+
+#-------------------------------- Space-filling (Morton) ordering ------------------------#
+
+def toint(x): return np.int32((x+1.)*65536)
+
+def morton_index(x,y,z):
+    cdef int nn
+    cdef int[::1] xx,yy,zz
+    cdef long[::1] idx_
+    nn, xx, yy, zz = x.size, toint(x), toint(y), toint(z)
+    idx=np.zeros(nn, dtype=np.int64)
+    idx_=idx
+    dynamico_morton_encode(nn, &xx[0],&yy[0],&zz[0], &idx_[0])
+    return idx