1 files changed, 93 insertions, 0 deletions

diff --git a/src/viz.c b/src/viz.c
index ea8a817..eb5631b 100644
--- a/src/viz.c
+++ b/src/viz.c
@@ -235,3 +235,96 @@ viz_symtables(Scope **scopes) {
     }
     printf("}\n");
 }
+
+static const char* basm_op_str[] = {
+    [OP_ADD]     = "add   ",
+    [OP_SUB]     = "sub   ",
+    [OP_MUL]     = "mul   ",
+    [OP_DIV]     = "div   ",
+    [OP_MOD]     = "mod   ",
+    [OP_LD8]     = "ld8   ",
+    [OP_LD16]    = "ld16  ",
+    [OP_LD32]    = "ld32  ",
+    [OP_LD64]    = "ld64  ",
+    [OP_ST8]     = "st8   ",
+    [OP_ST16]    = "st16  ",
+    [OP_ST32]    = "st32  ",
+    [OP_ST64]    = "st64  ",
+    [OP_CP8]     = "cp8   ",
+    [OP_CP16]    = "cp16  ",
+    [OP_CP32]    = "cp32  ",
+    [OP_CP64]    = "cp64  ",
+    [OP_NOT]     = "not   ",
+    [OP_AND]     = "and   ",
+    [OP_OR]      = "or    ",
+    [OP_XOR]     = "xor   ",
+    [OP_LSHIFT]  = "lshift",
+    [OP_RSHIFT]  = "rshift",
+    [OP_LROT]    = "lrot  ",
+    [OP_RROT]    = "rrot  ",
+    [OP_LABEL]   = "lab   ",
+    [OP_JMP]     = "jmp   ",
+    [OP_JMP_EQ]  = "jeq   ",
+    [OP_JMP_NEQ] = "jne   ",
+    [OP_JMP_GT]  = "jgt   ",
+    [OP_JMP_LT]  = "jlt   ",
+    [OP_JMP_GE]  = "jge   ",
+    [OP_JMP_LE]  = "jle   ",
+};
+
+void
+viz_operand(Operand op) {
+    switch (op.type) {
+        case OP_TYPE_LABEL: {
+            printf("l%zu", op.id);
+        } break;
+        case OP_TYPE_REG: {
+            printf("r%zu", op.id);
+        } break;
+        case OP_TYPE_CONST: {
+            printf("%lld", op.constant.uval);
+        } break;
+    }
+}
+
+void
+viz_instruction(Instruction *inst, LineInfo *line) {
+    printf("[%6ld:%-6ld] ", line->line, line->col);
+    printf("%s", basm_op_str[inst->op]);
+    switch (inst->op) {
+        case OP_ST8:
+        case OP_ST16:
+        case OP_ST32:
+        case OP_ST64:
+        case OP_LD8:
+        case OP_LD16:
+        case OP_LD32:
+        case OP_LD64: {
+            printf(" ");
+            viz_operand(inst->dst);
+            printf(", ");
+            viz_operand(inst->src_a);
+            if (inst->src_a.type != OP_TYPE_CONST) {
+                printf(", ");
+                viz_operand(inst->src_b);
+            }
+        } break;
+        default: {
+            printf(" ");
+            viz_operand(inst->dst);
+            printf(", ");
+            viz_operand(inst->src_a);
+            printf(", ");
+            viz_operand(inst->src_b);
+        } break;
+    }
+    printf("\n");
+}
+
+void
+viz_basm(ProgramBASM *program) {
+    for (size_t i = 0; i < array_size(program->inst); ++i) {
+        viz_instruction(&program->inst[i], &program->lines[i]);
+    }
+
+}