hCddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZddlmZmZdd lmZdd lmZdd lmZmZmZm Z dd lm!Z!dd l"m#Z#ddl$m%Z%gdZ&hdZ'ee e(fZ)GddZ*GddZ+dddZ, d ddZ- d ddZ. d d dZ/d!dZ0y)") annotations)IterableIteratorcastBinaryIOOptionalUnionAny)StringIO)Path)DXFStructureError) ExtendedTagsDXFTag) TagWriter) tag_compilerascii_tags_loader) dxf_file_info) fileindex) DXFGraphic DXFEntityPolylineInsert)factory) entity_linker) toencoding)opendxfsingle_pass_modelspace modelspace>ARCRAYLINEMESHTEXTHATCHHELIXIMAGEMLINEMTEXTPOINTSHAPESOLIDTRACEXLINE3DFACEATTDEFATTRIBCIRCLEINSERTLEADERSEQENDSPLINEVERTEXELLIPSEMLEADERWIPEOUTPOLYLINE DIMENSION LWPOLYLINEceZdZdZd d dZ ddZedZedZddZ ddZ d dd Z dd Z d Z y)IterDXFu4Iterator for DXF entities stored in the modelspace. Args: name: filename, has to be a seekable file. errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError`exception for invalid data Raises: DXFStructureError: invalid or incomplete DXF file UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs c|jt|\|_|_||_t |d|_d|jvr td|jjdkDrd|jvr tdyy)NrbmodeENTITIESzENTITIES section not found.AC1009OBJECTSzOBJECTS section not found.) _load_indexstr structuresectionserrorsopenfiler version)selfnamerJs 4lib/python3.12/site-packages/ezdxf/addons/iterdxf.py__init__zIterDXF.__init__Rs~(,(8(8T(C%  "4d3 T]] *#$AB B >> ! !H ,$--1O#$@A A2P ,ctj|}t}g}|jD]N}|jdk(r|j |$|jdk(s4t |dz ||j<P||_||fS)Nr)rloaddictindexcodeappendlenvalue)rNrOrHrI new_indexes rPrFzIterDXF._load_index[s{NN4( #'6 Avv{  #1$' NQ$6! ! $ (""rRc.|jjSN)rHencodingrNs rPrazIterDXF.encodingjs~~&&&rRc.|jjSr`)rHrMrbs rP dxfversionzIterDXF.dxfversionns~~%%%rRct||}|jj|jddzj}|j j d|j j|}|j||S)a^Returns a companion object to export parts from the source DXF file into another DXF file, the new file will have the same HEADER, CLASSES, TABLES, BLOCKS and OBJECTS sections, which guarantees all necessary dependencies are present in the new file. Args: name: filename, no special requirements rCrUr) IterDXFWriterrHrXrIlocationrLseekread write_data)rNrOdocrgdatas rPexportzIterDXF.exportrsmD$'>>'' j(AA(EFOO qyy~~h' t rRc|jd} |jjdd|}|jj |j }|jj |dzj }||z }|jj||jj|}|j|y#t$r t dwxYw)NrErENDSECz$ENDSEC of OBJECTS section not found.rU) rIrHget ValueErrorr rXrgrLrhriwrite)rNf start_index end_indexstart_location end_locationcountrls rPcopy_objects_sectionzIterDXF.copy_objects_sectionsmmI.  M**1h DI--k:CC~~++IM:CC ~- ~&yy~~e$   M#&JL L Ms CCNc#Kt}d}t|}|j|jddz|D]-}||r |jj dk(s&|r||}/|r|yyw)aReturns an iterator for all supported DXF entities in the modelspace. These entities are regular :class:`~ezdxf.entities.DXFGraphic` objects but without a valid document assigned. It is **not** possible to add these entities to other `ezdxf` documents. It is only possible to recreate the objects by factory functions base on attributes of the source entity. For MESH, POLYMESH and POLYFACE it is possible to use the :class:`~ezdxf.render.MeshTransformer` class to render (recreate) this objects as new entities in another document. Args: types: DXF types like ``['LINE', '3DFACE']`` which should be returned, ``None`` returns all supported types. NrCrUr)r_requested_types load_entitiesrIdxf paperspace)rNtypes linked_entityqueuedrequested_typesentitys rPrzIterDXF.modelspaces&& *51(( MM* % )? F!(VZZ-B-Ba-G L  L sAA4A4"A4c#Kdfd }|}jj|}jj|j|j dk7r|dz }jj|}|j|jz }jj |}|j |vr2tj||} tj| |}|j dk7ryyw)Ncr|jjjjddS)NrJz  )decoderarJreplace)rlrNs rPto_strz%IterDXF.load_entities..to_strs0;;t}}T[[;AII rRrorU)rlbytesreturnrG) rHrXrLrhrgr\rir from_textrrV) rNstartrrrXentry next_entrysizerlxtagss ` rPr|zIterDXF.load_entitiess  $$U+ u~~&kkX% QJE--e4J&&7D99>>$'D{{o-$..vd|<ll5))EkkX%s C9C?=C?c8|jjy)zSafe closing source DXF file.N)rLcloserbs rPrz IterDXF.closes rRsurrogateescape)rOFilenamerJrG)rOrGrz.tuple[fileindex.FileStructure, dict[str, int]])rOrrrf)rsrrNoner`)rOptional[Iterable[str]]rIterable[DXFGraphic])rintrset[str]rr)__name__ __module__ __qualname____doc__rQrFpropertyrardrmryrr|rrRrPr>r>As B # # 7 #''&&& 04 ,   D+3 *rRr>c*eZdZddZddZddZdZy) rfct||_t|d|_t |_t |j |j|_||_ y)NwbrA) rGrOrKrLr textrrd entity_writerloader)rNrOrs rPrQzIterDXFWriter.__init__sDI "4d3 J &tyy&2C2CD rRc:|jj|yr`)rLrr)rNrls rPrjzIterDXFWriter.write_datas rRc|jjd|jj|jjd|j _|j|j |jdk(rbtt|}|jD]}|j|j |jj|j n|jdk(rmtt|}|jrQ|jD]}|j|j |jj|j |jj!j#|j$j&}|j(j+|y)a%Write a DXF entity from the source DXF file to the export file. Don't write entities from different documents than the source DXF file, dependencies and resources will not match, maybe it will work once, but not in a reliable way for different DXF documents. rNFr:r2)rrhtruncater}handler write_handles export_dxfdxftyperrverticesseqendrattribs_followattribsgetvalueencoderrarLrr)rNrpolylinevertexinsertattribrls rPrrzIterDXFWriter.writesB" q  ::   $/4D   ,$,,- >> z )Hf-H"++!!$"4"45, OO & &t'9'9 : ^^  )&&)F$$$nnF%%d&8&89- ((););<yy!!#**4;;+?+?@ rRc |jjd|jjdkDr%|jj |j|jjd|jj y)zSafe closing of exported DXF file. Copying of OBJECTS section happens only at closing the file, without closing the new DXF file is invalid. s 0 ENDSEC rDs 0 EOF N)rLrrrrdryrrbs rPrzIterDXFWriter.closes\ ,- ;; ! !H , KK , ,TYY 7 )* rRN)rOrrr>)rlr)rr)rrrrQrjrrrrrRrPrfrfs$L rRrfct||S)uOpen DXF file for iterating, be sure to open valid DXF files, no DXF structure checks will be applied. Use this function to split up big DXF files as shown in the example above. Args: filename: DXF filename of a seekable DXF file. errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: invalid or incomplete DXF file UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs r)r>)filenamerJs rPrr s& 8F ++rRNc#Ktt|}d}d}d}t|}t|d|j|5}t |} d} g} t } t| D]} | j}| j}|r|dk(rgt| rX| dj|vrGtjt| }| |s!|jjdk(r| r| |} | g} n| j!| |dk(r|dk(r| r| dddy|d k(r|dk(r |d k(r|d k(}|}|} dddy#1swYyxYww) uIterate over all modelspace entities as :class:`DXFGraphic` objects of a seekable file. Use this function to iterate "quick" over modelspace entities of a DXF file, filtering DXF types may speed up things if many entity types will be skipped. Args: filename: filename of a seekable DXF file types: DXF types like ``['LINE', '3DFACE']`` which should be returned, ``None`` returns all supported types. errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: invalid or incomplete DXF file UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs Frt)rBrarJNrrorTSECTIONrC)rrGr{rKrarrrrYr\r[rrVrr}r~rZ)rrrJinfo prev_code prev_valueentitiesrfptaggerrtagsrtagrYr\rs rPrr"sI4 X 'DIJH&u-O hTDMM& IR"2&&*% 'C88DIIE194yT!W]]o%E!(l4.@!A -f 5 & 5 5 : &&, %+F5DKK$19(!2$ 9 J I:yA~* 2I J.IJ9( J I Is*?E CD? E D?6 E ?EE c #LKd}d}d}d}d}d}t|} t|D]{\} } | dk(r| dk(rnl| d k(r|dk(r | d k7r| d k(}nV| d k(r| d k(rd}nE| d k(r| dk(rd}n8|dk(rt| j}d}n|dk(r| j}d}| }}|dk\rd}d} g} t }t t|||D]}|j } |j} |r| dk(r | dk(r| r| y| dk(rut| rf| dj| vrUtttjt| }||s!|jjdk(r| r| |} |g} n| j!|| d k(r|dk(r |dk(r| dk(}| }| }yw)uIterate over all modelspace entities as :class:`DXFGraphic` objects in a single pass. Use this function to 'quick' iterate over modelspace entities of a **not** seekable binary DXF stream, filtering DXF types may speed up things if many entity types will be skipped. Args: stream: (not seekable) binary DXF stream types: DXF types like ``['LINE', '3DFACE']`` which should be returned, ``None`` returns all supported types. errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: Invalid or incomplete DXF file UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs Ncp1252rDrrFrsENDSECrTsHEADERsENTITIES s $DWGCODEPAGEENCODINGs$ACADVERVERSIONAC1021zutf-8rorrC)r{ binary_taggerrrrrrYr\r[rrrrVrr}r~rZ)streamrrJfetch_header_varrarMrrrrrYr\rrrrrs rPrrgs6'+HGIJH&u-O$V, e 19)+  QY9>ey.@ +H  QY5O3)  QY5K/(   +!%,,.1H#   *llnG#  %-((#'FD!OMM&(FCDxx  qyUh. Lqyt9a/!A!*gll<;M.NOF(0VZZ5J5Ja5O""(L!'u C  QY9>jI.E *H  3EsF"F$c#K t|j}|jj d}t ||r|j ||n|^#t$r tdwxYw#t$rYywxYww)NzInvalid group codes r)rreadlinerqr rstriprrIOError)rLrarJrYr\s rPrrs   ?4==?+MMO**73E9A Xf 5u   ?'*<>> ?   s:B A"AA:!B "A77A:: BB BB c|rltjt|}d|vr"|jd|jdd|vr"|jd|jd|St}|S)Nr:r4r6r2r0)SUPPORTED_TYPES intersectionsetadd)r requesteds rPr{r{sn #00U<  " MM( # MM( # y MM( # MM( # $ rRr)rrrJrGrr>)Nr)rrrrrJrGrr)rrrrrJrGrr)rLrraz Optional[str]rJrGrzIterator[DXFTag])rrrr)1 __future__rtypingrrrrrr r ior pathlibr ezdxf.lldxf.constr ezdxf.lldxf.extendedtagsrrezdxf.lldxf.tagwriterrezdxf.lldxf.taggerrrezdxf.filemanagementr ezdxf.lldxfrezdxf.entitiesrrrrrezdxf.entities.subentityrezdxf.tools.codepager__all__rrGrr>rfrrrrr{rrRrPrs #/9+>.!BB"2+ =B s KK\::z,0&*#BB "B B BN&*#W W "W W Wx##   ( rR