Changeset 2265

Timestamp:

Jan 23, 2006, 10:30:18 AM (17 years ago)

Author:

jack

Message:

Updated visualiser to handle quitting better.
Replaced vert_index and qty_index dictionaries with numeric arrays.

File:

• inundation/pyvolution/vtk_realtime_visualiser.py (modified) (4 diffs)

inundation/pyvolution/vtk_realtime_visualiser.py

@@ -5 +5 @@
 import Tkinter
 import vtk
+from Numeric import *
 from vtk.tk.vtkTkRenderWidget import vtkTkRenderWidget
 
@@ -116 +117 @@
 
         self.N_tri = len(self.domain.triangles)
+        self.N_vert = len(self.vertices)
         self.cells = vtk.vtkCellArray()
         self.vertices = self.domain.get_vertex_coordinates()
-        self.vert_index = {}
-
-        for n in range(len(self.vertices)):
+        self.vert_index = zeros((self.N_vert,2), Float)
+        for n in range(self.N_vert):
             for i in range(3):
                 self.vert_index[self.domain.triangles[n][i]] = self.vertices[n][i*2:i*2+2]
-
-        self.N_vert = len(self.vert_index.keys())
 
         # Prepare the list of cells
@@ -138 +137 @@
 
     def draw_quantity(self, q):
-        qty_index = {}
+        qty_index = zeros(self.N_vert, Float)
         for n in range(self.N_tri):
             for v in range(3):
@@ -201 +200 @@
     def shutdown(self):
         self.domain.visualise = False
+        self.idle.set()
         self.root.withdraw()
         self.root.quit()