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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
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="Lists the tests available to run", action="store_true")
parser.add_argument('--slow-tests',
help="Run potentially slow tests", action="store_true")
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")
# 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("'{}' is not a valid path to the pyrenderdoc module".format(args.pyrenderdoc))
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("'{}' is not a valid path to the renderdoc library".format(args.renderdoc))
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("<body><h1>Failed to import rdtest: {}</h1></body>".format(ex))
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("Test: {}".format(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)
# 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)
|
