Changeset 5447 for anuga_work/development/Hinwood_2008/slope.py

Timestamp:

Jun 27, 2008, 3:10:16 PM (16 years ago)

Author:

duncan

Message:

Current Hinwood scenario

File:

• anuga_work/development/Hinwood_2008/slope.py (modified) (6 diffs)

anuga_work/development/Hinwood_2008/slope.py

@@ -8 +8 @@
 
 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
@@ -64 +62 @@
     return times, slope_locations, slopes
 
-def graph_slopes(slope_file):
+def graph_slopes(slope_file, break_xs=None, break_times=None):
+    from pylab import meshgrid, cm, contourf, contour, ion, plot, xlabel, \
+         ylabel, close, legend, savefig, title, figure ,colorbar, show , axis
     origin = 'lower'
     times, slope_locations, slopes = load_slopes(slope_file)
-    X, Y = meshgrid(times, slope_locations)
-    #print "X", times
-    #print "Y", slope_locations
+    #print "times", times
+    #print "slope_locations", slope_locations
     #print "slopes", slopes
+
+    # Can't seem to reshape this info once it is in the function
     CS = contourf(slope_locations, times, slopes, 10, # [-1, -0.1, 0, 0.1],
-                        #alpha=0.5,
-                        cmap=cm.bone,
-                        origin=origin)
+                  #alpha=0.5,
+                  cmap=cm.bone,
+                  origin=origin)
     
     #CS2 = contour(slope_locations, times, slopes, CS.levels[::2],
@@ -84 +85 @@
     xlabel('x location')
     ylabel('Time, seconds')
+    axis([5.0, 5.5, 30, 60])
 
+
+    if break_times is not None and break_xs is not None:
+        plot(break_xs, break_times, 'ro')
+    
     # Make a colorbar for the ContourSet returned by the contourf call.
     cbar = colorbar(CS)
-    cbar.ax.set_ylabel('verbosity coefficient')
+    cbar.ax.set_ylabel('slope')
     # Add the contour line levels to the colorbar
     #cbar.add_lines(CS2)
@@ -105 +111 @@
                                        outputdir_name=outputdir_name)
         end = id + ".csv"
-        slope_file = pro_instance.outputdir + "slope_depth_" + end
-        graph_slopes(slope_file) 
+        slope_file = pro_instance.outputdir + "d3slope_stage_" + end
+        graph_slopes(slope_file, run_data['break_xs'],
+                     run_data['break_times']) 
     
 def gauges_for_slope():
+    from scenarios import scenarios
     from scenarios import scenarios
     
     outputdir_tag = "_good_tri_area_0.01_A"
     scenarios = [scenarios[1]] # !!!!!!!!!!!!!!!!!!!!!!
-    dx = 0.001
+    dx = 0.005
     for run_data in scenarios:
         point_x = arange(run_data['start_slope_x'],
@@ -131 +139 @@
         interpolate_sww2csv(pro_instance.outputdir + basename +".sww",
                             points,
-                            pro_instance.outputdir + "slope_depth_" + end,
-                            pro_instance.outputdir + "slope_velocity_x_" + end,
-                            pro_instance.outputdir + "slope_velocity_y_" + end)
+                            pro_instance.outputdir + "eslope_depth_" + end,
+                            pro_instance.outputdir + "eslope_velocity_x_" + end,
+                            pro_instance.outputdir + "eslope_velocity_y_" + end,
+                            pro_instance.outputdir + "eslope_stage_" + end,
+                            time_thinning=1)
   
     
@@ -142 +152 @@
     """ 
     """
+    #gauges_for_slope()
     auto_graph_slopes()