- Timestamp:
- Sep 14, 2006, 2:40:38 PM (17 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
anuga_core/source/anuga_parallel/test_parallel_sw_runup.py
r3591 r3593 25 25 from parallel_api import * 26 26 27 #------------------------------------------------------------------------------28 # Read in processor information29 #------------------------------------------------------------------------------30 27 31 numprocs = pypar.size() 32 myid = pypar.rank() 33 processor_name = pypar.Get_processor_name() 34 print 'I am processor %d of %d on node %s' %(myid, numprocs, processor_name) 28 #-------------------------------------------------------------------------- 29 # Setup computational domain 30 #-------------------------------------------------------------------------- 31 points, vertices, boundary = rectangular_cross(10, 10) # Basic mesh 32 domain = Domain(points, vertices, boundary) # Create domain 33 domain.set_name('runup') # Set sww filename 35 34 36 35 37 #-------------------------------------------------------------------------- ----38 # Initialise39 #-------------------------------------------------------------------------- ----36 #-------------------------------------------------------------------------- 37 # Setup initial conditions 38 #-------------------------------------------------------------------------- 40 39 41 if myid == 0: 42 #-------------------------------------------------------------------------- 43 # Setup computational domain 44 #-------------------------------------------------------------------------- 40 def topography(x,y): 41 return -x/2 # linear bed slope 45 42 46 points, vertices, boundary = rectangular_cross(10, 10) # Basic mesh 47 48 domain = Domain(points, vertices, boundary) # Create domain 43 domain.set_quantity('elevation', topography) # Use function for elevation 44 domain.set_quantity('friction', 0.1) # Constant friction 45 domain.set_quantity('stage', -.4) # Constant initial stage 49 46 50 47 51 # Unstructured mesh 52 #polygon = [[1,1],[0,1],[0,0],[1,0]] 53 #meshname = 'runup.msh' 54 #create_mesh_from_regions(polygon, 55 # boundary_tags={'top': [0], 56 # 'left': [1], 57 # 'bottom': [2], 58 # 'right': [3]}, 59 # maximum_triangle_area=0.01, 60 # filename=meshname) 61 #domain = Domain(meshname, use_cache=True, verbose = True) 48 #-------------------------------------------------------------------------- 49 # Create the parallel domain 50 #-------------------------------------------------------------------------- 51 domain = distribute(domain) 62 52 63 53 64 65 domain.set_name('runup') # Set sww filename66 67 54 68 #------------ ------------------------------------------------------------- 69 # Setup initial conditions 70 #-------------------------------------------------------------------------- 71 72 def topography(x,y): 73 return -x/2 # linear bed slope 74 75 domain.set_quantity('elevation', topography) # Use function for elevation 76 domain.set_quantity('friction', 0.1) # Constant friction 77 domain.set_quantity('stage', -.4) # Constant initial stage 78 79 80 #------------ ------------------------------------------------------------- 81 # Distribute the domain 82 #-------------------------------------------------------------------------- 83 84 # Subdivide the mesh 85 print 'Subdivide mesh' 86 nodes, triangles, boundary, triangles_per_proc, quantities = \ 87 pmesh_divide_metis(domain, numprocs) 88 89 # Build the mesh that should be assigned to each processor, 90 # this includes ghost nodes and the communicaiton pattern 91 print 'Build submeshes' 92 submesh = build_submesh(nodes, triangles, boundary,\ 93 quantities, triangles_per_proc) 94 95 # Send the mesh partition to the appropriate processor 96 print 'Distribute submeshes' 97 for p in range(1, numprocs): 98 send_submesh(submesh, triangles_per_proc, p) 99 100 101 # Build the local mesh for processor 0 102 points, vertices, boundary, quantities, ghost_recv_dict, full_send_dict = \ 103 extract_hostmesh(submesh, triangles_per_proc) 104 105 print 'Communication done' 106 107 108 else: 109 # Read in the mesh partition that belongs to this 110 # processor (note that the information is in the 111 # correct form for the GA data structure) 112 113 points, vertices, boundary, quantities, ghost_recv_dict, full_send_dict, \ 114 = rec_submesh(0) 115 116 #------------------------------------------------------------------------------ 117 # Start the computations on each subpartion 118 #------------------------------------------------------------------------------ 119 120 # Build the domain for this processor 121 domain = Parallel_Domain(points, vertices, boundary, 122 full_send_dict = full_send_dict, 123 ghost_recv_dict = ghost_recv_dict) 55 # TODO: Communicate all attributes of domain including boundary conditions 124 56 125 57 # Name and dir, etc currently has to be set here as they are not … … 127 59 domain.set_name('runup') # Set sww filename 128 60 129 130 #------------------------------------------------------------------------------131 # Setup initial conditions132 #------------------------------------------------------------------------------133 for q in quantities:134 domain.set_quantity(q, quantities[q]) # Distribute all quantities135 61 136 62 … … 144 70 145 71 # Associate boundary tags with boundary objects 146 #domain.set_boundary({'left': Br, 'right': Bd, 'top': Br, 'bottom': Br,147 # 'ghost': None, 'exterior': Bd})148 72 domain.set_boundary({'left': Br, 'right': Bd, 'top': Br, 'bottom': Br, 149 73 'ghost': None})
Note: See TracChangeset
for help on using the changeset viewer.
