1
2
3
4
5
6
7
8
9
10
11
12
13
14
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
59
60
61
62
63
64
65
66
67
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
|
#!/usr/bin/env python
import vtk
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
#
# All Plot3D vector functions
#
# Create the RenderWindow, Renderer and both Actors
#
renWin = vtk.vtkRenderWindow()
renWin.SetMultiSamples(0)
ren1 = vtk.vtkRenderer()
ren1.SetBackground(.8, .8, .2)
renWin.AddRenderer(ren1)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
vectorLabels = ["Velocity", "Vorticity", "Momentum", "Pressure_Gradient"]
vectorFunctions = ["200", "201", "202", "210"]
camera = vtk.vtkCamera()
light = vtk.vtkLight()
# All text actors will share the same text prop
textProp = vtk.vtkTextProperty()
textProp.SetFontSize(10)
textProp.SetFontFamilyToArial()
textProp.SetColor(.3, 1, 1)
i = 0
for vectorFunction in vectorFunctions:
exec("pl3d" + vectorFunction + " = vtk.vtkMultiBlockPLOT3DReader()")
eval("pl3d" + vectorFunction).SetXYZFileName(
VTK_DATA_ROOT + "/Data/bluntfinxyz.bin")
eval("pl3d" + vectorFunction).SetQFileName(
VTK_DATA_ROOT + "/Data/bluntfinq.bin")
eval("pl3d" + vectorFunction).SetVectorFunctionNumber(int(vectorFunction))
eval("pl3d" + vectorFunction).Update()
output = eval("pl3d" + vectorFunction).GetOutput().GetBlock(0)
exec("plane" + vectorFunction + " = vtk.vtkStructuredGridGeometryFilter()")
eval("plane" + vectorFunction).SetInputData(output)
eval("plane" + vectorFunction).SetExtent(25, 25, 0, 100, 0, 100)
exec("hog" + vectorFunction + " = vtk.vtkHedgeHog()")
eval("hog" + vectorFunction).SetInputConnection(
eval("plane" + vectorFunction).GetOutputPort())
maxnorm = output.GetPointData().GetVectors().GetMaxNorm()
eval("hog" + vectorFunction).SetScaleFactor(1.0 / maxnorm)
exec("mapper" + vectorFunction + " = vtk.vtkPolyDataMapper()")
eval("mapper" + vectorFunction).SetInputConnection(
eval("hog" + vectorFunction).GetOutputPort())
exec("actor" + vectorFunction + " = vtk.vtkActor()")
eval("actor" + vectorFunction).SetMapper(eval("mapper" + vectorFunction))
exec("ren" + vectorFunction + " = vtk.vtkRenderer()")
eval("ren" + vectorFunction).SetBackground(0.5, .5, .5)
eval("ren" + vectorFunction).SetActiveCamera(camera)
eval("ren" + vectorFunction).AddLight(light)
renWin.AddRenderer(eval("ren" + vectorFunction))
eval("ren" + vectorFunction).AddActor(eval("actor" + vectorFunction))
exec("textMapper" + vectorFunction + " = vtk.vtkTextMapper()")
eval("textMapper" + vectorFunction).SetInput(vectorLabels[i])
eval("textMapper" + vectorFunction).SetTextProperty(textProp)
exec("text" + vectorFunction + " = vtk.vtkActor2D()")
eval("text" + vectorFunction).SetMapper(eval("textMapper" + vectorFunction))
eval("text" + vectorFunction).SetPosition(2, 5)
eval("ren" + vectorFunction).AddActor2D(eval("text" + vectorFunction))
i += 1
#
# now layout renderers
column = 1
row = 1
deltaX = 1.0 / 2.0
deltaY = 1.0 / 2.0
for vectorFunction in vectorFunctions:
eval("ren" + vectorFunction).SetViewport(
(column - 1) * deltaX + (deltaX * .05),
(row - 1) * deltaY + (deltaY * .05),
column * deltaX - (deltaX * .05),
row * deltaY - (deltaY * .05))
column += 1
if (column > 2):
column = 1
row += 1
camera.SetViewUp(1, 0, 0)
camera.SetFocalPoint(0, 0, 0)
camera.SetPosition(.4, -.5, -.75)
ren200.ResetCamera()
camera.Dolly(1.25)
for vectorFunction in vectorFunctions:
eval("ren" + vectorFunction).ResetCameraClippingRange()
light.SetPosition(camera.GetPosition())
light.SetFocalPoint(camera.GetFocalPoint())
renWin.SetSize(350, 350)
renWin.Render()
iren.Initialize()
#iren.Start()
|
