Changeset 2105 for inundation/parallel/build_submesh.py

Timestamp:

Dec 1, 2005, 6:04:45 PM (19 years ago)

Author:

steve

Message:

Updating merimbula code

File:

• inundation/parallel/build_submesh.py (modified) (18 diffs)

inundation/parallel/build_submesh.py

@@ -51 +51 @@
     submesh = {}
     tsubnodes = concatenate((reshape(arrayrange(nnodes),(nnodes,1)), nodes), 1)
-    
+
     # loop over processors
 
@@ -63 +63 @@
 
         # find the boundary edges on processor p
-        
+
         subboundary = {}
         for k in boundary:
@@ -69 +69 @@
                 subboundary[k]=boundary[k]
         boundary_list.append(subboundary)
-        
+
         # find nodes in processor p
 
@@ -77 +77 @@
             nodemap[t[1]]=1
             nodemap[t[2]]=1
-                   
+
         node_list.append(take(tsubnodes,nonzero(nodemap)))
 
@@ -127 +127 @@
     ncoord = len(mesh.coordinates)
     ntriangles = len(mesh.triangles)
-    
+
     # find the first layer of boundary triangles
 
@@ -166 +166 @@
     nodemap = zeros(ncoord, 'i')
     fullnodes = submesh["full_nodes"][p]
-    
+
     subtriangles = []
     for i in range(len(trianglemap)):
@@ -178 +178 @@
     tsubtriangles = concatenate((trilist, mesh.triangles), 1)
     subtriangles = take(tsubtriangles, nonzero(trianglemap))
-    
+
     # keep a record of the triangle vertices, if they are not already there
 
@@ -188 +188 @@
     tsubnodes = concatenate((nodelist, mesh.coordinates), 1)
     subnodes = take(tsubnodes, nonzero(nodemap))
-     
+
     # clean up before exiting
 
@@ -223 +223 @@
 
     ghost_commun = zeros((len(subtri), 2), Int)
-    
+
     for i in range(len(subtri)):
         global_no = subtri[i][0]
@@ -241 +241 @@
 
         ghost_commun[i] = [global_no, neigh]
-    
+
     return ghost_commun
 
@@ -343 +343 @@
 
         # build the ghost boundary layer
-        
+
         [subnodes, subtri] = ghost_layer(submesh, mesh, p, tupper, tlower)
         ghost_triangles.append(subtri)
         ghost_nodes.append(subnodes)
-    
+
         # build the communication pattern for the ghost nodes
 
@@ -392 +392 @@
 
 def submesh_quantities(submesh, quantities, triangles_per_proc):
-    
+
     nproc = len(triangles_per_proc)
 
@@ -398 +398 @@
 
     # build an empty dictionary to hold the quantites
-    
+
     submesh["full_quan"] = {}
     submesh["ghost_quan"] = {}
@@ -406 +406 @@
 
     # loop trough the subdomains
-    
+
     for p in range(nproc):
         upper =   lower+triangles_per_proc[p]
 
         # find the global ID of the ghost triangles
-        
+
         global_id = []
         M = len(submesh["ghost_triangles"][p])
@@ -419 +419 @@
         # use the global ID to extract the quantites information from
         # the full domain
-        
+
         for k in quantities:
             submesh["full_quan"][k].append(quantities[k][lower:upper])
@@ -456 +456 @@
 
     submeshf = submesh_full(nodes, triangles, edges, triangles_per_proc)
-        
+
     # add any extra ghost boundary layer information
 
@@ -463 +463 @@
     # order the quantities information to be the same as the triangle
     # information
-    
+
     submesh = submesh_quantities(submeshg, quantities, triangles_per_proc)
-    
+
     return submesh
 
@@ -498 +498 @@
         submesh_cell["full_quan"][k] = submesh["full_quan"][k][0]
         submesh_cell["ghost_quan"][k] = submesh["ghost_quan"][k][0]
-        
+
     return submesh_cell
 
-