mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from manimlib.camera.moving_camera import MovingCamera
|
|
from manimlib.scene.scene import Scene
|
|
from manimlib.utils.iterables import list_update
|
|
from manimlib.utils.family_ops import extract_mobject_family_members
|
|
|
|
|
|
class MovingCameraScene(Scene):
|
|
CONFIG = {
|
|
"camera_class": MovingCamera
|
|
}
|
|
|
|
def setup(self):
|
|
Scene.setup(self)
|
|
assert(isinstance(self.camera, MovingCamera))
|
|
self.camera_frame = self.camera.frame
|
|
# Hmm, this currently relies on the fact that MovingCamera
|
|
# willd default to a full-sized frame. Is that okay?
|
|
return self
|
|
|
|
def get_moving_mobjects(self, *animations):
|
|
moving_mobjects = Scene.get_moving_mobjects(self, *animations)
|
|
all_moving_mobjects = extract_mobject_family_members(
|
|
moving_mobjects
|
|
)
|
|
movement_indicators = self.camera.get_mobjects_indicating_movement()
|
|
for movement_indicator in movement_indicators:
|
|
if movement_indicator in all_moving_mobjects:
|
|
# When one of these is moving, the camera should
|
|
# consider all mobjects to be moving
|
|
return list_update(self.mobjects, moving_mobjects)
|
|
return moving_mobjects
|