:vj8dZddlZddlmZmZddlmZddlm Z m Z dde de fd Z dde fd Z dde fd Z Gd d Z e jej_e jej_e jej_y#e$rYywxYw)zG permutate.py ------------- Randomly deform meshes in different ways. N)transformationsutil) triangles)NumberSeedtranslation_scaleseedc ^tj|}tj||}|j |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 seed : None or int Seed for deterministic results, otherwise OS entropy. Returns ---------- permutated : trimesh.Trimesh Mesh with same faces as input mesh but reordered and rigidly transformed in space. ) translater )Trimesh)r rr) rrandom_generatorrrandom_rotation_matrix permutationrreshapetransform_points type_namedtriangles_module to_kwargs)meshr r randommatrixr mesh_type permutateds b/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/permutate.py transformrs( " "4 (F  3 3#&F ""4>>2::7CI00FCIi0IW-77 8I8I*8UVWJ c||jdz }tj|}|j|jj dz |z}|jj |z}|j||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) seed : None or int Seed for deterministic results, otherwise OS entropy. Returns ---------- permutated : trimesh.Trimesh Input mesh with noise applied gY@g?rr) scalerrrverticesshapecopyrfacesrrr) r magnituder roffsetvertices_noiserrrs rnoiser*5s(JJ&  " "4 (FmmDMM//036) CF]]'')F2N"">$**#=>Ii0IC-77 BCJ r c Ptj|}|j|jjdz}||j dj dz}|j d|jzj d}tjt|t|jz}tj|j|f}tjtj|jddddgf|ftj|jddddgf|ftj|jddddgf|ff}|j|}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 seed : None or int Seed for deterministic results, otherwise OS entropy. Returns ---------- permutated : trimesh.Trimesh Mesh with remeshed facets g?r)axis)r r)r rrNrr)r#r&)rrrr&r$sumrrnparangelenr#vstack column_stackrr) rr r barycentric vertex_facevertex_face_idr#r&rrs r tessellationr7Ysc( " "4 (F-- 0 01D8K;???*227;;K&&z2T^^CHHaHPKYYs;/03t}}3EENyy$--56H II OOTZZAq6 2NC D OOTZZAq6 2NC D OOTZZAq6 2NC D  E   u %Ei0IHE:J r c<eZdZdZddefdZddefdZd defdZy) Permutatorc||_y)zL A convenience object to get permutated versions of a mesh. N)_mesh)selfrs r__init__zPermutator.__init__s  r Nr c2t|j||S)N)r r )rr;)r<r r s rrzPermutator.transforms7HtTTr c2t|j||SN)r )r*r;)r<r'r s rr*zPermutator.noisesTZZ66r c0t|j|Sr@)r7r;)r<r s rr7zPermutator.tessellationsDJJT22r )iNNNN)__name__ __module__ __qualname__r=rrr*r7rr rr9r9s- UdU7$733r r9)g@@NrBrC)__doc__numpyr/rrrrtypedrrrr*r7r9AttributeErrorrr rrLs#+#v#d#L!d!H.T.b33" $}}J#,#4#4J &2&:&:J#  sABBB