Changeset 23440

Timestamp:

10/30/18 17:59:04 (6 years ago)

Author:

schlegel

Message:

BUG: Python partitioning for 3D meshes had not been tested

File:

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

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

@@ -1 +1 @@
 import numpy as np
+import copy
 from pairoptions import *
 import MatlabFuncs as m
@@ -41 +42 @@
         recomputeadjacency=options.getfieldvalue('recomputeadjacency')
 
-        if(np.ndim(md.mesh)==3):
+        if(md.mesh.dimension()==3):
                 #partitioning essentially happens in 2D. So partition in 2D, then 
                 #extrude the partition vector vertically. 
-                md3d=md #save  for later
+                md3d = copy.deepcopy(md)
                 md.mesh.elements=md.mesh.elements2d
                 md.mesh.x=md.mesh.x2d
@@ -116 +117 @@
 
         #extrude if we are in 3D:
-        if np.ndim(md.mesh)==3:
+        if md.mesh.dimension()==3:
                 md3d.qmu.vertex_weight=md.qmu.vertex_weight
                 md3d.qmu.adjacency=md.qmu.adjacency
                 md=md3d
-                part=project3d(md,'vector',part.T,'type','node')
+                part=project3d(md,'vector',np.squeeze(part),'type','node')
 
         md.qmu.partition=part