blob: 57c5d5ad6188586039a69bce31e953f0dd3901f5 (
plain) (
blame)
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
|
namespace;
include rt, run;
fn ipow(a, b) { let r = 1; while (b > 0) { r *= a; b--; } return r; }
// a parse word that adds an operator to the compiler's tables
immediate fn pow_op() { run.next(); run.add_binop("**", 11, "ipow c"); return 0; }
pow_op
// unless (cond) stmt: a parse word that emits code itself
immediate fn unless() {
let a;
run.next();
run.expect(run.intern("(")); run.rval(); run.plainuse(run.expr()); run.rval(); run.expect(run.intern(")"));
run.emitseq(run.useq[0], run.ulen[0]); // not
a = run.hole(); run.op(run.O_JZ); // skip the body when the condition was true
run.substmt();
run.setlit(a, run.here);
return 0;
}
// repeat N { body }: a macro, built from token lists
immediate fn repeat() {
let n, body, i = 0;
run.next(); n = run.tv; run.next();
body = run.grab();
while (i < n) { run.feed(run.qdup(body)); i++; }
rt.free(d*body);
return 0;
}
fn main() {
rt.put_n(2 ** 10); rt.put_c(' '); rt.put_n(3 + 2 ** 3 * 2); rt.put_c(10);
unless (1 > 2) rt.put_s("unless ok\n");
unless (2 > 1) rt.put_s("wrong\n");
repeat 3 { rt.put_c('r'); }
rt.put_c(10);
let q = quote { rt.put_n(40 + 2); rt.put_c(10); return 7; };
rt.put_n(q[0]); rt.put_c(' ');
rt.put_n(run.eval(q)); rt.put_c(10);
let f = run.compile(q);
rt.put_n(f()); rt.put_c(10);
rt.free(d*q);
let defs = quote { fn sq(x) { return x * x; } };
run.load(defs);
rt.free(d*defs);
rt.put_n(run.evals("return sq(12);")); rt.put_c(10);
return 0;
}
|