Added user data serialization and expanded user data to actually use the user_index field so multiple types of user data are actually supported

test/test_serialize.py

@@ -0,0 +1,51 @@
+import os
+
+import numpy as np
+import pytest
+
+from atrcopy import DefaultSegment, SegmentData, get_xex, interleave_segments
+
+
+class TestSegment(object):
+    def setup(self):
+        data = np.ones([4000], dtype=np.uint8)
+        r = SegmentData(data)
+        self.segment = DefaultSegment(r, 0)
+
+    def test_s1(self):
+        s = self.segment
+        s.set_comment([[4,5]], "test1")
+        s.set_comment([[40,50]], "test2")
+        s.set_style_ranges([[2,100]], comment=True)
+        s.set_style_ranges([[200, 299]], data=True)
+        for i in range(1,4):
+            for j in range(1, 4):
+                # create some with overlapping regions, some without
+                r = [500*j, 500*j + 200*i + 200]
+                s.set_style_ranges([r], user=i)
+                s.set_user_data([r], i, i*10 + j)
+        r = [100, 200]
+        s.set_style_ranges([r], user=4)
+        s.set_user_data([r], 4, 99)
+        r = [3100, 3200]
+        s.set_style_ranges([r], user=4)
+        s.set_user_data([r], 4, 99)
+
+        out = dict()
+        s.serialize_extra_to_dict(out)
+        print "saved", out
+
+        data = np.ones([4000], dtype=np.uint8)
+        r = SegmentData(data)
+        s2 = DefaultSegment(r, 0)
+        s2.restore_extra_from_dict(out)
+        out2 = dict()
+        s2.serialize_extra_to_dict(out2)
+        print "loaded", out2
+        assert out == out2
+
+
+if __name__ == "__main__":
+    t = TestSegment()
+    t.setup()
+    t.test_s1()