Changeset 2469

Timestamp:

Mar 3, 2006, 9:52:00 AM (18 years ago)

Author:

sexton

Message:

Updating time series to write output to file (filename reflecting gauge name/location)

File:

• production/sydney_2006/get_timeseries.py (modified) (12 diffs)

production/sydney_2006/get_timeseries.py

@@ -9 +9 @@
 from matplotlib.ticker import MultipleLocator, FormatStrFormatter
 
-swwfile = project.outputname2 + '.sww'
+swwfile = project.outputname + '.sww' 
+
+
 
 #Time interval to plot
@@ -38 +40 @@
 gauges, lines, locations = get_gauges_from_file(project.gauge_filename)
 
+print 'number of gauges for Benfield:   ', len(gauges)
+
 #Read model output
 quantities = ['stage', 'elevation', 'xmomentum', 'ymomentum']
@@ -63 +67 @@
     ymom = []
     bearings = []
+    depths = []
 
     max_depth = 0
@@ -79 +84 @@
         uh = f(t, point_id = k)[2]
         vh = f(t, point_id = k)[3]
-        #myloc = locations[k]
+        myloc = locations[k]
+        depth = w-z
 
         m = sqrt(uh*uh + vh*vh)   #Absolute momentum
         vel = sqrt(uh*uh + vh*vh) / (w-z + 1.e-30) #Absolute velocity
         angle = degrees(atan(vh/(uh+1.e-15)))
-        print 'angle', angle
         if (0 < angle < 90.0):
             if vh > 0:
@@ -106 +111 @@
         ymom.append(vh)
         bearings.append(bearing)
+        depths.append(depth)
 
         if w-z > max_depth:
@@ -113 +119 @@
         if vel > max_velocity:
             max_velocity = vel
-            print 'max speed', max_velocity
 
     #Plot only those gauges that have been inundated by more than a threshold
@@ -128 +133 @@
         plot(model_time, stages, '-b',
              model_time, elevations, '-k')
-    name = 'Gauge_%d: (%.1f, %.1f)' %(k, g[0], g[1])
-    #name = 'Gauge_%d: (%.1f, %.1f) Location: %s' %(k, g[0], g[1], myloc)
+    #name = 'Gauge_%d: (%.1f, %.1f)' %(k, g[0], g[1])
+    name = 'Gauge_%d: (%.1f, %.1f) Location: %s' %(k, g[0], g[1], myloc)
     title(name)
 
@@ -138 +143 @@
            shadow=True,
            loc='upper right')
-    savefig('Gauge_%d_stage' %k) # savefig('Gauge_%s_stage' %myloc)
-
-    raw_input('Next')
-
-    #X Momentum plot
-    ion()
-    hold(False)
-    plot(model_time, xmom, '-r')
-    title(name)
-
-    title('%s (x momentum)' %name)
-    xlabel('time [s]')
-    ylabel('uh [m^2/s]')    
-    savefig('Gauge_%d_xmomentum' %k)
-    #savefig('Gauge_%s_momentum' %myloc)
-    
-    raw_input('Next')
-
-    #y Momentum plot
-    ion()
-    hold(False)
-    plot(model_time, ymom, '-r')
-    title(name)
-
-    title('%s (y momentum)' %name)
-    xlabel('time [s]')
-    ylabel('vh [m^2/s]')    
-    savefig('Gauge_%d_ymomentum' %k)
-    #savefig('Gauge_%s_momentum' %myloc)
-    
-    raw_input('Next')
+    #savefig('Gauge_%d_stage' %k)
+    savefig('Gauge_%s_stage' %myloc)
+
+    # raw_input('Next')
     
     #Momentum plot
@@ -179 +157 @@
     xlabel('time [s]')
     ylabel('sqrt( uh^2 + vh^2 ) [m^2/s]')    
-    savefig('Gauge_%d_momentum' %k)
-    #savefig('Gauge_%s_momentum' %myloc)
-    
-    raw_input('Next')
+    #savefig('Gauge_%d_momentum' %k)
+    savefig('Gauge_%s_momentum' %myloc)
+    
+    # raw_input('Next')
 
     #Bearing plot
@@ -193 +171 @@
     text(maxT+tstep, 90, 'East')
     text(maxT+tstep, 270, 'West')
-    majorLocator = MultipleLocator(45)
+    #majorLocator = MultipleLocator(3600)
     #print 'major', majorLocator[1]
-    #ax.yaxis.set_major_locator(majorLocator) #'yticklabels', range(30,390,30))
+    #ax.xaxis.set_major_locator(majorLocator) #'yticklabels', range(30,390,30))
     # set(labels,color='g',rotation=45)
 
@@ -201 +179 @@
     xlabel('time [s]')
     ylabel(' atan(vh/uh) [degrees from North]')    
-    savefig('Gauge_%d_bearing' %k)
-    #savefig('Gauge_%s_bearing' %myloc)
-    
-    raw_input('Next')
+    #savefig('Gauge_%d_bearing' %k)
+    savefig('Gauge_%s_bearing' %myloc)
+    
+    # raw_input('Next')
 
     #Speed plot
@@ -215 +193 @@
     xlabel('time [s]')
     ylabel('sqrt( uh^2 + vh^2 ) / depth [m/s]')    
-    savefig('Gauge_%d_speed' %k)
-    #savefig('Gauge_%s_speed' %myloc)
-    
-    raw_input('Next')
-    
-
+    #savefig('Gauge_%d_speed' %k)
+    savefig('Gauge_%s_speed' %myloc)
+    
+    # raw_input('Next')
+
+    whichone = '_%s' %myloc
+    thisfile = project.gaugetimeseries+whichone+'.csv'
+    fid = open(thisfile, 'w')
+    s = 'Time, Depth, Momentum, Velocity \n'
+    fid.write(s)
+    for i_t, i_d, i_m, i_vel in zip(model_time, depths, momenta, velocity):
+        s = '%.2f, %.2f, %.2f, %.2f\n' %(i_t, i_d, i_m, i_vel)
+        fid.write(s)
+        
 show()
 
+