source: development/stochastic_study/plot_spread.py @ 2967

"""Read in timeseries from N simulations and plot spread for each timestep at specified locations (ch 5,7,9)
"""

import sys, os
from caching import cache
import project

gauge_name = project.gauge_names[0]
from Numeric import zeros, Float, allclose


number_of_realisations = project.number_of_realisations

time = zeros(project.number_of_timesteps, Float)
data = zeros((project.number_of_timesteps, number_of_realisations), Float)

j = 0 # Count realisations
for filename in os.listdir('.'):
    if filename.startswith(project.basename) and filename.endswith('%s.txt' %gauge_name):
        if j < data.shape[1]:        
            print 'Reading filename %s (column %d)' %(filename, j)
            fid = open(filename)
            for i, line in enumerate(fid.readlines()):
                if i < data.shape[0]:
                    fields = line.strip().split()
                    time[i] = float(fields[0])
                    data[i,j] = float(fields[1])
        else:
            print 'Ignored %s (column %d)' %(filename, j)

        fid.close()
        j += 1    


# Plot
from pylab import ion, hold, plot, title, xlabel, ylabel, legend, savefig, show

ion()
hold(True)
hold(False)

#for i in range(project.number_of_timesteps):
#    print i, data[i,:]
#    plot(data[i,:], 'k.')
#    raw_input('Next')    

for j in range(number_of_realisations):
    print j, data[:,j]
    plot(data[:,j], 'k-')
    
    raw_input('Next')
#title('Gauge %s' %name)
#xlabel('time(s)')
#ylabel('stage (m)')    
#legend(('Observed', 'Modelled'), shadow=True, loc='upper left')
#savefig(name, dpi = 300)


show()



#from pylab import *
#plot(time, stage)
#show()