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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
typedef struct EnvEntry {
Object *symbol;
Object *value;
} EnvEntry;
typedef struct Environment {
struct Environment *parent;
EnvEntry *buf;
size_t size;
size_t cap;
bool marked;
} Environment;
static Environment *global_env;
#define ENV_BUF_CAP 8
Environment *alloc_env(void);
Environment *
env_create(Environment *parent) {
Environment *env = alloc_env();
env->parent = parent;
env->buf = NULL;
env->size = 0;
env->cap = ENV_BUF_CAP;
return env;
}
void
env_add_symbol(Environment *env, Object *symbol, Object *value) {
if (symbol->type != OBJ_TYPE_SYMBOL) {
error_push((Error){
.type = ERR_TYPE_RUNTIME,
.value = ERR_NOT_A_SYMBOL,
.line = 0,
.col = 0,
});
return;
}
if (env->buf == NULL) {
env->size = 0;
env->cap = ENV_BUF_CAP;
env->buf = malloc(env->cap * sizeof(EnvEntry));
} else if (env->size == env->cap) {
env->cap *= 2;
env->buf = realloc(env->buf, env->cap * sizeof(EnvEntry));
}
env->buf[env->size++] = (EnvEntry){symbol, value};
}
Object *
env_lookup(Environment *env, Object *symbol) {
while (env != NULL) {
for (size_t i = 0; i < env->size; i++) {
EnvEntry entry = env->buf[i];
if (obj_eq(symbol, entry.symbol)) {
return entry.value;
}
}
env = env->parent;
}
return obj_err;
}
Object *
env_update(Environment *env, Object *symbol, Object *value) {
while (env != NULL) {
for (size_t i = 0; i < env->size; i++) {
EnvEntry entry = env->buf[i];
if (obj_eq(symbol, entry.symbol)) {
env->buf[i].value = value;
return obj_nil;
}
}
env = env->parent;
}
error_push((Error){
.type = ERR_TYPE_RUNTIME,
.value = ERR_SYMBOL_NOT_FOUND,
});
return obj_err;
}
ssize_t
env_index_current(Environment *env, Object *symbol) {
for (size_t i = 0; i < env->size; i++) {
EnvEntry entry = env->buf[i];
if (obj_eq(symbol, entry.symbol)) {
return i;
}
}
return -1;
}
void
env_add_or_update_current(Environment *env, Object *symbol, Object *value) {
ssize_t index = env_index_current(env, symbol);
if (index == -1) {
env_add_symbol(env, symbol, value);
} else {
env->buf[index].value = value;
}
}
Environment *
env_extend(Environment *parent, Environment *extra) {
Environment *env = env_create(parent);
for (size_t i = 0; i < extra->size; i++) {
EnvEntry entry = extra->buf[i];
Environment *tmp = env;
bool found = false;
while (tmp != NULL) {
if (env_index_current(tmp, entry.symbol) != -1) {
found = true;
break;
}
tmp = tmp->parent;
}
if (!found) {
env_add_symbol(env, entry.symbol, entry.value);
}
}
return env;
}
|