j dZddlmZddlZddlmZddlZdZdZeGddZddZ ddZ ddZ ddZ ddZ ddZddZddZdS)aTriangle mesh container shared by the 3D models, the file formats, the rasterizer and the renderer. Conventions (those of glTF 2.0, so that a Mesh is written to GLB without conversion): - Right-handed coordinates, +Y up, the front of the asset faces +Z (-X is the asset's right), units in metres. - Faces are counter-clockwise seen from outside: the right-hand-rule normal (v1 - v0) x (v2 - v0) points outwards. - One index buffer for every vertex attribute: vertices are duplicated along UV seams (and wherever an attribute differs). - uvs: (0, 0) is the top-left corner of the texture image, u grows to the right and v downwards, so texel [r, c] of an [H, W, C] image covers u in [c / W, (c + 1) / W] and v in [r / H, (r + 1) / H]. OBJ/PLY files store 1 - v; their readers and writers convert. - colors: per-vertex sRGB-encoded values in 0..1 (3 or 4 channels, alpha linear). The GLB writer stores them as linear COLOR_0, as glTF requires, and the GLB reader converts them back. - base_color: uint8 sRGB texture [H, W, 3|4]; metallic_roughness: uint8 linear texture [H, W, 3] with G = roughness and B = metallic (R unused, the glTF packing); normal_map: uint8 linear tangent-space normal map [H, W, 3], OpenGL/glTF convention (+X red, +Y green, +Z blue). Image rows are stored top to bottom. ) annotationsN) dataclass)verticesnormalsuvscolors) base_colormetallic_roughness normal_mapceZdZUded<ded<dZded<dZded<dZded<dZded <dZded <dZ ded <e ddZ ddZ ddZ dS)Mesh torch.TensorrfacesNztorch.Tensor | Nonerrrr r r return torch.devicec|jjSN)rdevice)selfs :C:\Users\johnn\OneDrive\Documents\AI\shared\mesh3d\mesh.pyrz Mesh.device's }##c`tdifdtjDS)Ncxi|]6}|jt|jxdn7Sr)namegetattrto).0frrts r zMesh.to..,sZAAAbcqvWT16-B-B(B'KttQRQUQUV\Q]Q]AAAr)r dataclassesfields)rrrs``@rrzMesh.to+skBBAAAAAAgrgyz~ggAAABB Brc &tj|fi|Sr)r"replace)rchangess rr%z Mesh.replace.s"4337333r)rr)rr )__name__ __module__ __qualname____annotations__rrrr r r propertyrrr%r!rrr r s#'G''''#C####"&F&&&&&*J****.22222&*J**** $$$X$BBBB444444rr meshindexrrrcHjdd|ifdtDS)zjMesh whose vertex i is the old vertex index[i] (every per-vertex attribute follows), with the given faces.rc^i|])}t||t|*Sr)r)rkr-r,s rr z#select_vertices..4s<'z'z'zq]deikl]m]m]y74+;+;E+B]y]y]yrr!)r% VERTEX_FIELDS)r,r-rs`` rselect_verticesr22s9 4< { {e {'z'z'z'z'zM'z'z'z { {{r face_maskcltj|j|d\}}t|||S)ziKeeps the selected faces (bool mask or indices) and only the vertices they use (in their original order).T)return_inverse)torchuniquerr2)r,r3usedrs rsubmeshr97s3,tz)4TJJJKD% 4u - --rr!tuple[torch.Tensor, torch.Tensor]c"|jrtj|ddS|dz}|dzt td|}|||dz z d|dz zdz zz }tjt|}tt|j dD]+}|tj |||fdj },||}tj t|tj }|dd |d d kd|dd <tj|}tj|ddz ||<||||fS) aUnique positions (exact float equality, -0.0 == 0.0) in lexicographic order and, for each vertex, the index of its unique position: torch.unique(vertices, dim=0, return_inverse=True). On the CPU, where torch.unique compares rows one by one (seconds for large meshes), the rows are ordered by stable radix sorts of integer keys that order like the coordinates, one column at a time.rT)dimr5int)stabledtypeN)is_cudar6r7 element_sizeviewrarangelenreversedrangeshapesortindicesonesboolany empty_likecumsum)rwidthkeysorderaxisfirstinverses rweldrZ=sB|H!DAAAA  ! ! # #a 'E sN  e !>!> ? ?D DUQY'Q519-=,BC DD LX ' 'Ex~a01122JJejeTk!24@@@HI ;D Js5zz 4 4 4EabbT#2#Y&++A..E!""Iu%%G\%++a/GEN E%L !7 **rc||}tj|dddf|dddfz |dddf|dddfz S)zUUnnormalized face normals (v1 - v0) x (v2 - v0); their length is twice the face area.Nr@r)r6linalgcross)rrvs r face_normalsr`Qs]A <  a1g!!!Q$/111a41QQQT71B C CCrct|\}}t||dd}t j|tjd||d||}| dd}t j gd|j }t j |dk|| d z |S) zArea-weighted unit vertex normals, smooth across vertices that share a position (UV seams). Vertices without any non-degenerate face get +Y, so that the result is always a valid glTF NORMAL.rrBrDr@T)r<keepdim)r>g?r>)rgKH9)rZr`floatrepeat_interleaver6 zeros_likefloat32 index_add_reshapenormtensorrwhere clamp_min)rrweldedrYcorner_normalssumslengthups rvertex_normalsrsWs8nnOFG!(.."2"2E::LLQPQRRN  F%- 8 8 8 C CAwu~G]G]^`GaGacq r rsz {D YY1dY + +F oooho > > >B ;vz4&*:*:5*A*A#A2 F FFrcV|d|dfS)Nr)aminamax)rs r bounding_boxrwbs% ==  X]]1-- --rxcttj|dk|dz |ddzdz dzS)z@sRGB transfer function decode (IEC 61966-2-1) of values in 0..1.g?ܵ?ףp= )@)\(?zG?g333333@r6rlrmrxs rsrgb_to_linearrfs; ;qG|QY!++g2F2F2NRW1W\_0_ ` ``rcttj|dk|dzd|ddzzdz S)z7sRGB transfer function encode of linear values in 0..1.gsåi?rzr|g?r{r}r~s rlinear_to_srgbrks; ;qI~q5y%!++i:P:PU\:]2]`e2e f ffr)r,r r-rrrrr )r,r r3rrr )rrrr:)rrrrrr)rxrrr)__doc__ __future__rr"rr6r1TEXTURE_FIELDSr r2r9rZr`rsrwrrr!rrrsH#"""""!!!!!! 8 C 4444444 4*|||| .... ++++(DDDD GGGG....aaaa ggggggr