From a3579eab419563d1d212e681305de9da6c9f74ba Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Wed, 20 Apr 2022 21:51:18 -0700 Subject: [PATCH] Have SceneFileWriter handle a location for saved mobjects --- manimlib/scene/scene_file_writer.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/manimlib/scene/scene_file_writer.py b/manimlib/scene/scene_file_writer.py index cb948ab5..9be88487 100644 --- a/manimlib/scene/scene_file_writer.py +++ b/manimlib/scene/scene_file_writer.py @@ -78,6 +78,8 @@ class SceneFileWriter(object): self.partial_movie_directory = guarantee_existence(os.path.join( movie_dir, "partial_movie_files", scene_name, )) + # A place to save mobjects + self.saved_mobject_directory = os.path.join(out_dir, "mobjects") def get_default_module_directory(self) -> str: path, _ = os.path.splitext(self.input_file_path) @@ -119,6 +121,12 @@ class SceneFileWriter(object): def get_movie_file_path(self) -> str: return self.movie_file_path + def get_saved_mobject_directory(self) -> str: + return guarantee_existence(os.path.join( + self.saved_mobject_directory, + str(self.scene), + )) + # Sound def init_audio(self) -> None: self.includes_sound: bool = False