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
83
84
85
86
87
|
from vtkmodules.test import Testing as vtkTesting
import re
class TestBlobs(vtkTesting.vtkTest):
def test(self):
from vtkmodules.vtkSerializationManager import vtkObjectManager
from vtkmodules.vtkCommonCore import vtkTypeFloat32Array, vtkUnsignedIntArray
arrayF32Id = ""
arrayUI32Id = ""
def serialize():
global arrayF32Id, arrayUI32Id
manager = vtkObjectManager()
if not manager.Initialize():
return
arrayF32 = vtkTypeFloat32Array()
arrayUI32 = vtkUnsignedIntArray()
for i in range(10):
arrayF32.InsertNextValue(10 - i)
arrayUI32.InsertNextValue(20 + i)
arrayF32Id = manager.RegisterObject(arrayF32)
self.assertEqual(arrayF32Id, 1)
arrayUI32Id = manager.RegisterObject(arrayUI32)
self.assertEqual(arrayUI32Id, 2)
manager.UpdateStatesFromObjects()
active_ids = manager.GetAllDependencies(0)
# these are not guaranteed to appear in the exact same order
# due to the unordered_map in use within vtkObjectManager.
self.assertEqual(len(active_ids), 2)
self.assertEqual(arrayF32Id in active_ids, True)
self.assertEqual(arrayUI32Id in active_ids, True)
states = [ manager.GetState(object_id) for object_id in active_ids ]
hash_to_blob_map = { blob_hash: manager.GetBlob(blob_hash) for blob_hash in manager.GetBlobHashes(active_ids) }
manager.UnRegisterObject(arrayF32Id)
manager.UnRegisterObject(arrayUI32Id)
return states, hash_to_blob_map
def deserialize(states, hash_to_blob_map):
global arrayF32Id, arrayUI32Id
if not len(states):
return
manager = vtkObjectManager()
if not manager.Initialize():
return
for state in states:
manager.RegisterState(state)
for hash_text, blob in hash_to_blob_map.items():
manager.RegisterBlob(hash_text, blob)
manager.UpdateObjectsFromStates()
active_ids = manager.GetAllDependencies(0)
self.assertEqual(len(active_ids), 2)
self.assertEqual(arrayF32Id in active_ids, True)
self.assertEqual(arrayUI32Id in active_ids, True)
arrayF32 = manager.GetObjectAtId(arrayF32Id)
self.assertIsInstance(arrayF32, vtkTypeFloat32Array)
arrayUI32 = manager.GetObjectAtId(arrayUI32Id)
self.assertIsInstance(arrayUI32, vtkUnsignedIntArray)
for i in range(10):
self.assertEqual(10 - i, arrayF32.GetValue(i))
self.assertEqual(20 + i, arrayUI32.GetValue(i))
deserialize(*serialize())
if __name__ == "__main__":
vtkTesting.main([(TestBlobs, 'test')])
|
