source: anuga_work/production/new_south_wales/batemans_bay/export_results_max.py @ 7327

Last change on this file since 7327 was 7327, checked in by jgriffin, 15 years ago
File size: 4.0 KB
Line 
1"""
2Generates ascii grids of nominated areas -
3Input: sww file from run_model.py
4       boundaries for grids from project.py
5Outputs: ascii grids of specified variables
6Stored in the 'outputs_folder' folder for respective .sww file
7
8Note:
9If producing a grid for the enitre extent cellsize should be greater than 30m
10If producing grids for inundation area resolution should be greater than mesh (ie ~22m)
11"""
12
13import project, os
14import sys
15from os.path import join
16from anuga.lib.maxasc.maxasc import MaxAsc
17from anuga.shallow_water.data_manager import convert_dem_from_ascii2netcdf, dem2pts
18from anuga.shallow_water.data_manager import sww2dem
19from os import sep
20
21
22directory = project.output_folder
23
24#Specify output directories
25time_dir1 = '20090529_143527_run_final_0.0_51424_jgriffin'
26#time_dir2 = '20090529_143458_run_final_0.0_58346_jgriffin'
27
28time_dirs = [time_dir1]#, time_dir2]   
29
30cellsize = 20   #dependent on data resolution in area of interest.
31
32timestep = None    # None means no timestep!
33##timestep = 0
34
35######
36# Set the special areas of interest.  If none, do: area='All'
37######
38
39#area = ['', '']  # strings must match keys in var_equations below
40area = ['All']      # 'All' means no special areas - the whole thing
41
42######
43# Define allowed variable names and associated equations to generate values.
44# This would not normally change.
45######
46var_equations = {'stage':     'stage',
47                 'momentum':  '(xmomentum**2 + ymomentum**2)**0.5',
48                 'depth':     'stage-elevation',
49                 'speed':     '(xmomentum**2 + ymomentum**2)**0.5/(stage-elevation+1.e-6)',
50                 'elevation': 'elevation' }
51
52# one or more key strings from var_equations above
53#var = ['depth', 'speed','stage']
54var = ['elevation']
55######
56# Start running the various conversions we require.
57######
58
59for which_var in var:
60    if which_var not in var_equations:
61        print 'Unrecognized variable name: %s' % which_var
62        break
63
64    for which_area in area:
65        if which_area == 'All':
66            easting_min = None
67            easting_max = None
68            northing_min = None
69            northing_max = None
70        else:
71            try:
72                easting_min = eval('project.xmin%s' % which_area)
73                easting_max = eval('project.xmax%s' % which_area)
74                northing_min = eval('project.ymin%s' % which_area)
75                northing_max = eval('project.ymax%s' % which_area)
76            except AttributeError:
77                print 'Unrecognized area name: %s' % which_area
78                break
79
80        for time_dir in time_dirs:
81
82            name1 = join(directory, time_dir, project.scenario_name)
83            name2 = join(directory, time_dir, project.scenario_name)+'_time_37860_0'
84
85           
86
87            names = [name1, name2]
88     
89            asc_name = []   
90               
91            for name in names:
92                         
93                outname = name + '_' + which_area + '_' + which_var
94                quantityname = var_equations[which_var]
95
96                print 'start sww2dem: time_dir=%s' % time_dir
97               
98                sww2dem(name, basename_out = outname,
99                            quantity = quantityname,
100                            timestep = timestep,
101                            cellsize = cellsize,     
102                            easting_min = easting_min,
103                            easting_max = easting_max,
104                            northing_min = northing_min,
105                            northing_max = northing_max,       
106                            reduction = max, 
107                            verbose = True,
108                            format = 'asc')
109
110                asc_name.append(outname + '.asc')
111   
112            maxasc_outname = join(directory, time_dir, project.scenario_name)+'_'+which_area+'_'+which_var+'_max.asc'
113
114            print 'max asc outname ', maxasc_outname
115            print 'asc_name ', str(asc_name)
116           
117            MaxAsc(maxasc_outname, asc_name)
Note: See TracBrowser for help on using the repository browser.