Changeset 3006

Timestamp:

May 29, 2006, 1:07:46 PM (19 years ago)

Author:

ole

Message:

Work on plot_spread.py

Location:

development/stochastic_study

Files:

• plot_spread.py (modified) (5 diffs)
• project.py (modified) (1 diff)
• read_realisations.py (added)

development/stochastic_study/plot_spread.py

@@ -7 +7 @@
 # Related major packages
 from Numeric import zeros, Float, allclose, arange
-from caching import cache
 
 # Application specific imports
 from utilities.numerical_tools import histogram, create_bins
+from read_realisations import read_realisations
 import project
 
 # Initialise
-gauge = '%s.txt' %project.gauge_names[0]
-number_of_realisations = project.number_of_realisations
-
-time = zeros(project.number_of_timesteps, Float)
-data = zeros((project.number_of_timesteps, number_of_realisations), Float)
-
-
 # Read in all realisations-timeseries
-
-j = 0 # Count realisations
-for filename in os.listdir(project.working_dir):
-    if filename.startswith(project.basename) and filename.endswith(gauge):
-        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    
+time, data, filenames = read_realisations('cyclone1', 
+                                          max_realisations = 200,
+                                          use_cache=True)
+#time, data, filenames = read_realisations('test', exclude='_0_',
+#                                          max_realisations = 200,
+#                                          use_cache=True)
+#time, data, filenames = read_realisations('nautilus1', exclude='_0_')
+number_of_realisations = data.shape[1]
 
 
@@ -45 +29 @@
 
 ion()
-#hold(True)
-hold(False)
+hold(True)
+#hold(False)
 
 # Simple plot of timeseries for different realisations 
-for j in range(number_of_realisations):
-    print j, data[:,j]
+for j in range(20):
+    print j, filenames[j]
     plot(time, data[:,j], 'k-')
     
     xlabel('time(s)')
     ylabel('stage (m)')
-    title('Realisation %d of %d' %(j, number_of_realisations-1))    #
-#raw_input('Next')
+    title('Timeseries for realisation %d of %d' %(j, number_of_realisations-1))
+    raw_input('Next')
+ 
 
 
 hold(False)
 # Plot spread of stage values for each timestep
-for i in range(200,project.number_of_timesteps):
-    # Plot histogram
+for i in range(300,320): #project.number_of_timesteps):
+    # Plot histogram#
 
     w = data[i,:]
@@ -68 +53 @@
     xlabel('realisations')
     ylabel('stage (m)')
-    title('Timestep %d of %d (t=%.2f)'\
+    title('Spread at timestep %d of %d (t=%.2f)'\
           %(i, project.number_of_timesteps-1, time[i]))        
     raw_input('Next')
@@ -75 +60 @@
 hold(False)
 # Plot histogram of stage values for each timestep
-for i in range(200,project.number_of_timesteps):
+for i in range(300,320):#project.number_of_timesteps):
     # Plot histogram
 
@@ -89 +74 @@
     xlabel('stage (m)')
     ylabel('frequency')
-    title('Timestep %d of %d (t=%.2f)'\
+    title('Histogram at timestep %d of %d (t=%.2f)'\
           %(i, project.number_of_timesteps-1, time[i]))        
-    #raw_input('Next')
+    raw_input('Next')
 
 

development/stochastic_study/project.py

@@ -34 +34 @@
 blocksize = 10 #How many realisations to fit at a time
 
-number_of_bins = 10
+number_of_bins = 100