Changeset 8241 for trunk/anuga_work/development/2010-projects/anuga_1d/channel/channel_domain.py

Timestamp:

Oct 27, 2011, 8:23:34 PM (14 years ago)

Author:

steve

Message:

Got the ctac figures working

File:

• trunk/anuga_work/development/2010-projects/anuga_1d/channel/channel_domain.py (modified) (7 diffs)

trunk/anuga_work/development/2010-projects/anuga_1d/channel/channel_domain.py

@@ -151 +151 @@
     def initialize_plotting(self,
                             stage_lim = [-1.0, 40.0],
-                            height_lim = [-1.0, 10.0],
+                            width_lim = [-1.0, 10.0],
                             velocity_lim = [-5.0, 5.0]):
 
@@ -159 +159 @@
 
         x = self.get_vertices().flatten()
+
         z = self.quantities['elevation'].vertex_values.flatten()
         w = self.quantities['stage'].vertex_values.flatten()
         h = self.quantities['height'].vertex_values.flatten()
         v = self.quantities['velocity'].vertex_values.flatten()
+        b = self.quantities['width'].vertex_values.flatten()
 
         print x.shape
         print z.shape
 
+        #-------------------------------
+        # Top plot
+        #-------------------------------
         self.plot1 = pylab.subplot(311)
 
@@ -173 +178 @@
 
         self.plot1.set_ylim(stage_lim)
-        pylab.xlabel('Position')
+        #pylab.xlabel('Position')
         pylab.ylabel('Stage')
 
+        #-------------------------------
+        # Middle Plot
+        #-------------------------------
         self.plot2 = pylab.subplot(312)
 
-        self.hplot, = pylab.plot(x, h)
-
-        self.plot2.set_ylim(height_lim)
-        pylab.xlabel('Position')
-        pylab.ylabel('Height')
-
+        self.bplot, = pylab.plot(x, b)
+
+        self.plot2.set_ylim(width_lim)
+        #pylab.xlabel('Position')
+        pylab.ylabel('Width')
+
+        #-------------------------------
+        # Bottom Plot
+        #-------------------------------
         self.plot3 = pylab.subplot(313)
 
@@ -203 +214 @@
         h = self.quantities['height'].vertex_values.flatten()
         v = self.quantities['velocity'].vertex_values.flatten()
+        b = self.quantities['width'].vertex_values.flatten()
 
 
         self.zplot.set_ydata(z)
         self.wplot.set_ydata(w)
-        self.hplot.set_ydata(h)
+        self.bplot.set_ydata(b)
         self.vplot.set_ydata(v)
 
@@ -213 +225 @@
 
 
-    def hold_plotting(self):
+    def hold_plotting(self,save=None):
 
         self.update_plotting()
@@ -219 +231 @@
         
         pylab.ioff()
+
+        if save != None:
+            file = save+".pdf"
+            pylab.savefig(file)
 
         pylab.show()
@@ -287 +303 @@
     w_C[:] = h_C + z_C
 
-    print w_C
+    #print w_C
 
     # Extrapolate velocity and stage as well as channel geometry.