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

Gave control of ''image'' subdirectory to Scene

Browse Source
This commit is contained in:
Grant Sanderson
2015-12-23 14:51:19 -08:00
parent 240742a613
commit c6e3db2529
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extract_scene.py
View File

@ -101,9 +101,7 @@ def handle_scene(scene, **config):
if config["save_image"]:
if not config["write_all"]:
scene.show_frame()
path = os.path.join(MOVIE_DIR, config["movie_prefix"], "images")
if not os.path.exists(path):
os.makedirs(path)
path = os.path.join(MOVIE_DIR, config["movie_prefix"])
scene.save_image(path, name)
if config["write"]:
scene.write_to_movie(os.path.join(config["movie_prefix"], name))

8
scene/scene.py
View File

@ -243,8 +243,12 @@ class Scene(object):
TkSceneRoot(self)
def save_image(self, directory = MOVIE_DIR, name = None):
path = os.path.join(directory, name or str(self)) + ".png"
Image.fromarray(self.get_frame()).save(path)
path = os.path.join(directory, "images")
file_name = (name or str(self)) + ".png"
full_path = os.path.join(path, file_name)
if not os.path.exists(path):
os.makedirs(path)
Image.fromarray(self.get_frame()).save(full_path)
# To list possible args that subclasses have
# Elements should always be a tuple