mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from manimlib.animation.animation import Animation
|
|
from manimlib.constants import ORIGIN, OUT
|
|
from manimlib.constants import PI, TAU
|
|
from manimlib.utils.rate_functions import linear
|
|
from manimlib.utils.rate_functions import smooth
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
import numpy as np
|
|
from typing import Callable
|
|
from manimlib.mobject.mobject import Mobject
|
|
|
|
|
|
class Rotating(Animation):
|
|
def __init__(
|
|
self,
|
|
mobject: Mobject,
|
|
angle: float = TAU,
|
|
axis: np.ndarray = OUT,
|
|
about_point: np.ndarray | None = None,
|
|
about_edge: np.ndarray | None = None,
|
|
run_time: float = 5.0,
|
|
rate_func: Callable[[float], float] = linear,
|
|
suspend_mobject_updating: bool = False,
|
|
**kwargs
|
|
):
|
|
self.angle = angle
|
|
self.axis = axis
|
|
self.about_point = about_point
|
|
self.about_edge = about_edge
|
|
super().__init__(
|
|
mobject,
|
|
run_time=run_time,
|
|
rate_func=rate_func,
|
|
suspend_mobject_updating=suspend_mobject_updating,
|
|
**kwargs
|
|
)
|
|
|
|
def interpolate_mobject(self, alpha: float) -> None:
|
|
pairs = zip(
|
|
self.mobject.family_members_with_points(),
|
|
self.starting_mobject.family_members_with_points(),
|
|
)
|
|
for sm1, sm2 in pairs:
|
|
sm1.data["points"][:] = sm2.data["points"]
|
|
self.mobject.rotate(
|
|
self.rate_func(alpha) * self.angle,
|
|
axis=self.axis,
|
|
about_point=self.about_point,
|
|
about_edge=self.about_edge,
|
|
)
|
|
|
|
|
|
class Rotate(Rotating):
|
|
def __init__(
|
|
self,
|
|
mobject: Mobject,
|
|
angle: float = PI,
|
|
axis: np.ndarray = OUT,
|
|
run_time: float = 1,
|
|
rate_func: Callable[[float], float] = smooth,
|
|
about_edge: np.ndarray = ORIGIN,
|
|
**kwargs
|
|
):
|
|
super().__init__(
|
|
mobject, angle, axis,
|
|
run_time=run_time,
|
|
rate_func=rate_func,
|
|
about_edge=about_edge,
|
|
**kwargs
|
|
)
|