211 lines
6.7 KiB
Python
211 lines
6.7 KiB
Python
|
from collections.abc import Callable, Sequence
|
||
|
|
import os
|
||
|
|
import pathlib
|
||
|
|
from typing import Any, BinaryIO, Literal
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
from numpy.typing import ArrayLike, NDArray
|
||
|
|
import PIL.Image
|
||
|
|
|
||
|
|
import matplotlib.artist as martist
|
||
|
|
from matplotlib.axes import Axes
|
||
|
|
from matplotlib import cm
|
||
|
|
from matplotlib.backend_bases import RendererBase, MouseEvent
|
||
|
|
from matplotlib.colors import Colormap, Normalize
|
||
|
|
from matplotlib.figure import Figure
|
||
|
|
from matplotlib.transforms import Affine2D, BboxBase, Bbox, Transform
|
||
|
|
|
||
|
|
#
|
||
|
|
# These names are re-exported from matplotlib._image.
|
||
|
|
#
|
||
|
|
|
||
|
|
BESSEL: int
|
||
|
|
BICUBIC: int
|
||
|
|
BILINEAR: int
|
||
|
|
BLACKMAN: int
|
||
|
|
CATROM: int
|
||
|
|
GAUSSIAN: int
|
||
|
|
HAMMING: int
|
||
|
|
HANNING: int
|
||
|
|
HERMITE: int
|
||
|
|
KAISER: int
|
||
|
|
LANCZOS: int
|
||
|
|
MITCHELL: int
|
||
|
|
NEAREST: int
|
||
|
|
QUADRIC: int
|
||
|
|
SINC: int
|
||
|
|
SPLINE16: int
|
||
|
|
SPLINE36: int
|
||
|
|
|
||
|
|
def resample(
|
||
|
|
input_array: NDArray[np.float32] | NDArray[np.float64] | NDArray[np.int8],
|
||
|
|
output_array: NDArray[np.float32] | NDArray[np.float64] | NDArray[np.int8],
|
||
|
|
transform: Transform,
|
||
|
|
interpolation: int = ...,
|
||
|
|
resample: bool = ...,
|
||
|
|
alpha: float = ...,
|
||
|
|
norm: bool = ...,
|
||
|
|
radius: float = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
#
|
||
|
|
# END names re-exported from matplotlib._image.
|
||
|
|
#
|
||
|
|
|
||
|
|
interpolations_names: set[str]
|
||
|
|
|
||
|
|
def composite_images(
|
||
|
|
images: Sequence[_ImageBase], renderer: RendererBase, magnification: float = ...
|
||
|
|
) -> tuple[np.ndarray, float, float]: ...
|
||
|
|
|
||
|
|
class _ImageBase(martist.Artist, cm.ScalarMappable):
|
||
|
|
zorder: float
|
||
|
|
origin: Literal["upper", "lower"]
|
||
|
|
axes: Axes
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
ax: Axes,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
norm: str | Normalize | None = ...,
|
||
|
|
interpolation: str | None = ...,
|
||
|
|
origin: Literal["upper", "lower"] | None = ...,
|
||
|
|
filternorm: bool = ...,
|
||
|
|
filterrad: float = ...,
|
||
|
|
resample: bool | None = ...,
|
||
|
|
*,
|
||
|
|
interpolation_stage: Literal["data", "rgba"] | None = ...,
|
||
|
|
**kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def get_size(self) -> tuple[int, int]: ...
|
||
|
|
def set_alpha(self, alpha: float | ArrayLike | None) -> None: ...
|
||
|
|
def changed(self) -> None: ...
|
||
|
|
def make_image(
|
||
|
|
self, renderer: RendererBase, magnification: float = ..., unsampled: bool = ...
|
||
|
|
) -> tuple[np.ndarray, float, float, Affine2D]: ...
|
||
|
|
def draw(self, renderer: RendererBase) -> None: ...
|
||
|
|
def write_png(self, fname: str | pathlib.Path | BinaryIO) -> None: ...
|
||
|
|
def set_data(self, A: ArrayLike | None) -> None: ...
|
||
|
|
def set_array(self, A: ArrayLike | None) -> None: ...
|
||
|
|
def get_shape(self) -> tuple[int, int, int]: ...
|
||
|
|
def get_interpolation(self) -> str: ...
|
||
|
|
def set_interpolation(self, s: str | None) -> None: ...
|
||
|
|
def get_interpolation_stage(self) -> Literal["data", "rgba"]: ...
|
||
|
|
def set_interpolation_stage(self, s: Literal["data", "rgba"]) -> None: ...
|
||
|
|
def can_composite(self) -> bool: ...
|
||
|
|
def set_resample(self, v: bool | None) -> None: ...
|
||
|
|
def get_resample(self) -> bool: ...
|
||
|
|
def set_filternorm(self, filternorm: bool) -> None: ...
|
||
|
|
def get_filternorm(self) -> bool: ...
|
||
|
|
def set_filterrad(self, filterrad: float) -> None: ...
|
||
|
|
def get_filterrad(self) -> float: ...
|
||
|
|
|
||
|
|
class AxesImage(_ImageBase):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
ax: Axes,
|
||
|
|
*,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
norm: str | Normalize | None = ...,
|
||
|
|
interpolation: str | None = ...,
|
||
|
|
origin: Literal["upper", "lower"] | None = ...,
|
||
|
|
extent: tuple[float, float, float, float] | None = ...,
|
||
|
|
filternorm: bool = ...,
|
||
|
|
filterrad: float = ...,
|
||
|
|
resample: bool = ...,
|
||
|
|
interpolation_stage: Literal["data", "rgba"] | None = ...,
|
||
|
|
**kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def get_window_extent(self, renderer: RendererBase | None = ...) -> Bbox: ...
|
||
|
|
def make_image(
|
||
|
|
self, renderer: RendererBase, magnification: float = ..., unsampled: bool = ...
|
||
|
|
) -> tuple[np.ndarray, float, float, Affine2D]: ...
|
||
|
|
def set_extent(
|
||
|
|
self, extent: tuple[float, float, float, float], **kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def get_extent(self) -> tuple[float, float, float, float]: ...
|
||
|
|
def get_cursor_data(self, event: MouseEvent) -> None | float: ...
|
||
|
|
|
||
|
|
class NonUniformImage(AxesImage):
|
||
|
|
mouseover: bool
|
||
|
|
def __init__(
|
||
|
|
self, ax: Axes, *, interpolation: Literal["nearest", "bilinear"] = ..., **kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def set_data(self, x: ArrayLike, y: ArrayLike, A: ArrayLike) -> None: ... # type: ignore[override]
|
||
|
|
# more limited interpolation available here than base class
|
||
|
|
def set_interpolation(self, s: Literal["nearest", "bilinear"]) -> None: ... # type: ignore[override]
|
||
|
|
|
||
|
|
class PcolorImage(AxesImage):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
ax: Axes,
|
||
|
|
x: ArrayLike | None = ...,
|
||
|
|
y: ArrayLike | None = ...,
|
||
|
|
A: ArrayLike | None = ...,
|
||
|
|
*,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
norm: str | Normalize | None = ...,
|
||
|
|
**kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def set_data(self, x: ArrayLike, y: ArrayLike, A: ArrayLike) -> None: ... # type: ignore[override]
|
||
|
|
|
||
|
|
class FigureImage(_ImageBase):
|
||
|
|
zorder: float
|
||
|
|
figure: Figure
|
||
|
|
ox: float
|
||
|
|
oy: float
|
||
|
|
magnification: float
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
fig: Figure,
|
||
|
|
*,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
norm: str | Normalize | None = ...,
|
||
|
|
offsetx: int = ...,
|
||
|
|
offsety: int = ...,
|
||
|
|
origin: Literal["upper", "lower"] | None = ...,
|
||
|
|
**kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def get_extent(self) -> tuple[float, float, float, float]: ...
|
||
|
|
|
||
|
|
class BboxImage(_ImageBase):
|
||
|
|
bbox: BboxBase
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
bbox: BboxBase | Callable[[RendererBase | None], Bbox],
|
||
|
|
*,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
norm: str | Normalize | None = ...,
|
||
|
|
interpolation: str | None = ...,
|
||
|
|
origin: Literal["upper", "lower"] | None = ...,
|
||
|
|
filternorm: bool = ...,
|
||
|
|
filterrad: float = ...,
|
||
|
|
resample: bool = ...,
|
||
|
|
**kwargs
|
||
|
|
) -> None: ...
|
||
|
|
def get_window_extent(self, renderer: RendererBase | None = ...) -> Bbox: ...
|
||
|
|
|
||
|
|
def imread(
|
||
|
|
fname: str | pathlib.Path | BinaryIO, format: str | None = ...
|
||
|
|
) -> np.ndarray: ...
|
||
|
|
def imsave(
|
||
|
|
fname: str | os.PathLike | BinaryIO,
|
||
|
|
arr: ArrayLike,
|
||
|
|
vmin: float | None = ...,
|
||
|
|
vmax: float | None = ...,
|
||
|
|
cmap: str | Colormap | None = ...,
|
||
|
|
format: str | None = ...,
|
||
|
|
origin: Literal["upper", "lower"] | None = ...,
|
||
|
|
dpi: float = ...,
|
||
|
|
*,
|
||
|
|
metadata: dict[str, str] | None = ...,
|
||
|
|
pil_kwargs: dict[str, Any] | None = ...
|
||
|
|
) -> None: ...
|
||
|
|
def pil_to_array(pilImage: PIL.Image.Image) -> np.ndarray: ...
|
||
|
|
def thumbnail(
|
||
|
|
infile: str | BinaryIO,
|
||
|
|
thumbfile: str | BinaryIO,
|
||
|
|
scale: float = ...,
|
||
|
|
interpolation: str = ...,
|
||
|
|
preview: bool = ...,
|
||
|
|
) -> Figure: ...
|