Typing scene

manim/animation/transform.py

@@ -49,6 +49,8 @@
 from ..utils.rate_functions import smooth, squish_rate_func
 
 if TYPE_CHECKING:
+    from typing import Any
+
     from ..scene.scene import Scene
 
 
@@ -516,7 +518,7 @@ def construct(self):
 
     def __init__(
         self,
-        function: types.MethodType,
+        function: Callable,
         mobject: Mobject,
         run_time: float = DEFAULT_POINTWISE_FUNCTION_RUN_TIME,
         **kwargs,
@@ -615,7 +617,9 @@ def __init__(self, mobject: Mobject, **kwargs) -> None:
 
 
 class ApplyFunction(Transform):
-    def __init__(self, function: types.MethodType, mobject: Mobject, **kwargs) -> None:
+    def __init__(
+        self, function: Callable[[Any], Any], mobject: Mobject, **kwargs: Any
+    ) -> None:
         self.function = function
         super().__init__(mobject, **kwargs)
 

manim/animation/transform_matching_parts.py

@@ -20,6 +20,8 @@
 from .transform import FadeTransformPieces, Transform
 
 if TYPE_CHECKING:
+    from typing import Any
+
     from ..scene.scene import Scene
 
 
@@ -74,7 +76,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         if isinstance(mobject, OpenGLVMobject):
             group_type = OpenGLVGroup
@@ -162,11 +164,11 @@ def clean_up_from_scene(self, scene: Scene) -> None:
         scene.add(self.to_add)
 
     @staticmethod
-    def get_mobject_parts(mobject: Mobject):
+    def get_mobject_parts(mobject: Mobject) -> None:
         raise NotImplementedError("To be implemented in subclass.")
 
     @staticmethod
-    def get_mobject_key(mobject: Mobject):
+    def get_mobject_key(mobject: Mobject) -> None:
         raise NotImplementedError("To be implemented in subclass.")
 
 
@@ -206,7 +208,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         super().__init__(
             mobject,
@@ -269,7 +271,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         super().__init__(
             mobject,

manim/camera/camera.py

@@ -274,7 +274,9 @@ def init_background(self):
             )
             self.background[:, :] = background_rgba
 
-    def get_image(self, pixel_array: np.ndarray | list | tuple | None = None):
+    def get_image(
+        self, pixel_array: np.ndarray | list | tuple | None = None
+    ) -> Image.Image:
         """Returns an image from the passed
         pixel array, or from the current frame
         if the passed pixel array is none.

manim/camera/moving_camera.py

@@ -10,6 +10,8 @@
 
 __all__ = ["MovingCamera"]
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from .. import config
@@ -19,6 +21,11 @@
 from ..mobject.mobject import Mobject
 from ..utils.color import WHITE
 
+if TYPE_CHECKING:
+    from typing import Any
+
+    from manim.utils.color import ParsableManimColor
+
 
 class MovingCamera(Camera):
     """
@@ -32,12 +39,12 @@ class MovingCamera(Camera):
 
     def __init__(
         self,
-        frame=None,
-        fixed_dimension=0,  # width
-        default_frame_stroke_color=WHITE,
-        default_frame_stroke_width=0,
-        **kwargs,
-    ):
+        frame: ScreenRectangle | None = None,
+        fixed_dimension: Any = 0,  # width
+        default_frame_stroke_color: ParsableManimColor = WHITE,
+        default_frame_stroke_width: float = 0,
+        **kwargs: Any,
+    ) -> None:
         """
         Frame is a Mobject, (should almost certainly be a rectangle)
         determining which region of space the camera displays
@@ -158,7 +165,7 @@ def cache_cairo_context(self, pixel_array, ctx):
     #         self.frame_shape = (self.frame.height, width)
     #     self.resize_frame_shape(fixed_dimension=self.fixed_dimension)
 
-    def get_mobjects_indicating_movement(self):
+    def get_mobjects_indicating_movement(self) -> list[ScreenRectangle]:
         """
         Returns all mobjects whose movement implies that the camera
         should think of all other mobjects on the screen as moving

manim/camera/multi_camera.py

@@ -5,7 +5,7 @@
 __all__ = ["MultiCamera"]
 
 
-from manim.mobject.types.image_mobject import ImageMobject
+from manim.mobject.types.image_mobject import ImageMobject, ImageMobjectFromCamera
 
 from ..camera.moving_camera import MovingCamera
 from ..utils.iterables import list_difference_update
@@ -38,7 +38,9 @@ def __init__(
         )
         super().__init__(**kwargs)
 
-    def add_image_mobject_from_camera(self, image_mobject_from_camera: ImageMobject):
+    def add_image_mobject_from_camera(
+        self, image_mobject_from_camera: ImageMobjectFromCamera
+    ):
         """Adds an ImageMobject that's been obtained from the camera
         into the list ``self.image_mobject_from_cameras``
 

manim/camera/three_d_camera.py

@@ -84,7 +84,7 @@ def capture_mobjects(self, mobjects, **kwargs):
         self.reset_rotation_matrix()
         super().capture_mobjects(mobjects, **kwargs)
 
-    def get_value_trackers(self):
+    def get_value_trackers(self) -> list[ValueTracker]:
         """A list of :class:`ValueTrackers <.ValueTracker>` of phi, theta, focal_distance,
         gamma and zoom.
 

manim/gui/gui.py

@@ -11,6 +11,8 @@
 
 
 from .. import __version__, config
+from ..renderer.cairo_renderer import CairoRenderer
+from ..renderer.opengl_renderer import OpenGLRenderer
 from ..utils.module_ops import scene_classes_from_file
 
 __all__ = ["configure_pygui"]
@@ -20,7 +22,9 @@
     window = dpg.generate_uuid()
 
 
-def configure_pygui(renderer, widgets, update=True):
+def configure_pygui(
+    renderer: CairoRenderer | OpenGLRenderer, widgets, update: bool = True
+) -> None:
     if not dearpygui_imported:
         raise RuntimeError("Attempted to use DearPyGUI when it isn't imported.")
     if update:

manim/mobject/geometry/arc.py

@@ -326,10 +326,10 @@ def __init__(
         if radius is None:  # apparently None is passed by ArcBetweenPoints
             radius = 1.0
         self.radius = radius
-        self.num_components = num_components
-        self.arc_center: Point3D = np.asarray(arc_center)
-        self.start_angle = start_angle
-        self.angle = angle
+        self.num_components: int = num_components
+        self.arc_center: Point3D = arc_center
+        self.start_angle: float = start_angle
+        self.angle: float = angle
         self._failed_to_get_center: bool = False
         super().__init__(**kwargs)
 

manim/mobject/geometry/polygram.py

@@ -325,7 +325,7 @@ def construct(self):
                 self.add(isosceles, square_and_triangles)
     """
 
-    def __init__(self, *vertices: Point3DLike, **kwargs: Any) -> None:
+    def __init__(self, *vertices: Point3D, **kwargs: Any) -> None:
         super().__init__(vertices, **kwargs)
 
 

manim/mobject/matrix.py

@@ -400,7 +400,7 @@ def add_background_to_entries(self):
             mob.add_background_rectangle()
         return self
 
-    def get_mob_matrix(self):
+    def get_mob_matrix(self) -> VGroup:
         """Return the underlying mob matrix mobjects.
 
         Returns
@@ -410,7 +410,7 @@ def get_mob_matrix(self):
         """
         return self.mob_matrix
 
-    def get_entries(self):
+    def get_entries(self) -> VGroup:
         """Return the individual entries of the matrix.
 
         Returns
@@ -435,7 +435,7 @@ def construct(self):
         """
         return self.elements
 
-    def get_brackets(self):
+    def get_brackets(self) -> VGroup:
         r"""Return the bracket mobjects.
 
         Returns

manim/mobject/text/tex_mobject.py

@@ -29,6 +29,7 @@
 from collections.abc import Iterable
 from functools import reduce
 from textwrap import dedent
+from typing import TYPE_CHECKING
 
 from manim import config, logger
 from manim.constants import *
@@ -38,6 +39,9 @@
 from manim.utils.tex import TexTemplate
 from manim.utils.tex_file_writing import tex_to_svg_file
 
+if TYPE_CHECKING:
+    from typing_extensions import Any
+
 tex_string_to_mob_map = {}
 
 
@@ -206,7 +210,7 @@ def _organize_submobjects_left_to_right(self):
         self.sort(lambda p: p[0])
         return self
 
-    def get_tex_string(self):
+    def get_tex_string(self) -> str:
         return self.tex_string
 
     def init_colors(self, propagate_colors=True):
@@ -255,13 +259,13 @@ def construct(self):
 
     def __init__(
         self,
-        *tex_strings,
+        *tex_strings: str,
         arg_separator: str = " ",
         substrings_to_isolate: Iterable[str] | None = None,
         tex_to_color_map: dict[str, ManimColor] = None,
         tex_environment: str = "align*",
-        **kwargs,
-    ):
+        **kwargs: Any,
+    ) -> None:
         self.tex_template = kwargs.pop("tex_template", config["tex_template"])
         self.arg_separator = arg_separator
         self.substrings_to_isolate = (
@@ -447,8 +451,12 @@ class Tex(MathTex):
     """
 
     def __init__(
-        self, *tex_strings, arg_separator="", tex_environment="center", **kwargs
-    ):
+        self,
+        *tex_strings: str,
+        arg_separator: str = "",
+        tex_environment: str = "center",
+        **kwargs: Any,
+    ) -> None:
         super().__init__(
             *tex_strings,
             arg_separator=arg_separator,

manim/mobject/value_tracker.py

@@ -5,12 +5,17 @@
 __all__ = ["ValueTracker", "ComplexValueTracker"]
 
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from manim.mobject.mobject import Mobject
 from manim.mobject.opengl.opengl_compatibility import ConvertToOpenGL
 from manim.utils.paths import straight_path
 
+if TYPE_CHECKING:
+    from typing_extensions import Any, Self
+
 
 class ValueTracker(Mobject, metaclass=ConvertToOpenGL):
     """A mobject that can be used for tracking (real-valued) parameters.
@@ -69,7 +74,7 @@ def construct(self):
 
     """
 
-    def __init__(self, value=0, **kwargs):
+    def __init__(self, value: float = 0, **kwargs: Any) -> None:
         super().__init__(**kwargs)
         self.set(points=np.zeros((1, 3)))
         self.set_value(value)
@@ -78,7 +83,7 @@ def get_value(self) -> float:
         """Get the current value of this ValueTracker."""
         return self.points[0, 0]
 
-    def set_value(self, value: float):
+    def set_value(self, value: float) -> Self:
         """Sets a new scalar value to the ValueTracker"""
         self.points[0, 0] = value
         return self

manim/renderer/cairo_renderer.py

@@ -17,7 +17,7 @@
     from typing import Any
 
     from manim.animation.animation import Animation
     from manim.scene.scene import Scene
 
     from ..typing import PixelArray
 
@@ -33,11 +33,11 @@
 
     def __init__(
         self,
-        file_writer_class=SceneFileWriter,
-        camera_class=None,
-        skip_animations=False,
-        **kwargs,
-    ):
+        file_writer_class: type[SceneFileWriter] = SceneFileWriter,
+        camera_class: Camera | None = None,
+        skip_animations: bool = False,
+        **kwargs: Any,
+    ) -> None:
         # All of the following are set to EITHER the value passed via kwargs,
         # OR the value stored in the global config dict at the time of
         # _instance construction_.
@@ -51,7 +51,7 @@
         self.time = 0
         self.static_image = None
 
-    def init_scene(self, scene):
+    def init_scene(self, scene: Scene) -> None:
         self.file_writer: Any = self._file_writer_class(
             self,
             scene.__class__.__name__,
@@ -119,12 +119,12 @@
 
     def update_frame(  # TODO Description in Docstring
         self,
-        scene,
+        scene: Scene,
         mobjects: typing.Iterable[Mobject] | None = None,
         include_submobjects: bool = True,
         ignore_skipping: bool = True,
-        **kwargs,
-    ):
+        **kwargs: Any,
+    ) -> None:
         """Update the frame.
 
         Parameters
@@ -263,7 +263,7 @@
             self.skip_animations = True
             raise EndSceneEarlyException()
 
-    def scene_finished(self, scene):
+    def scene_finished(self, scene: Scene) -> None:
         # If no animations in scene, render an image instead
         if self.num_plays:
             self.file_writer.finish()