brett/Zordon
1
0
Fork 0
mirror of https://github.com/blw1138/Zordon.git synced 2025-12-17 16:58:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

More worker cleanups and improvements

Browse Source
This commit is contained in:
Brett Williams
2022-10-31 00:03:27 -07:00
parent a36f6d3128
commit b244a258e2
4 changed files with 30 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
utilities/blender_worker.py
View File

@@ -22,13 +22,10 @@ class BlenderRenderWorker(BaseRenderWorker):
self.export_format = self.args.get('export_format', None) or 'JPEG'
self.camera = self.args.get('camera', None)
self.render_all_frames = self.args.get('render_all_frames', False)
self.frame = 0 #todo: remove this attribute
self.frame_to_render = 0
# Stats
self.memory_use = None
self.time_elapsed = None
self.time_remaining = None
self.frame_percent_complete = 0.0
self.__frame_percent_complete = 0.0
# Scene Info
self.scene_info = get_scene_info(input_path)
@@ -51,7 +48,10 @@ class BlenderRenderWorker(BaseRenderWorker):
def _generate_subprocess(self):
cmd = [self.renderer_path(), '-b', self.input_path]
cmd = [self.renderer_path()]
if self.args.get('background', True): # optionally run render not in background
cmd.append('-b')
cmd.append(self.input_path)
if self.camera:
cmd.extend(['--python-expr', f"import bpy;bpy.context.scene.camera = bpy.data.objects['{self.camera}'];"])
@@ -59,7 +59,7 @@ class BlenderRenderWorker(BaseRenderWorker):
cmd.extend(['-E', self.engine, '-o', self.output_path, '-F', self.export_format])
# all frames or single
cmd.extend(['-a'] if self.render_all_frames else ['-f', str(self.frame)])
cmd.extend(['-a'] if self.render_all_frames else ['-f', str(self.frame_to_render)])
return cmd
@@ -70,14 +70,14 @@ class BlenderRenderWorker(BaseRenderWorker):
found = pattern.search(line)
if found:
stats = found.groupdict()
self.memory_use = stats['memory']
self.time_elapsed = stats['time']
self.time_remaining = stats['remaining'] or 'Unknown'
memory_use = stats['memory']
time_elapsed = stats['time']
time_remaining = stats['remaining'] or 'Unknown'
sample_string = line.split('|')[-1].strip()
if "sample" in sample_string.lower():
samples = re.sub(r'[^\d/]', '', sample_string)
self.frame_percent_complete = int(samples.split('/')[0]) / int(samples.split('/')[-1])
self.__frame_percent_complete = int(samples.split('/')[0]) / int(samples.split('/')[-1])
# Calculate rough percent based on cycles
# EEVEE
@@ -93,8 +93,8 @@ class BlenderRenderWorker(BaseRenderWorker):
if int(stats['frame']) > self.current_frame:
self.current_frame = int(stats['frame'])
logger.debug(
'Frame:{0} | Mem:{1} | Time:{2} | Remaining:{3}'.format(self.current_frame, self.memory_use,
self.time_elapsed, self.time_remaining))
'Frame:{0} | Mem:{1} | Time:{2} | Remaining:{3}'.format(self.current_frame, memory_use,
time_elapsed, time_remaining))
elif 'error' in line.lower():
logger.error(line)
self.errors.append(line)
@@ -115,10 +115,10 @@ class BlenderRenderWorker(BaseRenderWorker):
def percent_complete(self):
if self.total_frames <= 1:
return self.frame_percent_complete
return self.__frame_percent_complete
else:
whole_frame_percent = (self.current_frame - 1) / self.total_frames
adjusted_frame_percent = self.frame_percent_complete / self.total_frames
adjusted_frame_percent = self.__frame_percent_complete / self.total_frames
total_percent = whole_frame_percent + adjusted_frame_percent
return max(total_percent, 0)