h~?ddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZerdd lmZhdZGddZGddZ y)) annotations)OptionalIterable TYPE_CHECKINGIterator)contextmanager)HandleGenerator)is_valid_handle) DXFEntity) DXFObject) AuditErrorAuditor)DXFInternalEzdxfError)factory) EntityQuery) default_copy)AbstractTagWriter>EOFCLASSTABLEENDSECENDTABSECTION ACDSRECORD ACDSSCHEMAceZdZdZGddZddZddZddZddZd dZ d!d Z d"d Z d#d Z d$d Z d%d Zd&dZd'dZd(dZd(dZd(dZd)dZd*dZd+dZed+dZddZd,dZd-dZd.d/dZy)0EntityDBzA 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&eZdZdZddZddZdZy)EntityDB.Trashcanz8Store handles to entities which should be deleted later.cD|j|_t|_yN) _databaseset_handles)selfdbs .lib/python3.12/site-packages/ezdxf/entitydb.py__init__zEntityDB.Trashcan.__init__0s\\DN&)eDMc:|jj|y)zPut handle into trashcan to delete the entity later, this is required for deleting entities while iterating the database. N)r$addr%handles r'r+zEntityDB.Trashcan.add4s MM  f %r)c|j}|jD]9}|j|}|r|jr|j ||vs7||=;|jj y)zfRemove handles in trashcan from database and destroy entities if still alive. N)r"r$getis_alivedestroyclear)r%r&r-entitys r'r2zEntityDB.Trashcan.clear:s[B--fooNN$R<6 ( MM   !r)N)r&r)r-str)__name__ __module__ __qualname____doc__r(r+r2r)r'Trashcanr-sF , & "r)r:c>i|_t|_d|_y)NF)r"r handleslockedr%s r'r(zEntityDB.__init__Is/1&( ! 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)c4t|tsJt|t|tsJt||jsJd|j r t d|dk(s t|std|d||j|<y)zSet `entity` for `handle`.zCan not store destroyed entity.Locked entity database.0zInvalid handle .N) isinstancer4typer r0r=rr ValueErrorr")r%r-r3s r' __setitem__zEntityDB.__setitem__Us&#&4V 4&&),:d6l:,A AA ;;'(AB B S= 7vha89 9!'vr)cL|jr td|j|=y)zkDelete entity by `handle`. Removes entity only from database, does not destroy the entity. rBN)r=rr"r,s r' __delitem__zEntityDB.__delitem__as# ;;'(AB B NN6 "r)cd|yt|tsJt|||jvS)z'``True`` if database contains `handle`.F)rEr4rFr"r,s r' __contains__zEntityDB.__contains__is2 >&#&4V 4&''r)c,t|jS)zCount of database items.)lenr"r>s r'__len__zEntityDB.__len__ps4>>""r)c"|jS)zjIterable of all handles, does filter destroyed entities but not entities in the trashcan. )keysr>s r'__iter__zEntityDB.__iter__tsyy{r)c8|jj|S)zoReturns entity for `handle` or ``None`` if no entry exist, does not filter destroyed entities. )r"r/r,s r'r/z EntityDB.getzs~~!!&))r)cZ |jj}||jvr|S+)zReturns next unique handle.)r<nextr"r,s r' next_handlezEntityDB.next_handles-\\&&(FT^^+ r)c0d|jDS)z8Iterable of all handles, does filter destroyed entities.c3&K|] \}}| ywr!r9.0r-r3s r' z EntityDB.keys..:\>66\itemsr>s r'rQz EntityDB.keys:TZZ\::r)c0d|jDS)z9Iterable of all entities, does filter destroyed entities.c3&K|] \}}| ywr!r9rYs r'r[z"EntityDB.values..r\r]r^r>s r'valueszEntityDB.valuesr`r)cDd|jjDS)zZIterable of all (handle, entities) pairs, does filter destroyed entities. c3BK|]\}}|jr||fywr!r0rYs r'r[z!EntityDB.items..s( "8V "8s)r"r_r>s r'r_zEntityDB.itemss! "&.."6"6"8  r)cH|jtvr0|jj|||jj<y|jj}|!|j }|j ||||<t |dr|j|yy)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-rV update_handlehasattrrhr%r3r-s r'r+z EntityDB.adds >> / /zz  ,+1VZZ&&' jj'' >%%'F   (V  69 :  / / 5 ;r)cl|jr(||jj=|jyy)z7Remove `entity` from database and destroy the `entity`.N)r0rkr-r1r%r3s r' delete_entityzEntityDB.delete_entitys* ??VZZ&&' NN  r)c|jrVt|dr|jfd|jj} j |=d|j_yy#t $rYywxYw)z?Discard `entity` from database without destroying the `entity`.process_sub_entitiesc&j|Sr!)discard)er%s r'z"EntityDB.discard..s dll1or)N)r0rmrsrkr-r"KeyErrorrns` r'ruzEntityDB.discardsi ??v56++,EFZZ&&F NN6*$( !   sA%% A10A1c |j}|Jd|j}|jt}||j_t j||t|tr|jj||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 zvalid DXF document required) copy_strategy) docrVcopyrrkr-rbindrEr objects add_object)r%r3r{ new_handle new_entitys r'duplicate_entityzEntityDB.duplicate_entityswjj= ==%%' & , G *  Z% j) , KK " ": .r)c Z|jdusJdg}|j5}|jD]\}}t|sG|j t j d|jd|d|j|||jjdk7swd|j|<|j||j| ddd|j|D]}|jjd}|4|j t j d|jd Tt|r|d k(r7|j t j d|jd|d|||<y#1swYxYw) 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.rC)r=trashcanr_r fixed_errorr INVALID_ENTITY_HANDLErir+rkr/r"appendpurge)r%auditor add_entitiestrashr-r3s r'auditzEntityDB.audits{{e#:%::# ]]_"&**,&v.'''=="1&..2B1CD##)(*A!C( IIf%VZZ^^H55.2DNN6*IIf% ''/#/$ "FZZ^^H-F~###99-fnn.>-?@,-$ "6*fm###99-fnn.>-?@%h&=?$ !DL!#)_sBF!03F!!F*c,tj|S)z_Returns a new trashcan, empty trashcan manually by: : func:`Trashcan.clear()`. )rr:r>s r' new_trashcanzEntityDB.new_trashcans  &&r)c#TK|j}||jyw)zoReturns a new trashcan in context manager mode, trashcan will be emptied when leaving context. N)rr2)r% trashcan_s r'rzEntityDB.trashcans' %%'  s&(c|j}|jDcgc]\}}|jr|}}}|D]}||=ycc}}w)z^Remove all destroyed entities from database, but does not empty the trashcan. N)r"r_r0)r%r&r-r3 dead_handless r'rzEntityDB.purgesH ^^57XXZWZ>66vZ W"F6 #Xs AAcBtd|jDS)Nc3<K|]}|jywr!)ri)rZr3s r'r[z,EntityDB.dxf_types_in_use..*s@-6>>#-s)r#rcr>s r'dxf_types_in_usezEntityDB.dxf_types_in_use)s@$++-@@@r)c||jvry|j|||j_|j |y)z|Try to reset the entity handle to a certain value. Returns ``True`` if successful and ``False`` otherwise. FT)r"rurkr-r+rns r' reset_handlezEntityDB.reset_handle,s: T^^ # V"  r)cXtd|jjD|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|]}|js|ywr!rfrZrvs r'r[z!EntityDB.query..CsM'>!!**A'>)rr"rc)r%querys r'rzEntityDB.query8s$Mt~~'<'<'>MuUUr)NreturnNone)r-r4rr )r-r4r3r rr)r-r4rrr-r4rboolrint)rz Iterator[str])r-r4rzOptional[DXFEntity])rr4)rz Iterable[str])rIterable[DXFEntity])rzIterable[tuple[str, DXFEntity]]r3r rr)r3r rr )rr)rr)rzset[str])r3r r-r4rr)*)rr4rr)r5r6r7r8r:r(r@rHrJrLrOrRr/rVrQrcr_r+rqrurrrrrrrrrr9r)r'rr$s""8" & (#(# * ;; 6. 20"d' A  Vr)rceZdZdZdddZddZddZddZddZdZ dd Z dd Z dd Z dd Z dd ZdddZddZddZy) 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. NcF|rtd|D|_yg|_y)Nc3:K|]}|js|ywr!rfrs r'r[z'EntitySpace.__init__..Ss3Hq Hrlistentities)r%rs r'r(zEntitySpace.__init__Qs#7?D3H3 3 EG r)c(d|jDS)z5Iterable of all entities, filters destroyed entities.c3:K|]}|js|ywr!rfrs r'r[z'EntitySpace.__iter__..Xs7=aAJJ=rrr>s r'rRzEntitySpace.__iter__Vs74==77r)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'r@zEntitySpace.__getitem__Zs}}U##r)c,t|jS)z/Count of entities including destroyed entities.)rNrr>s r'rOzEntitySpace.__len__fs4==!!r)clttsJttfd|DS)z@``True`` if `handle` is present, does filter destroyed entities.c3PK|]}|jjk(ywr!)rkr-)rZrvr-s r'r[z)EntitySpace.has_handle..ms84a155<<6)4s#&)rEr4rFanyr,s `r' has_handlezEntitySpace.has_handlejs-&#&4V 4&84888r)c$t||_y)z0Remove all destroyed entities from entity space.Nrr>s r'rzEntitySpace.purgeos T  r)ct|tsJt||jsJd|jj |y)z Add `entity`.z Can not store destroyed entitiesN)rEr rFr0rrrps r'r+zEntitySpace.addss?&),:d6l:,B BB V$r)c4|D]}|j|y)zAdd multiple `entities`.N)r+)r%rr3s r'extendzEntitySpace.extendysF HHV r)cFt|D]}|j|y)zRExport all entities into DXF file by `tagwriter`. (internal API) N)iter export_dxf)r% tagwriterr3s r'rzEntitySpace.export_dxf~s 4jF   i (!r)c:|jj|y)zRemove `entity`.N)rremoverps r'rzEntitySpace.removes V$r)c"t|_y)zRemove all entities.Nrr>s r'r2zEntitySpace.clears  r)c8|jj|Sr!)rpoprs r'rzEntitySpace.pops}}  ''r)c<|jj||yr!)rinsert)r%rr3s r'rzEntitySpace.inserts UF+r)cX|jj}g}|D]X}|jj}|||us$|j t j d|d|j|Z|sy|D]-}|jj||j/y)Nrz8 with a conflicting handle and without a database entry.) entitydbr/rkr-rr REMOVED_INVALID_DXF_OBJECTrrr _silent_kill)r%rdb_getrr3r-s r'rzEntitySpace.audits!!%%!#FZZ&&FVF^+##99%fX.&'  V$  F MM  (    ! r)r!)rzOptional[Iterable[DXFEntity]])rzIterator[DXFEntity])rr rrr)rrrr)rrrrr))rrrr )rrr3r rr)rrrr)r5r6r7r8r(rRr@rOrrr+rrrr2rrrr9r)r'rrFsM 8 $"9 #%  )% (,"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.tagwriterrrjrrr9r)r'rs]# &.-.++3"#,7 _V_VD j"j"r)