Added always_shift and always_rotate

This commit is contained in:
Grant Sanderson 2019-02-11 13:17:23 -08:00
parent 62c4967f21
commit a73cbb8014

View file

@ -1,5 +1,13 @@
import inspect
from manimlib.constants import DEGREES
from manimlib.constants import RIGHT
from manimlib.mobject.mobject import Mobject
def always(method, *args, **kwargs):
assert(inspect.ismethod(method))
mobject = method.__self__
assert(isinstance(mobject, Mobject))
func = method.__func__
mobject.add_updater(lambda m: func(m, *args, **kwargs))
@ -8,3 +16,15 @@ def always_redraw(func):
mob = func()
mob.add_updater(lambda m: mob.become(func()))
return mob
def always_shift(mobject, direction=RIGHT, rate=0.1):
mobject.add_updater(
lambda m, dt: m.shift(dt * rate * direction)
)
def always_rotate(mobject, rate=20 * DEGREES, **kwargs):
mobject.add_updater(
lambda m, dt: m.rotate(dt * rate, **kwargs)
)