120 lines
2.8 KiB
C
120 lines
2.8 KiB
C
#include "mcg.h"
|
|
|
|
static PMAPOF(struct basicblock, struct ir) pops;
|
|
static PMAPOF(struct basicblock, struct ir) pushes;
|
|
|
|
static struct ir* get_last_push(struct basicblock* bb)
|
|
{
|
|
int i;
|
|
|
|
for (i=bb->irs.count-1; i>=0; i--)
|
|
{
|
|
struct ir* ir = bb->irs.item[i];
|
|
|
|
if (ir->opcode == IR_PUSH)
|
|
return ir;
|
|
if (ir_find(ir, IR_POP))
|
|
return NULL;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static struct ir* get_first_pop(struct basicblock* bb)
|
|
{
|
|
int i;
|
|
|
|
for (i=0; i<bb->irs.count; i++)
|
|
{
|
|
struct ir* irr;
|
|
struct ir* ir = bb->irs.item[i];
|
|
|
|
if (ir->opcode == IR_PUSH)
|
|
return NULL;
|
|
|
|
irr = ir_find(ir, IR_POP);
|
|
if (irr)
|
|
return irr;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void convert_block(struct procedure* proc, struct basicblock* bb)
|
|
{
|
|
int i, j;
|
|
struct ir* ir;
|
|
|
|
pushes.count = pops.count = 0;
|
|
for (;;)
|
|
{
|
|
struct ir* lastpush = get_last_push(bb);
|
|
if (!lastpush)
|
|
return;
|
|
|
|
/* Abort unless *every* successor block of this one starts with a pop
|
|
* of the same size... */
|
|
|
|
for (i=0; i<bb->nexts.count; i++)
|
|
{
|
|
struct basicblock* outbb = bb->nexts.item[i];
|
|
|
|
ir = get_first_pop(outbb);
|
|
if (!ir || (ir->size != lastpush->size))
|
|
return;
|
|
pmap_add(&pops, outbb, ir);
|
|
|
|
/* Also abort unless *every* predecessor block of the one we've
|
|
* just found *also* ends in a push of the same size. */
|
|
|
|
for (j=0; j<outbb->prevs.count; j++)
|
|
{
|
|
struct basicblock* inbb = outbb->prevs.item[j];
|
|
|
|
ir = get_last_push(inbb);
|
|
if (!ir || (ir->size != lastpush->size))
|
|
return;
|
|
pmap_add(&pushes, inbb, ir);
|
|
}
|
|
}
|
|
|
|
/* If we didn't actually find anything, give up. */
|
|
|
|
if ((pushes.count == 0) || (pops.count == 0))
|
|
return;
|
|
|
|
/* Okay, now we can wire them all up. */
|
|
|
|
for (i=0; i<pushes.count; i++)
|
|
{
|
|
struct ir* ir = pushes.item[i].right;
|
|
*ir = *ir->left;
|
|
}
|
|
|
|
for (i=0; i<pops.count; i++)
|
|
{
|
|
struct basicblock* outbb = pops.item[i].left;
|
|
struct ir* ir = pops.item[i].right;
|
|
struct ir* phi = new_ir0(IR_PHI, ir->size);
|
|
|
|
for (j=0; j<pushes.count; j++)
|
|
pmap_add(&phi->u.phivalue,
|
|
pushes.item[j].left,
|
|
pushes.item[j].right);
|
|
phi->root = phi;
|
|
|
|
*ir = *phi;
|
|
}
|
|
}
|
|
}
|
|
|
|
void pass_convert_stack_ops(struct procedure* proc)
|
|
{
|
|
int i;
|
|
|
|
for (i=0; i<proc->blocks.count; i++)
|
|
convert_block(proc, proc->blocks.item[i]);
|
|
}
|
|
|
|
/* vim: set sw=4 ts=4 expandtab : */
|