U i2e\@sddlmZddlZddlmZmZmZddlmZ ddl m Z ddl ZddlmZmZddlmZmZmZddlmZerddlmZddlmZdd lmZddlmZ Gd d d eZ!Gd d d e!Z"Gddde!Z#dS)) annotationsN) TYPE_CHECKINGAnycast)FillTypeLineType)filled_to_mpl_pathslines_to_mpl_pathsmpl_codes_to_offsets)Renderer)Axes)Figure) ArrayLikec @s@eZdZUded<ded<ded<dGd d d dd dddddZddddZddddZdddddZdHdd dd!d"dd#d$d%ZdId(d(dd!d"d d"dd)d*d+Z dJd-d.dd!d"d"dd/d0d1Z dKd(d(d2dd!dd3d4d5Z dLd!ddd7d8d9Z d:dd;d<Z ddd=d>ZdMd!dd dd?d@dAZdNd(d(d(dd!d!dddDdEdFZd S)O MplRendererr _axesr _figbool _want_tight rTNinttuple[float, float]z str | Nonezdict[str, Any] | NoneNone)nrowsncolsfigsize show_framebackend gridspec_kwreturnc Cs|dk rddl}||t|dddd}|dk r<||d<ntdd|d<tj||f|\|_} | |_|s|jD]} | d qtd|_ dS) NrFT)rZsqueezeZsharexZshareyrequal)ZaspectZ subplot_kwoff) matplotlibZusedictpltZsubplotsrflattenrZaxisr) selfrrrrrrr#kwargsZaxesaxr*?/tmp/pip-unpacked-wheel-_24pu9r8/contourpy/util/mpl_renderer.py__init__'s      zMplRenderer.__init__)r cCst|drt|jdS)Nr)hasattrr%closerr'r*r*r+__del__Bs zMplRenderer.__del__cCsL|jD]"}t|ddr|jddd|_q|jrHt|jdkrH|jdS)N_need_autoscaleFT)Ztightr)rgetattrZautoscale_viewr1rlenrZ tight_layoutr'r)r*r*r+ _autoscaleFs    zMplRenderer._autoscale Axes | int)r)r cCst|tr|j|}|SN) isinstancerrr4r*r*r+_get_axQs  zMplRenderer._get_axrC0ffffff?cpy.FillReturnrstrfloat)filled fill_typer)coloralphar cCs<||}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): Type of ``filled`` data, as returned by :attr:`~contourpy.ContourGenerator.fill_type`. 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 edgecolorslwrBTN)r9r mcollectionsPathCollectionadd_collectionr1)r'r?r@r)rArBpaths collectionr*r*r+r?Vs   zMplRenderer.filledblack皙?r)xyr)rArB point_colorquad_as_tri_alphar c Cs||}|||\}}t||d}|j|||j|jf||dkrd|ddddf|ddddf|ddddf|ddddf} d|ddddf|ddddf|ddddf|ddddf} ||d<|jt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdf||dk r|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. )rArBrg?NrrB)rQo)rArBmarkerrET) r9 _grid_as_2dr$plotTnpstackZreshaper1) r'rMrNr)rArBrOrPr(xmidymidr*r*r+gridrs&   TT4444 zMplRenderer.grid?cpy.LineReturnr)lines line_typer)rArB linewidthr c Cs<||}t||}tj|d|||d}||d|_dS)a6Plot contour lines on a single Axes. Args: lines (sequence of arrays): Contour line data as returned by :func:`~contourpy.ContourGenerator.lines`. line_type (LineType): Type of ``lines`` data, as returned by :attr:`~contourpy.ContourGenerator.line_type`. 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``. rCrDTN)r9r rFrGrHr1) r'r_r`r)rArBrarIrJr*r*r+r_s   zMplRenderer.linesz'ArrayLike | np.ma.MaskedArray[Any, Any]rMrNzr)rAr cCsTtj|}|tjjkrdS||}|||\}}|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"``. NrS)c)rXmaZgetmaskZnomaskr9rUrV)r'rMrNrcr)rAmaskr*r*r+rfs    zMplRenderer.maskF)filename transparentr cCs||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``. )rhN)r5rsavefig)r'rgrhr*r*r+saveszMplRenderer.savez io.BytesIOcCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)formatr)r5ioBytesIOrriseek)r'bufr*r*r+save_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)r5r%showr/r*r*r+rqszMplRenderer.show)titler)rAr cCs.|r||j||dn|||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)r9 set_title)r'rrr)rAr*r*r+rrs zMplRenderer.titlegreen.1f)rMrNrcr)rAfmt quad_as_trir c Cs2||}|||\}}t|}|j\}} t|D]H} t| D]:} |j|| | f|| | f|| | f|dd|ddqBq6|r.t|dD]} t| dD]} t|| | d| | df} t|| | d| | df} t|| | d| | df}|j| | ||dd|ddqqdS)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. centerTZhavarAZclip_onrN)r9rUrXasarrayshaperangetextmean)r'rMrNrcr)rArvrwnynxjiZxxyyzzr*r*r+z_valuess&     , """zMplRenderer.z_values)rrrTNN)rr:r;)rrKrLNr)rr:r]r)rrK)F)rN)rrtruF)__name__ __module__ __qualname____annotations__r,r0r5r9r?r\r_rfrjrprqrrrr*r*r*r+rsP    3#  rcs.eZdZdZd dddddfdd 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()``. rrrrr)rrrr cshddddddd}tj|||dd|d|jD],}|d|d|g|gq0d|_dS) Ng{Gz?gGz?)leftrighttopbottomZwspaceZhspaceTZAgg)rrrF)superr,rZ set_xmarginZ set_ymarginZ set_xticksZ set_yticksr)r'rrrZgridspecr) __class__r*r+r,7s*     zMplTestRenderer.__init__)rrr)rrr__doc__r, __classcell__r*r*rr+r1s rc seZdZdZd:dddddd fd d Zd d d dddddddZddddddZddddddZd;ddd%ddddddddd& fd'd( Zdd-d-d-d%ddd.d2d3Z d?d-d-d-dd6d%ddd7d8d9Z Z S)@MplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rrTrrrr)rrrrr cst||||dSr7)rr,)r'rrrrrr*r+r,XszMplDebugRenderer.__init__r zcpy.CoordinateArrayr=r>)r) line_startline_endrArB arrow_sizer c Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-rdrB)rXsqrtdotr|rYrV) r'r)rrrArBrZmidZalongrZarrowr*r*r+_arrowas zMplDebugRenderer._arrowr<rz2tuple[list[cpy.PointArray], list[cpy.OffsetArray]])r?r@r c Cst|tjkr8trttj|}|d}dd|dD}n4|tjkrztrRttj|}dd|dD}dd|dD}n|tjkrtrttj |}|d}|d}n|tj krtrttj |}dd|dD}dd|dD}n|tj krtrttj |}g}g}t|D]v\}}}|dkr0qtrN|dk rJ|dk sNt|t||dd 7}t||dd }|d d|D7}qn|tjkr\trttj|}g}g}t|D]\}} }|dkrؐqtr| dk r|dk sttt|dD]N} | || || dd} ||| d| d || | dqqntd |d ||fS) NrcSsg|] }t|qSr*r .0codesr*r*r+ ~szKMplDebugRenderer._filled_to_lists_of_points_and_offsets..rcSsg|]}|dk r|qSr7r*rpointsr*r*r+rscSsg|]}|dk rt|qSr7rrr*r*r+rscSsg|]}|dk r|qSr7r*rr*r*r+rscSsg|]}|dk r|qSr7r*)roffsetsr*r*r+rsrQcSsg|] }t|qSr*rrr*r*r+rszRendering FillType  not implemented)rZ OuterCoderrcpyZFillReturn_OuterCodeChunkCombinedCodeZFillReturn_ChunkCombinedCodeZ OuterOffsetZFillReturn_OuterOffsetChunkCombinedOffsetZFillReturn_ChunkCombinedOffsetZChunkCombinedCodeOffsetZ"FillReturn_ChunkCombinedCodeOffsetzipAssertionErrorrXsplitZChunkCombinedOffsetOffsetZ$FillReturn_ChunkCombinedOffsetOffsetr~r3append RuntimeError) r'r?r@ all_points all_offsetsrrZ outer_offsetsZ all_codesrrZoffsr*r*r+&_filled_to_lists_of_points_and_offsetsusb               z7MplDebugRenderer._filled_to_lists_of_points_and_offsetsr^rzlist[cpy.PointArray])r_r`r c Csh|tjkr"trttj|}|}nB|tjkrHtr        $  &z)MplDebugRenderer._lines_to_list_of_pointsrC1r;r:redrLr6) r?r@r)rArB line_color line_alpharOstart_point_colorrr c  st||||||dkr(|dkr(dS||}|||\} } |dk rt| | D]\} }t|dd|ddD]\}}| ||}|j|dddf|dddf||d| dkrvt|}t|dD]$}||||||d||| qqvqT|dk rt| | D]\} }t j |dt d}d||ddd<| dk rZ|dd}d||<|j| dddf|| dddf|d||d| dk r|j| dddf|| dddf|d| |dqdS) NrQrrrr)ZdtypeFrS) rr?r9rrrVr3r~rrXZonesr)r'r?r@r)rArBrrrOrrrrrrstartendZxysnrrfZ start_indicesrr*r+r?sB  " *&    *zMplDebugRenderer.filledr]) r_r`r)rArBrarOrrr c  s>t||||||| dkr*|dkr*dS||}|||} | dkr| D]:} tt| dD]$} ||| | | | d||| q`qL|dk r:| D]} d} t| }|dk r |j| d| dd||dd} | dd| ddkr | dd| ddkr |d8}|j| | |df| | |dfd||dqdS) Nrrr)rr)rrrSrrQ)rr_r9rr~r3rrV)r'r_r`r)rArBrarOrrrliner start_indexZ end_indexrr*r+r_s,   $  4"zMplDebugRenderer.linesrrbc Cs||}|||\}}t|}|j\}}t|D]J}t|D]<} | ||} |j||| f||| ft| dd|ddqBq6dS)NrrTry)r9rUrXr|r}r~rr=) r'rMrNrcr)rArrrrquadr*r*r+ point_numbers(s      $zMplDebugRenderer.point_numbersbluec Cs||}|||\}}t|}|j\}}td|D]}td|D]|} | ||} ||d|d| d| df} ||d|d| d| df} |j| | t| dd|ddqFq8dS)NrrxTry) r9rUrXr|r}r~rrr=) r'rMrNrcr)rArrrrrrZr[r*r*r+ quad_numbers:s    ((zMplDebugRenderer.quad_numbersNrtz float | None)rMrNrc lower_level upper_levelr)rAr c Cs||}|||\}}t|}|j\}} t|D]n} t| D]`} || | f} |dk rh| |krhd} n| |krvd} nd} |j|| | f|| | f| dd|ddqBq6dS)Nr{rrrrTry)r9rUrXr|r}r~r)r'rMrNrcrrr)rArrrrrZz_levelr*r*r+z_levelsMs       zMplDebugRenderer.z_levels)rrrT)rrr;r:r;r:rrL)rr:r]rr:rrL)rr)rr)Nrrt)rrrrr,rrrr?r_rrrrr*r*rr+rRsH 9)(4&*r)$ __future__rrltypingrrrZmatplotlib.collections collectionsrFZmatplotlib.pyplotZpyplotr%ZnumpyrXZ contourpyrrZcontourpy.util.mpl_utilrr r Zcontourpy.util.rendererr Zmatplotlib.axesr Zmatplotlib.figurer Z numpy.typingrZcontourpy._contourpyZ _contourpyrrrrr*r*r*r+s$        !