jO 6ddlZddlZddlZddlmZddlZddlmZm Z ddl m Z m Z m Z mZddlmZddlmZddlmZdd lmZmZid d d d dddddddd dddddddddd ddddddd d!d"d#d$d!d%d%d&Zd d ddddddd d"d'd( Zd)Z dCd*eed+ed,eed-efd.Zd/Zd0Zd1Z d2Z! dDd3eed4efd5Z"d6Z#dEd7Z$d8Z%d9Z&d:Z'd;Z(d<Z)dFd=Z*d>Z+d?eiZ,d?e"iZ-e j\d@Z/e j\dAZ0e0e+e,dB<e/e*e-dB<yy)GN)Template)structured_to_unstructuredunstructured_to_structured)grouping resourcesutilvisual)log)triangulate_quads)Resolver)NDArrayOptionalchari1ucharu1shorti2ushortu2inti4int8int16int32int64i8uintu4uint8uint16uint32uint64u8floatf4float16f2float32f8)float64doubler-) rrrrrrr r%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 A/DATA/.local/lib/python3.12/site-packages/trimesh/exchange/ply.py_numpy_type_to_ply_typer46s ;??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)imageelementsr7r8) _parse_header _ply_ascii _ply_binary PIL.ImagegetImageopenr wrap_as_stream ImportErrorr debug BaseExceptionwarning_elements_to_kwargs) file_objr6r7r8r9argskwargsr>is_ascii image_namer=PILdatas r3load_plyrSEsH&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  ) Mr5c|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/rrV)rXr[r]npones) data_arrayr_r`rRs r3_add_attributes_to_data_arrayrlsf !&&( d 99q=*.**Q-"''$**Q-:P*PJ$v ' 4) r5c|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)valuesr[ ValueErrorrYnames)r_rRs r3_assert_attributes_validrqsM!!# 99F "NO O ::   'JK K $r5 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)rvasciiz encoding must be binary or asciiNvertex_normalsvertex_attributesr uvrrsr/tface_attributes)countrW)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 vrtt|dk(r'|d d\} } |t| z|dt| <nd|d vr|dd\} } } |t| zdz|zt| z|dt| <nGd|jvr5|jjddz} || dj }H|||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!rrwrU>bigNzHeader not terminated properly! end_headerelementrr/)r propertiespropertyrrlistrz , ($LIST,) texturefile ) r1readlinelowerrodecodestripr collections OrderedDictsplitr_dtypesr)rLrrOendianr>rPrawliner`rrYfield dtype_countrs r3r?r?s( C))+,2244*++  "))'288:@@BH("H3ZEX-. /F&&(HJ ! ;>? ?jj!'')yy{ 4   Q 8LD&f+)557HTN 47 "4yA~#ABx u;AGEN;R|,SZ847",0H) UEW[11L@6IGTYNZ|,SZ8ciik )IIK%%m4r9EUV**,JO R Xz ))r5c 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_rawrrxyzrR)rrzVertices were not (n,3)!geometry)nxnynzrxrN vertex_indexvertex_indices)r)rrtexcoordr/rf1r)) texture_u texture_v)ur)r{r|r)axis) unmerge_faces)rr)rzr=r )TextureVisuals)rzrrrrrvertex1vertex2z'failed to convert PLY edges from keys: Tr;) edges_to_path)"rirr is_shaperorrI isinstancedictndarrayrYrpKeyErrorr]r rreshapevisual.texturerr unique_rowsrr texturer_element_colorsrCr rHkeysrpath.exchange.miscrr)r>r7r=r8rNirjrx face_datar index_namesr face_blobr`r]texcoord_namesrpt_ut_vruv_alluniqueinversemask_vmask_vtrzr edge_datarrs r3rKrKs0: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 )redgreenbluealpharrRrN)rrir)rrrcandidate_colorss r3rrsZ" -D48WDqAAVz_ply_ascii..|sT%BrgQSm%Bs rr/rRN) r1readr splitlinesri fromstringrXrrosumrnr+r&) r>rLtextlinesrrrow_poskeyrnrRcol_count_equal list_count element_datas r3r@r@Us3 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)tellrXrirYrritemsizeseekr2r^ frombufferreplacer1poprrn)rLr>p_start p_currentelem_pop element_keyrprops prior_datarrYprop_poprarblobsizerErAs r3populate_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 r5cz|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. rrrrRzPLY failed to populate: N) rrrXrirYr2rArCrIr rJ)rLr>r9rXrYrRs r3 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)rnrirYrrA)r>rNrrYs r3_elements_sizez#_ply_binary.._elements_sizes^ (GHHSXXgl&;&B&B&DEFE GH%6 6D) r5zPLY is unexpected length!N)r distance_to_endro)r>rLrOrQrS size_file size_elementss r3rArAs^3j h)$$X.I#8,MM!455(H%r5c 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_encoderr1rr`rBr2)rbitstemp_plyencodedrRs r3 export_dracorhs(  $ $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 r[rYrXr\r]rN) r^r_r`r2rarbrc draco_decoderr`rBrS)rLrNtemp_drcrfs r3 load_dracorl&s  $ $F 3xx}}'  ( ( 78  # #hmmT8==I  MM! h'F 8 4 M 8 7 4 Ms%AC AB>,C >C C  Crrdrjdrc)NTNF)ruNT)N))1rrbr^stringrnumpyrinumpy.lib.recfunctionsrrrrrr r constantsr rr resolversr typedrrrr0r4boolr1rSrbrgrlrqrr?rKrr&r+r@rArhrl _ply_loaders_ply_exporterswhichrdrjr/r5r3rysOY00( %  D  T  T  d    4    D   T  T  T  D  T  d  d  d  T  t! "t# $' 0             0"$("& Ax AA3- A  AH24.L2$(# _D>_ _DK*\JZ0*Z&R3-lp&f%P8x $ ?+  ?+ $L(N5r5