def test_get_task_num_from_pixels(self):
offsets = generate_expected_offsets(30, 1920, 1080)
frame_offsets = generate_expected_offsets(15, 1920, 1080)
task_definition = RenderingTaskDefinition()
task_definition.options = BlenderRendererOptions()
task_definition.resolution = [1920, 1080]
for k in range(1, 31):
task_definition.options.use_frames = False
num = BlenderTaskTypeInfo.get_task_num_from_pixels(
6, offsets[k] + 1, task_definition, 30)
assert num == k
task_definition.options.use_frames = True
task_definition.options.frames = range(30)
num = BlenderTaskTypeInfo.get_task_num_from_pixels(
1, 0, task_definition, 30, k)
assert num == k
i = (k - 1) % 15 + 1
task_definition.options.frames = range(2)
num = BlenderTaskTypeInfo.get_task_num_from_pixels(
1, frame_offsets[i] + 3, task_definition, 30, (k - 1) / 15 + 1)
assert num == k
task_definition.options.use_frames = False
assert BlenderTaskTypeInfo.get_task_num_from_pixels(
5, 1200, task_definition, 30) == 30
def test_build(self):
definition = RenderingTaskDefinition()
definition.subtasks_count = 1
definition.options = BlenderRendererOptions()
builder = BlenderRenderTaskBuilder(owner=Node(),
task_definition=definition,
dir_manager=DirManager(
self.tempdir))
blender_task = builder.build()
self.assertIsInstance(blender_task, BlenderRenderTask)
def test_build(self):
definition = RenderingTaskDefinition()
definition.options = BlenderRendererOptions()
builder = BlenderRenderTaskBuilder(node_name="ABC",
task_definition=definition,
root_path=self.tempdir,
dir_manager=DirManager(
self.tempdir))
blender_task = builder.build()
self.assertIsInstance(blender_task, BlenderRenderTask)
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 _get_task_border(as_path=False):
offsets = generate_expected_offsets(30, 800, 600)
subtask = SubtaskState()
definition = RenderingTaskDefinition()
definition.options = BlenderRendererOptions()
definition.options.use_frames = False
definition.resolution = [800, 600]
for k in range(1, 31):
subtask.extra_data = {'start_task': k, 'end_task': k}
border = BlenderTaskTypeInfo.get_task_border(subtask,
definition,
30,
as_path=as_path)
assert min(border) == (0, offsets[k])
assert max(border) == (797, offsets[k + 1] - 1)
definition.options.use_frames = True
definition.options.frames = list(range(2))
offsets = generate_expected_offsets(15, 800, 600)
for k in range(1, 31):
subtask.extra_data = {'start_task': k, 'end_task': k}
border = BlenderTaskTypeInfo.get_task_border(subtask,
definition,
30,
as_path=as_path)
i = (k - 1) % 15 + 1
assert min(border) == (0, offsets[i])
assert max(border) == (798, offsets[i + 1] - 1)
subtask.extra_data = {'start_task': 2, 'end_task': 2}
definition.options.use_frames = True
definition.options.frames = list(range(30))
if as_path:
assert BlenderTaskTypeInfo.get_task_border(
subtask, definition, 30, as_path=as_path) == \
[(0, 600), (800, 600), (800, 0), (0, 0)]
else:
assert BlenderTaskTypeInfo.get_task_border(subtask,
definition,
30,
as_path=as_path) == []
definition.options.use_frames = False
definition.resolution = (0, 0)
assert BlenderTaskTypeInfo.get_task_border(subtask,
definition,
30,
as_path=as_path) == []
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 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
def __init__(self):
Benchmark.__init__(self)
self.normalization_constant = 9360
self.blender_task_path = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "test_task")
self.task_definition.output_file = os.path.join(
tempfile.gettempdir(), "blender_benchmark.png")
self.task_definition.task_type = "Blender"
self.task_definition.output_format = "png"
self.task_definition.options = BlenderRendererOptions()
self.task_definition.options.frames = [1]
self.task_definition.task_id = u"{}".format("blender_benchmark")
self.task_definition.main_scene_file = os.path.join(
self.blender_task_path, "scene-Helicopter-27-cycles.blend")
self.task_definition.main_program_file = BlenderEnvironment(
).main_program_file
self.task_definition.resources.add(
os.path.normpath(self.task_definition.main_scene_file))