Move custom type to manimlib.typing

manimlib/animation/creation.py

@@ -19,7 +19,7 @@ if TYPE_CHECKING:
     from typing import Callable
     from manimlib.mobject.mobject import Mobject
     from manimlib.scene.scene import Scene
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class ShowPartial(Animation, ABC):

manimlib/animation/growing.py

@@ -10,7 +10,7 @@ if TYPE_CHECKING:
 
     from manimlib.mobject.geometry import Arrow
     from manimlib.mobject.mobject import Mobject
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class GrowFromPoint(Transform):

manimlib/animation/indication.py

@@ -36,7 +36,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Callable
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
     from manimlib.mobject.mobject import Mobject
 
 

manimlib/animation/specialized.py

@@ -10,7 +10,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     import numpy as np
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class Broadcast(LaggedStart):

manimlib/animation/transform.py

@@ -18,7 +18,7 @@ if TYPE_CHECKING:
     from typing import Callable
     import numpy.typing as npt
     from manimlib.scene.scene import Scene
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class Transform(Animation):

manimlib/camera/camera.py

@@ -26,7 +26,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from manimlib.shader_wrapper import ShaderWrapper
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
     from typing import Sequence
 
     np_vector = np.ndarray[int, np.dtype[np.float64]]

manimlib/constants.py

@@ -1,19 +1,11 @@
 from __future__ import annotations
-
 import numpy as np
 
 from typing import TYPE_CHECKING
-
 if TYPE_CHECKING:
-    from typing import Union, Tuple
-    from colour import Color
-
     from typing import List
+    from manimlib.typing import ManimColor, np_vector
 
-    # Abbreviations for a common types
-    ManimColor = Union[str, Color, None]
-    RangeSpecifier = Tuple[float, float, float] | Tuple[float, float]
-    np_vector = np.ndarray[int, np.dtype[np.float64]]
 
 # Sizes relevant to default camera frame
 ASPECT_RATIO: float = 16.0 / 9.0

manimlib/mobject/changing.py

@@ -11,8 +11,8 @@ from manimlib.utils.rate_functions import smooth
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Callable, List, Tuple, Iterable
-    from manimlib.constants import ManimColor, np_vector
+    from typing import Callable, List, Iterable
+    from manimlib.typing import ManimColor, np_vector
 
 
 class AnimatedBoundary(VGroup):

manimlib/mobject/coordinate_systems.py

@@ -34,7 +34,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Callable, Iterable, Sequence, Type, TypeVar
     from manimlib.mobject.mobject import Mobject
-    from manimlib.constants import ManimColor, np_vector, RangeSpecifier
+    from manimlib.typing import ManimColor, np_vector, RangeSpecifier
 
     T = TypeVar("T", bound=Mobject)
 

manimlib/mobject/frame.py

@@ -6,7 +6,7 @@ from manimlib.mobject.geometry import Rectangle
 
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class ScreenRectangle(Rectangle):

manimlib/mobject/functions.py

@@ -11,7 +11,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Callable, Sequence, Tuple
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 class ParametricCurve(VMobject):

manimlib/mobject/geometry.py

@@ -30,7 +30,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Iterable
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 DEFAULT_DOT_RADIUS = 0.08

manimlib/mobject/interactive.py

@@ -26,7 +26,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Callable
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 # Interactive Mobjects

manimlib/mobject/matrix.py

@@ -21,7 +21,7 @@ if TYPE_CHECKING:
     from typing import Sequence
     import numpy.typing as npt
     from manimlib.mobject.mobject import Mobject
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 VECTOR_LABEL_SCALE_FACTOR = 0.8

manimlib/mobject/mobject.py

@@ -48,7 +48,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Callable, Iterable, Sequence, Union, Tuple
     import numpy.typing as npt
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
     TimeBasedUpdater = Callable[["Mobject", float], None]
     NonTimeUpdater = Callable[["Mobject"], None]

manimlib/mobject/number_line.py

@@ -16,8 +16,8 @@ from manimlib.utils.simple_functions import fdiv
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Iterable, Sequence
-    from manimlib.constants import ManimColor, np_vector, RangeSpecifier
+    from typing import Iterable
+    from manimlib.typing import ManimColor, np_vector, RangeSpecifier
 
 
 class NumberLine(Line):

manimlib/mobject/numbers.py

@@ -11,8 +11,8 @@ from manimlib.mobject.types.vectorized_mobject import VMobject
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Type, TypeVar
-    from manimlib.constants import ManimColor, np_vector
+    from typing import TypeVar
+    from manimlib.typing import ManimColor, np_vector
 
     T = TypeVar("T", bound=VMobject)
 

manimlib/mobject/probability.py

@@ -19,7 +19,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Iterable
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 EPSILON = 0.0001

manimlib/mobject/shape_matchers.py

@@ -16,7 +16,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Sequence
     from manimlib.mobject.mobject import Mobject
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 class SurroundingRectangle(Rectangle):

manimlib/mobject/svg/brace.py

@@ -27,8 +27,7 @@ if TYPE_CHECKING:
 
     from manimlib.animation.animation import Animation
     from manimlib.mobject.mobject import Mobject
-
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import np_vector
 
 
 class Brace(SingleStringTex):

manimlib/mobject/svg/drawings.py

@@ -60,7 +60,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Tuple, Sequence, Callable
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 class Checkmark(TexTextFromPresetString):

manimlib/mobject/svg/mtex_mobject.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
     from typing import Iterable, Union
 
     from manimlib.mobject.types.vectorized_mobject import VGroup
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
     Span = tuple[int, int]
     Selector = Union[

manimlib/mobject/svg/string_mobject.py

@@ -17,7 +17,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Callable, Iterable, Union
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
     Span = tuple[int, int]
     Selector = Union[

manimlib/mobject/svg/svg_mobject.py

@@ -22,7 +22,7 @@ from manimlib.utils.simple_functions import hash_string
 
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 

manimlib/mobject/svg/tex_mobject.py

@@ -17,8 +17,8 @@ from manimlib.utils.tex_file_writing import tex_content_to_svg_file
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Iterable, List, Dict, Sequence
-    from manimlib.constants import ManimColor
+    from typing import Iterable, List, Dict
+    from manimlib.typing import ManimColor
 
 
 SCALE_FACTOR_PER_FONT_POINT = 0.001

manimlib/mobject/svg/text_mobject.py

@@ -22,10 +22,10 @@ from manimlib.utils.simple_functions import hash_string
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Iterable, Union, Pattern
+    from typing import Iterable, Union
 
     from manimlib.mobject.types.vectorized_mobject import VGroup
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
     Span = tuple[int, int]
     Selector = Union[
         str,

manimlib/mobject/three_dimensions.py

@@ -23,7 +23,7 @@ from manimlib.utils.space_ops import z_to_vector
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Tuple, TypeVar
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
     T = TypeVar("T", bound=Mobject)
 

manimlib/mobject/types/dot_cloud.py

@@ -13,7 +13,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     import numpy.typing as npt
     from typing import Sequence, Tuple
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 DEFAULT_DOT_RADIUS = 0.05

manimlib/mobject/types/image_mobject.py

@@ -13,7 +13,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Sequence, Tuple
-    from manimlib.constants import np_vector
+    from manimlib.typing import np_vector
 
 
 class ImageMobject(Mobject):

manimlib/mobject/types/point_cloud_mobject.py

@@ -15,7 +15,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Callable, Sequence
     import numpy.typing as npt
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 class PMobject(Mobject):

manimlib/mobject/types/surface.py

@@ -18,7 +18,7 @@ if TYPE_CHECKING:
     from typing import Callable, Iterable, Sequence, Tuple
 
     from manimlib.camera.camera import Camera
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 
 class Surface(Mobject):

manimlib/mobject/types/vectorized_mobject.py

@@ -42,7 +42,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Callable, Iterable, Sequence, Tuple
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
 DEFAULT_STROKE_COLOR = GREY_A
 DEFAULT_FILL_COLOR = GREY_C

manimlib/mobject/vector_field.py

@@ -23,7 +23,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Callable, Iterable, Sequence, TypeVar, Tuple
     import numpy.typing as npt
-    from manimlib.constants import ManimColor, np_vector
+    from manimlib.typing import ManimColor, np_vector
 
     from manimlib.mobject.coordinate_systems import CoordinateSystem
     from manimlib.mobject.mobject import Mobject

manimlib/typing.py

@@ -0,0 +1,19 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Union, Tuple
+    from colour import Color
+    import numpy as np
+
+    # Abbreviations for a common types
+    ManimColor = Union[str, Color, None]
+    RangeSpecifier = Tuple[float, float, float] | Tuple[float, float]
+
+    # TODO, Nothing about these actually specifies length,
+    # they are so far just about code readability
+    np_vector = np.ndarray[int, np.dtype[np.float64]]
+    Vect2 = np.ndarray[int, np.dtype[np.float64]]  # TODO, specify length of 2
+    Vect3 = np.ndarray[int, np.dtype[np.float64]]  # TODO, specify length of 3
+    Vect4 = np.ndarray[int, np.dtype[np.float64]]  # TODO, specify length of 4
+    VectN = np.ndarray[int, np.dtype[np.float64]]
+    Matrix3x3 = np.ndarray[int, np.dtype[np.float64]]  # TODO, specify output size

manimlib/utils/color.py

@@ -14,7 +14,7 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from typing import Iterable
-    from manimlib.constants import ManimColor
+    from manimlib.typing import ManimColor
 
 
 def color_to_rgb(color: ManimColor) -> np.ndarray: