B î&]\ôã@svddlmZmZmZddlZddlmZdddgZ eddd„ƒZ d d „Z e dd d„ƒZ e dd d„ƒZ e dd d„ƒZdS)é)ÚdivisionÚprint_functionÚabsolute_importN)Ú decoratorÚdelaunay_plot_2dÚconvex_hull_plot_2dÚvoronoi_plot_2dc Ks’ddlm}|dkr8| ¡}| ¡}||fd|i|—ŽSt|ddd„ƒƒ}|rb||fd|i|—ŽSz| d¡||fd|i|—ŽS| |¡XdS)NrÚaxZisholdcSsdS)NT©r r r ú7lib/python3.7/site-packages/scipy/spatial/_plotutils.pyÚsz_held_figure..T)Zmatplotlib.pyplotZpyplotÚfigureZgcaÚgetattrZhold)ÚfuncÚobjr ÚkwZpltZfigZwas_heldr r r Ú _held_figure s  rcCs\d|jdd}|jdd|}|jdd|}| |d|d¡| |d|d¡dS)Ngš™™™™™¹?r)Úaxisé)ÚptpÚminÚmaxZset_xlimZset_ylim)r ÚpointsZmarginZxy_minZxy_maxr r r Ú_adjust_boundss rcCsX|jjddkrtdƒ‚|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: >>> points = np.random.rand(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Úyr r r r&s+  csˆddlm}ˆjjddkr$tdƒ‚| ˆ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: >>> points = np.random.rand(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)Úhullr r ú Žsz'convex_hull_plot_2d..ÚkÚsolid)ÚcolorsÚ linestyle) Úmatplotlib.collectionsr$rrrrr Úadd_collectionrr )r'r r$Z line_segmentsr )r'r r]s+ *  c Ksddlm}|jjddkr$tdƒ‚| dd¡rj| dd ¡}|j|jd d …df|jd d …dfd |d | d d¡r | |jd d …df|jd d …dfd ¡| dd¡}| dd¡}| dd¡}|jjdd}|jj dd} g} g} xðt |j |j ƒD]Þ\} } t  | ¡} t  | dk¡r,|  |j| ¡qø| | dkd}|j| d|j| d}|t j |¡}t  |d |dg¡}|j| jdd}t  t  |||¡¡|}|j|||  ¡}|  |j||g¡qøW| || |||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 >>> points = np.random.rand(10,2) #random 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() r)r$rrzVoronoi diagram is not 2-DZ show_pointsTÚ point_sizeNÚ.)Z markersizeZ show_verticesrÚ line_colorsr)Ú line_widthgð?Ú line_alpha)rr*)r+ZlwZalphar,Zdashed)r-r$rrrÚgetrZverticesZmeanrÚzipZ ridge_pointsZridge_verticesÚnpZasarrayÚallÚappendZlinalgZnormZarrayZsignÚdotrr.rr )Zvorr rr$r/r1r2r3ÚcenterZ ptp_boundZfinite_segmentsZinfinite_segmentsZpointidxr&ÚiÚtÚnZmidpointÚ directionZ far_pointr r r r—sN=   . *       )N)N)N)N)Z __future__rrrZnumpyr6Zscipy._lib.decoratorrZ _decoratorÚ__all__rrrrrr r r r Ús    6 9