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
|
from vtkmodules.vtkFiltersCore import vtkStripper
from vtkmodules.vtkFiltersGeometry import vtkDataSetSurfaceFilter
from vtkmodules.vtkFiltersParallel import vtkIntegrateAttributes
from vtkmodules.vtkIOLegacy import vtkDataSetReader
from vtkmodules.vtkIOXML import vtkXMLUnstructuredGridReader
from vtkmodules.vtkTestingRendering import vtkTesting
import sys
testing = vtkTesting()
for arg in sys.argv:
testing.AddArgument(arg)
VTK_DATA_ROOT = testing.GetDataRoot()
def Test1(datadir):
reader = vtkXMLUnstructuredGridReader()
reader.SetFileName(datadir + "/Data/quadraticTetra01.vtu")
reader.UpdateInformation();
reader.GetPointDataArraySelection().EnableAllArrays()
reader.GetCellDataArraySelection().EnableAllArrays()
f = vtkIntegrateAttributes()
f.SetInputConnection(reader.GetOutputPort())
f.Update()
result = f.GetOutputDataObject(0)
val = result.GetPointData().GetArray("scalars").GetValue(0)
assert (val > 0.0162 and val < 0.01621)
val = result.GetCellData().GetArray("Volume").GetValue(0)
assert (val > 0.128 and val < 0.1284)
def Test2(datadir):
reader = vtkXMLUnstructuredGridReader()
reader.SetFileName(datadir + "/Data/elements.vtu")
reader.UpdateInformation();
reader.GetPointDataArraySelection().EnableAllArrays()
reader.GetCellDataArraySelection().EnableAllArrays()
f = vtkIntegrateAttributes()
f.SetInputConnection(reader.GetOutputPort())
f.Update()
result = f.GetOutputDataObject(0)
val = result.GetPointData().GetArray("pointScalars").GetValue(0)
assert (val > 83.1 and val < 83.2)
val = result.GetCellData().GetArray("Volume").GetValue(0)
assert (val > 1.999 and val < 2.01)
def Test3(datadir):
reader = vtkDataSetReader()
reader.SetFileName(datadir + "/Data/blow.vtk")
reader.UpdateInformation();
reader.ReadAllScalarsOn()
reader.ReadAllVectorsOn()
dssf = vtkDataSetSurfaceFilter()
dssf.SetInputConnection(reader.GetOutputPort())
stripper = vtkStripper()
stripper.SetInputConnection(dssf.GetOutputPort())
f = vtkIntegrateAttributes()
f.SetInputConnection(stripper.GetOutputPort())
f.Update()
result = f.GetOutputDataObject(0)
val = result.GetPointData().GetArray("displacement1").GetValue(0)
assert (val > 463.64 and val < 463.642)
val = result.GetPointData().GetArray("thickness3").GetValue(0)
assert (val > 874.61 and val < 874.618)
val = result.GetCellData().GetArray("Area").GetValue(0)
assert (val > 1145.405 and val < 1145.415)
Test1(VTK_DATA_ROOT)
Test2(VTK_DATA_ROOT)
Test3(VTK_DATA_ROOT)
|
