hkUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z mZddlmZmZddlmZmZddlmZdd lmZdd lmZmZer(dd lmZdd lmZdd l m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*e jVdZ,GddZ-gdZ.idddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6iZ/d7e0d8<Gd9d:Z1y);) annotations)IterableIteratorOptional TYPE_CHECKINGUnioncastN) Dictionary)factory is_dxf_object)const validator) EntitySpaceEntityDB) EntityQuery)UnderlayKeyGenerator)Auditor AuditError)Drawing)AbstractTagWriter) GeoData DictionaryVar DXFTagStorage DXFObjectImageDefUnderlayDefinitionDictionaryWithDefaultXRecord Placeholder)ImageDefReactorezdxfceZdZd&d'dZed(dZd)dZdfd*dZd+dZd,dZ d-d Z d.d Z d/d Z d0d Z d1d Zd.dZd2dZdZed0dZd3dZd4dZd5dZdZd6d7dZd8dZd9d:dZ d; ddZd?d@dZd?dAdZ dB dCdZdDdZdEdFdZ d3d Z! d& dGd!Z"dHd"Z# dI dJd#Z$dKdLd$Z%d&dMd%Z&y)NObjectsSectionNc||_t|_t|_||j t |yyN)docrunderlay_key_generatorr _entity_space_builditer)selfr&entitiess 6lib/python3.12/site-packages/ezdxf/sections/objects.py__init__zObjectsSection.__init__*s9&:&<#(]   KKX ' c.|jjS)z/Returns drawing entity database. (internal API))r&entitydbr+s r-r1zObjectsSection.entitydb1sxx   r/c|jS)z$Returns entity space. (internal API)r(r2s r-get_entity_spacezObjectsSection.get_entity_space6s!!!r/cyNT)ks r-zObjectsSection.:sDr/cN |jj}||r|S%r%)r'next)r+ checkfunckeys r-next_underlay_keyz ObjectsSection.next_underlay_key:s*--224C~ r/ctdt|}|jdk7s|jddk7rt j d|D]}|j j|y)NrSECTION)OBJECTSz0Critical structure error in the OBJECTS section.)r r<dxftype base_classr DXFStructureErrorr(add)r+r, section_headentitys r-r)zObjectsSection._build@suOT(^<    !Y .,2I2I!2LQ 3 ))B F    " "6 *r/c|jd|jj||jddy)z0Export DXF entity by `tagwriter`. (internal API)z 0 SECTION 2 OBJECTS rENDSECN) write_strr( export_dxf write_tag2)r+ tagwriters r-rNzObjectsSection.export_dxfPs5:; %%i0Q)r/c~tj|||j}|jj ||S)zCreate new DXF object, add it to the entity database and to the entity space. Args: _type: DXF type like `DICTIONARY` dxfattribs: DXF attributes as dict (internal API) )r create_db_entryr&r(rH)r+_type dxfattribs dxf_entitys r- new_entityzObjectsSection.new_entityVs6,,UJI  z*r/cp|jj||jj|y)zDRemove `entity` from entity space and destroy object. (internal API)N)r(remover1 delete_entityr+rJs r-rYzObjectsSection.delete_entityds( !!&) ##F+r/c|j}|jD]}|j||jjy)z&Delete all DXF objects. (internal API)N)r1r(rYclear)r+dbrJs r-delete_all_entitiesz"ObjectsSection.delete_all_entitiesis; ]]((F   V $)   "r/ct|rtjdtj d|j ddS)zcCreate a root dictionary. Has to be the first object in the objects section. (internal API)z=Can not create root dictionary in none empty objects section.zCreating ROOT dictionary.0Fowner hard_owned)lenr rGloggerdebugadd_dictionaryr2s r-setup_rootdictzObjectsSection.setup_rootdictpsF t9))O   01"""??r/cPfd}d fd }tD]}j|}t|tr%tj d|d|dk(r|nj |dt|tsotj|s|||y ) z0Setup required management tables. (internal API)cjjjd}j|jj}|j |||d<|d<y)NFra)rbNormalACAD_PLOTSTYLENAME)add_dictionary_with_defaultdxfhandleadd_placeholder set_default)plot_style_name_dict placeholderrootdictr+s r-setup_plot_style_name_tablezRObjectsSection.setup_object_management_tables..setup_plot_style_name_table~sp#'#C#Cll)) $D$ ..5I5M5M5T5T.UK , ,[ 9-8  *-AH) *r/cj|}t|trCjjj ||rt jd|d|yyy)Nzreset handle of table z to #)get isinstancer r&r1 reset_handlererf) table_namero new_tablertr+s r-restore_table_handlezKObjectsSection.setup_object_management_tables..restore_table_handles]! Z0I)Z0TXX5F5F5S5S66 5j\vhOP60r/z creating z dictionaryrlF)rc)roN)rzstrror}returnNone) _OBJECT_TABLE_NAMESrwrxr reinfo add_new_dictr}r is_handle)r+rtrur|nametables`` r-setup_object_management_tablesz-ObjectsSection.setup_object_management_tables{s B Q(D2:,,t2DE%, KK)D65 6+++-%%du%=%%)*=*=e*D$T%8(r/ct|r|jj|ytjd|j d)z/Add `entity` to OBJECTS section. (internal API)zinvalid DXF type z for OBJECTS sectionN)r r(rHr DXFTypeErrorrErZs r- add_objectzObjectsSection.add_objectsF     " "6 *$$#FNN$4#55IJ r/cT|j||}|j|dg|S)z+Add DXF object with reactor. (internal API)rb)rV set_reactors)r+rErT dxfobjects r-add_dxf_object_with_reactorz*ObjectsSection.add_dxf_object_with_reactors.OOGZ8  7 345r/c8|jjyr%)r(purger2s r-rzObjectsSection.purges   "r/cVt|r|jdS|jS)z[Returns the root DICTIONARY, or as AutoCAD calls it: the named DICTIONARY. r)rdr(rhr2s r-rtzObjectsSection.rootdicts+ t9%%a( (&&( (r/c,t|jS)zReturns an iterator of all DXF objects in the OBJECTS section.)r*r(r2s r-__iter__zObjectsSection.__iter__sD&&''r/c |j|S)aGet entity at `index`. The underlying data structure for storing DXF objects is organized like a standard Python list, therefore `index` can be any valid list indexing or slicing term, like a single index ``objects[-1]`` to get the last entity, or an index slice ``objects[:10]`` to get the first 10 or fewer objects as ``list[DXFObject]``. r4)r+indexs r- __getitem__zObjectsSection.__getitem__s!!%((r/c|t|tr |j|}||jvS#t$rYywxYw)zReturns ``True`` if `entity` stored in OBJECTS section. Args: entity: :class:`DXFObject` or handle as hex string F)rxr}r1KeyErrorr(rZs r- __contains__zObjectsSection.__contains__sH fc " v.++++  s / ;;c,tt||S)z>9#f+G,-$  f%) 7   )r/cN|jd||d}tt|S)zAdd new :class:`~ezdxf.entities.Dictionary` object. Args: owner: handle to owner as hex string. hard_owned: ``True`` to treat entries as hard owned. DICTIONARYrarT)rVr r )r+rbrcrJs r-rgzObjectsSection.add_dictionarys4 (! J''r/cH|jd|||d}td|S)zAdd new :class:`~ezdxf.entities.DictionaryWithDefault` object. Args: owner: handle to owner as hex string. default: handle to default entry. hard_owned: ``True`` to treat entries as hard owned. ACDBDICTIONARYWDFLT)rbdefaultrcrrrVr )r+rbrrcrJs r-rmz*ObjectsSection.add_dictionary_with_defaults8 !"(! +V44r/c.|jd||dS)zAdd a new :class:`~ezdxf.entities.DictionaryVar` object. Args: owner: handle to owner as hex string. value: value as string DICTIONARYVAR)rbvaluerrV)r+rbrs r-add_dictionary_varz!ObjectsSection.add_dictionary_var's$ %%(H  r/c,|jdd|iS)z}Add a new :class:`~ezdxf.entities.XRecord` object. Args: owner: handle to owner as hex string. XRECORDrbrrr+rbs r- add_xrecordzObjectsSection.add_xrecord3sygu5EFFr/c,|jdd|iS)zAdd a new :class:`~ezdxf.entities.Placeholder` object. Args: owner: handle to owner as hex string. ACDBPLACEHOLDERrbrrrs r-rpzObjectsSection.add_placeholder<s$ 7E*:  r/c tjj|d} |jd}||j_||j_||j_y#tj$rH|jd|jjj|||d}||jd<YywxYw)aySet raster variables. Args: frame: ``0`` = do not show image frame; ``1`` = show image frame quality: ``0`` = draft; ``1`` = high units: units for inserting images. This defines the real world unit for one drawing unit for the purpose of inserting and scaling images with an associated resolution. ===== =========================== mm Millimeter cm Centimeter m Meter (ezdxf default) km Kilometer in Inch ft Foot yd Yard mi Mile none None ===== =========================== (internal API), public interface :meth:`~ezdxf.drawing.Drawing.set_raster_variables` rACAD_IMAGE_VARSRASTERVARIABLES)rbframequalityunitsrN) r RASTER_UNITSrwrtrnrrr DXFKeyErrorrro)r+rrrunits_ raster_varss r-set_raster_variablesz#ObjectsSection.set_raster_variablesIs6((,,UA6 +--(9:K%*KOO !&-KOO #$*KOO !  ;::!!]]..55"&# ;K0;DMM+ , ;sA%%AC?Cc |jd}|jj|jj tj j|jjdfS#tj$rYywxYw)Nr)rrBnoner) rtr rrnrrREVERSE_RASTER_UNITSrwr)r+rs r-get_raster_variablesz#ObjectsSection.get_raster_variableswsv --(9:K OO ! ! OO # #  & & * *;??+@+@& I      sA00BBc" |jd}t||j_y#tj $rO|j d|jjjt|d}||jd<YywxYw)zSet wipeout variables. Args: frame: ``0`` = do not show image frame; ``1`` = show image frame (internal API) ACAD_WIPEOUT_VARSWIPEOUTVARIABLES)rbrrN)rtintrnrr rrro)r+r wipeout_varss r-set_wipeout_variablesz$ObjectsSection.set_wipeout_variabless 0==)<=L&)ZL   "   >;;"!]]..55 Z<L2>DMM- . >s,AB Bc |jd}|jjS#tj$rYywxYw)Nrr)rtr rrnr)r+rs r-get_wipeout_frame_settingz(ObjectsSection.get_wipeout_frame_settingsD ==)<=L%%%    s '==c||}|jjd}|jd|jj||d}|||<t d|S)aAdd an image definition to the objects section. Add an :class:`~ezdxf.entities.image.ImageDef` entity to the drawing (objects section). `filename` is the image file name as relative or absolute path and `size_in_pixel` is the image size in pixel as (x, y) tuple. To avoid dependencies to external packages, `ezdxf` can not determine the image size by itself. Returns a :class:`~ezdxf.entities.image.ImageDef` entity which is needed to create an image reference. `name` is the internal image name, if set to ``None``, name is auto-generated. Absolute image paths works best for AutoCAD but not really good, you have to update external references manually in AutoCAD, which is not possible in TrueView. If the drawing units differ from 1 meter, you also have to use: :meth:`set_raster_variables`. Args: filename: image file name (absolute path works best for AutoCAD) size_in_pixel: image size in pixel as (x, y) tuple name: image name for internal use, None for using filename as name (best for AutoCAD) ACAD_IMAGE_DICTIMAGEDEF)rbfilename image_sizerr)rtget_required_dictrrnror )r+r size_in_pixelr image_dict image_defs r- add_image_defzObjectsSection.add_image_defsl8 <D]]445FG 44 #..$+5 % 4J **r/cF|jd||d}td|S)zZAdd required IMAGEDEF_REACTOR object for IMAGEDEF object. (internal API) IMAGEDEF_REACTOR)rb image_handlerr r)r+rimage_def_reactors r-add_image_def_reactorz$ObjectsSection.add_image_def_reactors7 !OO % ,, %'899r/ct|j}|dvr d|d}|d}ntjd|d||dk(rd}n |d k(rd }nd }|jj ||j |j j||d  }|jfd}||<td|S)aAdd an :class:`~ezdxf.entities.underlay.UnderlayDefinition` entity to the drawing (OBJECTS section). `filename` is the underlay file name as relative or absolute path and `fmt` as string (pdf, dwf, dgn). The underlay definition is required to create an underlay reference. Args: filename: underlay file name fmt: file format as string ``'pdf'|'dwf'|'dgn'`` name: pdf format = page number to display; dgn format = ``'default'``; dwf: ???? )PDFDWFDGNACAD_ DEFINITIONS DEFINITIONzUnsupported file format: ''r1rrModel)rbrrrc |vSr%r8)r9 underlay_dicts r-r:z1ObjectsSection.add_underlay_def..s q /Er/r) upperr DXFValueErrorrtrrVrnror?r ) r+rfmtrunderlay_dict_nameunderlay_def_entity underlay_defr>rs @r-add_underlay_defzObjectsSection.add_underlay_defsiik ' '#([!9 %(E"4 %%(B3%q&IJ J <e|  778JK  &**11$' $$%EF) c(,77r/cL|i}||d<td|jd|S)aCreates a new :class:`GeoData` entity and replaces existing ones. The GEODATA entity resides in the OBJECTS section and NOT in the layout entity space, and it is linked to the layout by an extension dictionary located in BLOCK_RECORD of the layout. The GEODATA entity requires DXF version R2010+. The DXF Reference does not document if other layouts than model space supports geo referencing, so getting/setting geo data may only make sense for the model space layout, but it is also available in paper space layouts. Args: owner: handle to owner as hex string dxfattribs: DXF attributes for :class:`~ezdxf.entities.GeoData` entity rbrGEODATA)r r)r+rbrTs r- add_geodatazObjectsSection.add_geodatas3  J# 7It?? :VWWr/c|(|jJdt|j}t||}|j|S)z;Validate and recreate the integrity of the OBJECTS section.zvalid document required)r&r _Sanitizerexecute)r+r sanitizers r-rzObjectsSection.reorgsH ?88' B)B B'dhh'Gw- r/r%)r&rr,zOptional[Iterable[DXFObject]])r~r)r~r)r~r})r,Iterator[DXFObject]r~r)rPrr~r)rSr}rTdictr~r)rJrr~rr~r)r~r )rtr r~r)rEr}r~r)r~r)r~r)r~r)*)rr}r~r)rrr~r)r`T)rbr}rcboolr~r )r`r`T)rcrr~r)r`)rbr}rr}r~r)r`)rbr}r~r)rbr}r~r)rrBm)rrrrrr}r~r)r~ztuple[int, int, str])r)rrr~r)rr}rztuple[int, int]r Optional[str]r~r)rr}r~r )pdfN)rr}rr}rrr~r)r`N)rbr}r~r)rzOptional[Auditor]r~r)'__name__ __module__ __qualname__r.propertyr1r5r?r)rNrVrYr^rhrrrrrtrrrrrrrgrmrrrprrrrrrrrrr8r/r-r#r#)sh(!!"+9 + * , # @'9R # ))'( ) ,.*2($:>5265 5*  G  >A,+,+'*,+7:,+ ,+\  0,&# (+(+'(+ (+  (+T :FJ*8*8"%*85B*8 *8XX*r/r#) ACAD_COLOR ACAD_GROUP ACAD_LAYOUT ACAD_MATERIALACAD_MLEADERSTYLEACAD_MLINESTYLEACAD_PLOTSETTINGSrlACAD_SCALELISTACAD_TABLESTYLEACAD_VISUALSTYLErDBCOLORrGROUPrrACAD_DETAILVIEWSTYLEACDBDETAILVIEWSTYLErLAYOUTr MATERIALr  MLEADERSTYLEr  MLINESTYLEr  PLOTSETTINGSACAD_RENDER_ACTIVE_SETTINGSMENTALRAYRENDERSETTINGSr SCALEACAD_SECTIONVIEWSTYLEACDBSECTIONVIEWSTYLEr TABLESTYLEACAD_PDFDEFINITIONS PDFDEFINITIONACAD_DWFDEFINITIONS DWFDEFINITIONACAD_DGNDEFINITIONS DGNDEFINITIONr VISUALSTYLEAcDbVariableDictionaryrzdict[str, str]KNOWN_DICT_CONTENTcPeZdZd dZd dZd ddZddZddZddZdZ dZ d Z y )rcP||_||_|j|_d|_yr7)objectsrrtremoved_entity)r+rr*s r-r.z_Sanitizer.__init__Gs%  (( "r/c#XK|jD]}t|ts|ywr%)r*rxr )r+ds r- dictionariesz_Sanitizer.dictionariesMs#A!Z(s **c|j|jd}d|_|jrC||kr>|dz }d|_|j|j |jr||kr>|j y)NrTrBF)+restore_owner_handles_of_dictionary_entriesvalidate_known_dictionariesr+remove_orphaned_dictionaries audit_objectscreate_required_structures)r+ max_loopsloopss r-rz_Sanitizer.executeRs 88: ((*"!!ei&7 QJE"'D   - - /    !!ei&7 '')r/c  dfd } fd}jj}jD]g jjj D]\ t tr|jjd}|k(r?|j|}t |tr|jr|y| D] j y)Ncj_jjtj ddy)NzFixed invalid owner handle of .)rnrbrrrINVALID_OWNER_HANDLE) dict_handlerJr+sr-reclaim_entityzN_Sanitizer.restore_owner_handles_of_dictionary_entries..reclaim_entityas6*FJJ  LL $ $//0: r/c jjjtjddt dy)NzRemoved invalid key z in r9)appendrrrr:r}) dictionaryr>rr+sr- purge_keyzI_Sanitizer.restore_owner_handles_of_dictionary_entries..purge_keyhsA LL  LL $ $//&se4J/@B r/rbr) rr1r.rnroitemsrxr}rwr find_keydiscard) r+r<r@r1 owner_handle parent_dictr;r?rJr>rs ` @@@@@r-r0z6_Sanitizer.restore_owner_handles_of_dictionary_entries`s  <<((++-J!E$..//K)//1 Vfc*K%zz~~g6 ;.&ll<8 k:6;;O;OPV;WK"$ 2""3'#.r/cLfd}|jj|j}|jD]g|urj j j d}||.kill_dictionarys   Z (  # # %r/rb) rr1rtr.rwrnrxr rB)r+rIrtrbr>r?r1s @@r-r2z'_Sanitizer.remove_orphaned_dictionariess &<<((==++-JX%LL!3!3G!<=E}  !eZ0..,C!!.r/c ddlm}|j}tj D]\}}|j j |}t|ts2g}|j D]t\}}t||s|j|k7s'|jtjd|d|d|d||j||j|v|D]}|j|y)Nr) DXFEntityzRemoved invalid type z from , expected type )ezdxf.entitiesrKrr'rArtrwrxr rErrrr>rrC) r+rKr dict_name expected_type object_dict purge_keysr>entrys r-r1z&_Sanitizer.validate_known_dictionariess,,,(:(@(@(B $I}--++I6Kk:6$&J)//1 UeY/EMMO}4T''"==/wf[M9+V))69 %%c*MM%(2"##C(")Cr/c|jj|j|jj}|jj |jj |jj |jj |jj |jj |jj yr%) r*rrtr& materialsupdate_object_dictcreate_required_entries mline_stylesmleader_stylesgroups)r+r&s r-r4z%_Sanitizer.create_required_structuress 33DMMBll ((* --/ ++- 002 --/ 224 %%'r/c|jj|jjjyr%)rempty_trashcanr1rr2s r-cleanup_entitydbz_Sanitizer.cleanup_entitydbs( ##% ##%r/c|j|j}t|j}|jD]}|j j ds@|jtjdt|d|j|^|j||j||j|t|jk7rd|_yy)NrbzDeleted z entity without owner handle.rT)r\rrdr1r*rnhasattrrrr:r}rcheck_owner_existrr[r+)r+rcurrent_db_sizerJs r-r3z_Sanitizer.audit_objectss ,,g../llF::%%g.###88&s6{m3PQ$ f%  % %f - LL !#  c'"2"23 3"&D  4r/N)rrr*r#r~r)r~zIterator[Dictionary])dr) rrrr.r.rr0r2r1r4r\r3r8r/r-rrFs0#  *"(H"2)* (&'r/r)2 __future__rtypingrrrrrr loggingezdxf.entities.dictionaryr rMr r ezdxf.lldxfr rezdxf.entitydbrr ezdxf.queryrezdxf.tools.handler ezdxf.auditrrezdxf.documentrezdxf.lldxf.tagwriterrrrrrrrrrrezdxf.entities.imager getLoggerrer#rr'__annotations__rr8r/r-rpsd#01(0#3+&7   5   7 #vvr &)&'&z&1 & 8 & Z &&|&&"#<&g&3&|&?&?& ?!&" #&$o%&N,O'O'r/