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
|
#!/usr/bin/env python
from vtkmodules.vtkCommonCore import (
vtkDoubleArray,
vtkMath,
)
from vtkmodules.vtkCommonSystem import vtkTimerLog
from vtkmodules.vtkFiltersGeneral import vtkWarpScalar
from vtkmodules.vtkFiltersSources import vtkPlaneSource
from vtkmodules.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
# Create a bunch of points and then transform them. Time the operations.
# Control resolution of test
res = 100
math = vtkMath()
# Create points and normals
plane = vtkPlaneSource()
plane.SetResolution(res,res)
plane.Update()
output = plane.GetOutput()
# Manually construct scalars
NPts = output.GetNumberOfPoints()
scalars = vtkDoubleArray()
scalars.SetNumberOfComponents(1)
scalars.SetNumberOfTuples(NPts)
for i in range(0,NPts):
scalars.SetTuple1(i, math.Random(0,10))
output.GetPointData().SetScalars(scalars)
# Output some statistics
print("Number of points: {0}".format(NPts))
# Time the warping
warpF = vtkWarpScalar()
warpF.SetInputData(output)
warpF.SetScaleFactor(2.5);
# For timing the various tests
timer = vtkTimerLog()
timer.StartTimer()
warpF.Update()
timer.StopTimer()
time = timer.GetElapsedTime()
print("Warp via scalar: {0}".format(time))
|
