o f5@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZddlZdd lmZmZdd lmZmZdd lmZer~dd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*GdddeZ+dS)) annotationsN) TYPE_CHECKINGAny) export_png export_svgshow)get_screenshot_as_png)gridplot)Label) Category10)figure) as_fill_type as_line_type)filled_to_bokehlines_to_bokeh)Renderer)GridPlot)Palette) ArrayLike) WebDriver)FillTypeLineType) FillReturn LineReturnc@seZdZUdZded<ded<ded<ded < dYdZddZd[ddZd\d d!Z " # $d]d^d+d,Z " - . / "d_d`d6d7Z " # 8 dadbd>d?Z " -dcdddBdCZ ded/dDdfdIdJZ d/dDdgdLdMZ dhdNdOZdidjdQdRZ " S T dkdldWdXZd/S)m BokehRendereraFUtility renderer using Bokeh 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 (assuming 100 dpi), default ``(9, 9)``. show_frame (bool, optional): Whether to show frame and axes ticks, default ``True``. want_svg (bool, optional): Whether output is required in SVG format or not, default ``False``. Warning: :class:`~contourpy.util.bokeh_renderer.BokehRenderer`, unlike :class:`~contourpy.util.mpl_renderer.MplRenderer`, needs to be told in advance if output to SVG format will be required later, otherwise it will assume PNG output. z list[figure]_figuresr_layoutr_palettebool _want_svg r"TFnrowsintncolsfigsizetuple[float, float] show_framewant_svgreturnNonec Cs||_td|_dtj|td}||}g|_|jrdnd}t|D]} t|d} d| j _ d| j _ |j | |sAd| _ d| j_ q#t|j|d|d||d |d |_dS) N d)ZdtypeZsvgZcanvas)Zoutput_backendFrr )r%Ztoolbar_locationwidthZheight)rr rnpasarrayr$rranger ZxgridZvisibleZygridappendZoutline_line_coloraxisr r) selfr#r%r&r(r)Z total_sizeZnfiguresZbackend_figr7=lib/python3.10/site-packages/contourpy/util/bokeh_renderer.py__init__2s&     zBokehRenderer.__init__colorstrcCs4t|tr|ddkrt|dd}|j|}|S)NrCr ) isinstancer;r$r)r4r:indexr7r7r8_convert_colorOs zBokehRenderer._convert_colorax figure | intr cCst|tr |j|}|S)N)r=r$r)r4r@r7r7r8 _get_figureUs  zBokehRenderer._get_figurerC0ffffff?filledr fill_typeFillType | stralphafloatc CsVt|}||}||}t||\}}t|dkr)|j|g|g||dddSdS)a Plot filled contours on a single plot. 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 a string equivalent. 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)xsysr:Z fill_alpha line_widthN)r rBr?rlenZmulti_polygons) r4rErFr@r:rHr6rJrKr7r7r8rEZs   zBokehRenderer.filledblack皙?Nxry point_color str | Nonequad_as_tri_alphac Cs^||}|||\}}t|t|j} t|t|j} ||d} |j| | fi| |dkrd|ddddf|ddddf|ddddf|ddddf} d|ddddf|ddddf|ddddf|ddddf} |jttj|ddddf| |ddddffddttj|ddddf| |ddddffddfi| |jttj|ddddf| |ddddffddttj|ddddf| |ddddffddfi| |dur-|j|||d|dd dSdS) 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_colorrHrg?Nr )r3)rPrQ fill_colorrUrHsize) rB _grid_as_2dlistTZ multi_lineZravelr/stackcircle)r4rPrQr@r:rHrRrTr6rJrKkwargsZxmidZymidr7r7r8gridvs4   XX>>>>  zBokehRenderer.grid?linesr line_typeLineType | str linewidthc CsNt|}||}||}t||\}} |dur%|j|| |||ddSdS)aPlot contour lines on a single plot. 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 a string equivalent. ax (int or Bokeh Figure, optional): Which plot to use, 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 ``Category10`` palette. Default ``"C0"``. alpha (float, optional): Opacity to plot lines with, default ``1.0``. linewidth (float, optional): Width of lines, default ``1``. Note: Assumes all lines are open line strips not closed line loops. N)rUZ line_alpharL)rrBr?rline) r4rbrcr@r:rHrer6rJrKr7r7r8rbs  zBokehRenderer.linesz'ArrayLike | np.ma.MaskedArray[Any, Any]cCs^tj|}|tjjurdS||}||}|||\}}|j|||||dddS)aPlot masked out grid points as circles 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 (masked array of shape (ny, nx): z-values. ax (int or Bokeh Figure, optional): Which plot to use, default ``0``. color (str, optional): Circle color, default ``"black"``. Nr,)rXrY)r/ZmaZgetmaskZnomaskrBr?rZr^)r4rPrQrgr@r:maskr6r7r7r8ris    zBokehRenderer.mask) webdriverfilename transparentrjWebDriver | NonecCsJ|r|jD]}d|_d|_q|jrt|j||ddSt|j||ddS)aSave plots to SVG or PNG file. Args: filename (str): Filename to save to. transparent (bool, optional): Whether background should be transparent, default ``False``. webdriver (WebDriver, optional): Selenium WebDriver instance to use to create the image. .. versionadded:: 1.1.1 Warning: To output to SVG file, ``want_svg=True`` must have been passed to the constructor. N)rkrj)rZbackground_fill_colorZborder_fill_colorrrrr)r4rkrlrjr6r7r7r8saves zBokehRenderer.save io.BytesIOcCs&t|j|d}t}||d|S)aSave plots to an ``io.BytesIO`` buffer. Args: webdriver (WebDriver, optional): Selenium WebDriver instance to use to create the image. .. versionadded:: 1.1.1 Return: BytesIO: PNG image buffer. )ZdriverZpng)rrioBytesIOrn)r4rjZimagebufferr7r7r8save_to_buffers  zBokehRenderer.save_to_buffercCst|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr)r4r7r7r8rszBokehRenderer.showtitlecCs6||}||_d|j_|dur|||j_dSdS)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)rBrtZalignr? text_color)r4rtr@r:r6r7r7r8rts zBokehRenderer.titlegreen.1ffmt quad_as_tric CsR||}||}|||\}}t|}|j\} } |ddd} t| D]&} t| D]} |td|| | f|| | f|| | f|d| q,q&|rt| dD]Q} t| dD]F} t || | d| | df}t || | d| | df}t || | d| | df}|td||||d| q]qUdSdS) 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. ruZmiddle)rvZ text_alignZ text_baseline)rPrQtextr Nr7) rBr?rZr/r0shaper1Z add_layoutr Zmean)r4rPrQrgr@r:ryrzr6nyZnxr_jiZxxZyyZzzr7r7r8z_values&s(       <"""$zBokehRenderer.z_values)r r r!TF) r#r$r%r$r&r'r(rr)rr*r+)r:r;r*r;)r@rAr*r )rrCrD) rErrFrGr@rAr:r;rHrIr*r+)rrNrONr)rPrrQrr@rAr:r;rHrIrRrSrTrIr*r+)rrCrar )rbrrcrdr@rAr:r;rHrIrerIr*r+)rrN) rPrrQrrgrhr@rAr:r;r*r+)F)rkr;rlrrjrmr*r+)rjrmr*ro)r*r+)rN)rtr;r@rAr:rSr*r+)rrwrxF)rPrrQrrgrr@rAr:r;ryr;rzrr*r+)__name__ __module__ __qualname____doc____annotations__r9r?rBrEr`rbrirnrsrrtrr7r7r7r8rsX       7 &   r),Z __future__rrptypingrrZbokeh.iorrrZbokeh.io.exportrZ bokeh.layoutsr Zbokeh.models.annotations.labelsr Zbokeh.palettesr Zbokeh.plottingr Znumpyr/Zcontourpy.enum_utilr rZcontourpy.util.bokeh_utilrrZcontourpy.util.rendererrZ bokeh.modelsrrZ numpy.typingrZ#selenium.webdriver.remote.webdriverrZ contourpyrrZcontourpy._contourpyrrrr7r7r7r8s*