source: anuga_work/production/onslow_2006/boundary_plot.py @ 6741

"""Read in csv files and plot time series

"""

import project
from pylab import plot, xlabel, ylabel, savefig, ion, hold, axis, close, figure, legend
from os import sep
 
time_dir1 = '20070608_060316_run_final_1.5_onslow_nbartzis'# HAT onslow
time_dir2 = '20070608_062811_run_final_1.5_exmouth_nbartzis'# HAT exmouth
directory = project.outputdir
boundary_dir=project.boundaries_in_dir

def get_time_series_from_file(filename):
    fid = open(filename)
    lines = fid.readlines()
    fid.close()
    
    t = []
    stage = []
    depth = []
    speed = []
    momentum = []
    for line in lines[1:]:
        fields = line.split(',')
        t.append((float(fields[0])/3600.0+5000./3600.))
        stage.append(float(fields[1]))
        depth.append(float(fields[1])-float(fields[4]))
        speed.append(float(fields[3]))
        momentum.append(float(fields[2]))        
    return t, stage, depth, speed, momentum

max_st = max_sp = max_mom = min_st = min_sp = min_mom = max_t = 0.
count = j = -1

#find maximums for axis for plots
for i in range(47*2):  #could alter get_all_swwfiles to get all file extension and get csv files??
    count += 1
    t = str(count/2.).rfind('.')
    if str(count/2.)[t+1] == str(0):
        j += 1
        extra = 'gauges_time_series_%d.csv' %(j)
        filename = directory + time_dir2 + sep + extra
    else:
        extra = 'gauges_time_series_%d.csv' %(j)
#        filename = directory + time_dir2 + sep + extra
        filename = boundary_dir + sep + extra

    t, stage, depth, speed, momentum = get_time_series_from_file(filename)
    if max(t) > max_t: max_t = max(t)
    if max(stage) > max_st: max_st = max(stage)
    if max(speed) > max_sp: max_sp = max(speed)
    if max(momentum) > max_mom: max_mom = max(momentum)
    if min(stage) < min_st: min_st = min(stage)
    if min(speed) < min_sp: min_sp = min(speed)
    if min(momentum) < min_mom: min_mom = min(momentum)
 
stage_axis = ([5000./3600.,max_t, min_st, max_st])
speed_axis = ([5000/3600.,max_t, min_sp, max_sp])
mom_axis = ([5000/3600.,max_t, min_mom, max_mom])

ion()
"""  
hold(False)

for i, filename in enumerate(filenames):
    t, stage, depth, speed, momentum = get_time_series_from_file(filename)

    i1 = filename.rfind('gauges')
    i2 = filename.rfind('csv')
    name = filename[i1:i2-1]
    loc = filename.rfind('1.5') + 4
    if filename[loc] is 'o': community = 'onslow'
    if filename[loc] is 'e': community = 'exmouth'

    figure(1)
    plot(t, stage, '-b', linewidth=1)
    xlabel('time (hour)')
    ylabel('wave height (m)')
    axis(stage_axis)
    figname = 'stage_%s_%s.png' %(name,community)
    savefig(figname)

    figure(2)
    plot(t, speed, '-b', linewidth=1)
    xlabel('time (hour)')
    ylabel('speed (m/s)')
    axis(speed_axis)
    figname = 'speed_%s_%s.png' %(name,community)
    savefig(figname)

    figure(3)
    plot(t, momentum, '-b', linewidth=1)
    xlabel('time (hour)')
    ylabel('momentum (m^2/sec)')
    axis(mom_axis)
    figname = 'mom_%s_%s.png' %(name,community)
    savefig(figname)

close('all')
"""

hold(True)
count = j = -1
figcount = 0
cstr = ['b', 'r']

for i in range(4*2):

    count += 1
    t = str(count/2.).rfind('.')
    if str(count/2.)[t+1] == str(0):
        j += 1
        ii = 0
        extra = 'gauges_time_series_%d.csv' %(j)
        filename = directory + time_dir2 + sep + extra
    else:
        ii = 1
        extra = 'gauges_time_series_%d.csv' %(j)
#        filename = directory + time_dir2 + sep + extra
        filename = boundary_dir + sep + extra
        
    t, stage, depth, speed, momentum = get_time_series_from_file(filename)

    i1 = filename.rfind('gauges')
    i2 = filename.rfind('csv')
    name = filename[i1:i2-1]
    loc = filename.rfind('1.5') + 4
#    if filename[loc] is 'o': community = 'onslow'
#    if filename[loc] is 'e': community = 'exmouth'

    print 'hello', community, filename
    figure(100+j)
    plot(t, stage, c = cstr[ii], linewidth=1)
    xlabel('time (hour)')
    ylabel('wave height (m)')
    axis(stage_axis)
    legend(('Onslow','Exmouth'),loc='upper right')
    figname = 'stage_%s_both.png' %(name)
    savefig(figname)

    figure(200+j)
    plot(t, speed, c = cstr[ii], linewidth=1)
    xlabel('time (hour)')
    ylabel('speed (m/s)')
    axis(speed_axis)
    legend(('Onslow','Exmouth'),loc='upper right')
    figname = 'speed_%s_both.png' %(name)
    savefig(figname)

    figure(300+j)
    plot(t, momentum, c = cstr[ii], linewidth=1)
    xlabel('time (hour)')
    ylabel('momentum (m^2/sec)')
    axis(mom_axis)
    legend(('Onslow','Exmouth'),loc='upper right')
    figname = 'mom_%s_both.png' %(name)
    savefig(figname)
  
close('all')