o f]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)rZsqueezeZsharexZshareyr#ZaspectZequalZ subplot_kwZoff) Z matplotlibZusepltZsubplotsrZflattenrZaxisr) selfrrrr r!r#ZmplkwargsZaxesaxr+;lib/python3.10/site-packages/contourpy/util/mpl_renderer.py__init__+s        zMplRenderer.__init__cCst|dr t|jdSdS)Nr)hasattrr'closerr(r+r+r,__del__Gs zMplRenderer.__del__cCsT|jD]}t|ddr|jddd|_q|jr&t|jdkr(|jdSdSdS)N_need_autoscaleFT)Ztightr)rgetattrZautoscale_viewr2rlenrZ tight_layoutr(r*r+r+r, _autoscaleKs   zMplRenderer._autoscaler* Axes | intrcCst|tr |j|}|SN) isinstancerrr5r+r+r,_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``. nonerZ facecolorsZ edgecolorslwrCTN)r r:r mcollectionsPathCollectionadd_collectionr2)r(r=r?r*rArCpaths collectionr+r+r,r=[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. )rArCrg?NrrC)rSo)rArCZmarkerrGT)r: _grid_as_2dplotTnpstackZreshaper2) r(rOrPr*rArCrQrRr)xmidymidr+r+r,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``. rErFTN)r r:r rHrIrJr2) r(r_rar*rArCrcrKrLr+r+r,r_s     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"``. NrU)c)rYZmaZgetmaskZnomaskr:rVrW)r(rOrPrdr*rAmaskr+r+r,rgs   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``. )riN)r6rsavefig)r(rhrir+r+r,saveszMplRenderer.save io.BytesIOcCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)formatr)r6ioBytesIOrrjseek)r(bufr+r+r,save_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)r6r'showr0r+r+r,rss 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. )rAN)r:Z set_title)r(rtr*rAr+r+r,rts 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. centerTZhaZvarAZclip_onrN)r:rVrYasarrayshaperangetextmean)r(rOrPrdr*rArwrxnynxjiZxxZyyzzr+r+r,z_values s,     ,"""zMplRenderer.z_values)rrrTNN)rrrrrrr rr!r"r#r$r%r&)r%r&)r*r7r%r)rr;r<) r=r>r?r@r*r7rArBrCrDr%r&)rrMrNNr)rOrrPrr*r7rArBrCrDrQr"rRrDr%r&)rr;r^r)r_r`rarbr*r7rArBrCrDrcrDr%r&)rrM) rOrrPrrdrer*r7rArBr%r&)F)rhrBrirr%r&)r%rl)rN)rtrBr*r7rAr"r%r&)rrurvF)rOrrPrrdrr*r7rArBrwrBrxrr%r&)__name__ __module__ __qualname____doc____annotations__r-r1r6r:r=r]r_rgrkrrrsrtrr+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()``. rrrrrrrr%r&cshddddddd}tj|||dd|d|jD]}|d|d|g|gqd|_dS) Ng{Gz?gGz?)leftrighttopbottomZwspaceZhspaceTZAgg)r r!r#F)superr-rZ set_xmarginZ set_ymarginZ set_xticksZ set_yticksr)r(rrrZgridspecr* __class__r+r,r->s        zMplTestRenderer.__init__)rrr)rrrrrrr%r&)rrrrr- __classcell__r+r+rr,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||||dSr8)rr-)r(rrrr rr+r,r-_szMplDebugRenderer.__init__r*r line_startcpy.CoordinateArrayline_endrArBrCrD arrow_sizec Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-rfrC)rYZsqrtdotr|rZrW) r(r*rrrArCrZmidZalongrZarrowr+r+r,_arrowhs 0zMplDebugRenderer._arrowrC1r<r;redrNr=r>r?r@r7 line_color line_alpharQstart_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) NrSrrrr)ZdtypeFrU)r rr=r:rrZChunkCombinedOffsetziprWr4r~rrYZonesr)r(r=r?r*rArCrrrQrrZpointsZoffsetsstartendZxysnrrgZ start_indicesrr+r,r=|sL  " *"  **zMplDebugRenderer.filledr^r_r`rarbrcc  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)rrrUrrS)r rr_r:r rZSeparaterrcpyZLineReturn_Separater~r4rrW)r(r_rar*rArCrcrQrrZseparate_linesliner start_indexZ end_indexrr+r,r_s6   "0"zMplDebugRenderer.linesrOrrPrdc Cs||}|||\}}t|}|j\}}t|D]%}t|D]} | ||} |j||| f||| ft| dd|ddq!qdS)NrrTrzr:rVrYr|r}r~rrB) r(rOrPrdr*rArrrrquadr+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)NrryTrz) r:rVrYr|r}r~rrrB) r(rOrPrdr*rArrrrrr[r\r+r+r, quad_numberss    ((zMplDebugRenderer.quad_numbersNru 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)Nr{rrrrTrzr)r(rOrPrdrrr*rArrrrrZz_levelr+r+r,z_levelss"      $zMplDebugRenderer.z_levels)rrrT) rrrrrrr rr%r&)r*rrrrrrArBrCrDrrDr%r&)rrr<r;r<r;rrN)r=r>r?r@r*r7rArBrCrDrrBrrDrQrBrrBrrDr%r&)rr;r^rr;rrN)r_r`rarbr*r7rArBrCrDrcrDrQrBrrBrrDr%r&)rr) rOrrPrrdrr*r7rArBr%r&)rr)Nrru)rOrrPrrdrrrDrrr*r7rArBr%r&) rrrrr-rr=r_rrrrr+r+rr,rYsF 9-  r)+Z __future__rrntypingrrrZmatplotlib.collections collectionsrHZmatplotlib.pyplotZpyplotr'ZnumpyrYZ contourpyrrZcontourpy.convertrr Zcontourpy.enum_utilr r Zcontourpy.util.mpl_utilr r Zcontourpy.util.rendererrZcollections.abcrZmatplotlib.axesrZmatplotlib.figurerZ numpy.typingrZcontourpy._contourpyZ _contourpyrrrrr+r+r+r,s,          !