1 files changed, 126 insertions, 4 deletions
diff --git a/sem_check.c b/sem_check.c
index 637d044..3d42996 100644
--- a/sem_check.c
+++ b/sem_check.c
@@ -1,12 +1,10 @@
+#include "sem_check.h"
+
 #include <assert.h>
 #include <stdio.h>
 
-#include "node.h"
-#include "scope.h"
-#include "tree.h"
 #include "y.tab.h"
 #include "pc.h"
-#include "sem_check.h"
 
 void check_id(s, n)
 scope *s;
@@ -34,3 +32,127 @@ char *n;
 
 	return tmp;
 }
+
+int check_ret_type(t)
+ptree *t;
+{
+	char buf[100];
+	int type;
+
+	if (!t)
+		fprintf(stderr, "TYPE: %d\n", t->type);
+
+	switch (t->type) {
+	case ID:
+		if (!(t->attr.nval))
+			yyerror("Missing ID\n");
+
+		return t->attr.nval->var_type;
+
+	case ADDOP :
+	case MULOP :
+		if (!(t->r && t->l))
+			yyerror("Missing nodes\n");
+
+		if (t->attr.opval == ADD || t->attr.opval == OR) {
+			if(t->l->ret_type == BOOL && t->r->ret_type ==BOOL)
+				return BOOL;
+			else {
+				type = t->l->ret_type == BOOL ?
+					t->r->ret_type : t->l->ret_type;
+
+				snprintf(buf, 100, "Mismached types:"
+						"Cannot use boolean "
+						"operator on type %s\n",
+						pretty_type(type));
+				yyerror(buf);
+			}
+		}
+
+		if (t->r->ret_type == t->l->ret_type)
+			return t->r->ret_type;
+		else {
+			snprintf(buf, 100, "Mismached types: "
+					"Type %s "
+					"cannot be used with type %s\n",
+					pretty_type(t->r->ret_type),
+					pretty_type(t->l->ret_type));
+			yyerror(buf);
+		}
+		break;
+	case RELOP :
+		if (!(t->r && t->l))
+			yyerror("Missing nodes\n");
+		if (t->r->ret_type == t->l->ret_type)
+			return BOOL;
+		else
+			snprintf(buf, 100, "Mismached types: "
+					"Type %s "
+					"cannot be compared to type %s\n",
+					pretty_type(t->r->ret_type),
+					pretty_type(t->l->ret_type));
+			yyerror(buf);
+		break;
+	case NOT:
+		if (t->l && t->l->ret_type == BOOL)
+			return BOOL;
+		yyerror("NOT needs bool input\n");
+		break;
+	case INUM:
+		return INT;
+	case RNUM:
+		return REAL;
+	case ASSIGNOP:
+		if (!(t->r && t->l))
+			yyerror("Incomplete parse tree\n");
+
+		if (t->l->ret_type == t->r->ret_type)
+			return 0;
+		else {
+			snprintf(buf, 100, "Mismached types: "
+					"Cannot assign type %s "
+					"to variable \"%s\" of type %s\n",
+					pretty_type(t->r->ret_type),
+					t->l->attr.nval->name,
+					pretty_type(t->l->attr.nval->var_type));
+			yyerror(buf);
+		}
+
+
+		break;
+	case ARRAY_ACCESS:
+		if (!(t->r && t->l && t->l->attr.nval))
+			yyerror("Incorrect Array Access\n");
+
+		if (t->r->ret_type != INT) {
+			snprintf(buf, 100, "Cannot access array"
+					"with type %s\n",
+					pretty_type(t->r->ret_type));
+			yyerror(buf);
+		}
+
+		type = t->l->attr.nval -> var_type;
+		if (type == ARRAY - INT || type == ARRAY - REAL)
+			return ARRAY - type;
+		break;
+	case IF:
+	case WHILE:
+		if (!(t->r && t->l))
+			yyerror("Incomplete parse tree\n");
+
+		if (t->l->ret_type != BOOL)
+			yyerror("If condition must be of type BOOL\n");
+		return 0;
+	case FOR:
+		/*TODO add for type checking after parsing is correct*/
+		break;
+	default:
+		return -200;
+		snprintf(buf, 101, "Unknown tree node: %d...\n", t->type);
+		yyerror(buf);
+	}
+
+	return -1;
+
+}
+