\e-N.ddlmZddlmZddlZddlmZmZmZddl m Z ddl m ZddlZddlmZmZddlmZmZddlmZmZddlmZmZdd lmZerdd lm Z dd l!m"Z"dd l#m$Z$ddl%m&Z'Gd deZ(Gdde(Z)Gdde(Z*dS)) annotations)SequenceN) TYPE_CHECKINGAnycast)FillTypeLineType)convert_filled convert_lines) as_fill_type as_line_type)filled_to_mpl_pathslines_to_mpl_paths)Renderer)Axes)Figure) ArrayLikeceZdZUdZded<ded<ded< dKdLdZdMdZdMdZdNdZ dOdPd*Z dQdRd2Z dSdTd9Z dUdVd<Z dWdXd@Z dYdBZdMdCZdZd[dEZ d\d]dJZd S)^ 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] | NonereturnNonecP|ddl}||t|ddd}|||d<ntd|d<tj||fi|\|_} | |_|s|jD]} | d d|_ dS) NrFT)r squeezesharexshareyr%equal)aspect subplot_kwoff) matplotlibusedictpltsubplotsrflattenraxisr) selfrrr r"r#r%r1kwargsaxesaxs ;lib/python3.11/site-packages/contourpy/util/mpl_renderer.py__init__zMplRenderer.__init__*s       NN7 # # #!%guTZ^!_!_!_  "$/F= ! !#'w#7#7#7F< ,ue>>v>> 4\\^^  j  c\t|drtj|jdSdS)Nr)hasattrr4closerr8s r<__del__zMplRenderer.__del__Es6 4  ! Idi  ! !r>c|jD]0}t|ddr|dd|_1|jr3t |jdkr|jdSdSdS)N_need_autoscaleFT)tightr)rgetattrautoscale_viewrErlenr tight_layoutr8r;s r< _autoscalezMplRenderer._autoscaleIs* + +Br,e44 +!!!---%*"   %DJ! 3 3 I " " $ $ $ $ $ % % 3 3r>r; Axes | intrcJt|tr |j|}|SN) isinstancerrrKs r<_get_axzMplRenderer._get_axTs$ b#   BB r>rC0ffffff?filledcpy.FillReturn fill_typeFillType | strcolorstralphafloatct|}||}t||}tj||dd|}||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 edgecolorslwrZTN)r rQr mcollectionsPathCollectionadd_collectionrE)r8rTrVr;rXrZpaths collections r<rTzMplRenderer.filledYsx*!++ \\"  #FI66!0 e1EKKK  *%%%!r>black皙?xry point_colorquad_as_tri_alphac 4||}|||\}}t||}|j|||j|jfi||dkrd|ddddf|ddddfz|ddddfz|ddddfzz} d|ddddf|ddddfz|ddddfz|ddddfzz} ||d<|jt j|ddddf| |ddddffdt j|ddddf| |ddddffdt j|ddddf| |ddddffdt j|ddddf| |ddddffdfi|||||||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. )rXrZrg?NrrZ)rno)rXrZmarkerraT) rQ _grid_as_2dr3plotTnpstackreshaperE) r8rirjr;rXrZrkrlr9xmidymids r<gridzMplRenderer.gridvs8\\"  1%%1!%E!?!?!?1ac13))&))) q 3B38qSbSz1Acrc122gJ>122qrr6JKD3B38qSbSz1Acrc122gJ>122qrr6JKD/F7O BG!CRC"H+tQqrr122vY788@@II!CRC"H+tQqrr122vY788@@II!ABBG*dAcrc122gJ788@@II!ABBG*dAcrc122gJ788@@II          " GGAq 5G K K K!r>?linescpy.LineReturn line_typeLineType | str linewidthct|}||}t||}tj|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``. r]r^TN)r rQrrbrcrdrE) r8r|r~r;rXrZrrerfs r<r|zMplRenderer.linessx.!++ \\"  "5)44!0 f9ESSS  *%%%!r>z'ArrayLike | np.ma.MaskedArray[Any, Any]ctj|}|tjjurdS||}|||\}}|||||d|dS)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"``. Nrp)c)rumagetmasknomaskrQrrrs)r8rirjrr;rXmasks r<rzMplRenderer.masks}"u}}Q 25<   F \\"  1%%1 $4#/////r>Ffilename transparentcf||j||dS)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)rLrsavefig)r8rrs r<savezMplRenderer.saves5  ( <<<< io.BytesIOc|tj}|j|d|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. png)formatr)rLioBytesIOrrseek)r8bufs r<save_to_bufferzMplRenderer.save_to_buffersN jll #e,,,   r>cT|tjdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)rLr4showrBs r<rzMplRenderer.shows#   r>titlec|r,||||dS|||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. )rXN)rQ set_title)r8rr;rXs r<rzMplRenderer.titles]  . LL   & &uE & : : : : : LL   & &u - - - - -r>green.1ffmt quad_as_tric ||}|||\}}tj|}|j\}} t |D]K} t | D]9} ||| | f|| | f|| | f|dd|d:L|rt |dz D]} t | dz D]} tj|| | dz| | dzf} tj|| | dz| | dzf} tj|| | dz| | dzf}|| | ||dd|ddSdS)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. centerThavarXclip_onrN)rQrrruasarrayshaperangetextmean)r8rirjrr;rXrrnynxjixxyyzzs r<z_valueszMplRenderer.z_values s6\\"  1%%1 JqMMBr 3 3A2YY 3 3!Q$1a4Qq!tWc*<*<h#T3333 3  *2a4[[ * *r!t**A1QqS5!AaC%<11B1QqS5!AaC%<11B1QqS5!AaC%<11BGGBr3MMh8SX$(**** * * * * *r>)rrrTNN)rrrrr r!r"rr#r$r%r&r'r()r'r()r;rMr'r)rrRrS) rTrUrVrWr;rMrXrYrZr[r'r()rrgrhNr)rirrjrr;rMrXrYrZr[rkr$rlr[r'r()rrRr{r)r|r}r~rr;rMrXrYrZr[rr[r'r()rrg) rirrjrrrr;rMrXrYr'r()F)rrYrrr'r()r'r)rN)rrYr;rMrXr$r'r()rrrF)rirrjrrrr;rMrXrYrrYrrr'r()__name__ __module__ __qualname____doc____annotations__r=rCrLrQrTrzr|rrrrrrr>r<rrs  LLL'-"-1     6!!!! % % % % """""B"&#$-"-"-"-"-"f"""""H 000000 = = = = =     .....*!**************r>rc,eZdZdZ d d fd ZxZS)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()``. rrrrrr r!r'r(c2ddddddd}t|||dd||jD]V}|d|d|g|gWd|_dS) Ng{Gz?gGz?)leftrighttopbottomwspacehspaceTAgg)r"r#r%F)superr=r set_xmargin set_ymargin set_xticks set_yticksr)r8rrr gridspecr; __class__s r<r=zMplTestRenderer.__init__<s     5'dEx    *  B NN3    NN3    MM"    MM"     r>)rrr)rrrrr r!r'r()rrrrr= __classcell__rs@r<rr6sW'- !!!!!!!!!!!r>rceZdZdZ d>d?fd Zd@dZ dAdBfd) Z dCdDfd0 Z dEdFd5Z dGdFd7Z dHdId=Z xZ S)JMplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rrTrrrr r!r"rr'r(cPt||||dSrO)rr=)r8rrr r"rs r<r=zMplDebugRenderer.__init__]s) w ;;;;;r>r;r line_startcpy.CoordinateArrayline_endrXrYrZr[ arrow_sizecd||zz}||z }|tjtj||z}tj|d|d f} tj||dz| z |zz ||dz|zz||dz| z|zz f} || dddf| dddfd||dS)Ng?rr-rrZ)rusqrtdotrrvrs) r8r;rrrXrZrmidalongrarrows r<_arrowzMplDebugRenderer._arrowfs:():% u--... E!HuQxi011 59u$j0 0 %)J& & 59u$j0 0    aaad U111a4[#eDDDDDr>rC1rSrRredrhrTrUrVrWrM line_color line_alpharkstart_point_colorc t|}t|||||||dS||}t ||t j}|t|D]\} } | t| dd| ddD]\} }| | |}||dddf|dddf||| dkrMt|}t|dz D]+}| |||||dz||| ,|t|D]\} } | tj | dt}d|| dddz <| | dd}d||<|| dddf|| dddf|d||| :|| dddf|| dddf|d| |dSdS) Nrnrrrr)dtypeFrp)r rrTrQr rChunkCombinedOffsetziprsrIrrruonesr)r8rTrVr;rXrZrrrkrrpointsoffsetsstartendxysnrr start_indicesrs r<rTzMplDebugRenderer.filledzs!++  vy"eU;;;  +"5 F \\"   83OPP  !#&< b b>"%gcrclGABBK"@"@bbJE3 s+CGGC1Is111a4yJjGQQQ!C''HH!&qsbbA KKCFC!Hj*V`aaaab  "#&< C C>wwr{$777&+WQRR[]#$0$+CRCLM*/D'111a4L&qqq!t T(:C;V`bbb%0GGF111a4L71m9TVY/zCCC # " C Cr>r{r|r}r~rrc :t|}t||||||| dkr|dS||}t ||t j} trttj | } | dkrP| D]M} tt| dz D]+} | || | | | dz||| ,N|| D]} d} t| }|i|| d| dd||d} | dd| ddkr#| dd| ddkr|dz}|| | |df| | |dfd||dSdS) Nrrr)rr)rrrprrn)r rr|rQr r SeparaterrcpyLineReturn_SeparaterrIrrs)r8r|r~r;rXrZrrkrrseparate_linesliner start_index end_indexrs r<r|zMplDebugRenderer.liness!++   eYE5)DDD   !4 F \\"  &ui9JKK  K!#"9>JJN   & R Rs4yy{++RRAKKDGT!A#YujQQQQR  "& . . II $0GGDJT C;LTYGZZZ"#KAwqzT"Xa[00T!WQZ48A;5N5N!Q [2A56[=RTU=U8VX[u.... # " . .r>rirrjrc h||}|||\}}tj|}|j\}}t |D]U}t |D]C} | ||zz} |||| f||| ft| dd|dDVdS)NrrTrrQrrrurrrrrY) r8rirjrr;rXrrrrquads r< point_numberszMplDebugRenderer.point_numberss\\"  1%%1 JqMMBr & &A2YY & &1R4x!Q$1a4#d))EQV $&&&& & & &r>bluec ||}|||\}}tj|}|j\}}t d|D]}t d|D]} | ||zz} ||dz |dz| dz | dzf} ||dz |dz| dz | dzf} || | t| dd|ddS)NrrTr) rQrrrurrrrrrY) r8rirjrr;rXrrrrrrxrys r< quad_numberszMplDebugRenderer.quad_numberss'\\"  1%%1 JqMMBq" d dA1b\\ d d1R4x1QqS!A#ac')*//111QqS!A#ac')*//11dCII(xu^bcccc  d d dr>Nr lower_level upper_level float | Nonec ||}|||\}}tj|}|j\}} t |D]m} t | D][} || | f} | | |krd} n | |krd} nd} ||| | f|| | ft| dd|d\ndS)NrrrrrTrr)r8rirjrrrr;rXrrrrrz_levels r<z_levelszMplDebugRenderer.z_levelss\\"  1%%1 JqMMBr & &A2YY & &q!tW*rK/?/?GG+%%GGG!Q$1a4#g,,6hV[ $&&&& & & &r>)rrrT) rrrrr r!r"rr'r()r;rrrrrrXrYrZr[rr[r'r()rrrSrRrSrRrrh)rTrUrVrWr;rMrXrYrZr[rrYrr[rkrYrrYrr[r'r()rrRr{rrRrrh)r|r}r~rr;rMrXrYrZr[rr[rkrYrrYrr[r'r()rr) rirrjrrrr;rMrXrYr'r()rr)Nrr)rirrjrrrrr[rrr;rMrXrYr'r() rrrrr=rrTr|rrrrrs@r<rrWs]'- <<<<<<<EEEE0!&3C3C3C3C3C3C3Cr!&&.&.&.&.&.&.&.Z &&&&&. ddddd2%)&&&&&&&&&r>r)+ __future__rcollections.abcrrtypingrrrmatplotlib.collections collectionsrbmatplotlib.pyplotpyplotr4numpyru contourpyrr contourpy.convertr r contourpy.enum_utilr r contourpy.util.mpl_utilrrcontourpy.util.rendererrmatplotlib.axesrmatplotlib.figurer numpy.typingrcontourpy._contourpy _contourpyrrrrrr>r<rs""""""$$$$$$ ++++++++++------((((((((;;;;;;;;::::::::KKKKKKKK,,,,,,'$$$$$$((((((&&&&&&&&&&&&Z*Z*Z*Z*Z*(Z*Z*Z*z!!!!!k!!!B}&}&}&}&}&{}&}&}&}&}&r>