Ti)ddlZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z Gdd e Zd Zd d ZdS)N)cachinggroupingutil)color)Visuals) PBRMaterialSimpleMaterialempty_materialceZdZddZdZedZedZdZedZ e j dZ dd Z d Z d Z d Zd ZdZdS)TextureVisualsNctj|_tj|jj|_||_|,|t|_nt||_n||_||_ dS)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 P/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/visual/texture.py__init__zTextureVisuals.__init__ s|&")!2!4!4mD$:$CDD   } . 0 0 !/U ; ; ; %DM,c8|jdS)z[ Dump the cache if anything in self.vertex_attributes has changed. N)rverifyrs r _verify_hashzTextureVisuals._verify_hash2s rcdS)z Return the type of visual data stored Returns ---------- kind : str What type of visuals are defined texturers rkindzTextureVisuals.kind9s yrc|jdu}|S)z Check if any data is stored Returns ---------- defined : bool Are UV coordinates and images set? N)r)roks rdefinedzTextureVisuals.definedEs]$ & rc4|jS)z Get a CRC of the stored data. Returns -------------- crc : int Hash of items in self.vertex_attributes )rrrs rrzTextureVisuals.__hash__Rs%..000rc8|jddS)z Get the stored UV coordinates. Returns ------------ uv : (n, 2) float or None Pixel position per-vertex. rN)rgetrs rrzTextureVisuals.uv]s%))$555rc||jddStj|tj|jd<dS)z Set the UV coordinates. Parameters -------------- values : (n, 2) float or None Pixel locations on a texture per- vertex Nrdtype)rpopnp asanyarrayfloat64)rvaluess rrzTextureVisuals.uvisJ >  " & &t , , , , ,+-=rz+R+R+RD "4 ( ( (rc||j}||}t||jtj|j}|S)z Return a copy of the current TextureVisuals object. Returns ---------- copied : TextureVisuals Contains the same information in a new object N)rrr)rcopyrrr)rrcopieds rr4zTextureVisuals.copyxsa :B >B]''))9T%899     rcn|j|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_colors3''00 v666 rc|jVtj|jj|}||j|S|S)z Get a copy of )rr/uniquemeshfacesflattenr4)r face_indexindicess r face_subsetzTextureVisuals.face_subsetsW 7 i  ; C C E EFFG99TWW-.. .99;; rcTi}|jD]X\}}t|dkr ||||<&#t$r&tjd|dYUwxYw|j|j|dS)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_verticess06688 F FJC5zzQ F$T{   F F F  !Dc!D!D!DEEEEE F $$&&& %%g.....s A-A21A2cdS)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  X   X  1 1 1 6 6X 6Y S SY S* ///4   )))))rrc|dd}|r|g}|}|tj|dz|D]}tjd|dzfdtj}ttd|j |j D]\}} } | ||| <tj |dk|tj } |tj | d tj|Stj|dg} |D]<}| tj|d=tj| } t#j| \}}| |}|d d df}||}tjt+|tj}tjt+|||<||d|jdf}|g}||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_facesrsR@ZZ 0%88N))++ bi! ,,--- L LCGQ! ,bAAAEuQxx#%88  1aa "eRV<s& 8EJbh / / /DRWRWT]]+++33BH==H?71BH--- > @||E(OOh|??? ::<<r)rF)r4numpyr/rrrrbaser rr r r rrrr#rrrs &&&&&&&&&&AAAAAAAAAAM)M)M)M)M)WM)M)M)````F      r