public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Change redered filed directory names

Browse source 
Default behavior is now to include the render quality in the name of the directory where files are written, so that low quality tests don't necessarily clobber the existing HD versions.
This commit is contained in:
Grant Sanderson 2018-03-06 13:39:59 -08:00
parent 1771b622ac
commit 27eaa322fb
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
scene/scene.py
View file

@ -40,7 +40,7 @@ class Scene(Container):
"random_seed" : 0, "random_seed" : 0,
"start_at_animation_number" : None, "start_at_animation_number" : None,
"end_at_animation_number" : None, "end_at_animation_number" : None,
"include_render_quality_in_name" : False, #TODO, nothing uses this right now "include_render_quality_in_output_directory" : True,
} }
def __init__(self, **kwargs): def __init__(self, **kwargs):
Container.__init__(self, **kwargs) # Perhaps allow passing in a non-empty *mobjects parameter? Container.__init__(self, **kwargs) # Perhaps allow passing in a non-empty *mobjects parameter?
@ -56,8 +56,6 @@ class Scene(Container):
self.original_skipping_status = self.skip_animations self.original_skipping_status = self.skip_animations
if self.name is None: if self.name is None:
self.name = self.__class__.__name__ self.name = self.__class__.__name__
if self.include_render_quality_in_name:
self.name += str(self.camera.pixel_shape[0])
if self.random_seed is not None: if self.random_seed is not None:
random.seed(self.random_seed) random.seed(self.random_seed)
np.random.seed(self.random_seed) np.random.seed(self.random_seed)
@ -552,15 +550,22 @@ class Scene(Container):
image.save(path) image.save(path)
def get_movie_file_path(self, name = None, extension = None): def get_movie_file_path(self, name = None, extension = None):
directory = self.output_directory
if self.include_render_quality_in_output_directory:
directory += "_%dp%d"%(
self.camera.pixel_shape[0],
int(1.0/self.frame_duration)
)
if not os.path.exists(directory):
os.makedirs(directory)
if extension is None: if extension is None:
extension = self.movie_file_extension extension = self.movie_file_extension
if name is None: if name is None:
name = self.name name = self.name
file_path = os.path.join(self.output_directory, name) file_path = os.path.join(directory, name)
if not file_path.endswith(extension): if not file_path.endswith(extension):
file_path += extension file_path += extension
if not os.path.exists(self.output_directory):
os.makedirs(self.output_directory)
return file_path return file_path
def open_movie_pipe(self): def open_movie_pipe(self):