mirror of
https://github.com/baldurk/renderdoc.git
synced 2026-09-17 19:25:39 +00:00
baldurk
12636a1607
* These were available for python 3.6 too so was never any reason not to use them except not being aware they existed.
149 lines
6.5 KiB
Python
149 lines
6.5 KiB
Python
import argparse
|
|
import os
|
|
import sys
|
|
|
|
script_dir = os.path.realpath(os.path.dirname(__file__))
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-r', '--renderdoc',
|
|
help="The location of the renderdoc library to use", type=str)
|
|
parser.add_argument('-p', '--pyrenderdoc',
|
|
help="The location of the renderdoc python module to use", type=str)
|
|
parser.add_argument('-l', '--list',
|
|
help="Lists the tests available to run", action="store_true")
|
|
parser.add_argument('-t', '--test_include', default=".*",
|
|
help="The tests to include, as a regexp filter", type=str)
|
|
parser.add_argument('-x', '--test_exclude', default="",
|
|
help="The tests to exclude, as a regexp filter", type=str)
|
|
parser.add_argument('--in-process',
|
|
help="Run test code in the same process as test runner", action="store_true")
|
|
parser.add_argument('--slow-tests',
|
|
help="Run potentially slow tests", action="store_true")
|
|
parser.add_argument('--test-timeout',
|
|
help="Timeout for output from tests", default=90, type=int)
|
|
parser.add_argument('--data', default=os.path.join(script_dir, "data"),
|
|
help="The folder that reference data is in. Will not be modified.", type=str)
|
|
parser.add_argument('--demos-binary', default="",
|
|
help="The path to the built demos binary.", type=str)
|
|
parser.add_argument('--demos-timeout', default=None,
|
|
help="The timeout to use when expecting the demos to run.", type=int)
|
|
parser.add_argument('--data-extra', default=os.path.join(script_dir, "data_extra"),
|
|
help="The folder that extra reference data is in (typically very large captures that aren't part "
|
|
"of the normal repo). Will not be modified.", type=str)
|
|
parser.add_argument('--artifacts', default=os.path.join(script_dir, "artifacts"),
|
|
help="The folder to put output artifacts in. Will be completely cleared.", type=str)
|
|
parser.add_argument('--temp', default=os.path.join(script_dir, "tmp"),
|
|
help="The folder to put temporary run data in. Will be completely cleared.", type=str)
|
|
parser.add_argument('--debugger',
|
|
help="Enable debugger mode, exceptions are not caught by the framework.", action="store_true")
|
|
parser.add_argument('--adb-device', required=False,
|
|
help="Use the specified ADB device to run the tests.", type=str)
|
|
# Internal command, when we fork out to run a test in a separate process
|
|
parser.add_argument('--internal_run_test', help=argparse.SUPPRESS, type=str, required=False)
|
|
# Internal command, when we re-run as admin to register vulkan layer
|
|
parser.add_argument('--internal_vulkan_register', help=argparse.SUPPRESS, action="store_true", required=False)
|
|
# Internal command, when we re-run as a remote server
|
|
parser.add_argument('--internal_remote_server', help=argparse.SUPPRESS, action="store_true", required=False)
|
|
args = parser.parse_args()
|
|
|
|
custom_pyrenderdoc = None
|
|
|
|
if args.pyrenderdoc is not None:
|
|
if os.path.isfile(args.pyrenderdoc):
|
|
custom_pyrenderdoc = os.path.abspath(os.path.dirname(args.pyrenderdoc))
|
|
elif os.path.isdir(args.pyrenderdoc):
|
|
custom_pyrenderdoc = os.path.abspath(args.pyrenderdoc)
|
|
else:
|
|
raise RuntimeError(f"'{args.pyrenderdoc}' is not a valid path to the pyrenderdoc module")
|
|
|
|
if args.renderdoc is not None:
|
|
if os.path.isfile(args.renderdoc):
|
|
renderdoc_dirpath = os.path.abspath(os.path.dirname(args.renderdoc))
|
|
elif os.path.isdir(args.renderdoc):
|
|
renderdoc_dirpath = os.path.abspath(args.renderdoc)
|
|
else:
|
|
raise RuntimeError(f"'{args.renderdoc}' is not a valid path to the renderdoc library")
|
|
os.environ["PATH"] += os.pathsep + renderdoc_dirpath
|
|
# Python 3.8 doesn't search PATH so add it to the DLL search path
|
|
if sys.platform == 'win32' and sys.version_info[1] >= 8:
|
|
os.add_dll_directory(renderdoc_dirpath)
|
|
|
|
# if the user didn't specify a pyrenderdoc but we do have a renderdoc, try the default location as a backup
|
|
if custom_pyrenderdoc is None:
|
|
if sys.platform == 'win32':
|
|
custom_pyrenderdoc = os.path.abspath(args.renderdoc) + os.path.sep + "pymodules"
|
|
else:
|
|
custom_pyrenderdoc = os.path.abspath(args.renderdoc)
|
|
|
|
if custom_pyrenderdoc is not None:
|
|
# explicit paths go at the start, implicit paths go at the end
|
|
if args.pyrenderdoc is not None:
|
|
sys.path.insert(0, custom_pyrenderdoc)
|
|
else:
|
|
sys.path.append(custom_pyrenderdoc)
|
|
|
|
sys.path.insert(0, os.path.realpath(os.path.dirname(__file__)))
|
|
|
|
data_path = os.path.realpath(args.data)
|
|
data_extra_path = os.path.realpath(args.data_extra)
|
|
temp_path = os.path.realpath(args.temp)
|
|
demos_binary = args.demos_binary
|
|
if demos_binary != "":
|
|
demos_binary = os.path.realpath(demos_binary)
|
|
demos_timeout = args.demos_timeout
|
|
|
|
os.chdir(sys.path[0])
|
|
|
|
artifacts_dir = os.path.realpath(args.artifacts)
|
|
|
|
try:
|
|
import rdtest
|
|
except (ModuleNotFoundError, ImportError) as ex:
|
|
# very simple output, to ensure we have *something*
|
|
import shutil
|
|
|
|
if os.path.exists(artifacts_dir):
|
|
shutil.rmtree(artifacts_dir, ignore_errors=True)
|
|
os.makedirs(artifacts_dir, exist_ok=True)
|
|
|
|
with open(os.path.join(artifacts_dir, 'output.log.html'), "w") as f:
|
|
f.write(f"<body><h1>Failed to import rdtest: {ex}</h1></body>")
|
|
|
|
print("Couldn't import renderdoc module. Try specifying path to python module with --pyrenderdoc " +
|
|
"or the path to the native library with --renderdoc")
|
|
print(ex)
|
|
|
|
sys.exit(1)
|
|
|
|
from tests import *
|
|
|
|
if args.list:
|
|
for test in rdtest.get_tests():
|
|
print(f"Test: {test.__name__}")
|
|
sys.exit(0)
|
|
|
|
rdtest.set_root_dir(os.path.realpath(os.path.dirname(__file__)))
|
|
rdtest.set_artifact_dir(artifacts_dir)
|
|
rdtest.set_data_dir(data_path)
|
|
rdtest.set_data_extra_dir(data_extra_path)
|
|
rdtest.set_temp_dir(temp_path)
|
|
rdtest.set_demos_binary(demos_binary)
|
|
rdtest.set_demos_timeout(demos_timeout)
|
|
|
|
if args.adb_device:
|
|
rdtest.create_adb_device(args.adb_device)
|
|
else:
|
|
rdtest.set_remote_server(None)
|
|
# debugger option implies in-process test running
|
|
if args.debugger:
|
|
args.in_process = True
|
|
|
|
if args.internal_vulkan_register:
|
|
rdtest.vulkan_register()
|
|
elif args.internal_remote_server:
|
|
rdtest.become_remote_server()
|
|
elif args.internal_run_test is not None:
|
|
rdtest.internal_run_test(args.internal_run_test)
|
|
else:
|
|
rdtest.run_tests(args.test_include, args.test_exclude, args.in_process, args.slow_tests, args.debugger, args.test_timeout)
|