public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-19 04:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Get rid of old methods

Browse source
This commit is contained in:
Grant Sanderson 2020-02-22 13:20:22 -08:00
parent ed6085d07c
commit 7b198545bb
1 changed files with 24 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
manimlib/scene/scene.py
View file

@ -35,6 +35,7 @@ class Scene(Container):
"end_at_animation_number": None, "end_at_animation_number": None,
"leave_progress_bars": False, "leave_progress_bars": False,
"preview": True, "preview": True,
"linger_after_completion": True,
} }
def __init__(self, **kwargs): def __init__(self, **kwargs):
@ -62,6 +63,7 @@ class Scene(Container):
self.zoom_on_scroll = False self.zoom_on_scroll = False
self.quit_interaction = False self.quit_interaction = False
# Much nice to work with deterministic scenes
if self.random_seed is not None: if self.random_seed is not None:
random.seed(self.random_seed) random.seed(self.random_seed)
np.random.seed(self.random_seed) np.random.seed(self.random_seed)
@ -83,14 +85,14 @@ class Scene(Container):
pass pass
def construct(self): def construct(self):
# Where all the animation happens
# To be implemented in subclasses # To be implemented in subclasses
pass pass
def tear_down(self): def tear_down(self):
self.stop_skipping() self.stop_skipping()
self.file_writer.finish() self.file_writer.finish()
self.print_end_message() if self.window and self.linger_after_completion:
if self.window:
self.interact() self.interact()
def interact(self): def interact(self):
@ -105,6 +107,7 @@ class Scene(Container):
def embed(self): def embed(self):
self.stop_skipping() self.stop_skipping()
self.linger_after_completion = False
self.update_frame() self.update_frame()
shell = InteractiveShellEmbed() shell = InteractiveShellEmbed()
@ -117,27 +120,6 @@ class Scene(Container):
def __str__(self): def __str__(self):
return self.__class__.__name__ return self.__class__.__name__
def print_end_message(self):
print(f"Played {self.num_plays} animations")
def set_variables_as_attrs(self, *objects, **newly_named_objects):
"""
This method is slightly hacky, making it a little easier
for certain methods (typically subroutines of construct)
to share local variables.
"""
caller_locals = inspect.currentframe().f_back.f_locals
for key, value in list(caller_locals.items()):
for o in objects:
if value is o:
setattr(self, key, value)
for key, value in list(newly_named_objects.items()):
setattr(self, key, value)
return self
def get_attrs(self, *keys):
return [getattr(self, key) for key in keys]
# Only these methods should touch the camera # Only these methods should touch the camera
def get_image(self): def get_image(self):
return self.camera.get_image() return self.camera.get_image()
@ -516,6 +498,25 @@ class Scene(Container):
self.update_frame(ignore_skipping=True) self.update_frame(ignore_skipping=True)
self.get_image().show() self.get_image().show()
# Helpers for interactive development
def save_state(self):
self.saved_state = {
"mobjects": self.mobjects,
"mobject_states": [
mob.copy()
for mob in self.mobjects
],
}
def restore(self):
if not hasattr(self, "saved_state"):
raise Exception("Trying to restore scene without having saved")
mobjects = self.saved_state["mobjects"]
states = self.saved_state["mobject_states"]
for mob, state in zip(mobjects, states):
mob.become(state)
self.mobjects = mobjects
# Event handling # Event handling
def on_mouse_motion(self, point, d_point): def on_mouse_motion(self, point, d_point):
self.mouse_point.move_to(point) self.mouse_point.move_to(point)