From 7ffaed0ac7c5f5cb0eec113ca82f13a63d08be2d Mon Sep 17 00:00:00 2001 From: GoThrones Date: Wed, 4 Mar 2026 18:41:22 +0530 Subject: [PATCH 1/4] Improvement to ComplexValueTracker.set_value to accept either a complex number(either in rectangular coordinate form or in polar form) or a tuple, list, string, or numpy array --- manim/mobject/value_tracker.py | 96 ++++++++++++++++++++++++++++++++-- 1 file changed, 91 insertions(+), 5 deletions(-) diff --git a/manim/mobject/value_tracker.py b/manim/mobject/value_tracker.py index 947a3c001f..c78a0a8960 100644 --- a/manim/mobject/value_tracker.py +++ b/manim/mobject/value_tracker.py @@ -7,6 +7,7 @@ from typing import TYPE_CHECKING, Any import numpy as np +from collections.abc import Sequence from manim.mobject.mobject import Mobject from manim.mobject.opengl.opengl_compatibility import ConvertToOpenGL @@ -235,8 +236,93 @@ def get_value(self) -> complex: # type: ignore [override] """Get the current value of this ComplexValueTracker as a complex number.""" return complex(*self.points[0, :2]) - def set_value(self, value: complex | float) -> Self: - """Sets a new complex value to the ComplexValueTracker.""" - z = complex(value) - self.points[0, :2] = (z.real, z.imag) - return self + def set_value( + self, + value: complex | float | int | str | Sequence[float | int] | np.ndarray = 0+0j, + mode: str = "rectangular", # "rectangular" or "polar" + angle_unit: str = "radians" # "radians" or "degrees" — only used when mode="polar" + ) -> Self: + """ + Sets a new complex value to the ComplexValueTracker. + + Parameters + ---------- + value : complex | float | int | str | Sequence[float | int] | np.ndarray + The value to set. It can be: + - a complex number: 2+3j + - a float or int: 5.0 or 5 + - a valid numeric string: "23" or "2+3j" + - a sequence of exactly 2 real numbers: (2, 3), [2, 3], np.array([2, 3]) + - if mode="rectangular": interpreted as (x, y) + - if mode="polar": interpreted as (r, theta) + - theta can be in radians or degrees, specified by angle_unit + mode : str + "rectangular" (default) or "polar". + Only relevant when value is a sequence. + angle_unit : str + "radians" (default) or "degrees". + Only relevant when mode="polar". + If "degrees", theta is converted to radians internally. + + Examples + -------- + set_value(2+3j) # rectangular complex + set_value((2, 3)) # rectangular sequence + set_value((1, 90), mode="polar", angle_unit="degrees") # polar, degrees + set_value((1, np.pi/2), mode="polar") # polar, radians + """ + + # validate mode + if mode not in ("rectangular", "polar"): + raise ValueError( + f"mode must be 'rectangular' or 'polar', got '{mode}'" + ) + + # validate angle_unit + if angle_unit not in ("radians", "degrees"): + raise ValueError( + f"angle_unit must be 'radians' or 'degrees', got '{angle_unit}'" + ) + + if isinstance(value, (list, tuple, np.ndarray)): + # length check + if len(value) != 2: + raise ValueError( + f"Expected exactly 2 numbers, got {len(value)}" + ) + # check for type of number provided and finiteness check + if not all(np.isreal(v) and np.isfinite(v) for v in value): + raise TypeError( + f"Elements must be real and finite numbers — no NAN(Not a Number) or infinity is allowed" + ) + a, b = value + + if mode == "polar": + r, theta = a, b + if r < 0: + raise ValueError( + f"Radius r must be non-negative in polar form, got {r}" + ) + # convert degrees to radians if needed + if angle_unit == "degrees": + theta = np.deg2rad(theta) + x = r * np.cos(theta) + y = r * np.sin(theta) + else: # rectangular + x, y = a, b + + else: + z = complex(value) # handles complex, float, int, valid strings + # check real and imag parts individually for finiteness + if not np.isfinite(z.real): + raise ValueError( + f"Real part must be finite, got {z.real}" + ) + if not np.isfinite(z.imag): + raise ValueError( + f"Imaginary part must be finite, got {z.imag}" + ) + x, y = z.real, z.imag + + self.points[0, :2] = (x, y) + return self \ No newline at end of file From 17e9ad195dac78ae44a36c65149b5fec9e0286c7 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 4 Mar 2026 18:28:47 +0000 Subject: [PATCH 2/4] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- manim/mobject/value_tracker.py | 28 ++++++++++------------------ 1 file changed, 10 insertions(+), 18 deletions(-) diff --git a/manim/mobject/value_tracker.py b/manim/mobject/value_tracker.py index c78a0a8960..aa190c5d5e 100644 --- a/manim/mobject/value_tracker.py +++ b/manim/mobject/value_tracker.py @@ -4,10 +4,10 @@ __all__ = ["ValueTracker", "ComplexValueTracker"] +from collections.abc import Sequence from typing import TYPE_CHECKING, Any import numpy as np -from collections.abc import Sequence from manim.mobject.mobject import Mobject from manim.mobject.opengl.opengl_compatibility import ConvertToOpenGL @@ -238,9 +238,10 @@ def get_value(self) -> complex: # type: ignore [override] def set_value( self, - value: complex | float | int | str | Sequence[float | int] | np.ndarray = 0+0j, + value: complex | float | int | str | Sequence[float | int] | np.ndarray = 0 + + 0j, mode: str = "rectangular", # "rectangular" or "polar" - angle_unit: str = "radians" # "radians" or "degrees" — only used when mode="polar" + angle_unit: str = "radians", # "radians" or "degrees" — only used when mode="polar" ) -> Self: """ Sets a new complex value to the ComplexValueTracker. @@ -271,12 +272,9 @@ def set_value( set_value((1, 90), mode="polar", angle_unit="degrees") # polar, degrees set_value((1, np.pi/2), mode="polar") # polar, radians """ - # validate mode if mode not in ("rectangular", "polar"): - raise ValueError( - f"mode must be 'rectangular' or 'polar', got '{mode}'" - ) + raise ValueError(f"mode must be 'rectangular' or 'polar', got '{mode}'") # validate angle_unit if angle_unit not in ("radians", "degrees"): @@ -287,13 +285,11 @@ def set_value( if isinstance(value, (list, tuple, np.ndarray)): # length check if len(value) != 2: - raise ValueError( - f"Expected exactly 2 numbers, got {len(value)}" - ) + raise ValueError(f"Expected exactly 2 numbers, got {len(value)}") # check for type of number provided and finiteness check if not all(np.isreal(v) and np.isfinite(v) for v in value): raise TypeError( - f"Elements must be real and finite numbers — no NAN(Not a Number) or infinity is allowed" + "Elements must be real and finite numbers — no NAN(Not a Number) or infinity is allowed" ) a, b = value @@ -315,14 +311,10 @@ def set_value( z = complex(value) # handles complex, float, int, valid strings # check real and imag parts individually for finiteness if not np.isfinite(z.real): - raise ValueError( - f"Real part must be finite, got {z.real}" - ) + raise ValueError(f"Real part must be finite, got {z.real}") if not np.isfinite(z.imag): - raise ValueError( - f"Imaginary part must be finite, got {z.imag}" - ) + raise ValueError(f"Imaginary part must be finite, got {z.imag}") x, y = z.real, z.imag self.points[0, :2] = (x, y) - return self \ No newline at end of file + return self From 44041423eef69673784a7ad938afe1078fbdd76d Mon Sep 17 00:00:00 2001 From: GoThrones Date: Thu, 5 Mar 2026 08:56:52 +0530 Subject: [PATCH 3/4] fixed: mypy type error in ComplexValueTracker.set_value --- manim/mobject/value_tracker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/manim/mobject/value_tracker.py b/manim/mobject/value_tracker.py index c78a0a8960..380e94ea60 100644 --- a/manim/mobject/value_tracker.py +++ b/manim/mobject/value_tracker.py @@ -4,7 +4,7 @@ __all__ = ["ValueTracker", "ComplexValueTracker"] -from typing import TYPE_CHECKING, Any +from typing import TYPE_CHECKING, Any, Union, cast import numpy as np from collections.abc import Sequence @@ -312,6 +312,7 @@ def set_value( x, y = a, b else: + value = cast(Union[complex, float, int, str], value) z = complex(value) # handles complex, float, int, valid strings # check real and imag parts individually for finiteness if not np.isfinite(z.real): From 1bbd1fcba417b03431ecadc753e5f97963c27de6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 5 Mar 2026 03:36:41 +0000 Subject: [PATCH 4/4] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- manim/mobject/value_tracker.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/manim/mobject/value_tracker.py b/manim/mobject/value_tracker.py index 64e01e4da7..381241f4ba 100644 --- a/manim/mobject/value_tracker.py +++ b/manim/mobject/value_tracker.py @@ -5,7 +5,8 @@ __all__ = ["ValueTracker", "ComplexValueTracker"] from collections.abc import Sequence -from typing import TYPE_CHECKING, Any, Union, cast +from typing import TYPE_CHECKING, Any, cast + import numpy as np from manim.mobject.mobject import Mobject @@ -237,7 +238,8 @@ def get_value(self) -> complex: # type: ignore [override] def set_value( self, - value: complex | float | int | str | Sequence[float | int] | np.ndarray = 0 + 0j, + value: complex | float | int | str | Sequence[float | int] | np.ndarray = 0 + + 0j, mode: str = "rectangular", # "rectangular" or "polar" angle_unit: str = "radians", # "radians" or "degrees" — only used when mode="polar" ) -> Self: @@ -306,7 +308,7 @@ def set_value( x, y = a, b else: - value = cast(Union[complex, float, int, str], value) + value = cast(complex | float | int | str, value) z = complex(value) # handles complex, float, int, valid strings # check real and imag parts individually for finiteness if not np.isfinite(z.real):