File: sequencer_video_output_tests.py

package info (click to toggle)
blender 5.0.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 329,128 kB
  • sloc: cpp: 2,489,823; python: 349,859; ansic: 261,364; xml: 2,103; sh: 999; javascript: 317; makefile: 193
file content (123 lines) | stat: -rw-r--r-- 3,462 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env python3
# SPDX-FileCopyrightText: 2015-2025 Blender Authors
#
# SPDX-License-Identifier: Apache-2.0

import argparse
import sys
import subprocess

from pathlib import Path
from modules import render_report


def get_movie_file_suffix(filepath):
    """
    Get suffix used for the video output.
    The script does not have access to the .blend file content, so deduct it from the .blend filename.
    """

    return Path(filepath).stem.split("_")[-1]


def get_arguments(filepath, output_filepath):
    suffix = get_movie_file_suffix(filepath)

    args = [
        "--background",
        "--factory-startup",
        "--enable-autoexec",
        "--debug-memory",
        "--debug-exit-on-error",
        filepath,
        "-o", f"{output_filepath}.{suffix}",
        "-a",
    ]

    return args


def create_argparse():
    parser = argparse.ArgumentParser(
        description="Run test script for each blend file in TESTDIR, comparing the render result with known output."
    )
    parser.add_argument("--blender", required=True)
    parser.add_argument("--testdir", required=True)
    parser.add_argument("--outdir", required=True)
    parser.add_argument("--oiiotool", required=True)
    parser.add_argument("--batch", default=False, action="store_true")
    return parser


class VideoOutputReport(render_report.Report):
    def postprocess_test(self, blender, test):
        suffix = get_movie_file_suffix(test.filepath)

        video_file = Path(f"{test.tmp_out_img_base}.{suffix}").as_posix()

        # If oiiotool supports the FFmpeg this could be used instead.
        """
        command = (
            self.oiiotool,
            "-i", video_file,
            "-o", test.tmp_out_img,
        )

        try:
            subprocess.check_output(command)
        except subprocess.CalledProcessError as e:
            pass
        """

        # Blender's render pipeline always appends frame suffix unless # is present in the file path.
        # Here we need the file name to match exactly, so we trick Blender by going 0001 -> #### mask
        # allowing render piepline to expand it back to 0001.
        out_filepath = test.tmp_out_img.replace("0001", "####")

        python_expr = (
            f"""
import bpy
scene = bpy.context.scene
scene.render.resolution_x = 1920
scene.render.resolution_y = 1080
scene.render.resolution_percentage = 25
ed = scene.sequence_editor_create()
strip = ed.strips.new_movie(name='input', filepath='{video_file}', channel=1, frame_start=1)
strip.colorspace_settings.name = 'Non-Color'
""")

        command = (
            blender,
            "--background",
            "--factory-startup",
            "--enable-autoexec",
            "--python-expr", python_expr,
            "-o", out_filepath,
            "-F", "PNG",
            "-f", "1",
        )

        try:
            subprocess.check_output(command)
        except subprocess.CalledProcessError as e:
            pass


def main():
    parser = create_argparse()
    args = parser.parse_args()

    report = VideoOutputReport("Sequencer", args.outdir, args.oiiotool)
    report.set_pixelated(True)
    # Default error tolerances are quite large, lower them.
    report.set_fail_threshold(2.0 / 255.0)
    report.set_fail_percent(0.01)
    report.set_reference_dir("reference")

    ok = report.run(args.testdir, args.blender, get_arguments, batch=args.batch)

    sys.exit(not ok)


if __name__ == "__main__":
    main()