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
|
#!/usr/bin/env python
# Test the PIO Reader using native binary PIO files
from vtkmodules.vtkCommonCore import vtkLookupTable
from vtkmodules.vtkFiltersHyperTree import vtkHyperTreeGridGeometry
from vtkmodules.vtkIOPIO import vtkPIOReader
from vtkmodules.vtkParallelCore import vtkMultiProcessController
from vtkmodules.vtkRenderingCore import (
vtkActor,
vtkDataSetMapper,
vtkRenderWindow,
vtkRenderWindowInteractor,
vtkRenderer,
)
from vtkmodules.vtkRenderingParallel import (
vtkCompositedSynchronizedRenderers,
vtkSynchronizedRenderWindows,
)
import vtkmodules.vtkInteractionStyle
import vtkmodules.vtkRenderingFreeType
import vtkmodules.vtkRenderingOpenGL2
from vtkmodules.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
controller = vtkMultiProcessController.GetGlobalController()
rank = controller.GetLocalProcessId()
pioreader = vtkPIOReader()
pioreader.SetFileName(VTK_DATA_ROOT + "/Data/PIO/simple.pio")
pioreader.UpdateInformation()
# confirm default arrays are enabled
default_arrays = ["tev", "pres", "rade", "cell_energy", "kemax",
"vel", "eng"]
selection = pioreader.GetCellDataArraySelection()
for name in default_arrays:
if not selection.ArrayExists(name) or selection.ArrayIsEnabled(name):
# all's well
pass
else:
raise RuntimeError("'%s' should have been enabled by default." % name)
pioreader.SetCurrentTimeStep(1)
pioreader.Update()
grid = pioreader.GetOutput()
block = grid.GetBlock(0)
piece = block.GetPieceAsDataObject(rank)
geometryFilter = vtkHyperTreeGridGeometry()
geometryFilter.SetInputData(piece)
geometryFilter.Update()
# ---------------------------------------------------------------------
# Rendering
# ---------------------------------------------------------------------
ren = vtkRenderer()
renWin = vtkRenderWindow()
renWin.AddRenderer(ren)
iren = vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
syncWindows = vtkSynchronizedRenderWindows()
syncWindows.SetRenderWindow(renWin)
syncWindows.SetParallelController(controller)
syncWindows.SetIdentifier(1)
syncRenderers = vtkCompositedSynchronizedRenderers()
syncRenderers.SetRenderer(ren);
syncRenderers.SetParallelController(controller);
lut = vtkLookupTable()
lut.SetHueRange(0.66, 0)
lut.SetSaturationRange(1.0, 0.25);
lut.SetTableRange(48.5, 50)
lut.Build()
mapper = vtkDataSetMapper()
mapper.SetLookupTable(lut)
mapper.SetColorModeToMapScalars()
mapper.SetScalarModeToUseCellFieldData()
mapper.SelectColorArray('cell_energy')
mapper.SetInputConnection(geometryFilter.GetOutputPort())
mapper.UseLookupTableScalarRangeOn()
mapper.SetScalarRange(48.5, 50)
actor = vtkActor()
actor.SetMapper(mapper)
ren.AddActor(actor)
renWin.SetSize(300,300)
ren.SetBackground(0.5,0.5,0.5)
ren.GetActiveCamera().SetPosition(0.0108652, -0.00586516, 0.0143301)
ren.GetActiveCamera().SetViewUp(0.707107, 0.707107, 0)
ren.GetActiveCamera().SetParallelScale(0.00433013)
ren.GetActiveCamera().SetFocalPoint(0.0025, 0.0025, 0.0025)
renWin.Render()
# prevent the tk window from showing up then start the event loop
pioreader.SetDefaultExecutivePrototype(None)
|
