Improve bytecode generation.
This commit is contained in:
James Roseborough
@@ -3,9 +3,15 @@ package org.luaj.vm2.luajc;
|
|||||||
import java.io.ByteArrayOutputStream;
|
import java.io.ByteArrayOutputStream;
|
||||||
import java.io.PrintStream;
|
import java.io.PrintStream;
|
||||||
|
|
||||||
|
import org.luaj.vm2.Buffer;
|
||||||
import org.luaj.vm2.Lua;
|
import org.luaj.vm2.Lua;
|
||||||
|
import org.luaj.vm2.LuaClosure;
|
||||||
|
import org.luaj.vm2.LuaTable;
|
||||||
|
import org.luaj.vm2.LuaValue;
|
||||||
import org.luaj.vm2.Print;
|
import org.luaj.vm2.Print;
|
||||||
import org.luaj.vm2.Prototype;
|
import org.luaj.vm2.Prototype;
|
||||||
|
import org.luaj.vm2.TailcallVarargs;
|
||||||
|
import org.luaj.vm2.UpValue;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Prototype information for static single-assignment analysis
|
* Prototype information for static single-assignment analysis
|
||||||
@@ -17,8 +23,8 @@ public class ProtoInfo {
|
|||||||
public final ProtoInfo[] subprotos; // one per enclosed prototype, or null
|
public final ProtoInfo[] subprotos; // one per enclosed prototype, or null
|
||||||
public final BasicBlock[] blocks; // basic block analysis of code branching
|
public final BasicBlock[] blocks; // basic block analysis of code branching
|
||||||
public final BasicBlock[] blocklist; // blocks in breadhth-first order
|
public final BasicBlock[] blocklist; // blocks in breadhth-first order
|
||||||
public final VarInfo[][] vars; // Each variable
|
|
||||||
public final VarInfo[] params; // Parameters and initial values of stack variables
|
public final VarInfo[] params; // Parameters and initial values of stack variables
|
||||||
|
public final VarInfo[][] vars; // Each variable
|
||||||
public final UpvalInfo[] upvals; // from outer scope
|
public final UpvalInfo[] upvals; // from outer scope
|
||||||
public final UpvalInfo[][] openups; // per slot, upvalues allocated by this prototype
|
public final UpvalInfo[][] openups; // per slot, upvalues allocated by this prototype
|
||||||
|
|
||||||
@@ -46,9 +52,9 @@ public class ProtoInfo {
|
|||||||
this.blocklist[0].mergeSlotInput(slot, v);
|
this.blocklist[0].mergeSlotInput(slot, v);
|
||||||
}
|
}
|
||||||
|
|
||||||
// find variables and block inputs
|
// find variables
|
||||||
this.vars = findVariables();
|
this.vars = findVariables();
|
||||||
findBasicBlockInputs();
|
replaceTrivialPhiVariables();
|
||||||
|
|
||||||
// find upvalues, create sub-prototypes
|
// find upvalues, create sub-prototypes
|
||||||
this.openups = new UpvalInfo[p.maxstacksize][];
|
this.openups = new UpvalInfo[p.maxstacksize][];
|
||||||
@@ -121,152 +127,273 @@ public class ProtoInfo {
|
|||||||
v[i] = new VarInfo[n];
|
v[i] = new VarInfo[n];
|
||||||
|
|
||||||
// process instructions
|
// process instructions
|
||||||
for ( int pc=0; pc<n; pc++ ) {
|
for ( int bi=0; bi<blocklist.length; bi++ ) {
|
||||||
|
BasicBlock b0 = blocklist[bi];
|
||||||
|
|
||||||
// propogate previous value except at block boundaries
|
// input from previous blocks
|
||||||
if ( pc>0 && blocks[pc].pc0 != pc )
|
int nprev = b0.prev!=null? b0.prev.length: 0;
|
||||||
for ( int j=0; j<m; j++ )
|
for ( int slot=0; slot<m; slot++ ) {
|
||||||
v[j][pc] = v[j][pc-1];
|
VarInfo var = null;
|
||||||
|
if ( nprev == 0 )
|
||||||
|
var = params[slot];
|
||||||
|
else if ( nprev == 1 )
|
||||||
|
var = v[slot][b0.prev[0].pc1];
|
||||||
|
else {
|
||||||
|
for ( int i=0; i<nprev; i++ ) {
|
||||||
|
BasicBlock bp = b0.prev[i];
|
||||||
|
if ( v[slot][bp.pc1] == VarInfo.INVALID )
|
||||||
|
var = VarInfo.INVALID;
|
||||||
|
}
|
||||||
|
if ( var == null )
|
||||||
|
var = VarInfo.PHI(this, slot, b0.pc0);
|
||||||
|
}
|
||||||
|
v[slot][b0.pc0] = var;
|
||||||
|
}
|
||||||
|
|
||||||
// account for assignments and invalidations
|
// process instructions for this basic block
|
||||||
int a,b,c;
|
for ( int pc=b0.pc0; pc<=b0.pc1; pc++ ) {
|
||||||
int ins = prototype.code[pc];
|
|
||||||
switch ( Lua.GET_OPCODE( ins ) ) {
|
// propogate previous for all but block boundaries
|
||||||
case Lua.OP_MOVE:/* A B R(A) := R(B) */
|
if ( pc>b0.pc0 )
|
||||||
case Lua.OP_LOADK:/* A Bx R(A) := Kst(Bx) */
|
|
||||||
case Lua.OP_LOADBOOL:/* A B C R(A) := (Bool)B; if (C) pc++ */
|
|
||||||
case Lua.OP_GETUPVAL: /* A B R(A) := UpValue[B] */
|
|
||||||
case Lua.OP_GETGLOBAL: /* A Bx R(A) := Gbl[Kst(Bx)] */
|
|
||||||
case Lua.OP_GETTABLE: /* A B C R(A) := R(B)[RK(C)] */
|
|
||||||
case Lua.OP_NEWTABLE: /* A B C R(A) := {} (size = B,C) */
|
|
||||||
case Lua.OP_ADD: /* A B C R(A) := RK(B) + RK(C) */
|
|
||||||
case Lua.OP_SUB: /* A B C R(A) := RK(B) - RK(C) */
|
|
||||||
case Lua.OP_MUL: /* A B C R(A) := RK(B) * RK(C) */
|
|
||||||
case Lua.OP_DIV: /* A B C R(A) := RK(B) / RK(C) */
|
|
||||||
case Lua.OP_MOD: /* A B C R(A) := RK(B) % RK(C) */
|
|
||||||
case Lua.OP_POW: /* A B C R(A) := RK(B) ^ RK(C) */
|
|
||||||
case Lua.OP_UNM: /* A B R(A) := -R(B) */
|
|
||||||
case Lua.OP_NOT: /* A B R(A) := not R(B) */
|
|
||||||
case Lua.OP_LEN: /* A B R(A) := length of R(B) */
|
|
||||||
case Lua.OP_CONCAT: /* A B C R(A) := R(B).. ... ..R(C) */
|
|
||||||
case Lua.OP_TESTSET: /* A B C if (R(B) <=> C) then R(A) := R(B) else pc++ */
|
|
||||||
case Lua.OP_FORPREP: /* A sBx R(A)-=R(A+2); pc+=sBx */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
break;
|
|
||||||
case Lua.OP_SELF: /* A B C R(A+1) := R(B); R(A) := R(B)[RK(C)] */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
v[a+1][pc] = new VarInfo(a+1,pc);
|
|
||||||
break;
|
|
||||||
case Lua.OP_FORLOOP: /* A sBx R(A)+=R(A+2);
|
|
||||||
if R(A) <?= R(A+1) then { pc+=sBx; R(A+3)=R(A) }*/
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
v[a+3][pc] = new VarInfo(a+1,pc);
|
|
||||||
break;
|
|
||||||
case Lua.OP_LOADNIL: /* A B R(A) := ... := R(B) := nil */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
b = Lua.GETARG_B( ins );
|
|
||||||
for ( ; a<=b; a++ )
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
break;
|
|
||||||
case Lua.OP_VARARG: /* A B R(A), R(A+1), ..., R(A+B-1) = vararg */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
b = Lua.GETARG_B( ins );
|
|
||||||
for ( int j=1; j<b; j++, a++ )
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
if ( b == 0 )
|
|
||||||
for ( ; a<m; a++ )
|
|
||||||
v[a][pc] = VarInfo.INVALID;
|
|
||||||
break;
|
|
||||||
case Lua.OP_CALL: /* A B C R(A), ... ,R(A+C-2) := R(A)(R(A+1), ... ,R(A+B-1)) */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
c = Lua.GETARG_C( ins );
|
|
||||||
for ( int j=0; j<=c-2; j++, a++ )
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
for ( ; a<m; a++ )
|
|
||||||
v[a][pc] = VarInfo.INVALID;
|
|
||||||
break;
|
|
||||||
case Lua.OP_TFORLOOP: /* A C R(A+3), ... ,R(A+2+C) := R(A)(R(A+1), R(A+2));
|
|
||||||
if R(A+3) ~= nil then R(A+2)=R(A+3) else pc++ */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
c = Lua.GETARG_C( ins );
|
|
||||||
a += 3;
|
|
||||||
for ( int j=0; j<c; j++, a++ )
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
for ( ; a<m; a++ )
|
|
||||||
v[a][pc] = VarInfo.INVALID;
|
|
||||||
break;
|
|
||||||
case Lua.OP_CLOSURE: /* A Bx R(A) := closure(KPROTO[Bx], R(A), ... ,R(A+n)) */
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
b = Lua.GETARG_Bx( ins );
|
|
||||||
v[a][pc] = new VarInfo(a,pc);
|
|
||||||
for ( int k=prototype.p[b].nups; --k>=0; ) {
|
|
||||||
++pc;
|
|
||||||
for ( int j=0; j<m; j++ )
|
for ( int j=0; j<m; j++ )
|
||||||
v[j][pc] = v[j][pc-1];
|
v[j][pc] = v[j][pc-1];
|
||||||
|
|
||||||
|
int a,b,c,nups;
|
||||||
|
int ins = prototype.code[pc];
|
||||||
|
int op = Lua.GET_OPCODE(ins);
|
||||||
|
|
||||||
|
// account for assignments, references and invalidations
|
||||||
|
switch ( op ) {
|
||||||
|
case Lua.OP_LOADK:/* A Bx R(A) := Kst(Bx) */
|
||||||
|
case Lua.OP_LOADBOOL:/* A B C R(A) := (Bool)B; if (C) pc++ */
|
||||||
|
case Lua.OP_GETUPVAL: /* A B R(A) := UpValue[B] */
|
||||||
|
case Lua.OP_GETGLOBAL: /* A Bx R(A) := Gbl[Kst(Bx)] */
|
||||||
|
case Lua.OP_NEWTABLE: /* A B C R(A) := {} (size = B,C) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_MOVE:/* A B R(A) := R(B) */
|
||||||
|
case Lua.OP_UNM: /* A B R(A) := -R(B) */
|
||||||
|
case Lua.OP_NOT: /* A B R(A) := not R(B) */
|
||||||
|
case Lua.OP_LEN: /* A B R(A) := length of R(B) */
|
||||||
|
case Lua.OP_TESTSET: /* A B C if (R(B) <=> C) then R(A) := R(B) else pc++ */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
v[b][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_ADD: /* A B C R(A) := RK(B) + RK(C) */
|
||||||
|
case Lua.OP_SUB: /* A B C R(A) := RK(B) - RK(C) */
|
||||||
|
case Lua.OP_MUL: /* A B C R(A) := RK(B) * RK(C) */
|
||||||
|
case Lua.OP_DIV: /* A B C R(A) := RK(B) / RK(C) */
|
||||||
|
case Lua.OP_MOD: /* A B C R(A) := RK(B) % RK(C) */
|
||||||
|
case Lua.OP_POW: /* A B C R(A) := RK(B) ^ RK(C) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
if (!Lua.ISK(b)) v[b][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(c)) v[c][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_SETTABLE: /* A B C R(A)[RK(B)]:= RK(C) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(b)) v[b][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(c)) v[c][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_CONCAT: /* A B C R(A) := R(B).. ... ..R(C) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
for ( ; b<=c; b++ )
|
||||||
|
v[b][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_FORPREP: /* A sBx R(A)-=R(A+2); pc+=sBx */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
v[a+2][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_GETTABLE: /* A B C R(A) := R(B)[RK(C)] */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
v[b][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(c)) v[c][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_SELF: /* A B C R(A+1) := R(B); R(A) := R(B)[RK(C)] */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
v[b][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(c)) v[c][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
v[a+1][pc] = new VarInfo(a+1,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_FORLOOP: /* A sBx R(A)+=R(A+2);
|
||||||
|
if R(A) <?= R(A+1) then { pc+=sBx; R(A+3)=R(A) }*/
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
v[a+2][pc].isreferenced = true;
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
v[a+3][pc] = new VarInfo(a+1,pc);
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
v[a+1][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_LOADNIL: /* A B R(A) := ... := R(B) := nil */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
for ( ; a<=b; a++ )
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_VARARG: /* A B R(A), R(A+1), ..., R(A+B-1) = vararg */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
for ( int j=1; j<b; j++, a++ )
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
if ( b == 0 )
|
||||||
|
for ( ; a<m; a++ )
|
||||||
|
v[a][pc] = VarInfo.INVALID;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_CALL: /* A B C R(A), ... ,R(A+C-2) := R(A)(R(A+1), ... ,R(A+B-1)) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
for ( int i=1; i<=b-1; i++ )
|
||||||
|
v[a+i][pc].isreferenced = true;
|
||||||
|
for ( int j=0; j<=c-2; j++, a++ )
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
for ( ; a<m; a++ )
|
||||||
|
v[a][pc] = VarInfo.INVALID;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_TAILCALL: /* A B C return R(A)(R(A+1), ... ,R(A+B-1)) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
for ( int i=1; i<=b-1; i++ )
|
||||||
|
v[a+i][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_RETURN: /* A B return R(A), ... ,R(A+B-2) (see note) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
for ( int i=0; i<=b-2; i++ )
|
||||||
|
v[a+i][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_TFORLOOP: /* A C R(A+3), ... ,R(A+2+C) := R(A)(R(A+1), R(A+2));
|
||||||
|
if R(A+3) ~= nil then R(A+2)=R(A+3) else pc++ */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
v[a++][pc].isreferenced = true;
|
||||||
|
v[a++][pc].isreferenced = true;
|
||||||
|
v[a++][pc].isreferenced = true;
|
||||||
|
for ( int j=0; j<c; j++, a++ )
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
for ( ; a<m; a++ )
|
||||||
|
v[a][pc] = VarInfo.INVALID;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_CLOSURE: /* A Bx R(A) := closure(KPROTO[Bx], R(A), ... ,R(A+n)) */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_Bx( ins );
|
||||||
|
nups = prototype.p[b].nups;
|
||||||
|
for ( int k=1; k<=nups; ++k ) {
|
||||||
|
int i = prototype.code[pc+k];
|
||||||
|
if ( (i&4) == 0 ) {
|
||||||
|
b = Lua.GETARG_B(i);
|
||||||
|
v[b][pc].isreferenced = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
v[a][pc] = new VarInfo(a,pc);
|
||||||
|
for ( int k=1; k<=nups; ++k ) {
|
||||||
|
for ( int j=0; j<m; j++ )
|
||||||
|
v[j][pc+k] = v[j][pc];
|
||||||
|
}
|
||||||
|
pc += nups;
|
||||||
|
break;
|
||||||
|
case Lua.OP_CLOSE: /* A close all variables in the stack up to (>=) R(A)*/
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
for ( ; a<m; a++ )
|
||||||
|
v[a][pc] = VarInfo.INVALID;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_SETLIST: /* A B C R(A)[(C-1)*FPF+i]:= R(A+i), 1 <= i <= B */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
for ( int i=1; i<=b; i++ )
|
||||||
|
v[a+i][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_SETGLOBAL: /* A Bx Gbl[Kst(Bx)]:= R(A) */
|
||||||
|
case Lua.OP_SETUPVAL: /* A B UpValue[B]:= R(A) */
|
||||||
|
case Lua.OP_TEST: /* A C if not (R(A) <=> C) then pc++ */
|
||||||
|
a = Lua.GETARG_A( ins );
|
||||||
|
v[a][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_EQ: /* A B C if ((RK(B) == RK(C)) ~= A) then pc++ */
|
||||||
|
case Lua.OP_LT: /* A B C if ((RK(B) < RK(C)) ~= A) then pc++ */
|
||||||
|
case Lua.OP_LE: /* A B C if ((RK(B) <= RK(C)) ~= A) then pc++ */
|
||||||
|
b = Lua.GETARG_B( ins );
|
||||||
|
c = Lua.GETARG_C( ins );
|
||||||
|
if (!Lua.ISK(b)) v[b][pc].isreferenced = true;
|
||||||
|
if (!Lua.ISK(c)) v[c][pc].isreferenced = true;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Lua.OP_JMP: /* sBx pc+=sBx */
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
throw new IllegalStateException("unhandled opcode: "+ins);
|
||||||
}
|
}
|
||||||
break;
|
|
||||||
case Lua.OP_CLOSE:
|
|
||||||
a = Lua.GETARG_A( ins );
|
|
||||||
for ( ; a<m; a++ )
|
|
||||||
v[a][pc] = VarInfo.INVALID;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return v;
|
return v;
|
||||||
}
|
}
|
||||||
|
|
||||||
private void findBasicBlockInputs() {
|
private void replaceTrivialPhiVariables() {
|
||||||
|
for ( int i=0; i<blocklist.length; i++ ) {
|
||||||
for ( boolean changed=true; changed; ) {
|
BasicBlock b0 = blocklist[i];
|
||||||
changed = false;
|
for ( int slot=0; slot<prototype.maxstacksize; slot++ ) {
|
||||||
|
VarInfo vold = vars[slot][b0.pc1];
|
||||||
// send inputs to next stage
|
VarInfo vnew = vold.resolvePhiVariableValues();
|
||||||
for ( int i=0; i<blocklist.length; i++ ) {
|
if ( vnew != null )
|
||||||
BasicBlock b0 = blocklist[i];
|
substituteVariable( slot, vold, vnew );
|
||||||
for ( int k=0, n=b0.next!=null? b0.next.length: 0; k<n; k++ ) {
|
|
||||||
BasicBlock b1 = b0.next[k];
|
|
||||||
for ( int slot=0; slot<prototype.maxstacksize; slot++ ) {
|
|
||||||
VarInfo v = vars[slot][b0.pc1];
|
|
||||||
if ( v != null ) {
|
|
||||||
if ( b1.mergeSlotInput(slot, v) ) {
|
|
||||||
if ( vars[slot][b1.pc0] == null )
|
|
||||||
if ( b1.ninputs[slot] == 2 || v == VarInfo.INVALID )
|
|
||||||
createInputVar( b1, slot );
|
|
||||||
changed = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// propogate up to one more variable per slot
|
|
||||||
if ( ! changed ) {
|
|
||||||
eachslot: for ( int slot=0; slot<prototype.maxstacksize; slot++ ) {
|
|
||||||
for ( int i=0; i<blocklist.length; i++ ) {
|
|
||||||
BasicBlock b0 = blocklist[i];
|
|
||||||
if ( vars[slot][b0.pc0] == null ) {
|
|
||||||
createInputVar( b0, slot );
|
|
||||||
changed = true;
|
|
||||||
continue eachslot;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private void createInputVar(BasicBlock b, int slot) {
|
private void substituteVariable(int slot, VarInfo vold, VarInfo vnew) {
|
||||||
int n = b.ninputs[slot];
|
for ( int i=0, n=prototype.code.length; i<n; i++ )
|
||||||
VarInfo v =
|
replaceAll( vars[slot], vars[slot].length, vold, vnew );
|
||||||
n==-1? VarInfo.INVALID:
|
for ( int i=0; i<blocklist.length; i++ ) {
|
||||||
n==0? VarInfo.NIL(slot):
|
BasicBlock b = blocklist[i];
|
||||||
n==1? b.inputs[slot][0]:
|
replaceAll( b.inputs[slot], b.ninputs[slot], vold, vnew );
|
||||||
VarInfo.PHI(slot,b.pc0);
|
}
|
||||||
for ( int pc=b.pc0; pc<=b.pc1 && vars[slot][pc] == null; ++pc )
|
}
|
||||||
vars[slot][pc] = v;
|
|
||||||
|
private void replaceAll(VarInfo[] v, int n, VarInfo vold, VarInfo vnew) {
|
||||||
|
for ( int i=0; i<n; i++ )
|
||||||
|
if ( v[i] == vold )
|
||||||
|
v[i] = vnew;
|
||||||
}
|
}
|
||||||
|
|
||||||
private void findUpvalues() {
|
private void findUpvalues() {
|
||||||
@@ -317,13 +444,23 @@ public class ProtoInfo {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public boolean isUpvalueRefer(int pc, int slot) {
|
public boolean isUpvalueRefer(int pc, int slot) {
|
||||||
// TODO: when it is a CALL
|
// special case for opcodes where overlap between refer and assign
|
||||||
|
if ( pc >= 0 ) {
|
||||||
|
switch ( Lua.GET_OPCODE(prototype.code[pc]) ) {
|
||||||
|
case Lua.OP_CLOSURE:
|
||||||
|
case Lua.OP_CALL:
|
||||||
|
case Lua.OP_TFORLOOP:
|
||||||
|
pc -= 1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
VarInfo v = pc<0? params[slot]: vars[slot][pc];
|
VarInfo v = pc<0? params[slot]: vars[slot][pc];
|
||||||
// return v.upvalue != null && v.upvalue.rw;
|
// return v.upvalue != null && v.upvalue.rw;
|
||||||
return v != null && v.upvalue != null;
|
return v != null && v.upvalue != null;
|
||||||
}
|
}
|
||||||
|
|
||||||
public boolean isInitialValueUsed(int slot) {
|
public boolean isInitialValueUsed(int slot) {
|
||||||
return true;
|
VarInfo v = params[slot];
|
||||||
|
return v.isreferenced;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,39 +3,42 @@
|
|||||||
*/
|
*/
|
||||||
package org.luaj.vm2.luajc;
|
package org.luaj.vm2.luajc;
|
||||||
|
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.Set;
|
||||||
|
|
||||||
public class VarInfo {
|
public class VarInfo {
|
||||||
|
|
||||||
public static VarInfo INVALID = new VarInfo(-1,-1);
|
public static VarInfo INVALID = new VarInfo(-1, -1);
|
||||||
|
|
||||||
public static VarInfo PARAM(int slot) {
|
public static VarInfo PARAM(int slot) {
|
||||||
return new VarInfo(slot,-1) {
|
return new VarInfo(slot, -1) {
|
||||||
public String toString() {
|
public String toString() {
|
||||||
return slot+".p";
|
return slot + ".p";
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
public static VarInfo NIL(final int slot) {
|
public static VarInfo NIL(final int slot) {
|
||||||
return new VarInfo(slot,-1) {
|
return new VarInfo(slot, -1) {
|
||||||
public String toString() {
|
public String toString() {
|
||||||
return "nil";
|
return "nil";
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
public static VarInfo PHI(int slot, int pc) {
|
public static VarInfo PHI(final ProtoInfo pi, final int slot, final int pc) {
|
||||||
return new VarInfo(slot,pc) {
|
return new PhiVarInfo(pi, slot, pc);
|
||||||
public String toString() {
|
|
||||||
return super.toString()+"{}";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public final int slot; // where assigned
|
public final int slot; // where assigned
|
||||||
public final int pc; // where assigned, or -1 if for block inputs
|
public final int pc; // where assigned, or -1 if for block inputs
|
||||||
|
|
||||||
public UpvalInfo upvalue; // not null if this var is an upvalue
|
public UpvalInfo upvalue; // not null if this var is an upvalue
|
||||||
public boolean allocupvalue; // true if this variable allocates r/w upvalue storage
|
public boolean allocupvalue; // true if this variable allocates r/w upvalue
|
||||||
|
// storage
|
||||||
|
public boolean isreferenced; // true if this variable is refenced by some
|
||||||
|
// opcode
|
||||||
|
|
||||||
public VarInfo(int slot, int pc) {
|
public VarInfo(int slot, int pc) {
|
||||||
this.slot = slot;
|
this.slot = slot;
|
||||||
@@ -43,7 +46,77 @@ public class VarInfo {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public String toString() {
|
public String toString() {
|
||||||
return slot<0? "x.x":
|
return slot < 0 ? "x.x" : (slot + "." + pc);
|
||||||
(slot+"."+pc);
|
}
|
||||||
|
|
||||||
|
/** Return replacement variable if there is exactly one value possible,
|
||||||
|
* otherwise compute entire collection of variables and return null.
|
||||||
|
* Computes the list of aall variable values, and saves it for the future.
|
||||||
|
*
|
||||||
|
* @return new Variable to replace with if there is only one value, or null to leave alone.
|
||||||
|
*/
|
||||||
|
public VarInfo resolvePhiVariableValues() {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected void collectUniqueValues(Set visitedBlocks, Set vars) {
|
||||||
|
vars.add(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static final class PhiVarInfo extends VarInfo {
|
||||||
|
private final ProtoInfo pi;
|
||||||
|
VarInfo[] values;
|
||||||
|
|
||||||
|
private PhiVarInfo(ProtoInfo pi, int slot, int pc) {
|
||||||
|
super(slot, pc);
|
||||||
|
this.pi = pi;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String toString() {
|
||||||
|
StringBuffer sb = new StringBuffer();
|
||||||
|
sb.append( super.toString() );
|
||||||
|
sb.append("={");
|
||||||
|
for (int i=0, n=(values!=null? values.length : 0); i<n; i++) {
|
||||||
|
if ( i>0 )
|
||||||
|
sb.append( "," );
|
||||||
|
sb.append(String.valueOf(values[i]));
|
||||||
|
}
|
||||||
|
sb.append("}");
|
||||||
|
return sb.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
public VarInfo resolvePhiVariableValues() {
|
||||||
|
Set visitedBlocks = new HashSet();
|
||||||
|
Set vars = new HashSet();
|
||||||
|
this.collectUniqueValues(visitedBlocks, vars);
|
||||||
|
if (vars.contains(INVALID))
|
||||||
|
return INVALID;
|
||||||
|
int n = vars.size();
|
||||||
|
Iterator it = vars.iterator();
|
||||||
|
if (n == 1) {
|
||||||
|
VarInfo v = (VarInfo) it.next();
|
||||||
|
v.isreferenced |= this.isreferenced;
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
this.values = new VarInfo[n];
|
||||||
|
for ( int i=0; i<n; i++ ) {
|
||||||
|
this.values[i] = (VarInfo) it.next();
|
||||||
|
this.values[i].isreferenced |= this.isreferenced;
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected void collectUniqueValues(Set visitedBlocks, Set vars) {
|
||||||
|
BasicBlock b = pi.blocks[pc];
|
||||||
|
for (int i = 0, n = b.prev != null ? b.prev.length : 0; i < n; i++) {
|
||||||
|
BasicBlock bp = b.prev[i];
|
||||||
|
if (!visitedBlocks.contains(bp)) {
|
||||||
|
visitedBlocks.add(bp);
|
||||||
|
VarInfo v = pi.vars[slot][bp.pc1];
|
||||||
|
if ( v != null )
|
||||||
|
v.collectUniqueValues(visitedBlocks, vars);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user