hNpLUddlmZddlmZmZmZmZmZddlZddl m Z m Z m Z m Z ddl mZddlmZmZmZddlmZdd lmZmZmZeegd fZiZd ed <ed ZdGdZe j>f dHdZ e jBf dIdZ"dJdZ#dZ$GddZ%e%Z&ded<Gdd e%Z' dKdZ(e$Gdde'Z)e$Gdde'Z*Gdde'Z+e$Gdd e+Z,e$Gd!d"e+Z-e$Gd#d$e'Z.e$Gd%d&e+Z/e$Gd'd(e+Z0e$Gd)d*e+Z1e$Gd+d,e+Z2e$Gd-d.e+Z3e$Gd/d0e3Z4e$Gd1d2e+Z5e$Gd3d4e+Z6e$Gd5d6e+Z7e$Gd7d8e+Z8e$Gd9d:e+Z9e$Gd;de:Z;e$Gd?d@e+Z<GdAdBejzZ>GdCdDe>Z?GdEdFe>Z@y)L) annotations)CallableTypeAnySequenceIteratorN)sabsatconsthdr)Features) DataLoaderAbstractEntity DataExporter) EncodedData)Matrix44Vec3NULLVEC AcisEntityzdict[str, Type[AcisEntity]] ENTITY_TYPESinfct|ttfrtj |St j |S)zReturns a list of :class:`Body` entities from :term:`SAT` or :term:`SAB` data. Accepts :term:`SAT` data as a single string or a sequence of strings and :term:`SAB` data as bytes or bytearray. ) isinstancebytes bytearray SabLoaderload SatLoader)datas 3lib/python3.12/site-packages/ezdxf/acis/entities.pyrrs1 $ *+~~d## >>$ c|tjkrtjd|tjt |}d|j _|D]}|j||jS)aXExport one or more :class:`Body` entities as text based :term:`SAT` data. ACIS version 700 is sufficient for DXF versions R2000, R2004, R2007 and R2010, later DXF versions require :term:`SAB` data. Raises: ExportError: ACIS structures contain unsupported entities InvalidLinkStructure: corrupt link structure invalid ACIS version: F) r MIN_EXPORT_VERSION ExportErrorr SatExporter_setup_export_headerheaderasm_end_markerexportdump_satbodiesversionexporterbodys r! export_satr2sp)))"8 BCC3G<=H%*HOO"    r"c|tjkrtjd|tjt |}d|j _|D]}|j||jS)aVExport one or more :class:`Body` entities as binary encoded :term:`SAB` data. ACIS version 21800 is sufficient for DXF versions R2013 and R2018, earlier DXF versions require :term:`SAT` data. Raises: ExportError: ACIS structures contain unsupported entities InvalidLinkStructure: corrupt link structure r$T) r r%r&r SabExporterr(r)r*r+dump_sabr-s r! export_sabr64sp)))"8 BCC3G<=H%)HOO"    r"ctj|stjd|tj}|j ||S)Nzinvalid export version: )r is_valid_export_versionr&r AcisHeader set_version)r/r)s r!r(r(KsH  ( ( 1":7) DEE ^^ F w Mr"c,|t|j<|SN)rtype)clss r!registerr?Ss L Jr"cBeZdZUejZded<eddZy) NoneEntitystrr=c<|jtjk(Sr<)r=r NONE_ENTITY_NAMEselfs r!is_nonezNoneEntity.is_none[syyE2222r"N)returnbool) __name__ __module__ __qualname__r rDr=__annotations__propertyrGr"r!rArAXs$&&D#& 33r"rArNONE_REFc|eZdZUdZdZded<dZded<eZded<dd Z dd Z dd Z dd Z dd Z ddZddZddZy)rauBase ACIS entity which also represents unsupported entities. Unsupported entities are entities whose internal structure are not fully known or user defined entity types. The content of these unsupported entities is not loaded and lost by exporting such entities, therefore exporting unsupported entities raises an :class:`ExportError` exception. zunsupported-entityrBr=intid attributesc:|jd|jdS)N())r=rTrEs r!__str__zAcisEntity.__str__ss))AdggYa((r"cJ|j|||j|y)z+Load the ACIS entity content from `loader`.N)restore_common restore_datarFloaderentity_factorys r!rzAcisEntity.loadvs  FN3 &!r"cy)z'Load the common part of an ACIS entity.NrOr]s r!r[zAcisEntity.restore_common{ r"cy)z%Load the data part of an ACIS entity.NrOrFr^s r!r\zAcisEntity.restore_datarar"cH|j||j|y)z,Write the ACIS entity content to `exporter`.N) write_common write_datarFr0s r!r+zAcisEntity.exports (# !r"cFtjd|j)zWrite the common part of the ACIS entity. It is not possible to export :class:`Body` entities including unsupported entities, doing so would cause data loss or worse data corruption! zunsupported entity type: )r r&r=rgs r!rezAcisEntity.write_commons!";DII; GHHr"cy)z'Write the data part of the ACIS entity.NrOrgs r!rfzAcisEntity.write_datarar"c#rKt|jD]}t|ts|yw)z7Yield all attributes of this entity of type AcisEntity.N)varsvaluesrrrFes r!entitieszAcisEntity.entitiess,d""$A!Z(%s-77N)rHrBr^rr_FactoryrHNoner^rrHrrr0rrHrr)rHzIterator[AcisEntity])rJrKrL__doc__r=rMrTrPrUrYrr[r\r+rerfrorOr"r!rrcsO %D#$BL%J %)"   " I r"c|j}|jrtS|jj |r||St j d|d|jd)Nzexpected entity type 'z', got '')read_ptr is_null_ptrrPnameendswithr ParsingError) expected_typer^r_ raw_entitys r!restore_entityrse"J .j))  $]O8JOO;LA N  r"c:eZdZUdZded<eZddZddZy) Transform transformrBr=c|j}|jdd|jdd|jdd|jdt||_y)N g?)read_transforminsertappendrmatrix)rFr^r s r!r\zTransform.restore_datasQ$$& As As B Ctn r"cd fd }g|jjD]#}||d||d||d%tddd}|jj|}|t |j d|j j| }j|rdndjdjd |jy) Nc,j|dy)Ng)r)valuer s r! write_doublez,Transform.write_common..write_doubles KK5) %r"rr rotate no_rotate no_reflectno_shear)rfloat) rrowsrtransform_directionround magnitude normalizeiscloserwrite_transform)rFr0rrow test_vectorresult is_rotatedr s @r!rezTransform.write_commons &;;##%C Q Q Q &1am 00=U6++Q/0))+33K@@  H < L! J  &r"Nrsrt) rJrKrLr=rMrrr\rerOr"r!rrsD# ZF%'r"rc6eZdZUdZded<dd dZd dZd dZy) AsmHeader asmheaderrBr=c||_yr<)r/rFr/s r!__init__zAsmHeader.__init__s  r"c.|j|_yr<)read_strr/r]s r!r[zAsmHeader.restore_commons( r"c:|j|jyr<) write_strr/rgs r!rezAsmHeader.write_common4<<(r"N))r/rBrprt)rJrKrLr=rMrr[rerOr"r!rrsD#))r"rc,eZdZUeZded<ddZddZy)SupportsPatternPatternpatterncd|jtjk\rtd|||_yy)Nr)r/rPATTERNrrr]s r!r[zSupportsPattern.restore_commons) >>X-- -))V^LDL .r"c:|j|jyr<) write_ptrrrgs r!rezSupportsPattern.write_commonrr"Nrprt)rJrKrLrPrrMr[rerOr"r!rrsGWM)r"rceZdZUdZded<eZded<eZded<eZded <eZ d ed <dfd Z dfd Z ddZ ddZ xZS)Bodyr1rBr=rrLumplumpWirewirerrct|||td|||_td|||_td|||_y)Nrrr)superr[rrrrrFr^r_ __class__s r!r[zBody.restore_commonsB v~6"66>B "66>B ' V^Lr"ct|||j|j|j|j|j|j yr<)rrerrrrrFr0rs r!rezBody.write_commonsF X&499%499%4>>*r"c||_|jjr||_y|j}|jjs#|j}|jjs#||_y)z+Append a :class:`Lump` entity as last lump.N)r1rrG next_lump)rFr current_lumps r! append_lumpzBody.append_lumpY 99  DI99L",,44+55 #,,44%)L "r"cg}|j}|js*|j||j}|js*|S)z4Returns all linked :class:`Lump` entities as a list.)rrGrr)rFlumpsrs r!rz Body.lumpsByy && LL &'11L&& r"rprt)rrrHrr)rHz list[Lump])rJrKrLr=rMrPrrrrr[rerr __classcell__rs@r!rrsJD#GWD$D$#Iy#M + *r"rceZdZUdZded<y)rrrBr=NrJrKrLr=rMrOr"r!rrs D#r"rceZdZUdZded<y)rrrBr=NrrOr"r!rrs D#r"rcveZdZUdZded<eZded<eZded<eZded<d fd Z dfd Z dd Z dd Z xZ S)rrrBr=rShellshellrr1ct|||td|||_td|||_td|||_y)Nrrr1)rr[rrrr1rs r!r[zLump.restore_common sB v~6'G#GV^D "66>B r"ct|||j|j|j|j|j|j yr<)rrerrrr1rs r!rezLump.write_common&sF X&4>>*4::&499%r"c||_|jjr||_y|j}|jjs#|j}|jjs#||_y)z-Append a :class:`Shell` entity as last shell.N)rrrG next_shell)rFr current_shells r! append_shellzLump.append_shell,sY ::  DJ JJM#..66 - 8 8 $..66',M $r"cg}|j}|js*|j||j}|js*|S)z5Returns all linked :class:`Shell` entities as a list.)rrGrr)rFshellsrs r!rz Lump.shells7sB '' MM- ()44M '' r"rprt)rrrHrr)rHz list[Shell])rJrKrLr=rMrPrrr1r[rerrrrs@r!rrs@D#ItE5D$C & -r"rceZdZUdZded<eZded<eZded<eZded<eZ d ed <eZ d ed <dfd Z dfd Z ddZ ddZxZS)rrrBr=rSubshellsubshellFacefacerrrrct|||td|||_td|||_td|||_td|||_td|||_y)Nrrrrr)rr[rrrrrrrs r!r[zShell.restore_commonJsb v~6(v~N&z6>J "66>B "66>B "66>B r"c2t|||j|j|j|j|j|j |j|j |j|jyr<)rrerrrrrrrs r!rezShell.write_commonRsj X&4??+4==)499%499%499%r"c||_|jjr||_y|j}|jjs#|j}|jjs#||_y)z+Append a :class:`Face` entity as last face.N)rrrG next_face)rFr current_faces r! append_facezShell.append_faceZsY 99  DI99L",,44+55 #,,44%)L "r"cg}|j}|js*|j||j}|js*|S)z4Returns all linked :class:`Face` entities as a list.)rrGrr)rFfacesrs r!rz Shell.faceserr"rprt)rrrHrr)rHz list[Face])rJrKrLr=rMrPrrrrrr[rerrrrs@r!rrAsTD# J !Hh!D$D$D$C& *r"rceZdZUdZded<y)rrrBr=NrrOr"r!rros D#r"rceZdZUdZded<eZded<eZded<eZded <eZ d ed <eZ d ed <dZ dZ dZ dfd Zdfd ZddZddZxZS)rrrBr=z'Face'rLooplooprrrrSurfacesurfaceFc~t|||td|||_td|||_td|||_td|||_td|||_|jdd|_ |jdd |_ |jr|jd d |_ yy) Nrrrrrreversedforwarddoublesingleinout) rr[rrrrrr read_boolsense double_sided containmentrs r!r[zFace.restore_commons v~6'G"66>B #GV^D &z6>J %iH %%j)< ",,Xx@   %//e>*499%4::&4==)4<<(DJJ I>D--xB       0 0$ > r"c||_|jjr||_y|j}|jjs#|j}|jjs#||_y)z+Append a :class:`Loop` entity as last loop.N)rrrG next_loop)rFr current_loops r! append_loopzFace.append_looprr"cg}|j}|js*|j||j}|js*|S)z4Returns all linked :class:`Loop` entities as a list.)rrGrr)rFloopsrs r!rz Face.loopsrr"rprt)rrrHrr)rHz list[Loop])rJrKrLr=rMrPrrrrrrrrr[rerrrrs@r!rrtsbD# Iv D$E5!Hh!GW ELK = ? *r"rcbeZdZUdZded<eefZeefZddZd dZ e jd dZ y) rrrBr=c|j|jf|_|j|jf|_yr<) read_intervalu_boundsv_boundsrcs r!r\zSurface.restore_datas>,,.0D0D0FF ,,.0D0D0FF r"c|j|jd|j|jd|j|jd|j|jdyNrr )write_intervalrrrgs r!rfzSurface.write_datas^ a 01 a 01 a 01 a 01r"cy)znReturns the spatial location at the parametric surface for the given parameters `u` and `v`. NrOrFuvs r!evaluatezSurface.evaluate r"Nrs)r0rr rr rrHr) rJrKrLr=rMINFrrr\rfabcabstractmethodr rOr"r!rrsCD#CxHCxHG2    r"rceZdZUdZded<edddZedddZedddZedddZ dZ d fd Z d fd Z d Z d d ZxZS)Planez plane-surfacerBr=rr Fc(t|||t|j|_t|j|_t|j|_|jdd|_|jyN reverse_v forward_v) rr[r read_vec3originnormalu_dirrr update_v_dirrs r!r[zPlane.restore_commonsq v~66++-. 6++-. &**,- ))+{C r"ct|||j|j|j |j |j |j |j|jddyr) rrewrite_loc_vec3rwrite_dir_vec3rrrrrs r!rezPlane.write_commons\ X& , , +DNNKEr"cz|jj|j}|jr| }||_yr<)rcrossrrv_dir)rFr#s r!rzPlane.update_v_dirs/ !!$**- >>FE r"cZ|j|j|zz|j|zzSr<)rrr#r s r!r zPlane.evaluates&{{djj1n-a@@r"rprtr)rJrKrLr=rMrrrrr#rr[rerr rrs@r!rrs^D# !Q]F !Q]F AqME AqMEIF Ar"rcxeZdZUdZded<eZded<eZded<eZded<d fd Z dfd Z ddd Z dd Z xZ S)rrrBr=rCoedgecoedgerrct|||td|||_td|||_td|||_y)Nrr'r)rr[rrr'rrs r!r[zLoop.restore_commonsB v~6'G$Xv~F "66>B r"ct|||j|j|j|j|j|j yr<)rrerrr'rrs r!rezLoop.write_commonsF X&4>>*4;;'499%r"cPt|dkDsJ|d|_|dd}|dd}|r*|j|d|jd|dn+|jt|jdtt |||D]\}}}||_||_||_y)z"Set all coedges of a loop at once.rr NrR) lenr'rrrPzipr prev_coedge next_coedge)rFcoedgesclose next_coedges prev_coedgesr'nextprevs r! set_coedgeszLoop.set_coedgess7|aaj qr{ s|     +   72; /    )   8 ,"%g|\"J FD$FK!%F !%F #Kr"cg}|j}|js;|j||j}||jur |S|js;|S)z6Returns all linked :class:`Coedge` entities as a list.)r'rGrr.)rFr/current_coedges r!r/z Loop.coedgessV (( NN> *+77N, !(( r"rprt)T)r/ list[Coedge]rHrrrHr8)rJrKrLr=rMrPrr'rr[rer5r/rrs@r!rrs@D#ItFFD$C & &$ r"rceZdZUdZded<eZded<eZded<eZded<eZ ded<d Z d ed <eZ d ed <dZ ded<eZ ded<dfd Zdfd ZddZddZddZxZS)r&r'rBr=r.r-partner_coedgeEdgeedgeTrIrrrrrSunknownPCurvepcurvecZt|||td|||_td|||_td|||_td|||_|jdd|_td|||_ |jd|_ td|||_ y) Nr'r=rrrrskip_satr@) rr[rr.r-r;r=rrrread_intr>r@rs r!r[zCoedge.restore_common1s v~6)(FNK)(FNK,Xv~N"66>B %%j)< "66>B 2 $Xv~F r"ct|||j|j|j|j|j|j |j|j |j|jdd|j|j|jdd|j|jy)NrrrTrB) rrerr.r-r;r=rrr write_intr@rs r!rezCoedge.write_common<s X&4++,4++,4../499%DJJ I>499%1t,4;;'r"c|jjsJ|j}|jr|}||_||_|jyr<)r;rGorder_partner_coedges)rFr'r;s r!add_partner_coedgezCoedge.add_partner_coedgeHsL$$,,,,,,  ! !!N$ . ""$r"cyr<rOrEs r!rHzCoedge.order_partner_coedgesRs r"cg}|j}|jr|S |j||j}|js||ur |S1)z:Returns all partner coedges of this coedge without `self`.)r;rGr)rFr/r;s r!partner_coedgeszCoedge.partner_coedgesWsW ",,  ! !N NN> *+::N%%4)? r"rprt)r'r&rHrr)rHrrr9)rJrKrLr=rMrPr.r-r;r=rrr>r@r[rerIrHrLrrs@r!r&r&s{D#"K""K" &NF%D$E4D$GSFF G (% r"r&ceZdZUdZded<eZded<dZded<eZded <dZ ded <eZ d ed <eZ d ed<dZ ded<dZ ded<dfd Zdfd ZxZS)r<r=rBr=Vertex start_vertexrr start_param end_vertex end_paramr&r'CurvecurveFrIrr> convexityct|||td|||_|jt j k\r|j|_td|||_ |jt j k\r|j|_ td|||_ td|||_ |jdd|_|jt j k\r|j|_yy)Nvertexr'rTrr)rr[rrOr/r TOL_MODELING read_doublerPrQrRr'rTrrrrUrs r!r[zEdge.restore_commonws v~6*8V^L >>X22 2%113D (6>J >>X22 2#//1DN$Xv~F #GV^D %%j)< >>X22 2#__.DN 3r"ct|||j|j|j |j |j|j |j |j|j|j|j|j|j|jdd|j|jy)Nrr)rrerrOrrPrQrRr'rTrrrrUrs r!rezEdge.write_commons X&4,,-d../4??+dnn-4;;'4::&DJJ I>4>>*r"rprt)rJrKrLr=rMrPrOrPrQrRr'rTrrUr[rerrs@r!r<r<esrD#$L&#K"J!IuFFE5E4Is / + +r"r<ceZdZUdZded<y)r?r@rBr=NrrOr"r!r?r?s D#r"r?cfeZdZUdZded<eZded<dZded<eZd ed <d fd Z dfd Z xZ S)rNrWrBr=r<r=rrS ref_countPointpointct|||td|||_|j d|_td|||_y)Nr=rrBr_)rr[rr=rDr]r_rs r!r[zVertex.restore_commonsD v~6"66>B !4#GV^D r"ct|||j|j|j |j d|j|j y)NTrB)rrerr=rFr]r_rs r!rezVertex.write_commonsK X&499%4>>D94::&r"rprt) rJrKrLr=rMrPr=r]r_r[rerrs@r!rNrNs;D#D$IsE5E ''r"rNcZeZdZUdZded<eefZddZd dZe jd dZ y) rSrTrBr=cN|j|jf|_yr<)rboundsrcs r!r\zCurve.restore_datas **,f.B.B.DD r"c||j|jd|j|jdyr)rrdrgs r!rfzCurve.write_datas. A/ A/r"cy)z_Returns the spatial location at the parametric curve for the given parameter. NrOrFparams r!r zCurve.evaluaterr"NrsrtrhrrHr) rJrKrLr=rMrrdr\rfrrr rOr"r!rSrSs:D# #XFE0   r"rScleZdZUdZded<edddZedddZd fd Zd fd Z d dZ xZ S) StraightCurvezstraight-curverBr=rr ct|j|_t|j|_t||yr<)rrr directionrr\)rFr^rs r!r\zStraightCurve.restore_datas:6++-. f..01 V$r"c|j|j|j|jt||yr<)rrr rmrrfrs r!rfzStraightCurve.write_datas4 ,/ 8$r"c:|j|j|zzSr<)rrmrgs r!r zStraightCurve.evaluates{{dnnu455r"rsrtri) rJrKrLr=rMrrrmr\rfr rrs@r!rkrks6 D# !Q]FQ1 I% % 6r"rkc:eZdZUdZded<eZded<d dZd dZy) r^r_rBr=rlocationc@t|j|_yr<)rrrqrcs r!r\zPoint.restore_datasV--/0 r"c:|j|jyr<)rrqrgs r!rfzPoint.write_datas .r"Nrsrt) rJrKrLr=rMrrqr\rfrOr"r!r^r^s D#Hd1/r"r^c\eZdZUded<d dZd dZd dZdZejd dZ y) FileLoaderz'Sequence[sat.SatEntity | sab.SabEntity]recordsc i|_||_yr<)ror/rs r!rzFileLoader.__init__s/1 # r"ct|} |j|S#t$r<tj |j t }||j|<|cYSwxYwr<)rTroKeyErrorrgetrzr)rFr~uidentitys r!r_zFileLoader.entity_factorysZn ==% % !%%joozBDF!'DMM# M sAA! A!c~|jjDcgc]}t|ts|c}Scc}wr<)rorlrrrms r!r.zFileLoader.bodiess2==//1I1aZ45H1IIIs::c|j}|jD]m}||}|j|_|j}|js |||_|j |j }|j||oyr<)r_rvrTrUrymake_data_loaderr r)rFr_r~r|rU data_loaders r! load_entitieszFileLoader.load_entitiesss,,,,J#J/F" FI#..J))$2:$>!// @K KK ^ 4'r"cyr<rOrFr s r!rzFileLoader.make_data_loaderrar"N)r/rS)r~rrHr)rH list[Body]r z list[Any]rHr) rJrKrLrMrr_r.rrrrrOr"r!rurus7 44$J 5   r"ruc:eZdZdfd ZddZeddZxZS)rctj|}t| |jj |j |_yr<)r parse_sabrrr)r/rorvrFr builderrs r!rzSabLoader.__init__5--% //0'' r"cBtj||jSr<)r SabDataLoaderr/rs r!rzSabLoader.make_data_loader   t||44r"cR||}|j|jSr<rr.r>r r^s r!rzSabLoader.load#T}}r")r bytes | bytearrayr)r rrHrrJrKrLrr classmethodrrrs@r!rr!( 5r"rc:eZdZdfd ZddZeddZxZS)rctj|}t| |jj |j |_yr<)r parse_satrrr)r/rorvrs r!rzSatLoader.__init__rr"cBtj||jSr<)r SatDataLoaderr/rs r!rzSatLoader.make_data_loaderrr"cR||}|j|jSr<rrs r!rzSatLoader.loadrr")r str | Sequence[str]r)r rrHrrrs@r!rrrr"r)r rrHr)r.Sequence[Body]r/rSrHz list[str])r.rr/rSrHr)rHzhdr.AcisHeader)r}rBr^rr_rqrHr)A __future__rtypingrrrrrrrr r r r rabstractrrr type_hintsr ezdxf.mathrrrrqrrMrrrDEFAULT_SAT_VERSIONr2DEFAULT_SAB_VERSIONr6r(r?rArPrrrrrrrrrrrrrrrr&r<r?rNrSrkr^ABCrurrrOr"r!rs?#:: "">>#.. N#\1 2,. ). El ,1+D+D %(.,1+D+D %( . 33 #77t    *