Changeset 3813

Timestamp:

Oct 17, 2006, 3:04:13 PM (18 years ago)

Author:

jack

Message:

Added support for overlaying polygons in the vtk visualisers. See the updated example.

Location:

anuga_core/source/anuga

Files:

• examples/sww_file_visualiser_example.py (modified) (1 diff)
• visualiser/visualiser.py (modified) (2 diffs)

anuga_core/source/anuga/examples/sww_file_visualiser_example.py

@@ -31 +31 @@
 o.render_axes()
 
+# Draw a polygon (here, a triangle) at height 10
+o.overlay_polygon([(20, 50), (40, 40), (50, 10), (30, 20), (10, 30)], 10, colour=(1.0, 1.0, 0.0))
+
 # Precaching the height-based quantities reduces the time taken to draw each
 # frame, but increases the time taken when the visualiser starts.

anuga_core/source/anuga/visualiser/visualiser.py

@@ -2 +2 @@
 from Tkinter import Tk, Button, Frame, N, E, S, W
 from types import FunctionType, TupleType
-from vtk import vtkActor, vtkCubeAxesActor2D, vtkFloatArray, vtkPolyDataMapper, vtkRenderer
+from vtk import vtkActor, vtkCubeAxesActor2D, vtkDelaunay2D, vtkFloatArray, vtkPoints, vtkPolyData, vtkPolyDataMapper, vtkRenderer
 from vtk.tk.vtkTkRenderWidget import vtkTkRenderWidget
 
@@ -162 +162 @@
         """
         self.colours_height[quantityName] = colour
-            
+
+    # --- Overlaid Polygons --- #
+
+    def overlay_polygon(self, coords, height=0.0, colour=(1.0, 0.0, 0.0)):
+        """Add a polygon to the output of the visualiser.
+
+        coords is a list of 2-tuples representing x and y coordinates.
+        These are triangulated by vtkDelaunay2D.
+
+        height is the z-value given to all points.
+
+        colour is the colour of the polygon, as a 3-tuple representing
+        r, g, b values between 0 and 1."""
+        points = vtkPoints()
+        for coord in coords:
+            points.InsertNextPoint(coord[0], coord[1], height)
+        profile = vtkPolyData()
+        profile.SetPoints(points)
+        delny = vtkDelaunay2D()
+        delny.SetInput(profile)
+        mesh = vtkPolyDataMapper()
+        mesh.SetInput(delny.GetOutput())
+        actor = vtkActor()
+        actor.SetMapper(mesh)
+        actor.GetProperty().SetColor(colour)
+        self.vtk_renderer.AddActor(actor)
+        
     # --- Vector Fields --- #