TiddlZddlZddlmZmZddlZ ddlmZn##e $rZ ddl m Z e e ZYdZ [ ndZ [ wwxYwddl mZddlmZmZddlmZdd lmZmZmZdd lmZdd lmZdd lmZmZ d&dedeededededeedededefdZ d'dZ!dZ"dZ#dZ$dededefd Z%d(d!Z& d)d$Z'd%e iZ(dS)*N) defaultdictdeque)Image)ExceptionWrapper)util)logtol) ResolverLike)DictLoadableOptional)to_floatSimpleMaterial)TextureVisuals unmerge_facesTFfile_objresolvergroup_materialskip_materialsmaintain_ordermetadata split_objects split_groupsreturnc +|} tj| } d| dd} | dd} i} | d} |s| dkr| | dz| d| } t|| | } d | D} nR#ttf$rtj d | Yn(t$rtj d | d YnwxYwt| \}}}}t| ||}|s|s|rt!||||}t#|s|d|i}|||d<|||d<|Si}t#|dkr|\}}}}dt'jd|t&jddD}t#|ddd+t-+fd|D}|ryt/jd|dddt.j}||dkxxdzcc<t7|+|d\}}}n'tj dt9|\}}}g}|r||||r||||r7t#| dkr$|"|t=||rd|} n?||} n:t?dtA|ddtA|d dd!fD} tj!| |} |"}!||9t#|t#|krtG||||"\}"}#}$}%nd}%tG|||"\}"}#}$tHj%rVt/j&||||#|"sJ|"'t#||#ksJ ||$}&n&#t$rtj d#d}&YnwxYw|!(||#"|"d$nSd}&tHj%r'|'t#|ksJ|4t/j)||j)krtG|||"\}"}#}%n|r)t/j*t#|tV%}#n(t/j,t#|tV%}#d |#|<t/j,t#|t.j%}'t/j-|#.|'|#<|'|}"d}%|!(|"||#"d&|3 ||#|!d<n&#t$rtj d'd YnwxYw|% ||%}(|(j)|!dj)krMt_d(t=|(j)t=|!dj)|(|!d<n&#t$rtj d)d YnwxYwd})|| vrta|&| |*})nS|&7t#|&t#|!dkrta|&+})n|tj d,|d-|)|!d.<|!|| <t#|dkd/|1D}*||*d0S)1a 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)rc.i|]\}}|tdi|S)r.0kvs N/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/exchange/obj.py zload_obj..as,TTTDAqN//Q//TTTzunable to load materials from: Texc_info)textNverticesvertex_normals vertex_colorscjg|]0}|ddd1Srr)splitstripr%is r( zload_obj..sG    GGD!  Q  % % ' '   r*z^f)flagsr3/ c3K|]=}t|ddkV>dS)r:r;N)lenreplacer4)r%fcolumnss r( zload_obj..sJYYC #s(;(;(A(A(C(C$D$DDYYYYYYr*sepdtype)arrayr@ sample_linez+faces have mixed data: using slow fallback!_c3K|]}||V dSNr#r6s r(rAzload_obj..s4 =  !===r* file_namenamegeometry)maintain_facesz)index failed on UV coordinates, skipping!)r.facesrD)rNr.zfailed to load vertex_colorszincorrect normals {} != {}zfailed to load vertex_normals)uvmaterial)rPzspecified material (z) not loaded!visualcg|]}||dS))rLframe_tor#)r%r&s r(r8zload_obj..:s E E E! + + E E Er*)rLgraph)2readr decode_textformatr5r>find parse_mtlitemsOSError TypeErrorr debug BaseException_parse_vertices_preprocess_faces _group_byr=poprer4 MULTILINEallnp fromstringjoinint64_parse_faces_vectorized_parse_faces_fallbackappendstrnextgetattr unique_namecopyrr strictallclosemaxupdateshapeonesboolzerosarangesum ValueErrorrkeys),rrrrrrrrkwargsr- materials mtl_positionmtl_pathmaterial_kwargsr'vnvtvc face_tuplespcrLrQcurrent_object current_groupchunk face_lines flat_arrayrErN faces_tex faces_norm name_partsrKmesh new_facesmask_vmask_vtmask_vnrPinversenormalsrRrUr@s, @r(load_objrs ^ ==??D  D ! !D ??4::<<//== > >D << # #DI99X&&L Sla// q(499T<+H+HHIOOQQ S'(:XNNNOTTO#%B >"$B  H k  Q  9D9J9J6.-  XdE>>>qrrB   jm++C55;;==>>YYYYjYYYYY   MM$$,,S#66CrxE %!)    !   ,CW*Q-,,, (E9jj IC D D D+@+L+L (E9j  .^7   n - - -  -M5   m , , ,  -c)nnq00X5I   c(mm , , ,  88J''DD  '!DDHk488Hfd33Dh//{{}}  %#j//SZZ*G*G6C9j7773 67GG-:9^...* 67z 8{1U8QvYy-ABBBBB }}QvY7777 [     EFFF  KKQvY^^%5%5 JJ K K K KBz ,yy{{SVV++++~"(:"6"6%+"E"E-::n...* 677":WSVV4888FFXc!ffD999F $u (3q66:::"$)FJJLL"9"9#EN  KK)69I9IJJ K K K > I(*6 _%%  I I I 84HHHHHH I   JW+=D$4$:::$4;; ..D4D4J0K0K *1%&&  J J J 9DIIIIII J y #rIh4GHHHFF ^B3tJ/?+@+@ @ @#r***FF  ! IEXEEE F F FXe k  Q  j F EX]]__ E E EE!5 1 11sO5D(E."EE9S S%$S%0 Y<< ZZ%A0\ \98\9c *tj|}d}i}tt|}ddddd}|D])}|}t |dkr=|d}|d kr-| |||d <d d |ddi}|d kr| d d z} || d} | | } tj tj | |d<tjtjt#|dd| |djd<r#t&$rt)jddYwxYw||vrk d|ddD} t | dkr| d} || |||<| ||<#t&$rt)jddYwxYw| |dd||<+|r |||d <|S)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)kdkaksnsr3rnewmtlrKr;map_kdr!imageparentr file_pathzfailed to load imageTr+c,g|]}t|Sr#)float)r%xs r(r8zparse_mtl..s555aq555r*zfailed to convert color!)rrWrn splitlinesr5r4r=lowerriindexgetropenwrap_as_streamospathabspathrpinfor_r r^r~) mtlrrQrlinesmappedliner4keyrrJ file_datavalues r(rZrZ@s"  3  CHI NN3s88>>++ , ,EYj U UF5&5& ""$$ u::?? Ahnn (??#.6 (6*+qrr 3 34HH H__JJLL&&x0014EUVV **,,I A$LL33 %*Jt/B9/M/M$N$N!68gooGLL8R!@!@)LL77!&{33! A A A 04@@@@@@ AFKKMM ! ! E5559555u::??!!HE',1HVC[)$)HSM  E E E 4tDDDDDD E !!!""IHSM/&. (6"# s&3BG  G10G1 AI I21I2c|d|f}t|}t ||z }t j||z}|dd|f}d\}}||dzkrdd|Dd} | |kr|dd|dzf}ne| t |dz kr|dd|dzf}n?tj d |n'||d zkr|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|]}|dVdS)r:N)r5r6s r(rAz*_parse_faces_vectorized..s,BB BBBBBBr*r:r3zface lines are weird: ) reshaper=r5r4intrgr{ricountr r^) rEr@rF group_countper_refrrNrrrs r(rkrks0 MM2w- ( (Ek''))//1122K'K'((G Ik " "W ,E !!!U(OE'Iz+/!! BBk.?.?.A.ABBBBBHHMM G  qqq%!)|,JJ c'A+&& & &aaal+II I<{<< = = = = K!O # #!!!UQY,' 111eai<( )Z ''r*c ggg}}}|D]}|dd t }|dkrn|dkr- d d d d d dg nd|dkr8g}|j fdt t dz D| n&t jdt d  D]}|d  |t d |t dn#t$rYnwxYw |t d#t$rYwxYwtj |tj  d }||dkxxdzcc<d \} } t|t|krGtj |tj  d } | | dkxxdzcc<t|t|krGtj |tj  d } | | dkxxdzcc<|| | fS)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 rrrr3rc g|]:}d|dz|dzg;S)rr3rr#)r%r7collect_appendr4s r(r8z)_parse_faces_fallback..sh     #N58,,"N5Q<00"N5Q<00   r*zface needs more values 3>z skipping!r:rO)rrr) r5r4r=rmranger r^rr_rgrErjr) rr'rrr len_splitcollectr?rNrrrr4s @@r(rlrls"B2rA.. ""4((+1133JJ >>  !^^1XuQxq58U1XuQxPEE ]]G%^N      s5zzA~..     EE IH#e**HHH I I I   AGGCLLE HHSq]] # # #  #eAh--((((       #eAh--((((      & HQbh ' ' ' / / 8 8E %!)#Iw 2ww#a&&HRrx00088AA )a-   A%    2ww#a&&(2RX...66w??! ! )W $$s$<(E%% E21E26(F F,+F,ctfddD}td|DsdSfd|D}td|D}t d|D}||dd d d fd |D}d |D}t d }|D]\}} tjd | dtj } t| ||f} t| tj | kr| | ||<gfd| DtdDtjfdDtj ||<|d} d} | .| jddkr| ddddf| ddddf} } n!| | jddkr| ddddf} |d}||ddddf}|d}t jr|t| dksJ|(t|dksJ|(t|dksJ| ||| fS)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 cDi|]}|d|dS)rr;)rY)r%r&r-s r(r)z#_parse_vertices..Ks/ A A A!a9999%% A A Ar*)r'rrc3"K|] }|dkV dSrNr#)r%r's r(rAz"_parse_vertices..Ns&//!qAv//////r*)NNNNc i|]L\}}|dk |dd|ddzt|zMS)rrr;r)rYrfindr=)r%r&r'r-s r(r)z#_parse_vertices..Rsb    Aq 66 499T4::i1iii0014s1vv= > > 66r*c3&K|] }|dk|V dSrr#)r%ss r(rAz"_parse_vertices..Ys&55aa1ffffff55r*c3&K|] }|dk|V dSrr#)r%es r(rAz"_parse_vertices..Zs& 1 1A!q&&a&&&& 1 1r*z+erz-ec zi|]7\}}|dk |dd|dddD8S)rcFg|]}|dddSr2r4r6s r(r8z._parse_vertices...`s+ D D DAAGGD!  Q  D D Dr*rr;r3Nr)r%r&r'rs r(r)z#_parse_vertices.._s\    Aq 66 D DY!YYY)?)?)C D D D 66r*cdi|]-\}}|t|d.Sr)r=r4r$s r(r)z#_parse_vertices..fs2===1q#adjjll##===r*cdSrIr#r#r*r(z!_parse_vertices..isr*r;rBc\g|](}fdt|D)S)cg|];}|..xs7 I I I!ell17799??,,-- I I Ir*)rnr)r%r7rs r(r8z#_parse_vertices..xs: Y Y Ya I I I Is~~a7H7H I I I Y Y Yr*c34K|]}t|VdSrI)r=)r%Ls r(rAz"_parse_vertices..zs(..1A......r*c$g|] }|d SrIr#)r%rrs r(r8z#_parse_vertices..|s!!;!;!;!FUF)!;!;!;r*rOr'Nr3r!rrrr v  vn  vt )anyvaluesr[minrur>rrgrhrifloat64r=prodrrErwr rsr)r-startsendsstartenddataper_rowresultr&rrErwr'rrrrrrs` @@@r(r`r`0s|6B A A A/@ A A AF //v}}/// / /&%%    LLNN   D 556==??555 5 5E 1 1 1 1 1 1 1C sO # #D# . . 6 6tS A AE    LLNN   D>= ===G & &FJJLLOO5 chhuoo3bjIIIUWQZ( u:: ' ' e,,F1II E Y Y Y YSX Y Y Y Y.......E!;!;!;!;U!;!;!;2:NNNF1II s A B}q!!!RaR%!AAAqsF)2 171:>> aaa!eH B ~ 2A2Y B z21vvF++++++ >r77djj111111 >r77djj111111 b"b=r*use_mtluse_obj use_groupctd}|D]Z\}}}}|r|nd|r|nd|r|ndf} ||| d<||| d<||| d<|| d|[|D],} d|| d|| d<-t |S)a For chunks of faces split by material group the chunks that share the same material. Parameters ------------ face_tuples : (n,) list of (material, obj, chunk) The data containing faces use_mtl Group tuples by `usemtl` commands use_obj Group tuples by `o` commands use_group Group tuples by `g` commands Returns ------------ grouped : (m,) list of tuples containing: `(material name, objectname, group name, raw chunk)` cdddggS)Nrr#r#r*r(rz_group_by..s2r2r"2r*Nrr3rrr)rrmr~rilistr) rrrrgroupedrQobjgrouprrs r(rbrbs.2233G'2 & &##ue )HHT $CC (EED  # Q Q Q Qu%%%%||~~55))GCLO44 Q   ! !!r*c vgd}t|}|D](}||d|}|dkr ||kr|})|d|ddz}|dkr |||}n ||d}tjr.|d|dksJt jdtj d|Dt } dt|g| g} |rK| t jd tj d |Dt |rK| t jd tj d |Dt t j t j t| } d} d} d}g}t| dd| ddD]7\}}|||dz}|dr6|dd\} }| dd} n|dr6|dd\} }| dd} nJ|dr5|dd\}}|dd}|dsd|vr| | | ||f9|S)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 rrrrNc8g|]}|dSrrr%ms r(r8z%_preprocess_faces..s"LLLq LLLr*usemtl rOc8g|]}|dSrrrs r(r8z%_preprocess_faces.."GGGQaggajjGGGr*rc8g|]}|dSrrrs r(r8z%_preprocess_faces.. rr*rrr3zo rusemtlr!zg f z f)r=rYrr rsrrgrErdfinditerrrmunique concatenatetuplezipr5 startswithr4)r-r use_groupsstartersf_startstsearchf_endf_chunkidx_mtl split_idxssplits current_obj current_mtlrrrrrs r(rarasg(=<.s+;;;$!Q1  ;;;r*)-ris_instance_nameddumpr}rsetrRkindr=r0rg column_stackr.rarray_to_string_cachecacher/rmr_r r^hasattrrQr&hashrqrKaddto_objrwrprP appendleftr rNrrXrirr[rendswithwarningencode)"rinclude_normals include_colorinclude_texturereturn_texture write_texturerr%mtl_nameheader face_formatsmeshesobjectscountsrmaterials_namecurrent face_typev_blobexport convertedrQhashedrKtex_name face_formatmtl_datamtl_librrGrJrprependr-s" ` r( export_objrQ+sh% L dI..6 g . .6 l 3 364555ggGAQ ' 'FIUUNf*f*E   &#'999GN011:_!8GN,HBQB,O#P#PQFF %F&cVF     ".'.2FFO  M M 0*!%!    &&& ei/0000  M M M 28[114'1133Hh**+HM>JJD"&&t,,,(0T(B(BIf%%V,Q/rxd C CDDEEJJ $ 4)SGTZ!!!I$$T***MM%)"3444!!"6H"6"67777  M M M v.FGG H H Htyy(())))H 9~~!'')) < r^s ********     .-----  Q  EEEEEE   $$$$$$,,,,,,,,,,######,,,,,,:::::::: (,  #b2b2b2|$b2b2 b2  b2 tn b2b2b2 b2b2b2b2J WWWWt@(@(@(FP%P%P%fmmm`)"D)"4)"D)")")")"X\\\\B   .iiiiXx s=8=