:vjjddlmZddlZddlmZmZmZddlmZ ddl m Z ddl m Z ddlmZd d lmZd d lmZmZmZmZid d d dddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/Zej/Dcic]\}}|| c}}Zd0Zd1Zd2Zd3Zd4Zd5Zd;d6Zd) defaultdictN)grouping resourcesutil)transformations)log)tol_path) multi_dict) to_threepoint)ArcBSplineLineTextinchesfeetmiles millimeters centimetersmeters kilometers microinches mils yards angstroms nanometers microns decimeters decameters hectometers gigametersAUz light yearsparsecs)z|<^>|zTRIMESH_METADATA:ipc |j}t|dr5|ddjddk7r td|j dd }t |j }|jd d j j}tjt j|jd }tjt j|jd }tj|ddd fdk(d}i}tj|ddd fdk(d}t|dkDr|tj||d} ||d| } i|d<dD]\} } t!| | | } | | |d| <dD]4\}} t!| | d}|||| <||z }|t"vs)t"||d<6d|vrt%j&dtj|ddd fdk(dd}|tj||}d}|||ddd f}d|vsd|vr tj|ddd fdk(dd}|tj||}|||}|||}tj|ddgk(j)d d}i}tj*tj,t||D]0} t/||||d\}}}t|dkDr||f||<2 t/|||||||!\}}|||d"}|S#t0$rY^wxYw#t0$rt%j2dd YPwxYw)#a Load a DXF file to a dictionary containing vertices and entities. Parameters ---------- file_obj: file or file- like object (has object.read method) Returns ---------- result: dict, keys are entities, vertices and metadata decodeNsAutoCAD Binary DXFzBinary DXF is not supported!zutf-8ignoreerrors )r<r rENDSECrHEADER DXF_HEADER))z$ACADVER1)z $DIMSCALE40)z$DIMALT70)z$DIMALTFrF)z$DIMUNITrG) $INSUNITSrG)$LUNITSrG))r<rI)rrHrGunitsz!DXF doesn't have units specified!ENTITIESINSERTBLOCKBLOCKS0axisT) return_namezfailed to parse blocks!exc_info)blocks)verticesentitiesmetadata)readhasattrfindNotImplementedErrorr:strstripreplaceuppernparray splitlinesreshapenonzerolen searchsortedget_key _DXF_UNITSr debugall array_splitarangeconvert_entities BaseExceptionerror) file_objkwargsrawcleanedblob_rawblobendsecrX header_start header_end header_blobkeygroupvalueoffsetrJ entity_start entity_endrU check_entity block_start block_end blob_blockblob_block_raw block_inflindexvenamerVrWresults j/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/path/exchange/dxf.pyload_dxfr6s$ --/CsH s8==. /2 5%&DE E**WX*6C c(.. Ckk#r"((*002Gxxs+,44W=H 88CNN7+ , 4 4W =DZZQT h. / 2FH::d1a4jH45a8L <1BOOFLODE <?Z8 "$ JCKe4E .3&s+ $?KFCKd3E}!HSM VOE "$.u$5!? ( " II9 :::d1a4jJ67:1=L =>J F Z0A6L<7l#: @**T!Q$Z8%;%A$F$FA$F$NOPQRJF #j/(BJO!1"5)>%+@d"JAq$1vz()1vt P* \*%x Z'HQWHh #h OF M% @ II/$ ? @s7 B,M:-M 'M MMMM M;:M;c f !"#||}d"!"#fd}!"#fd}!"#fd} !"#fd}!"#fd}!#fd} !#fd } t|ft|ft|ft|ft|ft| ft| fd } g#g!d} tj|ddd fd k(d } t d }tj tj t|| D]g}t|dkr|| d d}|dvr- tdt D}|||d |d<|dk(r t g} c| |dk(r| jt | }|dk(rwtj| ddDcgc] }|d|dgc}tj}d}d| d vr;t| d d}t|dz}|rtj ||ddf}tjt| Dcgc] \}}d|vs |c}}tj"}tjt| Dcgc]\}}d|vs t%|dc}}tj}t'||||\}}|D] }|xj(t#z c_"#j+|!j+|d} |dk(r,||}||ddd f|ddd f< | t|6|| vr| |\}}| }||T|dk7s[||xxdz cc<jt|d kDrHt/j0d j3d!j5d"|j7Dt9j:#j=tj#|r||ddd fd#k(d d}#!|fS#!fS#t$rYwxYwcc}wcc}}wcc}}w#t,$rt/j0ddYWwxYw)$av Convert a chunk of entities into trimesh entities. Parameters ------------ blob : (n, 2) str Blob of entities uppercased blob_raw : (n, 2) str Blob of entities not uppercased blocks : None or dict Blocks referenced by INSERT entities return_name : bool If True return the first '2' value Returns ---------- Ncddd}|jDcic]&\}}||vs ||tj|d(}}}|Scc}}w)zJ Pull metadata based on group code, and return as a dict. layerr)82r)itemsr make_sequence)rgetkrrenameds rinfozconvert_entities..infos[ &)AB V 1QRUX3q64--a033 VWs A A c jtd dttjdzi|j tj |d|dg|d|dggtjy) zG Convert DXF LINE entities into trimesh Line entities. pointsr 10201121dtypeN)appendrrfrarmextendrbfloat64)rrWrrVs r convert_linez&convert_entities..convert_linesp LCMBIIaL$@LDGLM HHqw$(1T7AdG*<=RZZ P c \t|d}tj|d|dgjtj}t |dd|}j td ttjdzdd |j|y) zJ Convert DXF CIRCLE entities into trimesh Circle entities rFrrNr )centerradiusrTrclosedr) floatrarbastyperr rrrfrmr)rRCrrWrrVs rconvert_circlez(convert_entities..convert_circles !D'N HHagqw' ( / / ;aeA6  NH  ! 4d Nd1g N  rc t|d}tj|d|dgtj}tjtj|d|dgtj}t |dd|| }j td ttjd zd d |j|y)zJ Convert DXF ARC entities into into trimesh Arc entities. rFrrr5051Nr )rranglesrFrr) rrarbrradiansr rrrfrmr)rrrArrWrrVs r convert_arcz%convert_entities..convert_arcs !D'N HHagqw'rzz : JJrxx4!D' 2"**E FaeAa@Y3x=299Q<#?YQUVWQXYZrc tj|d|dfjtj} |}d|vxrt |dddz}|rtj ||ddf}d|vrtj |dtj}tj dddfdk(d}tj dddfdk(d}|s||jk}||}||}tj||}t|||| \} }|D] } | xjtz c_ "j|  j|y jtd d tj t|tzi|j|y) zM Convert DXF LWPOLYLINE entities into trimesh Line entities. rrrGrrN42rlinesbulge bulge_idx is_closedrr)ra column_stackrrintvstackrbremaxrg bulge_to_arcsrrfrrrrm)rrpolyinforrvidbidbid_okrrichunkrWrrVs rconvert_polylinez*convert_entities..convert_polylines 4!D' 23::2::F7AI5#agaj/A"5 IIueBQi01E 19HHQtWBJJ7E**U1a4[D01!4C**U1a4[D01!4Cswwy&kf S1I 5IDAqCM) OOA  OOA   VBIIc%j$9CM$IVXVWrc tj|d|dfjtj}tj|djtj}t |dk(rSj td dt tjdzi|j|ytjj|d|dz tjk}j td tjt |t z||d |j|y) zL Convert DXF Spline entities into trimesh BSpline entities. rrrFr rNrr<)rknotsrr)rarrrrbrfrrrmrlinalgnormtolmerger)rrrrrWrrVs rconvert_bsplinez)convert_entities..convert_bspline7s!D'1T7!34;;BJJG4!((4 v;!  OODPH  ! (DPQP Q OOF #  q F2J 67#))C   yyV-H = q'    rc |dj} t|d} tjt|d}tj |d|dgtj }|tj|tj|gz}d d g} gd t|d |d <jtttd z|||j|j|y#t$rd}YwxYw#t$rd}YwxYw#t$rYwxYw)zF Convert a DXF TEXT entity into a native text entity. rErFNrrrrr)leftrright72rr)originvectorheighttextalign) r^rrorarrbrcossinrrrrf) rrranglerrrrWrVs r convert_textz&convert_entities..convert_textZs2 v||~ 1T7^F JJuQtW~.E 1T7AdG,BJJ?266%="&&-888$ 23qw<@E!H  8}8}q(    = F   E    s4D!D#%D4 D D # D10D14 E?Ec y|d}|vryt|jdd}tj|jdd|jddgtj}tj|jdd |jd d gtj}|\}}|D]B} j |j  d xjt z c_D jtj|tj|tj|| y) z Convert an INSERT entity, which inserts a named group of entities (i.e. a "BLOCK") at a specific location. Nrrrrrr41?rr<)r~thetascale)rrrarbrrcopyrrfrtftransform_points planar_matrixr) rrrr~rcvcerrUrWrVs rconvert_insertz(convert_entities..convert_inserts > v v  aeeD#&'155s+QUU4-=>bjjQ!%%c*AEE$,<=RZZPBA OOAFFH % RL  3x= 0      B$$F"**U:KSXY  r)LINE LWPOLYLINEARCCIRCLESPLINErLrMrrOcy)Nrrrrrz"convert_entities..sarr)rMrLc38K|]\}}|ddk(s|yw)rrNr).0rrs r z#convert_entities..s!!P0@1AaDCK!0@sPOLYLINEVERTEXSEQENDrrrFrGrrTEXTzfailed to load text entity!TrSrKzskipping dxf entities: {}z, c30K|]\}}|d|yw)z: Nr)rrrs rrz#convert_entities..s!F2E$!QQCr!+2Esr)dictr rarerrlrmrfnext enumerate StopIterationrrbrrboolrint64rrrrror rjformatjoinrr vstack_emptyr)$rvrurUrRrrrrrrrloaderspolyline inflection unsupportedr entity_type index_namerrrflagrrrrr chunk_rawchunkerloader entity_datarrrWrrVs$ ` @@@@rrnrnsI&      &1f! F' R  J|$!#34k"(/('GHHHDAJ#-.q1Ji(K #d) 4jA u:> U Ahqk  - - !!P %0@!PP '/z'B1'Ej!!$ * $U }H  !kX&= OODK (  !kX&=HH(12,G,Qqw$0,GrzzZEIx{"8A;t,- N IIueBQi&89E(2@2tq!dai2@IHH,5h,?M,?DAq419qw,?MjjE #5IFBCM) OOB  OOB H F "!I#5k!Q$/IadO HT)_- G #%k2OGV!%.K ;  J &  $ ) $eBf ;! ' . . F+2C2C2EFF    *11"**=HQT c)*1-a04'' X a!  HAN2! H 7$GG HsB=,O+<O;  P P P P ;P + O87O8 P0/P0c tjddddfdfd}fd}fd}fd}tjd ||||d }g}t |j |j D]\}} t|j} || |vr"| |vrE|| ||jj} t| d kDsY|j| ktjd| dj|} t dtj|j d dD cic] \} }d| | }} }|j#t dtj|j ddD cic] \} }d| | c}} t$j'|j(d|d<dj*di|jg}|jdj+| j|jdjdj|j-t.d}t0j2r\t4j7t5|}t|dzd k(sJt9d|d d dDsJ|j;d|Scc}} wcc}} w) a1 Export a 2D path object to a DXF file. Parameters ---------- path : trimesh.path.path.Path2D Input geometry to export only_layers : None or set If passed only export the layers specified Returns ---------- export : str Path formatted as a DXF file ztemplates/dxf.jsonc @tj|tj}tj|}|rRtj tj t|tjjdd}n0tjt|dftj}|gdz }|r|ddddf}|ddddf}djd t|jd |jd D}|S) a{ Format points into DXF- style point string. Parameters ----------- points : (n,2) or (n,3) float Points in space as_2D : bool If True only output 2 points per vertex increment : bool If True increment group code per point Example: [[X0, Y0, Z0], [X1, Y1, Z1]] Result, new lines replaced with spaces: True -> 10 X0 20 Y0 30 Z0 11 X1 21 Y1 31 Z1 False -> 10 X0 20 Y0 30 Z0 10 X1 20 Y1 30 Z1 Returns ----------- packed : str Points formatted with group code rr<r)rrr)r"r6Nr  c34K|]\}}|dd|dywdr.12gNr)rgrs rrz4export_dxf..format_points..Ys( +T41aqe2aX +Tr<) ra asanyarrayrrstack_3Dtilermrfrrdzerosrzip)ras_2D incrementthreer|packeds r format_pointsz!export_dxf..format_points3s.vRZZ8 f% GG #e*BHH5==gFEHHc%j!_BHH=E  !RaR%LE!RaR%LE +.u}}R/@%--PRBS+T   rcddtt|ddd}t|dr"tj|j |d<|S)a Pull layer, color, and name information about an entity Parameters ----------- entity : entity object Source entity to pull metadata Returns ---------- subs : dict Has keys 'COLOR', 'LAYER', 'NAME' rNr.)COLORLAYERNAMErr')r]idrZrto_asciir)entitysubss r entity_infozexport_dxf..entity_info_sJ 6 OCR(  67 # MM&,,7DM rc|j|}t|dkry|}|dd|d<d|d<t||d <tt|j|d <d j d i|}|S) aB Convert an entity to a discrete polyline Parameters ------------- line : entity Entity which has 'e.discrete' method vertices : (n, 2) float Vertices in space Returns ----------- as_dxf : str Entity exported as a DXF rrATF)rr POINTSrTYPEVCOUNTFLAGliner)discreterfrrrr)r3rVrr,rr-r#templates rrz export_dxf..convert_linexs"x( v;! 4 &vTUKX#V VX4 ,-V (&!((040 rc|j|dd}|}|j}t|dk(rtj|d}dj|}|d|j dz }|j rd |d <n9d |d <|d z }|d jtj|jz }||d<djdi|}|S)NTF) return_angle return_normalr rz 10 {:.12g} 20 {:.12g} 30 {:.12g}z 40 rrr0rz 100 AcDbArcz 50 {:.12g} 51 {:.12g}DATAarcr) rrfrarrrrdegreesr) r:rVrr,rdatarr-r5s rrzexport_dxf..convert_arcszz(UzK3 v;! YYvs+F=6==vF &T*++ ::#DL DL $ $D 7077DKK9PQ QDV '%''/$/ rc  ||jd}dt|jzj|jdd}ddddd d }|d }|jr||d z}gd }gd}dj dt ||D} |} | jd|||d|dt|jt|jd  djdi| } | S)NFr z 40 {:.12g} r<rr rrr.)rperiodicrationalplanarlinearrAr)rrr)rc34K|]\}}|dd|dywrr)rrjs rrz6export_dxf..convert_bspline..s&M9LAQqEAd8,9Lrrrr) r0r/KNOTSNORMALDEGREEr2FCOUNTKCOUNTPCOUNTbspliner)rrfrrrrrupdate)splinerVrrflagsr normaln_coden_strr,rr-r#r5s rrz#export_dxf..convert_bsplinesx 6%H=3v||#44<ORIGINANGLEr@asciir=r>rHEIGHTrr) rrar;rrr_ _SAFESPACEencoder:rr)txtrVsubrr-r#r5s rrz export_dxf..convert_texts #%h |&<NH zz#))H"56G  HH  S* - VGHV - VG_ F  H (&!((/3/ rc||S)zl For entities we don't know how to handle, return their discrete form as a polyline r)r+rVrs rconvert_genericz#export_dxf..convert_generics FH--rr&) precision)rrrBezierrNrzEntity type %s not exported!rXYZrEXTMIN_rEXTMAX_LUNITSheaderrW)rKfooterr@r c38K|]}t|dk\yw)rN)r)rrs rrzexport_dxf..(s3 13q6Q; srX)FTr)rget_jsonraset_printoptionsrrWlayerstype__name__rVr^rfrr rjrboundsrO _UNITS_TO_DXFrrJrr_rZrstrictr]rcrkr[)path only_layersrrrr_ conversions collectedrrr converted entities_strrrhsubsectionsrvrrr-r#r5s @@@@r export_dxfry s"!!"67H*X2 D.#J,."%!" KI t{{35Aw  "uK'?  ; ) D)!T]];AACI9~!  + II4d ;499Y'L*-UBIIdkk!nc4R)S T)SAgaSM1 )SD TKKc%4;;q>SV9W.XY.Xda71#!.XYZ"&&tzz15DN)"))1D1779:H OOHZ(///FLLNO OOHX&,,./ 99X  & &z3 7D zzs4y)E Q1$$$3cc 3333 G K9 UYs KK c L tj|tj}tj|tj}tj|tj}tj|dkD}||}||}|i}t |dk(r/tj t |}t|fi|g}||fStj|dz}tj||dz f} |r| t |z} || dddf|| dddfz } tjj| d} tjtj| tjt | ftjt | dfgd z} tj | dddd f} || j#d} | d z tj$|d z z }|tj&|d z |zz }tj|| dddf| | |j)d zz|| dddffj)d }t*j,rd dlm}|Dcgc]}||j2}}tj4tj|tj|sJ|Dcgc]}||j6}}tj4|tj|sJgg}}|D]N}|j9t;ddtj dt |zi||j=|Pt |dz t |kDrtj>tj t |}tAjB|tjD| dtjFD]_}|j9tddtj d t |zi||j=||jIatjJ|tj}||fScc}wcc}w)a Polylines can have "vertex bulge" which means the polyline has an arc tangent to segments, rather than meeting at a vertex. From Autodesk reference: The bulge is the tangent of one fourth the included angle for an arc segment, made negative if the arc goes clockwise from the start point to the endpoint. A bulge of 0 indicates a straight segment, and a bulge of 1 is a semicircle. Parameters ---------------- lines : (n, 2) float Polyline vertices in order bulge : (m,) float Vertex bulge value bulge_idx : (m,) float Which index of lines is bulge associated with is_closed : bool Is segment closed metadata : None, or dict Entity metadata to add Returns --------------- vertices : (a, 2) float New vertices for poly-arc entities : (b,) entities.Entity New entities, either line or arc rgh㈵>Nrg@rrPr)rrrr g@r)r<rr ) arc_centerrr)&rarrrabsrfrmrarctanrrrcrossronesrunitizemeanrrrdrrpr:r{spanallcloserrrr stack_linesr boolean_rowssort setdiff1drrb)rrrrrXokrrWrtidrlengthperpmidpointrr~r!r{r check_angle check_radiirV arc_pointsexistingline_idxs rrr/sD MM%rzz 2E MM%rzz 2E irxx8I  B "IE" I 5zQ #e*%+(+,h IIe s "E //9i!m4 5C s5z3q!t9 c!Q$i 0 0FYY^^F^ +F 88 #f+!678 Va !I- D <<QU $DSzA&HslbffUS[1 1FbffUQY'&0 0F OO s1a4y 8dV^^G-D&DDeCPQSTPTIFVW  gj  zz$3895az!}))5 9{{266%="&&*=>>>5:;Uz!}++U ;{{;v777RhH L299Q<#h-#?L8LM # E Q#e*$##BIIc%j$9: -- bggc*BLL H OODQ ! s8}(DQQ R OOE(O002 3  xx 3H X ;: Aw$ tAw<    7N s"+A A AAA%$A%dxf)NNF)N)FN)$ collectionsrnumpyrarArrrrr constantsr r rr r:r rWrrrrrirrorZXRECORD_METADATAXRECORD_SENTINELXRECORD_MAX_LINEXRECORD_MAX_INDEXrrnryrrh _dxf_loaders)rrs00rrs#))%(//xvw}  }  x |}v       !" #$ ) .#-"2"2"45"4$!QA"45  'DeP L^D&x 6s B?