Changeset 3625

Timestamp:

Sep 19, 2006, 4:55:48 PM (18 years ago)

Author:

jack

Message:

Updatd the offline visualiser example and prepared the realtime visualiser.

Location:

anuga_core/source/anuga

Files:

• examples/sww_file_visualiser_example.py (moved) (moved from anuga_core/source/anuga/examples/run_offline.py)
• visualiser/realtime.py (modified) (3 diffs)
• visualiser/visualiser.py (modified) (3 diffs)

anuga_core/source/anuga/visualiser/realtime.py

@@ +1 @@
+from Numeric import Float, zeros
 from Tkinter import Button, E, W
 from threading import Event
 from visualiser import Visualiser
+from vtk import vtkCellArray, vtkPoints, vtkPolyData
 
 class RealtimeVisualiser(Visualiser):
@@ -22 +24 @@
 
     def run(self):
+        self.tk_root.after(100, self.sync_idle.set)
         Visualiser.run(self)
-        self.tk_root.after(100, self.sync_idle.set)
+
+    def setup_grid(self):
+        self.vtk_cells = vtkCellArray()
+        # Also build vert_index - a list of the x & y values of each vertex
+        N_vert = len(self.source.vertex_coordinates)
+        self.vert_index = zeros((N_vert,2), Float)
+        for n in range(N_vert):
+            self.vtk_cells.InsertNextCell(3)
+            for v in range(3):
+                self.vert_index[self.source.triangles[n][v]] = self.source.vertex_coordinates[n][i*2:i*2+2]
+                self.vtk_cells.InsertCellPoint(self.source.triangles[n][v])
+
+    def update_height_quantity(self, quantityName, dynamic=True):
+        N_vert = len(self.source.vertex_coordinates)
+        qty_index = zeros(N_vert, Float)
+
+        for n in range(len(self.source.triangles)):
+            for v in range(3):
+                qty_index[self.source.triangles[n][v]] = self.source.quantities[quantityName].vertex_values[n][v]
+
+        points = vtkPoints()
+        for v in range(N_vert):
+            points.InsertNextPoint(self.vert_index[v][0],
+                                   self.vert_index[v][1],
+                                   qty_index[v] * self.height_zScales[quantityName]
+                                   + self.height_offset[quantityName])
+
+        polydata = self.vtk_polyData[quantityName] = vtkPolyData()
+        polydata.SetPoints(points)
+        polydata.SetPolys(self.vtk_cells)
+        
+    def build_quantity_dict(self):
+        N_vert = len(self.source.vertex_coordinates)
+        quantities = {}
+        for q in self.source.quantities.keys():
+            quantities[q] = zeros(N_vert, Float)
+            for n in range(len(self.source.triangles)):
+                for v in range(3):
+                    quantities[q][self.source.triangles[n][v]] = self.source.quantities[q].vertex_values[n][v]
+        return quantities
 
     def setup_gui(self):
@@ -43 +85 @@
         self.sync_idle.set()
         self.sync_unpaused.set()
+

anuga_core/source/anuga/visualiser/visualiser.py

@@ -51 +51 @@
         """Create the vtkCellArray instance that represents the
         triangles. Subclasses are expected to override this function
-        to read from their source as appropriate.
+        to read from their source as appropriate. The vtkCellArray should
+        be stored to self.vtk_cells.
         """
         pass
@@ -69 +70 @@
     def update_height_quantity(self, quantityName, dynamic=True):
         """Create a vtkPolyData object and store it in
-        self.vtk_polyData[q]. Subclasses are expected to override this
+        self.vtk_polyData[quantityName]. Subclasses are expected to override this
         function.
         """
@@ -115 +116 @@
 
     def build_quantity_dict(self):
-        """Build a dictionary mapping quantity name->Numeric array of vertex
+        """Build and return a dictionary mapping quantity name->Numeric array of vertex
         values for that quantity. Subclasses are expected to override
         this function."""