# 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_135427_run_final_0.6_27255_Bt_kvanputt'
time_dir2 = '20090324_154957_run_final_0.6_68693_Bt_kvanputt'
time_dir3 = '20090324_155028_run_final_0.6_27283_Bt_kvanputt'
time_dir4 = '20090327_155353_run_final_0_27283_Bt_kvanputt'
time_dir5 = '20090327_155659_run_final_0_27255_Bt_kvanputt'
time_dir6 = '20090327_155803_run_final_0_68693_Bt_kvanputt'

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

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

area_dictionary = {'Fremantle' : interior_region[0],
                  'Rockingham' : interior_region[1],
                     'Geordie' : interior_region[3],
                    'Sorrento' : interior_region[4]}

area = ['Fremantle', 'Rockingham', 'Geordie', 'Sorrento'] 

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