| #!/usr/bin/env python3 |
| import re, sys |
| |
| def fix_string(s): |
| TYPE = re.compile('\s*(i[0-9]+|float|double|x86_fp80|fp128|ppc_fp128|\[\[.*?\]\]|\[2 x \[\[[A-Z_0-9]+\]\]\]|<.*?>|{.*?}|\[[0-9]+ x .*?\]|%["a-z:A-Z0-9._]+({{.*?}})?|%{{.*?}}|{{.*?}}|\[\[.*?\]\])(\s*(\*|addrspace\(.*?\)|dereferenceable\(.*?\)|byval\(.*?\)|sret|zeroext|inreg|returned|signext|nocapture|align \d+|swiftself|swifterror|readonly|noalias|inalloca|nocapture))*\s*') |
| |
| counter = 0 |
| if 'i32{{.*}}' in s: |
| counter = 1 |
| |
| at_pos = s.find('@') |
| if at_pos == -1: |
| at_pos = 0 |
| |
| annoying_pos = s.find('{{[^(]+}}') |
| if annoying_pos != -1: |
| at_pos = annoying_pos + 9 |
| |
| paren_pos = s.find('(', at_pos) |
| if paren_pos == -1: |
| return s |
| |
| res = s[:paren_pos+1] |
| s = s[paren_pos+1:] |
| |
| m = TYPE.match(s) |
| while m: |
| res += m.group() |
| s = s[m.end():] |
| if s.startswith(',') or s.startswith(')'): |
| res += f' %{counter}' |
| counter += 1 |
| |
| next_arg = s.find(',') |
| if next_arg == -1: |
| break |
| |
| res += s[:next_arg+1] |
| s = s[next_arg+1:] |
| m = TYPE.match(s) |
| |
| return res+s |
| |
| def process_file(contents): |
| PREFIX = re.compile(r'check-prefix(es)?(=|\s+)([a-zA-Z0-9,]+)') |
| check_prefixes = ['CHECK'] |
| result = '' |
| for line in contents.split('\n'): |
| if 'FileCheck' in line: |
| m = PREFIX.search(line) |
| if m: |
| check_prefixes.extend(m.group(3).split(',')) |
| |
| found_check = False |
| for prefix in check_prefixes: |
| if prefix in line: |
| found_check = True |
| break |
| |
| if not found_check or 'define' not in line: |
| result += line + '\n' |
| continue |
| |
| # We have a check for a function definition. Number the args. |
| line = fix_string(line) |
| result += line + '\n' |
| return result |
| |
| def main(): |
| print(f'Processing {sys.argv[1]}') |
| f = open(sys.argv[1]) |
| content = f.read() |
| f.close() |
| |
| content = process_file(content) |
| |
| f = open(sys.argv[1], 'w') |
| f.write(content) |
| f.close() |
| |
| if __name__ == '__main__': |
| main() |
