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
|
#!/usr/bin/env python
"""
This tests a pipeline that consists of a chain of GPU resident operators.
The data, once on the GPU, should stay there until brought back with the
vtkPistonMapper or vtkPistonToDataSet algorithms.
"""
import sys
import vtk
from vtk.test import Testing
from PistonTestCommon import *
class TestSort(Testing.vtkTest):
def testSort(self):
global args
writefiles = "SaveData" in args
renderer = vtk.vtkRenderer()
renwin = vtk.vtkRenderWindow()
renwin.AddRenderer(renderer)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renwin)
renwin.Render()
src = vtk.vtkImageMandelbrotSource()
src.SetWholeExtent(0,10,0,10,0,10)
#scale and bias until piston understands origin and spacing
src.Update()
inputdata = src.GetOutput()
if "Normalize" in args:
testdata1 = inputdata.NewInstance()
testdata1.ShallowCopy(inputdata)
testdata1.SetSpacing(1,1,1)
testdata1.SetOrigin(0,0,0)
inputdata = testdata1
d2p = vtk.vtkDataSetToPiston()
d2p.SetInputData(inputdata)
#d2p.SetInputConnection(src.GetOutputPort())
sorterF = vtk.vtkPistonSort()
sorterF.SetInputConnection(d2p.GetOutputPort())
contourF = vtk.vtkPistonContour()
contourF.SetInputConnection(sorterF.GetOutputPort())
contourF.SetIsoValue(50.0)
p2d = vtk.vtkPistonToDataSet()
p2d.SetInputConnection(contourF.GetOutputPort())
p2d.SetOutputDataSetType(vtk.VTK_POLY_DATA)
p2d.Update()
if writefiles:
writeFile(p2d, "piston_sortcontour.vtk")
mapper = vtk.vtkDataSetMapper()
mapper.SetInputConnection(p2d.GetOutputPort())
mapper.Update()
actor = vtk.vtkActor()
actor.SetMapper(mapper)
renderer.AddActor(actor)
renderer.ResetCamera()
renwin.Render()
img_file = "TestSort.png"
Testing.compareImage(renwin, Testing.getAbsImagePath(img_file))
if Testing.isInteractive():
iren.Start()
if __name__ == "__main__":
global args
args = parseArgs()
Testing.main([(TestSort, 'test')])
|
