o e\4@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZddlZdd lmZmZdd lmZmZdd lmZerpdd lmZddlm 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) FillReturn LineReturnc@seZdZUded<ded<ded<ded< dUdVddZdWddZdXdd Z ! " #dYdZd*d+Z ! , - . !d[d\d5d6Z ! " 7 d]d^d=d>Z ! ,d_d`dAdBZ dadbdEdFZ dcdHdIZ dddJdKZ dedfdMdNZ ! O P dgdhdSdTZd.S)i BokehRendererz list[figure]_figuresr_layoutr_palettebool _want_svg rTFnrowsintncolsfigsizetuple[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)ZdtypesvgZcanvas)Zoutput_backendFrr)r"Ztoolbar_locationwidthheight)rr rnpasarrayr!rranger ZxgridvisibleZygridappendZoutline_line_coloraxisr r) selfr r"r#r%r& total_sizeZnfiguresbackend_figr9MD:\Projects\ConvertPro\env\Lib\site-packages\contourpy/util/bokeh_renderer.py__init__0s&     zBokehRenderer.__init__colorstrcCs4t|tr|ddkrt|dd}|j|}|S)NrCr) isinstancer=r!r)r4r<indexr9r9r:_convert_colorMs zBokehRenderer._convert_colorax figure | intr cCst|tr |j|}|S)N)r?r!r)r4rBr9r9r: _get_figureSs  zBokehRenderer._get_figurerC0ffffff?filledr fill_typer alphafloatc CsN||}||}t||\}}t|dkr%|j|g|g||dddSdS)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)rDrArlenZmulti_polygons) r4rGrHrBr<rIr8rKrLr9r9r:rGXs   zBokehRenderer.filledblack皙?Nxry point_color str | Nonequad_as_tri_alphac Cs||}|||\}}dd|Ddd|jD} dd|Ddd|jD} t||d} |j| | fi| |dkr0d|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 Dfi| |jd dtj|ddd df| |d dddffd d Dddtj|ddd df| |d dddffd d Dfi| |durF|j|||d|dddSdS)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. cSg|]}|qSr9r9.0rowr9r9r: z&BokehRenderer.grid..cSrVr9r9rWr9r9r:rZr[) line_colorrIrg?NrcSrVr9r9rWr9r9r:rZr[)r3cSrVr9r9rWr9r9r:rZr[cSrVr9r9rWr9r9r:rZr[cSrVr9r9rWr9r9r:rZr[)rQrR fill_colorr\rIsize) rD _grid_as_2dTdict multi_lineZravelr.stackcircle)r4rQrRrBr<rIrSrUr8rKrLkwargsZxmidZymidr9r9r:gridss4   XXDDDD  zBokehRenderer.grid?linesr line_typer linewidthc CsJ||}||}t||\}} t|dkr#|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): 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)r\Z line_alpharMN)rDrArrNrd) r4rjrkrBr<rIrlr8rKrLr9r9r:rjs   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))r_r`)r.maZgetmaskZnomaskrDrArarf)r4rQrRrmrBr<maskr8r9r9r:rps    zBokehRenderer.maskfilename transparentcCsF|r|jD]}d|_d|_q|jrt|j|ddSt|j|ddS)aOSave plots to SVG or PNG file. Args: filename (str): Filename to save to. transparent (bool, optional): Whether background should be transparent, default ``False``. Warning: To output to SVG file, ``want_svg=True`` must have been passed to the constructor. N)rq)rZbackground_fill_colorZborder_fill_colorrrrr)r4rqrrr8r9r9r:saves zBokehRenderer.save io.BytesIOcCs"t|j}t}||d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)rrioBytesIOrs)r4imagebufferr9r9r:save_to_buffers  zBokehRenderer.save_to_buffercCst|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr)r4r9r9r:rszBokehRenderer.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)rDrzalignrA text_color)r4rzrBr<r8r9r9r:rzs zBokehRenderer.titlegreen.1ffmt quad_as_tric CsT||}||}|||\}}t|}|j\} } t|ddd} t| D]&} t| D]} |t d|| | f|| | f|| | f|d| q-q'|rt| dD]Q} t| dD]F} t || | d| | df}t || | d| | df}t || | d| | df}|t d||||d| q^qVdSdS) 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. r{middle)r}Z text_alignZ text_baseline)rQrRtextrNr9) rDrArar.r/shapercr0Z add_layoutr mean)r4rQrRrmrBr<rrr8nynxrgjixxyyzzr9r9r:z_valuess(      <"""$zBokehRenderer.z_values)rrrTF) r r!r"r!r#r$r%rr&rr'r()r<r=r'r=)rBrCr'r )rrErF) rGrrHr rBrCr<r=rIrJr'r()rrOrPNr)rQrrRrrBrCr<r=rIrJrSrTrUrJr'r()rrErir)rjrrkrrBrCr<r=rIrJrlrJr'r()rrO) rQrrRrrmrnrBrCr<r=r'r()F)rqr=rrrr'r()r'rt)r'r()rN)rzr=rBrCr<rTr'r()rr~rF)rQrrRrrmrrBrCr<r=rr=rrr'r()__name__ __module__ __qualname____annotations__r;rArDrGrhrjrprsryrrzrr9r9r9r:rsR      7 %   r)' __future__rrutypingrrZbokeh.iorrrZbokeh.io.exportrZ bokeh.layoutsr Zbokeh.models.annotations.labelsr Zbokeh.palettesr Zbokeh.plottingr numpyr.Z contourpyr rZcontourpy.util.bokeh_utilrrZcontourpy.util.rendererrZ bokeh.modelsrrZ numpy.typingrZcontourpy._contourpyrrrr9r9r9r:s&