d-9UddlmZddlZejeZddlmZmZm Z ddl Z ddl m Z mZmZddlmZmZddlmZmZddlmZdd lmZdd lmZdd lmZmZer@dd lm Z dd l!m"Z"ddlm#Z#m$Z$ddl%m&Z&e e&ee&fZ'de(d<e e'e#e$e'fZ)de(d<dZ*edGddZ+edGddZ,edGddZ-edGdde+Z.edGdde,Z/edGd d!Z0 d=ddd"d>d-Z1 d=d?d/Z2edGd0d1Z3edGd2d3Z4d@d7Z5dAd9Z6dBd:Z7dCd;Z8dDd<Z9dS)E) annotationsN) TYPE_CHECKINGSequenceUnion) FillProps HatchProps LineProps) MultiLine MultiPolygons)ContourRenderer GlyphRenderer)ColumnDataSource)linear_palette)_process_sequence_literals) dataclassentries) ArrayLike) TypeAlias)PalettePaletteCollection) ColorLiker ContourColorContourColorOrPalette) contour_data from_contourT)frozenc(eZdZUdZded<ded<dS) FillCoordszR Coordinates for all filled polygons over a whole sequence of contour levels. zlist[list[list[np.ndarray]]]xsysN__name__ __module__ __qualname____doc____annotations__6lib/python3.11/site-packages/bokeh/plotting/contour.pyrr:s4$$$$$$$$$$r)rc(eZdZUdZded<ded<dS) LineCoordszP Coordinates for all contour lines over a whole sequence of contour levels. zlist[np.ndarray]r r!Nr"r(r)r*r,r,As4r)r,c(eZdZUdZded<ded<dS) ContourCoordszP Combined filled and line contours over a whole sequence of contour levels. zFillCoords | None fill_coordszLineCoords | None line_coordsNr"r(r)r*r.r.Hs4""""""""""r)r.c.eZdZUdZded<ded<dZdS)FillDatazY Complete geometry data for filled polygons over a whole sequence of contour levels. r lower_levels upper_levelsc:tt|SNdictrselfs r*asdictzFillData.asdictVGDMM"""r)Nr#r$r%r&r'r;r(r)r*r2r2OsG#####r)r2c$eZdZUdZded<dZdS)LineDatazW Complete geometry data for contour lines over a whole sequence of contour levels. rlevelsc:tt|Sr6r7r9s r*r;zLineData.asdict`r<r)Nr=r(r)r*r?r?Zs;#####r)r?c(eZdZUdZded<ded<dS) ContourDataa Complete geometry data for filled polygons and/or contour lines over a whole sequence of contour levels. :func:`~bokeh.plotting.contour.contour_data` returns an object of this class that can then be passed to :func:`bokeh.models.ContourRenderer.set_data`. zFillData | None fill_datazLineData | None line_dataNr"r(r)r*rCrCd6 r)rC) want_fill want_linexArrayLike | Noneyz$ArrayLike | np.ma.MaskedArray | Noner@rGboolrHreturnct|}t|dkrd}|s|stdt||||||}d}|jr4|j}t |j|j|dd|dd}d} |jr#|j} t| j| j|} t|| S) z Return the contour data of filled and/or line contours that can be passed to :func:`bokeh.models.ContourRenderer.set_data` rFz/Neither fill nor line requested in contour_dataN)r r!r3r4)r r!r@) _validate_levelslen ValueError_contour_coordsr/r2r r!r0r?rC) rIrKrLr@rGrHcoordsrDr/rEr0s r*rrosf % %F 6{{Q LYLJKKK Q1fi C CFI v(  ;>PVWZXZWZP[jpqrqsqsjtuuu I R(  ;>&QQQ y) , ,,r)r c t|}t|dkrd}t|}|dddu}|rtt|d||d<t }t t ||di} tjD]} | | d} | | | | < n| dd|dddu}|rBt|d|dz |d<t } t t|| dn| dd| tjz tjz } | r%tdd| t!|||||| }t#t%tt t%t t t'| }|||jr|jj}|D]\} }t3|| ||jj}| jD]\} }||| d |_d |_|jrx|j j}| D]\} }t3|| ||j j}|jD]\} }||| |S) an Creates a :class:`bokeh.models.ContourRenderer` containing filled polygons and/or contour lines. Usually it is preferable to call :func:`~bokeh.plotting.figure.contour` instead of this function. Filled contour polygons are calculated if ``fill_color`` is set, contour lines if ``line_color`` is set. Args: x (array-like[float] of shape (ny, nx) or (nx,), optional) : The x-coordinates of the ``z`` values. May be 2D with the same shape as ``z.shape``, or 1D with length ``nx = z.shape[1]``. If not specified are assumed to be ``np.arange(nx)``. Must be ordered monotonically. y (array-like[float] of shape (ny, nx) or (ny,), optional) : The y-coordinates of the ``z`` values. May be 2D with the same shape as ``z.shape``, or 1D with length ``ny = z.shape[0]``. If not specified are assumed to be ``np.arange(ny)``. Must be ordered monotonically. z (array-like[float] of shape (ny, nx)) : A 2D NumPy array of gridded values to calculate the contours of. May be a masked array, and any invalid values (``np.inf`` or ``np.nan``) will also be masked out. levels (array-like[float]) : The z-levels to calculate the contours at, must be increasing. Contour lines are calculated at each level and filled contours are calculated between each adjacent pair of levels so the number of sets of contour lines is ``len(levels)`` and the number of sets of filled contour polygons is ``len(levels)-1``. **visuals: |fill properties|, |hatch properties| and |line properties| Fill and hatch properties are used for filled contours, line properties for line contours. If using vectorized properties then the correct number must be used, ``len(levels)`` for line properties and ``len(levels)-1`` for fill and hatch properties. ``fill_color`` and ``line_color`` are more flexible in that they will accept longer sequences and interpolate them to the required number using :func:`~bokeh.palettes.linear_palette`, and also accept palette collections (dictionaries mapping from integer length to color sequence) such as `bokeh.palettes.Cividis`. rF line_colorN fill_colorrRz-Unknown keyword arguments in 'from_contour': z, )rIrKrLr@rGrH)glyph data_source) fill_renderer line_rendererr@r)!rSrTget_colorrrr r propertiespopr keysrr rUjoinrr rlistset_datarDr]r[itemssetattrr\dataadd line_alpha line_widthrEr^)rIrKrLr@visualsrGnlevelsrHline_cds line_visualsnamepropfill_cdsunknownnew_contour_datacontour_rendererr[valuecdss r*rrsgnf % %F 6{{Q &kkG L$//t;I( &w|'><== = ZZ!^^"5!,,E Lr)rc|s|stdddlm}m}m}|||||j|j} d} |rg} g} tt|dz D]j} | || || dz}t|}| |j | |j kt| | } d}|rqg} g} |D]Z}| |}t!|}| |j | |j [t#| | }t%| |S)zD Return the (xs, ys) coords of filled and/or line contours. z2Neither fill nor line requested in _contour_coordsr)FillTypeLineTypecontour_generator) line_type fill_typeNrR) RuntimeError contourpyrrrChunkCombinedOffset OuterOffsetrangerTfilled_filled_to_coordsappendr r!rlines_lines_to_coordsr,r.)rIrKrLr@rGrHrrrcont_genr/all_xsall_ysirrWr0levelrs r*rVrV5s QYQOPPP??????????  AqH4P\d\pqqqHK1s6{{1}%% % %A__VAYqs <z%_filled_to_coords..h4HHH!6'!*WQqS\1145HHHr)cFg|]}||dzdfS)rRr(rs r*rz%_filled_to_coords..irr))ziprTrrrz)rr r!r~rrs @@r*rr^s B B<JJ LL1  HHHHHuQxxHHHIII HHHHHuQxxHHHIIII B # ##r)cV|dd}|$tjd}t||S|dd}t|}t|dz }tj||zdz }tj||zdz }t |D]s}||} ||dz} |dkr*tj|| |zdz <tj|| |zdz <|| | df|| |z| |z<|| | df|| |z| |z<tt||S)NrrR)npemptyr|rTrnan) rrrrnpointsnlinesr r!rstartends r*rrls@ 1Xa[F ~ u---AhqkG&kkG \\A F 'F"1$ % %B 'F"1$ % %B 6]]11 acl q55FBuQwqyMFBuQwqyM"59a<0573q5="59a<0573q5= B # ##r)c tj|}|jdkst|dkrt dt|dkr9tj|dkrt d|S)NrzNo contour levels specifiedrRgz!Contour levels must be increasing)rasarrayndimrTrUdiffmin)r@s r*rSrSs~ Z  F {a3v;;!++6777 6{{Q276??..00C77<=== Mr))NNNN)rIrJrKrJrLrMr@rJrGrNrHrNrOrC) rIrJrKrJrLrMr@rJrOr )r}rr~rrOr)rIrJrKrJrLrMr@rrGrNrHrNrOr.)rOrz)rOr|)r@rJ): __future__rlogging getLoggerr#logtypingrrrnumpyrcore.property_mixinsrr r models.glyphsr r models.renderersr rmodels.sourcesrpalettesrplotting._rendererrutil.dataclassesrr numpy.typingrtyping_extensionsrrr transformrrr'r__all__rr,r.r2r?rCrrrzr|r`rVrrrSr(r)r*rs#""""""g!!2111111111DCCCCCCCCC44444444========------%%%%%%;;;;;;11111111e&&&&&&++++++55555555%%%%%%#Ix /B$BCLCCCC',\7DUWc-c'ddddd   $%%%%%%%%  $  $########  $#####z### $#####z### $.2# - ------D.2# {{{{{B $ $    '3'3'3'3R $ $ $ $$$$$4r)