Commit: 7664ab3d2931081abe12f7f9a773fa7a678673eb - em_gb_asm (git)

Emily's Z80 assembler for the Gameboy.

Commit MetaInfo

• Revision: 7664ab3d2931081abe12f7f9a773fa7a678673eb (tree)
• Time: 2021-03-05 14:37:00
• Author: AlaskanEmily <emily@alas...>
• Commiter: AlaskanEmily

Log Message

Add arguments for romsize and bank count.

Change Summary

• modified: em_gb_asm.py

Incremental Difference

--- a/em_gb_asm.py
+++ b/em_gb_asm.py
@@ -835,7 +835,81 @@ def assembleFile(block, path):
     if len(block.data) & 1:
         block.pushByte(0xFF)
 
-for p in sys.argv[1:]:
+# Default rom size.
+romsize = 0x8000
+def setRomSize(n):
+    romsize = int(n, 0)
+
+num_banks = None
+def setNumBanks(n):
+    num_banks = int(n, 0)
+
+i = 1
+num_args = len(sys.argv)
+
+OPTIONS=(
+    ("-s", "--size", setRomSize, "Sets the rom size."),
+    ("-b", "--banks", setNumBanks, "Sets the number of banks.")
+)
+
+def MaybeParseArgument(short, long, proc, _help):
+    global i
+    assert i < num_args
+    opt = sys.argv[i]
+    if opt == short or opt == long:
+        i += 1
+        if i >= num_args:
+            val = None
+        else:
+            val = sys.argv[i]
+    elif opt.startswith(short + "="):
+       val = sys.argv[len(short)+1:]
+    elif opt.startswith(long + "="):
+       val = sys.argv[len(long)+1:]
+    else:
+        return False
+    
+    if not val:
+        print(val + " requires an argument")
+    else:
+        try:
+            proc(val)
+        except:
+            return False
+    return True
+
+def usage():
+    print("Emily's Gameboy Assembler")
+    print("USAGE: " + sys.argv[0] + " [options] <files>")
+    print("OPTIONS:")
+    print("  --help|-h\tDisplays this message then exits.")
+    for opt in OPTIONS:
+        print("  " + opt[0] + "|" + opt[1] + "\t" + opt[3])
+
+while i < num_args:
+    if sys.argv[i] == "--" or not sys.argv[i].startswith("-"):
+        break
+    
+    arg = sys.argv[i]
+    if arg in ("--help", "-h"):
+        usage()
+        sys.exit(0)
+    
+    found = False
+    for opt in OPTIONS:
+        found = MaybeParseArgument(*opt)
+        if found:
+            break
+    
+    if not found:
+        print("Error in option '" + arg + "'")
+        usage()
+        sys.exit(1)
+    
+    i += 1
+
+while i < num_args:
+    p = sys.argv[i]
     included_names = set()
     b = Block(p)
     try:
@@ -844,30 +918,33 @@ for p in sys.argv[1:]:
     except:
         print("Error in " + p)
         raise
+    i += 1
 
 if len(blocks) == 0:
     sys.exit(0)
 
-max_bank = 0
-for b in blocks:
-    if b.bank == None:
-        b.bank = 0
-    else:
-        if b.bank > max_bank:
-            max_bank = b.bank
-        if b.base == None:
-            b.error("!bank specified, but no !orig")
+if num_banks == None:
+    max_bank = 0
+    for b in blocks:
+        if b.bank == None:
+            b.bank = 0
         else:
-            if b.bank == 0 and b.base >= 0x4000:
-                b.error("Invalid base for bank 0")
-            elif b.bank == 0 and b.base + len(b.data) >= 0x4000:
-                b.error("Block is too large to fit into bank 0")
-            if b.bank != 0 and (b.base < 0x4000 or b.base >= 0x8000):
-                b.error("Invalid base for bank " + str(b.bank))
-            elif b.bank == 0 and b.base + len(b.data) >= 0x8000:
-                b.error("Block is too large to fit into bank " + str(b.bank))
+            if b.bank > max_bank:
+                max_bank = b.bank
+            if b.base == None:
+                b.error("!bank specified, but no !orig")
+            else:
+                if b.bank == 0 and b.base >= 0x4000:
+                    b.error("Invalid base for bank 0")
+                elif b.bank == 0 and b.base + len(b.data) >= 0x4000:
+                    b.error("Block is too large to fit into bank 0")
+                if b.bank != 0 and (b.base < 0x4000 or b.base >= 0x8000):
+                    b.error("Invalid base for bank " + str(b.bank))
+                elif b.bank == 0 and b.base + len(b.data) >= 0x8000:
+                    b.error("Block is too large to fit into bank " + str(b.bank))
+    num_banks = max_bank
 
-bank_spans=[[]] * (max_bank + 1)
+bank_spans=[[]] * (num_banks + 1)
 
 # First, place all already originated blocks.
 for b in blocks:
@@ -957,11 +1034,13 @@ for b in blocks:
         asc = hex(a[1])[2:].upper().rjust(4, '0')
         debug.write(bank_name + ":" + addr + " .asc:" + asc + "\n")
 
-while at < 0x7FF0:
+print("ROM used " + str(at) + " bytes of " + str(romsize) + " (" + str(int(at / (romsize / 100.0))) + "%)")
+
+while at < romsize - 16:
     out.write(zero16)
     at += 16
 
-while (at & 0x0FFF) != 0 or at < 0x8000:
+while at < romsize:
     out.write(zero)
     at += 1