From d9034ef09af4d9cfba702911e55c7bfbf3bee2be Mon Sep 17 00:00:00 2001
From: octorock <79596758+octorock@users.noreply.github.com>
Date: Thu, 4 Mar 2021 22:09:44 +0100
Subject: Create parser that resolves the .incbin macros

---
 tools/script_disassembler/incbin_parser.py | 47 ++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 tools/script_disassembler/incbin_parser.py

(limited to 'tools/script_disassembler/incbin_parser.py')

diff --git a/tools/script_disassembler/incbin_parser.py b/tools/script_disassembler/incbin_parser.py
new file mode 100644
index 00000000..f20543a9
--- /dev/null
+++ b/tools/script_disassembler/incbin_parser.py
@@ -0,0 +1,47 @@
+# This python script reads the script.s file which contains all the .incbin macros
+# Then it fetches the corresponding data of the baserom, o
+TMC_FOLDER='../..'
+
+import subprocess
+from script_disassembler import disassemble_script
+
+def main():
+    # read baserom data
+    with open(f'{TMC_FOLDER}/baserom.gba', 'rb') as baserom:
+        baserom_data = bytearray(baserom.read())
+
+    # read scripts.s with incbins
+    with open(f'{TMC_FOLDER}/data/scripts.s', 'r') as scripts:
+
+        while True:
+            line = scripts.readline()
+
+            if not line:
+                # end of file
+                break
+
+            if ':: @' in line: # this might be a label
+                incbin_line = scripts.readline()
+
+                if '.incbin' in incbin_line: # found a label with incbin
+                    label = line.split('::')[0]
+                    (_, start, end) = incbin_line.split(',')
+                    start = int(start, 16)
+                    end = int(end, 16)
+                    
+                    # read data from rom
+                    data = baserom_data[start:start+end]
+
+                    print(f'DISASM {label}')
+                    disassemble_script(data)
+
+                    # print new include label
+                    print(f'.include "data/scripts/{label}.inc"')
+                else:
+                    print(line, end='')
+                    print(incbin_line, end='')
+            else:
+                print(line, end='')
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file
-- 
cgit v1.2.3