e955c94ed3
Listing the "Blender Foundation" as copyright holder implied the Blender Foundation holds copyright to files which may include work from many developers. While keeping copyright on headers makes sense for isolated libraries, Blender's own code may be refactored or moved between files in a way that makes the per file copyright holders less meaningful. Copyright references to the "Blender Foundation" have been replaced with "Blender Authors", with the exception of `./extern/` since these this contains libraries which are more isolated, any changed to license headers there can be handled on a case-by-case basis. Some directories in `./intern/` have also been excluded: - `./intern/cycles/` it's own `AUTHORS` file is planned. - `./intern/opensubdiv/`. An "AUTHORS" file has been added, using the chromium projects authors file as a template. Design task: #110784 Ref !110783.
131 lines
3.8 KiB
Python
131 lines
3.8 KiB
Python
# SPDX-FileCopyrightText: 2022 Blender Authors
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import enum
|
|
import time
|
|
|
|
|
|
class RecordStage(enum.Enum):
|
|
INIT = 0,
|
|
WAIT_SHADERS = 1,
|
|
WARMUP = 2,
|
|
RECORD = 3,
|
|
FINISHED = 4
|
|
|
|
|
|
WARMUP_SECONDS = 3
|
|
WARMUP_FRAMES = 10
|
|
SHADER_FALLBACK_SECONDS = 60
|
|
RECORD_PLAYBACK_ITER = 3
|
|
LOG_KEY = "ANIMATION_PERFORMANCE: "
|
|
|
|
|
|
def _run(args):
|
|
import bpy
|
|
|
|
global record_stage
|
|
record_stage = RecordStage.INIT
|
|
|
|
bpy.app.handlers.frame_change_post.append(frame_change_handler)
|
|
bpy.ops.screen.animation_play()
|
|
|
|
|
|
def frame_change_handler(scene):
|
|
import bpy
|
|
|
|
global record_stage
|
|
global start_time
|
|
global start_record_time
|
|
global start_warmup_time
|
|
global warmup_frame
|
|
global stop_record_time
|
|
global playback_iteration
|
|
|
|
if record_stage == RecordStage.INIT:
|
|
screen = bpy.context.window_manager.windows[0].screen
|
|
bpy.context.scene.sync_mode = 'NONE'
|
|
|
|
for area in screen.areas:
|
|
if area.type == 'VIEW_3D':
|
|
space = area.spaces[0]
|
|
space.shading.type = 'RENDERED'
|
|
space.overlay.show_overlays = False
|
|
|
|
start_time = time.perf_counter()
|
|
record_stage = RecordStage.WAIT_SHADERS
|
|
|
|
elif record_stage == RecordStage.WAIT_SHADERS:
|
|
shaders_compiled = False
|
|
if hasattr(bpy.app, 'is_job_running'):
|
|
shaders_compiled = not bpy.app.is_job_running("SHADER_COMPILATION")
|
|
else:
|
|
# Fallback when is_job_running doesn't exists by waiting for a time.
|
|
shaders_compiled = time.perf_counter() - start_time > SHADER_FALLBACK_SECONDS
|
|
|
|
if shaders_compiled:
|
|
start_warmup_time = time.perf_counter()
|
|
warmup_frame = 0
|
|
record_stage = RecordStage.WARMUP
|
|
|
|
elif record_stage == RecordStage.WARMUP:
|
|
warmup_frame += 1
|
|
if time.perf_counter() - start_warmup_time > WARMUP_SECONDS and warmup_frame > WARMUP_FRAMES:
|
|
start_record_time = time.perf_counter()
|
|
playback_iteration = 0
|
|
scene = bpy.context.scene
|
|
scene.frame_set(scene.frame_start)
|
|
record_stage = RecordStage.RECORD
|
|
|
|
elif record_stage == RecordStage.RECORD:
|
|
current_time = time.perf_counter()
|
|
scene = bpy.context.scene
|
|
if scene.frame_current == scene.frame_end:
|
|
playback_iteration += 1
|
|
|
|
if playback_iteration >= RECORD_PLAYBACK_ITER:
|
|
stop_record_time = current_time
|
|
record_stage = RecordStage.FINISHED
|
|
|
|
elif record_stage == RecordStage.FINISHED:
|
|
bpy.ops.screen.animation_cancel()
|
|
num_frames = RECORD_PLAYBACK_ITER * ((scene.frame_end - scene.frame_start) + 1)
|
|
elapse_seconds = stop_record_time - start_record_time
|
|
avg_frame_time = elapse_seconds / num_frames
|
|
fps = 1.0 / avg_frame_time
|
|
print(f"{LOG_KEY}{{'time': {avg_frame_time}, 'fps': {fps} }}")
|
|
bpy.app.handlers.frame_change_post.remove(frame_change_handler)
|
|
bpy.ops.wm.quit_blender()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
_run(None)
|
|
|
|
else:
|
|
import api
|
|
|
|
class EeveeTest(api.Test):
|
|
def __init__(self, filepath):
|
|
self.filepath = filepath
|
|
|
|
def name(self):
|
|
return self.filepath.stem
|
|
|
|
def category(self):
|
|
return "eevee"
|
|
|
|
def run(self, env, device_id):
|
|
args = {}
|
|
_, log = env.run_in_blender(_run, args, [self.filepath], foreground=True)
|
|
for line in log:
|
|
if line.startswith(LOG_KEY):
|
|
result_str = line[len(LOG_KEY):]
|
|
result = eval(result_str)
|
|
return result
|
|
|
|
raise Exception("No playback performance result found in log.")
|
|
|
|
def generate(env):
|
|
filepaths = env.find_blend_files('eevee/*')
|
|
return [EeveeTest(filepath) for filepath in filepaths]
|