Changeset 5426

Timestamp:

Jun 25, 2008, 9:27:56 AM (15 years ago)

Author:

duncan

Message:

Current Hinwood scenario

Location:

anuga_work/development/Hinwood_2008

Files:

• plot.py (modified) (3 diffs)
• run_dam.py (modified) (1 diff)
• slope.py (modified) (2 diffs)

anuga_work/development/Hinwood_2008/plot.py

@@ -5 +5 @@
 from os import sep
 import project
+from time import localtime, strftime
 
 def plot_compare_csv(location_sim, file_sim, location_exp, file_exp,
@@ -83 +84 @@
     from scenarios import scenarios
     outputdir_tag = "_good_tri_area_0.01_A"
-    #scenarios = [scenarios[0]] # !!!!!!!!!!!!!!!!!!!!!! 
+    #scenarios = scenarios[1] # !!!!!!!!!!!!!!!!!!!!!! 
+    scenarios = [scenarios[5]] # !!!!!!!!!!!!!!!!!!!!!! 
     # Supported formats: emf, eps, pdf, png, ps, raw, rgba, svg, svgz.
     plot_type = ".pdf"
@@ -99 +101 @@
                                                     save_as_list,
                                                     run_data['gauge_x']):
+            time_date = strftime('plot date: %d/%m/%Y Time: %H:%M:%S',
+                                      localtime())
             plot_title = "Scenario: " + outputdir_name + "\n" + \
-                         "X Gauge (m):" + str(gauge)
+                         "X Gauge (m):" + str(gauge) + "    " + time_date
+                         
             print "Doing ", plot_title
             plot_compare_csv(location_sim=loc_sim,

anuga_work/development/Hinwood_2008/run_dam.py

@@ -250 +250 @@
     
     from scenarios import scenarios
-    
-    run_type = 5
-    for run_data in [scenarios[0]]:
+
+
+    #4 is 0.01 5 is 0.001
+    run_type = 4
+    for run_data in [scenarios[5]]:
         main( run_data['scenario_id'] + '_boundary.tsm'  , run_data,
               run_type = run_type,

anuga_work/development/Hinwood_2008/slope.py

@@ -6 +6 @@
 import project
 #from scipy import arange
-from Numeric import arange
+
+from Numeric import arange, array, zeros, Float
+from pylab import meshgrid, cm, contourf, contour, ion, plot, xlabel, \
+     ylabel, close, legend, savefig, title, figure ,colorbar, show #, cbar
 
 from anuga.fit_interpolate.interpolate import interpolate_sww2csv
+from anuga.shallow_water.data_manager import csv2dict
 
+def load_depths(slope_file):
+    #slope, _ = csv2dict(file_sim)
+    
+    # Read the depth file
+    dfid = open(slope_file)
+    lines = dfid.readlines()
+    dfid.close()
 
-#-------------------------------------------------------------
-if __name__ == "__main__":
-    """ 
-    """
+    title = lines.pop(0)
+    n_time = len(lines)
+    n_sensors = len(lines[0].split(','))-1  # -1 to remove time
+    dtimes = zeros(n_time, Float)  #Time
+    depths = zeros(n_time, Float)  #
+    sensors = zeros((n_time,n_sensors), Float)
+    depth_locations = title.split(',') #(',')
+    depth_locations.pop(0) # remove 'time'
+    # !!!! -3 assumes a 
+    depths = [float(j.split(':')[0]) for j in depth_locations]
+    
+    for i, line in enumerate(lines):
+        fields = line.split(',') #(',')
+        fields = [float(j) for j in fields]
+        dtimes[i] = fields[0]
+        sensors[i] = fields[1:] # 1: to remove time
+
+    #print "dtimes",dtimes
+    #print "depths", depths
+    #print "sensors", sensors
+    return dtimes, depths, sensors
+   
+def load_slopes(slope_file):
+    times, depths, sensors = load_depths(slope_file)
+    n_slope_locations = len(depths)-1
+    n_time = len(times)
+    slope_locations = zeros(n_slope_locations, Float)  #
+    slopes = zeros((n_time,n_slope_locations), Float)
+
+    # An array of the sensor spacing values
+    delta_depths = zeros(n_slope_locations, Float)
+    
+    for i in arange(n_slope_locations):
+        slope_locations[i] = (depths[i+1] + depths[i+1])/2.
+        delta_depths[i] = (depths[i+1] - depths[i])
+    
+    for j in arange(n_time):
+        for i in arange(n_slope_locations):
+            slopes[j,i] = (sensors[j,i+1] - sensors[j,i])/delta_depths[i]
+
+    return times, slope_locations, slopes
+
+def graph_slopes(slope_file):
+    origin = 'lower'
+    times, slope_locations, slopes = load_slopes(slope_file)
+    X, Y = meshgrid(times, slope_locations)
+    #print "X", times
+    #print "Y", slope_locations
+    #print "slopes", slopes
+    CS = contourf(slope_locations, times, slopes, 10, # [-1, -0.1, 0, 0.1],
+                        #alpha=0.5,
+                        cmap=cm.bone,
+                        origin=origin)
+    
+    #CS2 = contour(slope_locations, times, slopes, CS.levels[::2],
+       #           colors = 'r',
+        #          origin=origin,
+         #         hold='on')
+
+    title('slope')
+    xlabel('x location')
+    ylabel('Time, seconds')
+
+    # Make a colorbar for the ContourSet returned by the contourf call.
+    cbar = colorbar(CS)
+    cbar.ax.set_ylabel('verbosity coefficient')
+    # Add the contour line levels to the colorbar
+    #cbar.add_lines(CS2)
+
+    figure()
+    show()
+
+def auto_graph_slopes():
     from scenarios import scenarios
     
     outputdir_tag = "_good_tri_area_0.01_A"
-    scenarios = [scenarios[1]] # !!!!!!!!!!!!!!!!!!!!!! 
+    scenarios = [scenarios[1]] # !!!!!!!!!!!!!!!!!!!!!!
+    for run_data in scenarios:
+        id = run_data['scenario_id']
+        outputdir_name = id + outputdir_tag
+        pro_instance = project.Project(['data','flumes','Hinwood_2008'],
+                                       outputdir_name=outputdir_name)
+        end = id + ".csv"
+        slope_file = pro_instance.outputdir + "slope_depth_" + end
+        graph_slopes(slope_file) 
+    
+def gauges_for_slope():
+    from scenarios import scenarios
+    
+    outputdir_tag = "_good_tri_area_0.01_A"
+    scenarios = [scenarios[1]] # !!!!!!!!!!!!!!!!!!!!!!
+    dx = 0.001
     for run_data in scenarios:
         point_x = arange(run_data['start_slope_x'],
                         run_data['finish_slope_x'],
-                        0.01).tolist()
+                        dx).tolist()
         flume_y_middle = 0.5
         points = []
@@ -36 +131 @@
         interpolate_sww2csv(pro_instance.outputdir + basename +".sww",
                             points,
-                            pro_instance.outputdir + "depth_" + end,
-                            pro_instance.outputdir + "velocity_x_" + end,
-                            pro_instance.outputdir + "velocity_y_" + end)
+                            pro_instance.outputdir + "slope_depth_" + end,
+                            pro_instance.outputdir + "slope_velocity_x_" + end,
+                            pro_instance.outputdir + "slope_velocity_y_" + end)
   
-        
+    
+    
 
+#-------------------------------------------------------------
+if __name__ == "__main__":
+    """ 
+    """
+    auto_graph_slopes() 
+