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
|
#include <time.h>
#include "lib.h"
#include "state.h"
#include "val.h"
#include "util.h"
static Val fn_clock(State *S, int nargs, Val *args) {
return VAL_NUM((double)clock() / CLOCKS_PER_SEC);
}
static Val fn_write(State *S, int nargs, Val *args) {
CHECK(nargs>0, "need 1 arg to write");
print_val(args[0]);
return VAL_NIL;
}
static Val fn_say(State *S, int nargs, Val *args) {
CHECK(nargs>0, "need 1 arg to say");
println_val(args[0]);
return VAL_NIL;
}
// lists
static Val fn_arr(State *S, int nargs, Val *args) {
CHECK(nargs == 0, "need 0 args to arr");
ObjArr *o = objarr_new(S);
return VAL_OBJ(o);
}
static Val fn_append(State *S, int nargs, Val *args) {
CHECK(nargs == 2, "need 2 args to append");
CHECK(IS_ARR(args[0]), "can only append to arr");
ObjArr *a = AS_ARR(args[0]);
objarr_append(S, a, args[1]);
return args[0];
}
static Val fn_len(State *S, int nargs, Val *args) {
CHECK(nargs == 1, "need 1 arg to len");
CHECK(IS_ARR(args[0]), "can only take length of arr");
ObjArr *a = AS_ARR(args[0]);
return VAL_NUM(a->len);
}
static Val fn_pend(State *S, int nargs, Val *args) {
CHECK(nargs > 0, "need at least 1 arg to ,");
ObjArr *a = objarr_new(S);
for (int i = 0; i < nargs; i++) {
Val v = args[i];
if (!IS_ARR(v))
objarr_append(S, a, v);
else {
ObjArr *b = AS_ARR(v);
for (int j = 0; j < b->len; j++)
objarr_append(S, a, b->d[j]);
}
}
return VAL_OBJ(a);
}
typedef struct {
char *name;
CFunc func;
} BuiltinFunc;
static BuiltinFunc builtin_funcs[] = {
{ "clock", fn_clock },
{ "say", fn_say },
{ "write", fn_write },
{ "arr", fn_arr },
{ "append", fn_append },
{ "len", fn_len },
{ ",", fn_pend },
{ 0 },
};
void load_stdlib(State *S) {
for (BuiltinFunc *b = builtin_funcs; b->name != NULL; b++) {
ObjString *oname = objstring_copy_cstr(S, b->name);
ht_put(S, &S->globals, oname, VAL_CFUNC(b->func));
}
}
|
