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