TidZddlZddlmZmZddlmZddlm Z dde fd Z dd Z d Z Gd d Z e je j _e je j _e je j _dS#e$rYdSwxYw)zG permutate.py ------------- Randomly deform meshes in different ways. N)transformationsutil) triangles)Number@@translation_scalec Ltj|}tj|jd}tj||}tj |d}|ditj |d}|S)a Return a permutated variant of a mesh by randomly reordering faces and rotatating + translating a mesh by a random matrix. Parameters ---------- mesh : trimesh.Trimesh Mesh, will not be altered by this function Returns ---------- permutated : trimesh.Trimesh Mesh with same faces as input mesh but reordered and rigidly transformed in space. ) translate)Trimesh)r r r ) rrandom_rotation_matrixnprandom permutationrreshapetransform_pointsr type_namedtriangles_module to_kwargs)meshr matrixr mesh_type permutateds K/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/permutate.py transformrs" 3>O P P PF %%dn55==gFFI0FCCIi00IWW-7 8I8I*8U8UVVWWJ cf| |jdz }tj|jjdz |z}|j|z}tj||j}tj |d}|ditj |}|S)a Add gaussian noise to every vertex of a mesh, making no effort to maintain topology or sanity. Parameters ---------- mesh : trimesh.Trimesh Input geometry, will not be altered magnitude : float What is the maximum distance per axis we can displace a vertex. If None, value defaults to (mesh.scale / 100.0) Returns ---------- permutated : trimesh.Trimesh Input mesh with noise applied NgY@g?rr) scalerrverticesshapecopyrfacesrrrr)r magnitudervertices_noiserrrs rnoiser(/s$J& it}233c9Y FF]''))F2N %%nTZ&@AAIi00ICC-7 BBCCJ rc Htj|jjdz}||ddz}|d|jzd}tjt|t|j z}tj |j |f}tj tj |jddddgf|ftj |jddddgf|ftj |jddddgf|ff}tj |}tj|d }||| }|S) a Subdivide each face of a mesh into three faces with the new vertex randomly placed inside the old face. This produces a mesh with exactly the same surface area and volume but with different tessellation. Parameters ------------ mesh : trimesh.Trimesh Input geometry Returns ---------- permutated : trimesh.Trimesh Mesh with remeshed facets g?r)axis)r r)r r rNrr)r"r%)rrr%r#sumrrarangelenr"vstack column_stackrrr)r barycentric vertex_facevertex_face_idr"r%rrs r tessellationr4Ps{()""4:#344t;K;???**227;;;K&&z22T^CHHaHPPKYs;//003t}3E3EENy$-566H I OTZAq6 2NC D D OTZAq6 2NC D D OTZAq6 2NC D D   E I ! !% ( (Ei00IHE:::J rc*eZdZdZddZddZdZdS) Permutatorc||_dS)zL A convenience object to get permutated versions of a mesh. N)_mesh)selfrs r__init__zPermutator.__init__~s rc.t|j|S)N)r )rr8)r9r s rrzPermutator.transforms7HIIIIrNc,t|j|SN)r(r8)r9r&s rr(zPermutator.noisesTZ+++rc*t|jSr>)r4r8)r9s rr4zPermutator.tessellationsDJ'''r)r;r>)__name__ __module__ __qualname__r:rr(r4rrrr6r6}s^ JJJJ,,,,(((((rr6)rr>)__doc__numpyrrrrrtypedrrr(r4r6AttributeErrorrrrrHs########++++++v@B***Z((((((((" $}J#,#4J &2&:J###   DD s3A,,A54A5