1 files changed, 84 insertions, 0 deletions

diff --git a/io_import_dxf/dxfgrabber/dxfentity.py b/io_import_dxf/dxfgrabber/dxfentity.py
new file mode 100755
index 00000000..26cfc436
--- /dev/null
+++ b/io_import_dxf/dxfgrabber/dxfentity.py
@@ -0,0 +1,84 @@
+# Purpose: generic tag wrapper
+# Created: 21.07.2012, taken from my ezdxf project
+# Copyright (C) 2012, Manfred Moitzi
+# License: MIT License
+from __future__ import unicode_literals
+__author__ = "mozman <mozman@gmx.at>"
+
+import os
+
+from .const import ENV_CYTHON, XTYPE_NONE, XTYPE_2D, XTYPE_3D, XTYPE_2D_3D
+
+cyDXFEntity = None
+OPTIMIZE = True
+if ENV_CYTHON in os.environ:
+    if os.environ[ENV_CYTHON].upper() in ('1', 'ON', 'TRUE'):
+        OPTIMIZE = True
+    else:
+        OPTIMIZE = False
+try:
+    if OPTIMIZE:
+        from.cydxfentity import cyDXFEntity
+except ImportError:
+    pass
+
+
+class pyDXFEntity(object):
+    DXFATTRIBS = {}
+
+    def __init__(self, tags):
+        self.tags = tags
+
+    def dxftype(self):
+        return self.tags.noclass[0].value
+
+    def get_dxf_attrib(self, key, default=ValueError):
+        # core function - every optimization is useful
+        try:
+            dxfattr = self.DXFATTRIBS[key]
+        except KeyError:
+            # attribute is not defined - returning the default value is useful
+            # to query newer DXF attributes on older DXF files.
+            # !! Problem: misspelled attributes with default values do not
+            # raise an Exception !!
+            if default is ValueError:
+                raise ValueError("DXFAttrib '%s' is not defined." % key)
+            else:
+                return default
+        try:
+            return self._get_dxf_attrib(dxfattr)
+        except ValueError:  # attribute is defined but no value is present
+            if default is ValueError:
+                raise ValueError("DXFAttrib '%s': value is not present." % key)
+            else:
+                return default
+
+    def _get_dxf_attrib(self, dxfattr):
+        # no subclass is subclass index 0
+        subclass_tags = self.tags.subclasses[dxfattr.subclass]
+        xtype = dxfattr.xtype
+        if xtype != XTYPE_NONE and xtype != XTYPE_2D_3D:
+            return self._get_extented_type(subclass_tags, dxfattr.code, xtype)
+        else:
+            return subclass_tags.get_value(dxfattr.code)
+
+    def paperspace(self):
+        return self.get_dxf_attrib('paperspace', default=0) == 1
+
+    def post_read_correction(self):
+        pass
+
+    @staticmethod
+    def _get_extented_type(tags, code, xtype):
+        value = tags.get_value(code)
+        if len(value) == 2:
+            if xtype == XTYPE_3D:
+                return value[0], value[1], 0.
+        elif xtype == XTYPE_2D:
+            return value[0], value[1]
+        return value
+
+if cyDXFEntity is not None:
+    DXFEntity = cyDXFEntity
+else:
+    DXFEntity = pyDXFEntity
\ No newline at end of file