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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
#include "uscript.h"
#include <stdio.h>
#include "dyn_arr.h"
#include "lex.h"
#include "val.h"
#include "vm.h"
#include "parser.h"
void core_print(int argc)
{
(void)argc;
char *str = val_to_str(vm_peek(0), NULL);
olog(str);
mem_free(str);
vm_push(create_zilch());
}
void core_len(int argc)
{
(void)argc;
struct us_val val = vm_peek(0);
switch (val.type) {
case VAL_ARR:
vm_push(create_num(da_len(get_arr(val)->e)));
break;
case VAL_STR:
vm_push(create_num(get_str(val)->len));
break;
default:
vm_push(create_zilch());
}
}
void core_range_next(int argc)
{
(void)argc;
struct us_cfunc *cfunc = get_cfunc(vm_get(-1));
double min = get_num(cfunc_get_upval(cfunc, 0));
double max = get_num(cfunc_get_upval(cfunc, 1));
double inc = get_num(cfunc_get_upval(cfunc, 2));
if (min == max) {
vm_push(create_zilch());
return;
}
vm_push(create_num(min));
cfunc_set_upval(cfunc, 0, create_num(min + inc));
}
void core_range(int argc)
{
if (vm_get(1).type != VAL_NUM || vm_get(2).type != VAL_NUM)
log_fatal(1, "core:range expected two numbers");
struct us_str *str = copy_str("rangenext", -1);
struct us_cfunc *cfunc = create_cfunc(str, core_range_next, 0);
for (int i = 0; i < argc; i++)
cfunc_add_upval(cfunc, vm_get(i));
vm_push(wrap_cfunc(cfunc));
}
void arr_add(int argc)
{
(void)argc;
if (vm_get(0).type != VAL_ARR)
log_fatal(1, "arr:add expected an array");
struct us_arr *arr = get_arr(vm_get(0));
da_append(struct us_val, &arr->e, vm_get(1));
}
void arr_iter_next(int argc)
{
(void)argc;
struct us_cfunc *cfunc = get_cfunc(vm_get(-1));
struct us_arr *arr = get_arr(cfunc_get_upval(cfunc, 0));
double i = get_num(cfunc_get_upval(cfunc, 1));
int len = da_len(arr->e);
if (i == len) {
vm_push(create_zilch());
return;
}
vm_push(arr->e[(int)i]);
cfunc_set_upval(cfunc, 1, create_num(i + 1));
}
void arr_iter(int argc)
{
(void)argc;
if (vm_get(0).type != VAL_ARR)
log_fatal(1, "arr:iter expected an array");
struct us_str *str = copy_str("arr:iternext", -1);
struct us_cfunc *cfunc = create_cfunc(str, arr_iter_next, 0);
cfunc_add_upval(cfunc, vm_get(0));
cfunc_add_upval(cfunc, create_num(0));
vm_push(wrap_cfunc(cfunc));
}
void us_init(void)
{
init_vm();
us_set_cfunc("core:len", core_len, 1);
us_set_cfunc("core:log", core_print, 1);
us_set_cfunc("core:range", core_range, 3);
us_set_cfunc("arr:iter", arr_iter, 1);
us_set_cfunc("arr:add", arr_add, 2);
}
void us_deinit(void)
{
deinit_vm();
}
void us_load_file(const char *file_path)
{
char *file = read_file(file_path, NULL);
us_load_src(file);
mem_free(file);
}
void us_load_src(const char *src)
{
struct us_proto *proto = compile("main", src);
if (!proto)
return;
us_exec(create_func(proto));
}
int us_declare_global(const char *name)
{
return declare_global(copy_str(name, -1));
}
void us_set_cfunc(const char *c_name, us_cfunc_sig c, int argc)
{
struct us_str *name = copy_str(c_name, -1);
int global = declare_global(name);
set_global(global, wrap_cfunc(create_cfunc(name, c, argc)));
}
|
