YcKddlZddlmZddlmZddlZddl m Z m Z ddl m Z mZmZGddZGddeZGd d eZdS) N)FillTypeLineType)filled_to_mpl_pathslines_to_mpl_pathsmpl_codes_to_offsetscxeZdZdZ ddZdZdZd Zd ZddZ ddZ d dZ d!dZ d"dZ dZdZd#dZd$dZdS)% 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 TNcP|ddl}||t|ddd}|||d<ntd|d<tj||fi|\|_} | |_|s|jD]} | d d|_ dS) NrFT)figsizesqueezesharexsharey gridspec_kwequal)aspect subplot_kwoff) matplotlibusedictpltsubplots_figflatten_axesaxis _want_tight) selfnrowsncolsr show_framebackendrrkwargsaxesaxs ;lib/python3.11/site-packages/contourpy/util/mpl_renderer.py__init__zMplRenderer.__init__s  $     NN7 # # #guT$OOO  8$/F= ! !#'w#7#7#7F< ,ue>>v>> 4\\^^  j  c\t|drtj|jdSdS)Nr)hasattrrcloserr!s r)__del__zMplRenderer.__del__.s6 4  ! Idi  ! !r+c|jD]0}t|ddr|dd|_1|jr3t |jdkr|jdSdSdS)N_need_autoscaleFT)tightr)rgetattrautoscale_viewr2r lenr tight_layoutr!r(s r) _autoscalezMplRenderer._autoscale2s* + +Br,e44 +!!!---%*"   %DJ! 3 % I " " $ $ $ $ $ % % % %r+cJt|tr |j|}|SN) isinstanceintrr8s r)_get_axzMplRenderer._get_ax=s$ b#   BB r+ctj|}tj|}|jdkrtj||\}}||fS)Nr)npasarrayndimmeshgrid)r!xys r) _grid_as_2dzMplRenderer._grid_as_2dBsH JqMM JqMM 6Q; %;q!$$DAq!t r+rC0ffffff?c||}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): 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``. noner facecolors edgecolorslwalphaTN)r>r mcollectionsPathCollectionadd_collectionr2)r!filled fill_typer(colorrOpaths collections r)rSzMplRenderer.filledIsk\\"  #FI66!0 e1EKKK  *%%%!r+black皙?c 2||}|||\}}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 |_ 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. )rUrOrg?NrrO)r[o)rUrOmarkerT) r>rFrplotTr@stackreshapescatterr2) r!rDrEr(rUrO point_colorquad_as_tri_alphar&xmidymids r)gridzMplRenderer.grid^s&\\"  1%%1E///1ac13))&))) q 3B38qSbSz1Acrc122gJ>122qrr6JKD3B38qSbSz1Acrc122gJ>122qrr6JKD/F7O BG!CRC"H+tQqrr122vY788@@II!CRC"H+tQqrr122vY788@@II!ABBG*dAcrc122gJ788@@II!ABBG*dAcrc122gJ788@@II          I JJq!;eCJ H H H!r+?c||}t||}tj|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``. rJrKTN)r>rrPrQrRr2) r!lines line_typer(rUrO linewidthrVrWs r)rkzMplRenderer.linessk\\"  "5)44!0 f9ESSS  *%%%!r+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"``. Nr])c)r@magetmasknomaskr>rFr_)r!rDrEzr(rUmasks r)rtzMplRenderer.masksu}}Q 25<   F \\"  1%%1 $4#/////r+Fcf||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``. ) transparentN)r9rsavefig)r!filenamervs r)savezMplRenderer.saves5  ( <<<< set_title)r!titler(rUs r)rzMplRenderer.titles]  . LL   & &uE & : : : : : LL   & &u - - - - -r+green.1fc ||}|||\}}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. centerThavarUclip_onrN)r>rFr@rAshaperangetextmean)r!rDrErsr(rUfmt quad_as_trinynxjixxyyzzs r)z_valueszMplRenderer.z_valuess$\\"  1%%1 JqMMBr 3 3A2YY 3 3!Q$1a4Qq!tWc*<*<h#T3333 3  *2a4[[ * *r!t**A1QqS5!AaC%<11B1QqS5!AaC%<11B1QqS5!AaC%<11BGGBr3MMh8SX$(**** * * * * *r+)rrr TNN)rrGrH)rrXrYNr)rrGrir)rrX)F)rN)rrrF)__name__ __module__ __qualname____doc__r*r0r9r>rFrSrhrkrtryrrrrr+r)r r s   \`    *!!! % % % """"*$"$"$"$"L"""",0000" = = = =    . . . .!*!*!*!*!*!*r+r c$eZdZdZdfd 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()``. rr c2ddddddd}t|||dd||jD]V}|d|d|g|gWd|_dS) Ng{Gz?gGz?)leftrighttopbottomwspacehspaceTAgg)r$r%rF)superr*r set_xmargin set_ymargin set_xticks set_yticksr )r!r"r#rgridspecr( __class__s r)r*zMplTestRenderer.__init__s     5'dEx    *  B NN3    NN3    MM"    MM"     r+)rrr )rrrrr* __classcell__rs@r)rrsG !!!!!!!!!!r+rcbeZdZdZdfd ZdZ dfd Z dfd ZddZddZ ddZ xZ S)MplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rr TcPt||||dSr;)rr*)r!r"r#rr$rs r)r*zMplDebugRenderer.__init__s' w ;;;;;r+cd||zz}||z }|tjtj||z}tj|d|d f} tj||dz| z |zz ||dz|zz||dz| z|zz f} || dddf| dddfd||dS)Ng?rr-rorO)r@sqrtdotrArar_) r!r( line_startline_endrUrO arrow_sizemidalongrarrows r)_arrowzMplDebugRenderer._arrow s:():% u--... E!HuQxi011 59u$j0 0 %)J& & 59u$j0 0    aaad U111a4[#eDDDDDr+rC1rHrGredrYc t|||||||dS||}|tjkr|d} d|dD} n|tjkr&d|dD} d|dD} n|tjkr|d} |d} nj|tjkr&d|dD} d|dD} n4|tjkrdg} g} t|D]U\} }}|  | tj | |ddz } tj ||dd}| d |Dz } Vn|tj krg} g} t|D]\} }}|  tt|dz D]e}|||||dzdz}| | |d|d| ||dz fnt!d |d |t| | D]\} }t|dd|ddD]\}}| ||}||dddf|dddf|| | d krMt|}t|dz D]+}||||||dz||| ,|t| | D]\} }tj|dt(}d||dddz <| |dd}d||<|| dddf|| dddf|d|| | :|| dddf|| dddf|d| | dSdS)Nrc,g|]}t|Srr.0codess r) z+MplDebugRenderer.filled..7s!NNN5/66NNNr+rcg|]}||Sr;rrpointss r)rz+MplDebugRenderer.filled..9OOOVFO&OOOr+c0g|]}|t|Sr;rrs r)rz+MplDebugRenderer.filled..:s'ccc5QVc/66cccr+cg|]}||Sr;rrs r)rz+MplDebugRenderer.filled..?rr+cg|]}||Sr;r)roffsetss r)rz+MplDebugRenderer.filled..@sSSSwwS7SSSr+r[c,g|]}t|Srrrs r)rz+MplDebugRenderer.filled..Is!OOO 4U ; ;OOOr+zRendering FillType  not implementedrr)dtypeFr])rrSr>r OuterCodeChunkCombinedCode OuterOffsetChunkCombinedOffsetChunkCombinedCodeOffsetzipr@splitChunkCombinedOffsetOffsetrr6append RuntimeErrorr_ronesbool)r!rSrTr(rUrO line_color line_alphardstart_point_colorr all_points all_offsetsrr outer_offsetsrroffsstartendxysnrt start_indicesrs r)rSzMplDebugRenderer.filled,s vy"eU;;;  +  F \\"   * * RJNNF1INNNKK (4 4 ROOvayOOOJccF1IcccKK (. . RJ )KK (6 6 ROOvayOOOJSS&)SSSKK (: : RJK03V  P P,}bhv}QrT/BCCC  ad(;<<OOOOOO  P (< < RJK25v, 7 7.s=11!34477A"=#3M!A#4Fq4H#HID%%fT!WT"X-=&>???&&td1g~66667 7PYPPPQQ Q  b#&z;#?#? b b"%gcrclGABBK"@"@bbJE3 s+CGGC1Is111a4yJjGQQQ!C'bHH!&qsbbA KKCFC!Hj*V`aaaab  C#&z;#?#? C Cwwr{$777&+WQRR[]#$0$+CRCLM*/D'111a4L&qqq!t T(:C;V`bbb%CGGF111a4L71m9TVY/zCCC C C C Cr+ric  t||||||| dkr|dS||}|tjkr|} n |tjkr |d} n|tjkrqg} t|D]d\} } | ]t| } tt| dz D].}| | | || |dz/en|tj krbg} t|D]U\} } | Ntt| dz D].}| | | || |dz/Vntd|d| dkrP| D]M}tt|dz D]+}||||||dz||| ,N|| D]}d}t|}|i||d|dd|| d}|dd|d dkr#|dd|d dkr|dz}||||df|||dfd|| dSdS) NrrrzRendering LineType r)rr)rrr]rr[)rrkr>rSeparate SeparateCoderrrrr6rrrrr_)r!rkrlr(rUrOrmrdrr all_linesrrrrline start_index end_indexrs r)rkzMplDebugRenderer.linesrs:  eYE5)DDD     F \\"   ) ) RII (/ / RaII (4 4 RI!$e J J J2599G"3w<<>22JJ!(( 71Q3<0G)HIIII  J (6 6 RI#&; J JJ"3w<<>22JJ!(( 71Q3<0G)HIIII J PYPPPQQ Q   R! R Rs4yy{++RRAKKDGT!A#YujQQQQR  .! . . II $'GGDJT C;LTYGZZZ"#KAwqzT"Xa[0'T!WQZ48A;5N'!Q [2A56[=RTU=U8VX[u.... . . . .r+c h||}|||\}}tj|}|j\}}t |D]U}t |D]C} | ||zz} |||| f||| ft| dd|dDVdS)NrrTr)r>rFr@rArrrstr) r!rDrErsr(rUrrrrquads 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) r>rFr@rArrrrr) r!rDrErsr(rUrrrrrrfrgs r) quad_numberszMplDebugRenderer.quad_numberss% \\"  1%%1 JqMMBq" d dA1b\\ d d1R4x1QqS!A#ac')*//111QqS!A#ac')*//11dCII(xu^bcccc  d d dr+Nrc ~||}|||\}}tj|}|j\}} t |D]`} t | D]N} || | f} | | |krd} n | |krd} nd} ||| | f|| | f| dd|dOadS)NrrrrrTr)r>rFr@rArrr)r!rDrErs lower_level upper_levelr(rUrrrrrz_levels r)z_levelszMplDebugRenderer.z_levelss \\"  1%%1 JqMMBr & &A2YY & &q!tW rK/? GG+% GGG!Q$1a4'fQV $&&&& & & &r+)rrr T)rrrHrGrHrGrrY)rrGrirrGrrY)rr)rr)Nrr) rrrrr*rrSrkrrrrrs@r)rrs <<<<<< E E EQUUXDCDCDCDCDCDCL]a25,.,.,.,.,.,.\ & & & & d d d d&&&&&&&&r+r)r}matplotlib.collections collectionsrPmatplotlib.pyplotpyplotrnumpyr@ contourpyrrmpl_utilrrrr rrrr+r)r s  ------((((((((SSSSSSSSSSl*l*l*l*l*l*l*l*^!!!!!k!!!8o&o&o&o&o&{o&o&o&o&o&r+