summaryrefslogtreecommitdiff
path: root/eval.c
blob: 9e1e946b9873479662fe1d3e1f12dd9d3bb4b134 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <string.h>
#include <stdio.h>

#include "types.h"

typedef struct val {
	type t;
	char *name;
	union {
		int i;
	};
} val;

val *vars[4096];
int varcount = 0;

values *eval(ast *a);

values *getval(ast *a, int index, type t) { 
	values *tmp;
	if (t == INT && a->vals->ts[index] == INT)
		return a->vals;
	else if  ((tmp = eval(a->vals->as[index]))->ts[0] == t)
		return tmp;
	else
		fprintf(stderr, "expected value, but got nothing\n");
}

void checkcount(ast *a, int argc) {
	if (a->vals->argc != argc)
		fprintf(stderr, "incorrect number of args given!\n");
}

values *eval(ast *a) {
	values *v = malloc(sizeof(values));

	if (strcmp(a->fname, "print") == 0) {
		checkcount(a, 1);
		printf("%d\n", getval(a, 0, INT)->is[0]);
		return NULL;
	} 

	else if (strcmp(a->fname, "set") == 0) {
		checkcount(a, 2);
		printf("%d\n", getval(a, 0, INT)->is[0]);
		return NULL;
	} 

	else if (strcmp(a->fname, "+") == 0) {
		v->is[0] = getval(a, 0, INT)->is[0];
		for (int i = 1; i < a->vals->argc; i++) 
			v->is[0] += getval(a, i, INT)->is[i];

		v->ts[0] = INT;
		v->argc++;
	}
	else if (strcmp(a->fname, "-") == 0) {
		if (a->vals->argc == 1) 
			v->is[0] = -getval(a, 0, INT)->is[0];
		else {
			v->is[0] = getval(a, 0, INT)->is[0];
			for (int i = 1; i < a->vals->argc; i++) 
				v->is[0] -= getval(a, i, INT)->is[i];
		}

		v->ts[0] = INT;
		v->argc++;
	}
	else if (strcmp(a->fname, "*") == 0) {
		v->is[0] = getval(a, 0, INT)->is[0]; 
		for (int i = 1; i < a->vals->argc; i++) 
			v->is[0] *= getval(a, i, INT)->is[i];

		v->ts[0] = INT;
		v->argc++;
	}
	else if (strcmp(a->fname, "/") == 0) {
		v->is[0] = getval(a, 0, INT)->is[0];
		for (int i = 1; i < a->vals->argc; i++) 
			v->is[0] /= getval(a, i, INT)->is[i];

		v->ts[0] = INT;
		v->argc++;
	}
	else if (strcmp(a->fname, "%") == 0) {
		v->is[0] = getval(a, 0, INT)->is[0] % getval(a, 1, INT)->is[1];
		v->ts[0] = INT;
		v->argc++;
	}

	return v;
}