Newer
Older
#!/usr/bin/env python
# Translated from hawaii.tcl
import vtk
def main():
file_name, color_scheme = get_program_parameters()
colors = vtk.vtkNamedColors()
# Set the background color. Match those in VTKTextbook.pdf.
bkg = map(lambda x: x / 256.0, [25, 51, 102])
colors.SetColor("BkgColor", *bkg)
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# Read a vtk file
#
hawaii = vtk.vtkPolyDataReader()
hawaii.SetFileName(file_name)
hawaii.Update()
bounds = [0.0] * 6
hawaii.GetOutput().GetBounds(bounds)
elevation = vtk.vtkElevationFilter()
elevation.SetInputConnection(hawaii.GetOutputPort())
elevation.SetLowPoint(0, 0, 0)
elevation.SetHighPoint(0, 0, 1000)
elevation.SetScalarRange(0, 1000)
lut = MakeLUT(color_scheme)
hawaiiMapper = vtk.vtkDataSetMapper()
hawaiiMapper.SetInputConnection(elevation.GetOutputPort())
hawaiiMapper.SetScalarRange(0, 1000)
hawaiiMapper.ScalarVisibilityOn()
hawaiiMapper.SetLookupTable(lut)
hawaiiActor = vtk.vtkActor()
hawaiiActor.SetMapper(hawaiiMapper)
# Create the RenderWindow, Renderer and both Actors
#
ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
# Add the actors to the renderer, set the background and size
#
ren.AddActor(hawaiiActor)
# Match the window shape to the object.
# renWin.SetSize(500, int(500 * bounds[1] / bounds[3]))
renWin.SetSize(500, 500)
iren.Initialize()
# Render the image.
# Diamond Head is the crater lower left.
# Punchbowl is the crater in the centre.
renWin.Render()
ren.SetBackground(colors.GetColor3d("BkgColor"))
ren.GetActiveCamera().Zoom(1.5)
ren.GetActiveCamera().Roll(-90)
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
iren.Start()
def get_program_parameters():
import argparse
description = 'Produce figure 6–12 from the VTK Textbook.'
epilogue = '''
Produce figure 6–12 from the VTK Textbook.
It is a translation of the original hawaii.tcl with a few additional enhancements.
The image is centered on Honolulu, O'ahu.
Diamond Head is the crater lower left. Punchbowl is the crater in the centre.
The color_scheme option allows you to select a series of colour schemes.
0: The default, a lookup using a "Brewer" palette.
1: The original: A lookup table of 256 colours ranging from deep blue (water) to yellow-white (mountain top).
2: A lookup table with a preset number of colours.
'''
parser = argparse.ArgumentParser(description=description, epilog=epilogue)
parser.add_argument('filename', help='honolulu.vtk.')
parser.add_argument('color_scheme', default=0, type=int, nargs='?', help='The particular color scheme to use.')
args = parser.parse_args()
return args.filename, args.color_scheme
def MakeLUT(color_scheme=0):
"""
Make a lookup table.
:param color_scheme: Select the type of lookup table.
:return: The lookup table.
"""
colors = vtk.vtkNamedColors()
if color_scheme == 1:
# A lookup table of 256 colours ranging from
# deep blue (water) to yellow-white (mountain top)
# is used to color map this figure.
lut = vtk.vtkLookupTable()
lut.SetHueRange(0.7, 0)
lut.SetSaturationRange(1.0, 0)
lut.SetValueRange(0.5, 1.0)
elif color_scheme == 2:
# Make the lookup table with a preset number of colours.
colorSeries = vtk.vtkColorSeries()
colorSeries.SetNumberOfColors(8)
colorSeries.SetColorSchemeName('Hawaii')
colorSeries.SetColor(0, colors.GetColor3ub("turquoise_blue"))
colorSeries.SetColor(1, colors.GetColor3ub("sea_green_medium"))
colorSeries.SetColor(2, colors.GetColor3ub("sap_green"))
colorSeries.SetColor(3, colors.GetColor3ub("green_dark"))
colorSeries.SetColor(4, colors.GetColor3ub("tan"))
colorSeries.SetColor(5, colors.GetColor3ub("beige"))
colorSeries.SetColor(6, colors.GetColor3ub("light_beige"))
colorSeries.SetColor(7, colors.GetColor3ub("bisque"))
lut = vtk.vtkLookupTable()
colorSeries.BuildLookupTable(lut, colorSeries.ORDINAL)
lut.SetNanColor(1, 0, 0, 1)
else:
# Make the lookup using a Brewer palette.
colorSeries = vtk.vtkColorSeries()
colorSeries.SetNumberOfColors(8)
colorSeriesEnum = colorSeries.BREWER_DIVERGING_BROWN_BLUE_GREEN_8
colorSeries.SetColorScheme(colorSeriesEnum)
lut = vtk.vtkLookupTable()
colorSeries.BuildLookupTable(lut, colorSeries.ORDINAL)
lut.SetNanColor(1, 0, 0, 1)
return lut
if __name__ == '__main__':
main()