1bf215546Sopenharmony_ci#!/usr/bin/env python3
2bf215546Sopenharmony_ci
3bf215546Sopenharmony_ciimport argparse
4bf215546Sopenharmony_ciimport difflib
5bf215546Sopenharmony_ciimport errno
6bf215546Sopenharmony_ciimport os
7bf215546Sopenharmony_ciimport pathlib
8bf215546Sopenharmony_ciimport subprocess
9bf215546Sopenharmony_ciimport sys
10bf215546Sopenharmony_ci
11bf215546Sopenharmony_ci# The meson version handles windows paths better, but if it's not available
12bf215546Sopenharmony_ci# fall back to shlex
13bf215546Sopenharmony_citry:
14bf215546Sopenharmony_ci    from meson.mesonlib import split_args
15bf215546Sopenharmony_ciexcept ImportError:
16bf215546Sopenharmony_ci    from shlex import split as split_args
17bf215546Sopenharmony_ci
18bf215546Sopenharmony_ciparser = argparse.ArgumentParser()
19bf215546Sopenharmony_ciparser.add_argument('--i965_asm',
20bf215546Sopenharmony_ci                    help='path to i965_asm binary')
21bf215546Sopenharmony_ciparser.add_argument('--gen_name',
22bf215546Sopenharmony_ci                    help='name of the hardware generation (as understood by i965_asm)')
23bf215546Sopenharmony_ciparser.add_argument('--gen_folder',
24bf215546Sopenharmony_ci                    type=pathlib.Path,
25bf215546Sopenharmony_ci                    help='name of the folder for the generation')
26bf215546Sopenharmony_ciargs = parser.parse_args()
27bf215546Sopenharmony_ci
28bf215546Sopenharmony_ciwrapper = os.environ.get('MESON_EXE_WRAPPER')
29bf215546Sopenharmony_ciif wrapper is not None:
30bf215546Sopenharmony_ci    i965_asm = split_args(wrapper) + [args.i965_asm]
31bf215546Sopenharmony_cielse:
32bf215546Sopenharmony_ci    i965_asm = [args.i965_asm]
33bf215546Sopenharmony_ci
34bf215546Sopenharmony_cisuccess = True
35bf215546Sopenharmony_ci
36bf215546Sopenharmony_cifor asm_file in args.gen_folder.glob('*.asm'):
37bf215546Sopenharmony_ci    expected_file = asm_file.stem + '.expected'
38bf215546Sopenharmony_ci    expected_path = args.gen_folder / expected_file
39bf215546Sopenharmony_ci
40bf215546Sopenharmony_ci    try:
41bf215546Sopenharmony_ci        command = i965_asm + [
42bf215546Sopenharmony_ci            '--type', 'hex',
43bf215546Sopenharmony_ci            '--gen', args.gen_name,
44bf215546Sopenharmony_ci            asm_file
45bf215546Sopenharmony_ci        ]
46bf215546Sopenharmony_ci        with subprocess.Popen(command,
47bf215546Sopenharmony_ci                              stdout=subprocess.PIPE,
48bf215546Sopenharmony_ci                              stderr=subprocess.DEVNULL) as cmd:
49bf215546Sopenharmony_ci            lines_after = [line.decode('ascii') for line in cmd.stdout.readlines()]
50bf215546Sopenharmony_ci    except OSError as e:
51bf215546Sopenharmony_ci        if e.errno == errno.ENOEXEC:
52bf215546Sopenharmony_ci            print('Skipping due to inability to run host binaries.',
53bf215546Sopenharmony_ci                  file=sys.stderr)
54bf215546Sopenharmony_ci            exit(77)
55bf215546Sopenharmony_ci        raise
56bf215546Sopenharmony_ci
57bf215546Sopenharmony_ci    with expected_path.open() as f:
58bf215546Sopenharmony_ci        lines_before = f.readlines()
59bf215546Sopenharmony_ci
60bf215546Sopenharmony_ci    diff = ''.join(difflib.unified_diff(lines_before, lines_after,
61bf215546Sopenharmony_ci                                        expected_file, asm_file.stem + '.out'))
62bf215546Sopenharmony_ci
63bf215546Sopenharmony_ci    if diff:
64bf215546Sopenharmony_ci        print('Output comparison for {}:'.format(asm_file.name))
65bf215546Sopenharmony_ci        print(diff)
66bf215546Sopenharmony_ci        success = False
67bf215546Sopenharmony_ci    else:
68bf215546Sopenharmony_ci        print('{} : PASS'.format(asm_file.name))
69bf215546Sopenharmony_ci
70bf215546Sopenharmony_ciif not success:
71bf215546Sopenharmony_ci    exit(1)
72