Changeset 3593 for anuga_core/source/anuga_parallel/parallel_api.py

Timestamp:

Sep 14, 2006, 2:40:38 PM (17 years ago)

Author:

ole

Message:

Got first version of parallel api going

Still todo: Communicate all attributes of domain such as name and also boundary conditions across to other processes automatically.

File:

• anuga_core/source/anuga_parallel/parallel_api.py (modified) (1 diff)

anuga_core/source/anuga_parallel/parallel_api.py

@@ -21 +21 @@
 processor_name = pypar.Get_processor_name()
 print 'I am processor %d of %d on node %s' %(myid, numprocs, processor_name)
+
+
+
+
+def distribute(domain):
+
+    if myid == 0:
+        #-------------------------------------------------------------------
+        # Distribute the domain
+        #-------------------------------------------------------------------
+
+        points, vertices, boundary, quantities,\
+                ghost_recv_dict, full_send_dict,\
+                = distribute_mesh(domain)
+        print 'Communication done'        
+        
+    else:
+        # Read in the mesh partition that belongs to this
+        # processor (note that the information is in the
+        # correct form for the GA data structure)
+
+        points, vertices, boundary, quantities,\
+                ghost_recv_dict, full_send_dict,\
+                = rec_submesh(0)
+
+    #------------------------------------------------------------------------
+    # Start the computations on each subpartion
+    #------------------------------------------------------------------------
+
+    # Build the domain for this processor
+    domain = Parallel_Domain(points, vertices, boundary,
+                             full_send_dict  = full_send_dict,
+                             ghost_recv_dict = ghost_recv_dict)
+
+    #------------------------------------------------------------------------
+    # Setup initial conditions
+    #------------------------------------------------------------------------
+    for q in quantities:
+        domain.set_quantity(q, quantities[q]) # Distribute all quantities    
+
+
+    #------------------------------------------------------------------------
+    # Return parallel domain to all nodes
+    #------------------------------------------------------------------------
+    return domain    
+
+
+