opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-02 02:35:22 +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

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