Vf-NddlmZddlmZddlZddlmZmZmZddl 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*y)) 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) ArrayLikecteZdZUdZded<ded<ded< d dd Zdd Zdd Zdd Z d dd Z d ddZ d ddZ d d!dZ d"d#dZ d$dZddZd%d&dZ d' d(dZy)) 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_tightNc<|ddl}|j|t|ddd}|||d<ntd|d<tj||fi|\|_} | j |_|s"|jD]} | jd d|_ y) NrFT)figsizesqueezesharexsharey gridspec_kwequal)aspect subplot_kwoff) matplotlibusedictpltsubplotsrflattenraxisr) selfnrowsncolsr show_framebackendrr$kwargsaxesaxs d/var/www/html/software/conda/envs/catlas/lib/python3.12/site-packages/contourpy/util/mpl_renderer.py__init__zMplRenderer.__init__*s    NN7 #!%guTZ^!_  "$/F= !#'w#7F< ,,ue>v> 4\\^ jj   c\t|dr tj|jyy)Nr)hasattrr'closerr+s r3__del__zMplRenderer.__del__Es! 4 IIdii  !r5c|jD])}t|dds|jdd|_+|jr4t |jdkDr|j jyyy)N_need_autoscaleFT)tight)rgetattrautoscale_viewr<rlenr tight_layoutr+r2s r3 _autoscalezMplRenderer._autoscaleIso** +Br,e4!!!-%*" +   DJJ! 3 II " " $!4 r5cDt|tr|j|}|SN) isinstanceintrrCs r3_get_axzMplRenderer._get_axTs b# BB r5ct|}|j|}t||}tj||dd|}|j |d|_y)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 edgecolorslwalphaTN)r rIr mcollectionsPathCollectionadd_collectionr<)r+filled fill_typer2colorrPpaths collections r3rTzMplRenderer.filledYsZ*!+ \\" #FI6!00 e1EK  *%!r5c |j|}|j||\}}t||}|j|||j|jfi||dkDr[d|ddddf|ddddfz|ddddfz|ddddfzz} d|ddddf|ddddfz|ddddfz|ddddfzz} ||d<|jt j |ddddf| |ddddffjdt j |ddddf| |ddddffjdt j |ddddf| |ddddffjdt j |ddddf| |ddddffjdfi|||j||||d d d |_y) 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. )rVrPrg?Nr>rP)rZo)rVrPmarkerrOT) rI _grid_as_2dr&plotTnpstackreshaper<) r+xyr2rVrP point_colorquad_as_tri_alphar0xmidymids r3gridzMplRenderer.gridvs08\\" 1%1!%E!?1acc133)&) q 3B38qSbSz1Acrc12gJ>12qr6JKD3B38qSbSz1Acrc12gJ>12qr6JKD/F7O BGG!CRC"H+tQqr12vY78@@I!CRC"H+tQqr12vY78@@I!ABG*dAcrc12gJ78@@I!ABG*dAcrc12gJ78@@I       " GGAq 5G K!r5ct|}|j|}t||}tj|d|||}|j |d|_y)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``. rKrLTN)r rIrrQrRrSr<) r+lines line_typer2rVrP linewidthrWrXs r3rlzMplRenderer.linessZ.!+ \\" "5)4!00 f9ES  *%!r5ctjj|}|tjjury|j |}|j ||\}}|j ||||d|y)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"``. Nr\)c)ramagetmasknomaskrIr^r_)r+rdrezr2rVmasks r3ruzMplRenderer.masksi"uu}}Q 255<<   \\" 1%1 $4#/r5c^|j|jj||y)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)rDrsavefig)r+filenamerws r3savezMplRenderer.saves%  ( } t | D].} |j || | f|| | f|| | f|ddd|d0@|rt |dz D]} t | dz D]} tj|| | dz| | dzf} tj|| | dz| | dzf} tj|| | dz| | dzf}|j | | ||ddd|dyy)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. centerThavarVclip_onr>N)rIr^raasarrayshaperangetextmean)r+rdrertr2rVfmt quad_as_trinynxjixxyyzzs r3z_valueszMplRenderer.z_values s6\\" 1%1 JJqMBr 3A2Y 3!Q$1a4Qq!tWcU FOh#T3 3 3 2a4[ *r!t*A1QqS5!AaC%<1B1QqS5!AaC%<1B1QqS5!AaC%<1BGGBr3%pjh8SX$(* * * r5)r>r> rTNN)r,rHr-rHrtuple[float, float]r.rr/ str | Nonerzdict[str, Any] | NonereturnNone)rr)r2 Axes | intrr)rC0ffffff?) rTcpy.FillReturnrUFillType | strr2rrVstrrPfloatrr)rblack皙?Nr)rdrrerr2rrVrrPrrfrrgrrr)rr?r>)rlcpy.LineReturnrmLineType | strr2rrVrrPrrnrrr)rr) rdrrerrtz'ArrayLike | np.ma.MaskedArray[Any, Any]r2rrVrrr)F)ryrrwrrr)rz io.BytesIO)rN)rrr2rrVrrr)rgreenz.1fF)rdrrerrtrr2rrVrrrrrrr)__name__ __module__ __qualname____doc____annotations__r4r:rDrIrTrjrlrurzrrrrr5r3rrsJ   L'-"-1   %      +   6! % """"  "  "  " "B"&#$-" -" -"  -"  -"  -" -"!-" -"f""""  "  "  "" "H 0 0 0 3 0  0  0 00 =  .*!** ** **  **  **  ****** **r5rc:eZdZdZ d dfd 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()``. cddddddd}t||||dd||jD]F}|jd|j d|j g|j gHd|_y) Ng{Gz?gGz?)leftrighttopbottomwspacehspaceTAgg)r.r/rF)superr4r set_xmargin set_ymargin set_xticks set_yticksr)r+r,r-rgridspecr2 __class__s r3r4zMplTestRenderer.__init__<s    5'dEx  ** B NN3  NN3  MM"  MM"    !r5)r>r>r)r,rHr-rHrrrr)rrrrr4 __classcell__rs@r3rr6sC'- !!!% !  !!r5rcdeZdZdZ d d fd Z d dZ d d fd Z d dfd Z d ddZ d ddZ d ddZ xZ S)MplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. c*t|||||yrF)rr4)r+r,r-rr.rs r3r4zMplDebugRenderer.__init__]s w ;r5cld||zz}||z }|tjtj||z}tj|d|d f} tj||dz| z |zz ||dz|zz||dz| z|zz f} |j | dddf| dddfd||y)Ng?r>r-rprP)rasqrtdotrrbr_) r+r2 line_startline_endrVrP arrow_sizemidalongrarrows r3_arrowzMplDebugRenderer._arrowfs:():% u-.. E!HuQxi01 59u$j0 0 %)J& & 59u$j0 0   ad U1a4[#eDr5c t|}t| |||||||y|j|}t ||t j }|t|D]\} } |  t| dd| ddD]p\} }| | |}|j|dddf|dddf||| dkDs4t|}t|dz D]!}|j|||||dz||| #r|t|D]\} } |  tj| dt}d|| dddz <| | dd}d||<|j| dddf|| dddf|d||| k|j| dddf| dddf|d| |yy) NrZr>rrr)dtypeFr\)r rrTrIr rChunkCombinedOffsetzipr_rArrraonesr)r+rTrUr2rVrP line_color line_alpharfstart_point_colorrpointsoffsetsstartendxysnrru start_indicesrs r3rTzMplDebugRenderer.filledzs !+  vy"eU;  +"5  \\"  83O3OP  !#&< b>"%gcrlGABK"@bJE3 s+CGGC1Is1a4yJjGQ!C'H!&qsbA KKCFC!Hj*V`ab b b  "#&< C>wwwr{$7&+WQR[]#$0$+CRLM*/D'1a4L&q!t T(:C;V`b%0GGF1a4L71m9TVY/zC C #r5c ~t|}t| ||||||| dk(r|y|j|}t ||t j } trttj| } | dkDrB| D]=} tt| dz D]!} |j|| | | | dz||| #?|| D]z} d} t| }|E|j| d| dd||d} | dd| ddk(r| dd| ddk(r|dz}|j| | |df| | |dfd|||yy) Nrr>r)rr)rr>r\rrZ)r rrlrIr r SeparaterrcpyLineReturn_SeparaterrArr_)r+rlrmr2rVrPrnrfrrseparate_linesliner start_index end_indexrs r3rlzMplDebugRenderer.liness!+   eYE5)D  !4  \\" &ui9J9JK !#"9"9>JN  & Rs4y{+RAKKDGT!A#YujQR R  "& . I $0GGDJT C;LTYGZ"#KAwqzT"Xa[0T!WQZ48A;5N!Q [2A56[=RTU=U8VX[u. . #r5c >|j|}|j||\}}tj|}|j\}}t |D]E}t |D]5} | ||zz} |j ||| f||| ft| dd|d7Gy)NrrTrrIr^rarrrrr) r+rdrertr2rVrrrrquads r3 point_numberszMplDebugRenderer.point_numberss\\" 1%1 JJqMBr &A2Y &1R4x!Q$1a4#d)EQV $& & &r5c |j|}|j||\}}tj|}|j\}}t d|D]}t d|D]u} | ||zz} ||dz |dz| dz | dzfj } ||dz |dz| dz | dzfj } |j| | t| dd|dwy)Nr>rTr) rIr^rarrrrrr) r+rdrertr2rVrrrrrrhris r3 quad_numberszMplDebugRenderer.quad_numberss\\" 1%1 JJqMBq" dA1b\ d1R4x1QqS!A#ac')*//11QqS!A#ac')*//1dCI(xu^bc  d dr5c d|j|}|j||\}}tj|}|j\}} t |D]X} t | D]H} || | f} || |kDrd} n | |kDrd} nd} |j || | f|| | ft| dd|dJZy)Nrr>rrrTrr)r+rdrert lower_level upper_levelr2rVrrrrrz_levels r3z_levelszMplDebugRenderer.z_levelss\\" 1%1 JJqMBr &A2Y &q!tW*rK/?G+%GG!Q$1a4#g,6hV[ $& & &r5)r>r>rT) r,rHr-rHrrr.rrr)r2rrcpy.CoordinateArrayrrrVrrPrrrrr)rC1rrrrredr)rTrrUrr2rrVrrPrrrrrrfrrrrrrr)rrrr>rrr)rlrrmrr2rrVrrPrrnrrfrrrrrrr)rr) rdrrerrtrr2rrVrrr)rblue)Nrr)rdrrerrtrrrrz float | Noner2rrVrrr) rrrrr4rrTrlrrrrrs@r3rrWs'- <<<% <  <  <E E(E& E  E  EE E0!&3C3C"3C  3C  3C  3C3C3C3C3C3C 3Cr!&&.&."&.  &.  &.  &.&.&.&.&. &.Z & & &  &  &  & &. d d d  d  d  d d2%)& & &  &  & " & && &r5r)+ __future__rcollections.abcrr~typingrrrmatplotlib.collections collectionsrQmatplotlib.pyplotpyplotr'numpyra contourpyrr contourpy.convertr r contourpy.enum_utilr r contourpy.util.mpl_utilrrcontourpy.util.rendererrmatplotlib.axesrmatplotlib.figurer numpy.typingrcontourpy._contourpy _contourpyrrrrrr5r3rse"$ ++-(;:K,$(&&Z*(Z*z!k!B}&{}&r5