'j=ddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZerdd lmZhdZGddZGddZ dS)) annotations)OptionalIterable TYPE_CHECKINGIterator)contextmanager)HandleGenerator)is_valid_handle) DXFEntity) DXFObject) AuditErrorAuditor)DXFInternalEzdxfError)factory) EntityQuery) default_copy)AbstractTagWriter>EOFCLASSTABLEENDSECENDTABSECTION ACDSRECORD ACDSSCHEMAceZdZdZGddZd0dZd1d Zd2d Zd3d Zd4dZ d5dZ d6dZ d7dZ d8dZ d9dZd:dZd;dZdd#Zd?d%Zed?d&Zd0d'Zd@d)ZdAd*ZdBdCd.Zd/S)DEntityDBzA simple key/entity database. Every entity/object, except tables and sections, are represented as DXFEntity or inherited types, these entities are stored in the DXF document database, database-key is the `handle` as string. c(eZdZdZd dZd dZdZd S) EntityDB.Trashcanz8Store handles to entities which should be deleted later.dbrcD|j|_t|_dSN) _databaseset_handles)selfr s H/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/entitydb.py__init__zEntityDB.Trashcan.__init__0s\DN&)eeDMMMhandlestrc:|j|dS)zPut handle into trashcan to delete the entity later, this is required for deleting entities while iterating the database. N)r%addr&r*s r'r-zEntityDB.Trashcan.add4s  M  f % % % % %r)c|j}|jD];}||}|r|jr|||vr||=<|jdS)zfRemove handles in trashcan from database and destroy entities if still alive. N)r#r%getis_alivedestroyclear)r&r r*entitys r'r3zEntityDB.Trashcan.clear:swB- # #%fo%NN$$$R<<6 M   ! ! ! ! !r)N)r r)r*r+)__name__ __module__ __qualname____doc__r(r-r3r)r'Trashcanr-sQFF , , , , & & & & " " " " "r)r:returnNonecHi|_t|_d|_dS)NF)r#r handleslockedr&s r'r(zEntityDB.__init__Is!/1&(( ! r)r*r+r c|j|S)ziGet entity by `handle`, does not filter destroyed entities nor entities in the trashcan. )r#r.s r' __getitem__zEntityDB.__getitem__Os~f%%r)r4cht|tsJt|t|tsJt||js Jd|jrt d|dkst|std|d||j |<dS)zSet `entity` for `handle`.zCan not store destroyed entity.Locked entity database.0zInvalid handle .N) isinstancer+typer r1r?rr ValueErrorr#)r&r*r4s r' __setitem__zEntityDB.__setitem__Us&#&&44V 44&&),,::d6ll::,AA AAA ; C'(ABB B S== 7 7=8v88899 9!'vr)cB|jrtd|j|=dS)zkDelete entity by `handle`. Removes entity only from database, does not destroy the entity. rDN)r?rr#r.s r' __delitem__zEntityDB.__delitem__as. ; C'(ABB B N6 " " "r)boolct|dSt|tsJt|||jvS)z'``True`` if database contains `handle`.NF)rGr+rHr#r.s r' __contains__zEntityDB.__contains__is= >5&#&&44V 44&''r)intc*t|jS)zCount of database items.)lenr#r@s r'__len__zEntityDB.__len__ps4>"""r) Iterator[str]c*|S)zjIterable of all handles, does filter destroyed entities but not entities in the trashcan. )keysr@s r'__iter__zEntityDB.__iter__tsyy{{r)Optional[DXFEntity]c6|j|S)zoReturns entity for `handle` or ``None`` if no entry exist, does not filter destroyed entities. )r#r0r.s r'r0z EntityDB.getzs~!!&)))r)cN |j}||jvr|S%)zReturns next unique handle.)r>nextr#r.s r' next_handlezEntityDB.next_handles1 \&&((FT^++  r) Iterable[str]c>d|DS)z8Iterable of all handles, does filter destroyed entities.c3 K|] \}}|V dSr"r9.0r*r4s r' z EntityDB.keys..&::>66::::::r)itemsr@s r'rVz EntityDB.keys::TZZ\\::::r)Iterable[DXFEntity]c>d|DS)z9Iterable of all entities, does filter destroyed entities.c3 K|] \}}|V dSr"r9r`s r'rbz"EntityDB.values..rcr)rdr@s r'valueszEntityDB.valuesrfr)Iterable[tuple[str, DXFEntity]]cHd|jDS)zZIterable of all (handle, entities) pairs, does filter destroyed entities. c32K|]\}}|j ||fVdSr"r1r`s r'rbz!EntityDB.items..sG   V       r)r#rer@s r'rezEntityDB.itemss1  "&."6"6"8"8    r)c<|tvr|jj|||jj<dS|jj}|)|}|||||<t |dr||dSdS)zAdd `entity` to database, assigns a new handle to the `entity` if :attr:`entity.dxf.handle` is ``None``. Adding the same entity multiple times is possible and creates only a single database entry. Nadd_sub_entities_to_entitydb)dxftypeDATABASE_EXCLUDEdxfr*r\ update_handlehasattrrqr&r4r*s r'r-z EntityDB.adds >>  / / /z ,+1VZ&' Fj' >%%''F   ( ( (V  69 : : 6  / / 5 5 5 5 5 6 6r)cZ|jr#||jj=|dSdS)z7Remove `entity` from database and destroy the `entity`.N)r1rtr*r2r&r4s r' delete_entityzEntityDB.delete_entitys9 ? VZ&' NN       r)c|jr\t|dr|fd|jj} j|=d|j_dS#t $rYdSwxYwdS)z?Discard `entity` from database without destroying the `entity`.process_sub_entitiesc.|Sr")discard)er&s r'z"EntityDB.discard..sdll1oor)N)r1rvr|rtr*r#KeyErrorrws` r'r~zEntityDB.discards ? v566 G++,E,E,E,EFFFZ&F N6*$( !!!      sA A"!A"c*|j}| Jd|}|t}||j_t j||t|tr|j ||S)aDuplicates `entity` and its sub entities (VERTEX, ATTRIB, SEQEND) and store them with new handles in the entity database. Graphical entities have to be added to a layout by :meth:`~ezdxf.layouts.BaseLayout.add_entity`. DXF objects will automatically added to the OBJECTS section. A new owner handle will be set by adding the duplicated entity to a layout. Raises: CopyNotSupported: copying of `entity` is not supported Nzvalid DXF document required) copy_strategy) docr\copyrrtr*rbindrGr objects add_object)r&r4r new_handle new_entitys r'duplicate_entityzEntityDB.duplicate_entitysj =%%'' & , G G *  Z%%% j) , , / K " ": . . .r)auditorrc |jdus Jdg}|5}|D]\}}t|sO|t jd|d|d||||j dkr4d|j |<||| | dddn #1swxYwY| |D]}|j d}|8|t jd|d Vt|r|d kr;|t jd|d|d|||<dS) aRestore database integrity: - restore database entries with modified handles (key != entity.dxf.handle) - remove entities with invalid handles - empty trashcan - destroy all entities in the trashcan - removes destroyed database entries (purge) FzDatabase is locked!Removed entity z with invalid handle "z" from entity database.)codemessager*Nz% without handle from entity database.rE)r?trashcanrer fixed_errorr INVALID_ENTITY_HANDLErrr-rtr0r#appendpurge)r&r add_entitiestrashr*r4s r'auditzEntityDB.auditsm{e###%:### ]]__ 0"&**,, 0 0&v..&'''=!C&..2B2B!C!C#)!C!C!C( IIf%%%VZ^^H5555.2DN6*IIf%%% ''/// 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0$ " " "FZ^^H--F~###9-fnn.>.>---$ "6** fmm###9?fnn.>.>??%???$ !DLL! " "sC DD Drc6t|S)z_Returns a new trashcan, empty trashcan manually by: : func:`Trashcan.clear()`. )rr:r@s r' new_trashcanzEntityDB.new_trashcans  &&&r)c#bK|}|V|dS)zoReturns a new trashcan in context manager mode, trashcan will be emptied when leaving context. N)rr3)r& trashcan_s r'rzEntityDB.trashcans: %%''  r)c`|j}d|D}|D]}||=dS)z^Remove all destroyed entities from database, but does not empty the trashcan. c&g|]\}}|j |Sr9rnr`s r' z"EntityDB.purge..%s#WWW>66vWWWWr)Nro)r&r dead_handlesr*s r'rzEntityDB.purgesH ^WWRXXZZWWW "  F6   r)set[str]cXtd|DS)Nc3>K|]}|VdSr")rr)rar4s r'rbz,EntityDB.dxf_types_in_use..*s,@@6>>##@@@@@@r))r$rjr@s r'dxf_types_in_usezEntityDB.dxf_types_in_use)s'@@$++--@@@@@@r)c||jvrdS||||j_||dS)z|Try to reset the entity handle to a certain value. Returns ``True`` if successful and ``False`` otherwise. FT)r#r~rtr*r-rws r' reset_handlezEntityDB.reset_handle,sI T^ # #5 V"  tr)*queryrcdtd|jD|S)zEntity query over all entities in the DXF document. Args: query: query string .. seealso:: :ref:`entity query string` and :ref:`entity queries` c3(K|] }|j |VdSr"rnrars r'rbz!EntityDB.query..Cs)MM!!*MAMMMMMMr))rr#rj)r&rs r'rzEntityDB.query8s1MMt~'<'<'>'>MMMuUUUr)Nr;r<)r*r+r;r )r*r+r4r r;r<)r*r+r;r<r*r+r;rMr;rP)r;rT)r*r+r;rX)r;r+)r;r])r;rg)r;rkr4r r;r<)r4r r;r )rr)r;r)r;r)r4r r*r+r;rM)r)rr+r;r)r5r6r7r8r:r(rBrJrLrOrSrWr0r\rVrjrer-rzr~rrrrrrrrrr9r)r'rr$s/""""""""8"""" &&&& ( ( ( (####((((#### **** ;;;;;;;;    6666.    20"0"0"0"d'''' ^AAAA     V V V V V V Vr)rceZdZdZd#d$dZd%dZd&d Zd'd Zd(dZdZ d)dZ d*dZ d+dZ d)dZ d,dZd-d.dZd/dZd0d"ZdS)1 EntitySpaceat An :class:`EntitySpace` is a collection of :class:`~ezdxf.entities.DXFEntity` objects, that stores only references to :class:`DXFEntity` objects. The :class:`~ezdxf.layouts.Modelspace`, any :class:`~ezdxf.layouts.Paperspace` layout and :class:`~ezdxf.layouts.BlockLayout` objects have an :class:`EntitySpace` container to store their entities. NentitiesOptional[Iterable[DXFEntity]]cJ|rtd|Dng|_dS)Nc3(K|] }|j |VdSr"rnrs r'rbz'EntitySpace.__init__..Ss)33q 3333333r)listr)r&rs r'r(zEntitySpace.__init__Qs17? GD33H333 3 3 3R r)r;Iterator[DXFEntity]c$d|jDS)z5Iterable of all entities, filters destroyed entities.c3(K|] }|j |VdSr"rnrs r'rbz'EntitySpace.__iter__..Xs)77aAJ7777777r)rr@s r'rWzEntitySpace.__iter__Vs774=7777r)r c|j|S)aGet entity at index `item` :class:`EntitySpace` has a standard Python list like interface, therefore `index` can be any valid list indexing or slicing term, like a single index ``layout[-1]`` to get the last entity, or an index slice ``layout[:10]`` to get the first 10 or fewer entities as ``list[DXFEntity]``. Does not filter destroyed entities. rr&indexs r'rBzEntitySpace.__getitem__Zs}U##r)rPc*t|jS)z/Count of entities including destroyed entities.)rRrr@s r'rSzEntitySpace.__len__fs4=!!!r)r*r+rMcttsJttfd|DS)z@``True`` if `handle` is present, does filter destroyed entities.c38K|]}|jjkVdSr")rtr*)rarr*s r'rbz)EntitySpace.has_handle..ms,88a15<6)888888r))rGr+rHanyr.s `r' has_handlezEntitySpace.has_handlejsJ&#&&44V 44&88884888888r)c.t||_dS)z0Remove all destroyed entities from entity space.Nrr@s r'rzEntitySpace.purgeosT  r)r4r<ct|tsJt||js Jd|j|dS)z Add `entity`.z Can not store destroyed entitiesN)rGr rHr1rrrys r'r-zEntitySpace.addssX&),,::d6ll::,BB BBB V$$$$$r)rgc:|D]}||dS)zAdd multiple `entities`.N)r-)r&rr4s r'extendzEntitySpace.extendys0  F HHV      r) tagwriterrcTt|D]}||dS)zRExport all entities into DXF file by `tagwriter`. (internal API) N)iter export_dxf)r&rr4s r'rzEntitySpace.export_dxf~s: 4jj ) )F   i ( ( ( ( ) )r)c:|j|dS)zRemove `entity`.N)rremoverys r'rzEntitySpace.removes V$$$$$r)c,t|_dS)zRemove all entities.Nrr@s r'r3zEntitySpace.clears r)rc6|j|Sr")rpoprs r'rzEntitySpace.pops}  '''r)c<|j||dSr")rinsert)r&rr4s r'rzEntitySpace.inserts  UF+++++r)rrc>|jj}g}|D]T}|jj}|||ur9|t jd|d||U|sdS|D]0}|j || 1dS)Nrz8 with a conflicting handle and without a database entry.) entitydbr0rtr*rr REMOVED_INVALID_DXF_OBJECTrrr _silent_kill)r&rdb_getrr4r*s r'rzEntitySpace.audits!%!# % %FZ&FVVF^^++##9'f'''  V$$$  F " "F M  ( ( (    ! ! ! !  " "r)r")rr)r;r)r;r rrr)rrgr;r<)rrr;r<r)r)rrPr;r )rrPr4r r;r<)rrr;r<)r5r6r7r8r(rWrBrSrrr-rrrr3rrrr9r)r'rrFs;     8888 $ $ $ $""""9999 ###%%%%  ))))%%%% (((((,,,,""""""r)rN)! __future__rtypingrrrr contextlibrezdxf.tools.handler ezdxf.lldxf.typesr ezdxf.entities.dxfentityr ezdxf.entities.dxfobjr ezdxf.auditr rezdxf.lldxf.constrezdxf.entitiesr ezdxf.queryrezdxf.entities.copyrezdxf.lldxf.tagwriterrrsrrr9r)r'rs#""""" &%%%%%......------......++++++++++++++333333""""""######,,,,,,8777777   _V_V_V_V_V_V_V_VD j"j"j"j"j"j"j"j"j"j"r)