1 files changed, 9 insertions, 7 deletions
diff --git a/lab1asm.py b/lab1asm.py
index 6f09bff..dfe9fb3 100644
--- a/lab1asm.py
+++ b/lab1asm.py
@@ -69,11 +69,13 @@ def compile(lines):
     return linked, labels
 
 
+def write(prog, labels):
+    prog_pad = 0x100 - len(prog)
+    prog = "\n".join(prog + [f"{len(prog)+n:02x}: 0000" for n in range(prog_pad)])
+
+    print("PM:")
+    print(prog)
+
+
 if __name__ == "__main__":
-    prog, labels = compile([line.strip() for line in sys.stdin])
-    print("prog:")
-    print("\n".join(prog))
-    print()
-    print("labels:")
-    for label, label_nr in labels.items():
-        print(label, hex(label_nr))
+    write(*compile([line.strip() for line in sys.stdin]))