Blender image sequences now generate a preview mp4 on completion

This commit is contained in:
Brett Williams
2023-05-27 16:01:54 -05:00
parent 55591c10e1
commit eaacbf4ec0
5 changed files with 28 additions and 5 deletions

View File

@@ -113,6 +113,24 @@ class BlenderRenderWorker(BaseRenderWorker):
total_percent = whole_frame_percent + adjusted_frame_percent
return max(total_percent, 0)
def post_processing(self):
output_dir = os.listdir(os.path.dirname(self.output_path))
if self.render_all_frames and len(output_dir) > 1:
from ..utilities.ffmpeg_helper import image_sequence_to_video
logger.info("Generating preview for image sequence")
# get proper file extension
found_output = next(obj for obj in output_dir if os.path.basename(self.output_path) in obj)
glob_pattern = self.output_path + '%04d' + ('.' + found_output.split('.')[-1] if found_output else "")
try:
image_sequence_to_video(source_glob_pattern=glob_pattern,
output_path=self.output_path + '_preview.mp4',
framerate=self.scene_info['fps'])
logger.info('Successfully generated preview image sequence')
except Exception as e:
logger.error('Error generating image sequence')
if __name__ == '__main__':