mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 11:26:18 -04:00
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
import re
|
|
|
|
# Quick and dirty script to generate decompiler test cases from text file format
|
|
|
|
with open("test-cases.txt") as f:
|
|
content = f.readlines()
|
|
content = [x.strip() for x in content]
|
|
|
|
test_cases = {}
|
|
|
|
for case in content:
|
|
args = re.split(",(?=(?:[^\"]*\"[^\"]*\")*[^\"]*$)", case)
|
|
assembly_lines = args[0].replace("\"", "").strip().split("\\n")
|
|
main_instruction = assembly_lines[0].split(" ")[0]
|
|
if re.match("^L\d*:\s*$", main_instruction):
|
|
main_instruction = assembly_lines[1].strip().split(" ")[0]
|
|
main_instruction = main_instruction.upper().replace(".", "_")
|
|
assembly_lines = "{{{}}}".format(", ".join(["\"{}\"".format(x.replace("\\n", "").strip()) for x in assembly_lines]))
|
|
output_lines = args[1].replace("\\n", "").strip()
|
|
write_regs = "{{{}}}".format(args[2].replace("\\n", "").strip().replace(" ", "\",\""))
|
|
read_regs = "{{{}}}".format(args[3].replace("\\n", "").strip().replace(" ", "\",\""))
|
|
clob_regs = "{{{}}}".format(args[4].replace("\\n", "").strip().replace(" ", "\",\""))
|
|
|
|
test_case = "test_case(assembly_from_list({}), {}, {}, {}, {});".format(assembly_lines, output_lines, write_regs, read_regs, clob_regs);
|
|
|
|
if main_instruction in test_cases:
|
|
test_cases[main_instruction].append(test_case)
|
|
else:
|
|
test_cases[main_instruction] = []
|
|
test_cases[main_instruction].append(test_case)
|
|
|
|
with open("test-cases.cpp", "a") as f:
|
|
instructions = test_cases.keys()
|
|
instructions = sorted(instructions)
|
|
for instr in instructions:
|
|
f.write("TEST(DecompilerAtomicOpBuilder, {}) {{".format(instr))
|
|
for case in test_cases[instr]:
|
|
f.write(case)
|
|
f.write("}\n\n")
|
|
|