Changeset 23231 for issm/trunk-jpl/src/m/partition/partitioner.py

Timestamp:

09/08/18 13:03:10 (7 years ago)

Author:

kruegern

Message:

NEW: added Dakota/QMU Chaco tests (234,235,413,414,417,444), minor bug fixes, left out 418 due to unresolved memory issues

File:

• issm/trunk-jpl/src/m/partition/partitioner.py (modified) (2 diffs)

issm/trunk-jpl/src/m/partition/partitioner.py

@@ -3 +3 @@
 import MatlabFuncs as m
 from adjacency import *
-#from Chaco import *
+from Chaco import *
 #from Scotch import *
-#from MeshPartition import *
+from MeshPartition import *
 from project3d import *
+from mesh2d import *
 
 def partitioner(md,*varargin):
@@ -60 +61 @@
 
         if m.strcmpi(package,'chaco'):
-                raise RuntimeError('Chaco is not currently supported for this function')
+                #raise RuntimeError('Chaco is not currently supported for this function')
 
                 #  default method (from chaco.m)
-                #method=np.array([1,1,0,0,1,1,50,0,.001,7654321]).reshape(-1,1)
-                #method[0]=3    #  global method (3=inertial (geometric))
-                #method[2]=0    #  vertex weights (0=off, 1=on)
+                method=np.array([1,1,0,0,1,1,50,0,.001,7654321])
+                method[0]=3    #  global method (3=inertial (geometric))
+                method[2]=0    #  vertex weights (0=off, 1=on)
 
                 #specify bisection
-                #method[5]=options.getfieldvalue('section')#  ndims (1=bisection, 2=quadrisection, 3=octasection)
+                method[5]=options.getfieldvalue('section')#  ndims (1=bisection, 2=quadrisection, 3=octasection)
 
                 #are we using weights? 
-                #if m.strcmpi(options.getfieldvalue('weighting'),'on'):
-                        #weights=np.floor(md.qmu.vertex_weight/min(md.qmu.vertex_weight))
-                        #method[2]=1
-                #else:
-                        #weights=[]
+                if m.strcmpi(options.getfieldvalue('weighting'),'on'):
+                        weights=np.floor(md.qmu.vertex_weight/min(md.qmu.vertex_weight))
+                        method[2]=1
+                else:
+                        weights=[]
         
+                method = method.reshape(-1,1)   # transpose to 1x10 instead of 10
+
                 #  partition into nparts
-                #if isinstance(md.mesh,mesh2d):
-                        #part=Chaco(md.qmu.adjacency,weights,[],md.mesh.x, md.mesh.y,np.zeros((md.mesh.numberofvertices,)),method,npart,[]).T+1 #index partitions from 1 up. like metis.
-                #else:
-                        #part=Chaco(md.qmu.adjacency,weights,[],md.mesh.x, md.mesh.y,md.mesh.z,method,npart,[]).T+1 #index partitions from 1 up. like metis.
-        
+                if isinstance(md.mesh,mesh2d):
+                        part=np.array(Chaco(md.qmu.adjacency,weights,np.array([]),md.mesh.x, md.mesh.y,np.zeros((md.mesh.numberofvertices,)),method,npart,np.array([]))).T+1 #index partitions from 1 up. like metis.
+                else:
+                        part=np.array(Chaco(md.qmu.adjacency,weights,np.array([]),md.mesh.x, md.mesh.y,md.mesh.z,method,npart,np.array([]))).T+1 #index partitions from 1 up. like metis.
         
         elif m.strcmpi(package,'scotch'):