source: anuga_work/production/wa/busselton_2009/get_runup.py @ 7436

# Standard modules
from os import sep
import os
from os.path import dirname, basename
from os import mkdir, access, F_OK
from os.path import join
from anuga.utilities.polygon import read_polygon
from shutil import copy
import time
import sys
# Related major packages
from anuga.shallow_water.data_manager import get_maximum_inundation_data
        
# Application specific imports
from setup_model import project  # Definition of file names and polygons
import csv
directory = project.output_folder

time_dir1 = '20090324_134328_run_final_0.6_27255_Bt_kvanputt'
time_dir2 = '20090324_140747_run_final_0.6_68693_Bt_kvanputt'
time_dir3 = '20090327_152717_run_final_0.6_27283_Bt_kvanputt'
time_dir4 = '20090327_160646_run_final_0_27255_Bt_kvanputt'
time_dir5 = '20090327_161520_run_final_0_68693_Bt_kvanputt'
time_dir6 = '20090330_120458_run_final_0_27283_Bt_kvanputt'

time_dirs = [time_dir6] #3, time_dir4, time_dir5]

interior_region = []
for region in project.interior_regions_data[:2]:
    polygon = read_polygon(join(project.polygons_folder,
                                region[0]))
    interior_region.append(polygon)

area_dictionary = {'Busselton': interior_region[0],
                    'Bunbury' : interior_region[1]}

area = ['Busselton', 'Bunbury'] 

for time_dir in time_dirs:

    basename = join(directory, time_dir, project.scenario_name)

    out_file = basename+'_runup.csv'
    output_file = open(out_file, 'w')
    s = 'area,q,x,y\n' 
    output_file.write(s)
    print 'basename ',basename 

    for which_area in area:
        if which_area not in area_dictionary:
            print 'Unrecognized variable name: %s' % which_area
            break

        polygon = area_dictionary[which_area]
        
        print 'Which area ',which_area

        q, loc = get_maximum_inundation_data(basename+'.sww',
                                             polygon=polygon, verbose=True)

        print '%s, %.6f, %.6f, %.6f\n' %(str(which_area), q, loc[0],loc[1])
        s = '%s, %.6f, %.6f, %.6f\n' %(str(which_area), q, loc[0],loc[1])

        output_file.write(s)

        print 'output file printed to: %s' %out_file
    output_file.close()