o Xεi]N@sddlmZddlZddlmZmZmZddlmZ ddl m Z ddl ZddlmZmZddlmZmZddlmZmZddlmZmZddlmZerjdd lmZdd lm Z dd l!m"Z"dd l#m$Z$ddl%m&Z'Gd ddeZ(Gddde(Z)Gddde(Z*dS)) annotationsN) 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#mplkwargsaxesaxr9O/home/ubuntu/.local/lib/python3.10/site-packages/contourpy/util/mpl_renderer.py__init__+s        zMplRenderer.__init__cCst|dr t|jdSdS)Nr)hasattrr0closerr4r9r9r:__del__Gs zMplRenderer.__del__cCsT|jD]}t|ddr|jddd|_q|jr&t|jdkr(|jdSdSdS)N_need_autoscaleFT)tightr)rgetattrautoscale_viewr@rlenr tight_layoutr4r8r9r9r: _autoscaleKs   zMplRenderer._autoscaler8 Axes | intrcCst|tr |j|}|SN) isinstancerrrFr9r9r:_get_axVs  zMplRenderer._get_axrC0ffffff?filledcpy.FillReturn fill_typeFillType | strcolorstralphafloatcCsDt|}||}t||}tj||dd|d}||d|_dS)aPlot filled contours on a single Axes. Args: filled (sequence of arrays): Filled contour data as returned by :func:`~contourpy.ContourGenerator.filled`. fill_type (FillType or str): Type of ``filled`` data as returned by :attr:`~contourpy.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 edgecolorslwrTTN)r rKr mcollectionsPathCollectionadd_collectionr@)r4rNrPr8rRrTpaths collectionr9r9r:rN[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. )rRrTrg?NrrT)rfo)rRrTmarkerrZT)rK _grid_as_2dplotTnpstackreshaper@) r4rbrcr8rRrTrdrer6xmidymidr9r9r:gridxs&  TT4444 zMplRenderer.grid?linescpy.LineReturn line_typeLineType | str linewidthc CsDt|}||}t||}tj|d|||d}||d|_dS)aRPlot contour lines on a single Axes. Args: lines (sequence of arrays): Contour line data as returned by :func:`~contourpy.ContourGenerator.lines`. line_type (LineType or str): Type of ``lines`` data as returned by :attr:`~contourpy.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``. rVrWTN)r rKr r[r\r]r@) r4rtrvr8rRrTrxr^r_r9r9r:rts     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"``. Nrh)c)rmmagetmasknomaskrKrjrk)r4rbrcryr8rRmaskr9r9r: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)rGrsavefig)r4rrr9r9r:saveszMplRenderer.save io.BytesIOcCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. png)formatr)rGioBytesIOrrseek)r4bufr9r9r:save_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)rGr0showr>r9r9r: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. )rRN)rK set_title)r4rr8rRr9r9r:rs zMplRenderer.titlegreen.1ffmt quad_as_tric Cs4||}|||\}}t|}|j\}} t|D]$} t| D]} |j|| | f|| | f|| | f|dd|ddq!q|rt|dD]O} t| dD]D} t|| | d| | df} t|| | d| | df} 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. centerThavarRclip_onrN)rKrjrmasarrayshaperangetextmean)r4rbrcryr8rRrrnynxjixxyyzzr9r9r:z_values s,     ,"""zMplRenderer.z_values)rrrTNN)rrrrrrr rr!r"r#r$r%r&)r%r&)r8rHr%r)rrLrM) rNrOrPrQr8rHrRrSrTrUr%r&)rr`raNr)rbrrcrr8rHrRrSrTrUrdr"rerUr%r&)rrLrsr)rtrurvrwr8rHrRrSrTrUrxrUr%r&)rr`) rbrrcrryrzr8rHrRrSr%r&)F)rrSrrr%r&)r%r)rN)rrSr8rHrRr"r%r&)rrrF)rbrrcrryrr8rHrRrSrrSrrr%r&)__name__ __module__ __qualname____doc____annotations__r;r?rGrKrNrrrtrrrrrrr9r9r9r: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()``. rrrrrrrr%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)r4rrrgridspecr8 __class__r9r:r;>s        zMplTestRenderer.__init__)rrr)rrrrrrr%r&)rrrrr; __classcell__r9r9rr:r8s 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. rrTrrrrrr rr%r&cst||||dSrI)rr;)r4rrrr rr9r:r;_szMplDebugRenderer.__init__r8r line_startcpy.CoordinateArrayline_endrRrSrTrU arrow_sizec Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-r{rT)rmsqrtdotrrnrk) r4r8rrrRrTrmidalongrarrowr9r9r:_arrowhs 0zMplDebugRenderer._arrowrC1rMrLredrarNrOrPrQrH line_color line_alphardstart_point_colorc  st|}t||||||dur|durdS||}t||tj}|durt|D]W\} } | dur5q,t| dd| ddD]@\} }| | |}|j|dddf|dddf||d| dkrt |}t |dD]}| |||||d||| qoqBq,|durt|D]d\} } | durqt j | dtd}d|| ddd<| dur| dd}d||<|j| dddf|| dddf|d||d| dur|j| dddf|| dddf|d| |dqdSdS) Nrfrrrr)dtypeFrh)r rrNrKrrChunkCombinedOffsetziprkrDrrrmonesr)r4rNrPr8rRrTrrrdrrpointsoffsetsstartendxysnrr start_indicesrr9r:rN|sL  " *"  **zMplDebugRenderer.filledrsrtrurvrwrxc  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)rrrhrrf)r rrtrKr rSeparaterrcpyLineReturn_SeparaterrDrrk)r4rtrvr8rRrTrxrdrrseparate_linesliner start_index end_indexrr9r:rts6   "0"zMplDebugRenderer.linesrbrrcryc Cs||}|||\}}t|}|j\}}t|D]%}t|D]} | ||} |j||| f||| ft| dd|ddq!qdS)NrrTrrKrjrmrrrrrS) r4rbrcryr8rRrrrrquadr9r9r: 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) rKrjrmrrrrrrS) r4rbrcryr8rRrrrrrrprqr9r9r: 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)r4rbrcryrrr8rRrrrrrz_levelr9r9r:z_levelss"      $zMplDebugRenderer.z_levels)rrrT) rrrrrrr rr%r&)r8rrrrrrRrSrTrUrrUr%r&)rrrMrLrMrLrra)rNrOrPrQr8rHrRrSrTrUrrSrrUrdrSrrSrrUr%r&)rrLrsrrLrra)rtrurvrwr8rHrRrSrTrUrxrUrdrSrrSrrUr%r&)rr) rbrrcrryrr8rHrRrSr%r&)rr)Nrr)rbrrcrryrrrUrrr8rHrRrSr%r&) rrrrr;rrNrtrrrrr9r9rr:rYsF 9-  r)+ __future__rrtypingrrrmatplotlib.collections collectionsr[matplotlib.pyplotpyplotr0numpyrm contourpyrrcontourpy.convertrr contourpy.enum_utilr r contourpy.util.mpl_utilr r contourpy.util.rendererrcollections.abcrmatplotlib.axesrmatplotlib.figurer numpy.typingrcontourpy._contourpy _contourpyrrrrr9r9r9r:s,          !