_Mh 5XdZddlZddlZddlmZddlmZdgZdZ GddZ dS) z3 Spherical Voronoi Code .. versionadded:: 0.18.0 N)_voronoi)cKDTreeSphericalVoronoic tj|}dtjd|dddf|dddftjd|dddf|dddfztjd|dddf|dddfzz}tjdtj||zS)zCalculates the solid angles of plane triangles. Implements the method of Van Oosterom and Strackee [VanOosterom]_ with some modifications. Assumes that input points have unit norm.rij,ij->iNr)nplinalgdeteinsumabsarctan2)R numerator denominators `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/spatial/_spherical_voronoi.pycalculate_solid_anglesrs a  IryQqqq!tWa1g>>yQqqq!tWa1g>>?yQqqq!tWa1g>>?@K 6!bjK888 9 99c8eZdZdZd dZdZdZdZd Zd Z dS) ra Voronoi diagrams on the surface of a sphere. .. versionadded:: 0.18.0 Parameters ---------- points : ndarray of floats, shape (npoints, ndim) Coordinates of points from which to construct a spherical Voronoi diagram. radius : float, optional Radius of the sphere (Default: 1) center : ndarray of floats, shape (ndim,) Center of sphere (Default: origin) threshold : float Threshold for detecting duplicate points and mismatches between points and sphere parameters. (Default: 1e-06) Attributes ---------- points : double array of shape (npoints, ndim) the points in `ndim` dimensions to generate the Voronoi diagram from radius : double radius of the sphere center : double array of shape (ndim,) center of the sphere vertices : double array of shape (nvertices, ndim) Voronoi vertices corresponding to points regions : list of list of integers of shape (npoints, _ ) the n-th entry is a list consisting of the indices of the vertices belonging to the n-th point in points Methods ------- calculate_areas Calculates the areas of the Voronoi regions. For 2D point sets, the regions are circular arcs. The sum of the areas is ``2 * pi * radius``. For 3D point sets, the regions are spherical polygons. The sum of the areas is ``4 * pi * radius**2``. Raises ------ ValueError If there are duplicates in `points`. If the provided `radius` is not consistent with `points`. Notes ----- The spherical Voronoi diagram algorithm proceeds as follows. The Convex Hull of the input points (generators) is calculated, and is equivalent to their Delaunay triangulation on the surface of the sphere [Caroli]_. The Convex Hull neighbour information is then used to order the Voronoi region vertices around each generator. The latter approach is substantially less sensitive to floating point issues than angle-based methods of Voronoi region vertex sorting. Empirical assessment of spherical Voronoi algorithm performance suggests quadratic time complexity (loglinear is optimal, but algorithms are more challenging to implement). References ---------- .. [Caroli] Caroli et al. Robust and Efficient Delaunay triangulations of points on or close to a sphere. Research Report RR-7004, 2009. .. [VanOosterom] Van Oosterom and Strackee. The solid angle of a plane triangle. IEEE Transactions on Biomedical Engineering, 2, 1983, pp 125--126. See Also -------- Voronoi : Conventional Voronoi diagrams in N dimensions. Examples -------- Do some imports and take some points on a cube: >>> import numpy as np >>> import matplotlib.pyplot as plt >>> from scipy.spatial import SphericalVoronoi, geometric_slerp >>> from mpl_toolkits.mplot3d import proj3d >>> # set input data >>> points = np.array([[0, 0, 1], [0, 0, -1], [1, 0, 0], ... [0, 1, 0], [0, -1, 0], [-1, 0, 0], ]) Calculate the spherical Voronoi diagram: >>> radius = 1 >>> center = np.array([0, 0, 0]) >>> sv = SphericalVoronoi(points, radius, center) Generate plot: >>> # sort vertices (optional, helpful for plotting) >>> sv.sort_vertices_of_regions() >>> t_vals = np.linspace(0, 1, 2000) >>> fig = plt.figure() >>> ax = fig.add_subplot(111, projection='3d') >>> # plot the unit sphere for reference (optional) >>> u = np.linspace(0, 2 * np.pi, 100) >>> v = np.linspace(0, np.pi, 100) >>> x = np.outer(np.cos(u), np.sin(v)) >>> y = np.outer(np.sin(u), np.sin(v)) >>> z = np.outer(np.ones(np.size(u)), np.cos(v)) >>> ax.plot_surface(x, y, z, color='y', alpha=0.1) >>> # plot generator points >>> ax.scatter(points[:, 0], points[:, 1], points[:, 2], c='b') >>> # plot Voronoi vertices >>> ax.scatter(sv.vertices[:, 0], sv.vertices[:, 1], sv.vertices[:, 2], ... c='g') >>> # indicate Voronoi regions (as Euclidean polygons) >>> for region in sv.regions: ... n = len(region) ... for i in range(n): ... start = sv.vertices[region][i] ... end = sv.vertices[region][(i + 1) % n] ... result = geometric_slerp(start, end, t_vals) ... ax.plot(result[..., 0], ... result[..., 1], ... result[..., 2], ... c='k') >>> ax.azim = 10 >>> ax.elev = 40 >>> _ = ax.set_xticks([]) >>> _ = ax.set_yticks([]) >>> _ = ax.set_zticks([]) >>> fig.set_size_inches(4, 4) >>> plt.show() rNư>c|tdt||_tj|tj|_|jjd|_ |tj |j |_ n tj|t|_ tj |j|jdz ||jz|_|j|j krtd|j t|j||jzrtdtj |j|j z d}tj||jz }|||jzkrtd |dS) NzM`radius` is `None`. Please provide a floating point number (i.e. `radius=1`).r)dtyper)tolz&Rank of input points must be at least zDuplicate generators present.axisz$Radius inconsistent with generators.) ValueErrorfloatradiusr arrayastypefloat64pointsshape_dimzeroscenterr matrix_rank_rankr query_pairsnormrmax_calc_vertices_regions)selfr#rr' thresholdradiimax_discrepancys r__init__zSphericalVoronoi.__init__s >233 3Fmm hv&&--bj99 K%a( >(49--DKK(6777DKY**4;Q+G/84;/F+HH : ! !QdiQQRR R 4;   + +I ,C D D ><== = t{T[8qAA&!45599;; i$+5 5 5CDD D ##%%%%%rctj|j}|j|jddddfz|jz|_|j|_ tj t|j }tj |g|jz}|j }tj|d}||tjtjtj|dzfdt+tdz D}||_dS)a: Calculates the Voronoi vertices and regions of the generators stored in self.points. The vertices will be stored in self.vertices and the regions in self.regions. This algorithm was discussed at PyData London 2015 by Tyler Reddy, Ross Hemsley and Nikolai Nowaczyk N mergesort)kindrc \g|](}t||dz)S)r)list).0iflattened_groups intervalss r z;SphericalVoronoi._calc_vertices_regions..sI666' ! Yq1u5E(EFGG666r)scipyspatial ConvexHullr#r equationsr'vertices simplices _simplicesr arangelen column_stackr%ravelargsortr!intpcumsumbincountrangeregions) r.convsimplex_indices tri_indices point_indicesindicesgroupsr;r<s @@rr-z'SphericalVoronoi._calc_vertices_regionss;}'' 44 dnQQQV&<>MNN N)$/4<HHHHHrc|d|jD}tj|}|d}dt |D}tjd|jD}tj|d}tj|d}d|d<||dz ||<|j|jz |j z }|j |jz |j z } tj ||| || |g |ddf} t| } tj| |dz } | ddxx| ddzcc<| |j dzzS) Nc,g|]}t|S)rF)r9regions rr=z8SphericalVoronoi._calculate_areas_3d.. s888V888rr4c<g|]\}}t|D]}|Sr\)rM)r9r:sizejs rr=z8SphericalVoronoi._calculate_areas_3d..sF///wq$"'++//////rcg|] }|D]}| Sr\r\)r9r]rs rr=z8SphericalVoronoi._calculate_areas_3d..s%GGGGG1!GGGGrrrrVr )rXrNr rK enumerater rollr#r'rrBhstackreshaper) r.sizescsizes num_regionsrRnbrs1nbrs2rS pnormalized vnormalized trianglestriangle_solid_angles solid_angless r_calculate_areas_3dz$SphericalVoronoi._calculate_areas_3ds %%'''884<8885!!Rj //)E*:*:/// GG$,GGGHHq!!'&!$$ vz*g{T[0DK? }t{2dkA I{=9*51*51!""#*';1*=">"> !7y A Ay!677 C QRRL"--dk1n,,rc|j|j|jz }tj|dddf|dddfz dzd}tjd|d|jdzzz z }|j|z}tjtjd|dddf|j |jz }tj |dk}dtj z|jz||z ||<|S)Nrrr rr) r#rDr'r sumarccosrsignr rBwherepi)r.arcsdthetaareassignsrSs r_calculate_areas_2dz$SphericalVoronoi._calculate_areas_2d.s{4?+dk9 FDAJaaad+1 : : : !qA)9$:;<== e# *d111a4j.2mdk.IKKLL(519%%RUT[05>Ag rc|jdkr|S|jdkr|Std)aCalculates the areas of the Voronoi regions. For 2D point sets, the regions are circular arcs. The sum of the areas is ``2 * pi * radius``. For 3D point sets, the regions are spherical polygons. The sum of the areas is ``4 * pi * radius**2``. .. versionadded:: 1.5.0 Returns ------- areas : double array of shape (npoints,) The areas of the Voronoi regions. r rVz'Only supported for 2D and 3D point sets)r%r}rqrWrYs rcalculate_areasz SphericalVoronoi.calculate_areas@sM 9>>++-- - Y!^^++-- -EFF Fr)rNr) __name__ __module__ __qualname____doc__r2r-rXrqr}rr\rrrr$sAAD&&&&>BIII@%-%-%-N$GGGGGr) rnumpyr r>r scipy.spatialr__all__rrr\rrrs !!!!!!   : : :qGqGqGqGqGqGqGqGqGqGr