l0jdZddlZddlmZmZddlmZmZm Z m Z m Z m Z ddl mZ dded e efd Zded e e fd Z dded e ed e e fdZddede e fdZded e e fdZdS)zG sample.py ------------ Randomly sample surface and volume of meshes. N)transformationsutil) ArrayLikeIntegerNDArrayNumberOptionalfloat64)uv_to_interpolated_colorFcount face_weightcX||j}tj|}|tjj}n$tj|j}|||dz}tj||}|j|jdddf} |j|jddddf} | tj | d dz} | |} | |} |rt|j dr|j j |jdddf} |j j |jddddf} tj | d d} | | z} | |} | |} |t| d df}|d  dd k}||xxd zcc<tj|}| |zd }|| z}|rlt|j dr@| |zd }|| z}|j jj}t'||}n|j j|}|||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. seed : None or int If passed as an integer will provide deterministic results otherwise pulls the seed from operating system 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 Nrr)r)rruv)rrrr)axisg?) area_facesnpcumsumrandom default_rng searchsortedverticesfacescopytilereshapehasattrvisualrlensumabsmaterialimager face_colors)meshr r sample_colorseed weight_cumr face_pick face_index tri_origins tri_vectors uv_origins uv_vectorsuv_origins_tilerandom_lengths random_test sample_vectorsamplessample_uv_vector uv_samplestexturecolorss Q/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/trimesh/sample.pysample_surfacer<sRo ;''J |!&&t,,3u 2.IY77J- 111a4 01K- 111abb5 127799K27;//77 CCCKj)Kj)K, T22,[^DJqqq!t$45 [^DJqqq!""u$56;;== '*f55==jIIo%  +  + VS--q!455N !$$!$,,44R883>K;3&VN++N!>166A6>>Mk)G + 4; % % 9 *^ ;@@a@HH )J6Jk*0G-j'BBFF[,Z8F F** J returnctj|df|jz|jdz}||}||d|}|S)aX 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 Returns --------- samples : (n, 3) float Points in the volume of the mesh where n <= count rrN)rrextentsboundscontains)r(r points containedr6s r; volume_meshrE|sX$iqz**T\9T[^ KF f%%IY'G Nr= transformctj|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 Returns --------- samples : (count, 3) float Points in requested volume rg?)rrrtransform_points)r@r rFr6s r;volume_rectangularrIsI*iqz**S0G wG!27IFF Nr=radiuscrddlm}|tj|jd|zz }t ||dz|\}}|||\}}t ||kr|d|||d|fStj dt |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)r*z only got /z samples!) rCrLrsqrtarear<r"rlogwarning)r(r rJr*rLrCindexmasks r;sample_surface_evenrTs8%$$$$$~a%i011#4>>>MFE <//LFD 6{{efuf~uT{6E6222 H?V??u???@@@ 5; r=ctjd|f\}}tjdz|z}tjd|zdz }t jtj||f}|S)aU 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 Returns ---------- points : (count, 3) float Random points on the surface of a unit sphere rr)rrpiarccosrspherical_to_vector column_stack)r uvthetaphirCs r;sample_surface_spherer^sj$ 9  QJ ' 'DAq EAIME )QUaK C  %boucl&C&C D DF Mr=)NFN)N)NN)__doc__numpyrrrtypedrrrr r r r!r r<rErIrTintr^r=r;res########IIIIIIIIIIIIIIII,,,,,, (, jj j)$jjjjZW)92?C(0(; W8//W/hv6F////d)9r=