mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2024-11-27 07:31:28 +00:00
b71edfaa4e
This is the first commit in a series that will reformat all the python files in the LLVM repository. Reformatting is done with `black`. See more information here: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, JDevlieghere, MatzeB Differential Revision: https://reviews.llvm.org/D150545
89 lines
2.2 KiB
Python
Executable File
89 lines
2.2 KiB
Python
Executable File
#!/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()
|