Yc8-ddlZddlmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlZdd lmZmZGd d ZdS) N) export_png export_svgshow)get_screenshot_as_png)gridplot)Label) Category10)figure)filled_to_bokehlines_to_bokehcpeZdZdZddZdZdZd Zdd ZddZ ddZ d dZ d!dZ dZ dZd"dZd#dZdS)$ 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||_td|_dtj|z}||z}g|_|jrdnd}t |D]Y} t|} d| j_ d| j _ |j | |sd| _ d| j _ Zt|j|d|d|z|d|z |_dS) N dsvgcanvas)output_backendFrr )ncolstoolbar_locationwidthheight) _want_svgr _palettenpasarray_figuresranger xgridvisibleygridappendoutline_line_coloraxisr_layout) selfnrowsrfigsize show_framewant_svg total_sizenfiguresbackend_figs =lib/python3.11/site-packages/contourpy/util/bokeh_renderer.py__init__zBokehRenderer.__init__s!"2 G,,, ; >7%%xx ) )A000C %CI  %CI  M  % % % ))-&#(  MQ-5(A%1GIII ct|tr0|ddkr$t|dd}|j|}|S)NrCr ) isinstancestrintr)r)colorindexs r3_convert_colorzBokehRenderer._convert_color5sG eS ! ! )eAh#o )abb NNEM%(E r5cJt|tr |j|}|S)N)r8r:r )r)axs r3 _get_figurezBokehRenderer._get_figure;s% b#   #r"B r5ctj|}tj|}|jdkrtj||\}}||fS)Nr )rrndimmeshgrid)r)xys r3 _grid_as_2dzBokehRenderer._grid_as_2d@sH JqMM JqMM 6Q; %;q!$$DAq!t r5rC0ffffff?c||}||}t||\}}t|dkr||g|g||ddSdS)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; fill_alpha line_widthN)r@r=r lenmulti_polygons) r)filled fill_typer?r;alphar2rJrKs r3rPzBokehRenderer.filledGsr""##E** 33B r77Q; ^   2$B4u[\  ] ] ] ] ] ^ ^r5black皙?Nc ||}|||\}}d|Dd|jDz} d|Dd|jDz} t||} |j| | fi| |dkr2d|dd dd f|d ddd fz|dd d dfz|d dd dfzz} d|dd dd f|d ddd fz|dd d dfz|d dd dfzz} |jd t j|dd dd f| |d dd dffd Dd t j|dd dd f| |d dd dffd Dfi| |jdt j|dd d df| |d ddd ffd Ddt j|dd d df| |d ddd ffd Dfi| |A||||d|ddSdS)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. cg|]}|SrW.0rows r3 z&BokehRenderer.grid..q   cc   r5cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..q"6"6"633"6"6"6r5cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..rr\r5cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..rr^r5) line_colorrRrg?Nr cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..zaaaaaar5)r'cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..{rdr5cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..~rdr5cg|]}|SrWrWrXs r3r[z&BokehRenderer.grid..rdr5)rDrE fill_colorrarRsize) r@rFTdict multi_lineravelrstackcircle)r)rDrEr?r;rR point_colorquad_as_tri_alphar2rJrKkwargsxmidymids r3gridzBokehRenderer.grid[s(r""1%%1  Q   "6"6!#"6"6"6 6  Q   "6"6!#"6"6"6 6e444r2((((( q !CRC"H+!""crc' 2QssABBwZ?!ABBF)KLSSUUD!CRC"H+!""crc' 2QssABBwZ?!ABBF)KLSSUUD CNaa!CRC"H+*;*;*=*=tQqrr122vY__EVEV)W^_ ` ` `aaaaa!CRC"H+*;*;*=*=tQqrr122vY__EVEV)W^_ ` ` `aaa      CNaa!CRCG**:*:*<*tj|}|tjjurdS||}||}|||\}}||||||ddS)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)rirj)rmagetmasknomaskr@r=rFrp)r)rDrEzr?r;maskr2s r3rzBokehRenderer.masksu}}Q 25<   Fr""##E**1%%1 1T7AdGB ?????r5c|r|jD]}d|_d|_|jrt |j|dSt |j|dS)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)r background_fill_colorborder_fill_colorrrr(r)r)r transparentr2s r3savezBokehRenderer.savesp  -} - -,0)(,%% > 8 t|h 7 7 7 7 7 7 t|h 7 7 7 7 7 7r5ct|j}tj}||d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. png)rr(ioBytesIOr)r)imagebuffers r3save_to_bufferzBokehRenderer.save_to_buffers7 &dl33 65!!! r5c.t|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr()r)s r3rzBokehRenderer.shows T\r5c||}||_d|j_|!|||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)r@titlealignr= text_color)r)rr?r;r2s r3rzBokehRenderer.titlesWr"" "   >#'#6#6u#=#=CI  > >r5green.1fc <||}||}|||\}}tj|}|j\} } t |dd} t| D]R} t| D]@} |td|| | f|| | f|| | f|d| AS|rt| dz D]} t| dz D]} tj || | dz| | dzf}tj || | dz| | dzf}tj || | dz| | dzf}|td||||d| dSdS) 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)rDrEtextr NrW) r@r=rFrrshaperlr! add_layoutrmean)r)rDrErr?r;fmt quad_as_trir2nynxrsjixxyyzzs r3z_valueszBokehRenderer.z_valuess$r""##E**1%%1 JqMMB88TTTr _ _A2YY _ _u]qAw!AqD'1QT7C@R@R]]V\]]^^^^ _  T2a4[[ T Tr!tTTA1QqS5!AaC%<11B1QqS5!AaC%<11B1QqS5!AaC%<11BNN5#R2R--#R#R6#R#RSSSS T T T T Tr5)r r rTF)rrGrH)rrSrTNr)rrGrwr )rrS)F)rN)rrrF)__name__ __module__ __qualname____doc__r4r=r@rFrPrvrzrrrrrrrWr5r3rrs$ IIII,  ^^^^((b(b(b(bT]]]]0@@@@$8888*    >>>> !T!T!T!T!T!Tr5r)rbokeh.iorrrbokeh.io.exportr bokeh.layoutsr bokeh.modelsrbokeh.palettesr bokeh.plottingr numpyr bokeh_utilr r rrWr5r3rs 1111111111111111""""""%%%%%%!!!!!!77777777wTwTwTwTwTwTwTwTwTwTr5