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
117
118
119
120
121
|
#!/usr/bin/env python
from vtkmodules.vtkCommonDataModel import (
vtkImageData,
vtkSphere,
)
from vtkmodules.vtkCommonSystem import vtkTimerLog
from vtkmodules.vtkFiltersCore import vtkSimpleElevationFilter
from vtkmodules.vtkFiltersExtraction import vtkExtractGeometry
from vtkmodules.vtkFiltersGeneral import vtkDataSetTriangleFilter
from vtkmodules.vtkFiltersGeometry import (
vtkDataSetSurfaceFilter,
vtkGeometryFilter,
)
from vtkmodules.vtkRenderingCore import (
vtkActor,
vtkPolyDataMapper,
vtkRenderWindow,
vtkRenderWindowInteractor,
vtkRenderer,
)
import vtkmodules.vtkInteractionStyle
import vtkmodules.vtkRenderingFreeType
import vtkmodules.vtkRenderingOpenGL2
from vtkmodules.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
# Test and compare vtkGeometryFilter verus
# vtkDataSetSurfaceFilter.
# Control test resolution
res = 300
res = 50
# Generate hexes or tets
genHexes = 0
# Create a synthetic source, then convert to unstructured grid
vol = vtkImageData()
vol.SetDimensions(res,res,res)
sphere = vtkSphere()
sphere.SetRadius(10000)
grid = vtkExtractGeometry()
grid.SetInputData(vol)
grid.SetImplicitFunction(sphere)
# Alternative way to create tetra
tetras = vtkDataSetTriangleFilter()
tetras.SetInputData(vol)
if genHexes:
grid.Update()
print("Processing {0} hexes".format(grid.GetOutput().GetNumberOfCells()))
else:
tetras.Update()
print("Processing {0} tets".format(tetras.GetOutput().GetNumberOfCells()))
# Create a scalar field
ele = vtkSimpleElevationFilter()
if genHexes == 1:
ele.SetInputConnection(grid.GetOutputPort())
else:
ele.SetInputConnection(tetras.GetOutputPort())
ele.Update()
# Extract the surface with vtkGeometryFilter and time it. Use
# the fast extraction mode.
geomF = vtkGeometryFilter()
geomF.SetInputConnection(ele.GetOutputPort())
geomF.FastModeOn()
geomF.MergingOn()
timer = vtkTimerLog()
timer.StartTimer()
geomF.Update()
timer.StopTimer()
time = timer.GetElapsedTime()
print("Geometry Filter: {0}".format(time))
print("\tNumber points: {0}".format(geomF.GetOutput().GetNumberOfPoints()))
print("\tNumber faces: {0}".format(geomF.GetOutput().GetNumberOfCells()))
print("")
# Extract the surface with vtkDataSetSurfaceFilter and time it
geomF2 = vtkDataSetSurfaceFilter()
geomF2.SetInputConnection(ele.GetOutputPort())
timer.StartTimer()
geomF2.Update()
timer.StopTimer()
time = timer.GetElapsedTime()
print("DataSet Surface Filter: {0}".format(time))
print("\tNumber points: {0}".format(geomF2.GetOutput().GetNumberOfPoints()))
print("\tNumber faces: {0}".format(geomF2.GetOutput().GetNumberOfCells()))
print("")
# Show the result
mapper = vtkPolyDataMapper()
mapper.SetInputConnection(geomF.GetOutputPort())
mapper.SetScalarRange(0,float(res-1))
actor = vtkActor()
actor.SetMapper(mapper)
# Define graphics objects
ren1 = vtkRenderer()
ren1.SetBackground(0,0,0)
ren1.AddActor(actor)
ren1.GetActiveCamera().SetPosition(1,0,0)
ren1.ResetCamera()
renWin = vtkRenderWindow()
renWin.SetMultiSamples(0)
renWin.AddRenderer(ren1)
iren = vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
iren.Initialize()
iren.Start()
# --- end of script --
|
