j 6*ddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZddlZdd lmZmZdd lmZmZdd lmZer.dd lmZddl m!Z!ddlm"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,GddeZ-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)OutputBackendType)GridPlot)Palette) ArrayLike) WebDriver)FillTypeLineType) FillReturn LineReturnceZdZUdZded<ded<ded<ded < dMdNdZdOdZdPdZ dQdRd(Z dSdTd2Z dUdVd9Z dWdXd<Z dYd+d=dZdBZ d+d=d[dDZ d\dEZd]d^dGZ d_d`dLZd+S)a BokehRendereraUtility 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:`~.BokehRenderer`, unlike :class:`~.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||_td|_dtj|t 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 d)dtypesvgcanvas)output_backendFrr#)r(toolbar_locationwidthheight)r"r r npasarrayr'rranger xgridvisibleygridappendoutline_line_coloraxisr r) selfr&r(r)r+r, total_sizenfiguresbackend_figs /Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/contourpy/util/bokeh_renderer.py__init__zBokehRenderer.__init__2s""2 G37777 ; .2n%JUU(x ) )A000C %CI  %CI  M  % % % ))-&#(  MQ-5(A%1GIII colorstrct|tr0|ddkr$t|dd}|j|}|S)NrCr#) isinstancerLr'r )rBrKindexs rH_convert_colorzBokehRenderer._convert_colorOsD eS ! ! )eAh#ooabb NNEM%(E rJax figure | intr cJt|tr |j|}|S)N)rOr'r)rBrRs rH _get_figurezBokehRenderer._get_figureUs% b#   #r"B rJrC0ffffff?filledr fill_typeFillType | stralphafloatct|}||}||}t||\}}t |dkr||g|g||ddSdS)aPlot filled contours on a single plot. Args: filled (sequence of arrays): Filled contour data as returned by :meth:`~.ContourGenerator.filled`. fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` data as returned by :attr:`~.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)xsysrK fill_alpha line_widthN)rrUrQrlenmulti_polygons) rBrXrYrRrKr[rGr^r_s rHrXzBokehRenderer.filledZs*!++ r""##E** 33B r77Q;;   2$B4u[\  ] ] ] ] ] ;rJblack皙?Nxry point_color str | Nonequad_as_tri_alphac ,||}|||\}}t|t|jz} t|t|jz} ||d} |j| | fi| |dkr>d|ddddf|ddddfz|ddddfz|ddddfzz} d|ddddf|ddddfz|ddddfz|ddddfzz} |jtt j|ddddf| |ddddffdtt j|ddddf| |ddddffdfi| |jtt j|ddddf| |ddddffdtt j|ddddf| |ddddffdfi| |B||||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_colorr[rg?Nr#)rAcircle)rfrg fill_colorrlr[markersize) rU _grid_as_2dlistT multi_lineravelr9stackscatter)rBrfrgrRrKr[rhrjrGr^r_kwargsxmidymids rHgridzBokehRenderer.gridvsd:r""1%%1 !WWtACyy  !WWtACyy  %66r2((((( q !CRC"H+!""crc' 2QssABBwZ?!ABBF)KLSSUUD!CRC"H+!""crc' 2QssABBwZ?!ABBF)KLSSUUD CNRXq"crc{0022D!ABBF)//:K:KLSTUUUVVRXq"crc{0022D!ABBF)//:K:KLSTUUUVV      CNRXq"abbz//114122ss79I9I9K9KLSTUUUVVRXq"abbz//114122ss79I9I9K9KLSTUUUVV       " KK''))qwwyyUtSXa  ) ) ) ) ) # "rJ?linesr line_typeLineType | str linewidthct|}||}||}t||\}} | | J||| |||dSdS)aPlot contour lines on a single plot. Args: lines (sequence of arrays): Contour line data as returned by :meth:`~.ContourGenerator.lines`. line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` data as returned by :attr:`~.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)rl line_alphara)rrUrQrline) rBrrrRrKr[rrGr^r_s rHrzBokehRenderer.liness4!++ r""##E**y11B >>>> HHR%IH V V V V V >rJz'ArrayLike | np.ma.MaskedArray[Any, Any]c@tj|}|tjjurdS||}||}|||\}}||||||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"``. Nrnr0)rprqrr)r9magetmasknomaskrUrQrsry)rBrfrgrrRrKmaskrGs rHrzBokehRenderer.masks"u}}Q 25<   Fr""##E**1%%1 AdGQtWxb QQQQQrJ) webdriverfilename transparentrWebDriver | Nonec|r|jD]}d|_d|_|jrt |j||dSt |j||dS)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)rr)rbackground_fill_colorborder_fill_colorr"rrr)rBrrrrGs rHsavezBokehRenderer.savesu(  -} - -,0)(,%% > M t|h) L L L L L L t|h) L L L L L LrJ io.BytesIOct|j|}tj}||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. )driverpng)r rioBytesIOr)rBrimagebuffers rHsave_to_bufferzBokehRenderer.save_to_buffers<&dl9EEE 65!!! rJc.t|jdS)z:Show plots in web browser, in usual Bokeh manner. N)rr)rBs rHrzBokehRenderer.shows T\rJtitlec||}||_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)rUralignrQ text_color)rBrrRrKrGs rHrzBokehRenderer.titlesSr"" "   #'#6#6u#=#=CI   rJgreen.1ffmt quad_as_tric (||}||}|||\}}tj|}|j\} } |ddd} t | D]T} t | D]B} td|| | f|| | f|| | f|d| }||CU|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)rfrgtextr#N) rUrQrsr9r:shaper;r add_layoutmean)rBrfrgrrRrKrrrGnynxrzjilabelxxyyzzs rHz_valueszBokehRenderer.z_values(s6r""##E**1%%1 JqMMB %XPXYYr & &A2YY & &V!Q$1QT7AadGS9K9KVVvVVu%%%% &  T2a4[[ T Tr!tTTA1QqS5!AaC%<11B1QqS5!AaC%<11B1QqS5!AaC%<11BNN5#R2R--#R#R6#R#RSSSS T T T T TrJ)r#r#r$TF) r&r'r(r'r)r*r+r!r,r!r-r.)rKrLr-rL)rRrSr-r )rrVrW) rXrrYrZrRrSrKrLr[r\r-r.)rrdreNr)rfrrgrrRrSrKrLr[r\rhrirjr\r-r.)rrVr~r#)rrrrrRrSrKrLr[r\rr\r-r.)rrd) rfrrgrrrrRrSrKrLr-r.)F)rrLrr!rrr-r.)rrr-r)r-r.)rN)rrLrRrSrKrir-r.)rrrF)rfrrgrrrrRrSrKrLrrLrr!r-r.)__name__ __module__ __qualname____doc____annotations__rIrQrUrXr}rrrrrrrrrJrHrrsOOO'- IIIII:  ^^^^^@"&#$2)2)2)2)2)p W W W W WN RRRRR8"M '+ MMMMMM<?C  >>>>>*!+T+T+T+T+T+T+TrJr). __future__rrtypingrrbokeh.iorrrbokeh.io.exportr bokeh.layoutsr bokeh.models.annotations.labelsr bokeh.palettesr bokeh.plottingr numpyr9contourpy.enum_utilrrcontourpy.util.bokeh_utilrrcontourpy.util.rendererrbokeh.core.enumsr bokeh.modelsrr numpy.typingr#selenium.webdriver.remote.webdriverr contourpyrrcontourpy._contourpyrrrrrJrHrs"""""" %%%%%%%%1111111111111111""""""111111%%%%%%!!!!!!::::::::EEEEEEEE,,,,,,<222222%%%%%%&&&&&&&&&&&&======,,,,,,,,;;;;;;;;vTvTvTvTvTHvTvTvTvTvTrJ