opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 22:03:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update extract_scene.py args to better support previewing, showing in finder, opening output files, etc.

Browse Source
This commit is contained in:
Grant Sanderson
2018-01-19 17:31:31 -08:00
parent 4919dbf652
commit 4e468af656
2 changed files with 38 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
scene/scene.py
View File

@ -33,6 +33,7 @@ class Scene(object):
"save_pngs" : False,
"pngs_mode" : "RGBA",
"output_directory" : ANIMATIONS_DIR,
"movie_file_extension" : ".mp4",
"name" : None,
"always_continually_update" : False,
"random_seed" : 0,
@ -470,24 +471,30 @@ class Scene(object):
def preview(self):
TkSceneRoot(self)
def save_image(self, name = None, mode = "RGB", dont_update = False):
def get_image_file_path(self, name = None, dont_update = False):
folder = "images"
if dont_update:
folder = str(self)
path = os.path.join(self.output_directory, folder)
file_name = (name or str(self)) + ".png"
full_path = os.path.join(path, file_name)
return os.path.join(path, file_name)
def save_image(self, name = None, mode = "RGB", dont_update = False):
path = self.get_image_file_path(name, dont_update)
if not os.path.exists(path):
os.makedirs(path)
if not dont_update:
self.update_frame()
image = self.get_image()
image = image.convert(mode)
image.save(full_path)
image.save(path)
def get_movie_file_path(self, name, extension):
def get_movie_file_path(self, name = None, extension = None):
if extension is None:
extension = self.movie_file_extension
if name is None:
name = self.name
file_path = os.path.join(self.output_directory, name)
if not file_path.endswith(extension):
file_path += extension
@ -497,8 +504,8 @@ class Scene(object):
def open_movie_pipe(self):
name = str(self)
file_path = self.get_movie_file_path(name, ".mp4")
temp_file_path = file_path.replace(".mp4", "Temp.mp4")
file_path = self.get_movie_file_path(name)
temp_file_path = file_path.replace(name, name + "Temp")
print("Writing to %s"%temp_file_path)
self.args_to_rename_file = (temp_file_path, file_path)