mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
3dfc40898c
The `__future__` we relied on is now, where the 3 specific things are all included [since Python 3.0](https://docs.python.org/3/library/__future__.html): * absolute_import * print_function * unicode_literals * division These import statements are no-ops and are no longer necessary.
43 lines
1.3 KiB
Python
Executable File
43 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# The following script takes as input a SIL fragment and changes all
|
|
# SSA variables into FileCheck variables. This significantly reduces
|
|
# the amount of time required for creating complicated FileCheck
|
|
# Tests.
|
|
|
|
import argparse
|
|
import re
|
|
import sys
|
|
import textwrap
|
|
|
|
parser = argparse.ArgumentParser(description=textwrap.dedent("""
|
|
Takes an input SIL fragment and changes all SSA variables into FileCheck
|
|
variables.
|
|
"""))
|
|
|
|
parser.add_argument('input', type=argparse.FileType('r'),
|
|
help='Input file. \'-\' for stdin.')
|
|
parser.add_argument('-o', type=argparse.FileType('w'),
|
|
metavar='output',
|
|
help='Output file. Defaults to stdout.',
|
|
default=sys.stdout)
|
|
|
|
args = parser.parse_args()
|
|
|
|
seen_variables = set([])
|
|
ssa_re = re.compile(r'[%](\d+)')
|
|
for line in args.input.readlines():
|
|
line = line[:line.find('//')].rstrip() + "\n"
|
|
have_match = False
|
|
for match in ssa_re.finditer(line):
|
|
have_match = True
|
|
var = match.groups()[0]
|
|
if var not in seen_variables:
|
|
line = line.replace('%' + var, '[[VAR_%s:%%[0-9]+]]' % var)
|
|
seen_variables.add(var)
|
|
else:
|
|
line = line.replace('%' + var, '[[VAR_%s]]' % var)
|
|
if have_match:
|
|
line = '// CHECK: ' + line
|
|
args.o.write(line)
|