mirror of
https://github.com/3b1b/manim.git
synced 2025-08-05 16:49:03 +00:00
138 lines
4.3 KiB
Python
138 lines
4.3 KiB
Python
|
|
from mobject.vectorized_mobject import VMobject, VGroup, VectorizedPoint
|
|
from mobject.tex_mobject import TexMobject
|
|
from animation import Animation
|
|
from animation.continual_animation import ContinualAnimation
|
|
from scene import Scene
|
|
from helpers import *
|
|
|
|
class DecimalNumber(VMobject):
|
|
CONFIG = {
|
|
"num_decimal_points" : 2,
|
|
"digit_to_digit_buff" : 0.05,
|
|
"show_ellipsis" : False,
|
|
"unit" : None,
|
|
}
|
|
def __init__(self, number, **kwargs):
|
|
digest_config(self, kwargs, locals())
|
|
num_string = '%.*f'%(self.num_decimal_points, number)
|
|
negative_zero_string = "-%.*f"%(self.num_decimal_points, 0.)
|
|
if num_string == negative_zero_string:
|
|
num_string = num_string[1:]
|
|
VMobject.__init__(self, *[
|
|
TexMobject(char)
|
|
for char in num_string
|
|
], **kwargs)
|
|
|
|
if self.show_ellipsis:
|
|
self.add(TexMobject("\\dots"))
|
|
|
|
if self.unit is not None:
|
|
self.add(TexMobject(self.unit))
|
|
|
|
self.arrange_submobjects(
|
|
buff = self.digit_to_digit_buff,
|
|
aligned_edge = DOWN
|
|
)
|
|
|
|
if num_string.startswith("-"):
|
|
minus = self.submobjects[0]
|
|
minus.next_to(
|
|
self.submobjects[1], LEFT,
|
|
buff = self.digit_to_digit_buff
|
|
)
|
|
if self.unit == "\\circ":
|
|
self[-1].align_to(self, UP)
|
|
|
|
class Integer(VGroup):
|
|
CONFIG = {
|
|
"digit_buff" : 0.8*SMALL_BUFF
|
|
}
|
|
def __init__(self, integer, **kwargs):
|
|
self.number = integer
|
|
num_str = str(integer)
|
|
VGroup.__init__(self, *map(TexMobject, num_str), **kwargs)
|
|
self.arrange_submobjects(
|
|
RIGHT, buff = self.digit_buff, aligned_edge = DOWN
|
|
)
|
|
if num_str[0] == "-":
|
|
self[0].next_to(self[1], LEFT, buff = SMALL_BUFF)
|
|
|
|
class ChangingDecimal(Animation):
|
|
CONFIG = {
|
|
"num_decimal_points" : None,
|
|
"show_ellipsis" : None,
|
|
"position_update_func" : None,
|
|
"tracked_mobject" : None
|
|
}
|
|
def __init__(self, decimal_number_mobject, number_update_func, **kwargs):
|
|
digest_config(self, kwargs, locals())
|
|
self.decimal_number_config = dict(
|
|
decimal_number_mobject.initial_config
|
|
)
|
|
for attr in "num_decimal_points", "show_ellipsis":
|
|
value = getattr(self, attr)
|
|
if value is not None:
|
|
self.decimal_number_config[attr] = value
|
|
if self.tracked_mobject:
|
|
dmc = decimal_number_mobject.get_center()
|
|
tmc = self.tracked_mobject.get_center()
|
|
self.diff_from_tracked_mobject = dmc - tmc
|
|
Animation.__init__(self, decimal_number_mobject, **kwargs)
|
|
|
|
def update_mobject(self, alpha):
|
|
self.update_number(alpha)
|
|
self.update_position()
|
|
|
|
def update_number(self, alpha):
|
|
decimal = self.decimal_number_mobject
|
|
new_number = self.number_update_func(alpha)
|
|
new_decimal = DecimalNumber(
|
|
new_number, **self.decimal_number_config
|
|
)
|
|
new_decimal.match_height(decimal)
|
|
new_decimal.move_to(decimal)
|
|
new_decimal.match_style(decimal)
|
|
|
|
decimal.submobjects = new_decimal.submobjects
|
|
decimal.number = new_number
|
|
|
|
def update_position(self):
|
|
if self.position_update_func is not None:
|
|
self.position_update_func(self.decimal_number_mobject)
|
|
elif self.tracked_mobject is not None:
|
|
self.decimal_number_mobject.move_to(self.tracked_mobject.get_center() + self.diff_from_tracked_mobject)
|
|
|
|
class ChangeDecimalToValue(ChangingDecimal):
|
|
def __init__(self, decimal_number_mobject, target_number, **kwargs):
|
|
start_number = decimal_number_mobject.number
|
|
func = lambda alpha : interpolate(start_number, target_number, alpha)
|
|
ChangingDecimal.__init__(self, decimal_number_mobject, func, **kwargs)
|
|
|
|
class ContinualChangingDecimal(ContinualAnimation):
|
|
def __init__(self, decimal_number_mobject, number_update_func, **kwargs):
|
|
self.anim = ChangingDecimal(decimal_number_mobject, number_update_func, **kwargs)
|
|
ContinualAnimation.__init__(self, decimal_number_mobject, **kwargs)
|
|
|
|
def update_mobject(self, dt):
|
|
self.anim.update(self.internal_time)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|