Changeset 680 for codes/icosagcm/devel/Python/src

Timestamp:

02/09/18 16:24:33 (6 years ago)

Author:

dubos

Message:

devel/unstructured : moved mesh partitioning code into meshes.py

File:

• codes/icosagcm/devel/Python/src/unstructured.pyx (modified) (3 diffs)

codes/icosagcm/devel/Python/src/unstructured.pyx

@@ -1 +1 @@
 import time
 import math
-import dynamico.wrap as wrap
-from dynamico.libs import libicosa
+import wrap
+from libs import libicosa
+from util import list_stencil
 
 from ctypes import c_void_p, c_int, c_double, c_bool
@@ -289 +290 @@
 
 # Helper functions and interface to ParMETIS
-# list_stencil converts an adjacency graph from array format index[num_cells, MAX_EDGES] to compressed format
 # loc_stencil returns the start/end indices (vtxdist) expected by ParMETIS
 # i.e. index[start:end] with start=vtxdist[cell], end=vtxdist[cell+1] lists the edges of cell 'cell'
@@ -295 +295 @@
 import numpy as np
 
-def list_stencil(degree, stencil, cond=lambda x:True):
-    for i in range(degree.size):
-        for j in range(degree[i]):
-            s=stencil[i,j]
-            if cond(s): yield stencil[i,j]
-                
 def loc_stencil(degree, stencil):
     loc=0