technical_drawings_extraction/venv/lib/python3.7/site-packages/ezdxf/legacy/trace.py

# Created: 25.03.2011
# Copyright (c) 2011-2018, Manfred Moitzi
# License: MIT License
from typing import cast
from ezdxf.lldxf.const import VERTEXNAMES

from .graphics import GraphicEntity, ExtendedTags, make_attribs, DXFAttr, XType


class QuadrilateralMixin:
    __slots__ = ()

    def __getitem__(self, num):
        return cast(GraphicEntity, self).get_dxf_attrib(VERTEXNAMES[num])

    def __setitem__(self, num, value):
        return cast(GraphicEntity, self).set_dxf_attrib(VERTEXNAMES[num], value)


_TRACE_TPL = """0
TRACE
5
0
8
0
10
0.0
20
0.0
30
0.0
11
0.0
21
0.0
31
0.0
12
0.0
22
0.0
32
0.0
13
0.0
23
0.0
33
0.0
"""


class Trace(GraphicEntity, QuadrilateralMixin):
    __slots__ = ()
    TEMPLATE = ExtendedTags.from_text(_TRACE_TPL)
    DXFATTRIBS = make_attribs({
        'vtx0': DXFAttr(10, xtype=XType.any_point),
        'vtx1': DXFAttr(11, xtype=XType.any_point),
        'vtx2': DXFAttr(12, xtype=XType.any_point),
        'vtx3': DXFAttr(13, xtype=XType.any_point),
    })


class Solid(Trace):
    __slots__ = ()
    TEMPLATE = ExtendedTags.from_text(_TRACE_TPL.replace('TRACE', 'SOLID'))


class Face(Trace):
    __slots__ = ()
    TEMPLATE = ExtendedTags.from_text(_TRACE_TPL.replace('TRACE', '3DFACE'))
    DXFATTRIBS = make_attribs({
        'vtx0': DXFAttr(10, xtype=XType.point3d),
        'vtx1': DXFAttr(11, xtype=XType.point3d),
        'vtx2': DXFAttr(12, xtype=XType.point3d),
        'vtx3': DXFAttr(13, xtype=XType.point3d),
        'invisible_edge': DXFAttr(70, default=0),
    })