TijtddlmZddlZddlmZmZmZddlmZ 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/Zd0eDZd1Zd2Zd3Zd4Zd5Zd6Zd=d8Zd>d9Z d?d:Z!d;Z"dr>%s555$!QA555z|<^>|zTRIMESH_METADATA:ipc b|}t|drG|ddddkrtd|dd }t |}|d d }tj t | d }tj t | d }tj |ddd fdkd}i}tj |ddd fdkd}t|dkr|tj||d} ||d| } i|d<dD]#\} } t!| | | } | | |d| <$dD]<\}} t!| | d}|||| <||z }|t"vrt"||d<=d|vrt%jdtj |ddd fdkdd}|tj||}d}|||ddd f}d|vsd|vr? tj |ddd fdkdd}|tj||}|||}|||}tj |ddgkd d}i}tjtjt||D]O} t/||||d\}}}t|dkr||f||<@#t0$rYLwxYwn&#t0$rt%jdd YnwxYwt/|||||||!\}}|||d"}|S)#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 )rEr rENDSECrHEADER DXF_HEADER))z$ACADVER1)z $DIMSCALE40)z$DIMALT70)z$DIMALTFrO)z$DIMUNITrP) $INSUNITSrP)$LUNITSrP))rErR)rrQrPunitsz!DXF doesn't have units specified!ENTITIESINSERTBLOCKBLOCKS0axisT) return_namezfailed to parse blocks!exc_info)blocks)verticesentitiesmetadata)readhasattrfindNotImplementedErrorrCstrstripreplaceuppernparray splitlinesreshapenonzerolen searchsortedget_key _DXF_UNITSr debugall array_splitarangeconvert_entities BaseExceptionerror) file_objkwargsrawcleanedblob_rawblobendsecra header_start header_end header_blobkeygroupvalueoffsetrS entity_start entity_endr^ check_entity block_start block_end blob_blockblob_block_raw block_inflindexr<enamer_r`results r=load_dxfr6s$ --//CsH 7 ss8==. / /2 5 5%&DEE E**WX*66C c((..  Ckk#r""((**0022Gxs++,,44W==H 8CNN7++ , , 4 4W = =DZQQQT h. / / 2FH:d111a4jH455a8L <1BOFLODDE <?Z78 "$ 4 4JCKe44E .3&s+? 6 6KFCKd33E}!HSM VOE ""$.u$5! ( " " I9 : : ::d111a4jJ677:1=L ==>J F Z/0A6L<7l#:#: @*T!!!Q$Z8%;<%A$F$FA$F$N$NOOPQRJF #j//(B(BJOO  !1"5)>%+@d"""JAq$1vvzz()1vt $D  @ @ @ I/$ ? ? ? ? ? ? @* \* $%x Z0G'HQWHh #h O OF Ms7C Oz2convert_entities..info..s9VVV1QRUXX3q64-a003XXXr?)items)rrenamedrs @r=infozconvert_entities..infos; &))WVVV VVVr?c Btd dttjdzi|tj|d|dg|d|dggtjdS) zG Convert DXF LINE entities into trimesh Line entities. pointsr 10201121dtypeNr8)appendrrorjrvextendrkfloat64)rr`rr_s r= convert_linez&convert_entities..convert_lines LLCMMBIaLL$@LDDGGLLMMM Hqw$(1T7AdG*<=RZ P P P     r?c t|d}tj|d|dgtj}t |dd|}td ttj dzdd | |dS) zJ Convert DXF CIRCLE entities into trimesh Circle entities rOrrNr )centerradiusrTrclosedr8) floatrjrkastyperr rrrorvr)rRCrr`rr_s r=convert_circlez(convert_entities..convert_circles !D'NN Hagqw' ( ( / / ; ;aeA666  NH  ! 4d N Ndd1gg N N    r?c t|d}tj|d|dgtj}tjtj|d|dgtj}t |dd|| }td ttj d zd d | |dS)zJ Convert DXF ARC entities into into trimesh Arc entities. rOrrr5051Nr )rranglesrFrr8) rrjrkrradiansr rrrorvr)rrrArr`rr_s r= convert_arcz%convert_entities..convert_arcs !D'NN Hagqw'rz : : : Jrx4!D' 2"*EEE F FaeAa@@@Y3x==29Q<<#?YYQUQUVWQXQXYYZZZr?c &tj|d|dftj} |}d|vot |dddz}|rtj||ddf}d|vrtj|dtj}tj dddfdkd}tj dddfdkd}|s(||k}||}||}tj ||}t|||| \} }|D]} | xj tz c_  |  |dS td d tjt|tzi| |dS) zM Convert DXF LWPOLYLINE entities into trimesh Line entities. rrrPrrN42rlinesbulge bulge_idx is_closedrr8)rj column_stackrrintvstackrkrnmaxrp bulge_to_arcsrrorrrrv)rrpolyinforrvidbidbid_okrr<ichunkr`rr_s r=convert_polylinez*convert_entities..convert_polylines 4!D' 233::2:FF477AI5#agaj//A"5  2IueBQBi011E 199HQtWBJ777E*U111a4[D011!4C*U111a4[D011!4C &swwyy&kf S11I 5IDAq * *CMM) OOA    OOA    F VVBIc%jj$9$9CMM$IVXVVWWWr?c 8tj|d|dftj}tj|dtj}t |dkrctd dt tjdzi| |dStj |d|dz tj k}td tjt |t z||d | |dS) zL Convert DXF Spline entities into trimesh BSpline entities. rrrOr rNrrE)rknotsrr8)rjrrrrkrorrrvrlinalgnormtolmerger)rrrrr`rr_s r=convert_bsplinez)convert_entities..convert_bspline7sf!D'1T7!344;;BJGG4!!((44 v;;!   OODPPH  ! (DPQPP Q Q Q OOF # # # F q F2J 677#)C   yV--H =  $q''       r?c |d} t|d}n#t$rd}YnwxYw tjt|d}n#t$rd}YnwxYwtj|d|dgtj}|tj|tj|gz}d d g} gd t|d |d <n#t$rYnwxYw tttd z||| | |dS)zF Convert a DXF TEXT entity into a native text entity. rNrONrrrrr)leftrright72rr)originvectorheighttextalign) rgrrxrjrrkrcossinrrrro) rrranglerrrr`r_s r= convert_textz&convert_entities..convert_textZs v||~~ 1T7^^FF   FFF  JuQtW~~..EE   EEE 1T7AdG,BJ???26%=="&--888$ 2223qw<<@E!HH    D   8}}8}}q(        s33 AA'A.. A=<A= C?? D  D c dS|d}|vrdSt|dd}tj|dd|ddgtj}tj|dd |d d gtj}|\}}|D]L} | d xjt z c_M tj |tj |tj || dS) z Convert an INSERT entity, which inserts a named group of entities (i.e. a "BLOCK") at a specific location. Nrrrrrr41?rrE)rthetascale)rrrjrkrrcopyrrortftransform_points planar_matrixr) rrrrrcvcerr^r`r_s r=convert_insertz(convert_entities..convert_inserts^ > Fv v   FaeeD#&&''155s++QUU4-=-=>bjQQQ!%%c**AEE$,<,<=RZPPPB 1 1A OOAFFHH % % % RL  3x== 0     B$F"*U:K:KSXYYY       r?)LINE LWPOLYLINEARCCIRCLESPLINErUrVrrXcdS)Nrr8r8r?r=z"convert_entities..sar?r)rVrUc38K|]\}}|ddk|VdS)rrNr8)r:rr<s r= z#convert_entities..s.!P!P1AaDCKK!KKKK!P!Pr?POLYLINEVERTEXSEQENDc.g|]}|d|dgS)rrr8r:rs r= z$convert_entities..s%GGGQqw$0GGGr?rFrPc g|] \}}d|v | Srr8r:rrs r=rz$convert_entities..s!@@@tq!daiiiiir?cFg|]\}}d|v t|dSr)rrs r=rz$convert_entities..s+MMMDAq4199qw999r?rTEXTzfailed to load text entity!Tr\rTzskipping dxf entities: {}z, c3*K|]\}}|d|VdS)z: Nr8r9s r=rz#convert_entities..s0FF$!QQ++!++FFFFFFr?r)dictr rjrnrrurvronext enumerate StopIterationrrkrrboolrint64rrrrxr rsformatjoinrr vstack_emptyr)#rr~r^r[rrrrrrrloaderspolyline inflection unsupportedr entity_type index_namerrflagrrrrr chunk_rawchunkerloader entity_datarrr`rr_s# ` @@@@r=rwrws&                          &11111111f! ! ! ! ! ! ! F' ' ' ' ' ' R        J|$!#34k"(/('GHHHDAJ#-..q1Jii((K #d)) 4 4jAAR*R* u::>> U Ahqk  - - - !!P!P %0@0@!P!P!PPP '/z'B1'Ej!!$$      * $ $U }HH  !kX&=&= OODKK ( ( ( (  !kX&=&=HGG(122,GGGrzZZZEIx{""8A;t,-- NN :IueBQBi&899E@@y22@@@IHMMIh,?,?MMMjE #5IFB * *CMM) OOB    OOB   HH F " "!I#5k!!!Q$/IaaadO H T)__----  H H H 7$GGGGGG HG # #%k2OGV!'%..K F;     J & &  $ $ $ ) $ $ $ ;! ' . . FF+2C2C2E2EFFFFF      **11"*==H(QQQT c)*1-a04'' X s%AE!! E.-E.M NNc tjdd ddfdfd}fd}fd }fd }tjd ||||d }g}t |j|jD]\}} t|j} || |vr | |vrR|| ||j  } t| dkr| | vtjd| d|} dt dtj |jddD} | dt dtj |jddDt$|jd| d<djd!i|  g}| d|  | d d|t.d}t0jrxt4t5|}t|dzdksJt9d|dddDsJ|d|S)"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.jsonFTc htj|tj}tj|}|rTtjtjt|tj dd}n/tj t|dftj}|gdz }|r|ddddf}|ddddf}d d t| d | d 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 rrEr)rrr)r"r6Nr  c3.K|]\}}|dd|dVdSdr.12gNr8)r:gr<s r=rz4export_dxf..format_points..YsK  #'1aq   a         r?rE) rj asanyarrayrrstack_3Dtilervror rmzerosr zip)ras_2D incrementthreerpackeds r= format_pointsz!export_dxf..format_points3s).vRZ888 f%%  >G #e**BH555==gFFEEHc%jj!_BH===E   !!!!RaR%LE!!!RaR%LE  +.u}}R/@/@%--PRBSBS+T+T      r?cddtt|ddd}t|drtj|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.)COLORLAYERNAMErr1)rfidrcrto_asciir)entitysubss r= entity_infozexport_dxf..entity_info_s\ 6 OOCRC(   67 # # 8 M&,77DM r?c0||}t|dkrdS|}|dd|d<d|d<t||d <tt|j|d <d jd 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 rrJTF)r)r*POINTSrTYPEVCOUNTFLAGliner8)discreterorr rr )r=r_rr6rr7r-templates r=rz export_dxf..convert_linexs"x(( v;;!  2{4  &vTUKKKX#V VX4 ,,--V (&!(00400 r?c||dd}|}|j}t|dkrtj|d}dj|}|d|jdz }|jrd |d <n.d |d <|d z }|d jtj|jz }||d<djdi|}|S)NTF) return_angle return_normalr rz 10 {:.12g} 20 {:.12g} 30 {:.12g}z 40 r"rr:rz 100 AcDbArcz 50 {:.12g} 51 {:.12g}DATAarcr8) rrorjrr rrdegreesr) rDr_rr6rdatarr7r?s r=rzexport_dxf..convert_arcszz(UzKK{3 v;;!  Yvs++F=6=vF +++++ : R#DLL DL $ $D 707DK9P9PQ QDV '%'//$// r?c  ||jd}dt|jzj|jdd}ddddd d }|d }|jr ||d z}gd }gd}ddt ||D} |} | d|||d|dt|jt|jd  djdi| } | S)NFr*z 40 {:.12g} rErr rrr.)rperiodicrationalplanarlinearrKr)rrr)rc3.K|]\}}|dd|dVdSr r8)r:rjs r=rz6export_dxf..convert_bspline..s:MMAQ,,,A,,,MMMMMMr?rrr) r:r9KNOTSNORMALDEGREEr<FCOUNTKCOUNTPCOUNTbspliner8)rrorr rr r(update)spliner_rrflagsrnormaln_coden_strr6rr7r-r?s r=rz#export_dxf..convert_bsplines>x 6%HHH=3v|#4#44.convert_texts k#% h |&<NNNH z#))H"5"566G  H  S* - - VGHV - - VG__ F  H (&!(//3// r?c||S)zl For entities we don't know how to handle, return their discrete form as a polyline r8)r5r_rs r=convert_genericz#export_dxf..convert_generics |FH---r?r&) precision)rrrBezierrNrzEntity type %s not exported!rc i|] \}}d|| S)EXTMIN_r8r9s r=r>zexport_dxf..s$ T T TAMaMM1 T T Tr?XYZrc i|] \}}d|| S)EXTMAX_r8r9s r=r>zexport_dxf..s$YYYda1YYYr?rLUNITSheaderr`)rTfooterrIr c3<K|]}t|dkVdS)rN)rrs r=rzexport_dxf..(s,3313q66Q;333333r?rb)FTr8)rget_jsonrjset_printoptionsr(r`layerstype__name__r_rgrorr rsr boundsrY _UNITS_TO_DXFrrSr rhrdrstrictrfrlrtre)path only_layersrrrri conversions collectedrrr converted entities_strhsubsectionsrrrr7r-r?s @@@@r= export_dxfr s"!"677H****X2       D.#######J,....."%%%%!" KI t{33 < <5Aww  "uK'?'?  ;  ) D)!T];;AACCI9~~!!  +++ I4d ; ; ; ;99Y''L U TUBIdk!nc4R4R)S)S T T TDKKYYc%4;q>SV9W9W.X.XYYYZZZ"&&tz155DN)")11D117799:H OOHZ(///FFLLNNOOO OOHX&,,../// 99X   & &z3 7 7D zs4yy))E Q1$$$$33ccc 33333333 G Kr?c r tj|tj}tj|tj}tj|tj}tj|dk}||}||}|i}t |dkr3tjt |}t|fi|g}||fStj|dz}tj ||dz f} |r| t |z} || dddf|| dddfz } tj | d} tj tj | tj t | ftjt | dfgd z} tj| dddd f} || d} | d z tj|d z z }|tj|d z |zz }tj || dddf| | |d zz|| dddffd }t*jrd dlmfd|D}tjtj|tj|sJfd|D}tj|tj|sJgg}}|D]Z}|t7ddtjdt |zi|||[t |dz t |krtjtjt |}t=j|tj | dtj!D]r}|tddtjd t |zi||||"stj#|tj}||fS)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@rrYr)rrrr g@r)rErr ) arc_centerc0g|]}|jSr8)spanr:rrs r=rz!bulge_to_arcs..s$999azz!}})999r?c0g|]}|jSr8)rrs r=rz!bulge_to_arcs..s$;;;zz!}}+;;;r?rr8)$rjr$rr absrorvrarctanrrrcrossr'onesrunitizemeanrrrmrr|rDrallcloserrr stack_linesr boolean_rowssort setdiff1drrk)rrrrraokrr`rtidrlengthperpmidpointrrr+ check_angle check_radiir_ arc_pointsexistingline_idxrs @r=rr/s}D M%rz 2 2 2E M%rz 2 2 2E irx888I  B "IE" I 5zzQ #e**%%++(++,h Ie  s "E /9i!m4 5 5C s5zz3qqq!t9 c!!!Q$i 0 0FY^^F^ + +F 8 #f++!6!6788 Va !!III-  D <QQQU $ $DSzA&&HslbfUS[11 1FbfUQY''&0 0F O s111a4y 8dV^^G-D-D&DDeCPQPQPQSTPTIFVW   gj  z8$$$$$$99995999 {26%=="&*=*=>>>>>;;;;U;;; {;v77777RhH$$ LL29Q<<#h--#?L8LLMMM #### E Q#e**$$#BIc%jj$9$9:: - bgc***BL   4 4H OODQQ ! s8}}(DQQQ R R R OOE(O0022 3 3 3 3x 333H X r?c |tj|dddf|kdddz}n#t$rYdSwxYw|d|kr. t|dS#t$r |dcYSwxYwdS)zN Given a loaded (n, 2) blob and a field name get a value by code. Nrr)rjrn IndexErrorr ValueError)rfieldcoder=s r=rqrqs BJtAAAqDzU233A6q9A=> tt Aw$ tAw<<    7NNN ts!7: AAA--BBdxf)NNF)N)FN)$ collectionsrnumpyrjrJrrrrr constantsr r rr rDr r`rrrrrrrr{rdXRECORD_METADATAXRECORD_SENTINELXRECORD_MAX_LINEXRECORD_MAX_INDEXrrwrrrq _dxf_loadersr8r?r=rsi######))))))))))%%%%%%((((((////////////xvw}  }  x |}v       !" #$ ) .65*"2"2"4"4555  'DeeeeP LLLL^D&x r?