#!/usr/bin/env python
##########
# Demonstration of the VTK sww Visualiser
# Jack Kelly
# September 2006
##########

# Import the offline visualiser
from anuga.visualiser import OfflineVisualiser
from vtk import vtkCubeAxesActor2D

# The argument to OfflineVisualiser is the path to a sww file
o = OfflineVisualiser("../../anuga_viewer/tests/cylinders.sww")
#o = OfflineVisualiser("../../../../anuga_validation/analytical solutions/circular_second_order.sww")

# Specify the height-based-quantities to render.
# Remember to set dynamic=True for time-varying quantities
o.render_quantity_height("elevation", dynamic=False)
o.render_quantity_height("stage", dynamic=True)

# Colour the stage:
# Either with an RGB value as a 3-tuple of Floats,
#o.colour_height_quantity('stage', (0.0, 0.0, 0.8))
# Or with a function of the quantities at that point, such as the stage height:
# 0 and 10 are the minimum and maximum values of the stage.
o.colour_height_quantity('stage', (lambda q:q['stage'], 0, 10))
# Or with the magnitude of the momentum at that point:
# Needs the sqrt function from Numeric. Again, 0 and 10 define the colour range.
#o.colour_height_quantity('stage', (lambda q:sqrt((q['xmomentum'] ** 2) +
#                                                 (q['ymomentum'] ** 2)), 0, 10))

# Draw some axes on the visualiser so we can see how big the wave is
o.render_axes()

# Increase the number of labels on the axes
o.alter_axes(vtkCubeAxesActor2D.SetNumberOfLabels, (5,))

# Draw a yellow polygon 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.
o.precache_height_quantities()

# Start the visualiser (in its own thread).
o.start()

# Wait for the visualiser to terminate before shutting down.
o.join()