source: trunk/anuga_work/development/Busselton/Analysis/export_results_max.py @ 8452

"""
Generates ascii grids of nominated areas -
Input: sww file from run_perth.py
       boundaries for grids from project.py
Outputs: ascii grids of specified variables
Stored in the 'outputs_dir' folder for respective .sww file

Note:
If producing a grid for the enitre extent cellsize should be greater than 30m
If producing grids for inundation area resolution should be greater than mesh (ie ~22m)
"""

import glob
import os
import sys
from os.path import join
from os import sep
import anuga
from anuga_parallel import myid, numprocs, distribute, finalize, barrier

import time
t0 = time.time()

rootDir = r"/short/w85/Shane/Output"
scenario_name = "Busselton"
event = "B2Full"

time_dirs = [#"final_"+event+"_sfm547_np1",\
#             "final_"+event+"_sfm547_np1_From28800",\
#             "final_"+event+"_sfm547_np1_From57600",\
#             "final_"+event+"_sfm547_np1_From86400",\
#             "final_"+event+"_sfm547_np1_From108000",\
             "final_"+event+"_sfm547_np1_From115200"] 

#Modify the cellsize value to set the size of the raster you require
#Take into account mesh size when aplying this paramater
cellsize = 25 #250

#Now set the timestep at which you want the raster generated.
#Either set the actual timestep required or use 'None' to indicate that
#you want the maximum values in the raster over all timesteps
timestep = None

# Set the special areas of interest.  If none, do: area=['All']
# Areas identified below are specified by cooridinates in project.py  

#area = ['South', 'NW', 'Hobart'] ### JANE - CHECK THIS #####
area = ['All']      

#one or more key strings from var_equations below
#var = ['stage','depth','momentum','elevation']
vars = ['depth','momentum']

######
# Define allowed variable names and associated equations to generate values.
# This would not normally change.
######
var_equations = {'stage':     'stage',
                 'momentum':  '(xmomentum**2 + ymomentum**2)**0.5',
                 'depth':     'stage-elevation',
                 'speed':     '(xmomentum**2 + ymomentum**2)**0.5/(stage-elevation+1.e-6)',
                 'elevation': 'elevation' }

######
# Start script, running through variables, area, folder, sww file (determine by times)
# Then creates a maximum asci if there is more than one sww file
######

proc = 0
Needed = True

for dirID in range(len(time_dirs)):    
    
    if myid == proc:
        time_dir = time_dirs[dirID] 
        var = vars[0]
        
    elif myid == proc + 1:
        time_dir = time_dirs[dirID] 
        var = vars[1]
        
    proc+=2  

if myid >= len(time_dirs)*2:
    print "I am not needed: Proc ", myid
    Needed = False

if Needed:
        
    if var not in var_equations:
        print 'Unrecognized variable name: %s' % which_var
        sys.exit()
    
    for which_area in area:
        if which_area == 'All':
            easting_min = None
            easting_max = None
            northing_min = None
            northing_max = None
        else:
            try:
                easting_min = eval('project.xmin%s' % which_area)
                easting_max = eval('project.xmax%s' % which_area)
                northing_min = eval('project.ymin%s' % which_area)
                northing_max = eval('project.ymax%s' % which_area)
            except AttributeError:
                print 'Unrecognized area name: %s' % which_area
                break
    
        names = []
        Name = glob.glob(join(rootDir,event,time_dir,scenario_name) +'*.sww')
        names.append(Name)
    
        for names in names:
            
            asc_name = []   
            print 'start sww2dem: time_dir = %s' % time_dir               
            for name in names:
    
                test = name
                name = test[:-4]
                outname = name + '_' + which_area + '_' + var + str(cellsize) + "m"
                quantityname = var_equations[var]
    
                print 'start sww2dem: name = %s' % os.path.basename(name)
                print outname
                anuga.sww2dem(name+'.sww', outname+'.asc',
                            quantity = quantityname,
                            cellsize = cellsize,      
                            easting_min = easting_min,
                            easting_max = easting_max,
                            northing_min = northing_min,
                            northing_max = northing_max,        
                            reduction = timestep, 
                            verbose = True)
    
                asc_name.append(outname + '.asc')
                
barrier()

if myid == 0:
    
    Duration = (int((time.time()-t0)/3600), int(((time.time()-t0)%3600)/60), int((time.time()-t0)%60))
    
    print "Finished"
    print "Time taken:"
    print "Hours: %i, Minutes: %i, Seconds: %i" %Duration
    print "Output written to ~", join(rootDir,scenario_name)