]>
Commit | Line | Data |
---|---|---|
e395c057 RD |
1 | |
2 | from wxPython.wx import * | |
3 | try: | |
4 | from wxPython.lib import vtk | |
5 | haveVTK = true | |
6 | except ImportError: | |
7 | haveVTK = false | |
8 | ||
9 | #---------------------------------------------------------------------- | |
10 | ||
11 | def runTest(frame, nb, log): | |
12 | if haveVTK: | |
13 | win = vtk.wxVTKRenderWindow(nb, -1) | |
14 | ||
15 | # Get the render window | |
16 | renWin = win.GetRenderWindow() | |
17 | ||
18 | # Next, do the VTK stuff | |
19 | ren = vtk.vtkRenderer() | |
20 | renWin.AddRenderer(ren) | |
21 | cone = vtk.vtkConeSource() | |
22 | cone.SetResolution(80) | |
23 | coneMapper = vtk.vtkPolyDataMapper() | |
24 | coneMapper.SetInput(cone.GetOutput()) | |
25 | coneActor = vtk.vtkActor() | |
26 | coneActor.SetMapper(coneMapper) | |
27 | ren.AddActor(coneActor) | |
28 | coneMapper.GetLookupTable().Build() | |
29 | ||
30 | # Create a scalar bar | |
31 | scalarBar = vtk.vtkScalarBarActor() | |
32 | scalarBar.SetLookupTable(coneMapper.GetLookupTable()) | |
33 | scalarBar.SetTitle("Temperature") | |
34 | scalarBar.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport() | |
35 | scalarBar.GetPositionCoordinate().SetValue(0.1, 0.01) | |
36 | scalarBar.SetOrientationToHorizontal() | |
37 | scalarBar.SetWidth(0.8) | |
38 | scalarBar.SetHeight(0.17) | |
39 | ren.AddActor2D(scalarBar) | |
40 | ||
41 | return win | |
42 | ||
43 | ||
44 | else: | |
45 | wxMessageBox("Unable to import VTK, which is a required component " | |
46 | "of this demo. You need to download and install the " | |
47 | "Python extension module for VTK from http://www.kitware.com/", | |
48 | "Import Error") | |
49 | return None | |
50 | ||
51 | #---------------------------------------------------------------------- | |
52 | ||
53 | ||
54 | overview = """\ | |
55 | wxVTKRenderWindow is a wrapper around the vtkRenderWindow from the | |
56 | VTK Visualization Toolkit. The VTK Python extensions are required, | |
57 | they can be obtained from http://www.kitware.com/ where you can also | |
58 | find some nifty pictures and stuff. | |
59 | ||
60 | """ |