summaryrefslogtreecommitdiff
path: root/codegen/expr2rust.mac
blob: e4e6be807bfc7d30aea9e552392c62719e4b625b (plain)
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
/*
 * Rust code generation for gipfel-rs.
 *
 * Port of gipfel's expr2c.mac (Copyright 2006-2009 Johannes Hofmann).
 * Emits one Rust function per line (grep "^pub fn"); integer literals are
 * converted to float literals by fixup.py afterwards.
 */

load("stringproc")$

varsubst(v) := ssubst("_", "%", string(v))$

printdecl(vars) := (
	sprint("let mut", varsubst(first(vars)), ": f64 = 0.0;"),
	for v in rest(vars) do sprint("let mut", varsubst(v), ": f64 = 0.0;"))$

printdef(expr) := sprint(varsubst(first(expr)), "=", varsubst(second(expr)), ";")$

block2rust(expr) :=
	if ?equal(op(expr), block) then (
		printdecl(first(expr)),
		for d in reverse(rest(reverse(rest(expr)))) do printdef(d),
		sprint("return", varsubst(last(expr)), ";"))
	else
		sprint("return", varsubst(expr), ";")$

expr2rust(funcname, argstr, expr) := (
	sprint("pub fn", funcname, "(", argstr, ") -> f64 {"),
	block2rust(subst(pow, "^", optimize(expr))),
	sprint("}", "
"))$