Archived
1
0
Fork 0

Added ability to define variables

This commit is contained in:
Brandon Rozek 2018-06-09 19:45:12 -04:00
parent faf8a22331
commit a8882d1cb4
3 changed files with 32 additions and 0 deletions

View file

@ -93,4 +93,32 @@ void lenv_add_builtins(lenv* e) {
lenv_add_builtin(e, "%", builtin_mod);
lenv_add_builtin(e, "min", builtin_min);
lenv_add_builtin(e, "max", builtin_max);
lenv_add_builtin(e, "def", builtin_def);
}
lval* builtin_def(lenv* e, lval* a) {
LASSERT(a, a->cell[0]->type == LVAL_QEXPR,
"Function 'def' passed incorrect type")
// First argument is the symbol list
lval* syms = a->cell[0];
// Ensure all elements of the first list are symbols
for (int i = 0; i < syms->count; i++) {
LASSERT(a, syms->cell[i]->type == LVAL_SYM,
"Function 'def' cannot define non-symbol")
}
// Check correct number of symbols and values
LASSERT(a, syms->count == a->count - 1,
"Function 'def' cannot define incorrect number of values to symbols")
// Assign copies of values to symbols
for (int i = 0; i < syms->count; i++) {
lenv_put(e, syms->cell[i], a->cell[i + 1]);
}
lval_del(a);
return lval_sexpr();
}

View file

@ -21,5 +21,7 @@ lval* lval_fun(lbuiltin func);
void lenv_add_builtin(lenv* e, char* name, lbuiltin func);
void lenv_add_builtins(lenv* e);
lval* builtin_def(lenv* e, lval* a);
#endif

View file

@ -96,6 +96,8 @@ int main (int argc, char** argv) {
free(input);
}
lenv_del(e);
mpc_cleanup(8, Number, Long, Double, Symbol, Sexpr, Qexpr, Expr, Lispy);
return 0;
}