mirror of
https://github.com/3b1b/manim.git
synced 2025-08-05 16:49:03 +00:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from scene import Scene
|
||
|
|
from animation.transform import Transform
|
||
|
|
|
||
|
|
from helpers import *
|
||
|
|
|
||
|
|
class ReconfigurableScene(Scene):
|
||
|
|
CONFIG = {
|
||
|
|
"allow_recursion" : True,
|
||
|
|
}
|
||
|
|
def setup(self):
|
||
|
|
self.states = []
|
||
|
|
self.num_recursions = 0
|
||
|
|
|
||
|
|
def show_alt_config(self, return_to_original_configuration = True, **new_config):
|
||
|
|
original_state = self.get_mobjects()
|
||
|
|
state_copy = [m.copy() for m in original_state]
|
||
|
|
self.states.append(state_copy)
|
||
|
|
if not self.allow_recursion:
|
||
|
|
return
|
||
|
|
alt_scene = self.__class__(
|
||
|
|
skip_animations = True,
|
||
|
|
allow_recursion = False,
|
||
|
|
**new_config
|
||
|
|
)
|
||
|
|
alt_state = alt_scene.states[len(self.states)-1]
|
||
|
|
|
||
|
|
if return_to_original_configuration:
|
||
|
|
self.clear()
|
||
|
|
self.transition_between_states(state_copy, alt_state)
|
||
|
|
self.transition_between_states(state_copy, original_state)
|
||
|
|
self.clear()
|
||
|
|
self.add(*original_state)
|
||
|
|
else:
|
||
|
|
self.transition_between_states(original_state, alt_state)
|
||
|
|
self.__dict__.update(new_config)
|
||
|
|
|
||
|
|
def transition_between_states(self, start_state, target_state):
|
||
|
|
self.play(*[
|
||
|
|
Transform(*pair)
|
||
|
|
for pair in zip(start_state, target_state)
|
||
|
|
])
|
||
|
|
self.dither()
|
||
|
|
|
||
|
|
|
||
|
|
|