Update Scene.save_state and Scene.restore

3b1b · Apr 21, 2022 · 97400a5 · 97400a5
1 parent cb768c2
commit 97400a5
Showing 1 changed file with 8 additions and 12 deletions.
diff --git a/manimlib/scene/scene.py b/manimlib/scene/scene.py
@@ -575,22 +575,18 @@ def add_sound(
 
     # Helpers for interactive development
     def save_state(self) -> None:
-        self.saved_state = {
-            "mobjects": self.mobjects,
-            "mobject_states": [
-                mob.copy()
-                for mob in self.mobjects
-            ],
-        }
+        self.saved_state = [
+            (mob, mob.copy())
+            for mob in self.mobjects
+        ]
 
     def restore(self) -> None:
         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
+        self.mobjects = []
+        for mob, mob_state in self.saved_state:
+            mob.become(mob_state)
+            self.mobjects.append(mob)
 
     # Event handling