llvm/utils/add_argument_names.py

   1 #!/usr/bin/env python3
   2 import re, sys
   3
   4 def fix_string(s):
   5     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*')
   6
   7     counter = 0
   8     if 'i32{{.*}}' in s:
   9         counter = 1
  10
  11     at_pos = s.find('@')
  12     if at_pos == -1:
  13         at_pos = 0
  14
  15     annoying_pos = s.find('{{[^(]+}}')
  16     if annoying_pos != -1:
  17         at_pos = annoying_pos + 9
  18
  19     paren_pos = s.find('(', at_pos)
  20     if paren_pos == -1:
  21         return s
  22
  23     res = s[:paren_pos+1]
  24     s = s[paren_pos+1:]
  25
  26     m = TYPE.match(s)
  27     while m:
  28         res += m.group()
  29         s = s[m.end():]
  30         if s.startswith(',') or s.startswith(')'):
  31             res += f' %{counter}'
  32             counter += 1
  33
  34         next_arg = s.find(',')
  35         if next_arg == -1:
  36             break
  37
  38         res += s[:next_arg+1]
  39         s = s[next_arg+1:]
  40         m = TYPE.match(s)
  41
  42     return res+s
  43
  44 def process_file(contents):
  45     PREFIX = re.compile(r'check-prefix(es)?(=|\s+)([a-zA-Z0-9,]+)')
  46     check_prefixes = ['CHECK']
  47     result = ''
  48     for line in contents.split('\n'):
  49         if 'FileCheck' in line:
  50             m = PREFIX.search(line)
  51             if m:
  52                 check_prefixes.extend(m.group(3).split(','))
  53
  54         found_check = False
  55         for prefix in check_prefixes:
  56             if prefix in line:
  57                 found_check = True
  58                 break
  59
  60         if not found_check or 'define' not in line:
  61             result += line + '\n'
  62             continue
  63
  64         # We have a check for a function definition. Number the args.
  65         line = fix_string(line)
  66         result += line + '\n'
  67     return result
  68
  69 def main():
  70     print(f'Processing {sys.argv[1]}')
  71     f = open(sys.argv[1])
  72     content = f.read()
  73     f.close()
  74
  75     content = process_file(content)
  76
  77     f = open(sys.argv[1], 'w')
  78     f.write(content)
  79     f.close()
  80
  81 if __name__ == '__main__':
  82     main()