public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Self type to numbers.py

Browse source
This commit is contained in:
Grant Sanderson 2023-01-31 14:16:15 -08:00
parent 74a11bb05c
commit 576d8c996b
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
manimlib/mobject/numbers.py
View file

@ -11,7 +11,7 @@ from manimlib.mobject.types.vectorized_mobject import VMobject
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
if TYPE_CHECKING: if TYPE_CHECKING:
from typing import TypeVar from typing import TypeVar, Self
from manimlib.typing import ManimColor, Vect3 from manimlib.typing import ManimColor, Vect3
T = TypeVar("T", bound=VMobject) T = TypeVar("T", bound=VMobject)
@ -163,7 +163,7 @@ class DecimalNumber(VMobject):
def get_tex(self): def get_tex(self):
return self.num_string return self.num_string
def set_value(self, number: float | complex): def set_value(self, number: float | complex) -> Self:
move_to_point = self.get_edge_center(self.edge_to_fix) move_to_point = self.get_edge_center(self.edge_to_fix)
style = self.family_members_with_points()[0].get_style() style = self.family_members_with_points()[0].get_style()
self.set_submobjects_from_number(number) self.set_submobjects_from_number(number)
@ -171,14 +171,16 @@ class DecimalNumber(VMobject):
self.set_style(**style) self.set_style(**style)
return self return self
def _handle_scale_side_effects(self, scale_factor: float) -> None: def _handle_scale_side_effects(self, scale_factor: float) -> Self:
self.uniforms["font_size"] = scale_factor * self.uniforms["font_size"] self.uniforms["font_size"] = scale_factor * self.uniforms["font_size"]
return self
def get_value(self) -> float | complex: def get_value(self) -> float | complex:
return self.number return self.number
def increment_value(self, delta_t: float | complex = 1) -> None: def increment_value(self, delta_t: float | complex = 1) -> Self:
self.set_value(self.get_value() + delta_t) self.set_value(self.get_value() + delta_t)
return self
class Integer(DecimalNumber): class Integer(DecimalNumber):