fb3e138cee
Being able to compare Eevee reference images is useful for refactoring I'm working on so might as well add them now, even if we can still improve them. Workbench tests are just rendering the same files as Cycles and Eevee. This doesn't really tests many workbench settings until we add tests specifically for them, but does cover how it it handles the different object types.
118 lines
3.4 KiB
Python
Executable File
118 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Apache License, Version 2.0
|
|
|
|
import argparse
|
|
import os
|
|
import shlex
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def render_file(filepath, output_filepath):
|
|
dirname = os.path.dirname(filepath)
|
|
basedir = os.path.dirname(dirname)
|
|
subject = os.path.basename(dirname)
|
|
|
|
frame_filepath = output_filepath + '0001.png'
|
|
|
|
common_args = [
|
|
"-noaudio",
|
|
"--factory-startup",
|
|
"--enable-autoexec",
|
|
filepath,
|
|
"-E", "CYCLES",
|
|
"-o", output_filepath,
|
|
"-F", "PNG"]
|
|
|
|
# OSL and GPU examples
|
|
# custom_args += ["--python-expr", "import bpy; bpy.context.scene.cycles.shading_system = True"]
|
|
# custom_args += ["--python-expr", "import bpy; bpy.context.scene.cycles.device = 'GPU'"]
|
|
custom_args = os.getenv('CYCLESTEST_ARGS')
|
|
custom_args = shlex.split(custom_args) if custom_args else []
|
|
common_args += custom_args
|
|
|
|
if subject == 'opengl':
|
|
command = [BLENDER, "--window-geometry", "0", "0", "1", "1"]
|
|
command += common_args
|
|
command += ['--python', os.path.join(basedir, "util", "render_opengl.py")]
|
|
elif subject == 'bake':
|
|
command = [BLENDER, "--background"]
|
|
command += common_args
|
|
command += ['--python', os.path.join(basedir, "util", "render_bake.py")]
|
|
elif subject == 'denoise_animation':
|
|
command = [BLENDER, "--background"]
|
|
command += common_args
|
|
command += ['--python', os.path.join(basedir, "util", "render_denoise.py")]
|
|
else:
|
|
command = [BLENDER, "--background"]
|
|
command += common_args
|
|
command += ["-f", "1"]
|
|
|
|
try:
|
|
# Success
|
|
output = subprocess.check_output(command)
|
|
if os.path.exists(frame_filepath):
|
|
shutil.copy(frame_filepath, output_filepath)
|
|
os.remove(frame_filepath)
|
|
if VERBOSE:
|
|
print(" ".join(command))
|
|
print(output.decode("utf-8"))
|
|
return None
|
|
except subprocess.CalledProcessError as e:
|
|
# Error
|
|
if os.path.exists(frame_filepath):
|
|
os.remove(frame_filepath)
|
|
if VERBOSE:
|
|
print(" ".join(command))
|
|
print(e.output.decode("utf-8"))
|
|
if b"Error: engine not found" in e.output:
|
|
return "NO_ENGINE"
|
|
elif b"blender probably wont start" in e.output:
|
|
return "NO_START"
|
|
return "CRASH"
|
|
except BaseException as e:
|
|
# Crash
|
|
if os.path.exists(frame_filepath):
|
|
os.remove(frame_filepath)
|
|
if VERBOSE:
|
|
print(" ".join(command))
|
|
print(e)
|
|
return "CRASH"
|
|
|
|
|
|
def create_argparse():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-blender", nargs="+")
|
|
parser.add_argument("-testdir", nargs=1)
|
|
parser.add_argument("-outdir", nargs=1)
|
|
parser.add_argument("-idiff", nargs=1)
|
|
return parser
|
|
|
|
|
|
def main():
|
|
parser = create_argparse()
|
|
args = parser.parse_args()
|
|
|
|
global BLENDER, VERBOSE
|
|
|
|
BLENDER = args.blender[0]
|
|
VERBOSE = os.environ.get("BLENDER_VERBOSE") is not None
|
|
|
|
test_dir = args.testdir[0]
|
|
idiff = args.idiff[0]
|
|
output_dir = args.outdir[0]
|
|
|
|
from modules import render_report
|
|
report = render_report.Report("Cycles Test Report", output_dir, idiff)
|
|
report.set_pixelated(True)
|
|
report.set_reference_dir("cycles_renders")
|
|
report.set_compare_engines('cycles', 'eevee')
|
|
ok = report.run(test_dir, render_file)
|
|
|
|
sys.exit(not ok)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|