#-----------------------------------------------------------------------------
# Copyright (c) Anaconda, Inc., and Bokeh Contributors.
# All rights reserved.
#
# The full license is in the file LICENSE.txt, distributed with this software.
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Boilerplate
#-----------------------------------------------------------------------------
from __future__ import annotations
import logging # isort:skip
log = logging.getLogger(__name__)
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
# Standard library imports
from typing import (
TYPE_CHECKING,
Sequence,
TypeAlias,
cast,
)
# External imports
import numpy as np
# Bokeh imports
from ..core.property_mixins import FillProps, HatchProps, LineProps
from ..models.glyphs import MultiLine, MultiPolygons
from ..models.renderers import ContourRenderer, GlyphRenderer
from ..models.sources import ColumnDataSource
from ..palettes import interp_palette
from ..plotting._renderer import _process_sequence_literals
from ..util.dataclasses import dataclass, entries
if TYPE_CHECKING:
from contourpy._contourpy import FillReturn_OuterOffset, LineReturn_ChunkCombinedNan
from numpy.typing import ArrayLike, NDArray
from ..palettes import Palette, PaletteCollection
from ..transform import ColorLike
ContourColor: TypeAlias = ColorLike | Sequence[ColorLike]
ContourColorOrPalette: TypeAlias = ContourColor | Palette | PaletteCollection | ContourColor
#-----------------------------------------------------------------------------
# Globals and constants
#-----------------------------------------------------------------------------
__all__ = (
'contour_data',
'from_contour',
)
#-----------------------------------------------------------------------------
# General API
#-----------------------------------------------------------------------------
@dataclass(frozen=True)
class FillCoords:
''' Coordinates for all filled polygons over a whole sequence of contour levels.
'''
xs: list[list[list[np.ndarray]]]
ys: list[list[list[np.ndarray]]]
@dataclass(frozen=True)
class LineCoords:
''' Coordinates for all contour lines over a whole sequence of contour levels.
'''
xs: list[np.ndarray]
ys: list[np.ndarray]
@dataclass(frozen=True)
class ContourCoords:
''' Combined filled and line contours over a whole sequence of contour levels.
'''
fill_coords: FillCoords | None
line_coords: LineCoords | None
@dataclass(frozen=True)
class FillData(FillCoords):
''' Complete geometry data for filled polygons over a whole sequence of contour levels.
'''
lower_levels: ArrayLike
upper_levels: ArrayLike
def asdict(self):
# Convert to dict using shallow copy. dataclasses.asdict uses deep copy.
return dict(entries(self))
@dataclass(frozen=True)
class LineData(LineCoords):
''' Complete geometry data for contour lines over a whole sequence of contour levels.
'''
levels: ArrayLike
def asdict(self):
# Convert to dict using shallow copy. dataclasses.asdict uses deep copy.
return dict(entries(self))
@dataclass(frozen=True)
class ContourData:
''' 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`.
'''
fill_data: FillData | None
line_data: LineData | None
[docs]
def contour_data(
x: ArrayLike | None = None,
y: ArrayLike | None = None,
z: ArrayLike | np.ma.MaskedArray | None = None,
levels: ArrayLike | None = None,
*,
want_fill: bool = True,
want_line: bool = True,
) -> ContourData:
''' Return the contour data of filled and/or line contours that can be
passed to :func:`bokeh.models.ContourRenderer.set_data`
'''
levels = _validate_levels(levels)
if len(levels) < 2:
want_fill = False
if not want_fill and not want_line:
raise ValueError("Neither fill nor line requested in contour_data")
coords = _contour_coords(x, y, z, levels, want_fill, want_line)
fill_data = None
if coords.fill_coords:
fill_coords = coords.fill_coords
fill_data = FillData(xs=fill_coords.xs, ys=fill_coords.ys, lower_levels=levels[:-1], upper_levels=levels[1:])
line_data = None
if coords.line_coords:
line_coords = coords.line_coords
line_data = LineData(xs=line_coords.xs, ys=line_coords.ys, levels=levels)
return ContourData(fill_data, line_data)
[docs]
def from_contour(
x: ArrayLike | None = None,
y: ArrayLike | None = None,
z: ArrayLike | np.ma.MaskedArray | None = None,
levels: ArrayLike | None = None,
**visuals, # This is union of LineProps, FillProps and HatchProps
) -> ContourRenderer:
''' 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`.
'''
levels = _validate_levels(levels)
if len(levels) < 2:
want_fill = False
nlevels = len(levels)
want_line = visuals.get("line_color", None) is not None
if want_line:
# Handle possible callback or interpolation for line_color.
visuals["line_color"] = _color(visuals["line_color"], nlevels)
line_cds = ColumnDataSource()
_process_sequence_literals(MultiLine, visuals, line_cds, False)
# Remove line visuals identified from visuals dict.
line_visuals = {}
for name in LineProps.properties():
prop = visuals.pop(name, None)
if prop is not None:
line_visuals[name] = prop
else:
visuals.pop("line_color", None)
want_fill = visuals.get("fill_color", None) is not None
if want_fill:
# Handle possible callback or interpolation for fill_color and hatch_color.
visuals["fill_color"] = _color(visuals["fill_color"], nlevels-1)
if "hatch_color" in visuals:
visuals["hatch_color"] = _color(visuals["hatch_color"], nlevels-1)
fill_cds = ColumnDataSource()
_process_sequence_literals(MultiPolygons, visuals, fill_cds, False)
else:
visuals.pop("fill_color", None)
# Check for extra unknown kwargs.
unknown = visuals.keys() - FillProps.properties() - HatchProps.properties()
if unknown:
raise ValueError(f"Unknown keyword arguments in 'from_contour': {', '.join(unknown)}")
new_contour_data = contour_data(x=x, y=y, z=z, levels=levels, want_fill=want_fill, want_line=want_line)
# Will be other possibilities here like logarithmic....
contour_renderer = ContourRenderer(
fill_renderer=GlyphRenderer(glyph=MultiPolygons(), data_source=ColumnDataSource()),
line_renderer=GlyphRenderer(glyph=MultiLine(), data_source=ColumnDataSource()),
levels=list(levels))
contour_renderer.set_data(new_contour_data)
if new_contour_data.fill_data:
glyph = contour_renderer.fill_renderer.glyph
for name, value in visuals.items():
setattr(glyph, name, value)
cds = contour_renderer.fill_renderer.data_source
for name, value in fill_cds.data.items():
cds.add(value, name)
glyph.line_alpha = 0 # Don't display lines around fill.
glyph.line_width = 0
if new_contour_data.line_data:
glyph = contour_renderer.line_renderer.glyph
for name, value in line_visuals.items():
setattr(glyph, name, value)
cds = contour_renderer.line_renderer.data_source
for name, value in line_cds.data.items():
cds.add(value, name)
return contour_renderer
#-----------------------------------------------------------------------------
# Private API
#-----------------------------------------------------------------------------
@dataclass(frozen=True)
class SingleFillCoords:
''' Coordinates for filled contour polygons between a lower and upper level.
The first list contains a list for each polygon. The second list contains
a separate NumPy array for each boundary of that polygon; the first array
is always the outer boundary, subsequent arrays are holes.
'''
xs: list[list[np.ndarray]]
ys: list[list[np.ndarray]]
@dataclass(frozen=True)
class SingleLineCoords:
''' Coordinates for contour lines at a single contour level.
The x and y coordinates are stored in a single NumPy array each, with a
np.nan separating each line.
'''
xs: np.ndarray
ys: np.ndarray
def _color(color: ContourColorOrPalette, n: int) -> ContourColor:
# Dict to sequence of colors such as palettes.cividis
if isinstance(color, dict):
return _palette_from_collection(color, n)
if isinstance(color, Sequence) and not isinstance(color, bytes | str) and len(color) != n:
return interp_palette(color, n)
return color
def _contour_coords(
x: ArrayLike | None,
y: ArrayLike | None,
z: ArrayLike | np.ma.MaskedArray | None,
levels: ArrayLike,
want_fill: bool,
want_line: bool,
) -> ContourCoords:
'''
Return the (xs, ys) coords of filled and/or line contours.
'''
if not want_fill and not want_line:
raise RuntimeError("Neither fill nor line requested in _contour_coords")
from contourpy import FillType, LineType, contour_generator
cont_gen = contour_generator(x, y, z, line_type=LineType.ChunkCombinedNan, fill_type=FillType.OuterOffset)
fill_coords = None
if want_fill:
all_xs = []
all_ys = []
for i in range(len(levels)-1):
filled = cont_gen.filled(levels[i], levels[i+1])
# This is guaranteed by use of fill_type=FillType.OuterOffset in contour_generator call.
filled = cast("FillReturn_OuterOffset", filled)
coords = _filled_to_coords(filled)
all_xs.append(coords.xs)
all_ys.append(coords.ys)
fill_coords = FillCoords(all_xs, all_ys)
line_coords = None
if want_line:
all_xs = []
all_ys = []
for level in levels:
lines = cont_gen.lines(level)
# This is guaranteed by use of line_type=LineType.ChunkCombinedNan in contour_generator call.
lines = cast("LineReturn_ChunkCombinedNan", lines)
coords = _lines_to_coords(lines)
all_xs.append(coords.xs)
all_ys.append(coords.ys)
line_coords = LineCoords(all_xs, all_ys)
return ContourCoords(fill_coords, line_coords)
def _filled_to_coords(filled: FillReturn_OuterOffset) -> SingleFillCoords:
# Processes polygon data returned from a single call to
# contourpy.ContourGenerator.filled(lower_level, upper_level)
# ContourPy filled data format is FillType.OuterOffset.
xs = []
ys = []
for points, offsets in zip(*filled):
# Polygon with outer boundary and zero or more holes.
n = len(offsets) - 1
xs.append([points[offsets[i]:offsets[i+1], 0] for i in range(n)])
ys.append([points[offsets[i]:offsets[i+1], 1] for i in range(n)])
return SingleFillCoords(xs, ys)
def _lines_to_coords(lines: LineReturn_ChunkCombinedNan) -> SingleLineCoords:
# Processes line data returned from a single call to
# contourpy.ContourGenerator.lines(level).
# ContourPy line data format is LineType.ChunkCombinedNan.
points = lines[0][0]
if points is None:
empty = np.empty(0)
return SingleLineCoords(empty, empty)
xs = points[:, 0]
ys = points[:, 1]
return SingleLineCoords(xs, ys)
def _palette_from_collection(collection: PaletteCollection, n: int) -> Palette:
# Return palette of length n from the specified palette collection, which
# is a dict[int, Palette]. If the required length palette is in the
# collection then return that. If the required length is bigger than the
# longest palette then interpolate that. If the required length is smaller
# than the shortest palette then interpolate that.
if len(collection) < 1:
raise ValueError("PaletteCollection is empty")
palette = collection.get(n, None)
if palette is not None:
return palette
max_key = max(collection.keys())
if isinstance(max_key, int) and n > max_key:
return interp_palette(collection[max_key], n)
min_key = min(collection.keys())
if isinstance(min_key, int) and n < min_key:
return interp_palette(collection[min_key], n)
raise ValueError(f"Unable to extract or interpolate palette of length {n} from PaletteCollection")
def _validate_levels(levels: ArrayLike | None) -> NDArray[float]:
levels = np.asarray(levels)
if levels.ndim == 0 or len(levels) == 0:
raise ValueError("No contour levels specified")
if len(levels) > 1 and np.diff(levels).min() <= 0.0:
raise ValueError("Contour levels must be increasing")
return levels