ref: bdf0ed62196f1e67d65b3346ae45d1a0e322bd00
dir: /mi/flatten.c/
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <inttypes.h>
#include <ctype.h>
#include <string.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "util.h"
#include "parse.h"
#include "mi.h"
#include "../config.h"
typedef struct Flattenctx Flattenctx;
typedef struct Loop Loop;
struct Loop {
Node *lcnt;
Node *lbrk;
Stab *body;
};
/* takes a list of nodes, and reduces it (and it's subnodes) to a list
* following these constraints:
* - All nodes are expression node
* - Nodes with side effects are root node
* - All nodes operate on machine-primitive types and tuple
*/
struct Flattenctx {
int isglobl;
Node **stmts;
size_t nstmts;
/* return handling */
Node *tret;
/* pre/postinc handling */
Node **incqueue;
size_t nqueue;
/* break/continue handling */
Loop loop;
unsigned inloop;
Stab *curst;
/* location handling */
Htab *globls;
size_t stksz;
};
static Node *flatten(Flattenctx *s, Node *n);
static Node *rval(Flattenctx *s, Node *n);
static Node *lval(Flattenctx *s, Node *n);
static Node *assign(Flattenctx *s, Node *lhs, Node *rhs);
static void draininc(Flattenctx *c);
static void
append(Flattenctx *s, Node *n)
{
if (debugopt['F'])
dumpn(n, stdout);
lappend(&s->stmts, &s->nstmts, n);
}
static void
cjmp(Flattenctx *s, Node *cond, Node *iftrue, Node *iffalse)
{
Node *jmp;
jmp = mkexpr(cond->loc, Ocjmp, cond, iftrue, iffalse, NULL);
jmp->expr.type = mktype(cond->loc, Tyvoid);
append(s, jmp);
}
static void
jmp(Flattenctx *s, Node *lbl)
{
Node *n;
n = mkexpr(lbl->loc, Ojmp, lbl, NULL);
n->expr.type = mktype(n->loc, Tyvoid);
append(s, n);
}
static Node *
asn(Node *a, Node *b)
{
Node *n;
assert(a != NULL && b != NULL);
if (tybase(exprtype(a))->type == Tyvoid)
return a;
n = mkexpr(a->loc, Oasn, a, b, NULL);
n->expr.type = exprtype(a);
return n;
}
static int
islbl(Node *n)
{
Node *l;
if (exprop(n) != Olit)
return 0;
l = n->expr.args[0];
return l->type == Nlit && l->lit.littype == Llbl;
}
static Node *
temp(Flattenctx *flatten, Node *e)
{
Node *t;
assert(e->type == Nexpr);
t = gentemp(e->loc, e->expr.type, NULL);
return t;
}
static Node *
add(Node *a, Node *b)
{
Node *n;
n = mkexpr(a->loc, Oadd, a, b, NULL);
n->expr.type = a->expr.type;
return n;
}
static Node *
addk(Node *n, uvlong v)
{
Node *k;
k = mkintlit(n->loc, v);
k->expr.type = exprtype(n);
return add(n, k);
}
static Node *
sub(Node *a, Node *b)
{
Node *n;
n = mkexpr(a->loc, Osub, a, b, NULL);
n->expr.type = a->expr.type;
return n;
}
static Node *
subk(Node *n, uvlong v)
{
Node *k;
k = mkintlit(n->loc, v);
k->expr.type = exprtype(n);
return sub(n, k);
}
static Node *
seqlen(Flattenctx *s, Node *n, Type *ty)
{
Node *r;
if (!ty)
ty = n->expr.type;
if (exprtype(n)->type == Tyarray) {
r = mkexpr(n->loc, Osllen, rval(s, n), NULL);
r->expr.type = ty;
} else if (exprtype(n)->type == Tyslice) {
r = mkexpr(n->loc, Osllen, rval(s, n), NULL);
r->expr.type = ty;
} else {
die("invalid seq type for len");
}
return r;
}
static Node *
visit(Flattenctx *s, Node *n)
{
size_t i;
Node *r;
for (i = 0; i < n->expr.nargs; i++)
n->expr.args[i] = rval(s, n->expr.args[i]);
if (opispure[exprop(n)]) {
r = n;
} else {
if (exprtype(n)->type == Tyvoid) {
r = mkexpr(n->loc, Olit, mkvoid(n->loc), NULL);
r->expr.type = mktype(n->loc, Tyvoid);
append(s, n);
} else {
r = temp(s, n);
append(s, asn(r, n));
}
}
return r;
}
static Node *
traitfn(Srcloc loc, Trait *tr, char *fn, Type *ty)
{
Node *proto, *dcl, *var;
char *name;
size_t i;
for (i = 0; i < tr->nproto; i++) {
name = declname(tr->proto[i]);
if (!strcmp(fn, name)) {
proto = tr->proto[i];
dcl = htget(proto->decl.impls, ty);
var = mkexpr(loc, Ovar, dcl->decl.name, NULL);
var->expr.type = dcl->decl.type;
var->expr.did = dcl->decl.did;
return var;
}
}
return NULL;
}
static void
dispose(Flattenctx *s, Stab *st, size_t n)
{
Node *e, *call, *func;
Trait *tr;
Type *ty;
size_t i;
tr = traittab[Tcdisp];
/* dispose in reverse order of appearance */
for (i = st->nautotmp; i-- > n;) {
e = st->autotmp[i];
ty = exprtype(e);
func = traitfn(Zloc, tr, "__dispose__", ty);
call = mkexpr(Zloc, Ocall, func, e, NULL);
call->expr.type = mktype(Zloc, Tyvoid);
flatten(s, call);
}
}
static void
flattencond(Flattenctx *s, Node *n, Node *ltrue, Node *lfalse)
{
Node **args;
Node *v, *lnext;
args = n->expr.args;
switch (exprop(n)) {
case Oland:
lnext = genlbl(n->loc);
flattencond(s, args[0], lnext, lfalse);
append(s, lnext);
flattencond(s, args[1], ltrue, lfalse);
break;
case Olor:
lnext = genlbl(n->loc);
flattencond(s, args[0], ltrue, lnext);
append(s, lnext);
flattencond(s, args[1], ltrue, lfalse);
break;
case Olnot:
flattencond(s, args[0], lfalse, ltrue);
break;
default:
v = rval(s, n);
cjmp(s, v, ltrue, lfalse);
break;
}
}
static Node*
flattentern(Flattenctx *s, Node *n)
{
Node *l1, *l2, *l3;
Node *res, *t;
l1 = genlbl(n->loc);
l2 = genlbl(n->loc);
l3 = genlbl(n->loc);
res = temp(s, n);
flattencond(s, n->expr.args[0], l1, l2);
append(s, l1);
t = assign(s, res, rval(s, n->expr.args[1]));
append(s, t);
jmp(s, l3);
append(s, l2);
t = assign(s, res, rval(s, n->expr.args[2]));
append(s, t);
append(s, l3);
return res;
}
/* flatten
* a || b
* to
* if a || b
* t = true
* else
* t = false
* ;;
*/
static Node *
flattenlazy(Flattenctx *s, Node *n)
{
Node *r, *t, *u;
Node *ltrue, *lfalse, *ldone;
/* set up temps and labels */
r = temp(s, n);
ltrue = genlbl(n->loc);
lfalse = genlbl(n->loc);
ldone = genlbl(n->loc);
/* flatten the conditional */
flattencond(s, n, ltrue, lfalse);
/* if true */
append(s, ltrue);
u = mkexpr(n->loc, Olit, mkbool(n->loc, 1), NULL);
u->expr.type = mktype(n->loc, Tybool);
t = asn(r, u);
append(s, t);
jmp(s, ldone);
/* if false */
append(s, lfalse);
u = mkexpr(n->loc, Olit, mkbool(n->loc, 0), NULL);
u->expr.type = mktype(n->loc, Tybool);
t = asn(r, u);
append(s, t);
jmp(s, ldone);
/* finish */
append(s, ldone);
return r;
}
static Node *
destructure(Flattenctx *s, Node *lhs, Node *rhs)
{
Node *lv, *rv, *idx;
Node **args;
size_t i;
args = lhs->expr.args;
rhs = rval(s, rhs);
for (i = 0; i < lhs->expr.nargs; i++) {
idx = mkintlit(rhs->loc, i);
idx->expr.type = mktype(rhs->loc, Tyuint64);
rv = mkexpr(rhs->loc, Otupget, rhs, idx, NULL);
rv->expr.type = lhs->expr.type;
if (exprop(args[i]) == Otup) {
destructure(s, args[i], rv);
} else {
lv = lval(s, args[i]);
append(s, assign(s, lv, rv));
}
}
return rhs;
}
static Node *
comparecomplex(Flattenctx *s, Node *n, Op op)
{
Type *ty;
Node *l, *r, *e;
/* special case: unions with nullary constructors can be compared easily. */
ty = tybase(exprtype(n->expr.args[0]));
if (ty->type == Tyunion && isenum(ty)) {
l = mkexpr(n->loc, Outag, rval(s, n->expr.args[0]), NULL);
r = mkexpr(n->loc, Outag, rval(s, n->expr.args[1]), NULL);
l->expr.type = mktype(n->loc, Tyuint32);
r->expr.type = mktype(n->loc, Tyuint32);
if (op == Oeq)
e = mkexpr(n->loc, Oueq, l, r, NULL);
else if (op == One)
e = mkexpr(n->loc, Oune, l, r, NULL);
else
fatal(n, "unsupported operator %s for enum union", opstr[op]);
e->expr.type = mktype(n->loc, Tybool);
return e;
}
fatal(n, "cannot compare values of type %s for equality\n",
tystr(exprtype(n->expr.args[0])));
return NULL;
}
static Node *
compare(Flattenctx *s, Node *n, int fields)
{
const Op cmpmap[Numops][3] = {
[Oeq] = {Oeq, Oueq, Ofeq},
[One] = {One, Oune, Ofne},
[Ogt] = {Ogt, Ougt, Ofgt},
[Oge] = {Oge, Ouge, Ofge},
[Olt] = {Olt, Oult, Oflt},
[Ole] = {Ole, Oule, Ofle}
};
Node *r;
Type *ty;
Op newop;
/* void is always void */
if (tybase(exprtype(n->expr.args[0]))->type == Tyvoid)
return mkboollit(n->loc, 1);
newop = Obad;
ty = tybase(exprtype(n->expr.args[0]));
if (istysigned(ty))
newop = cmpmap[n->expr.op][0];
else if (istyunsigned(ty))
newop = cmpmap[n->expr.op][1];
else if (istyunsigned(ty))
newop = cmpmap[n->expr.op][1];
else if (ty->type == Typtr)
newop = cmpmap[n->expr.op][1];
else if (istyfloat(ty))
newop = cmpmap[n->expr.op][2];
if (newop != Obad) {
n->expr.op = newop;
r = visit(s, n);
} else if (fields) {
r = comparecomplex(s, n, exprop(n));
} else {
fatal(n, "unsupported comparison on type %s", tystr(ty));
}
r->expr.type = mktype(n->loc, Tybool);
return r;
}
static Node *
assign(Flattenctx *s, Node *lhs, Node *rhs)
{
Node *r, *t, *e, *u;
if (exprop(lhs) == Otup) {
r = destructure(s, lhs, rhs);
} else if (tybase(exprtype(lhs))->type != Tyvoid) {
t = lval(s, lhs);
u = rval(s, rhs);
e = asn(t, u);
append(s, e);
r = t;
} else {
r = rval(s, rhs);
}
return r;
}
/*
* If protect is not null, returns a node with the
* unmodified value.
*/
static void
draininc(Flattenctx *fc)
{
size_t i;
for (i = 0; i < fc->nqueue; i++)
append(fc, fc->incqueue[i]);
lfree(&fc->incqueue, &fc->nqueue);
}
/* returns 1 when the exit jump needs to be emitted */
static int
exitscope(Flattenctx *s, Stab *stop, Srcloc loc, Exit x)
{
Node *exit;
Stab *st;
for (st = s->curst;; st = st->super) {
exit = st->exit[x];
if (st->ndisposed[x] < st->nautotmp) {
st->exit[x] = genlbl(loc);
flatten(s, st->exit[x]);
dispose(s, st, st->ndisposed[x]);
st->ndisposed[x] = st->nautotmp;
}
if (exit) {
jmp(s, exit);
return 0;
}
if ((!stop && st->isfunc) || st == stop) {
return 1;
}
}
}
static Node *
rval(Flattenctx *s, Node *n)
{
Node *t, *u, *v; /* temporary nodes */
Node *r; /* expression result */
Node **args;
Type *ty;
const Op fusedmap[Numops] = {
[Oaddeq] = Oadd,
[Osubeq] = Osub,
[Omuleq] = Omul,
[Odiveq] = Odiv,
[Omodeq] = Omod,
[Oboreq] = Obor,
[Obandeq] = Oband,
[Obxoreq] = Obxor,
[Obsleq] = Obsl,
[Obsreq] = Obsr,
};
r = NULL;
args = n->expr.args;
switch (exprop(n)) {
case Oauto:
t = temp(s, n);
r = assign(s, t, n->expr.args[0]);
lappend(&s->curst->autotmp, &s->curst->nautotmp, t);
break;
case Osize:
r = n; /* don't touch subexprs; they're a pseudo decl */
break;
case Olor: case Oland:
r = flattenlazy(s, n);
break;
case Oidx:
t = rval(s, n->expr.args[0]);
u = rval(s, n->expr.args[1]);
r = mkexpr(n->loc, Oidx, t, u, NULL);
r->expr.type = n->expr.type;
break;
/* array.len slice.len are magic 'virtual' members.
* they need to be special cased. */
case Omemb:
ty = tybase(exprtype(args[0]));
if (ty->type == Tyslice || ty->type == Tyarray) {
r = seqlen(s, args[0], exprtype(n));
} else {
case Otupmemb:
t = rval(s, args[0]);
r = mkexpr(n->loc, exprop(n), t, args[1], NULL);
r->expr.type = n->expr.type;
}
break;
case Oucon:
if (n->expr.nargs > 1)
t = rval(s, args[1]);
else
t = NULL;
r = mkexpr(n->loc, Oucon, args[0], t, NULL);
r->expr.type = n->expr.type;
break;
/* fused ops:
* foo ?= blah
* =>
* foo = foo ? blah*/
case Oaddeq: case Osubeq: case Omuleq: case Odiveq: case Omodeq:
case Oboreq: case Obandeq: case Obxoreq: case Obsleq: case Obsreq:
assert(fusedmap[exprop(n)] != Obad);
u = lval(s, args[0]);
v = rval(s, args[1]);
v = mkexpr(n->loc, fusedmap[exprop(n)], u, v, NULL);
v->expr.type = u->expr.type;
r = asn(u, v);
break;
/* ++expr(x)
* => args[0] = args[0] + 1
* expr(x) */
case Opreinc:
t = lval(s, args[0]);
v = asn(t, addk(t, 1));
append(s, v);
r = rval(s, t);
break;
case Opredec:
t = lval(s, args[0]);
v = asn(t, subk(t, 1));
append(s, v);
r = rval(s, t);
break;
/* expr(x++)
* => expr
* x = x + 1
*/
case Opostinc:
r = lval(s, args[0]);
t = asn(r, addk(r, 1));
lappend(&s->incqueue, &s->nqueue, t);
break;
case Opostdec:
r = lval(s, args[0]);
t = asn(r, subk(r, 1));
lappend(&s->incqueue, &s->nqueue, t);
break;
case Olit:
switch (args[0]->lit.littype) {
case Lvoid:
case Lchr:
case Lbool:
case Llbl:
case Lint:
case Lstr:
case Lflt:
case Lfunc:
r = n; //capture(s, n, dst);
break;
}
break;
case Ovar:
r = n;
break;;
case Oret:
v = rval(s, args[0]);
if (!s->tret)
s->tret = temp(s, v);
flatten(s, asn(lval(s, s->tret), v));
draininc(s);
if (exitscope(s, NULL, Zloc, Xret))
append(s, mkexpr(n->loc, Oret, s->tret, NULL));
break;
case Otern:
r = flattentern(s, n);
break;
case Oasn:
r = assign(s, args[0], args[1]);
break;
case Obreak:
r = NULL;
if (s->inloop == 0)
fatal(n, "trying to break when not in loop");
if (exitscope(s, s->loop.body, n->loc, Xbrk))
jmp(s, s->loop.lbrk);
break;
case Ocontinue:
r = NULL;
if (s->inloop == 0)
fatal(n, "trying to continue when not in loop");
if (exitscope(s, s->loop.body, n->loc, Xcnt))
jmp(s, s->loop.lcnt);
break;
case Oeq: case One:
r = compare(s, n, 1);
break;
case Ogt: case Oge: case Olt: case Ole:
r = compare(s, n, 0);
break;
default:
if (istyfloat(exprtype(n))) {
switch (exprop(n)) {
case Oadd: n->expr.op = Ofadd; break;
case Osub: n->expr.op = Ofsub; break;
case Omul: n->expr.op = Ofmul; break;
case Odiv: n->expr.op = Ofdiv; break;
default: break;
}
}
r = visit(s, n);
break;
}
if (r && n->expr.idx)
r->expr.idx = n->expr.idx;
return r;
}
static Node *
lval(Flattenctx *s, Node *n)
{
Node *r;
switch (exprop(n)) {
case Ovar: r = n; break;
case Oidx: r = rval(s, n); break;
case Oderef: r = rval(s, n); break;
case Otupmemb: r = rval(s, n); break;
case Omemb: r = rval(s, n); break;
case Ostruct: r = rval(s, n); break;
/* for chaining */
case Oucon: r = rval(s, n); break;
case Oarr: r = rval(s, n); break;
case Ogap: r = temp(s, n); break;
default:
fatal(n, "%s cannot be an lvalue", opstr[exprop(n)]);
break;
}
return r;
}
static void
flattenblk(Flattenctx *s, Node *n)
{
Stab *st;
size_t i;
st = s->curst;
s->curst = n->block.scope;
for (i = 0; i < n->block.nstmts; i++) {
n->block.stmts[i] = fold(n->block.stmts[i], 0);
flatten(s, n->block.stmts[i]);
}
assert(s->curst == n->block.scope);
if (st->isfunc)
exitscope(s, NULL, Zloc, Xret);
else
dispose(s, s->curst, 0);
s->curst = st;
}
/* init; while cond; body;;
* => init
* jmp :cond
* :body
* ...body...
* ...step...
* :cond
* ...cond...
* cjmp (cond) :body :end
* :end
*/
static void
flattenloop(Flattenctx *s, Node *n)
{
Stab *b;
Loop l;
Node *lbody;
Node *lend;
Node *ldec;
Node *lcond;
Node *lstep;
lbody = genlbl(n->loc);
ldec = genlbl(n->loc);
lcond = genlbl(n->loc);
lstep = genlbl(n->loc);
lend = genlbl(n->loc);
b = s->curst;
s->curst = n->loopstmt.scope;
l = s->loop;
s->loop.lcnt = lstep;
s->loop.lbrk = lend;
s->loop.body = n->loopstmt.scope;
s->inloop++;
flatten(s, n->loopstmt.init); /* init */
jmp(s, lcond); /* goto test */
flatten(s, lbody); /* body lbl */
flatten(s, n->loopstmt.body); /* body */
flatten(s, lstep); /* test lbl */
flatten(s, n->loopstmt.step); /* step */
flatten(s, lcond); /* test lbl */
flattencond(s, n->loopstmt.cond, ldec, lend); /* repeat? */
flatten(s, ldec); /* drain decrements */
draininc(s);
jmp(s, lbody); /* goto test */
flatten(s, lend); /* exit */
draininc(s);
s->inloop--;
s->loop = l;
s->curst = b;
}
/* if foo; bar; else baz;;
* => cjmp (foo) :bar :baz */
static void
flattenif(Flattenctx *s, Node *n, Node *exit)
{
Node *l1, *l2, *l3;
Node *cond, *iftrue, *iffalse, *t;
l1 = genlbl(n->loc);
l2 = genlbl(n->loc);
if (exit)
l3 = exit;
else
l3 = genlbl(n->loc);
cond = rval(s, n->ifstmt.cond);
iftrue = n->ifstmt.iftrue;
iffalse = n->ifstmt.iffalse;
if (s->incqueue) {
t = temp(s, cond);
flatten(s, asn(t, cond));
draininc(s);
cond = t;
}
flattencond(s, cond, l1, l2);
flatten(s, l1);
draininc(s);
/* goto test */
flatten(s, iftrue);
jmp(s, l3);
flatten(s, l2);
draininc(s);
/* because lots of bunched up end labels are ugly,
* coalesce them by handling 'elif'-like construct
* separately */
if (iffalse && iffalse->type == Nifstmt) {
flattenif(s, iffalse, exit);
} else {
flatten(s, iffalse);
jmp(s, l3);
}
if (!exit)
flatten(s, l3);
}
static void
flattenloopmatch(Flattenctx *s, Node *pat, Node *val, Node *ltrue, Node *lfalse)
{
Node **cap, **out, *lload;
size_t i, ncap, nout;
/* pattern match */
lload = genlbl(pat->loc);
out = NULL;
nout = 0;
cap = NULL;
ncap = 0;
genonematch(pat, val, lload, lfalse, &out, &nout, &cap, &ncap);
for (i = 0; i < nout; i++)
flatten(s, out[i]);
flatten(s, lload);
for (i = 0; i < ncap; i++)
flatten(s, cap[i]);
jmp(s, ltrue);
}
/* pat; seq;
* body;;
*
* =>
* .pseudo = seqinit
* jmp :cond
* :body
* ...body...
* :step
* ...step...
* :cond
* ...cond...
* cjmp (cond) :match :end
* :match
* ...match...
* cjmp (match) :load :step
* :load
* matchval = load
* :end
*/
static void
flattenidxiter(Flattenctx *s, Node *n)
{
Loop l;
Node *lbody, *lstep, *lcond, *lmatch, *lend;
Node *idx, *len, *dcl, *seq, *val, *done;
Node *zero;
Type *idxtype;
lbody = genlbl(n->loc);
lstep = genlbl(n->loc);
lcond = genlbl(n->loc);
lmatch = genlbl(n->loc);
lend = genlbl(n->loc);
s->inloop++;
l = s->loop;
s->loop.lcnt = lstep;
s->loop.lbrk = lend;
s->loop.body = n->iterstmt.body->block.scope;
/* FIXME: pass this in from main() */
idxtype = mktype(n->loc, Tyuint64);
zero = mkintlit(n->loc, 0);
zero->expr.type = idxtype;
seq = rval(s, n->iterstmt.seq);
idx = gentemp(n->loc, idxtype, &dcl);
/* setup */
append(s, asn(idx, zero));
jmp(s, lcond);
flatten(s, lbody);
/* body */
flatten(s, n->iterstmt.body);
/* step */
flatten(s, lstep);
flatten(s, asn(idx, addk(idx, 1)));
/* condition */
flatten(s, lcond);
len = seqlen(s, seq, idxtype);
done = mkexpr(n->loc, Olt, idx, len, NULL);
done->expr.type = mktype(n->loc, Tybool);
cjmp(s, done, lmatch, lend);
flatten(s, lmatch);
val = mkexpr(n->loc, Oidx, seq, idx, NULL);
val->expr.type = tybase(exprtype(seq))->sub[0];
/* pattern match */
flattenloopmatch(s, n->iterstmt.elt, val, lbody, lstep);
jmp(s, lbody);
flatten(s, lend);
s->inloop--;
s->loop = l;
}
/* for pat in seq
* body;;
* =>
* .seq = seq
* .elt = elt
* :body
* ..body..
* :step
* __iterfin__(&seq, &elt)
* cond = __iternext__(&seq, &eltout)
* cjmp (cond) :match :end
* :match
* ...match...
* cjmp (match) :load :step
* :load
* ...load matches...
* :end
*/
static void
flattentraititer(Flattenctx *s, Node *n)
{
Loop l;
Node *lbody, *lclean, *lstep, *lmatch, *lend;
Node *done, *val, *iter, *valptr, *iterptr;
Node *func, *call, *seq;
Trait *tr;
val = temp(s, n->iterstmt.elt);
valptr = mkexpr(val->loc, Oaddr, val, NULL);
valptr->expr.type = mktyptr(n->loc, exprtype(val));
iter = temp(s, n->iterstmt.seq);
iterptr = mkexpr(val->loc, Oaddr, iter, NULL);
iterptr->expr.type = mktyptr(n->loc, exprtype(iter));
tr = traittab[Tciter];
/* create labels */
lbody = genlbl(n->loc);
lclean = genlbl(n->loc);
lstep = genlbl(n->loc);
lmatch = genlbl(n->loc);
lend = genlbl(n->loc);
s->inloop++;
l = s->loop;
s->loop.lcnt = lstep;
s->loop.lbrk = lend;
s->loop.body = n->iterstmt.body->block.scope;
seq = rval(s, n->iterstmt.seq);
append(s, asn(iter, seq));
jmp(s, lstep);
flatten(s, lbody);
/* body */
flatten(s, n->iterstmt.body);
flatten(s, lclean);
/* call iterator cleanup */
func = traitfn(n->loc, tr, "__iterfin__", exprtype(iter));
call = mkexpr(n->loc, Ocall, func, iterptr, valptr, NULL);
call->expr.type = mktype(n->loc, Tyvoid);
append(s, call);
flatten(s, lstep);
/* call iterator step */
func = traitfn(n->loc, tr, "__iternext__", exprtype(iter));
call = mkexpr(n->loc, Ocall, func, iterptr, valptr, NULL);
done = gentemp(n->loc, mktype(n->loc, Tybool), NULL);
call->expr.type = exprtype(done);
append(s, asn(done, call));
cjmp(s, done, lmatch, lend);
/* pattern match */
flatten(s, lmatch);
flattenloopmatch(s, n->iterstmt.elt, val, lbody, lclean);
jmp(s, lbody);
flatten(s, lend);
s->inloop--;
s->loop = l;
}
static void
flatteniter(Flattenctx *s, Node *n)
{
switch (tybase(exprtype(n->iterstmt.seq))->type) {
case Tyarray: flattenidxiter(s, n); break;
case Tyslice: flattenidxiter(s, n); break;
default: flattentraititer(s, n); break;
}
}
static void
flattenmatch(Flattenctx *fc, Node *n)
{
Node *v, *r, **match;
size_t i, nmatch;
r = rval(fc, n->matchstmt.val);
v = temp(fc, r);
flatten(fc, asn(v, r));
draininc(fc);
match = NULL;
nmatch = 0;
genmatch(n, v, &match, &nmatch);
for (i = 0; i < nmatch; i++)
flatten(fc, match[i]);
}
static void
flattenexpr(Flattenctx *fc, Node *n)
{
Node *r;
if (islbl(n)) {
append(fc, n);
return;
}
r = rval(fc, n);
if (r)
append(fc, r);
draininc(fc);
}
static Node *
flatten(Flattenctx *fc, Node *n)
{
Node *r, *u, *t;
if (!n)
return NULL;
r = NULL;
switch (n->type) {
case Nblock: flattenblk(fc, n); break;
case Nloopstmt: flattenloop(fc, n); break;
case Niterstmt: flatteniter(fc, n); break;
case Nifstmt: flattenif(fc, n, NULL); break;
case Nmatchstmt: flattenmatch(fc, n); break;
case Nexpr: flattenexpr(fc, n); break;
case Ndecl:
append(fc, n);
r = mkexpr(n->loc, Ovar, n->decl.name, NULL);
if (n->decl.init) {
t = rval(fc, n->decl.init);
u = mkexpr(n->loc, Oasn, r, t, NULL);
u->expr.type = n->decl.type;
r->expr.type = n->decl.type;
r->expr.did = n->decl.did;
flatten(fc, u);
}
break;
default:
dumpn(n, stderr);
die("bad node passsed to flatten()");
break;
}
return r;
}
static Node *
flatteninit(Node *dcl)
{
Flattenctx fc = {0,};
Node *lit, *fn, *blk, *body;
lit = dcl->decl.init->expr.args[0];
fn = lit->lit.fnval;
body = fn->func.body;
fc.curst = fn->func.scope;
flatten(&fc, fn->func.body);
blk = mkblock(fn->loc, body->block.scope);
blk->block.stmts = fc.stmts;
blk->block.nstmts = fc.nstmts;
fn->func.body = blk;
return dcl;
}
static int
ismain(Node *n)
{
n = n->decl.name;
if (n->name.ns)
return 0;
return strcmp(n->name.name, "main") == 0;
}
Node *
flattenfn(Node *dcl)
{
if (ismain(dcl))
dcl->decl.vis = Vishidden;
if (dcl->decl.isextern || dcl->decl.isgeneric)
return dcl;
if (isconstfn(dcl)) {
dcl = flatteninit(dcl);
//lappend(fn, nfn, f);
}
//lappend(fn, nfn, f);
return dcl;
}
int
isconstfn(Node *n)
{
Node *d, *e;
Type *t;
if (n->type == Nexpr) {
if (exprop(n) != Ovar)
return 0;
d = decls[n->expr.did];
} else {
d = n;
}
t = tybase(decltype(d));
if (!d || !d->decl.isconst || d->decl.isgeneric)
return 0;
if (!d->decl.isglobl && !d->decl.isextern)
return 0;
if (t->type != Tyfunc && t->type != Tycode)
return 0;
e = d->decl.init;
if (e && (exprop(e) != Olit || e->expr.args[0]->lit.littype != Lfunc))
return 0;
if (!e && !d->decl.isextern && !d->decl.isimport)
return 0;
return 1;
}