From 0d7d00a97fbe21d2f863a34eda05ad0d93a53fd5 Mon Sep 17 00:00:00 2001
From: tehzz <tehzz@users.noreply.github.com>
Date: Fri, 30 Dec 2022 12:14:51 -0500
Subject: [PATCH] add ASM to GOBAL_ASM C file util

---
 utils/split_asm.py | 97 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100755 utils/split_asm.py

diff --git a/utils/split_asm.py b/utils/split_asm.py
new file mode 100755
index 0000000..965c6f4
--- /dev/null
+++ b/utils/split_asm.py
@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+
+''' Split an asm (.s) file into a series of function.s files 
+    and a GLOBAL_ASM .c file that imports those funcion.s files
+'''
+
+import sys
+from pathlib import Path
+
+if len(sys.argv) != 4:
+    print(f"Usage: {sys.argv[0]} <ASM file> <non-matching dir> <C output dir>")
+    exit()
+
+def fail(msg):
+    print(msg, file=sys.stderr)
+    exit(1)
+
+
+def assert_is_dir(p):
+    if not p.is_dir():
+        fail(f"Path <{p}> is not a directory")
+
+def is_routine_glabel(line):
+    return "glabel" in line and "D_" not in line and "jtgt" not in line and "L8" not in line
+
+def id_routines(input):
+    # routine name => String<asm statements>
+    routines = dict()
+    def store_routine(state):
+        if state is not None: 
+            routines[state[0]] = state[1]
+        state = None
+    
+    with open(input, 'r') as f:
+        # (name, textbuf)
+        state = None
+        text_section = False
+        for line in f:
+            if line.startswith(".section"):
+                text_section = line.split()[1] in [".text", ".text,"]
+                store_routine(state)
+            elif is_routine_glabel(line) and text_section:
+                name = line.split()[-1]
+                # store "completed" prior routine
+                store_routine(state)
+                # and start collecting the next routine
+                state = (name, line)
+            elif state is not None and text_section:
+                state = (state[0], state[1] + line)
+
+        # output final routine
+        store_routine(state)
+
+    return routines
+
+def write_split_files(nm_dir, c_out, routines):
+    # routineName => routineNonMatchingFile
+    with open(c_out, "w+") as cf:
+        cf.write("#include <PR/ultratypes.h>\n\n")
+        cf.write('#pragma GCC diagnostic push\n')
+        cf.write('#pragma GCC diagnostic ignored "-Wunknown-pragmas"\n')
+
+        for (routine, text) in routines.items():
+            nm_out = nm_dir / Path(routine).with_suffix('.s')
+
+            f = open(nm_out, "w")
+            f.write(".section .text\n")
+            f.write(text)
+            f.close()
+
+            cf.write(f'\n#ifdef MIPS_TO_C\n#else\n#pragma GLOBAL_ASM("{nm_out}")\n#endif\n')
+        
+        cf.write('#pragma GCC diagnostic pop\n')
+
+def main():
+    s_in = Path(sys.argv[1])
+    s_out_base = Path(sys.argv[2])
+    c_out_dir = Path(sys.argv[3])
+    
+    input_filename = s_in.stem
+    s_out_dir = s_out_base / Path(*c_out_dir.parts[1:]) / input_filename
+    c_out_file = c_out_dir / s_in.with_suffix('.c').name
+
+    s_out_dir.relative_to
+
+    s_out_dir.mkdir(parents=True, exist_ok=True)
+    c_out_dir.mkdir(parents=True, exist_ok=True)
+    assert_is_dir(s_out_dir)
+    assert_is_dir(c_out_dir)
+
+    routines = id_routines(s_in)
+    write_split_files(s_out_dir, c_out_file, routines)
+    # print(routines)
+
+
+if __name__ == '__main__':
+    main()