From c19b3138084c3e636cf63f0b55a2e036bb613e99 Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Thu, 17 Mar 2016 23:53:59 -0700 Subject: [PATCH] freeze_background for scenes --- camera.py | 2 +- scene/scene.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/camera.py b/camera.py index 6ec9d1e3..39a2d7d9 100644 --- a/camera.py +++ b/camera.py @@ -41,7 +41,7 @@ class Camera(object): self.space_shape = (space_height, space_width) def init_background(self): - if self.background: + if self.background is not None: self.pixel_shape = self.background.shape[:2] else: background_rgb = color_to_int_rgb(self.background_color) diff --git a/scene/scene.py b/scene/scene.py index 196959d3..0d01b22e 100644 --- a/scene/scene.py +++ b/scene/scene.py @@ -60,6 +60,11 @@ class Scene(object): else: self.camera.reset() self.camera.capture_mobjects(mobjects, **kwargs) + + def freeze_background(self): + self.update_frame() + self.set_camera(Camera(self.get_frame())) + self.clear() ### def add(self, *mobjects):