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
|
#!/usr/bin/env python
import vtk
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
arrayType = ['Bit', 'Char', 'Double', 'Float', 'Int', 'Long', 'Short',\
'UnsignedChar', 'UnsignedInt', 'UnsignedLong', 'UnsignedShort']
for array in arrayType:
print(array + ' array')
vtkClass = 'vtk' + array + 'Array'
a = getattr(vtk, vtkClass)()
a.Allocate(1,1)
a.SetNumberOfComponents(3)
a.SetNumberOfTuples(4)
# InsertComponent
k = 0
for i in range(0, a.GetNumberOfTuples()):
for j in range(0, a.GetNumberOfComponents()):
a.InsertComponent(i, j, 1)
k += 1
# SetComponent
k = 0
for i in range(0, a.GetNumberOfTuples()):
for j in range(0, a.GetNumberOfComponents()):
a.SetComponent(i, j, 1)
k += 1
# DeepCopy
b = getattr(vtk, vtkClass)()
b.Allocate(1000, 100)
# force a resize
b.InsertComponent(2001, 0, 1)
b.DeepCopy(a)
# NewInstance
m = b.NewInstance()
m.UnRegister(b)
# confirm the deep copy
k = 0
for i in range(0, a.GetNumberOfTuples()):
for j in range(0, a.GetNumberOfComponents()):
if a.GetComponent(i, j) != b.GetComponent(i, j):
s = '%s: bad component %d, %d' % (array, i, j)
print(s)
k += 1
b.InsertComponent(2001, 0, 1)
b.Resize(3000)
a.Squeeze()
a.Initialize()
a = None
b = None
exit()
|
