Changeset 5607 for anuga_work/production/perth/get_gauges.py

Timestamp:

Aug 5, 2008, 1:17:37 PM (16 years ago)

Author:

kristy

Message:

 

File:

• anuga_work/production/perth/get_gauges.py (modified) (4 diffs)

anuga_work/production/perth/get_gauges.py

@@ -10 +10 @@
 from Scientific.IO.NetCDF import NetCDFFile
 
-def get_sts_gauge_data(filename,polygon,verbose=False):
+def get_sts_gauge_data(filename,verbose=False):
     from Numeric import asarray,transpose,sqrt,argmax,argmin,arange,Float,\
         compress,zeros,fabs,take
@@ -17 +17 @@
     y = fid.variables['y'][:]+fid.yllcorner   #y-coordinates of vertices
     points=transpose(asarray([x.tolist(),y.tolist()]))
-    time=fid.variables['time'][:]
+    time=fid.variables['time'][:]+fid.starttime
     elevation=fid.variables['elevation'][:]
-    indices=inside_polygon(points, polygon, closed=True, verbose=False)
-    inside_points=take(points,indices,0)
-    elevation=take(elevation,indices,0)
-    if verbose:
-        inside_points_array=ensure_numeric(inside_points)#for plotting only
-        #plot(points[:,0],points[:,1],'o')
-        #plot(inside_points_array[:,0][::5],inside_points_array[:,1][::5],'o')
-        polygon=ensure_numeric(polygon)
-        #plot(polygon[:,0],polygon[:,1])
-        #show()
-
-    
+       
     basename='sts_gauge'
     quantity_names=['stage','xmomentum','ymomentum']
@@ -36 +25 @@
     for i, name in enumerate(quantity_names):
         quantities[name] = fid.variables[name][:]
-##        if inside_points is not None:
-##            quantities[name] = take(quantities[name],indices,1)
-##        else:
-##            msg = 'No gauges found inside polygon'
-##            raise msg
-##            
-    for j in range(len(x)-1):
+
+    maxname = 'max_sts_stage.csv'
+    fid_max = open(project.boundaries_dir+sep+maxname,'w')
+    s = 'x, y, max_stage \n'
+    fid_max.write(s)    
+    for j in range(len(x)):
         locx=int(x[j])
         locy=int(y[j])
+        stage = quantities['stage'][:,j]
+        xmomentum = quantities['xmomentum'][:,j]
+        ymomentum = quantities['ymomentum'][:,j]
+
+        s = '%.6f, %.6f, %.6f\n' %(x[j], y[j], max(stage))
+        fid_max.write(s)
+        
         fid_sts = open(project.boundaries_dir+sep+basename+'_'+str(locx)+'_'+str(locy)+'.csv', 'w')
         s = 'time, stage, xmomentum, ymomentum \n'
         fid_sts.write(s)
-        stage = quantities['stage'][:,j]
-        xmomentum = quantities['xmomentum'][:,j]
-        ymomentum = quantities['ymomentum'][:,j]
+        
         for k in range(len(time)-1):
-            s = '%.2f, %.2f, %.2f, %.2f\n' %(time[k], stage[k], xmomentum[k], ymomentum[k])
+            s = '%.6f, %.6f, %.6f, %.6f\n' %(time[k], stage[k], xmomentum[k], ymomentum[k])
             fid_sts.write(s)
 
@@ -59 +52 @@
     fid.close()
         
-    return inside_points,quantities,elevation,time
+    return quantities,elevation,time
 
-polygon=project.poly_all
-points,quantities,elevation,time=get_sts_gauge_data(os.path.join(project.boundaries_dir,project.scenario_name),polygon,verbose=False)
-print quantities['stage'][0,:]
-print len(points), len(elevation), len(quantities['stage'][0,:])
-assert len(points)==len(elevation)==len(quantities['stage'][0,:])
+quantities,elevation,time=get_sts_gauge_data(os.path.join(project.boundaries_dir,project.scenario_name),verbose=False)
+
+print len(elevation), len(quantities['stage'][0,:])