" f3ddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZddlZdd lmZmZdd lmZmZdd lmZerdd lmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%GddeZ&y)) annotationsN) TYPE_CHECKINGAny) export_png export_svgshow)get_screenshot_as_png)gridplot)Label) Category10)figure)FillTypeLineType)filled_to_bokehlines_to_bokeh)Renderer)GridPlot)Palette) ArrayLike) FillReturn LineReturncneZdZUded<ded<ded<ded< d dd Zdd Zdd Z d dd Z d ddZ d ddZ d d!dZ d"d#dZ d$dZ d%dZ d&d'dZ d( d)dZy )* BokehRendererz list[figure]_figuresr_layoutr_palettebool _want_svgc||_td|_dtj|z}||z}g|_|jrdnd}t |D]f} t|} d| j_ d| j_ |j j| |rOd| _ d| j_ ht|j |d|d|z|d|z |_y) N dsvgcanvas)output_backendFr)ncolstoolbar_locationwidthheight)rr rnpasarrayrranger xgridvisibleygridappendoutline_line_coloraxisr r) selfnrowsr&figsize show_framewant_svg total_sizenfiguresbackend_figs ?/usr/lib/python3/dist-packages/contourpy/util/bokeh_renderer.py__init__zBokehRenderer.__init__0s""2 G,, ; >>%xx )A0C %CII  %CII  MM  %)-&#(  ) MMQ-5(A%1GI cpt|tr%|ddk(rt|dd}|j|}|S)NrCr%) isinstancestrintr)r3colorindexs r=_convert_colorzBokehRenderer._convert_colorMs8 eS !eAh#oab NEMM%(E r?cDt|tr|j|}|S)N)rBrDr)r3axs r= _get_figurezBokehRenderer._get_figureSs b# r"B r?c|j|}|j|}t||\}}t|dkDr|j |g|g||dyy)aPlot filled contours on a single plot. 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 Bokeh Figure, optional): Which plot to use, 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 ``Category10`` palette. Default ``"C0"``. alpha (float, optional): Opacity to plot with, default ``0.7``. r)xsysrE fill_alpha line_widthN)rJrGrlenmulti_polygons) r3filled fill_typerIrEalphar<rLrMs r=rRzBokehRenderer.filledXsb*r"##E* 3B r7Q;   2$B4u[\  ] r?Nc |j|}|j||\}}|D cgc]} | c} |jD cgc]} | c} z} |D cgc]} | c} |jD cgc]} | c} z} t||} |j| | fi| |dkDrd|ddddf|ddddfz|ddddfz|ddddfzzj } d|ddddf|ddddfz|ddddfz|ddddfzzj }|jt j|ddddfj | |ddddfj fdD cgc]} | c} t j|ddddfj ||ddddfj fdD cgc]} | c} fi| |jt j|ddddfj | |ddddfj fdD cgc]} | c} t j|ddddfj ||ddddfj fdD cgc]} | c} fi| |4|j|j |j |d|d yycc} wcc} wcc} wcc} wcc} wcc} wcc} wcc} w) a Plot quad grid lines on a single plot. 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 Bokeh Figure, optional): Which plot to use, 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 ``Category10`` palette. Warning: ``quad_as_tri_alpha > 0`` plots all quads as though they are unmasked. ) line_colorrTrg?Nr%)r2)xy fill_colorrVrTsize) rJ _grid_as_2dTdict multi_lineravelr*stackcircle)r3rYrZrIrErT point_colorquad_as_tri_alphar<rowrLrMkwargsxmidymids r=gridzBokehRenderer.gridss:r"1%1 cc !##"633"6 6 cc !##"633"6 6e4r2(( q !CRC"H+!"crc' 2QssABwZ?!ABF)KLSSUD!CRC"H+!"crc' 2QssABwZ?!ABF)KLSSUD CNN "!CRC"H+*;*;*=tQqr12vY__EV)W^_ `aa "!CRC"H+*;*;*=tQqr12vY__EV)W^_ `aa   CNN "!CRCG**:*:*> t||h 7 t||h 7r?c|t|j}tj}|j |d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. png)r rioBytesIOr{)r3imagebuffers r=save_to_bufferzBokehRenderer.save_to_buffers0 &dll3 65! r?c.t|jy)z:Show plots in web browser, in usual Bokeh manner. N)rr)r3s r=rzBokehRenderer.shows T\\r?c|j|}||_d|j_|!|j||j_yy)aSet the title of a single plot. Args: title (str): Title text. ax (int or Bokeh Figure, optional): Which plot 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 ``Category10`` palette. Default ``None`` which is ``black``. centerN)rJtitlealignrG text_color)r3rrIrEr<s r=rzBokehRenderer.titlesHr" "   #'#6#6u#=CII  r?c |j|}|j|}|j||\}}tj|}|j \} } t |dd} t| D]D} t| D]4} |jtd || | f|| | f|| | f|dd| 6F|rt| dz D]} t| dz D]} tj|| | dz| | dzf}tj|| | dz| | dzf}tj|| | dz| | dzf}|jtd ||||dd| yy) aShow ``z`` values on a single plot. 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 Bokeh Figure, optional): Which plot to use, 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 ``Category10`` palette. 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`` centres of quads. Warning: ``quad_as_tri=True`` shows z-values for all quads, even if masked. rmiddle)r text_align text_baseline)rYrZtextr%N) rJrGr]r*r+shaper_r, add_layoutr mean)r3rYrZrtrIrEfmt quad_as_trir<nynxrgjixxyyzzs r=z_valueszBokehRenderer.z_valuess6r"##E*1%1 JJqMB88Tr _A2Y _u]qAw!AqD'1QT7C5PP&/]V\]^ _ _ 2a4[ Tr!tTA1QqS5!AaC%<1B1QqS5!AaC%<1B1QqS5!AaC%<1BNN5#R2RPvJ#R6#RS T T r?)r%r%) rTF) r4rDr&rDr5ztuple[int, int]r6rr7rreturnNone)rErCrrC)rI figure | intrr )rC0gffffff?) rRrrSrrIrrErCrTfloatrr)rblackg?Nr)rYrrZrrIrrErCrTrrd str | Nonererrr)rrg?r%)rmrrnrrIrrErCrTrrorrr)rr) rYrrZrrtz'ArrayLike | np.ma.MaskedArray[Any, Any]rIrrErCrr)F)rwrCrzrrr)rz io.BytesIO)rr)rN)rrCrIrrErrr)rgreenz.1fF)rYrrZrrtrrIrrErCrrCrrrr)__name__ __module__ __qualname____annotations__r>rGrJrRrjrmrur{rrrrrr?r=rrsj O$#) III! I  I  I I:  ^^^  ^  ^  ^ ^>"&#$1b 1b 1b  1b  1b  1b 1b!1b 1bn]]]  ]  ]  ]] ]J @ @ @ 3 @  @  @ @28*  >*!*T *T *T  *T  *T  *T*T*T *Tr?r)' __future__rr~typingrrbokeh.iorrrbokeh.io.exportr bokeh.layoutsr bokeh.models.annotations.labelsr bokeh.palettesr bokeh.plottingr numpyr* contourpyrrcontourpy.util.bokeh_utilrrcontourpy.util.rendererr bokeh.modelsrr numpy.typingrcontourpy._contourpyrrrrr?r=rsN" %111"1%!(E,%&&;dTHdTr?