mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Union, Tuple, Annotated, Literal, Iterable
|
|
from colour import Color
|
|
import numpy as np
|
|
import re
|
|
|
|
# Abbreviations for a common types
|
|
ManimColor = Union[str, Color, None]
|
|
RangeSpecifier = Tuple[float, float, float] | Tuple[float, float]
|
|
|
|
|
|
Span = tuple[int, int]
|
|
SingleSelector = Union[
|
|
str,
|
|
re.Pattern,
|
|
tuple[Union[int, None], Union[int, None]],
|
|
]
|
|
Selector = Union[SingleSelector, Iterable[SingleSelector]]
|
|
|
|
UniformDict = Dict[str, float | bool | np.ndarray | tuple]
|
|
|
|
# These are various alternate names for np.ndarray meant to specify
|
|
# certain shapes.
|
|
#
|
|
# In theory, these annotations could be used to check arrays sizes
|
|
# at runtime, but at the moment nothing actually uses them, and
|
|
# the names are here primarily to enhance readibility and allow
|
|
# for some stronger type checking if numpy has stronger typing
|
|
# in the future
|
|
FloatArray = np.ndarray[int, np.dtype[np.float64]]
|
|
Vect2 = Annotated[FloatArray, Literal[2]]
|
|
Vect3 = Annotated[FloatArray, Literal[3]]
|
|
Vect4 = Annotated[FloatArray, Literal[4]]
|
|
VectN = Annotated[FloatArray, Literal["N"]]
|
|
Matrix3x3 = Annotated[FloatArray, Literal[3, 3]]
|
|
Vect2Array = Annotated[FloatArray, Literal["N", 2]]
|
|
Vect3Array = Annotated[FloatArray, Literal["N", 3]]
|
|
Vect4Array = Annotated[FloatArray, Literal["N", 4]]
|
|
VectNArray = Annotated[FloatArray, Literal["N", "M"]]
|