mirror of https://github.com/PCSX2/pcsx2.git
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
import argparse
|
|
import glob
|
|
import sys
|
|
import os
|
|
import subprocess
|
|
import multiprocessing
|
|
from pathlib import Path
|
|
from functools import partial
|
|
|
|
def is_gs_path(path):
|
|
ppath = Path(path)
|
|
for extension in [[".gs"], [".gs", ".xz"], [".gs", ".zst"]]:
|
|
if ppath.suffixes == extension:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def run_regression_test(runner, dumpdir, renderer, gspath):
|
|
args = [runner]
|
|
gsname = Path(gspath).name
|
|
while gsname.rfind('.') >= 0:
|
|
gsname = gsname[:gsname.rfind('.')]
|
|
|
|
real_dumpdir = os.path.join(dumpdir, gsname)
|
|
if not os.path.exists(real_dumpdir):
|
|
os.mkdir(real_dumpdir)
|
|
|
|
if renderer is not None:
|
|
args.extend(["-renderer", renderer])
|
|
args.extend(["-dumpdir", real_dumpdir])
|
|
|
|
# loop a couple of times for those stubborn merge/interlace dumps that don't render anything
|
|
# the first time around
|
|
args.extend(["-loop", "2"])
|
|
|
|
args.append("--")
|
|
args.append(gspath)
|
|
|
|
print("Running '%s'" % (" ".join(args)))
|
|
subprocess.run(args)
|
|
|
|
|
|
def run_regression_tests(runner, gsdir, dumpdir, renderer, parallel=1):
|
|
paths = glob.glob(gsdir + "/*.*", recursive=True)
|
|
gamepaths = list(filter(is_gs_path, paths))
|
|
|
|
if not os.path.isdir(dumpdir):
|
|
os.mkdir(dumpdir)
|
|
|
|
print("Found %u GS dumps" % len(gamepaths))
|
|
|
|
if parallel <= 1:
|
|
for game in gamepaths:
|
|
run_regression_test(runner, dumpdir, renderer, game)
|
|
else:
|
|
print("Processing %u games on %u processors" % (len(gamepaths), parallel))
|
|
func = partial(run_regression_test, runner, dumpdir, renderer)
|
|
pool = multiprocessing.Pool(parallel)
|
|
pool.map(func, gamepaths)
|
|
pool.close()
|
|
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Generate frame dump images for regression tests")
|
|
parser.add_argument("-runner", action="store", required=True, help="Path to PCSX2 GS runner")
|
|
parser.add_argument("-gsdir", action="store", required=True, help="Directory containing GS dumps")
|
|
parser.add_argument("-dumpdir", action="store", required=True, help="Base directory to dump frames to")
|
|
parser.add_argument("-renderer", action="store", required=False, help="Renderer to use")
|
|
parser.add_argument("-parallel", action="store", type=int, default=1, help="Number of proceeses to run")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not run_regression_tests(args.runner, os.path.realpath(args.gsdir), os.path.realpath(args.dumpdir), args.renderer, args.parallel):
|
|
sys.exit(1)
|
|
else:
|
|
sys.exit(0)
|
|
|