Changeset 2484

Timestamp:

Mar 3, 2006, 3:01:35 PM (19 years ago)

Author:

ole

Message:

Added timeseries extraction

File:

• development/stochastic_study/run_model.py (modified) (4 diffs)

development/stochastic_study/run_model.py

@@ -31 +31 @@
 from pyvolution.data_manager import xya2pts
 from pyvolution.util import file_function
+from caching.caching import cache
 
 # Application specific imports
@@ -75 +76 @@
 
 # Get prefitted realisations
+
+finaltime = 22.5
+timestep = 0.05
+
+
+
 realisation = 0
 for filename in os.listdir('.'):
@@ -85 +92 @@
         # For each column (each realisation)
         for i in range(V.shape[1]):
-            name = project.basename + '_realisation_%d' %realisation
-            domain.set_name(name)                     #Output name
+            domain.set_name(project.basename)         #Output name
             domain.set_quantity('elevation', V[:,i])  #Assign bathymetry
+            domain.starttime = 0.0                    #Reset time
 
             #---------------------------------------------------
@@ -95 +102 @@
                   %(i, V.shape[1], filename)
             t0 = time.time()
-            for t in domain.evolve(yieldstep = 0.05, finaltime = 22.5):
+            for t in domain.evolve(yieldstep = timestep, finaltime = finaltime):
                 domain.write_time()
-                #domain.write_boundary_statistics(tags = 'wave')         
+                
         
             print 'Realisation %d took %.2f seconds'\
                   %(realisation, time.time()-t0)
 
-            realisation += 1
+
+
+
+            #---------------------------------------------------
+            # Now extract the 3 timeseries (Ch 5-7-9) and store them
+            # in three files for this realisation
+            gauges = [[4.521, 1.196],  [4.521, 1.696],  [4.521, 2.196]] 
+            gauge_names = ['ch5', 'ch7', 'ch9']
+
+            
+            f = file_function(domain.filename + '.sww',
+                              quantities='stage',
+                              interpolation_points=gauges,
+                              verbose = True)
+
+
+            simulation_name = domain.filename + '_realisation_%d' %realisation
+
+            for k, name in enumerate(gauge_names):
+                fid = open(simulation_name + '_' + name + '.txt', 'w')
+                for t in f.T:
+                    #For all precomputed timesteps
+                    val = f(t, point_id = k)[0]
+                    fid.write('%f %f\n' %(t, val))
+
+                fid.close()
+
+                    
+
+            realisation += 1