Fix and use make_shim.py

4 files changed, 25 insertions, 201 deletions

diff --git a/tools/Makefile b/tools/Makefile
index c591d9c..6c49ccb 100644
--- a/tools/Makefile
+++ b/tools/Makefile
@@ -4,8 +4,7 @@ CFLAGS := -O3 -std=c99 -Wall -Wextra
 
 tools := \
 	pkmncompress \
-	gfx \
-	make_shim
+	gfx
 
 all: $(tools)
 
diff --git a/tools/make_shim.c b/tools/make_shim.c
deleted file mode 100644
index a025905..0000000
--- a/tools/make_shim.c
+++ /dev/null
@@ -1,179 +0,0 @@
-#include <stdio.h>
-#include <getopt.h>
-#include <stdbool.h>
-#include <stdlib.h>
-#include <string.h>
-#include <errno.h>
-
-struct Section {
-    unsigned short end;
-    bool invalid;
-    char * name;
-    bool banked;
-};
-
-typedef struct Section section_t;
-
-// These functions are like strspn and strcspn, but from the right of the string rather than the left.
-size_t strrspn(const char *s1, const char *s2) {
-    const char * _s1 = s1;
-    const char * _s2;
-
-    while (*_s1++);
-    _s1--;
-    while (_s1 > s1) {
-        for (_s2 = s2; *_s2; _s2++) {
-            if (_s1[-1] == *_s2) {
-                break;
-            }
-        }
-        if (*_s2 == 0)
-            break;
-        _s1--;
-    }
-
-    return _s1 - s1;
-}
-
-size_t strrcspn(const char *s1, const char *s2) {
-    const char * _s1 = s1;
-    const char * _s2;
-
-    while (*_s1++);
-    _s1--;
-    while (_s1 > s1) {
-        for (_s2 = s2; *_s2; _s2++) {
-            if (_s1[-1] == *_s2)
-                break;
-        }
-        if (*_s2)
-            break;
-        _s1--;
-    }
-
-    return _s1 - s1;
-}
-
-#define RIP(errmsg) {                                                \
-    fprintf(stderr, "%s:%d:%s: %s\n", fname, lineno, eline, errmsg); \
-    if (file) fclose(file);                                          \
-    return 1;                                                        \
-}
-
-int main(int argc, char * argv[]) {
-    int ch;
-    char line[16*1024];
-    char eline[16*1024];
-    char *lineptr = NULL;
-    char *fname;
-    int lineno;
-
-    section_t section_list[] = {
-        {0x4000, false, "ROM0", false},
-        {0x8000, false, "ROMX", true},
-        {0xA000, false, "VRAM", true},
-        {0xC000, false, "SRAM", true},
-        {0xD000, false, "WRAM0", false},
-        {0xE000, false, "WRAMX", true},
-        {0xFE00, true, "Echo RAM", false},
-        {0xFEA0, false, "OAM", false},
-        {0xFF80, true, "FEXX / IO", false},
-        {0xFFFF, false, "HRAM", false},
-        {}
-    };
-
-    while ((ch = getopt(argc, argv, "wdt")) != -1) {
-        switch (ch) {
-            case 'w':
-            case 'd':
-                section_list[4].end = 0xE000;
-                break;
-            case 't':
-                section_list[0].end = 0x8000;
-                break;
-        }
-    }
-
-    for (int arg_idx = optind; arg_idx < argc; arg_idx++) {
-        int last = -1;
-        int curr;
-        eline[0] = 0;
-        lineno = 0;
-        fname = argv[arg_idx];
-        FILE * file = fopen(fname, "r");
-        if (file == NULL)
-            RIP("Unable to open file");
-        while ((lineptr = fgets(line, sizeof(line), file)) != NULL) {
-            // Assume it's already sorted
-            lineno++;
-            unsigned short bank = 0;
-            unsigned short pointer = 0;
-            char * symbol = NULL;
-            char * end;
-            char * addr_p;
-            strcpy(eline, line);  // Unmodified copy for tracebacks
-            *strrchr(eline, '\n') = 0;
-
-            // line = line.split(";")[0].strip()
-            lineptr += strspn(lineptr, " \t\n");
-            end = strchr(lineptr, ';');
-            if (end) *end = 0;
-            lineptr[strrspn(lineptr, " \t\n")] = 0;
-            if (!*lineptr)
-                continue;
-
-            // Get the bank, address, and symbol
-            end = lineptr + strcspn(lineptr, " \t\n");
-            symbol = end + strspn(end, " \t\n");
-            if (!*symbol)
-                RIP("Declaration not in \"bank:addr Name\" format");
-            *end = 0;
-            addr_p = strchr(lineptr, ':');
-            if (!addr_p)
-                RIP("Pointer not in bank:addr format");
-            *addr_p++ = 0;
-            pointer = strtoul(addr_p, &end, 16);
-            if (pointer == 0 && end == addr_p)
-                RIP("Unable to parse symbol address");
-            bank = strtoul(lineptr, &end, 16);
-            if (bank == 0 && end == lineptr)
-                RIP("Unable to parse bank number");
-            curr = (bank << 14) | (pointer & 0x3fff);
-
-            // Main loop
-            const char * section = NULL;
-            section_t * section_type;
-
-            for (section_type = section_list; section_type->end; section_type++) {
-                if (pointer < section_type->end) {
-                    if (section_type->invalid) {
-                        fprintf(stderr, "Warning: cannot shim '%s' in section type '%s'\n", symbol, section_type->name);
-                    } else {
-                        section = section_type->name;
-                        if (!section_type->banked)
-                            bank = 0;
-                    }
-                    break;
-                }
-            }
-
-            if (section == NULL)
-                // Found section, but cannot shim it
-                continue;
-
-            if (curr != last) {
-                if (last != -1)
-                    fputc('\n', stdout);
-                printf("SECTION \"Shim for %s\", %s[$%04X]", symbol, section, pointer);
-                if (bank)
-                    printf(", BANK[$%04X]", bank);
-                printf("\n%s::\n", symbol);
-                last = curr;
-            } else
-                printf("%s::\n", symbol);
-            fflush(stdout);
-        }
-        fclose(file);
-    }
-    return 0;
-}
diff --git a/tools/make_shim.py b/tools/make_shim.py
index 65d9362..e2e4bf2 100644
--- a/tools/make_shim.py
+++ b/tools/make_shim.py
@@ -5,19 +5,19 @@ import argparse
 from sys import stderr
 from collections import OrderedDict, namedtuple
 
-Section = namedtuple('Section', ('end', 'invalud', 'banked'))
-section_list = OrderedDict(
-    ROM0=Section(0x4000, False, False),
-    ROMX=Section(0x8000, False, True),
-    VRAM=Section(0xA000, False, True),
-    SRAM=Section(0xC000, False, True),
-    WRAM0=Section(0xD000, False, False),
-    WRAMX=Section(0xE000, False, True),
-    EchoRAM=Section(0xFE00, True, False),
-    OAM=Section(0xFEA0, False, False),
-    IO=Section(0xFF80, True, False),
-    HRAM=Section(0xFFFF, False, False)
-)
+Section = namedtuple('Section', ('end', 'invalid', 'banked'))
+section_list = OrderedDict((
+    ('ROM0', Section(0x4000, False, False)),
+    ('ROMX', Section(0x8000, False, True)),
+    ('VRAM', Section(0xA000, False, True)),
+    ('SRAM', Section(0xC000, False, True)),
+    ('WRAM0', Section(0xD000, False, False)),
+    ('WRAMX', Section(0xE000, False, True)),
+    ('EchoRAM', Section(0xFE00, True, False)),
+    ('OAM', Section(0xFEA0, False, False)),
+    ('IO', Section(0xFF80, True, False)),
+    ('HRAM', Section(0xFFFF, False, False))
+))
 
 parser = argparse.ArgumentParser()
 parser.add_argument('files', nargs='+', type=argparse.FileType())
@@ -28,14 +28,14 @@ args = parser.parse_args()
 
 
 if args.w or args.d:
-    section_list['WRAM0'].end = 0xE000
+    section_list['WRAM0'] = Section(0xE000, *section_list['WRAM0'][1:])
 
 if args.t:
-    section_list['ROM0'].end = 0x8000
+    section_list['ROM0'] = Section(0x8000, *section_list['ROM0'][1:])
 
 
-for file_name in args.files:
-    for line in open(file_name, "rt"):
+for f in args.files:
+    for line in f:
 
         # Strip out the comment
         line = line.split(";")[0].strip()
@@ -50,13 +50,13 @@ for file_name in args.files:
             pointer = int(pointer, 16)
         except ValueError:
             print("Error: Cannot parse line: %s" % line, file=stderr)
-            raise from None
+            raise
 
         section = None
         for name, section_type in section_list.items():
             if pointer < section_type.end:
                 if section_type.invalid:
-                    print("Warning: cannot shim '%s' in section type '%s'" % (symbol, section_type['name']), file=stderr)
+                    print("Warning: cannot shim '%s' in section type '%s'" % (symbol, name), file=stderr)
                     section = False
                 else:
                     section = name
diff --git a/tools/scan_includes.py b/tools/scan_includes.py
index 592d6bd..e659dca 100644
--- a/tools/scan_includes.py
+++ b/tools/scan_includes.py
@@ -42,10 +42,14 @@ def shallow_dependencies_of(asm_file_path):
             
             keyword = m.group(1).upper()
             line = line.split(';', 1)[0]
-            path = line[line.index('"') + 1:line.rindex('"')]
+            # RGBDS treats absolute(-looking) paths as relative,
+            # so leading slashes should be stripped.
+            path = line[line.index('"') + 1:line.rindex('"')].lstrip('/')
             if keyword == 'INCLUDE':
                 asm_dependencies.add(path)
             else:
+                if not os.path.isfile(path):
+                    path = 'build/' + path
                 bin_dependencies.add(path)
     
     return asm_dependencies, bin_dependencies