o hN@sddlmZddlZddlmZddlmZmZmZddl m Z ddl m ZddlZddlmZmZddlmZmZddlmZmZddlmZmZdd lmZerpdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z)GdddeZ*Gddde*Z+Gddde*Z,dS)) annotationsN)pairwise) TYPE_CHECKINGAnycast)FillTypeLineType)convert_filled convert_lines) as_fill_type as_line_type)filled_to_mpl_pathslines_to_mpl_paths)Renderer)Sequence)Axes)Figure) ArrayLikec@seZdZUdZded<ded<ded<   dXdYddZdZddZdZddZd[d!d"Z # $ %d\d]d.d/Z # 0 1 #d^d_d7d8Z # $ 9 d`dad?d@Z # 0dbdcdCdDZ dddedHdIZ dfdKdLZdZdMdNZdgdhdPdQZ # R S EdidjdVdWZd S)k MplRendereraUtility renderer using Matplotlib to render a grid of plots over the same (x, y) range. Args: nrows (int, optional): Number of rows of plots, default ``1``. ncols (int, optional): Number of columns of plots, default ``1``. figsize (tuple(float, float), optional): Figure size in inches, default ``(9, 9)``. show_frame (bool, optional): Whether to show frame and axes ticks, default ``True``. backend (str, optional): Matplotlib backend to use or ``None`` for default backend. Default ``None``. gridspec_kw (dict, optional): Gridspec keyword arguments to pass to ``plt.subplots``, default None. zSequence[Axes]_axesr_figbool _want_tight rTNnrowsintncolsfigsizetuple[float, float] show_framebackend str | None gridspec_kwdict[str, Any] | NonereturnNonec Cs|dur ddl}|||dddd}|dur||d<nddi|d<tj||fi|\|_} | |_|sB|jD]} | d q:d|_dS) NrFT)rsqueezesharexshareyr$aspectequal subplot_kwoff) matplotlibusepltsubplotsrflattenraxisr) selfrrrr!r"r$mplkwargsaxesaxr:o/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/contourpy/util/mpl_renderer.py__init__,s        zMplRenderer.__init__cCst|dr t|jdSdS)Nr)hasattrr1closerr5r:r:r;__del__Hs zMplRenderer.__del__cCsT|jD]}t|ddr|jddd|_q|jr&t|jdkr(|jdSdSdS)N_need_autoscaleFT)tightr)rgetattrautoscale_viewrArlenr tight_layoutr5r9r:r:r; _autoscaleLs   zMplRenderer._autoscaler9 Axes | intrcCst|tr |j|}|SN) isinstancerrrGr:r:r;_get_axWs  zMplRenderer._get_axrC0ffffff?filledcpy.FillReturn fill_typeFillType | strcolorstralphafloatcCsDt|}||}t||}tj||dd|d}||d|_dS)a Plot filled contours on a single Axes. Args: filled (sequence of arrays): Filled contour data as returned by :meth:`~.ContourGenerator.filled`. fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` data as returned by :attr:`~.ContourGenerator.fill_type`, or string equivalent ax (int or Maplotlib Axes, optional): Which axes to plot on, default ``0``. color (str, optional): Color to plot with. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"C0"``. alpha (float, optional): Opacity to plot with, default ``0.7``. noner facecolors edgecolorslwrUTN)r rLr mcollectionsPathCollectionadd_collectionrA)r5rOrQr9rSrUpaths collectionr:r:r;rO\s     zMplRenderer.filledblack皙?xry point_colorquad_as_tri_alphac Cs||}|||\}}||d}|j|||j|jfi||dkrd|ddddf|ddddf|ddddf|ddddf} d|ddddf|ddddf|ddddf|ddddf} ||d<|jt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdfi||dur|j||||d dd d |_dS) aPlot quad grid lines on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color to plot grid lines, default ``"black"``. alpha (float, optional): Opacity to plot lines with, default ``0.1``. point_color (str, optional): Color to plot grid points or ``None`` if grid points should not be plotted, default ``None``. quad_as_tri_alpha (float, optional): Opacity to plot ``quad_as_tri`` grid, default 0. Colors may be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Warning: ``quad_as_tri_alpha > 0`` plots all quads as though they are unmasked. )rSrUrg?NrrU)rgo)rSrUmarkerr[T)rL _grid_as_2dplotTnpstackreshaperA) r5rcrdr9rSrUrerfr7xmidymidr:r:r;gridys&  TT4444 zMplRenderer.grid?linescpy.LineReturn line_typeLineType | str linewidthc CsDt|}||}t||}tj|d|||d}||d|_dS)aWPlot contour lines on a single Axes. Args: lines (sequence of arrays): Contour line data as returned by :meth:`~.ContourGenerator.lines`. line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` data as returned by :attr:`~.ContourGenerator.line_type`, or string equivalent. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color to plot lines. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"C0"``. alpha (float, optional): Opacity to plot lines with, default ``1.0``. linewidth (float, optional): Width of lines, default ``1``. rWrXTN)r rLrr\r]r^rA) r5rurwr9rSrUryr_r`r:r:r;rus     zMplRenderer.linesz'ArrayLike | np.ma.MaskedArray[Any, Any]cCsTtj|}|tjjurdS||}|||\}}|j||||d|ddS)aPlot masked out grid points as circles on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. z (masked array of shape (ny, nx): z-values. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Circle color, default ``"black"``. Nri)c)rnmagetmasknomaskrLrkrl)r5rcrdrzr9rSmaskr:r:r;rs   zMplRenderer.maskFfilename transparentcCs||jj||ddS)zSave plots to SVG or PNG file. Args: filename (str): Filename to save to. transparent (bool, optional): Whether background should be transparent, default ``False``. )rN)rHrsavefig)r5rrr:r:r;saveszMplRenderer.save io.BytesIOcCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. png)formatr)rHioBytesIOrrseek)r5bufr:r:r;save_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)rHr1showr?r:r:r;rs zMplRenderer.showtitlecCs0|r||j||ddS|||dS)aSet the title of a single Axes. Args: title (str): Title text. ax (int or Matplotlib Axes, optional): Which Axes to set the title of, default ``0``. color (str, optional): Color to set title. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default is ``None`` which uses Matplotlib's default title color that depends on the stylesheet in use. )rSN)rL set_title)r5rr9rSr:r:r;rs zMplRenderer.titlegreen.1ffmt quad_as_tric Cs@||}|||\}}t|}|j\}} t|D]$} t| D]} |j|| | f|| | f|| | f|dd|ddq!q|rt|dD]U} t| dD]J} tj|| | d| | dftjd} tj|| | d| | dftjd} t|| | d| | df}|j| | ||dd|ddqPqHdSdS)aShow ``z`` values on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. z (array-like of shape (ny, nx): z-values. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color of added text. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"green"``. fmt (str, optional): Format to display z-values, default ``".1f"``. quad_as_tri (bool, optional): Whether to show z-values at the ``quad_as_tri`` centers of quads. Warning: ``quad_as_tri=True`` shows z-values for all quads, even if masked. centerThavarSclip_onrdtypeN) rLrkrnasarrayshaperangetextmeanfloat64)r5rcrdrzr9rSrrnynxjixxyyzzr:r:r;z_values s,     ,(("zMplRenderer.z_values)rrrTNN)rrrrrr r!rr"r#r$r%r&r')r&r')r9rIr&r)rrMrN) rOrPrQrRr9rIrSrTrUrVr&r')rrarbNr)rcrrdrr9rIrSrTrUrVrer#rfrVr&r')rrMrtr)rurvrwrxr9rIrSrTrUrVryrVr&r')rra) rcrrdrrzr{r9rIrSrTr&r')F)rrTrrr&r')r&r)rN)rrTr9rIrSr#r&r')rrrF)rcrrdrrzrr9rIrSrTrrTrrr&r')__name__ __module__ __qualname____doc____annotations__r<r@rHrLrOrsrurrrrrrr:r:r:r;rsT       ! 3 $  rcs*eZdZdZ   d dfd d ZZS)MplTestRendererzTest renderer implemented using Matplotlib. No whitespace around plots and no spines/ticks displayed. Uses Agg backend, so can only save to file/buffer, cannot call ``show()``. rrrrrrr r&r'cshddddddd}tj|||dd|d|jD]}|d|d|g|gqd|_dS) Ng{Gz?gGz?)leftrighttopbottomwspacehspaceTAgg)r!r"r$F)superr<r set_xmargin set_ymargin set_xticks set_yticksr)r5rrrgridspecr9 __class__r:r;r<?s        zMplTestRenderer.__init__)rrr)rrrrrr r&r')rrrrr< __classcell__r:r:rr;r9s rcseZdZdZ    dEdFfdd ZdGddZ       !dHdIfd+d, Z   -   !dJdKfd3d4 Z  dLdMd9d:Z  ;dNdMd  ?dOdPdCdDZ Z S)QMplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rrTrrrrr r!rr&r'cst||||dSrJ)rr<)r5rrrr!rr:r;r<`szMplDebugRenderer.__init__r9r line_startcpy.CoordinateArrayline_endrSrTrUrV arrow_sizec Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-r|rU)rnsqrtdotrrorl) r5r9rrrSrUrmidalongrarrowr:r:r;_arrowis 0zMplDebugRenderer._arrowrC1rNrMredrbrOrPrQrRrI line_color line_alpharestart_point_colorc  st|}t||||||dur|durdS||}t||tj}|dur{t|D]N\} } | dur5q,t| D]@\} }| | |}|j |dddf|dddf||d| dkryt |}t |dD]}| |||||d||| qfq9q,|durt|D]d\} } | durqt j| dtd}d|| ddd<| dur| dd}d||<|j | dddf|| dddf|d||d| dur|j | dddf|| dddf|d| |dqdSdS) NrrrrrgrFri)r rrOrLr rChunkCombinedOffsetziprrlrErrrnonesr)r5rOrQr9rSrUrrrerrpointsoffsetsstartendxysnrr start_indicesrr:r;rO}sL   *"  **zMplDebugRenderer.filledrtrurvrwrxryc  sTt|}t||||||| dkr|durdS||}t||tj} tr-tt j | } | dkrQ| D]} t t | dD]} | || | | | d||| q=q3|dur| D]P} d} t | }|dur|j| d| dd||dd} | dd| ddkr| dd| ddkr|d8}|j| | |df| | |dfd||dqWdSdS) Nrrr)rr)rrrirrg)r rrurLr rSeparaterrcpyLineReturn_SeparaterrErrl)r5rurwr9rSrUryrerrseparate_linesliner start_index end_indexrr:r;rus6   "0"zMplDebugRenderer.linesrcrrdrzc Cs||}|||\}}t|}|j\}}t|D]%}t|D]} | ||} |j||| f||| ft| dd|ddq!qdS)NrrTrrLrkrnrrrrrT) r5rcrdrzr9rSrrrrquadr:r:r; point_numberss      $zMplDebugRenderer.point_numbersbluec Cs||}|||\}}t|}|j\}}td|D]F}td|D]>} | ||} ||d|d| d| df} ||d|d| d| df} |j| | t| dd|ddq#qdS)NrrTr) rLrkrnrrrrrrT) r5rcrdrzr9rSrrrrrrqrrr:r:r; quad_numberss    ((zMplDebugRenderer.quad_numbersNr lower_level upper_level float | Nonec Cs||}|||\}}t|}|j\}} t|D]9} t| D]2} || | f} |dur4| |kr4d} n | |kr;d} nd} |j|| | f|| | ft| dd|ddq!qdS)NrrrrrTrr)r5rcrdrzrrr9rSrrrrrz_levelr:r:r;z_levelss"      $zMplDebugRenderer.z_levels)rrrT) rrrrrr r!rr&r')r9rrrrrrSrTrUrVrrVr&r')rrrNrMrNrMrrb)rOrPrQrRr9rIrSrTrUrVrrTrrVrerTrrTrrVr&r')rrMrtrrMrrb)rurvrwrxr9rIrSrTrUrVryrVrerTrrTrrVr&r')rr) rcrrdrrzrr9rIrSrTr&r')rr)Nrr)rcrrdrrzrrrVrrr9rIrSrTr&r') rrrrr<rrOrurrrrr:r:rr;rZsF 9-  r)- __future__rr itertoolsrtypingrrrmatplotlib.collections collectionsr\matplotlib.pyplotpyplotr1numpyrn contourpyrrcontourpy.convertr r contourpy.enum_utilr r contourpy.util.mpl_utilr rcontourpy.util.rendererrcollections.abcrmatplotlib.axesrmatplotlib.figurer numpy.typingrcontourpy._contourpy _contourpyrrrrr:r:r:r;s.           !