gyN:ddlmZddlZddlmZddlmZmZmZddl m Z ddl m ZddlZddlmZmZddlmZmZddlmZmZddlmZmZdd lmZerdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z)GddeZ*Gdde*Z+Gdde*Z,dS)) annotationsN)pairwise) TYPE_CHECKINGAnycast)FillTypeLineType)convert_filled convert_lines) as_fill_type as_line_type)filled_to_mpl_pathslines_to_mpl_paths)Renderer)Sequence)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] | NonereturnNonec |ddl}|||dddd}|||d<nddi|d<tj||fi|\|_} | |_|s|jD]} | d d|_dS) NrFT)r!squeezesharexshareyr&aspectequal subplot_kwoff) matplotlibusepltsubplotsrflattenraxisr) selfrr r!r#r$r&mplkwargsaxesaxs Q/granary/edev/quantum/lib/python3.11/site-packages/contourpy/util/mpl_renderer.py__init__zMplRenderer.__init__,s   $ $ $ $ GGG   -4,0D"B"B  "$/F= ! !$,g#6F< ,ue>>v>> 4\\^^  j  c\t|drtj|jdSdS)Nr)hasattrr4closerr8s r=__del__zMplRenderer.__del__Hs6 4  ! Idi  ! !r?c|jD]0}t|ddr|dd|_1|jr3t |jdkr|jdSdSdS)N_need_autoscaleFT)tightr)rgetattrautoscale_viewrFrlenr tight_layoutr8r<s r= _autoscalezMplRenderer._autoscaleLs* + +Br,e44 +!!!---%*"   %DJ! 3 3 I " " $ $ $ $ $ % % 3 3r?r< Axes | intrcJt|tr |j|}|SN) isinstancerrrLs r=_get_axzMplRenderer._get_axWs$ b#   BB r?rC0ffffff?filledcpy.FillReturn fill_typeFillType | strcolorstralphafloatct|}||}t||}tj||dd|}||d|_dS)a Plot filled contours on a single Axes. Args: filled (sequence of arrays): Filled contour data as returned by :meth:`~.ContourGenerator.filled`. fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` data as returned by :attr:`~.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 edgecolorslwr[TN)r rRr mcollectionsPathCollectionadd_collectionrF)r8rUrWr<rYr[paths collections r=rUzMplRenderer.filled\sx*!++ \\"  #FI66!0 e1EKKK  *%%%!r?black皙?xry point_colorquad_as_tri_alphac ||}|||\}}||d}|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. )rYr[rg?Nrr[)roo)rYr[markerrbT)rR _grid_as_2dplotTnpstackreshaperF) r8rjrkr<rYr[rlrmr:xmidymids r=gridzMplRenderer.gridys8\\"  1%%1+05!A!A1ac13))&))) 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)aWPlot contour lines on a single Axes. Args: lines (sequence of arrays): Contour line data as returned by :meth:`~.ContourGenerator.lines`. line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` data as returned by :attr:`~.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 rRrrcrdrerF) r8r}rr<rYr[rrfrgs 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"``. Nrq)c)rvmagetmasknomaskrRrsrt)r8rjrkrr<rYmasks 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)rMrsavefig)r8rrs r=savezMplRenderer.saves5  ( <<<<rDrMrRrUr{r}rrrrrrr?r=rrs  LLL'-"-1     8!!!! % % % % """""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()``. rrrrr r!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)r8rr r!gridspecr< __class__s r=r>zMplTestRenderer.__init__?s     5'dEx    *  B NN3    NN3    MM"    MM"     r?)rrr)rrr rr!r"r(r))rrrrr> __classcell__rs@r=rr9sW'- !!!!!!!!!!!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. rrTrrr r!r"r#rr(r)cPt||||dSrP)rr>)r8rr r!r#rs r=r>zMplDebugRenderer.__init__`s) w ;;;;;r?r<r line_startcpy.CoordinateArrayline_endrYrZr[r\ 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-rr[)rvsqrtdotrrwrt) r8r<rrrYr[rmidalongrarrows r=_arrowzMplDebugRenderer._arrowis:():% u--... E!HuQxi011 59u$j0 0 %)J& & 59u$j0 0    aaad U111a4[#eDDDDDr?rC1rTrSredrirUrVrWrXrN line_color line_alpharlstart_point_colorc t|}t|||||||dS||}t ||t j}|t|D]\} } | t| D]\} }| | |}| |dddf|dddf||| dkrMt|}t|dz D]+}| |||||dz||| ,|t|D]\} } | tj| dt}d|| dddz <| | dd}d||<| | dddf|| dddf|d||| :| | dddf|| dddf|d| |dSdS) Nrrrrro)dtypeFrq)r rrUrRr rChunkCombinedOffsetziprrtrJrrrvonesr)r8rUrWr<rYr[rrrlrrpointsoffsetsstartendxysnrr start_indicesrs r=rUzMplDebugRenderer.filled}sx!++  vy"eU;;;  +"5 F \\"   83OPP  !#&< b b>"*7"3"3bbJE3 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~rrrc :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)rrrqrro)r rr}rRr r SeparaterrcpyLineReturn_SeparaterrJrrt)r8r}rr<rYr[rrlrrseparate_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?rjrrkrc h||}|||\}}tj|}|j\}}t |D]U}t |D]C} | ||zz} |||| f||| ft| dd|dDVdS)NrrTrrRrsrvrrrrrZ) r8rjrkrr<rYrrrrquads 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) rRrsrvrrrrrrZ) r8rjrkrr<rYrrrrrryrzs 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)r8rjrkrrrr<rYrrrrrz_levels r=z_levelszMplDebugRenderer.z_levelss\\"  1%%1 JqMMBr & &A2YY & &q!tW*rK/?/?GG+%%GGG!Q$1a4#g,,6hV[ $&&&& & & &r?)rrrT) rrr rr!r"r#rr(r))r<rrrrrrYrZr[r\rr\r(r))rrrTrSrTrSrri)rUrVrWrXr<rNrYrZr[r\rrZrr\rlrZrrZrr\r(r))rrSr|rrSrri)r}r~rrr<rNrYrZr[r\rr\rlrZrrZrr\r(r))rr) rjrrkrrrr<rNrYrZr(r))rr)Nrr)rjrrkrrrrr\rrr<rNrYrZr(r)) rrrrr>rrUr}rrrrrs@r=rrZs]'- <<<<<<<EEEE0!&3C3C3C3C3C3C3Cr!&&.&.&.&.&.&.&.Z &&&&&. ddddd2%)&&&&&&&&&r?r)- __future__rr itertoolsrtypingrrrmatplotlib.collections collectionsrcmatplotlib.pyplotpyplotr4numpyrv contourpyrr contourpy.convertr r contourpy.enum_utilr r contourpy.util.mpl_utilrrcontourpy.util.rendererrcollections.abcrmatplotlib.axesrmatplotlib.figurer numpy.typingrcontourpy._contourpy _contourpyrrrrrr?r=rs"""""" ++++++++++------((((((((;;;;;;;;::::::::KKKKKKKK,,,,,,'(((((($$$$$$((((((&&&&&&&&&&&&[*[*[*[*[*([*[*[*|!!!!!k!!!B}&}&}&}&}&{}&}&}&}&}&r?