:vjxvddlZddlZddlmZmZddlZ ddlmZddl mZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZ d"d ededzdededededzdededefdZd#dZ dZ!dZ"dZ#dededefdZ$d$dZ% d%d Z&d!eiZ'y#e $rZ ddl m Z e e ZYdZ [ dZ [ wwxYw)&N) defaultdictdeque)Image)ExceptionWrapper)util)logtol) ResolverLike)Loadable)to_float)SimpleMaterial)TextureVisuals unmerge_facesfile_objresolvergroup_materialskip_materialsmaintain_ordermetadata split_objects split_groupsreturnc X-|j} tj| } dj| j j dd} | j dd} i} | j d} |sh| dk\rc| | dz| j d| j } t|| | } | jDcic]\}}|td/i|} }}t| \}}}}t!| ||}|s|s|rt#||||}t%|s|d|i}|||d<|||d<|Si}t%|dkDr|j'\}}}}t)j*d|t(j,ddDcgc]%}|j+dddj '}}t%|dj ddj+-t/-fd|D}|rkt1j2dj5|j dddt0j6}||dkDxxdzcc<t9|-|d\}}} n$tjdt;|\}}} g}!|r||!j=||r||!j=||r*t%| dkDr||!j=t?||!rdj5|!}"n0||}"n+tAdtC|ddtC|dddfD}"tjD|"|}"|jG}#|| ,t%| t%|k(rtI||| | \}$}%}&}'nd}'tI||| \}$}%}&tJjLrBt1jN||||%|$sJ|$jQt%||%ksJ ||&}(|#jS||%jG|$d"n7d}(tJjLr|jQt%|ksJ|5t1jT| |jTk(rtI|| | \}$}%}'n|r%t1jVt%|tX#}%n$t1jZt%|tX#}%d |%|<t1jZt%|t0j6#})t1j\|%j_|)|%<|)|}$d}'|#jS|$||%jGd$| ||%|#d<|'k ||'}*|*jT|#djTk7rDtad&jt?|*jTt?|#djT|*|#d<d}+|| vrtc|(| |(}+nD|('t%|(t%|#dk(r tc|()}+n|tjd*|d+|+|#d,<|#||"<t%|dkDr|jeDcgc]}||d- },}||,d.Scc}}w#ttf$rtjd | Yt$rtjd | d YwxYwcc}w#t$rtjd!d}(YwxYw#t$rtjd%d YwxYw#t$rtjd'd YiwxYwcc}w)0a Load a Wavefront OBJ file into kwargs for a trimesh.Scene object. Parameters -------------- file_obj : file like object Contains OBJ data resolver : trimesh.visual.resolvers.Resolver Allow assets such as referenced textures and material files to be loaded group_material : bool Group faces that share the same material into the same mesh. skip_materials Don't load any materials. maintain_order Make the strongest attempt possible to not reorder faces or vertices which may result in visual artifacts and other odd behavior. The OBJ data structure is quite different than the "flat matching array" used by Trimesh and GLTF so this may not be completely possible. split_objects Whenever the loader encounters an `o` directive in the OBJ file, split the loaded result into a new Trimesh object. split_groups Whenever the loader encounters a `g` directive in the OBJ file, split the loaded result into a new Trimesh object. Returns ------------- kwargs : dict Keyword arguments which can be loaded by trimesh.exchange.load.load_kwargs into a trimesh.Scene z {} z  z\ mtllibr)rzunable to load materials from: Texc_info)textNverticesvertex_normals vertex_colorsz^f)flags/ c3vK|]0}t|jddjk(2yw)r'r(N)lenreplacesplit).0fcolumnss e/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/exchange/obj.py zload_obj..s0YjC #s(;(A(A(C$DDjs69sepdtype)arrayr/ sample_linez+faces have mixed data: using slow fallback!_c3$K|]}|| ywNr-is r0r1zload_obj..s#A = s file_namenamegeometry)maintain_facesz)index failed on UV coordinates, skipping!)r"facesr4)rAr"zfailed to load vertex_colorszincorrect normals {} != {}zfailed to load vertex_normals)uvmaterial)rCzspecified material (z) not loaded!visual)r?frame_to)r?graphr:)3readr decode_textformatstripr+find parse_mtlitemsrOSError TypeErrorr debug BaseException_parse_vertices_preprocess_faces _group_byr*poprer, MULTILINEallnp fromstringjoinint64_parse_faces_vectorized_parse_faces_fallbackappendstrnextgetattr unique_namecopyrr strictallclosemaxupdateshapeonesboolzerosarangesum ValueErrorrkeys).rrrrrrrrkwargsr! materials mtl_positionmtl_pathmaterial_kwargskvvnvtvc face_tuplespcr?rDcurrent_object current_groupchunkr< face_lines flat_arrayr5rA faces_tex faces_norm name_partsr>mesh new_facesmask_vmask_vtmask_vnrCinversenormalsrErGr/s. @r0load_objrs*^ ==?D   D !D ??4::<//= >D << #DI99X&L la/ q(499T<+HIOOQ S'(:XNO#%B >"$B  H k Q 9D9J6.-XXdE>qrB B GGD! Q  % % 'B  jm++C5;;=>YjYY  MM$,,S#6CrxxE %!)  ! ,CW*Q-, (E9j IIC D+@+L (E9j ^7   n - M5   m , c)nq0X5I   c(m , 88J'D  '!DHk48Hfd3Dh/{{}  %#j/SZ*G6C9j73 67G-:9^.* 67zz{{1U8QvYy-ABBB }}QvY777 [ KKQvY^^%5 J KBzzyy{SV+++~"((:"6%++"E-::n.* 67"WWSV48FXXc!fD9F $u ((3q6:"$))FJJL"9#EN  KK)69IJ K > I(*6 _%   JW+==D$4$:$::$4;; .D4D4J4J0K *1%& y #rIh4GHF ^B3tJ/?+@ @#r*F  ! II,XJnE FXe k Q j6>]]_ E_! +_E E!5 11yU# D II7zB C S II7zBT R R S^ v!  EF V! I 84HH I! J 9DII J& Fsm)#Y Y"Y *Z-Z2[A*\ \'Y'Z*"Z*)Z*2 [[ [=<[= \$#\$ctj|}d}i}tjt|j }ddddd}|D]Q}|j j }t |dkr1|dj}|d k(r!||||d <d d j|ddi}j|d k(r|jjd d z} || dj } |j| } tjtj| |d<| |djd<||j%vr@ |ddD cgc] } t'| } } t | dk(r| d} | | |||<| ||<F|J|dd||<T|r|||d <|S#t$rt!j"ddYwxYwcc} w#t$rt!j"ddYwxYw)a8 Parse a loaded MTL file. Parameters ------------- mtl : str or bytes Data from an MTL file resolver : trimesh.Resolver Fetch assets by name from file system, web, or other Returns ------------ mtllibs : list of dict Each dict has keys: newmtl, map_Kd, Kd Ndiffuseambientspecular glossiness)kdkaksnsr&rnewmtlr>r(map_kdrimage file_pathzfailed to load imageTrzfailed to convert color!)rrIra splitlinesrKr,r*lowerr\indexgetropenwrap_as_streaminforRr rQrqfloat)mtlrrDrslinesmappedliner,keyrr= file_dataxvalues r0rMrM@s$"   3 CHI NN3s8>>+ ,EYj UF ""$ u:? Ahnn (?#.6 (6*+qr 34H H_JJL&&x014EUV **,I A$LL3 %*JJt/B/B9/M$N!6?!&&{3 FKKM ! E+0959aq95u:?!!HE',1HVC[)$)HSM !!!"IHSMgj&. (6"# 3! A 04@@ A 6! E 4tDD Es74AF0GG/$G0 GGG H?Hc4|jd|f}t|jj}t ||z }t j ||z}|dd|f}d\}}||dzk(r~djd|jDjd} | |k(r |dd|dzf}nW| t |dz k(r |dd|dzf}n9tjd |n ||d zk(r|dd|dzf}|dd|dzf}|||fS) a Parse loaded homogeneous (tri/quad) face data in a vectorized manner. Parameters ------------ array : (n,) int Indices in order columns : int Number of columns in the file sample_line : str A single line so we can assess the ordering Returns -------------- faces : (n, d) int Faces in space faces_tex : (n, d) int or None Texture for each vertex in face faces_norm : (n, d) int or None Normal index for each vertex in face NNNrrc3>K|]}|jdyw)r'N)rKr;s r0r1z*_parse_faces_vectorized..sB.A .Asr'r&zface lines are weird: ) reshaper*rKr,intrZrnr\countr rQ) r5r/r6 group_countper_refrrArrrs r0r^r^s;0 MM2w- (Ek'')//12K'K'(G IIk "W ,E !U(OE'Iz+/! Bk.?.?.ABBHHM G q%!)|,J c'A+& &al+I II.{m< = K!O #!UQY,' 1eai<( )Z ''c ggg}}}|D]W}|jjddj}t|}|dk(rn|dk(r|d|d|d|d|d|dg}n|dkDrZg}|j}t t|dz D cgc](} ||d||| dz||| dzg*c} |}n#t j dt|d|D]l} | jd }|jt|d |jt|d |jt|dnZtj|tj jd } | | dkDxxdzcc<d \} } t|t|k(rDtj|tj jd } | | dkDxxdzcc<t|t|k(rDtj|tj jd } | | dkDxxdzcc<| | | fScc} w#t$rY8wxYw#t$rYwxYw) aC Use a slow but more flexible looping method to process face lines as a fallback option to faster vectorized methods. Parameters ------------- lines : (n,) str List of lines with face information Returns ------------- faces : (m, 3) int Clean numpy array of face triangles rrrr&rzface needs more values 3>z skipping!r'rB)rrr) rKr,r*r`ranger rQrrRrZr5r]r)rrxrzryrr, len_splitcollectcollect_appendr<r.rArrs r0r_r_s}"B2rA ""4(+113J >  !^1XuQxq58U1XuQxPE ]G%^^Ns5zA~.   /A #58,"5Q<0"5Q<0 /  E II1#e*ZH I AGGCLE HHSq] #  #eAh-(  #eAh-(?d HHQbhh ' / / 8E %!)#Iw 2w#a&HHRrxx088A )a- A%  2w#a&((2RXX.66w?! ! )W $$W .!   !  s*-I($I-I=- I:9I:= J  J c dDcic]}||jd|d}}td|jDsy|jDcic]?\}}|dk\r5||jd|j d|ddzt |zA}}}t d|jD}td |jD}|||jd d jd d }|jDcic]D\}}|dk\r:||jd|dd dDcgc]}|jdd dc}F} }}}| jDcic]"\}}|t |dj$} }}td} | jD],\}} tjdj| dtj} t | | |f}t | tj|k(r| j!|| |<g}| Dcgc]P}t"j%|Dcgc]/}|j'|j)j1c}Rc}}t d|D}tj*|Dcgc]}|d| c}tj| |</| d}d}|)|j,d dk\r|ddddf|ddddf}}n||j,d dkDr |ddddf}| d}| |ddddf}| d}t.j0rat ||j3dk(sJ|t ||j3dk(sJ|t ||j3dk(sJ||||fScc}wcc}}wcc}wcc}}}wcc}}wcc}wcc}}wcc}w)a Parse raw OBJ text into vertices, vertex normals, vertex colors, and vertex textures. Parameters ------------- text : str Full text of an OBJ file Returns ------------- v : (n, 3) float Vertices in space vn : (m, 3) float or None Vertex normals vt : (p, 2) float or None Vertex texture coordinates vc : (n, 3) float or None Per-vertex color )rxrzryrr(c3&K|] }|dk\ ywrNr:)r-rxs r0r1z"_parse_vertices..Ls/!qAvs)NNNNrrc3,K|] }|dk\s |ywrr:)r-ss r0r1z"_parse_vertices..Ws5?aa1f? c3,K|] }|dk\s |ywrr:)r-es r0r1z"_parse_vertices..Xs 1A!q&arz+erz-er&Ncyr9r:r:rr0z!_parse_vertices..gsrr2c32K|]}t|ywr9)r*)r-Ls r0r1z"_parse_vertices..xs.1AsrBrxrrrzry v  vn  vt )rLanyvaluesrNrfindr*minrhr+r,rrZr[r\float64prodrrarr`rKr5rjr rfr)r!rwstartsrxendsstartendrr<dataper_rowresultrr5rjrrrr{rzrys r0rSrS.s60A A/@!aRs!9%%/@F A /v}}/ /% LLN "DAq 6 499T4::1#Qi014s1v= >>"   56==?5 5E 1 1 1C sO # #D# . 6 6tS AE LLN "DAq 6 r!AY)?)C D)CAAGGD! Q )C DD"  15 = 1q#adjjl## G= &FJJL5 chhuo3bjjIUWQZ( u: ' e,F1I ESX YSXas~~a7H I7H!ell1779??,-7H ISX Y...EU!;U!FU)U!;2::NF1I%!* s A B}q!RaR%!AqsF)2 1771:> a!eH B ~ 2A2Y B zz1vF++++ >r7djj11 11 >r7djj11 11 b"b=eB  E >$J Y".s 2r2r"2rNrr&rrr)rr`rqr\listr) r|rrrgroupedrDobjgrouprrs r0rUrUs.23G'2##ue HTCED  # Q Q Q Qu%(3||~))GCLO4 Q  !!rc gd}t|}|D]#}|j|d|}|dkr||ks"|}%|jd|jddz}|dk\r|||}n||d}tjr%|j d|j dk(sJt jtjd|D cgc]} | jdc} t} dt|g| g} |r[| jt jtjd |D cgc]} | jdc} t|r[| jt jtjd |D cgc]} | jdc} tt jt jt| } d} d}d}g}t!j"| D]\}}|||j%dz}|j'd r)|j)dd \} }| d dj%} ns|j'dr)|j)dd \}}|ddj%}n9|j'dr(|j)dd \}}|d dj%}|j'dsd|vs|j|| ||f|Scc} wcc} wcc} w)a Pre-Process Face Text Rather than looking at each line in a loop we're going to split lines by directives which indicate a new mesh, specifically 'usemtl', 'o', and 'g' keys search for materials, objects, faces, or groups Parameters ------------ text : str Raw file Returns ------------ triple : (n, 3) tuple Tuples of (material, object, data-chunk) )z usemtl  o  f  g z s rrrrNusemtl rBrrzo r&rusemtlrzg f z f)r*rLrr rfrrZr5rWfinditerrrr`unique concatenatetuple itertoolspairwiserK startswithr,)r!r use_groupsstartersf_startstsearchf_endf_chunkmidx_mtl split_idxssplits current_obj current_mtlrr|rrrs r0rTrTs(=H$iG2q'* A:  G G IIdDJJv.2 3E zwu%wx. zz}}V$ 6(:::: hhBKK 7,KL,Kq ,KLTWXGc'l#W-J HH"++fg*FG*FQaggaj*FGs S  HH"++fg*FG*FQaggaj*FGs S YYr~~eJ&78 9FKKMK((0 sc"((*T1   D !!&T1!5 K%ab///1K   h '!&T1!5 K%ab///1K   d ##(;;tQ#7 M5)!"-335M   D !Ue^    [-O P!1" UMH HsK<0L L c  ddddd} tj|dr|g} nLtj|dr|j} n%tj|dr|g} n td t } d d d d } i}t }| D]7}d g}|r|j jd vrgt|j jrHtj|jt|j jddddff}n |j}t dtj|dd|zg}|d|jj v}|rI tj|j"dd|}|j%d|j%d|z|rCt-|j dr, |j j.}t-|dr|j1}t3|}||vrFtj4|j6|}|j9||j;|||<||d}ttj<t?|j dddk(rRtj|j j@dd |}|j%d!|j%d"|z|jCd#|| tE|}t-|d%r>|j%d&tj|jFdz| d zdd'|(z| d xxt|jz cc<d)|jHvr-|jCd*jK|jHd)| j%d+jM|:i}t|d kDrg}|jOD]p\}}|jQD]X\}}|jSjUd,r|j%|7||vr|||<At)jVd-|Zr|d.}| d/| d0jY} nd1} | d2jM|z||<| jCd3|| | jCd/| | j%d+d+jM| }!|rC|At|d kDr3|jQD"#cgc]\}"}#|j[|"|#c}#}"|r|!|fS|!S#t&$rt)j*ddYwxYw#t&$rt)j*d$dYwxYwcc}#}"w)4a Export a mesh as a Wavefront OBJ file. TODO: scenes with textured meshes Parameters ----------- mesh : trimesh.Trimesh Mesh to be exported include_normals : bool or None Include vertex normals in export. If None will only be included if vertex normals are in cache. include_color : bool Include vertex color in export include_texture : bool Include `vt` texture in file text return_texture : bool If True, return a dict with texture files write_texture : bool If True and a writable resolver is passed write the referenced texture files with resolver resolver : None or trimesh.resolvers.Resolver Resolver which can write referenced text objects digits : int Number of digits to include for floating point mtl_name : None or str If passed, the file name of the MTL file. header : str or None Header string for top of file or None for no header. Returns ----------- export : str OBJ format output texture : dict Contains files that need to be saved in the same directory as the exported mesh: {file name : bytes} z{}z{}//{}z{}/{}z{}/{}/{}))rx)rxry)rxrzrxryrzTrimeshScene PointCloudzmust be Trimesh or Scene!rrrx)vertexfaceNrzv r(r) col_delim row_delimdigitsr#rryzvn z failed to convert vertex normalsTrrC to_simple)r>r&rrrzzvt rz failed to convert UV coordinatesrArr)rr value_formatr>z o {}rz.mtlz not writing z material.mtlz# z rs zmtllib ).ris_instance_nameddumprprsetrEkindr*r$rZ column_stackr"r array_to_string_cachecacher#r`rRr rQhasattrrDrhashrdr>addto_objrjrcrC appendleftrrArrJr\rrNrendswithwarningencodewrite)$rinclude_normals include_colorinclude_texturereturn_texture write_texturerrmtl_nameheader face_formatsmeshesobjectscountsrsmaterials_namecurrent face_typev_blobexport convertedrDhashedr>tex_name face_formatmtl_datamtl_librr7r=rprependr!rwrxs$ r0 export_objr0)sKh% L dI.  g .  l 3455gGAQ 'FIUNE  ##'99GNN001__!!8GNN,H,HBQB,O#PQF %%F&&cVF    ".'..2F2FFO  M 00**!%!    & ei/0 ww~~t< M">>228[1'113Hh*++HMM>JD"&&t,(0T(BIf%%V,Q/rxxd CDEJ $ 4 4))SGTZ!I$$T*MM%)"34!!GH:"67 #5#34 7G $ MM&&MMA%s 3!$!,   s s7++,,  W%% %   hoog.>.>v.FG Htyy()MRH 9~!'')GD!(, $ 9??$--f5NN9-h.*3HY'KK,yk :;)5*  %H  6($'..0GG%w||G'<<WXJ/0 Rx=) NN4 99W D-#i.12D*2..*:;*:$!Q1 *:;X~ Kc! M r?sH *   $#,: %)   b2b2T!b2b2 b2  b2 Tk b2b2b2 b2J Up@(FP%fm`)"D)"4)"D)"X\B   .iXx U  . Q E  sBB8 B33B8