aboutsummaryrefslogtreecommitdiff
path: root/tests/programs/meta.zc
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;
}