source: anuga_work/production/carnarvon/carnarvon_rerun/get_runup.py @ 7116

# 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_134442_run_final_1_27255_Bt_kvanputt'
time_dir2 = '20090324_154738_run_final_1_27283_Bt_kvanputt'
time_dir3 = '20090324_154824_run_final_1_68693_Bt_kvanputt'
time_dir4 = '20090327_154730_run_final_0_68693_Bt_kvanputt'
time_dir5 = '20090327_154824_run_final_0_27255_Bt_kvanputt'
time_dir6 = '20090327_160753_run_final_0_27283_Bt_kvanputt'


time_dirs = [time_dir1, time_dir2, time_dir3, time_dir4, time_dir5, time_dir6]

interior_region = project.interior_regions_data[0]
polygon = read_polygon(join(project.polygons_folder, first_region))

area_dictionary = {'Carnarvon': interior_region}

area = ['Carnarvon'] 

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()