def _create_blender_task(self, dir_manager):
definition = RenderingTaskDefinition()
definition.options = BlenderRendererOptions()
definition.task_id = "deadbeef"
definition.task_type = "Blender"
task_file_name = os.path.join(self.path, 'task_file.gt')
with open(task_file_name, 'wb') as task_file:
task_file.write(jsonpickle.dumps(definition))
return task_file_name
def test_compositing(self):
task_definition = RenderingTaskDefinition()
task_definition.options = BlenderRendererOptions()
task_definition.options.use_frames = True
task_definition.options.frames = [7, 8, 10]
task_definition.main_scene_file = self.temp_file_name("example.blend")
task_definition.output_file = self.temp_file_name('output')
task_definition.output_format = 'PNG'
task_definition.resolution = [2, 300]
task_definition.task_id = "ABC"
def _get_blender_task(task_definition, total_tasks=6):
return BlenderRenderTask(
owner=Node(node_name="exmaple-node-name"),
task_definition=task_definition,
total_tasks=total_tasks,
root_path=self.tempdir,
)
# Compostiting set to False
task_definition.options.compositing = False
bt = _get_blender_task(task_definition)
assert not bt.compositing
# Compositing True, use frames, more subtasks than frames
task_definition.options.compositing = True
bt = _get_blender_task(task_definition)
assert not bt.compositing
# Compositing True, use frames, as many subtasks as frames
bt = _get_blender_task(task_definition, 3)
assert not bt.compositing
# Compositing True, use frames, less subtasks than frames
bt = _get_blender_task(task_definition, 1)
assert not bt.compositing
# Compositing True, use frames is False, as many subtask as frames
task_definition.options.use_frames = False
bt = _get_blender_task(task_definition, 3)
assert not bt.compositing
def setUp(self):
super(TestBlenderFrameTask, self).setUp()
task_definition = RenderingTaskDefinition()
task_definition.options = BlenderRendererOptions()
task_definition.options.compositing = False
task_definition.options.use_frames = True
task_definition.options.frames = [7, 8, 10]
task_definition.main_scene_file = self.temp_file_name("example.blend")
task_definition.output_file = self.temp_file_name('output')
task_definition.output_format = 'PNG'
task_definition.resolution = [200, 300]
task_definition.task_id = str(uuid.uuid4())
BlenderRenderTask.VERIFICATION_QUEUE._reset()
self.bt = BlenderRenderTask(
owner=Node(node_name="example-node-name"),
task_definition=task_definition,
total_tasks=6,
root_path=self.tempdir,
)
dm = DirManager(self.path)
self.bt.initialize(dm)
def build_bt(self, res_x, res_y, total_tasks, frames=None):
output_file = self.temp_file_name('output')
if frames is None:
use_frames = False
frames = [1]
else:
use_frames = True
task_definition = RenderingTaskDefinition()
task_definition.options = BlenderRendererOptions()
task_definition.options.use_frames = use_frames
task_definition.options.frames = frames
task_definition.output_file = output_file
task_definition.output_format = "PNG"
task_definition.resolution = [res_x, res_y]
task_definition.main_scene_file = path.join(self.path, "example.blend")
task_definition.task_id = str(uuid.uuid4())
bt = BlenderRenderTask(owner=Node(node_name="example-node-name"),
task_definition=task_definition,
total_tasks=total_tasks,
root_path=self.tempdir)
bt.initialize(DirManager(self.tempdir))
return bt