source: anuga_work/production/hobart_2009/get_runup_v2.py @ 7120

Last change on this file since 7120 was 7120, checked in by kristy, 15 years ago
File size: 2.2 KB
Line 
1# Standard modules
2from os import sep
3import os
4from os.path import dirname, basename
5from os import mkdir, access, F_OK
6from os.path import join
7from anuga.utilities.polygon import read_polygon
8from shutil import copy
9import time
10import sys
11# Related major packages
12from anuga.shallow_water.data_manager import get_maximum_inundation_data
13from anuga.shallow_water.data_manager import csv2polygons
14
15# Application specific imports
16from setup_model import project  # Definition of file names and polygons
17import csv
18directory = project.output_folder
19
20time_dir1 = '20090505_150430_run_final_0.8_58292_None_kvanputt'
21time_dir2 = '20090505_150517_run_final_0_58292_None_kvanputt'
22time_dir3 = '20090505_150711_run_final_0_58280_None_kvanputt'
23time_dir4 = '20090505_150805_run_final_0.8_58280_None_kvanputt'
24time_dir5 = '20090505_151322_run_final_0.8_64477_None_kvanputt'
25time_dir6 = '20090505_151447_run_final_0_64477_None_kvanputt'
26
27time_dirs = [time_dir1, time_dir2, time_dir3, time_dir4, time_dir5, time_dir6]
28
29
30# polygons is a dictionary of polygon data indexed by 'id'
31# values is a dictionary of polygon names indexed by 'id'
32polygons, values = csv2polygons(project.images)
33print 'Creating %d highest runup' % len(polygons)
34
35# create a dictionary {<polygon_name>: <polygon_data>, ...}
36area_dictionary = {}
37for id in polygons:
38    area_dictionary[values[id]] = polygons[id] 
39
40# for each time directory ...
41for time_dir in time_dirs:
42    print 'time_dir ',time_dir
43    basename = join(directory, time_dir, project.scenario_name)
44
45    out_file = basename+'_runup.csv'
46    output_file = open(out_file, 'w')
47    s = 'area,q,x,y\n' 
48    output_file.write(s)
49
50    # iterate over polygon names
51    for which_area in values.itervalues():
52        print 'which_area = %s' % which_area
53        polygon = area_dictionary[which_area]
54       
55        q, loc = get_maximum_inundation_data(basename+'.sww',
56                                             polygon=polygon, verbose=True)
57
58        print '%s, %.6f, %.6f, %.6f\n' %(str(which_area), q, loc[0],loc[1])
59        s = '%s, %.6f, %.6f, %.6f\n' %(str(which_area), q, loc[0],loc[1])
60
61        output_file.write(s)
62
63        print 'output file printed to: %s' %out_file
64    output_file.close()
Note: See TracBrowser for help on using the repository browser.