j"dZddlZddlmZmZddlmZddlm Z d de fdZ dd Z d Z Gd d Z e je j_e je j_e je j_y#e$rYywxYw)zG permutate.py ------------- Randomly deform meshes in different ways. N)transformationsutil) triangles)Numbertranslation_scalec Ntj|}tjj |j j d}tj||}tj|d}|ditj|j 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)meshrmatrixr mesh_type permutateds >/DATA/.local/lib/python3.12/site-packages/trimesh/permutate.py transformrs" 3 3>O PF %%dnn5==gFI00FCIi0IW-77 8I8I*8UVWJ c||jdz }tjj|jjdz |z}|jj |z}tjj ||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 gY@g?r r) scalerrverticesshapecopyrfacesrrrr)r magnitudervertices_noiserrrs rnoiser'/s$JJ& iit}}223c9Y FF]]'')F2N %%nTZZ&@AIi0IC-77 BCJ rc ^tjj|jjdz}||j dj dz}|j d|j zj d}tjt|t|jz}tj|j|f}tjtj|jddddgf|ftj|jddddgf|ftj|jddddgf|ff}tjj|}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 tessellationr3Ps[())""4::#3#34t;K;???*227;;K&&z2T^^CHHaHPKYYs;/03t}}3EENyy$--56H II OOTZZAq6 2NC D OOTZZAq6 2NC D OOTZZAq6 2NC D  E II ! !% (Ei0IHE:J rc(eZdZdZddZddZdZy) Permutatorc||_y)zL A convenience object to get permutated versions of a mesh. N)_mesh)selfrs r__init__zPermutator.__init__~s  rc0t|j|S)N)r)rr7)r8rs rrzPermutator.transforms7HIIrNc.t|j|SN)r'r7)r8r%s rr'zPermutator.noisesTZZ++rc,t|jSr<)r3r7)r8s rr3zPermutator.tessellationsDJJ''r)ir<)__name__ __module__ __qualname__r9rr'r3rrrr5r5}s J,(rr5)g@@r<)__doc__numpyrrrrrtypedrrr'r3r5AttributeErrorrrrrFs#+v@B*Z((" $}}J#,#4#4J &2&:&:J#  sABB B