tornavis/tests/python/eevee_render_tests.py

#!/usr/bin/env python3
# Apache License, Version 2.0

import argparse
import os
import shlex
import shutil
import subprocess
import sys


def setup():
    import bpy

    for scene in bpy.data.scenes:
        scene.render.engine = 'BLENDER_EEVEE'

    # Enable Eevee features
    scene = bpy.context.scene
    eevee = scene.eevee

    eevee.use_sss = True
    eevee.use_ssr = True
    eevee.use_ssr_refraction = True
    eevee.use_gtao = True
    eevee.use_dof = False

    eevee.use_volumetric = True
    eevee.use_volumetric_shadows = True
    eevee.volumetric_tile_size = '2'

    for mat in bpy.data.materials:
        mat.use_screen_refraction = True
        mat.use_sss_translucency = True


# When run from inside Blender, render and exit.
try:
    import bpy
    inside_blender = True
except ImportError:
    inside_blender = False

if inside_blender:
    try:
        setup()
    except Exception as e:
        print(e)
        sys.exit(1)


def render_files(filepaths, output_filepaths):
    command = [
        BLENDER,
        "--background",
        "-noaudio",
        "--factory-startup",
        "--enable-autoexec"]

    for filepath, output_filepath in zip(filepaths, output_filepaths):
        frame_filepath = output_filepath + '0001.png'
        if os.path.exists(frame_filepath):
            os.remove(frame_filepath)

        command.extend([
            filepath,
            "-E", "BLENDER_EEVEE",
            "-P",
            os.path.realpath(__file__),
            "-o", output_filepath,
            "-F", "PNG",
            "-f", "1"])

    error = None
    try:
        # Success
        output = subprocess.check_output(command)
        if VERBOSE:
            print(" ".join(command))
            print(output.decode("utf-8"))
    except subprocess.CalledProcessError as e:
        # Error
        if VERBOSE:
            print(" ".join(command))
            print(e.output.decode("utf-8"))
        error = "CRASH"
    except BaseException as e:
        # Crash
        if VERBOSE:
            print(" ".join(command))
            print(e.decode("utf-8"))
        error = "CRASH"

    # Detect missing filepaths and consider those errors
    errors = []
    for output_filepath in output_filepaths:
        frame_filepath = output_filepath + '0001.png'
        if os.path.exists(frame_filepath):
            shutil.copy(frame_filepath, output_filepath)
            os.remove(frame_filepath)
            errors.append(None)
        else:
            errors.append(error)
            error = 'SKIPPED'

    return errors


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("Eevee Test Report", output_dir, idiff)
    report.set_pixelated(True)
    report.set_reference_dir("eevee_renders")
    report.set_compare_engines('eevee', 'cycles')
    ok = report.run(test_dir, render_files)

    sys.exit(not ok)


if not inside_blender and __name__ == "__main__":
    main()