1 files changed, 88 insertions, 0 deletions
diff --git a/io_import_dxf/dxfgrabber/acdsdata.py b/io_import_dxf/dxfgrabber/acdsdata.py
new file mode 100755
index 00000000..60538d5f
--- /dev/null
+++ b/io_import_dxf/dxfgrabber/acdsdata.py
@@ -0,0 +1,88 @@
+# Purpose: acdsdata section manager
+# Created: 05.05.2014
+# Copyright (C) 2014, Manfred Moitzi
+# License: MIT License
+
+from __future__ import unicode_literals
+__author__ = "mozman <mozman@gmx.at>"
+
+from itertools import islice
+from .tags import TagGroups, DXFStructureError, Tags, binary_encoded_data_to_bytes
+
+
+class AcDsDataSection(object):
+    name = 'acdsdata'
+
+    def __init__(self):
+        # Standard_ACIS_Binary (SAB) data store, key = handle of DXF Entity in the ENTITIES section: BODY, 3DSOLID
+        # SURFACE, PLANESURFACE, REGION
+        self.sab_data = {}
+
+    @classmethod
+    def from_tags(cls, tags, drawing):
+        data_section = cls()
+        data_section._build(tags, drawing.dxfversion)
+        return data_section
+
+    def _build(self, tags, dxfversion):
+        if len(tags) == 3:  # empty entities section
+            return
+
+        for group in TagGroups(islice(tags, 2, len(tags)-1)):
+            data_record = AcDsDataRecord(Tags(group))
+            if data_record.dxftype == 'ACDSRECORD':
+                asm_data = data_record.get_section('ASM_Data', None)
+                if asm_data is not None:
+                    self.add_asm_data(data_record)
+
+    def add_asm_data(self, acdsrecord):
+        """ Store SAB data as binary string in the sab_data dict, with handle to owner Entity as key.
+        """
+        try:
+            asm_data = acdsrecord.get_section('ASM_Data')
+            entity_id = acdsrecord.get_section('AcDbDs::ID')
+        except ValueError:
+            return
+        else:
+            handle = entity_id[2].value
+            binary_data_text = (tag.value for tag in asm_data if tag.code == 310)
+            binary_data = binary_encoded_data_to_bytes(binary_data_text)
+            self.sab_data[handle] = binary_data
+
+
+class Section(Tags):
+    @property
+    def name(self):
+        return self[0].value
+
+    @property
+    def type(self):
+        return self[1].value
+
+    @property
+    def data(self):
+        return self[2:]
+
+
+class AcDsDataRecord(object):
+    def __init__(self, tags):
+        self.dxftype = tags[0].value
+        start_index = 2
+        while tags[start_index].code != 2:
+            start_index += 1
+        self.sections = [Section(tags) for tags in TagGroups(islice(tags, start_index, None), split_code=2)]
+
+    def has_section(self, name):
+        return self.get_section(name, default=None) is not None
+
+    def get_section(self, name, default=KeyError):
+        for section in self.sections:
+            if section.name == name:
+                return section
+        if default is KeyError:
+            raise KeyError(name)
+        else:
+            return default
+
+    def __getitem__(self, name):
+        return self.get_section(name)