:vj)xddlZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z Gdd e Zd Zd d Zy) N)cachinggroupingutil)color)Visuals) PBRMaterialSimpleMaterialempty_materialceZdZddZdZedZedZdZedZ e jdZ dd Z d Z d Z d Zd ZdZy)TextureVisualsNc$tj|_tj|jj|_||_|2|t|_||_ yt||_||_ y||_||_ y)a Store a single material and per-vertex UV coordinates for a mesh. If passed UV coordinates and a single image it will create a SimpleMaterial for the image. Parameters -------------- uv : (n, 2) float UV coordinates for the mesh material : Material Store images and properties image : PIL.Image Can be passed to automatically create material N)image) r DataStorevertex_attributesCache__hash___cacheuvr materialr face_materials)selfrrrrs g/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/visual/texture.py__init__zTextureVisuals.__init__ s&")!2!2!4mmD$:$:$C$CD   } . 0 - !/U ; -%DM,c8|jjy)z[ Dump the cache if anything in self.vertex_attributes has changed. N)rverifyrs r _verify_hashzTextureVisuals._verify_hash2s rcy)z Return the type of visual data stored Returns ---------- kind : str What type of visuals are defined texturers rkindzTextureVisuals.kind9src"|jdu}|S)z Check if any data is stored Returns ---------- defined : bool Are UV coordinates and images set? N)r)roks rdefinedzTextureVisuals.definedEs]]$ & rc6|jjS)z Get a CRC of the stored data. Returns -------------- crc : int Hash of items in self.vertex_attributes )rrrs rrzTextureVisuals.__hash__Rs%%..00rc:|jjddS)z Get the stored UV coordinates. Returns ------------ uv : (n, 2) float or None Pixel position per-vertex. rN)rgetrs rrzTextureVisuals.uv]s%%))$55rc||jjdytj|tj|jd<y)z Set the UV coordinates. Parameters -------------- values : (n, 2) float or None Pixel locations on a texture per- vertex Nrdtype)rpopnp asanyarrayfloat64)rvaluess rrzTextureVisuals.uvis< >  " " & &t ,+-==rzz+RD " "4 (rc| |j}||j}t||jjtj|j}|S)z Return a copy of the current TextureVisuals object. Returns ---------- copied : TextureVisuals Contains the same information in a new object )rrr)rcopyrrr)rrcopieds rr4zTextureVisuals.copyxsX :B >B]]'')99T%8%89   rc||jj|j}tj|}|S)z Convert textured visuals to a ColorVisuals with vertex color calculated from texture. Returns ----------- vis : trimesh.visuals.ColorVisuals Contains vertex color from texture ) vertex_colors)rto_colorrr ColorVisuals)rcolorsviss rr8zTextureVisuals.to_colors1''0  v6 rc|jXtj|jj|j }|j |j|S|j S)z Get a copy of )rr/uniquemeshfacesflattenr4)r face_indexindicess r face_subsetzTextureVisuals.face_subsetsV 77 ii  ; C C EFG99TWWW-. .99; rcJi}|jjD]\}}t|dk(r ||||<|jj|jj|y#t$r&tj j d|dYwxYw)z Apply a mask to remove or duplicate vertex properties. Parameters ------------ mask : (len(vertices),) bool or (n,) int Mask which can be used like: `vertex_attribute[mask]` rzfailed to update visual: ``N) ritemslen BaseExceptionrlogwarningclearupdate)rmaskupdateskeyvalues rupdate_verticeszTextureVisuals.update_verticess00668JC5zQ F$T{ 9 $$& %%g. ! F  #=cU!!DE FsA33,B"!B"cy)zq Apply a mask to remove or duplicate face properties, not applicable to texture visuals. Nr#)rrMs r update_faceszTextureVisuals.update_facessrc ddlm}|||S)af Concatenate this TextureVisuals object with others and return the result without modifying this visual. Parameters ----------- others : (n,) Visuals Other visual objects to concatenate Returns ----------- concatenated : TextureVisuals Concatenated visual objects r) concatenate)objectsrU)rothersrUs rrUzTextureVisuals.concatenates )4((r)NNNN)N)__name__ __module__ __qualname__rr propertyr$r'rrsetterr4r8rCrQrSrUr#rrrr s$-L     1 6 6YY S S* /4 )rrc|jdd}|r|g}|j}|jtj|dz|D]}tj d|dzfdtj }ttd|j|jD]\}} } | ||| <tj|dk7|tj} |jtj| djtj |Stj|jdg} |D]5}| jtj|jd7tj | } t#j$| \}}| |}|d d dfj'}||}tj(t+|tj }tjt+|||<||jd|j,df}|g}|j/|j|S) a Textured meshes can come with faces referencing vertex indices (`v`) and an array the same shape which references vertex texture indices (`vt`) and sometimes even normal (`vn`). Vertex locations with different values of any of these can't be considered the "same" vertex, and for our simple data model we need to not combine these vertices. Parameters ------------- faces : (n, d) int References vertex indices *args : (n, d) int Various references of corresponding values This is usually UV coordinates or normal indexes maintain_faces : bool Do not alter original faces and return no-op masks. Returns ------------- new_faces : (m, d) int New faces for masked vertices mask_v : (p,) int A mask to apply to vertices mask_* : (p,) int A mask to apply to vt array to get matching UV coordinates Returns as many of these as args were passed maintain_facesFrr,r)axisN)r*maxappendr/arangefullint64ziprangeTwherenan nanmedianastyper0reshape column_stackr unique_rowsargsortzerosrGshapeextend)r?argskwargsr^resultmax_idxargmasksifa masks_nan stackablestackr=inversepairsorderremap new_facess r unmerge_facesrs@ZZ 0%8N))+ bii! ,-CGGQ! ,bAEuQx#%%81aa 9"eRVVs& 88EJJbhh /DRWWRWWT]++33BHH=H771BHH- > ||E(Oh|?? ::<r)rF)r4numpyr/rrrrbaser rr r r rrrr#rrrs8 &&AAM)WM)``F r