o èEb‘ã@s`ddlZddlmZgd¢Zeddd„ƒZdd„Zeddd „ƒZedd d „ƒZ edd d „ƒZ dS)éN)Ú decorator)Údelaunay_plot_2dÚconvex_hull_plot_2dÚvoronoi_plot_2dc Ksšddlm}|dur| ¡}| ¡}||fd|i|¤ŽSt|ddd„ƒƒ}|r1||fd|i|¤ŽSz| d¡||fd|i|¤ŽW| |¡S| |¡w)NrÚaxZisholdcSsdS)NT©rrrú:/usr/lib/python3/dist-packages/scipy/spatial/_plotutils.pyÚsz_held_figure..T)Zmatplotlib.pyplotZpyplotÚfigureZgcaÚgetattrZhold)ÚfuncÚobjrÚkwZpltZfigZwas_heldrrrÚ _held_figures  rcCs\d|jdd}|jdd|}|jdd|}| |d|d¡| |d|d¡dS)Ngš™™™™™¹?r©Zaxisé)ÚptpÚminÚmaxZset_xlimZset_ylim)rÚpointsZmarginZxy_minZxy_maxrrrÚ_adjust_boundss rcCsX|jjddkr tdƒ‚|jj\}}| ||d¡| |||j ¡¡t||jƒ|j S)a' Plot the given Delaunay triangulation in 2-D Parameters ---------- tri : scipy.spatial.Delaunay instance Triangulation to plot ax : matplotlib.axes.Axes instance, optional Axes to plot on Returns ------- fig : matplotlib.figure.Figure instance Figure for the plot See Also -------- Delaunay matplotlib.pyplot.triplot Notes ----- Requires Matplotlib. Examples -------- >>> import matplotlib.pyplot as plt >>> from scipy.spatial import Delaunay, delaunay_plot_2d The Delaunay triangulation of a set of random points: >>> rng = np.random.default_rng() >>> points = rng.random((30, 2)) >>> tri = Delaunay(points) Plot it: >>> _ = delaunay_plot_2d(tri) >>> plt.show() réz!Delaunay triangulation is not 2-DÚo) rÚshapeÚ ValueErrorÚTÚplotZtriplotÚ simplicesÚcopyrr )ZtrirÚxÚyrrrr$s,  rcsˆddlm}ˆjjddkrtdƒ‚| ˆjdd…dfˆjdd…dfd¡‡fdd „ˆjDƒ}| ||d d d ¡t|ˆjƒ|j S) a  Plot the given convex hull diagram in 2-D Parameters ---------- hull : scipy.spatial.ConvexHull instance Convex hull to plot ax : matplotlib.axes.Axes instance, optional Axes to plot on Returns ------- fig : matplotlib.figure.Figure instance Figure for the plot See Also -------- ConvexHull Notes ----- Requires Matplotlib. Examples -------- >>> import matplotlib.pyplot as plt >>> from scipy.spatial import ConvexHull, convex_hull_plot_2d The convex hull of a random set of points: >>> rng = np.random.default_rng() >>> points = rng.random((30, 2)) >>> hull = ConvexHull(points) Plot it: >>> _ = convex_hull_plot_2d(hull) >>> plt.show() r©ÚLineCollectionrrzConvex hull is not 2-DNrcsg|]}ˆj|‘qSr)r)Ú.0Úsimplex©ÚhullrrÚ Žsz'convex_hull_plot_2d..ÚkÚsolid)ÚcolorsÚ linestyle) Úmatplotlib.collectionsr"rrrrrÚadd_collectionrr )r&rr"Z line_segmentsrr%rr\s ,*þ rc Ks"ddlm}|jjddkrtdƒ‚| dd¡r5| dd ¡}|j|jd d …df|jd d …dfd |d | d d¡rP| |jd d …df|jd d …dfd ¡| dd¡}| dd¡}| dd¡}|jjdd}|jj dd} g} g} t |j |j ƒD]t\} } t  | ¡} t  | dk¡r”|  |j| ¡q{| | dkd}|j| d|j| d}|t j |¡}t  |d |dg¡}|j| jdd}t  t  |||¡¡|}|jrÚ| }|j|||  ¡}|  |j||g¡q{| || |||dd¡| || |||dd¡t||jƒ|jS)aØ Plot the given Voronoi diagram in 2-D Parameters ---------- vor : scipy.spatial.Voronoi instance Diagram to plot ax : matplotlib.axes.Axes instance, optional Axes to plot on show_points: bool, optional Add the Voronoi points to the plot. show_vertices : bool, optional Add the Voronoi vertices to the plot. line_colors : string, optional Specifies the line color for polygon boundaries line_width : float, optional Specifies the line width for polygon boundaries line_alpha: float, optional Specifies the line alpha for polygon boundaries point_size: float, optional Specifies the size of points Returns ------- fig : matplotlib.figure.Figure instance Figure for the plot See Also -------- Voronoi Notes ----- Requires Matplotlib. Examples -------- Set of point: >>> import matplotlib.pyplot as plt >>> rng = np.random.default_rng() >>> points = rng.random((10,2)) Voronoi diagram of the points: >>> from scipy.spatial import Voronoi, voronoi_plot_2d >>> vor = Voronoi(points) using `voronoi_plot_2d` for visualisation: >>> fig = voronoi_plot_2d(vor) using `voronoi_plot_2d` for visualisation with enhancements: >>> fig = voronoi_plot_2d(vor, show_vertices=False, line_colors='orange', ... line_width=2, line_alpha=0.6, point_size=2) >>> plt.show() rr!rrzVoronoi diagram is not 2-DZ show_pointsTÚ point_sizeNÚ.)Z markersizeZ show_verticesrÚ line_colorsr(Ú line_widthgð?Ú line_alpharr))r*ZlwZalphar+Zdashed)r,r"rrrÚgetrZverticesZmeanrÚzipZ ridge_pointsZridge_verticesÚnpZasarrayÚallÚappendZlinalgZnormZarrayÚsignÚdotZ furthest_siterr-rr )Zvorrrr"r.r0r1r2ÚcenterZ ptp_boundZfinite_segmentsZinfinite_segmentsZpointidxr$ÚiÚtÚnZmidpointZ directionZ far_pointrrrr—sV >  . *    üü r)N) Znumpyr5Zscipy._lib.decoratorrZ _decoratorÚ__all__rrrrrrrrrÚs   7 :