:vj 2ddlZddlZddlZddlmZddlZddlmZm Z ddl m Z m Z m Z mZddlmZddlmZddlmZdd lmZid d d d dddddddd dddddddddd ddddddd d!d"d#d$d!d%d%d&Zd d ddddddd d"d'd( Zd)Z dCd*edzd+ed,edzd-efd.Zd/Zd0Zd1Zd2Z dDd3edzd4efd5Z!d6Z"dEd7Z#d8Z$d9Z%d:Z&d;Z'd<Z(dFd=Z)d>Z*d?eiZ+d?e!iZ,e jZd@Z.e jZdAZ/e/e*e+dB<e.e)e,dB<yy)GN)Template)structured_to_unstructuredunstructured_to_structured)grouping resourcesutilvisual)log)triangulate_quads)Resolver) ArrayLikechari1ucharu1shorti2ushortu2inti4int8int16int32int64i8uintu4uint8uint16uint32uint64u8floatf4float16f2float32f8)float64doubler,) rrrrrrrr$r&r(r*c.t|jddS)z Returns the closest ply equivalent of a numpy type Parameters --------- _numpy_type : a numpy datatype Returns --------- ply_type : string N)_inverse_dtypesstr) _numpy_types e/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/exchange/ply.py_numpy_type_to_ply_typer36s ;??12. //resolver fix_texture prefer_colorskip_materialsct|\}}} |r t||n t||d} |sF ddl} | ?|j | } | j j tj| } t| |||}|S#t$rtjdY1t$rtjddYRwxYw)a Load a PLY file from an open file object. Parameters --------- file_obj : an open file- like object Source data, ASCII or binary PLY resolver Object which can resolve assets fix_texture If True, will re- index vertices and faces so vertices with different UV coordinates are disconnected. skip_materials If True, will not load texture (if present). prefer_color None, 'vertex', or 'face' Which kind of color to prefer if both defined Returns --------- mesh_kwargs : dict Data which can be passed to Trimesh constructor, eg: a = Trimesh(**mesh_kwargs) Nrz%textures require `pip install pillow`zunable to load image!Texc_info)imageelementsr6r7) _parse_header _ply_ascii _ply_binary PIL.ImagegetImageopenr wrap_as_stream ImportErrorr debug BaseExceptionwarning_elements_to_kwargs) file_objr5r6r7r8argskwargsr=is_ascii image_namer<PILdatas r2load_plyrREsH&38%<"Hh 8X&Hh' E  @ %||J/ t':':4'@A!hKlF M ? II= > @ KK/$ ? @sABC&CCc|jD]k\}}|jjd}|jdkDr)|j |ddf|||j dfgY|j ||fm|S)a! Parses attribute datatype to populate a numpy dtype list Parameters ---------- dtype : list of numpy datatypes operated on in place attributes : dict contains all the attributes to parse Returns ---------- dtype : list of numpy datatypes  MMI&=djj&I%J!D6QST U MM&'>tzz'J&K1TFRTU  ) Mr4c|jD]Q\}}|jdkDr8|jdtj|jdz||d<|||<S|S)a Parses attribute data in to a custom array, assumes datatype has been defined appropriately Parameters ---------- data_array : numpy array with custom datatype datatype reflects all the data to be stored for a given ply element attributes : dict contains all the attributes to parse Returns ---------- data_array : numpy array with custom datatype r.rrU)rWrZr\npones) data_arrayr^r_rQs r2_add_attributes_to_data_arrayrksf !&&( d 99q=*.**Q-"''$**Q-:P*PJ$v ' 4) r4c|jD]<}|jdvr td|jj3tdy)z Asserts that a set of attributes is valid for PLY export. Parameters ---------- attributes : dict Contains the attributes to validate Raises -------- ValueError If passed attributes aren't valid. )r.rz/PLY attributes are limited to 1 or 2 dimensionsNz+PLY attributes must be of a single datatype)valuesrZ ValueErrorrXnames)r^rQs r2_assert_attributes_validrpsM!!# 99F "NO O ::   'JK K $r4 vertex_normalinclude_attributesc |dk(rd}n|dvr td|d|jv}|rt|drt|drt|jd rt j |jj t|jd fk(rT|jj ddd f|jd <|jj ddd f|jd<t|jt|drt|jddg}dg}d}d}dg}tjd} | dg} d|i} d} d} d}t|dr3t|jr'|jj ddk7r tdt|jd kDr|jDcgc]}|j|j}}t j |Dcgc]}|j d c}}t j"d gt j$|ddf}|j'}t j(t j*d |d z t j*d |f}t j,t/|d z |Dcgc] \}}|d||zc}}}t j,|}t|}t|}|d kDr^|d kDrX| j1| dt j2||} t j4|tj6| d<| j1| dt9||} | j;||d nt|d!r| j1| dt|j}|| d"<|r%| j1| d#|j1|t|dxrP|jj<dk(xr5t|jj>t|jk(}|r%| j1| d$|j1||rt|drut|j}|jjADcic]"\}}t|d%rt||k(r||$} }}tC| | tE|| nd} t j2||} |j| d<|r|jF| d&<|r|jj>| d'<|r tI| | t|d(rB| j1| d)|jj<d)k(r*|d*k7r%| j1| d$|j1||r8t|dr,tC| |jtE||jt j2t|jJ|}d|d+<|jJ|d,<|jj<d)k(r|d*k7r|jjL|d'<t|jJ| d-<|r"t|drtI||j| j1| d.tOd/jQ| jS| jUd0g}!|dk(rd| |!j1| jW||!j1|jW| |!j1| jWn|d*k(r| 6|!j1tYjZ| d1d23jUd0|8|!j]d4tYjZ|d1d23jUd0g| 8|!j]d4tYjZ| d1d23jUd0g|!j1d4n td5d6jQ|!Scc}wcc}wcc}}wcc}}w)7a/ Export a mesh in the PLY format. Parameters ---------- mesh : trimesh.Trimesh Mesh to export. encoding : str PLY encoding: 'ascii' or 'binary_little_endian' vertex_normal : None or include vertex normals Returns ---------- export : bytes of result binarybinary_little_endian)ruasciiz encoding must be binary or asciiNvertex_normalsvertex_attributesr uvrrsr.tface_attributes)countrV)indexFG G t}}  !;?==I=a 4==1=HI88$BAQWWQZ$BCL>>A3 ,0G0L*MNL#&&(GOORYYq'A+%> !W@U$VWEII+.lQ.> *M*M'6NV+*MEyy*Hx=LE I1}!1 i12 hh|<H (* 82::(N H% i/07ZP $$#,lK z " i)*4==) (4 n%  MM)O4 5    3 4 D( # E   H, EDKK--.#dmm2DD   MM)G, -    , t01"4==1 !% 6 6 < < >% >1q),Q<1GqD >"% *&2CD(7HI$(!hh|<@ $ H %)%8%8K " "&++";";K  "3"? )+7H ItW i'( ;;  v %(g*= MM)G, -   k * '$0A"B %fd.B.B C $Z1E1E FXXc$**oZ@  7"jj 7 ;;  v %(g*=!%!8!8Jv &)$**o l# '$0A"B )*d6J6J K MM)G$%rwwv'22=AHHQ RF))  " MM+--/ 0  ! MM*,,. /  ! MM*,,. / W   " MM//3$&/   ! MM33"cTfWo    ! MM33"cTfWo    e<== 88F iJ%Cf%s"a=a ;a ='acdt|jjvr td|jj dj j}d|v}ddgt d|v}tj}d} |j}| td |j dj }|j}d |vrnd |d vr,|d d\}} t | tjd||<nd|d vr|s tdt|dk(r'|d d\} } |t| z|dt| <nd|d vr|dd\} } } |t| zdz|zt| z|dt| <nGd|jvr5|jjddz} || dj }V|||fS)a Read the ASCII header of a PLY file, and leave the file object at the position of the start of data but past the header. Parameters ----------- file_obj : open file object Positioned at the start of the file Returns ----------- elements : collections.OrderedDict Fields and data types populated is_ascii : bool Whether the data is ASCII or binary image_name : None or str File name of TextureFile plyzNot a ply file!rrvrT>bigNzHeader not terminated properly! end_headerelementrr.)r propertiespropertyz$Property defined before any element!rrlistrz , ($LIST,) texturefile ) r0readlinelowerrndecodestripr collections OrderedDictsplitr_dtypesr~)rKrrNendianr=rOrawliner_rrXfield dtype_countr~s r2r>r>s+( C))+,2244*++  "))'288:@@BH("H3ZEX-. /F&&(HJ ! ;>? ?jj!'')yy{ 4   Q 8LD&f+)557HTN 47 " !GHH4yA~#ABx u;AGEN;R|,SZ847",0H) UEW[11L@6IGTYNZ|,SZ8ciik )IIK%%m4r9EUV**,JY \ Xz ))r4c dd|ii}d|vrU|ddrMtjdDcgc] }|dd|c}}tj|dst di|d <|S tjd Dcgc] }|dd|c}}t |t |k(r||d <d |vr|d dr |d d} nd } d } ddg} d } tj| dr| } nt| tr| D] }|| vs| |} nd| vr| d} nt| tjr|d d} t | jjdk(r| jjd}nEt | jjdkDr#| jjD] }|| vs|}n| d} | dd}  1tj| }t |dk7r t| } | gd}|D]{} |dd|d}|dd|d}tj|| j!d|| j!dfdj!| jddf} ntj| }|  t |dk(r| j| jd| jddzfk(r|rcddlm}| j!d}t'j(|\}}|| |j!| j\} }}||}|||}nCtj*t |df}| j!d|| j!d<t,j.j1|||d<nW| Uddlm}tj*t |df}| j!d|| j!d<|||d<| |d<||d <d |vrt3|d |d!<d|vrt3|d|d"<|j5d#ij5dd }|d }t|tr tj|d$|d%f}nRt|tjr8t |jdk(r|jddk(r|}n t=|}|!dd)lm }|jC|||d |Scc}wcc}w#t $rY]wxYw#ttf$rY[wxYw#ttf$rY8wxYw#t $r+t7j8d&|j;d'(YwxYw)*a` Given an elements data structure, extract the keyword arguments that a Trimesh object constructor will expect. Parameters ------------ elements : OrderedDict object With fields and data loaded fix_texture : bool If True, will re- index vertices and faces so vertices with different UV coordinates are disconnected. image : PIL.Image Image to be viewed prefer_color : None, 'vertex', or 'face' Which kind of color to prefer if both defined Returns ----------- kwargs : dict Keyword arguments for Trimesh constructor metadata_ply_rawrrxyzrQ)rrzVertices were not (n,3)!geometry)nxnynzrwrN vertex_indexvertex_indices)r)rrtexcoordr.rf1r)) texture_u texture_v)ur)rzr{r)axis) unmerge_faces)rr)ryr<r )TextureVisuals)ryrrrrrvertex1vertex2z'failed to convert PLY edges from keys: Tr:) edges_to_path)"rhrr is_shapernrrH isinstancedictndarrayrXroKeyErrorr\r rreshapevisual.texturerr unique_rowsrr texturer_element_colorsrBr rGkeysrpath.exchange.miscrr)r=r6r<r7rMirjrw face_datar index_namesr face_blobr_r\texcoord_namesrot_ut_vruv_alluniqueinversemask_vmask_vtryr edge_datarrs r2rJrJs0:x0 1F8 28 <??5#Q5aHX$6v$>q$A5#QR}}Xw/78 8 z  4F G4FqXh  ' *4F G  ~ #h- /'5F# $hv.x8V$V,  "#34KH }}Y - It $AI~!!   " ,H Irzz *V$V,  y$$ % *??((+D && '! +__** #D+ $%  ,T2H   u:?%e,E  RN' "8,V4U1X>C"8,V4U1X>C!xxU]]2./U]]25F1GHr gu{{1~r23(  E a5;;q>5;;q>A3E"FF:"))'2#+"6"6v"> *77??5;;7*&vw$F+F^G,XXs8}a01(0(8(8(A5==$% &~~<<%<PF8   ! 73x=!,-B$,$4$4W$=Bu}}R !-4F8 w!F: /0@ A}8"1(82D"E VR(,,VT:I i & 9)=y?S(TU  2:: .9??#q(Y__Q-?1-D!39=   : MM-vj/AB C M_$R H    XH%   2#H-T!  =inn>N=OP! s`S7(T;S< #TTA7T'"T=<T T TT$#T$'T:9T:=1U10U1cgd}|Dcgc]}||dvs |d|}}t|dk\rtj|Sycc}w)aT Given an element, try to extract RGBA color from properties and return them as an (n,3|4) array. Parameters ------------- element : dict Containing color keys Returns ------------ colors : (n, 3) or (n, 4) float Colors extracted from the element signal : float Estimate of range )redgreenbluealpharrQrN)rrhr)rrrcandidate_colorss r2rrsZ" -D48WDqAAVz_ply_ascii..sT%BrgQSm%Bs rr.rQN) r0readr splitlinesrh fromstringrWrrnsumrmr*r%) r=rKtextlinesrrrow_poskeyrmrQcol_count_equal list_count element_datas r2r?r?Zs3 x}}%%g. /D NN4 E05 61R]]1# &E 6G ~~' V 6 !VH%5%: Ww)99:6(## $99T?D"O TVL%9%@%@%BTT zQ0|0DdKL 36,3GNL , f7( 7  $#O $sDD  DDcd}d}d}|||tj|}||}||k7r td|||y)ac Load the data from a binary PLY file into the elements data structure. Parameters ------------ elements : OrderedDict Populated from the file header. Object will be modified to add data by this function. file_obj : open file object With current position at the start of the data section (past the header) c|j}|j}g}|jD]\}}|d}d}|jD]\} } g} d| vrtj| j dd} t |dk(rd} ntj|j } |j|| z|j| j }t |dk(r| j| nHtj|| d}|| jdt||| <||| dzz }t  dkDr9| D]}|j|t |dk(r|j|`tjdj|jj }||d|zz }|j||D]}|j|y ) a1 Given a set of elements populated from the header if there are any list properties seek in the file the length of the list. Note that if you have a list where each instance is different length (if for example you mixed triangles and quads) this won't work at all rrr,rrz, rN)tellrWrhrXrritemsizeseekr1r] frombufferreplacer0poprrm)rKr=p_start p_currentelem_pop element_keyrprops prior_datarrXprop_popr`rblobsizerDr@s r2populate_listsizez&_ply_binary..populate_listsizes--/MMO $,NN$4 KL)EJ!KKM5e##%((5;;s+;A+>"?K:!+!"!#*!5!>!>MM)f"45#==)=)=>D4yA~ *==[A!DD$Qx//TCE!HeAhn, '*(8}q #CIIcN$u:?OOK0xx %,,. 9:CCH *X5 5IG%5J  gC LL r4cz|jD]x}t||dj}tj|}|j ||d|j z} tj||||d<z|S#t$r#tjd|d||d<YwxYw)z Given the data type and field information from the header, read the data and add it to a 'data' field in the element. rrrrQzPLY failed to populate: N) rrrWrhrXr1r@rBrHr rI)rKr=r8rWrXrQs r2 populate_dataz"_ply_binary..populate_datas ==?C#|4::<=EHHUOE==#x!85>>!IJD -(* d%(H f% #! - 6se<=(, f% -s-B)B:9B:cd}|jD]L}tjdj|dj}||d|jzz }N|S)z Given an elements data structure populated from the header, calculate how long the file should be if it is intact. rr>rr)rmrhrXrr@)r=rMrrXs r2_elements_sizez#_ply_binary.._elements_sizes^ (GHHSXXgl&;&B&B&DEFE GH%6 6D) r4zPLY is unexpected length!N)r distance_to_endrn)r=rKrNrPrR size_file size_elementss r2r@r@s^3j h)$$X.I#8,MM!455(H%r4c tjd5}|jt||j tjd5}t j tdtt|d|jd|jg|jd|j}ddddddS#1swYxYw#1swYSxYw) a Export a mesh using Google's Draco compressed format. Only works if draco_encoder is in your PATH: https://github.com/google/draco Parameters ---------- mesh : Trimesh object Mesh to export bits : int Bits of quantization for position tol.merge=1e-8 is roughly 25 bits Returns ---------- data : str or bytes DRC file bytes .plysuffix.drcz-qp-i-orN) tempfileNamedTemporaryFilewriterflush subprocess check_output draco_encoderr0rr_rAr1)rbitstemp_plyencodedrQs r2 export_dracorgs(  $ $F 3xz$'(  ( ( 77  # #!D NMMLL  LLO<<>D8 4" K8 7 4" Ks%ACA(CCC CC(c tjd5}|j|j|j tjd5}t j td|jd|jg|jdt|}dddddd|S#1swYxYw#1swY|SxYw)z Load a mesh from Google's Draco format. Parameters ---------- file_obj : file- like object Contains data Returns ---------- kwargs : dict Keyword arguments to construct a Trimesh object rZrXrWr[r\rN) r]r^r_r1r`rarb draco_decoderr_rArR)rKrMtemp_drcres r2 load_dracork+s  $ $F 3xx}}'  ( ( 78  # #hmmT8==I  MM! h'F 8 4 M 8 7 4 Ms%AC AB>,C >C C  Crrcridrc)NTNF)rtNT)N))0rrar]stringrnumpyrhnumpy.lib.recfunctionsrrrrrr r constantsr rr resolversr typedrrr/r3boolr0rRrarfrkrprr>rJrr%r*r?r@rgrk _ply_loaders_ply_exporterswhichrcrir.r4r2rxsNY00(   D  T  T  d    4    D   T  T  T  D  T  d  d  d  T  t! "t# $' 0             0"!%# AoAA* A  AH24.L2!%# _$;_ _DP*fJZ0*Z&R3-lp&f%P8x $ ?+  ?+ $L(N5r4