a |=icK@s~ddlZddlmZddlmZddlZddl m Z m Z ddl m Z mZmZGdddZGdddeZGd d d eZdS) N)FillTypeLineType)filled_to_mpl_pathslines_to_mpl_pathsmpl_codes_to_offsetsc@seZdZdZd+ddZdd Zd d Zd d ZddZd,ddZ d-ddZ d.ddZ d/ddZ d0dd Z d!d"Zd#d$Zd1d%d&Zd2d)d*ZdS)3 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. r r TNc Cs|durddl}||t|dddd}|dur<||d<ntdd|d<tj||fi|\|_} | |_|s|jD]} | d qxd|_ dS) NrFT)figsizeZsqueezeZsharexZsharey gridspec_kwequal)ZaspectZ subplot_kwoff) matplotlibZusedictpltZsubplots_figflatten_axesZaxis _want_tight) selfnrowsncolsr show_framebackendr rkwargsZaxesaxrl/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/contourpy/util/mpl_renderer.py__init__s     zMplRenderer.__init__cCst|drt|jdS)Nr)hasattrrcloserrrrr__del__.s zMplRenderer.__del__cCsL|jD]"}t|ddr|jddd|_q|jrHt|jdkrH|jdS)N_need_autoscaleFT)Ztightr)rgetattrZautoscale_viewr$rlenrZ tight_layoutrrrrr _autoscale2s    zMplRenderer._autoscalecCst|tr|j|}|SN) isinstanceintrr'rrr_get_ax=s  zMplRenderer._get_axcCs6t|}t|}|jdkr.t||\}}||fS)Nr)npasarrayndimZmeshgrid)rxyrrr _grid_as_2dBs    zMplRenderer._grid_as_2drC0ffffff?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 edgecolorsZlwalphaTN)r,r mcollectionsPathCollectionadd_collectionr$)rfilled fill_typercolorr7paths collectionrrrr;Is    zMplRenderer.filledblack皙?c Cs||}|||\}}t||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 d 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. )r=r7rg?Nrr7)rBo)r=r7markerT) r,r2rplotTr-stackZreshapeZscatterr$) rr0r1rr=r7 point_colorZquad_as_tri_alpharxmidymidrrrgrid^s&   TT4444 zMplRenderer.grid?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``. r5r6TN)r,rr8r9r:r$) rlines line_typerr=r7 linewidthr>r?rrrrNs    zMplRenderer.linescCsTtj|}|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"``. NrD)c)r-maZgetmaskZnomaskr,r2rF)rr0r1zrr=maskrrrrTs   zMplRenderer.maskFcCs||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``. ) transparentN)r(rsavefig)rfilenamerUrrrsaveszMplRenderer.savecCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)formatr)r(ioBytesIOrrVseek)rbufrrrsave_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)r(rshowr"rrrr_szMplRenderer.showcCs.|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 ``None`` which is ``black``. )r=N)r, set_title)rtitlerr=rrrras zMplRenderer.titlegreen.1fc 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. centerTZhavar=Zclip_onrN)r,r2r-r.shaperangetextmean)rr0r1rSrr=fmtZ quad_as_trinynxjixxyyzzrrrz_valuess$     , """zMplRenderer.z_values)rrr TNN)rr3r4)rr@rANr)rr3rMr)rr@)F)rN)rrbrcF)__name__ __module__ __qualname____doc__rr#r(r,r2r;rLrNrTrXr^r_rartrrrrr s     &     rcs"eZdZdZdfdd 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()``. rr cshddddddd}tj|||dd|d|jD],}|d|d|g|gq0d|_dS) Ng{Gz?gGz?)leftrighttopbottomZwspaceZhspaceTZAgg)rrr F)superrrZ set_xmarginZ set_ymarginZ set_xticksZ set_yticksr)rrrr Zgridspecr __class__rrrs       zMplTestRenderer.__init__)rrr )rurvrwrxr __classcell__rrrrrysrycsdeZdZdZdfdd ZddZdfdd Zdfdd Zd ddZd!ddZ d"ddZ Z S)#MplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rr Tcst||||dSr))rr)rrrr rrrrrszMplDebugRenderer.__init__c Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-rQr7)r-sqrtdotr.rHrF) rr line_startline_endr=r7 arrow_sizemidZalongr{arrowrrr_arrow s zMplDebugRenderer._arrowrC1r4r3redrAc  st||||||dur(|dur(dS||}|tjkrZ|d} dd|dD} n|tjkrdd|dD} dd|dD} nn|tjkr|d} |d} nP|tjkrdd|dD} dd|dD} n|tjkrPg} g} t |D]T\} }}| durq| t | |dd 7} t ||dd }| d d|D7} qn|tj krg} g} t |D]z\} }}| durqlt t|dD]N}|||||dd}| | |d|d | ||dqqlntd |d |durt | | D]\} }t |dd |ddD]\}}| ||}|j|dddf|dddf||d | dkr0t|}t |dD]&}||||||d||| qq0q|durt | | D]\} }t j|d td}d||ddd<| dur|dd }d||<|j| dddf|| dddf|d||d | dur|j| dddf|| dddf|d| |d qdS)NrcSsg|] }t|qSrr.0codesrrr 7z+MplDebugRenderer.filled..rcSsg|]}|dur|qSr)rrpointsrrrr9rcSsg|]}|durt|qSr)rrrrrr:rcSsg|]}|dur|qSr)rrrrrr?rcSsg|]}|dur|qSr)r)roffsetsrrrr@rrBcSsg|] }t|qSrrrrrrrIrzRendering FillType  not implementedrr~)ZdtypeFrD)rr;r,rZ OuterCodeChunkCombinedCodeZ OuterOffsetChunkCombinedOffsetZChunkCombinedCodeOffsetzipr-splitZChunkCombinedOffsetOffsetrir&append RuntimeErrorrFrZonesbool)rr;r<rr=r7Z line_colorZ line_alpharIstart_point_colorrZ all_pointsZ all_offsetsrrZ outer_offsetsrrpZoffsstartendZxysnrTZ start_indicesrrrr;,sv           " * ,   * *zMplDebugRenderer.filledrMc  s4t||||||| dkr*|dur*dS||}|tjkrD|} n|tjkrX|d} n|tjkrg} t|D]L\} } | durnt| } t t | dD]"}| | | || |dqqnnp|tj krg} t|D]D\} } | durt t | dD]"}| | | || |dqqnt d|d| dkr|| D]>}t t |dD]&}||||||d||| qPq<|dur0| D]}d}t |}|dur|j|d|dd||d d}|dd|d dkr|dd|d dkr|d8}|j|||df|||dfd||d qdS) Nr~rrzRendering LineType r)rr)rrrDrrB)rrNr,rZSeparateZ SeparateCoderrrrir&rrrrrF)rrNrOrr=r7rPrIrr all_linesrrrrpline start_indexZ end_indexrrrrNrsL     $ $ (  4"zMplDebugRenderer.linesc Cs||}|||\}}t|}|j\}}t|D]J}t|D]<} | ||} |j||| f||| ft| dd|ddqBq6dS)Nr{r|Tre)r,r2r-r.rhrirjstr) rr0r1rSrr=rmrnrorpquadrrr point_numberss      $zMplDebugRenderer.point_numbersbluec Cs||}|||\}}t|}|j\}}td|D]}td|D]|} | ||} ||d|d| d| df} ||d|d| d| df} |j| | t| dd|ddqFq8dS)NrrdTre) r,r2r-r.rhrirkrjr) rr0r1rSrr=rmrnrorprrJrKrrr quad_numberss    ((zMplDebugRenderer.quad_numbersNrbc Cs||}|||\}}t|}|j\}} t|D]n} t| D]`} || | f} |durh| |krhd} n| |krvd} nd} |j|| | f|| | f| dd|ddqBq6dS)Nrgrrrzr}Tre)r,r2r-r.rhrirj)rr0r1rSZ lower_levelZ upper_levelrr=rmrnrorprsZz_levelrrrz_levelss       zMplDebugRenderer.z_levels)rrr T)rrr4r3r4r3rrA)rr3rMrr3rrA)rr)rr)Nrrb) rurvrwrxrrr;rNrrrrrrrrrs F. r)rZZmatplotlib.collections collectionsr8Zmatplotlib.pyplotZpyplotrnumpyr-Z contourpyrrZmpl_utilrrrrryrrrrrs  p