source: anuga_work/production/shark_bay_2007/plot_gauges.py @ 4885

"""Read in csv files and plot time series

"""
import project
from pylab import plot, xlabel, ylabel, savefig, ion, hold, axis, close, figure, legend, title
from os import sep

directory = project.output_dir
dirnames = [
#'20070710_023849_run_final_-0.85_shark_bay_onielsen_10000_tide-0.85_coastpolys',
#'20070710_001018_run_final_0.85_shark_bay_onielsen_10000_tide0.85_coastpolys',
#'20070710_001110_run_final_0.85_shark_bay_onielsen_july2006_tide0.85_coastpolys',
#'20070710_023824_run_final_-0.85_shark_bay_onielsen_july2006_tide-0.85_coastpolys',
'20070712_003638_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T6',
'20070712_003626_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T12',
'20070712_003616_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T25',
'20070712_003606_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T50',
'20070712_001432_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T100',
'20070712_002232_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T200',
'20070712_003334_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T300',
'20070712_002254_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T400',
'20070712_003400_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T500',
'20070712_003410_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T600',
'20070712_003421_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T700',
'20070712_002317_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T800',
'20070712_002331_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T900',
'20070712_003433_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T1000',
'20070712_003443_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T1100',
'20070712_003453_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T1200',
'20070712_003527_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T1300',
'20070712_002526_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T1800',
'20070712_002604_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T3600',
'20070712_002625_run_final_0.0_shark_bay_onielsen_experimental_A0.5_T7200'
]

period = [6,12,15,50,100,200,300,400,500,600,700,800,900,1000,1100,1200,1300,1800,3600,7200]

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))
        stage.append(float(fields[1]))
        #depth.append(float(fields[1])-float(fields[4]))
        speed.append(float(fields[3]))
        #momentum.append(float(fields[2]))
    return stage, speed
    
from Numeric import zeros, Float, size
gauges = range(3,80)
plot_gauges = range(77)
max_st = max_sp = min_st = min_sp = max_m = min_m = zeros((len(gauges),len(dirnames)), Float)
max_t = []

for j, dirname in enumerate(dirnames):
    for i, gauge in enumerate(gauges):  #could alter get_all_swwfiles to get all file extension and get csv files??
        extra = 'gauges_time_series_%d.csv' %(gauge)
        filename = directory + dirname + sep + extra
        stage, speed = get_time_series_from_file(filename)
        #max_t.append(max(t))
        max_st[i,j] = max(stage)
        max_sp[i,j] = max(speed)
        min_st[i,j] = min(stage)
        min_sp[i,j] = min(speed)
        #min_m[i,j] = min(mom)
        #max_m[i,j] = max(mom)

#find maximums for axis for plots
#stage_axis = ([5000./3600.,max(max_t), min(min(min_st)), max(max(max_st))])
#speed_axis = ([5000/3600.,max(max_t), min(min(min_sp)), max(max(max_sp))])
#mom_axis = ([5000/3600.,max(max_t), min(min(min_m)), max(max(max_m))])

ion()
hold(False)

for i in plot_gauges:
    plot(period,min_st[i,:],'+',period,max_st[i,:],'+',period,min_st[i,:],period,max_st[i,:])
    #plot(period,max_st[i,:],'+',period,max_st[i,:])
    xlabel('Period')
    ylabel('Stage (m)')
    axis(stage_axis)
    name = 'stage_period_%d' %(i+3)
    savefig(name)
    plot(period,min_sp[i,:],'+',period,max_sp[i,:],'+',period,min_sp[i,:],period,max_sp[i,:])
    #plot(period,max_sp[i,:],'+',period,max_sp[i,:])
    xlabel('Period')
    ylabel('Speed (m/s)')
    axis(speed_axis)
    name = 'speed_period_%d' %(i)
    savefig(name)

for i, dirname in enumerate(dirnames):
    plot(gauges,min_st[:,i],'+',gauges,max_st[:,i],'+',gauges,min_st[:,i],gauges,max_st[:,i])
    #plot(gauges,max_st[:,i],'+',gauges,max_st[:,i])
    xlabel('Gauge number')
    ylabel('Stage (m)')
    axis(stage_axis)
    title(dirname)
    j = dirname.rfind('_')
    name = 'stage_%s' %(dirname[j+1:])
    savefig(name)
    plot(gauges,min_sp[:,i],'+',gauges,max_sp[:,i],'+',gauges,min_sp[:,i],gauges,max_sp[:,i])
    #plot(gauges,max_sp[:,i],'+',gauges,max_sp[:,i])
    xlabel('Gauge number')
    ylabel('Speed (m/s)')
    axis(speed_axis)
    title(dirname)
    name = 'speed_%s' %(dirname[j+1:])
    savefig(name)

close('all')