Pass Scene to play() and wait()

Author: eulertour

manim/renderer/cairo_renderer.py

@@ -8,6 +8,15 @@
 from ..utils.caching import handle_caching_play, handle_caching_wait
 
 
+def manage_scene_reference(func):
+    def wrapper(self, scene, *args, **kwargs):
+        setattr(self, "temp_scene_ref", scene)
+        func(self, *args, **kwargs)
+        delattr(self, "temp_scene_ref")
+
+    return wrapper
+
+
 def handle_play_like_call(func):
     """
     This method is used internally to wrap the
@@ -74,11 +83,13 @@ def __init__(self, scene, camera_class, **kwargs):
         self.num_plays = 0
         self.time = 0
 
+    @manage_scene_reference
     @handle_caching_play
     @handle_play_like_call
     def play(self, *args, **kwargs):
         self.scene.play_internal(*args, **kwargs)
 
+    @manage_scene_reference
     @handle_caching_wait
     @handle_play_like_call
     def wait(self, duration=DEFAULT_WAIT_TIME, stop_condition=None):

manim/scene/scene.py

@@ -761,10 +761,10 @@ def finish_animations(self, animations):
             self.update_mobjects(0)
 
     def wait(self, duration=DEFAULT_WAIT_TIME, stop_condition=None):
-        self.renderer.wait(duration=duration, stop_condition=stop_condition)
+        self.renderer.wait(self, duration=duration, stop_condition=stop_condition)
 
     def play(self, *args, **kwargs):
-        self.renderer.play(*args, **kwargs)
+        self.renderer.play(self, *args, **kwargs)
 
     def play_internal(self, *args, **kwargs):
         """