Changeset 3589

Timestamp:

Sep 14, 2006, 10:00:44 AM (19 years ago)

Author:

ole

Message:

New parallel test example

File:

• anuga_core/source/anuga_parallel/test_parallel_sw_runup.py (modified) (4 diffs)

anuga_core/source/anuga_parallel/test_parallel_sw_runup.py

@@ -21 +21 @@
 from parallel_api import *
 
+#------------------------------------------------------------------------------
+# Read in processor information
+#------------------------------------------------------------------------------
+
+numprocs = pypar.size()
+myid = pypar.rank()
+processor_name = pypar.Get_processor_name()
+print 'I am processor %d of %d on node %s' %(myid, numprocs, processor_name)
+
 
 #------------------------------------------------------------------------------
@@ -31 +40 @@
     #--------------------------------------------------------------------------
 
-    points, vertices, boundary = rectangular_cross(20, 20) # Basic mesh
+    points, vertices, boundary = rectangular_cross(10, 10) # Basic mesh
 
     domain = Domain(points, vertices, boundary) # Create domain
@@ -49 +58 @@
 
 
+    #------------ -------------------------------------------------------------
+    # Distribute the domain
     #--------------------------------------------------------------------------
-    # Setup boundary conditions
-    #--------------------------------------------------------------------------
+    
+    # Subdivide the mesh
+    print 'Subdivide mesh'
+    nodes, triangles, boundary, triangles_per_proc, quantities = \
+           pmesh_divide_metis(domain, numprocs)
 
-    Br = Reflective_boundary(domain)      # Solid reflective wall
-    Bd = Dirichlet_boundary([-0.2,0.,0.]) # Constant boundary values
+    # Build the mesh that should be assigned to each processor,
+    # this includes ghost nodes and the communicaiton pattern
+    print 'Build submeshes'    
+    submesh = build_submesh(nodes, triangles, boundary,\
+                            quantities, triangles_per_proc)
 
-    # Associate boundary tags with boundary objects
-    domain.set_boundary({'left': Br, 'right': Bd, 'top': Br, 'bottom': Br})
+    # Send the mesh partition to the appropriate processor
+    print 'Distribute submeshes'        
+    for p in range(1, numprocs):
+      send_submesh(submesh, triangles_per_proc, p)
 
+    # Build the local mesh for processor 0
+    points, vertices, boundary, quantities, ghost_recv_dict, full_send_dict = \
+              extract_hostmesh(submesh, triangles_per_proc)
 
-#------------------------------------------------------------------------------
-# Parallel stuff
-#------------------------------------------------------------------------------
-
-if myid == 0:
-    # Distribute the domain
-    points, vertices, boundary, quantities, ghost_recv_dict, full_send_dict,\
-            = distribute_mesh(domain)
     print 'Communication done'        
     
@@ -102 +116 @@
 
 
+
 #------------------------------------------------------------------------------
 # Setup parallel boundary conditions