mirror of https://github.com/PrimeDecomp/prime.git
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# MIT License
|
||
|
|
# See https://github.com/ethteck/m2ctx/blob/main/m2ctx.py
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import subprocess
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
root_dir = os.path.abspath(os.path.join(script_dir, ".."))
|
||
|
|
src_dir = root_dir + "src/"
|
||
|
|
|
||
|
|
CPP_FLAGS = [
|
||
|
|
"-Iinclude",
|
||
|
|
"-Isrc",
|
||
|
|
"-DGEKKO",
|
||
|
|
"-DHW2",
|
||
|
|
"-D__attribute__(...)=",
|
||
|
|
"-D__asm__(...)=",
|
||
|
|
"-ffreestanding",
|
||
|
|
"-DM2CTX",
|
||
|
|
]
|
||
|
|
|
||
|
|
def import_c_file(in_file) -> str:
|
||
|
|
in_file = os.path.relpath(in_file, root_dir)
|
||
|
|
cpp_command = ["gcc", "-E", "-P", "-dM", *CPP_FLAGS, in_file]
|
||
|
|
cpp_command2 = ["gcc", "-E", "-P", *CPP_FLAGS, in_file]
|
||
|
|
|
||
|
|
with tempfile.NamedTemporaryFile(suffix=".c") as tmp:
|
||
|
|
stock_macros = subprocess.check_output(["gcc", "-E", "-P", "-dM", tmp.name], cwd=root_dir, encoding="utf-8")
|
||
|
|
|
||
|
|
out_text = ""
|
||
|
|
try:
|
||
|
|
out_text += subprocess.check_output(cpp_command, cwd=root_dir, encoding="utf-8")
|
||
|
|
out_text += subprocess.check_output(cpp_command2, cwd=root_dir, encoding="utf-8")
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
print(
|
||
|
|
"Failed to preprocess input file, when running command:\n"
|
||
|
|
+ cpp_command,
|
||
|
|
file=sys.stderr,
|
||
|
|
)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if not out_text:
|
||
|
|
print("Output is empty - aborting")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
for line in stock_macros.strip().splitlines():
|
||
|
|
out_text = out_text.replace(line + "\n", "")
|
||
|
|
return out_text
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description="""Create a context file which can be used for mips_to_c"""
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"c_file",
|
||
|
|
help="""File from which to create context""",
|
||
|
|
)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
output = import_c_file(args.c_file)
|
||
|
|
|
||
|
|
with open(os.path.join(root_dir, "ctx.c"), "w", encoding="UTF-8") as f:
|
||
|
|
f.write(output)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|