def test_presets(self): tdf = RenderingTaskDefinition() tdf.subtasks_count = 12 tdf.options = "Some option" tdf.optimize_total = True tdf.verification_options = "Option" tdf.resolution = [1111, 2222] tdf.output_format = ".exr" preset = tdf.make_preset() assert len(preset) == 6 assert preset["subtasks_count"] == 12 assert preset["options"] == "Some option" assert preset["optimize_total"] assert preset["verification_options"] == "Option" assert preset["output_format"] == ".exr" assert preset["resolution"] == [1111, 2222] tdf2 = RenderingTaskDefinition() assert tdf2.subtasks_count == 0 assert tdf2.options is None assert not tdf2.optimize_total assert tdf2.verification_options is None assert tdf2.resolution == [0, 0] assert tdf2.output_format == "" tdf2.load_preset(preset) assert tdf2.subtasks_count == 12 assert tdf2.options == "Some option" assert tdf2.optimize_total assert tdf2.verification_options == "Option" assert tdf2.resolution == [1111, 2222] assert tdf2.output_format == ".exr"
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() program_file = self.temp_file_name('program') 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 = [2, 300] self.bt = BlenderRenderTask( 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 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 _get_frame_task(self, use_frames=True, num_tasks=3): files_ = self.additional_dir_content([3]) rt = RenderingTaskDefinition() rt.options = FrameRendererOptions() rt.options.use_frames = use_frames rt.options.frames = list(range(6)) rt.main_scene_file = files_[1] rt.output_format = "PNG" rt.output_file = files_[2] rt.resources = [] rt.resolution = [800, 600] rt.timeout = 3600 rt.subtask_timeout = 600 rt.estimated_memory = 1000 rt.max_price = 15 task = FrameRenderingTaskMock(files_[0], owner=Node(node_name="ABC"), task_definition=rt, total_tasks=num_tasks, root_path=self.path) dm = DirManager(self.path) task.initialize(dm) return task
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