Changeset 4429 for anuga_work/production/broome_2006/export_results.py

Timestamp:

May 15, 2007, 4:13:58 PM (17 years ago)

Author:

nick

Message:

update broome

File:

• anuga_work/production/broome_2006/export_results.py (modified) (3 diffs)

anuga_work/production/broome_2006/export_results.py

@@ -5 +5 @@
 from os import sep
 
-time_dir = '20061113_040953' 
+time_dir = '20070423_040235_run' #
+cellsize = 25
+timestep = None
+directory = project.output_dir
+#name = directory + time_dir + sep + project.scenario_name
+name = directory+time_dir+sep+project.scenario_name
 
-directory = project.outputdir
-name = directory + time_dir +sep + 'source'
+is_parallel = True
+if is_parallel == True: nodes = 4
+print 'output dir:', name
 
-print 'output dir:', name
-which_var = 4
+which_var = 2
 if which_var == 0:  # Stage
     outname = name + '_stage'
@@ -17 +22 @@
 
 if which_var == 1:  # Absolute Momentum
-    outname = name + '_momentum'
-    quantityname = '(xmomentum**2 + ymomentum**2)**0.5'  #Absolute momentum
+    outname = name + '_momentum_i1'
+    quantityname = '(xmomentum**2 + ymomentum**2)**0.5'  
 
 if which_var == 2:  # Depth
     outname = name + '_depth'
-    quantityname = 'stage-elevation'  #Depth
+    quantityname = 'stage-elevation'  
 
 if which_var == 3:  # Speed
-    outname = name + '_speed'
+    outname = name + '_speed_i0'
     quantityname = '(xmomentum**2 + ymomentum**2)**0.5/(stage-elevation+1.e-6/(stage-elevation))'  #Speed
 
@@ -32 +37 @@
     quantityname = 'elevation'  #Elevation
 
-print 'start sww2dem'
 
-sww2dem(name, basename_out = outname,
-            quantity = quantityname,
-            cellsize = 25,      # would prefer this at 25
-            # define region for viz purposes
-            easting_min = project.eastingmin,
-            easting_max = project.eastingmax,
-            northing_min = project.northingmin,
-            northing_max = project.northingmax,        
-            reduction = max, #this is because we want max quantityname
-            verbose = True,
-            format = 'asc')
 
+if is_parallel == True:
+#    print 'is_parallel',is_parallel
+    for i in range(0,nodes):
+        namei = name + '_P%d_%d' %(i,nodes)
+        outnamei = outname + '_P%d_%d' %(i,nodes)
+        print 'start sww2dem for sww file %d' %(i)
+        sww2dem(namei, basename_out = outnamei,
+                    quantity = quantityname,
+                    timestep = timestep,
+                    cellsize = cellsize,      
+                    easting_min = project.e_min_area,
+                    easting_max = project.e_max_area,
+                    northing_min = project.n_min_area,
+                    northing_max = project.n_max_area,        
+                    reduction = max, 
+                    verbose = True,
+                    format = 'asc')
+else:
+    print 'start sww2dem'
+    sww2dem(name, basename_out = outname,
+                quantity = quantityname,
+                timestep = timestep,
+                cellsize = cellsize,      
+                easting_min = project.e_min_area,
+                easting_max = project.e_max_area,
+                northing_min = project.n_min_area,
+                northing_max = project.n_max_area,        
+                reduction = max, 
+                verbose = True,
+                format = 'asc')