Changeset 3960

Timestamp:

Nov 9, 2006, 4:47:53 PM (18 years ago)

Author:

jack

Message:

Converted about 1/2 the realtime vis to use accessor functions rather than accessing internals directly.

File:

• anuga_core/source/anuga/visualiser/realtime.py (modified) (3 diffs)

anuga_core/source/anuga/visualiser/realtime.py

@@ -45 +45 @@
     def setup_grid(self):
         self.vtk_cells = vtkCellArray()
-        triangles = self.source.triangles
-        N_tri = self.source.number_of_triangles
+        triangles = self.source.get_triangles()
+        N_tri = len(self.source)
         verticies = self.source.get_vertex_coordinates()
         N_vert = len(verticies)
@@ -58 +58 @@
 
     def update_height_quantity(self, quantityName, dynamic=True):
-        N_vert = len(self.source.vertex_coordinates)
+        N_vert = len(self.source.get_vertex_coordinates())
         qty_index = zeros(N_vert, Float)
+        triangles = self.source.get_triangles()
 
-        for n in range(len(self.source.triangles)):
+        for n in range(len(triangles)):
             for v in range(3):
-                qty_index[self.source.triangles[n][v]] = self.source.quantities[quantityName].vertex_values[n][v]
+                qty_index[triangles[n][v]] = self.source.get_quantity(quantityName).vertex_values[n][v]
 
         points = vtkPoints()
@@ -92 +93 @@
         
     def build_quantity_dict(self):
-        N_vert = len(self.source.vertex_coordinates)
+        triangles = self.source.get_triangles()
         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]
+            quantities[q], _ = self.source.get_quantity(q).get_vertex_values(xy=False)
         return quantities