:vj! dZddlmZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlmZmZddlmZeeZ dd e d edzd ed e fd Zdd e d e de e fdZ dd e dedzd e de e fdZ dd e de dzd e fdZdd ed e de e fdZy)zG sample.py ------------ Randomly sample surface and volume of meshes. ) getLoggerN)transformations) ArrayLikeIntegerNDArrayNumberSeedfloat64)random_generatorspherical_to_vector)uv_to_interpolated_colorcount face_weightreturn_barycentricseedc| |j}tj|}t|j}|||dz}tj ||} |j |jdddf} |j |jddddfj} | tj| djdz} | | } | | } |rt|jdr|jj|jdddf} |jj|jddddfj} tj| djd}| |z} | | } | | } |t| d df}|jd jdd kD}||xxd zcc<tj |}|r4tj"d|jd z |dddddff}| |zjd }|| z}|rt|jdrG |zjd }| z}|jj$j&}t)||}n|jj*| }|r|| |fS|| |fS|r|| fS|| fS) a Sample the surface of a mesh, returning the specified number of points For individual triangle sampling uses this method: http://mathworld.wolfram.com/TrianglePointPicking.html Parameters ----------- mesh : trimesh.Trimesh Geometry to sample the surface of count : int Number of points to return face_weight : None or len(mesh.faces) float Weight faces by a factor other than face area. If None will be the same as face_weight=mesh.area sample_color : bool Option to calculate the color of the sampled points. Default is False. return_barycentric : bool If True will also return the barycentric coordinates of each sampled point. seed : None or int Seed for deterministic results, otherwise OS entropy. Returns --------- samples : (count, 3) float Points in space on the surface of mesh face_index : (count,) int Indices of faces for each sampled point colors : (count, 4) float Colors of each sampled point Returns only when the sample_color is True barycentric : (count, 3) float Coordinates on `mesh.faces[face_index]` Returned only when return_barycentric is True Nrr)r)rruv)rrrr)axisg?) area_facesnpcumsumr random searchsortedverticesfacescopytilereshapehasattrvisualrlensumabshstackmaterialimager face_colors)meshrr sample_colorrr weight_cumr face_pick face_index tri_origins tri_vectors uv_origins uv_vectorsuv_origins_tilerandom_lengths random_test barycentric sample_vectorsamplessample_uv_vector uv_samplestexturecolorss _/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/sample.pysample_surfacer@s^oo ;'J d # * *Fu 2.IY7J-- 1a4 01K-- 1ab5 12779K277;/77 CCKj)Kj)K T2[[^^DJJq!t$45 [[^^DJJq!"u$56;;= ''*f5==jIo%  +  + S-q!45N !$$!$,44R83>K;3&VVN+Nii^%7%7Q%7%?!?PQSTVWPWAX YZ !>166A6>Mk)G 4;; % *^ ;@@a@H )J6Jkk**00G-j'BF[[,,Z8F J ; ; F** K// J returnct|j}||df|jz|jdz}|j |}||d|}|S)a Use rejection sampling to produce points randomly distributed in the volume of a mesh. Parameters ----------- mesh : trimesh.Trimesh Geometry to sample count : int Number of points to return seed : None or int Seed for deterministic results, otherwise OS entropy. Returns --------- samples : (n, 3) float Points in the volume of the mesh where n <= count rrN)r rextentsboundscontains)r,rrrpoints containedr:s r? volume_meshrIs\(d # * *FeQZ 4<</4;;q> AF f%IY'G NrA transformczt|j|dfdz |z}|tj||}|S)a Return random samples inside a rectangular volume, useful for sampling inside oriented bounding boxes. Parameters ----------- extents : (3,) float Side lengths of rectangular solid count : int Number of points to return transform : (4, 4) float Homogeneous transformation matrix seed : None or int Seed for deterministic results, otherwise OS entropy. Returns --------- samples : (count, 3) float Points in requested volume rg?)r rrtransform_points)rDrrJrr:s r?volume_rectangularrMsD4 %,,eQZ83>'IG!227IF NrAradiusc*ddlm}|%tj|jd|zz }t ||dz|\}}|||\}}t ||k\r |d|||d|fStjdt |d|d|||fS) a, Sample the surface of a mesh, returning samples which are VERY approximately evenly spaced. This is accomplished by sampling and then rejecting pairs that are too close together. Note that since it is using rejection sampling it may return fewer points than requested (i.e. n < count). If this is the case a log.warning will be emitted. Parameters ----------- mesh : trimesh.Trimesh Geometry to sample the surface of count : int Number of points to return radius : None or float Removes samples below this radius seed : None or int Provides deterministic values Returns --------- samples : (n, 3) float Points in space on the surface of mesh face_index : (n,) int Indices of faces for each sampled point r) remove_closeNr)rz only got /z samples!) rGrPrsqrtarear@r%logwarning)r,rrNrrPrGindexmasks r?sample_surface_evenrXs<%~a%i01#4>MFE /LFD 6{efu~uT{6E222KK)CK=% :; 5; rAct|jd|f\}}tjdz|z}tjd|zdz }t tj ||fS)a Correctly pick random points on the surface of a unit sphere Uses this method: http://mathworld.wolfram.com/SpherePointPicking.html Parameters ----------- count : int Number of points to return seed : None or int Seed for deterministic results, otherwise OS entropy. Returns ---------- points : (count, 3) float Random points on the surface of a unit sphere rr)r rrpiarccosr column_stack)rruvthetaphis r?sample_surface_sphereras`( D ! ( (!U 4DAq EEAIME ))QUaK C rs|< ==rA)NFFN)N)NN)__doc__loggingrnumpyrrtypedrrrr r r utilr r r$r__name__rTboolr@rIrMrXintrarAr?rlsEE7, %)$ w wT!w  w  wtWDGGB1h>>D>GGrA