compilation of basic arithmetic exprs to bytecode

1 files changed, 70 insertions, 0 deletions

diff --git a/com.c b/com.c
new file mode 100644
index 0000000..f7c7f2d
--- /dev/null
+++ b/com.c
@@ -0,0 +1,70 @@
+#include <stdlib.h>
+#include <stdio.h>
+
+#include "vm.h"
+#include "ast.h"
+#include "read.h"
+
+static void compile_node(Chunk *ch, AstNode a) {
+	switch (a.ty) {
+	case AST_SYMBOL:
+		printf("can't compile symbol\n");
+		exit(1);
+		break;
+	case AST_NUM:
+		chunk_wbc(ch, OP_LOADK);
+		chunk_wbc(ch, chunk_wconst(ch, VAL_NUM(a.as.num)));
+		break;
+	case AST_LIST:;
+		AstVec l = a.as.list;
+		#define CK(cond, msg) if (!(cond)) { puts(msg); exit(1); }
+			CK(l.len == 3, "can only compile binary ops");
+			CK(l.vals[0].ty == AST_SYMBOL, "can only call ops");
+		#undef CK
+		char opchar = l.vals[0].as.str[0];
+		Op op;
+		switch (opchar) {
+		#define OP(char, code) case char: op = code; break;
+			OP('+', OP_ADD)
+			OP('-', OP_SUB)
+			OP('*', OP_MUL)
+			OP('/', OP_DIV)
+		#undef OP
+			default:
+				printf("unkown op %s\n",l.vals[0].as.str);
+				exit(1);
+				break;
+		}
+		compile_node(ch, l.vals[1]);
+		compile_node(ch, l.vals[2]);
+		chunk_wbc(ch, op);
+	}
+}
+
+
+
+
+
+int main() {
+	Chunk ch = chunk_new();
+	AstNode an = read();
+	compile_node(&ch, an);
+
+	chunk_wbc(&ch, OP_PRINT);
+	chunk_wbc(&ch, OP_RET);
+
+
+	// chunk_wbc(&ch, OP_LOADK);
+	// chunk_wbc(&ch, chunk_wconst(&ch, VAL_NUM(5.0)));
+	// chunk_wbc(&ch, OP_LOADK);
+	// chunk_wbc(&ch, chunk_wconst(&ch, VAL_NUM(10.0)));
+	// chunk_wbc(&ch, OP_LOADK);
+	// chunk_wbc(&ch, chunk_wconst(&ch, VAL_NUM(2.0)));
+	// chunk_wbc(&ch, OP_MUL);
+	// chunk_wbc(&ch, OP_ADD);
+	// chunk_wbc(&ch, OP_PRINT);
+	// chunk_wbc(&ch, OP_RET);
+
+	runvm(&ch);
+}
+