source: anuga_work/production/dampier_2006/get_timeseries_and.py @ 4308

"""
Generate images of "gauges" for production run

Inputs:

production dirs: dictionary of production directories with a
                 association to that simulation run, eg high tide,
                 low tide and MSL.
                    
Outputs:

* figures used for report stored in the report_figure directory

"""

from os import sep, getcwd, access, F_OK, mkdir, getenv
#from anuga.abstract_2d_finite_volumes.util import get_data_from_file
import os
from Numeric import zeros, array, allclose
from os import getcwd, sep, altsep, mkdir, access, F_OK
import project
from anuga.abstract_2d_finite_volumes.util import start_screen_catcher, sww2timeseries, get_data_from_file

#makes the csv files from the evolved model


home = getenv('INUNDATIONHOME') #Sandpit's parent dir   
#user = get_user_name()
data = 'data'
state = 'western_australia'
scenario_name = 'dampier.sww'
scenario = 'dampier_tsunami_scenario_2006'
#scenario = 'test_dampier'
an = 'anuga'
bo = 'boundaries'

run_time = '20070312_030201_run'
production_dirs = {#run_time: 'URS evolved data'#,
                   'boundaries': 'URS boundary condition'
                   }

topo = 'topographies'
out = 'outputs'
urs = 'urs'
gridded = '1_10000'

#gauge_filename = os.path.join(home,data,state,scenario,an,out,scenario_name,'.sww')
#sww_filename ={}
#swwfiles = {}

#sww_filename[os.path.join(home,data,state,scenario,an,out,run_time,scenario_name)] = run_time

gauge_boundary_filename = 'gauges_time_series_near_top.csv'
gauge_evolved_filename = 'gauges_time_series_near_top.csv'
#gauge_boundary_filename = 'gauges_time_series_middle.csv'
#gauge_evolved_filename = 'gauges_time_series_middle.csv'

#gauge_boundary_filename = 'gauges_time_series_first.csv'
#gauge_evolved_filename = 'gauges_time_series_first.csv'

boundary_dir_filename = os.path.join(home,data,state,scenario,an,bo,gauge_boundary_filename)
print'boundary_dir_filename',boundary_dir_filename
out_dir=os.path.join(home,data,state,scenario,an,out,run_time)
out_dir_name= out_dir+sep

evolved_dir_filename= os.path.join(home,data,state,scenario,an,out,run_time,gauge_evolved_filename)

#start_screen_catcher(out_dir_name, print_to_screen=True)
start_screen_catcher(out_dir_name, extra_info='get_time', print_to_screen=True)
print'boundary_dir_filename',boundary_dir_filename
print'evolved_dir_filename',evolved_dir_filename

#file_loc = project.output_dir + label_id + sep
#swwfile = file_loc + project.scenario_name + '.sww'
#swwfiles[swwfile] = label_id
print "hella what the!"
'''
#get the timeseries for the evolved sww file
texname, elev_output = sww2timeseries(sww_filename,
                                      project.gauges_dir_name_simple,
                                      production_dirs,
                                      report = False,
                                      #reportname = report_name,
                                      plot_quantity = ['stage', 'momentum'],
                                      surface = False,
                                      time_min = None,
                                      time_max = None,
                                      title_on = False,
                                      verbose = True)
                                      
'''                                      
swwfiles = {}
                                      
for label_id in production_dirs.keys():
        
    if label_id == 'boundaries':
        print 'boundaries'
#        file_loc = project.boundaries_in_dir
#        swwfile = project.boundaries_dir_name3 + '.sww'
        swwfile = project.boundaries_dir_name6+'.sww'
    else:
#        file_loc = project.output_dir + label_id + sep
        file_loc = out_dir_name
        swwfile = file_loc + project.scenario_name + '.sww'
    swwfiles[swwfile] = label_id
    print"swwfiles",swwfiles
        
texname, elev_output = sww2timeseries(swwfiles,
                                      project.gauges_dir_name_simple,
                                      production_dirs,
                                      report = False,
                                      #reportname = report_name,
                                      plot_quantity = ['stage', 'xmomentum', 'ymomentum'],
#                                      plot_quantity = ['stage', 'momentum'],
                                      surface = False,
                                      time_min = None,
                                      time_max = None,
                                      title_on = False,
#                                      use_cache = True,
                                      verbose = True)

#makes the csv files from the evolved model



e_time, e_stage, e_momentum, e_speed, e_elevation = get_data_from_file(evolved_dir_filename)

print'boundary_dir_filename',boundary_dir_filename
b_time, b_stage, b_momentum, b_speed, b_elevation = get_data_from_file(boundary_dir_filename)

# compares the 2 models
j=0
k=2
b_sample = []
e_sample = []
for i in range(len(b_time)):
#    if j<(len(e_time)) and b_time[i] == e_time[j]:
    if j<(len(e_time)) and k<(len(e_time)) and b_time[i] == e_time[j]:
        b_sample.append(float(b_stage[i]))
        e_sample.append(float(e_stage[k]))
        if k <len(e_time): print 'time e equal b:', b_time[i], b_stage[i],i, j, b_sample[j],e_stage[j], e_stage[k],(len(e_time)-1)
        j = j +1
        k = k +1


e_stage.pop()
e_stage.pop()
 
print len(b_sample), len(e_stage)
#assert allclose (b_sample, e_sample, 0.5, 0.5)

assert allclose (b_sample, e_stage, 0.5, 0.5) 
print "test successful"