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
|
#!/usr/bin/env python
import os
from vtkmodules.vtkFiltersCore import vtkContourFilter
from vtkmodules.vtkFiltersSources import vtkSphereSource
from vtkmodules.vtkIOLegacy import (
vtkDataSetReader,
vtkDataSetWriter,
)
from vtkmodules.vtkImagingHybrid import vtkVoxelModeller
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()
# Now create the RenderWindow, Renderer and Interactor
#
ren1 = vtkRenderer()
renWin = vtkRenderWindow()
renWin.AddRenderer(ren1)
iren = vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
sphereModel = vtkSphereSource()
sphereModel.SetThetaResolution(10)
sphereModel.SetPhiResolution(10)
voxelModel = vtkVoxelModeller()
voxelModel.SetInputConnection(sphereModel.GetOutputPort())
voxelModel.SetSampleDimensions(21, 21, 21)
voxelModel.SetModelBounds(-1.5, 1.5, -1.5, 1.5, -1.5, 1.5)
voxelModel.SetScalarTypeToBit()
voxelModel.SetForegroundValue(1)
voxelModel.SetBackgroundValue(0)
#
# If the current directory is writable, then test the writers
#
try:
channel = open("voxelModel.vtk", "wb")
channel.close()
aWriter = vtkDataSetWriter()
aWriter.SetFileName("voxelModel.vtk")
aWriter.SetInputConnection(voxelModel.GetOutputPort())
aWriter.Update()
aReader = vtkDataSetReader()
aReader.SetFileName("voxelModel.vtk")
voxelSurface = vtkContourFilter()
voxelSurface.SetInputConnection(aReader.GetOutputPort())
voxelSurface.SetValue(0, .999)
voxelMapper = vtkPolyDataMapper()
voxelMapper.SetInputConnection(voxelSurface.GetOutputPort())
voxelActor = vtkActor()
voxelActor.SetMapper(voxelMapper)
sphereMapper = vtkPolyDataMapper()
sphereMapper.SetInputConnection(sphereModel.GetOutputPort())
sphereActor = vtkActor()
sphereActor.SetMapper(sphereMapper)
ren1.AddActor(sphereActor)
ren1.AddActor(voxelActor)
ren1.SetBackground(.1, .2, .4)
renWin.SetSize(256, 256)
ren1.ResetCamera()
ren1.GetActiveCamera().SetViewUp(0, -1, 0)
ren1.GetActiveCamera().Azimuth(180)
ren1.GetActiveCamera().Dolly(1.75)
ren1.ResetCameraClippingRange()
# cleanup
#
try:
os.remove("voxelModel.vtk")
except OSError:
pass
iren.Initialize()
# render the image
# iren.Start()
except IOError:
print("Unable to test the writer/reader.")
|
