1 files changed, 98 insertions, 0 deletions
diff --git a/sem1/osc/miniproject/cnasm/ast.c b/sem1/osc/miniproject/cnasm/ast.c
new file mode 100644
index 0000000..035b75d
--- /dev/null
+++ b/sem1/osc/miniproject/cnasm/ast.c
@@ -0,0 +1,98 @@
+#include "ast.h"
+
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+
+static ast_node_t *create_empty_node() {
+	ast_node_t *n = malloc(sizeof(ast_node_t));
+	memset(n, 0, sizeof(ast_node_t));
+}
+
+ast_node_t *insert_ctrl(enum ntype t, cond_t *c, ast_node_t *iftrue, ast_node_t *iffalse) {
+	ast_node_t *n = create_empty_node();
+
+	n->t = t;
+	n->flowctrl.condition = c;
+	n->flowctrl.iftrue = iftrue;
+	n->flowctrl.iffalse = iffalse;
+
+	return n;
+}
+
+ast_node_t *insert_for(char *pre, cond_t *c, char *inc, ast_node_t *stuff) {
+	ast_node_t *n = create_empty_node();
+
+	n->t = TFOR;
+	n->forloop.condition = c;
+	n->forloop.pre = pre;
+	n->forloop.inc = inc;
+	n->forloop.stuff = stuff;
+
+	return n;
+}
+
+ast_node_t *insert_stm(ast_node_t *stm, ast_node_t *stm_list) {
+	ast_node_t *n = create_empty_node();
+
+	n->t = TSTM_LIST;
+	n->list.children[0] = stm_list;
+	n->list.children[1] = stm;
+
+	return n;
+}
+ast_node_t *insert_ident(char *ident) {
+	ast_node_t *n = create_empty_node();
+
+	n->t = TIDENT;
+	n->ident = ident;
+
+	return n;
+}
+
+cond_t *insert_cond(uint8_t cmp, char *a, char *b) {
+	cond_t *c = malloc( sizeof(cond_t));
+
+	c->cmp = cmp;
+	c->a = a;
+	c->b = b;
+}
+
+void node_print(ast_node_t *node) {
+	if( !node ){
+		printf("Nil");
+		return;
+	}
+	switch(node->t) {
+		case TSTM_LIST: 
+			printf("Stm_list(");
+			node_print(node->list.children[0]);
+			printf(",");
+			node_print(node->list.children[1]);
+			printf(")");
+			break;
+		case TIF:
+			printf("If");
+			printf("(%s %c %s) {", node->flowctrl.condition->a,
+					node->flowctrl.condition->cmp,
+					node->flowctrl.condition->b);
+			node_print(node->flowctrl.iftrue);
+			printf("}{");
+			node_print(node->flowctrl.iffalse);
+			printf("}");
+			break;
+		case TIDENT:
+			printf("%s", node->ident);
+			break;
+		case TWHILE:
+			printf("while");
+			printf("(%s %c %s) {", node->flowctrl.condition->a,
+					node->flowctrl.condition->cmp,
+					node->flowctrl.condition->b);
+			node_print(node->flowctrl.iftrue);
+			printf("}");
+			break;
+		default:
+			printf("invalid");
+	}
+}