Changeset 3493

Timestamp:

Aug 15, 2006, 4:45:28 PM (17 years ago)

Author:

jack

Message:

Offline visualiser is working fairly well.

Location:

inundation/visualiser

Files:

• offline.py (added)
• realtime.py (modified) (2 diffs)
• run_offline.py (added)
• visualiser.py (modified) (5 diffs)

inundation/visualiser/realtime.py

@@ -5 +5 @@
     """A VTK-powered realtime visualiser which runs in its own thread.
     """
-    def __init__(self, domain):
-        Visualiser.__init__(self,domain)
+    def __init__(self, source):
+        """The source parameter is assumed to be a Domain.
+        """
+        Visualiser.__init__(self, source)
+
+        self.running = True
+
+        # Synchronisation Constructs
+        self.sync_idle = Event()
+        self.sync_idle.clear()
+        self.sync_unpaused = Event()
+        self.sync_unpaused.set()
+        self.sync_redrawReady = Event()
+        self.sync_redrawReady.clear()
+
+    def run(self):
+        Visualiser.run(self)
+        self.tk_root.after(100, self.sync_idle.set)
 
     def setup_gui(self):
@@ -20 +36 @@
             self.sync_unpaused.set()
             self.tk_pauseResume.config(text="Pause")
+
+    def shutdown(self):
+        Visualiser.shutdown(self)
+        self.running = False
+        self.sync_idle.set()
+        self.sync_unpaused.set()

inundation/visualiser/visualiser.py

@@ -1 +1 @@
 from threading import Event, Thread
-from Tkinter import Tk
-from vtk import vtkRenderer
+from Tkinter import Tk, Button, N, E, S, W
+from vtk import vtkActor, vtkPolyDataMapper, vtkRenderer
 from vtk.tk.vtkTkRenderWidget import vtkTkRenderWidget
 
@@ -7 +7 @@
     """Superclass of both the realtime and offline VTK visualisers
     """
-    def __init__(self, domain):
+    def __init__(self, source):
         Thread.__init__(self)
 
-        self.domain = domain
-        self.running = True
-
-        # Synchronisation Constructs
-        self.sync_idle = Event()
-        self.sync_idle.clear()
-        self.sync_unpaused = Event()
-        self.sync_unpaused.set()
-        self.sync_redrawReady = Event()
-        self.sync_redrawReady.clear()
+        self.source = source
 
         # Structures for Height Based quantities
@@ -25 +16 @@
         self.height_zScales = {}
         self.height_dynamic = {}
+        self.height_offset = {}
+
+        # Structures used for VTK
+        self.vtk_actors = {}
+        self.vtk_mappers = {}
+        self.vtk_polyData = {}
 
     def run(self):
         self.setup_gui()
-        self.tk_root.after(100, self.sync_idle.set)
+        self.setup_grid()
+        # Draw Height Quantities
+        for q in self.height_quantities:
+            self.update_height_quantity(q, self.height_dynamic[q])
+            self.draw_height_quantity(q)
         self.tk_root.mainloop()
+
+    def redraw_quantities(self, dynamic_only=False):
+        """Redraw all dynamic quantities, unless dynamic_only is True.
+        """
+        # Height quantities
+        for q in self.height_quantities:
+            if (dynamic_only is False) or (self.height_dynamic[q]):
+                self.update_height_quantity(q, self.height_dynamic[q])
+                self.draw_height_quantity(q)
 
     # --- Height Based Rendering --- #
 
-    def render_quantity_height(self, quantityName, zScale=1.0, dynamic=True):
-        """Instruct the visualiser to render a quantity
-        using the value at a point as its height.
-        The value at each point is multiplied by z_scale,
-        and if dynamic=False, the quantity is not recalculated
-        on each update.
+    def setup_grid(self):
+        """Create the vtkCellArray instance that represents the
+        triangles. Subclasses are expected to override this function
+        to read from their source as appropriate.
         """
-        if quantityName in self.domain.quantities:
-            self.height_quantities.append(quantityName)
-            self.height_zScales[quantityName] = zScale
-            self.height_dynamic[quantityName] = dynamic
+        pass
+
+    def render_quantity_height(self, quantityName, zScale=1.0, offset=0.0, dynamic=True):
+        """Instruct the visualiser to render a quantity using the
+        value at a point as its height.  The value at each point is
+        multiplied by z_scale and is added to offset, and if
+        dynamic=False, the quantity is not recalculated on each
+        update.
+        """
+        self.height_quantities.append(quantityName)
+        self.height_zScales[quantityName] = zScale
+        self.height_offset[quantityName] = offset
+        self.height_dynamic[quantityName] = dynamic
+
+    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
+        function.
+        """
+        pass
+
+
+    def draw_height_quantity(self, quantityName):
+        """Use the vtkPolyData and prepare/update the rest of the VTK
+        rendering pipeline.
+        """
+        if self.vtk_mappers.has_key(quantityName):
+            mapper = self.vtk_mappers[quantityName]
+        else:
+            mapper = self.vtk_mappers[quantityName] = vtkPolyDataMapper()
+        mapper.SetInput(self.vtk_polyData[quantityName])
+        mapper.Update()
+
+        if not self.vtk_actors.has_key(quantityName):
+            actor = self.vtk_actors[quantityName] = vtkActor()
+            actor.SetMapper(mapper)
+            actor.GetProperty().SetColor(0.5, 0.5, 0.5)
+            self.vtk_renderer.AddActor(actor)
 
     # --- Colour Coding --- #
@@ -58 +100 @@
 
         self.tk_renderWidget = vtkTkRenderWidget(self.tk_root, width=400, height=400)
-        self.tk_renderWidget.grid(row=0, column=0)
+        self.tk_renderWidget.grid(row=0, column=0, sticky=N+S+E+W)
+        self.tk_quit = Button(self.tk_root, text="Quit", command=self.shutdown)
+        self.tk_quit.grid(row=2, column=0, sticky=E+W)
         self.vtk_renderer = vtkRenderer()
         self.tk_renderWidget.GetRenderWindow().AddRenderer(self.vtk_renderer)
@@ -69 +113 @@
 
     def redraw(self):
-        pass
+        self.tk_renderWidget.GetRenderWindow().Render()
+        self.tk_root.update_idletasks()
+        self.tk_root.after(100, self.redraw)
 
     def shutdown(self):
-        self.running = False
-        self.sync_idle.set()
-        self.sync_unpaused.set()
         self.tk_root.withdraw()
         self.tk_root.quit()