Changeset 2092

Timestamp:

Nov 29, 2005, 1:58:09 PM (19 years ago)

Author:

jack

Message:

pmesh_divide_metis doesn't catch fire anymore when called with n_procs = 1. It now assigns all
the triangles to the one processor. Could probably be made more efficient.

File:

• inundation/parallel/pmesh_divide.py (modified) (2 diffs)

inundation/parallel/pmesh_divide.py

@@ -11 +11 @@
 #  Authors: Linda Stals and Matthew Hardy, June 2005
 #  Modified: Linda Stals, Nov 2005
+#            Jack Kelly, Nov 2005
 #
 #
@@ -310 +311 @@
     # Prepare variables for the metis call
     n_tri = len(domain.triangles)
-    n_vert = len(domain.coordinates)
-    t_list = domain.triangles.copy()
-    t_list = reshape(t_list, (-1,))
-    
-    # The 1 here is for triangular mesh elements.
-    edgecut, epart, npart = partMeshNodal(n_tri, n_vert, t_list, 1, n_procs)
-    
-    # print edgecut
+    if n_procs != 1: #Because metis chokes on it...
+        n_vert = len(domain.coordinates)
+        t_list = domain.triangles.copy()
+        t_list = reshape(t_list, (-1,))
+    
+        # The 1 here is for triangular mesh elements.
+        edgecut, epart, npart = partMeshNodal(n_tri, n_vert, t_list, 1, n_procs)
+        del edgecut
+        del npart
+        # print edgecut
+        # print npart
+    else:
+        epart = [0] * n_tri
+
     # print epart
-    # print npart
-
-    del edgecut
-    del npart
+
 
     # Assign triangles to processes, according to what metis told us.