From fbe23d3506e8f0ab3000ad20633ddf2588f0ee9a Mon Sep 17 00:00:00 2001
From: Zack Buhman <zack@buhman.org>
Date: Mon, 3 Nov 2025 22:04:20 -0600
Subject: [PATCH] assembler: add support for binary output

---
 regs/assembler/fs/__main__.py | 24 ++++++++++++++++++++----
 regs/assembler/vs/__main__.py | 20 +++++++++++++++++---
 regs/pvs_disassemble.py       | 13 ++++++++++---
 regs/us_disassemble.py        | 12 +++++++++---
 regs/us_disassemble2.py       | 13 ++++++++++---
 5 files changed, 66 insertions(+), 16 deletions(-)

diff --git a/regs/assembler/fs/__main__.py b/regs/assembler/fs/__main__.py
index 9dd9519..4c509ca 100644
--- a/regs/assembler/fs/__main__.py
+++ b/regs/assembler/fs/__main__.py
@@ -1,4 +1,5 @@
 import sys
+import struct
 
 from assembler.lexer import Lexer, LexerError
 from assembler.parser import ParserError
@@ -17,12 +18,11 @@ def frontend_inner(buf):
         ins = validate_instruction(ins_ast)
         code = [0] * 6
         emit_instruction(code, ins)
-        print("\n".join(f"0x{code[i]:08x}," for i in range(6)))
-        print()
+        yield code
 
 def frontend(filename, buf):
     try:
-        frontend_inner(buf)
+        yield from frontend_inner(buf)
     except LexerError as e:
         print_error(filename, buf, e)
         raise
@@ -34,7 +34,23 @@ def frontend(filename, buf):
         raise
 
 if __name__ == "__main__":
+    assert len(sys.argv) in {2, 3}
     input_filename = sys.argv[1]
+    binary = len(sys.argv) == 3
+    if binary:
+        output_filename = sys.argv[2]
+
     with open(input_filename, 'rb') as f:
         buf = f.read()
-    frontend(input_filename, buf)
+
+    code_gen = list(frontend(input_filename, buf))
+
+    if not binary:
+        for cw in code_gen:
+            print("\n".join(f"0x{cw[i]:08x}," for i in range(6)))
+            print()
+    else:
+        with open(output_filename, 'wb') as f:
+            for cw in code_gen:
+                data = struct.pack("<IIIIII", *cw)
+                f.write(data)
diff --git a/regs/assembler/vs/__main__.py b/regs/assembler/vs/__main__.py
index 58b9506..9edc611 100644
--- a/regs/assembler/vs/__main__.py
+++ b/regs/assembler/vs/__main__.py
@@ -1,4 +1,5 @@
 import sys
+import struct
 
 from assembler.lexer import Lexer, LexerError
 from assembler.validator import ValidatorError
@@ -48,9 +49,22 @@ def frontend(filename, buf):
         raise
 
 if __name__ == "__main__":
+    assert len(sys.argv) in {2, 3}
     input_filename = sys.argv[1]
-    #output_filename = sys.argv[2]
+    binary = len(sys.argv) == 3
+    if binary:
+        output_filename = sys.argv[2]
+
     with open(input_filename, 'rb') as f:
         buf = f.read()
-    for cw in frontend(input_filename, buf):
-        print(f"0x{cw[0]:08x}, 0x{cw[1]:08x}, 0x{cw[2]:08x}, 0x{cw[3]:08x},")
+
+    code_gen = list(frontend(input_filename, buf))
+
+    if not binary:
+        for cw in code_gen:
+            print(f"0x{cw[0]:08x}, 0x{cw[1]:08x}, 0x{cw[2]:08x}, 0x{cw[3]:08x},")
+    else:
+        with open(output_filename, 'wb') as f:
+            for cw in code_gen:
+                data = struct.pack("<IIII", *cw)
+                f.write(data)
diff --git a/regs/pvs_disassemble.py b/regs/pvs_disassemble.py
index c4f07f5..36b9bf2 100644
--- a/regs/pvs_disassemble.py
+++ b/regs/pvs_disassemble.py
@@ -6,6 +6,7 @@ import pvs_dual_math
 import itertools
 from functools import partial
 import sys
+import struct
 
 def out(level, *args):
     sys.stdout.write("    " * level + " ".join(args))
@@ -248,8 +249,14 @@ def parse_hex(s):
 
 if __name__ == "__main__":
     filename = sys.argv[1]
-    with open(filename) as f:
-        buf = f.read()
-    code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
+    if filename.endswith(".bin"):
+        with open(filename, 'rb') as f:
+            buf = f.read()
+        code = [struct.unpack("<I", buf[i*4:i*4+4])[0] for i in range(len(buf) // 4)]
+    else:
+        with open(filename) as f:
+            buf = f.read()
+        code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
+
     for i in range(len(code) // 4):
         parse_instruction(code[i*4:i*4+4])
diff --git a/regs/us_disassemble.py b/regs/us_disassemble.py
index fbd15b9..489fc57 100644
--- a/regs/us_disassemble.py
+++ b/regs/us_disassemble.py
@@ -1,3 +1,4 @@
+import struct
 import sys
 import parse_bits
 from collections import OrderedDict
@@ -149,8 +150,13 @@ def parse_hex(s):
 
 if __name__ == "__main__":
     filename = sys.argv[1]
-    with open(filename) as f:
-        buf = f.read()
-    code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
+    if filename.endswith(".bin"):
+        with open(filename, 'rb') as f:
+            buf = f.read()
+        code = [struct.unpack("<I", buf[i*4:i*4+4])[0] for i in range(len(buf) // 4)]
+    else:
+        with open(filename) as f:
+            buf = f.read()
+        code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
     for i in range(len(code) // 6):
         disassemble(code, i * 6)
diff --git a/regs/us_disassemble2.py b/regs/us_disassemble2.py
index b7b9872..5eb865e 100644
--- a/regs/us_disassemble2.py
+++ b/regs/us_disassemble2.py
@@ -4,6 +4,7 @@ import parse_bits
 from collections import OrderedDict
 from functools import partial
 from pprint import pprint
+import struct
 
 VERBOSE = environ.get("VERBOSE", "false").lower() == "true"
 
@@ -479,9 +480,15 @@ def parse_hex(s):
 
 if __name__ == "__main__":
     filename = sys.argv[1]
-    with open(filename) as f:
-        buf = f.read()
-    code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
+    if filename.endswith(".bin"):
+        with open(filename, 'rb') as f:
+            buf = f.read()
+        code = [struct.unpack("<I", buf[i*4:i*4+4])[0] for i in range(len(buf) // 4)]
+    else:
+        with open(filename) as f:
+            buf = f.read()
+        code = [parse_hex(c.strip()) for c in buf.split(',') if c.strip()]
+
     for i in range(len(code) // 6):
         start = (i + 0) * 6
         end   = (i + 1) * 6