a |=ic8-@sddlZddlmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlZdd lmZmZGd d d ZdS) N) export_png export_svgshow)get_screenshot_as_png)gridplot)Label) Category10)figure)filled_to_bokehlines_to_bokehc@seZdZdZd)ddZdd Zd d Zd d Zd*ddZd+ddZ d,ddZ d-ddZ d.ddZ dd Z d!d"Zd/d#d$Zd0d'd(ZdS)1 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. r  rTFc Cs||_td|_dt|}||}g|_|jr6dnd}t|D]<} t|d} d| j_ d| j _ |j | |sBd| _ d| j _ qBt|j|d|d||d|d |_dS) N dsvgZcanvas)Zoutput_backendFrr )ncolsZtoolbar_locationwidthheight) _want_svgr_palettenpasarray_figuresranger ZxgridvisibleZygridappendZoutline_line_coloraxisr_layout) selfZnrowsrZfigsizeZ show_frameZwant_svg total_sizeZnfiguresbackend_figr%n/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/contourpy/util/bokeh_renderer.py__init__s$     zBokehRenderer.__init__cCs4t|tr0|ddkr0t|dd}|j|}|S)NrCr ) isinstancestrintr)r colorindexr%r%r&_convert_color5s zBokehRenderer._convert_colorcCst|tr|j|}|S)N)r)r+r)r axr%r%r& _get_figure;s  zBokehRenderer._get_figurecCs6t|}t|}|jdkr.t||\}}||fS)Nr )rrndimZmeshgrid)r xyr%r%r& _grid_as_2d@s    zBokehRenderer._grid_as_2drC0ffffff?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)r0r.r lenZmulti_polygons) r filledZ fill_typer/r,alphar$r7r8r%r%r&r;Gs    zBokehRenderer.filledblack皙?Nc Cs||}|||\}}dd|Ddd|jD} dd|Ddd|jD} t||d} |j| | fi| |dkr`d|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| |dur|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|]}|qSr%r%.0rowr%r%r& qz&BokehRenderer.grid..cSsg|]}|qSr%r%r?r%r%r&rBrrC) line_colorr<rg?Nr cSsg|]}|qSr%r%r?r%r%r&rBzrC)rcSsg|]}|qSr%r%r?r%r%r&rB{rCcSsg|]}|qSr%r%r?r%r%r&rB~rCcSsg|]}|qSr%r%r?r%r%r&rBrC)r2r3 fill_colorrDr<size) r0r4Tdict multi_lineZravelrstackcircle)r r2r3r/r,r<Z point_colorZquad_as_tri_alphar$r7r8kwargsZxmidZymidr%r%r&grid[s2   XXDDDD zBokehRenderer.grid?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)rDZ line_alphar9N)r0r.r r:rK) r lines line_typer/r,r<Z linewidthr$r7r8r%r%r&rQs    zBokehRenderer.linescCs^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)rGrH)rmaZgetmaskZnomaskr0r.r4rM)r r2r3zr/r,maskr$r%r%r&rUs    zBokehRenderer.maskcCsD|r|jD]}d|_d|_q |jr2t|j|dnt|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)filename)rZbackground_fill_colorZborder_fill_colorrrrr)r rV transparentr$r%r%r&saves  zBokehRenderer.savecCs"t|j}t}||d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)rrioBytesIOrX)r imagebufferr%r%r&save_to_buffers  zBokehRenderer.save_to_buffercCst|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr)r r%r%r&rszBokehRenderer.showcCs2||}||_d|j_|dur.|||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)r0titlealignr. text_color)r r_r/r,r$r%r%r&r_s zBokehRenderer.titlegreen.1fc 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. r^middle)raZ text_alignZ text_baseline)r2r3textr N) r0r.r4rrshaperJrZ add_layoutrmean)r r2r3rTr/r,fmtZ quad_as_trir$nynxrNjixxyyzzr%r%r&z_valuess       >"""zBokehRenderer.z_values)r r rTF)rr5r6)rr=r>Nr)rr5rPr )rr=)F)rN)rrbrcF)__name__ __module__ __qualname____doc__r'r.r0r4r;rOrQrUrXr]rr_rqr%r%r%r&r s   *     r )rYZbokeh.iorrrZbokeh.io.exportrZ bokeh.layoutsrZ bokeh.modelsrZbokeh.palettesrZbokeh.plottingr numpyrZ bokeh_utilr r r r%r%r%r&s