Changeset 8649 for trunk/anuga_core/source/anuga/visualiser

Timestamp:

Nov 26, 2012, 8:57:18 PM (12 years ago)

Author:

steve

Message:

Testing visualising height quantity

Location:

trunk/anuga_core/source/anuga/visualiser

Files:

• commandline_viewer.py (modified) (1 diff)
• numerical_dam_break_dry.py (modified) (2 diffs)
• visualiser.py (modified) (1 diff)

trunk/anuga_core/source/anuga/visualiser/commandline_viewer.py

@@ -39 +39 @@
         # the stage height:
         # 0 and 10 are the minimum and maximum values of the stage.
-        o.colour_height_quantity('stage', (lambda q: q['stage'], 0, 10))
+        o.colour_height_quantity('stage', (lambda q: q['stage'], 1.0, 5.0))
         # Or with the magnitude of the momentum at that point:
         # Needs the sqrt function from numeric. Again, 0 and 10
         # define the colour range.
-        # o.colour_height_quantity('stage',
+        from numpy import sqrt
+        #o.colour_height_quantity('stage',
         #                          (lambda q:sqrt((q['xmomentum'] ** 2) +
         #                                         (q['ymomentum'] ** 2)),
         #                                          0, 10))
+        o.colour_height_quantity('stage',
+                                  (lambda q: q['xmomentum']/
+                                             (q['stage'] - q['elevation']),
+                                              0, 5))
+
+
+        # Draw some axes on the visualiser so we can see how big the wave is
+
 
         # Draw some axes on the visualiser so we can see how big the wave is

trunk/anuga_core/source/anuga/visualiser/numerical_dam_break_dry.py

@@ -93 +93 @@
     from anuga.visualiser import RealtimeVisualiser
     vis = RealtimeVisualiser(domain)
-    vis.render_quantity_height("stage", zScale =1.0, dynamic=True)
+    vis.render_quantity_height("height",dynamic=True)
+    #vis.render_quantity_height("stage", zScale =1.0, dynamic=True)
     vis.colour_height_quantity('stage', (0.0, 0.0, 1.0))
     vis.start()
@@ -110 +111 @@
         fileName = 'stage_%03d' % i + '.vtk'
         i = i+1
-        vis.store_height_quantity('stage',fileName)
-
+        vis.store_height_quantity('height',fileName)
 
 #test against know data

trunk/anuga_core/source/anuga/visualiser/visualiser.py

@@ -195 +195 @@
                     map(scalars.InsertNextValue, colour[0](self.build_quantity_dict()))
                     self.vtk_polyData[quantityName].GetPointData().SetScalars(scalars)
-                    #mapper.SetScalarRange(colour[1:])
+                    mapper.SetScalarRange(colour[1:])
                     mapper.Update()
                 else: