h"azddlmZddlmZmZmZmZmZmZddl Z ddl Z ddl m Z ddl mZddlmZmZmZddlmZmZddlmZerdd lmZdd lmZdd lmZd d gZe j>dZ dZ!dZ"dddddZ#e$gdZ%Gdd Z& d,dZ'hdZ(Gdd Z)dddddd d!d"d#d$d%d&d'd(d)d*Z*d-d+Z+y).) annotations)AnyOptionalUnionIterable TYPE_CHECKINGSetN)options)const)XType DXFAttributesDXFAttr) cast_valuedxftag)Tags) ExtendedTags) DXFEntity)AbstractTagWriter DXFNamespaceSubclassProcessorezdxfz(Invalid DXF attribute "{}" for entity {}z!DXF attribute "{}" does not existon_layer_changeon_linetype_changeon_style_changeon_dimstyle_change)layerlinetypestyledimstyle)_entityhandleownercDeZdZdZ d ddZd dZd!d"dZd#dZd$dZdZ d d%d Z d&d Z d'd Z d(d Z d!d)d Zd&dZd'dZd'dZd*dZddd d+dZd(dZd,dZd,dZed-dZed.dZd&dZ d/dZ d0dZ d0dZy)1raz:class:`DXFNamespace` manages all named DXF attributes of an entity. The DXFNamespace.__dict__ is used as DXF attribute storage, therefore only valid Python names can be used as attrib name. The namespace can only contain immutable objects: string, int, float, bool, Vec3. Because of the immutability, copy and deepcopy are the same. (internal class) Nc<|ry|j}|djdk(rdnd}d}d}|D]:}|j}||k(r|j}|s#n|dk(s+|j}|s:n|j|||y|j |j|y)NrDIMSTYLEiJ) base_classvaluecoderewire reset_handles) self processorentityr) handle_coder!r"tag group_codes 4lib/python3.12/site-packages/ezdxf/entities/dxfns.py__init__zDXFNamespace.__init__3s "--J!+A!4!4 !B#KFE! XX , YYF3&IIE" KK .    KK c|j}|jjD]\}}||j|<|j||SN) __class____dict__itemsr,)r.r0 namespacekvs r4copyzDXFNamespace.copyOsMNN$ MM'')DAq$%I  q !* r6c8|j|jSr8)r?r )r.memodicts r4 __deepcopy__zDXFNamespace.__deepcopy__Vsyy&&r6c|jSr8r:r.s r4 __getstate__zDXFNamespace.__getstate__Ys }}r6ct|ts!tdt|jt j |d|y)Nzinvalid state: r:) isinstancedict TypeErrortype__name__object __setattr__)r.states r4 __setstate__zDXFNamespace.__setstate__\s;%&od5k.B.B-CDE E4U3r6c@d|jd<d|jd<y)zReset handle and owner to None.Nr!r"rDrEs r4r-zDXFNamespace.reset_handlesbs"& h!% gr6ch||jd<|||jd<|||jd<yy)zRewire DXF namespace with parent entity Args: entity: new associated entity handle: new handle or None owner: new entity owner handle or None r Nr!r"rD)r.r0r!r"s r4r,zDXFNamespace.rewiregs?$* i  &,DMM( #  %*DMM' " r6c*|jj|}|rD|jtjk(r|j |j S|jStjtj||j)zCalled if DXF attribute `key` does not exist, returns the DXF default value or ``None``. Raises: DXFAttributeError: attribute `key` is not supported ) dxfattribsgetxtyper callbackget_callback_valuer defaultr DXFAttributeErrorERR_INVALID_DXF_ATTRIBformatdxftype)r.key attrib_defs r4 __getattr__zDXFNamespace.__getattr__|su)-(;(;C(@ 5>>1!44T\\BB!)))))&--c4<<@ r6cdfd fd}jjrPjtjk(rj j |nI||j<n3tjtjjtvr*tj td}|r ||yyy)~Set DXF attribute `key` to `value`. Raises: DXFAttributeError: attribute `key` is not supported czjd}j}|r|jd|dzSd|S)Nr!z(#)#)rUr r])r!r r.s r4r0z(DXFNamespace.__setattr__..entitysCXXh'FllG(Rxq>996(|#r6c Htj|}j|syjr@j|}tj dddt |d|Stjdt |ddd|S)NzFixed invalid attribute "z " in entity z to "z".zInvalid value z for attribute ".) rr+is_valid_valuefixerloggerdebugstrr DXFValueError)r*r_r0r^s r4checkz'DXFNamespace.__setattr__..checksz6E,,U3##&,,U3ELL3C59"H:U3u:,b:L  --(U 4DSEJ""((1.Lr6Nreturnrl)rTrUrVr rWset_callback_valuer r:r rZr[r\r] SETTER_EVENTSgetattr)r.r^r*rnhandlerr_r0s`` @@r4rNzDXFNamespace.__setattr__s $  )-(;(;C(@ 5>>1--dllEB%*5\ c"))&--c4<<@  - dllM#,>EG r6c|j|r|j|=ytjtj |)ztDelete DXF attribute `key`. Raises: DXFAttributeError: attribute `key` does not exist N)hasattrr:r rZERR_DXF_ATTRIB_NOT_EXITSr\r.r^s r4 __delattr__zDXFNamespace.__delattr__s9 <<  c"))*B*I*I#*NO Or6cV|j|r|j|S|jj|}|r:|jt j k(r|j|jS|Stjtj||j)zReturns value of DXF attribute `key` or the given `default` value not DXF default value for unset attributes. Raises: DXFAttributeError: attribute `key` is not supported )rvr:rTrUrVr rWrXr r rZr[r\r])r.r^rYr_s r4rUzDXFNamespace.gets << ==% %*.//*=*=c*B 5>>1!44T\\BB))&--c4<<@ r6cP|j|d}||j|S|S)z8Returns DXF default value for unset DXF attribute `key`.N)rUdxf_default_valuer.r^r*s r4 get_defaultzDXFNamespace.get_defaults,d#.3mt%%c*FFr6c(|j||y)rbN)rNr}s r4setzDXFNamespace.sets e$r6c"||j|<y)zSet DXF attribute `key` to `value` without any validity checks. Used for fast attribute setting without validity checks at loading time. (internal API) NrDr}s r4unprotected_setzDXFNamespace.unprotected_sets# cr6c6t|j}|d=|S)z@Returns all existing DXF attributes, except DXFEntity back-link.r )rIr:)r.attribss r4all_existing_dxf_attribsz%DXFNamespace.all_existing_dxf_attribsst}}% I r6F)exclude ignore_errorsc|t}ndh|z}|j}|jD]\}}||vs |||y#ttf$r|sY,wxYw)z,Update DXF namespace attributes from a dict.Nr )EXCLUDE_FROM_UPDATErNr;AttributeError ValueError)r.rTrr set_attributer=r>s r4updatezDXFNamespace.updatest ?)G kG+G(( $$&DAq!!Q''' 3()s AAAc> |j|=y#t$rYywxYw)z:Delete DXF attribute `key` silently without any exception.N)r:KeyErrorrxs r4discardzDXFNamespace.discards%  c"   s  c||jvS)aReturns True if DXF attribute `key` is supported else False. Does not grant that attribute `key` really exists and does not check if the actual DXF version of the document supports this attribute, unsupported attributes will be ignored at export. )rTrxs r4 is_supportedzDXFNamespace.is_supportedsdoo%%r6c||jvS)z9Returns True if attribute `key` really exists else False.rDrxs r4rvzDXFNamespace.hasattr&sdmm##r6c.|jjS)zReturns the DXF entity type.)r DXFTYPErEs r4r]zDXFNamespace.dxftype*s||###r6c.|jjS)z%Returns the DXF attribute definition.)r DXFATTRIBSrEs r4rTzDXFNamespace.dxfattribs/s||&&&r6cV|jj|}|r |jSy)z9Returns the default value as defined in the DXF standard.N)rTrUrY)r.r^attribs r4r|zDXFNamespace.dxf_default_value4s&$(OO$7$7$< >> !r6c|t|tr|j||y|D]}|j||y)aExports DXF attribute `name` by `tagwriter`. Non-optional attributes are forced and optional tags are only written if different to default value. DXF version check is always on: does not export DXF attribs which are not supported by tagwriter.dxfversion. Args: tagwriter: tag writer object attribs: DXF attribute name as string or an iterable of names N)rHrl_export_dxf_attribute_optional)r. tagwriterrnames r4export_dxf_attribszDXFNamespace.export_dxf_attribs<s8 gs #  / / 7 C33ItD r6c|jj|}|3tjtj ||j |j}|j}|j|d}||s|}|tjd|dy|r|js|||k(ryt|||y)zExports DXF attribute `name` by `tagwriter`. Optional tags are only written if they differ from the default value. NDXF attribute ''' not written because its a None value.) rTrUr rZr[r\r]optionalrYrjrkforce_optional_export_group_codes)r.rrrrrYr*s r4rz+DXFNamespace._export_dxf_attribute_optionalOs%)OO$7$7$= >))&--dDLLA ??..t$ =E = LL!$'NO   --#5 Ivu5r6c|j|sy|jj|}| Jd|d|j|}|tj d|dyt |||y)aKExports DXF attribute `name` by `tagwriter` if exists. If the attribute exists, and it's not None it will be written, the optional-flag is ignored. No default value will be written if the attribute doesn't exist! This method can not be used for attributes that are required (e.g. layer)! Nzexisting DXF attribute 'z*' has no definition class - internal errorrr)rvrTrUrjrkr)r.rrrr*s r4export_dxf_attribute_if_existsz+DXFNamespace.export_dxf_attribute_if_existsvs||D! $(OO$7$7$=   W %dV+U V W  = LL!$'NO  Ivu5r6)NN)r/zOptional[SubclassProcessor]r0Optional[DXFEntity])r0rr8)rAzOptional[dict])rprM)rOrMrpNone)r0rr! Optional[str]r"rrpr)r^rlrpr)r^rlr*rrpr)r^rlrpr)r^rlrYrrpr)rprI)rTzdict[str, Any]rzOptional[Set[str]]rpr)r^rlrpboolro)rpr )rrrzUnion[str, Iterable]rpr)rrrrlrpr)rL __module__ __qualname____doc__r5r?rBrFrPr-r,r`rNryrUr~rrrrrrrvpropertyr]rTr|rrrr6r4rr's^ 26&* . $ 8'4 &!%# +#++ +  +*&/b P.G %#'+ "$   ,&$$$''E*E5IE E&%6*%625%6 %6N6*6256 6r6c|J|J|j|jkry|jtjk(r't |dkDr |j |j f}t|trd|vsJdd|vsJdt|j|}|j|y#t$r|dd}Y^wxYw)N zline break '\n' not allowed zline break '\r' not allowed) dxfversionrVr point2dlenxyrrHrlrr+ write_tag)rrr*r2s r4rrs      f///||u}}$Ua WWegg&E%5 @"@@ 5 @"@@ e $C  "1IE sB55CC>rr'r(fceZdZdZdddZedZ d ddZddZddZ ddZ d d d  dd Z dd Z ddZ y)rz=Helper class for loading tags into entities. (internal class)Nct|jdk(r tdt|j|_|jxsg|_||_|t jk(xst|jdk(|_|j|_ | |j|_ y#t j$r d|_ YywxYw)Nrz Invalid tags.)r subclassesrlistembedded_objectsrr DXF12r12r]r get_handler!rm)r.tagsrs r4r5zSubclassProcessor.__init__s t 1 $_- -&*4??&;,0,A,A,GR)3% 3ST__9MQR9R   //+DK"" DK s$B::CCc |jdSNr)rrEs r4r)zSubclassProcessor.base_classsq!!r6ctjr8|D]2}d}|rd|}tjdt |d||z4yy)Nz in entity #zignored z in subclass )r log_unprocessed_tagsrjinforepr)r.unprocessed_tagssubclassr!r2r0s r4rz&SubclassProcessor.log_unprocessed_tagssR  ' ''+F84F htCykxjIFRS ( (r6cj|jD]$}t|s|dj|k(s"|cSyr)rrr*)r.rrs r4 find_subclasszSubclassProcessor.find_subclasss2H8}!!2!2d!:(r6c@ |j|S#t$rYywxYwr8)r IndexError)r.indexs r4subclass_by_indexz#SubclassProcessor.subclass_by_indexs( ??5) )  s  c|j|}|r.t|dkDr |d}|j|k(r |jS|S)Nr)rrr+r*)r.subclass_indexr3rYrr2s r4detect_implementation_versionz/SubclassProcessor.detect_implementation_versionsF)).9 H )1+Cxx:%yy r6FT)recoverlogcp|jr|jd}nFt|tr|j |}n$t|t r|j |}n|}t}|t|dk(r|St}|j} |j} |j} |j} |djdvrdnd} || dD]z}| |j}t|tr|}d}|D]}||vs|}| |n|r1|ddk7sK| |t!|j|j"s| |||jr|S|rt|r t%||}t|r4|r2|dj"}|j'|||jd|S)a2Load DXF attributes into the DXF namespace and returns the unprocessed tags without leading subclass marker(100, AcDb...). Bypasses the DXF attribute validity checks. Args: dxf: entity DXF namespace group_code_mapping: group code to DXF attribute name mapping, callback attributes have to be marked with a leading "*" subclass: subclass by index, by name or as Tags() recover: recover graphic attributes log: enable/disable logging of unprocessed tags rN)rdr*r!)rr!)rrrHintrrlrrrrrUappendraddr+rrr*recover_graphic_attributesr)r.dxfgroup_code_mappingrrrrrprocessed_namesget_attrib_nameappend_unprocessed_tagunprotected_set_attribmark_attrib_as_processedstartr2rnamesname_s r4fast_load_dxfattribsz&SubclassProcessor.fast_load_dxfattribss, 88??1%D(C(--h7Hc*))(36 <3t9># #$'E,00!1!8!8!$!4!4#2#6#6 !W\\X-1*j399='s+- 0 88 $ # s+, ::JCP   S7==D  % % 48I &  r6c|jry|jd}|ddk(r&|jd|jdDyy)zIt is valid to mix up the base class with AcDbEntity class. This method appends all none base class group codes to the AcDbEntity class. Nrr)r AcDbEntityc3FK|]}|jtvs|ywr8)r+BASE_CLASS_CODES).0r2s r4 zESubclassProcessor.append_base_class_to_acdb_entity..Ws!$1SXXEU5U1s!!)rrextend)r.acdb_entity_tagss r4 append_base_class_to_acdb_entityz2SubclassProcessor.append_base_class_to_acdb_entityHsT 88 ??1- A "5 5  # #$#q1$  6r6cLtjj|j}|j}|j }|D]^}||j }t|ts&|jdr8||t|j |j`y)aLoad DXF attributes from all subclasses into the DXF namespace. Can not handle same group codes in different subclasses, does not remove processed tags or log unprocessed tags and bypasses the DXF attribute validity checks. This method ignores the subclass structure and can load data from very malformed DXF files, like such in issue #604. This method works only for very simple DXF entities with unique group codes in all subclasses, the old DXF R12 entities: - POINT - LINE - CIRCLE - ARC - INSERT - SHAPE - SOLID/TRACE/3DFACE - TEXT (ATTRIB/ATTDEF bypasses TEXT loader) - BLOCK/ENDBLK - POLYLINE/VERTEX/SEQEND - DIMENSION and subclasses - all table entries: LAYER, LTYPE, ... And the newer DXF entities: - ELLIPSE - RAY/XLINE The recover mode for graphical attributes is automatically included. Logging of unprocessed tags is not possible but also not required for this simple and well known entities. Args: dxf: entity DXF namespace group_code_mapping: group code name mapping for all DXF attributes from all subclasses, callback attributes have to be marked with a leading "*" rN) itertoolschain from_iterablerrUrr+rHrl startswithrr*)r.rrrrrr2rs r4simple_dxfattribs_loaderz*SubclassProcessor.simple_dxfattribs_loader[szX,,T__=,00!$!4!4C"388,D$$T__S-A&tZ#))-LMr6r8)rrrr)rN)rrr!rrpr)rrlrpOptional[Tags])rrrpr)rrr3rrYrrpr)rrrzdict[int, Union[str, list]]rzUnion[int, str, Tags]rpr)rpr)rrrzdict[int, str]rpr)rLrrrr5rr)rrrrrrrrr6r4rrsG (""  $ T" T T  T  ! /2 =@   $ W W 8W ( W  W r&2N2N5C2N 2Nr6rrcolor paperspace lineweightltscale invisible true_color color_name transparency shadow_modematerial_handlevisualstyle_handleplotstyle_enumplotstyle_handle)>Cir0<iiiii[i\i|ict}|D]b}tj|j}|r.|j |s|j ||j R|j|d|Sr8)rGRAPHIC_ATTRIBUTES_TO_RECOVERrUr+rvrr*r)rrrr2 attrib_names r4rrs`v377A  s{{;7 GGK +  # #C ( r6)rrrrr*rrpr)rrrrrpr), __future__rtypingrrrrrr loggingrrr ezdxf.lldxfr ezdxf.lldxf.attributesr r rezdxf.lldxf.typesrrezdxf.lldxf.tagsrezdxf.lldxf.extendedtagsrezdxf.entitiesrezdxf.lldxf.tagwriterr__all__ getLoggerrjr[rwrr frozensetrrrrrrrrr6r4rs#EE@@0!5(7 . /   7 #C>$ $   >?i6i6X  *1:= 2$^N^ND         !* r6