refactor: excel parse
This commit is contained in:
Blizzard
@@ -0,0 +1,235 @@
|
||||
# Copyright (c) 2019-2023, Manfred Moitzi
|
||||
# License: MIT-License
|
||||
from __future__ import annotations
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from typing_extensions import Self
|
||||
import copy
|
||||
from ezdxf.lldxf import validator
|
||||
from ezdxf.lldxf.const import SUBCLASS_MARKER, DXF2000, DXFStructureError
|
||||
from ezdxf.lldxf.attributes import (
|
||||
DXFAttributes,
|
||||
DefSubclass,
|
||||
DXFAttr,
|
||||
group_code_mapping,
|
||||
)
|
||||
from ezdxf.lldxf.tags import Tags
|
||||
from .dxfentity import base_class, SubclassProcessor
|
||||
from .dxfobj import DXFObject
|
||||
from .factory import register_entity
|
||||
from .copy import default_copy
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from ezdxf.entities import DXFNamespace, DXFEntity
|
||||
from ezdxf.lldxf.tagwriter import AbstractTagWriter
|
||||
|
||||
__all__ = ["VisualStyle"]
|
||||
|
||||
acdb_visualstyle = DefSubclass(
|
||||
"AcDbVisualStyle",
|
||||
{
|
||||
"description": DXFAttr(2),
|
||||
# Style type:
|
||||
# 0 = Flat
|
||||
# 1 = FlatWithEdges
|
||||
# 2 = Gouraud
|
||||
# 3 = GouraudWithEdges
|
||||
# 4 = 2dWireframe
|
||||
# 5 = Wireframe
|
||||
# 6 = Hidden
|
||||
# 7 = Basic
|
||||
# 8 = Realistic
|
||||
# 9 = Conceptual
|
||||
# 10 = Modeling
|
||||
# 11 = Dim
|
||||
# 12 = Brighten
|
||||
# 13 = Thicken
|
||||
# 14 = Linepattern
|
||||
# 15 = Facepattern
|
||||
# 16 = ColorChange
|
||||
# 20 = JitterOff
|
||||
# 21 = OverhangOff
|
||||
# 22 = EdgeColorOff
|
||||
# 23 = Shades of Gray
|
||||
# 24 = Sketchy
|
||||
# 25 = X-Ray
|
||||
# 26 = Shaded with edges
|
||||
# 27 = Shaded
|
||||
"style_type": DXFAttr(
|
||||
70, validator=validator.is_in_integer_range(0, 28)
|
||||
),
|
||||
# Face lighting type:
|
||||
# 0 = Invisible
|
||||
# 1 = Visible
|
||||
# 2 = Phong
|
||||
# 3 = Gooch
|
||||
"face_lighting_model": DXFAttr(
|
||||
71, validator=validator.is_in_integer_range(0, 4)
|
||||
),
|
||||
# Face lighting quality:
|
||||
# 0 = No lighting
|
||||
# 1 = Per face lighting
|
||||
# 2 = Per vertex lighting
|
||||
# 3 = Unknown
|
||||
"face_lighting_quality": DXFAttr(
|
||||
72, validator=validator.is_in_integer_range(0, 4)
|
||||
),
|
||||
# Face color mode:
|
||||
# 0 = No color
|
||||
# 1 = Object color
|
||||
# 2 = Background color
|
||||
# 3 = Custom color
|
||||
# 4 = Mono color
|
||||
# 5 = Tinted
|
||||
# 6 = Desaturated
|
||||
"face_color_mode": DXFAttr(
|
||||
73, validator=validator.is_in_integer_range(0, 6)
|
||||
),
|
||||
# Face modifiers:
|
||||
# 0 = No modifiers
|
||||
# 1 = Opacity
|
||||
# 2 = Specular
|
||||
# 3 = Unknown
|
||||
"face_modifiers": DXFAttr(
|
||||
90, validator=validator.is_in_integer_range(0, 4)
|
||||
),
|
||||
"face_opacity_level": DXFAttr(40),
|
||||
"face_specular_level": DXFAttr(41),
|
||||
"color1": DXFAttr(62),
|
||||
"color2": DXFAttr(63),
|
||||
"face_style_mono_color": DXFAttr(421),
|
||||
# Edge style model:
|
||||
# 0 = No edges
|
||||
# 1 = Isolines
|
||||
# 2 = Facet edges
|
||||
"edge_style_model": DXFAttr(
|
||||
74, validator=validator.is_in_integer_range(0, 3)
|
||||
),
|
||||
"edge_style": DXFAttr(91),
|
||||
"edge_intersection_color": DXFAttr(64),
|
||||
"edge_obscured_color": DXFAttr(65),
|
||||
"edge_obscured_linetype": DXFAttr(75),
|
||||
"edge_intersection_linetype": DXFAttr(175),
|
||||
"edge_crease_angle": DXFAttr(42),
|
||||
"edge_modifiers": DXFAttr(92),
|
||||
"edge_color": DXFAttr(66),
|
||||
"edge_opacity_level": DXFAttr(43),
|
||||
"edge_width": DXFAttr(76),
|
||||
"edge_overhang": DXFAttr(77),
|
||||
"edge_jitter": DXFAttr(78),
|
||||
"edge_silhouette_color": DXFAttr(67),
|
||||
"edge_silhouette_width": DXFAttr(79),
|
||||
"edge_halo_gap": DXFAttr(170),
|
||||
"edge_isoline_count": DXFAttr(171),
|
||||
"edge_hide_precision": DXFAttr(290), # flag
|
||||
"edge_style_apply": DXFAttr(174), # flag
|
||||
"style_display_settings": DXFAttr(93),
|
||||
"brightness": DXFAttr(44),
|
||||
"shadow_type": DXFAttr(173),
|
||||
"unknown1": DXFAttr(177), # required if xdata is present?
|
||||
"internal_use_only_flag": DXFAttr(
|
||||
291
|
||||
), # visual style only use internal
|
||||
# Xdata must follow tag 291 (AutoCAD -> 'Xdata wasn't read' error)
|
||||
# 70: Xdata count (count of tag groups)
|
||||
# any code, value: multiple tags build a tag group
|
||||
# 176, 1: end of group marker
|
||||
# e.g. (291, 0) (70, 2) (62, 7) (420, 16777215) (176, 1) (90, 1) (176, 1)
|
||||
},
|
||||
)
|
||||
acdb_visualstyle_group_codes = group_code_mapping(acdb_visualstyle)
|
||||
|
||||
|
||||
# undocumented Xdata in DXF R2018
|
||||
@register_entity
|
||||
class VisualStyle(DXFObject):
|
||||
"""DXF VISUALSTYLE entity"""
|
||||
|
||||
DXFTYPE = "VISUALSTYLE"
|
||||
DXFATTRIBS = DXFAttributes(base_class, acdb_visualstyle)
|
||||
MIN_DXF_VERSION_FOR_EXPORT = DXF2000 # official supported in R2007
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.acad_xdata = None # to preserve AutoCAD xdata
|
||||
|
||||
def copy_data(self, entity: Self, copy_strategy=default_copy) -> None:
|
||||
"""Copy acad internal data."""
|
||||
assert isinstance(entity, VisualStyle)
|
||||
entity.acad_xdata = copy.deepcopy(self.acad_xdata)
|
||||
|
||||
def load_dxf_attribs(
|
||||
self, processor: Optional[SubclassProcessor] = None
|
||||
) -> DXFNamespace:
|
||||
dxf = super().load_dxf_attribs(processor)
|
||||
if processor:
|
||||
tags = processor.subclass_by_index(1)
|
||||
if tags:
|
||||
self.acad_xdata = self.store_acad_xdata(tags)
|
||||
processor.fast_load_dxfattribs(
|
||||
dxf, acdb_visualstyle_group_codes, subclass=tags
|
||||
)
|
||||
else:
|
||||
raise DXFStructureError(
|
||||
f"missing 'AcDbVisualStyle' subclass in VISUALSTYLE(#{dxf.handle})"
|
||||
)
|
||||
|
||||
return dxf
|
||||
|
||||
@staticmethod
|
||||
def store_acad_xdata(tags: Tags):
|
||||
try:
|
||||
index = tags.tag_index(291)
|
||||
except IndexError:
|
||||
return None
|
||||
else: # store tags after 291
|
||||
index += 1
|
||||
xdata = tags[index:]
|
||||
del tags[index:] # remove xdata from subclass
|
||||
return xdata
|
||||
|
||||
def export_entity(self, tagwriter: AbstractTagWriter) -> None:
|
||||
"""Export entity specific data as DXF tags."""
|
||||
super().export_entity(tagwriter)
|
||||
tagwriter.write_tag2(SUBCLASS_MARKER, acdb_visualstyle.name)
|
||||
self.dxf.export_dxf_attribs(
|
||||
tagwriter,
|
||||
[
|
||||
"description",
|
||||
"style_type",
|
||||
"face_lighting_model",
|
||||
"face_lighting_quality",
|
||||
"face_color_mode",
|
||||
"face_modifiers",
|
||||
"face_opacity_level",
|
||||
"face_specular_level",
|
||||
"color1",
|
||||
"color2",
|
||||
"face_style_mono_color",
|
||||
"edge_style_model",
|
||||
"edge_style",
|
||||
"edge_intersection_color",
|
||||
"edge_obscured_color",
|
||||
"edge_obscured_linetype",
|
||||
"edge_intersection_linetype",
|
||||
"edge_crease_angle",
|
||||
"edge_modifiers",
|
||||
"edge_color",
|
||||
"edge_opacity_level",
|
||||
"edge_width",
|
||||
"edge_overhang",
|
||||
"edge_jitter",
|
||||
"edge_silhouette_color",
|
||||
"edge_silhouette_width",
|
||||
"edge_halo_gap",
|
||||
"edge_isoline_count",
|
||||
"edge_hide_precision",
|
||||
"edge_style_apply",
|
||||
"style_display_settings",
|
||||
"brightness",
|
||||
"shadow_type",
|
||||
"unknown1",
|
||||
"internal_use_only_flag",
|
||||
],
|
||||
)
|
||||
if self.acad_xdata:
|
||||
tagwriter.write_tags(self.acad_xdata)
|
||||
Reference in New Issue
Block a user