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
|
#!/usr/bin/env python
from vtkmodules.vtkCommonCore import (
vtkMath,
vtkUnsignedCharArray,
)
from vtkmodules.vtkCommonTransforms import vtkTransform
from vtkmodules.vtkFiltersCore import (
vtkAppendPolyData,
vtkStripper,
)
from vtkmodules.vtkFiltersGeneral import vtkTransformFilter
from vtkmodules.vtkFiltersModeling import vtkLinearExtrusionFilter
from vtkmodules.vtkFiltersSources import vtkArrowSource
from vtkmodules.vtkRenderingCore import (
vtkActor,
vtkPolyDataMapper,
vtkRenderWindow,
vtkRenderWindowInteractor,
vtkRenderer,
)
from vtkmodules.vtkRenderingFreeType import vtkVectorText
import vtkmodules.vtkInteractionStyle
import vtkmodules.vtkRenderingFreeType
import vtkmodules.vtkRenderingOpenGL2
from vtkmodules.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
disk = vtkVectorText()
disk.SetText("o")
t = vtkTransform()
t.Translate(1.1,0,0)
tf = vtkTransformFilter()
tf.SetTransform(t)
tf.SetInputConnection(disk.GetOutputPort())
strips = vtkStripper()
strips.SetInputConnection(tf.GetOutputPort())
strips.Update()
app = vtkAppendPolyData()
app.AddInputData(disk.GetOutput())
app.AddInputData(strips.GetOutput())
app.Update()
model = app.GetOutput()
extrude = vtkLinearExtrusionFilter()
extrude.SetInputData(model)
# create random cell scalars for the model before extrusion.
rn = vtkMath()
rn.RandomSeed(1230)
cellColors = vtkUnsignedCharArray()
cellColors.SetNumberOfComponents(3)
cellColors.SetNumberOfTuples(model.GetNumberOfCells())
i = 0
while i < model.GetNumberOfCells():
cellColors.InsertComponent(i,0,rn.Random(100,255))
cellColors.InsertComponent(i,1,rn.Random(100,255))
cellColors.InsertComponent(i,2,rn.Random(100,255))
i = i + 1
model.GetCellData().SetScalars(cellColors)
# Lets test the arrow source instead of creating another test.
arrow1 = vtkArrowSource()
mapper1 = vtkPolyDataMapper()
mapper1.SetInputConnection(arrow1.GetOutputPort())
actor1 = vtkActor()
actor1.SetMapper(mapper1)
actor1.SetPosition(0,-0.2,1)
arrow2 = vtkArrowSource()
arrow2.SetShaftResolution(2)
arrow2.SetTipResolution(1)
mapper2 = vtkPolyDataMapper()
mapper2.SetInputConnection(arrow2.GetOutputPort())
actor2 = vtkActor()
actor2.SetMapper(mapper2)
actor2.SetPosition(1,-0.2,1)
# Create the RenderWindow, Renderer and both Actors
#
ren1 = vtkRenderer()
renWin = vtkRenderWindow()
renWin.AddRenderer(ren1)
iren = vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
mapper = vtkPolyDataMapper()
mapper.SetInputConnection(extrude.GetOutputPort())
actor = vtkActor()
actor.SetMapper(mapper)
# Add the actors to the renderer, set the background and size
#
ren1.AddActor(actor)
ren1.AddActor(actor1)
ren1.AddActor(actor2)
ren1.SetBackground(0.1,0.2,0.4)
renWin.SetSize(300,300)
# render the image
#
cam1 = ren1.GetActiveCamera()
cam1.Azimuth(20)
cam1.Elevation(40)
ren1.ResetCamera()
cam1.Zoom(1.5)
iren.Initialize()
# prevent the tk window from showing up then start the event loop
# --- end of script --
|
