U i2e4@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZddlZdd lmZmZdd lmZmZdd lmZerdd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'GdddeZ(dS)) 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) WebDriver) FillReturn LineReturnc @sJeZdZUded<ded<ded<ded<dKd d dddddddZdddddZdddddZdLdd ddd!dd"d#d$ZdMd(d(ddd!d)d!dd*d+d,ZdNd.d/ddd!d!dd0d1d2Z dOd(d(d3dddd4d5d6Z dPd'd7ddd8dd9d:d;Z d'd7d8dd?Z dd@dAdBZ dQddd)ddCdDdEZdRd(d(d(ddddddHdIdJZd'S)S BokehRendererz list[figure]_figuresr_layoutr_palettebool _want_svg r TFintztuple[float, float]None)nrowsncolsfigsize show_framewant_svgreturnc Cs||_td|_dtj|td}||}g|_|jr:dnd}t|D]<} t|d} d| j _ d| j _ |j | |sFd| _ d| j_ qFt|j|d|d||d |d |_dS) N d)ZdtypesvgZcanvas)Zoutput_backendFrr)r$Ztoolbar_locationwidthheight)rr rnpasarrayr!rranger ZxgridZvisibleZygridappendZoutline_line_coloraxisr r) selfr#r$r%r&r'Z total_sizeZnfiguresbackend_figr7A/tmp/pip-unpacked-wheel-_24pu9r8/contourpy/util/bokeh_renderer.py__init__1s*      zBokehRenderer.__init__str)colorr(cCs4t|tr0|ddkr0t|dd}|j|}|S)NrCr) isinstancer:r!r)r3r;indexr7r7r8_convert_colorNs zBokehRenderer._convert_colorz figure | intr )axr(cCst|tr|j|}|S)N)r=r!r)r3r@r7r7r8 _get_figureTs  zBokehRenderer._get_figurerC0ffffff?rr float)filled fill_typer@r;alphar(c CsJ||}||}t||\}}t|dkrF|j|g|g||dddS)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)xsysr;Z fill_alpha line_widthN)rAr?rlenZmulti_polygons) r3rErFr@r;rGr6rHrIr7r7r8rEYs    zBokehRenderer.filledblack皙?Nrz str | None)xyr@r;rG point_colorquad_as_tri_alphar(c Cs||}|||\}}dd|Ddd|jD} dd|Ddd|jD} t||d} |j| | f| |dkrTd|ddddf|d dddf|ddd df|d dd df} d|ddddf|d dddf|ddd df|d dd df} |jd dtj|ddddf| |d dd dffd d Dd dtj|ddddf| |d dd dffd d Df| |jd dtj|ddd df| |d dddffd d Dddtj|ddd df| |d dddffd d Df| |dk r||j|||d|dddS)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. cSsg|]}|qSr7r7.0rowr7r7r8 sz&BokehRenderer.grid..cSsg|]}|qSr7r7rRr7r7r8rUs) line_colorrGrg?NrcSsg|]}|qSr7r7rRr7r7r8rUs)r2cSsg|]}|qSr7r7rRr7r7r8rUscSsg|]}|qSr7r7rRr7r7r8rUscSsg|]}|qSr7r7rRr7r7r8rUs)rNrO fill_colorrVrGsize) rA _grid_as_2dTdict multi_lineZravelr.stackcircle)r3rNrOr@r;rGrPrQr6rHrIkwargsZxmidZymidr7r7r8gridts<   XXDDDD zBokehRenderer.grid?rr)lines line_typer@r;rG linewidthr(c CsF||}||}t||\}} t|dkrB|j|| |||ddS)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): Type of ``lines`` data, as returned by :attr:`~contourpy.ContourGenerator.line_type`. 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. r)rVZ line_alpharJN)rAr?rrKr^) r3rdrer@r;rGrfr6rHrIr7r7r8rds    zBokehRenderer.linesz'ArrayLike | np.ma.MaskedArray[Any, Any])rNrOzr@r;r(cCs^tj|}|tjjkrdS||}||}|||\}}|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))rYrZ)r.maZgetmaskZnomaskrAr?r[r`)r3rNrOrgr@r;maskr6r7r7r8ris    zBokehRenderer.mask) webdriverzWebDriver | None)filename transparentrjr(cCsH|r|jD]}d|_d|_q |jr4t|j||dnt|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. Warning: To output to SVG file, ``want_svg=True`` must have been passed to the constructor. N)rkrj)rZbackground_fill_colorZborder_fill_colorrrrr)r3rkrlrjr6r7r7r8saves zBokehRenderer.savez io.BytesIO)rjr(cCs&t|j|d}t}||d|S)zSave plots to an ``io.BytesIO`` buffer. Args: webdriver (WebDriver, optional): Selenium WebDriver instance to use to create the image. Return: BytesIO: PNG image buffer. )ZdriverZpng)rrioBytesIOrm)r3rjimagebufferr7r7r8save_to_buffers  zBokehRenderer.save_to_buffer)r(cCst|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr)r3r7r7r8r szBokehRenderer.show)titler@r;r(cCs2||}||_d|j_|dk r.|||j_dS)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)rArsZalignr? text_color)r3rsr@r;r6r7r7r8rss zBokehRenderer.titlegreen.1f)rNrOrgr@r;fmt quad_as_trir(c CsR||}||}|||\}}t|}|j\} } t|ddd} t| D]L} t| D]>} |t f|| | f|| | f|| | f|d| qZqN|rNt| dD]} t| dD]} t || | d| | df}t || | d| | df}t || | d| | df}|t f||||d| qqdS)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. rtZmiddle)ruZ text_alignZ text_baseline)rNrOtextrN) rAr?r[r.r/shaper]r0Z add_layoutr Zmean)r3rNrOrgr@r;rxryr6nyZnxrajiZxxyyzzr7r7r8z_valuess       >"""zBokehRenderer.z_values)rrrTF)rrBrC)rrLrMNr)rrBrcr)rrL)F)rN)rrvrwF)__name__ __module__ __qualname____annotations__r9r?rArErbrdrirmrrrrsrr7r7r7r8rsT  7%r)) __future__rrntypingrrZbokeh.iorrrZbokeh.io.exportrZ bokeh.layoutsr Zbokeh.models.annotations.labelsr Zbokeh.palettesr Zbokeh.plottingr Znumpyr.Z contourpyr rZcontourpy.util.bokeh_utilrrZcontourpy.util.rendererrZ bokeh.modelsrrZ numpy.typingrZ#selenium.webdriver.remote.webdriverrZcontourpy._contourpyrrrr7r7r7r8s&