TiO DddlZddlZddlZddlmZddlZddlmZm 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 dGd,eed-ed.eed/efd0Zd1Zd2Zd3Z d4Z! dHd6eed7efd8Z"d9Z#dId:Z$d;Z%d<Z&d=Z'd>Z(d?Z)dJdAZ*dBZ+dCeiZ,dCe"iZ-e j.dDZ/e j.dEZ0e0e+e,dF<e/e*e-dF<dSdS)KN)Template)structured_to_unstructuredunstructured_to_structured)grouping resourcesutilvisual)log)triangulate_quads)Resolver)NDArrayOptionalchari1ucharu1shorti2ushortu2inti4int8int16int32int64i8uintu4uint8uint16uint32uint64u8floatf4float16f2float32f8)float64doubler-) rrrrrrr r%r'r)r+c6t|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 N/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/exchange/ply.py_numpy_type_to_ply_typer46s ;?122. //TFresolver fix_texture prefer_colorskip_materialsct|\}}} |rt||nt||d} |s ddl} | A|| } | jtj| } nE#t$rtj dYn%t$rtj ddYnwxYwt| |||}|S)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_plyrSEs*H&38%<%<"Hh (8X&&&&Hh''' E  @ @    %||J// t':4'@'@AA ? ? ? I= > > > > > @ @ @ K/$ ? ? ? ? ? ? @!hKlF MsABC$CCc|D]l\}}|jd}|jdkr+||ddf|||jdfgU|||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 > MMT&=dj&I&ITTDTTT U U U U MMU'>tz'J'JUUTUUU     Mr5c|D]H\}}|jdkr3|jdtj|jdz||d<|||<I|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_arrayrlsl !&&((  d 9q==*.*Q-"'$*Q-:P:P*PJ$ ' 4 r5c|D]5}|jdvrtd|jjtd6dS)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_validrqsj!!##LL 9F " "NOO O :  'JKK K ( LLr5binary vertex_normalinclude_attributesc|dkrd}n|dvrtd| djv}|rtdrtdrtjd rrt jjjtjd fkr<jjddd fj d <jjddd fj d<tj tdrtj ddg}dg}d}d}dg}tj d} | dg} d|i} d} d} d}tdrtjr%jjddkrtdtjd krfdjD}t jd|D}t jd gt j|ddf}|}t jt jd |d z t jd |ft jfdt-|d z |D}t j|}t|}t|}|d kr|d kr| | d t j||!} t j|tj!| d <| | d"t7||!} | ||d#ntd$r| | d tj}|| d%<|r0| | d&||tdo>jjd ko.tjjtjk}|r0| | d'|||rltdrZtjfd(j D}tA| |tC||nd}t j||!} j| d <|r j"| d)<|rjj| d*<|r|tG| |td+r7| | d,jjd,kr6|d-kr0| | d'|||r:tdr*tA| j tC|j t jtj$|!}d|d.<j$|d/<jjd,kr|d-krjj%|d*<tj$| d0<|r%tdrtG|j | | d1tMd2'| (| )d3g}|dkr|| '|| *|'||*| '|| *n|d-kr| =|tWj,| d4d56)d3|?|-d7tWj,|d4d56)d3g| ?|-d7tWj,| d4d56)d3g|d7ntd8d9'|S):a/ 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 rrbinary_little_endian)rvasciiz encoding must be binary or asciiNvertex_normalsvertex_attributesr uvrrsr/tface_attributes)countrW)indexzexport_ply..5s'IIIa 4=11IIIr5c(g|]}|jdS)r)r])rds r3rzexport_ply..8s$B$B$BAQWQZ$B$B$Br5c0g|]\}}d||zSNr)rlengthoffsetstacks r3rzexport_ply..Bs:&'6'NV+r5rrYedge) edge_count vertex_countrrrscolorcbi|]+\}}t|drt|k(||,S)__len__)hasattrlen)rkvrs r3 zexport_ply..usN%%%1q),,%25Q<1G1Gq1G1G1Gr5rrfacesfacerwr~r face_countoutroutf-8rdre) col_delim row_delim z!encoding must be ascii or binary!r5).ro_cacherr rir]rzrrryrqr}rget_jsonrarray concatenatecumsummax column_stackarangevstackzipr^zerosasarrayr*rupdatekind vertex_colorsrXrgrbrxrlr face_colorsrjoin substituteencodetobytesr structured_array_to_stringr\)rrrsrt dtype_face dtype_vertexdtype_vertex_normal dtype_color dtype_edge templatesrf header_params pack_edges pack_vertex pack_facesr discrete_len discrete_offlongestedgesr num_vertices num_edges vertex_colorryexportrrs` @@r3 export_plyrs.8) : : :;<<<(DK7  ; 4, - - =h'' CDK.. CHT[^,,T]1C1CQ0GGG.2knQQQT.B&s+.2knQQQT.B&s+ $T%; < < < 4* + + ; $T%9 : : :#$9:J*+L1&K'(J"#788I !F*M%)J%)K$(JtZ  YJ t}   H$-"5b"9Q">">FGG G t}   ! !IIII4=IIIH8$B$B$B$B$BCCL>A3 ,0G0G0L*MNNL#&&((GORYq'A+%>%> !W@U@U$VWWEI*-lQ.> *M*MEy**Hx==LE I1}}!1!1 i1222 h|<HHH (* 82:(N(N(N H% i/0007ZPPP $$#,lKK z " "+J i)***4=)) (4 n%  5 MM)O4 5 5 5    3 4 4 4 D( # # E  H, EDK-..#dm2D2DD   - MM)G, - - -    , , ,  )t011 )"4=11 %%%% $ 6 < < > >%%%! *&2CDDD(7HIIII$(!h|<@@@ $ H  9%)%8K "  <"&+";K   J"3"? )+7H I I ItWL i'((( ; v % %(g*=*= MM)G, - - -   k * * *  G'$0A"B"B G %fd.B C C C $Z1E F F FXc$*ooZ@@@  7"j 7 ; v % %(g*=*=!%!8Jv &)$*oo l#  L'$0A"B"B L )*d6J K K K MM)G$%%%rwwv''22=AAHHQQ RF)))  " MM+--// 0 0 0  ! MM*,,.. / / /  ! MM*,,.. / / / W    " MM/3$&//     ! MM3"cTfWoo      ! MM3"cTfWoo      e<=== 88F  r5cdt|vrtd|d}d|v}ddgt d|v}tj}d} |}|td |d}| }d |vrnMd |d vr5|dd\}} t | tjd||<n d|d vrt|dkr:|dd\} } |t| z||dt| <nd|dvrN|dd\} } } |t| zdz|zt| z||dt| <n\d|vrF| ddz} || d}|||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>bigNTzHeader not terminated properly! end_headerelementrr/)r propertiespropertyrrlistrz , ($LIST,) texturefile ) r1readlinelowerrodecodestripr collections OrderedDictsplitr_dtypesr)rLrrOendianr>rPrawliner`rrYfield dtype_countrs r3r?r?s~( C))++,,224444*+++  ""))'2288::@@BBH("H3ZEX-.. /F&((HJ'-!! ;>?? ?jj!!''))yy{{ 4    Q  8LD&f++)577HTNN 47 " "4yyA~~#ABBx u;AGEN;R|,SZZ8847"",0H) UEW[11L@6IGTYNZ|,SZZ8ciikk ) )IIKK%%m44r9EUVV**,,JO'-R Xz ))r5c ddii}dvrSddrEtjfddD}tj|dst dni|d <|S tjfd d D}t |t |kr||d <n#t $rYnwxYwd vrd drd d}nd}d}ddg} d} tj|dr|}nt|tr | D]} | |vr || }nd|vr|d} nt|tj rd d} t | j j dkr| j j d} n4t | j j dkr| j j D] } | | vr| } n | | d} | dd} n#ttf$rYnwxYw|tj |}t |dkrt|}| gd}|D]} dd|d}dd|d}tj||d||dfd|j ddf} n#ttf$rYwxYwtj |}|  t |dkr | j |j d|j ddzfkr|rqddlm}| d}t'j|\}}||||j \}}}||}|||}nNtjt |df}| d||d<t,j|||d<ne| cddlm}tjt |df}| d||d<|| |d<||d!<||d"<d vrt3d |d#<dvrt3d|d$<d%idd}|d}t|tr_ tj|d&|d'f}n#t $r.t7jd(|d)*YnYwxYwt|tj r;t |j dkr|j ddkr|}nt=|}|+dd+lm }|!|||d"|S),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_rawrrc8g|]}dd|SrrRr)rir>s r3rz'_elements_to_kwargs..5s(#Q#Q#QaHX$6v$>q$A#Q#Q#Qr5xyz)rrzVertices were not (n,3)!geometryc8g|]}dd|Srr)rjr>s r3rz'_elements_to_kwargs..?s( G G GqXh  ' * G G Gr5)nxnynzrxrrRN 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=r8rNrrx face_datar index_namesrr face_blobr`r]texcoord_namesrpt_ut_vruv_alluniqueinversemask_vmask_vtrzr edge_datarrs` r3rKrKs0:x0 1F8 28 <?#Q#Q#Q#Q5#Q#Q#QRR}Xw// 9788 8 9 z   G G G G4F G G G   ~  #h-- / /'5F# $      hv.x8V$V,  "#34KH }Y --! It $ $  AI~~!!   " " ,H Irz * *V$V,  y$ % % * *?(+DD & ' '! + +_*   ##DE$$%  ,T2HHH%    D    u::??%e,,E  RQQN'   "8,V4U1X>C"8,V4U1X>C!xU]]2../U]]25F5F1GHr   gu{1~r233E"H-D  E a5;q>5;q>A3E"FFF B::::::"))'22#+"6v">"> *77??5;77**&vw$F+F^G,Xs8}}a011(0(8(8(A(A5==$$% &~<<%<PPF8    ! 7 7 7 7 7 73x==!,--B$,$4$4W$=$=Bu}}R  !-~444F8 w!F: /0@ A A}8"1(82D"E"E VR((,,VT::I i & & > 9)=y?S(TUU     Pinn>N>NPP!   2: . . >9?##q((Y_Q-?1-D-D!39==   : : : : : : MM--vj/ABB C C C MsJ'AB-- B:9B:GG32G3:BKK.-K."T665U.-U.czgd}fd|D}t|dkrtj|SdS)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 )redgreenbluealphac@g|]}|dv d|S)rrRr)rrrs r3rz#_element_colors..s1WWWqAAV. s . . .qQ . . .r5rr/$LIST($LIST,)rcBi|]\}}|tj|dS)objectr)rir)rrrs r3rz+_load_element_different..s-HHH$!Qq"(1H---HHHr5ci|]V\}}t|jdk|||djWS)rr)rr]squeezeastyperY)rrredatas r3rz+_load_element_different.."s]   117||q   qyy{{!!%(1+"344   r5)rrXrrr^r0r) rrRrowstartr`dtrendr/r1s @r3_load_element_differentr6s? / .JOO-- . . .E"((**  HD"F"}}XXj))"-SZ &.C $K  s59~44R88 9 9 9EE  IH%++--HHHG NN         Nr5c|d}i}d}|D]\}}|t|krnd|vrf|dd}t||}|dd|dz|dz|zf|||<||dzz }|dd||dzf|||<|dz }|S)a Load element data with lists of a single length based on the element's property-definitions. Parameters ------------ properties : dict Property definitions encoded in a dict where the property name is the key and the property data type the value. data : array Data rows for this element, if the data contains list-properties all lists belonging to one property must have the same length. rr*r+rNr/)rXrrrr0) rrRfirstcolumnscurrentr`r4rYrs r3_load_element_singler;,s" GEGG$$&&b c%jj E b==HHZ((,Ew((F GaK'A+2F$F!FGNNuUUGDM vz !GG Ggk$9!9:AA"EEGDM qLGG Nr5ct|d}t|}d|D}d}|D]\}}d|vs |ddkr||||dz}||dz } t j|}d} n#t$rd} YnwxYwtd|d D} | r| d krt|d|} nt|d|} | ||d <d S) aX Load data from an ASCII PLY file into an existing elements data structure. Parameters ------------ elements : OrderedDict Populated from the file header, data will be added in-place to this object file_obj : file-like-object Current position at the start of the data section (past the header). rc:g|]}tj|dS)rd)sep)ri fromstring)rrs r3rz_ply_ascii..hs' 6 6 61R]1# & & & 6 6 6r5rrTFc3"K|] }d|vdV dS)r*r/Nr)rr4s r3 z_ply_ascii..|s'TTrgQSmmmmmmTTr5rr/rRN) r1readr splitlinesrXrirrosumrnr;r6) r>rLtextlinesrrow_poskeyrnrRcol_count_equal list_count element_datas r3r@r@Usz x}}%%g.. / /D NN4 E 6 6 6 6 6EG ~~''-- V 6 ! !VH%5%:%: Ww)999:6(## $9T??D"OO $ $ $#OOO $TTVL%9%@%@%B%BTTTTT  OzQ0|0DdKKLL 36,3GNNL , f7--s&B== C  C cd}d}d}|||tj|}||}||krtd|||dS)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|}|}g}|D]\}}|d}d}|D]\} } g} d| vrtj| dd} t |dkrd} ntj|j} ||| z|| j}t |dkr| | nXtj || d}||  dt||| <||| dzz }t | dkrD| D]}| |t |dkr| |tjd|j}||d|zz }|||D]}| |d S) 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)tellrXrirYrritemsizeseekrBr^ frombufferreplacer1poprrn)rLr>p_start p_currentelem_pop element_keyrprops prior_datarrYprop_poprarblobsizerTrPs r3populate_listsizez&_ply_binary..populate_listsizes;--//MMOO $,NN$4$4# 6# 6 KL)EJ!KKMM - -5e###%(5;;s+;+;A+>"?"?K:!++!"!#*!5!5!>MM)f"4555#==)=>>D4yyA~~ ***=[AAA!DD$Qx//TCCE!HeAhn, 8}}q  ###CIIcNNNNu::??OOK000x %,,.. 9 9::CH *X5 5II g  C LL      r5c|D]}t||d}tj|}|||d|jz} tj||||d<#t$r%tj d|d||d<YwxYw|S)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) rrrXrirYrBrPrRrIr rJ)rLr>rHrXrYrRs r3 populate_dataz"_ply_binary..populate_datas ==?? - -C#|4::<<==EHUOOE==#x!85>!IJJD -(* d%(H(H(H f%%  - - - r]rrYs r3_elements_sizez#_ply_binary.._elements_sizesk (( 7 7GHSXXgl&;&B&B&D&DEEFFE GH%6 6DD r5zPLY is unexpected length!N)r distance_to_endro)r>rLr^r`rb size_file size_elementss r3rArAs333j   h)))$X..I#N8,,MM!!4555M(H%%%%%r5c tjd5}|t||tjd5}t jtdtt|d|j d|j g| d| }dddn #1swxYwYdddn #1swxYwY|S) 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`rQrB)rbitstemp_plyencodedrRs r3 export_dracorxs^(  $F 3 3 3"xz$''(((  ( 7 7 7 "7  #!D NNML    LLOOO<<>>D " " " " " " " " " " " " " " """"""""""""""""" Ks7A C3"A.C C3C C3#C $C33C7:C7c tjd5}|||tjd5}t jtd|jd|jg| dt|}dddn #1swxYwYdddn #1swxYwY|S)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 rkrirhrlrmrN) rnrorprBrqrrrs draco_decoderr`rQrS)rLrNtemp_drcrvs r3 load_dracor|&sI  $F 3 3 3 (xx}}'''  ( 7 7 7 (8  #hmT8=I    MM!   h''F  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( Ms7AC'A C4 CC CC CCCrrtrzdrc)NTNF)rrNTr)rf)1rrrrnstringrnumpyrinumpy.lib.recfunctionsrrrrrr r constantsr rr resolversr typedrrrr0r4boolr1rSrbrgrlrqrr?rKrr6r;r@rArxr| _ply_loaders_ply_exporterswhichrtrzrr5r3rsYYYYYYYY000000000000(((((( %%%%%%%%  D  T  T  d    4    D   T  T  T  D  T  d  d  d  T  t! "t# $'   0              0 0 0"$("& AAx AA3- A  AAAAH24.LLL2$(# __D>_ ____DK*K*K*\JJJJZ0***Z&&&R3-3-3-lp&p&p&f%%%%P8x $ ?++  ?++ $L(N5r5