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
|
from vtkmodules import vtkCommonCore as cc
from vtkmodules import vtkCommonDataModel as dm
from vtkmodules import vtkCommonExecutionModel as em
from vtkmodules import vtkImagingCore as ic
from vtkmodules import vtkIOLegacy as il
from vtkmodules.test import Testing
from vtkmodules.util.misc import vtkGetTempDir
import os
class TestCompositeWriterReader(Testing.vtkTest):
def test(self):
p = dm.vtkPartitionedDataSet()
s = ic.vtkRTAnalyticSource()
s.SetWholeExtent(0, 10, 0, 10, 0, 5)
s.Update()
p1 = dm.vtkImageData()
p1.ShallowCopy(s.GetOutput())
s.SetWholeExtent(0, 10, 0, 10, 5, 10)
s.Update()
p2 = dm.vtkImageData()
p2.ShallowCopy(s.GetOutput())
p.SetPartition(0, p1)
p.SetPartition(1, p2)
p2 = dm.vtkPartitionedDataSet()
p2.ShallowCopy(p)
c = dm.vtkPartitionedDataSetCollection()
c.SetPartitionedDataSet(0, p)
c.SetPartitionedDataSet(1, p2)
tmpdir = vtkGetTempDir()
fname = tmpdir+"/testcompowriread.vtk"
w = il.vtkCompositeDataWriter()
w.SetInputData(c)
w.SetFileName(fname)
w.Write()
r = il.vtkCompositeDataReader()
r.SetFileName(fname)
r.Update()
o = r.GetOutputDataObject(0)
self.assertTrue(o.IsA("vtkPartitionedDataSetCollection"))
nd = o.GetNumberOfPartitionedDataSets()
self.assertEqual(nd, 2)
for i in range(nd):
p = o.GetPartitionedDataSet(i)
p2 = c.GetPartitionedDataSet(i)
self.assertTrue(p.IsA("vtkPartitionedDataSet"))
self.assertEqual(p.GetNumberOfPartitions(), 2)
self.assertEqual(p.GetPartition(0).GetNumberOfCells(), p2.GetPartition(0).GetNumberOfCells())
del(r)
import gc
gc.collect()
os.remove(fname)
if __name__ == "__main__":
Testing.main([(TestCompositeWriterReader, 'test')])
|
