Improve __concat metatag
This commit is contained in:
@@ -583,14 +583,19 @@ public class JavaCodeGen {
|
||||
case Lua.OP_NEQ: out(evalLuaValue(exp.lhs)+".neq("+evalLuaValue(exp.rhs)+")"); return;
|
||||
case Lua.OP_CONCAT:
|
||||
if ( isConcatExp(exp.rhs) ) {
|
||||
out( "new Buffer().append("+evalLuaValue(exp.lhs)+")" );
|
||||
out( evalLuaValue(exp.lhs) );
|
||||
Exp e = exp.rhs;
|
||||
for ( ; isConcatExp(e); e=((BinopExp)e).rhs )
|
||||
out( ".append("+evalLuaValue(((BinopExp)e).lhs)+")" );
|
||||
out( ".append("+evalLuaValue(e)+")" );
|
||||
out( ".tostring()" );
|
||||
} else
|
||||
out(evalLuaValue(exp.lhs)+".concat("+evalLuaValue(exp.rhs)+")");
|
||||
String close = "";
|
||||
for ( ; isConcatExp(e); e=((BinopExp)e).rhs ) {
|
||||
out( ".concat("+evalLuaValue(((BinopExp)e).lhs) );
|
||||
close += ')';
|
||||
}
|
||||
out( ".concat("+evalLuaValue(e)+".buffer())" );
|
||||
out( close );
|
||||
out( ".value()" );
|
||||
} else {
|
||||
out(evalLuaValue(exp.lhs)+".concat("+evalLuaValue(exp.rhs)+")");
|
||||
}
|
||||
return;
|
||||
default: throw new IllegalStateException("unknown bin op:"+exp.op);
|
||||
}
|
||||
|
||||
@@ -109,7 +109,6 @@ public class JavaBuilder {
|
||||
private static final Type[] ARG_TYPES_STRING = { Type.STRING };
|
||||
private static final Type[] ARG_TYPES_CHARARRAY = { TYPE_CHARARRAY };
|
||||
private static final Type[] ARG_TYPES_VARARGS_INT = { TYPE_VARARGS, Type.INT };
|
||||
private static final Type[] ARG_TYPES_LUASTRING = { TYPE_LUASTRING };
|
||||
private static final Type[] ARG_TYPES_INT_LUAVALUE = { Type.INT, TYPE_LUAVALUE };
|
||||
private static final Type[] ARG_TYPES_INT_VARARGS = { Type.INT, TYPE_VARARGS };
|
||||
private static final Type[] ARG_TYPES_LUAVALUE_VARARGS = { TYPE_LUAVALUE, TYPE_VARARGS };
|
||||
@@ -121,6 +120,7 @@ public class JavaBuilder {
|
||||
private static final Type[] ARG_TYPES_LUAVALUE_LUAVALUE = { TYPE_LUAVALUE, TYPE_LUAVALUE };
|
||||
private static final Type[] ARG_TYPES_INT_INT = { Type.INT, Type.INT };
|
||||
private static final Type[] ARG_TYPES_LUAVALUE = { TYPE_LUAVALUE };
|
||||
private static final Type[] ARG_TYPES_BUFFER = { TYPE_BUFFER };
|
||||
|
||||
// names, arg types for main prototype classes
|
||||
private static final String[] SUPER_NAME_N = { STR_FUNC0, STR_FUNC1, STR_FUNC2, STR_FUNC3, STR_FUNCV, };
|
||||
@@ -773,14 +773,19 @@ public class JavaBuilder {
|
||||
append(factory.createInvoke(STR_LUAVALUE, "rawsetlist", Type.VOID, ARG_TYPES_INT_VARARGS, Constants.INVOKEVIRTUAL));
|
||||
}
|
||||
|
||||
public void newBuffer() {
|
||||
append(factory.createNew(TYPE_BUFFER));
|
||||
append(InstructionConstants.DUP);
|
||||
append(factory.createInvoke(STR_BUFFER, "<init>", Type.VOID, Type.NO_ARGS, Constants.INVOKESPECIAL));
|
||||
public void concatvalue() {
|
||||
append(factory.createInvoke(STR_LUAVALUE, "concat", TYPE_LUAVALUE, ARG_TYPES_LUAVALUE, Constants.INVOKEVIRTUAL));
|
||||
}
|
||||
|
||||
public void concatbuffer() {
|
||||
append(factory.createInvoke(STR_LUAVALUE, "concat", TYPE_BUFFER, ARG_TYPES_BUFFER, Constants.INVOKEVIRTUAL));
|
||||
}
|
||||
|
||||
public void appendBuffer() {
|
||||
append(factory.createInvoke(STR_LUAVALUE, "checkstring", TYPE_LUASTRING, Type.NO_ARGS, Constants.INVOKEVIRTUAL));
|
||||
append(factory.createInvoke(STR_BUFFER, "append", TYPE_BUFFER, ARG_TYPES_LUASTRING, Constants.INVOKEVIRTUAL));
|
||||
public void tobuffer() {
|
||||
append(factory.createInvoke(STR_LUAVALUE, "buffer", TYPE_BUFFER, Type.NO_ARGS, Constants.INVOKEVIRTUAL));
|
||||
}
|
||||
|
||||
public void tovalue() {
|
||||
append(factory.createInvoke(STR_BUFFER, "value", TYPE_LUAVALUE, Type.NO_ARGS, Constants.INVOKEVIRTUAL));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -21,6 +21,7 @@
|
||||
******************************************************************************/
|
||||
package org.luaj.vm2.luajc;
|
||||
|
||||
import org.luaj.vm2.Buffer;
|
||||
import org.luaj.vm2.Lua;
|
||||
import org.luaj.vm2.Prototype;
|
||||
|
||||
@@ -180,12 +181,16 @@ public class JavaGen {
|
||||
break;
|
||||
|
||||
case Lua.OP_CONCAT: /* A B C R(A):= R(B).. ... ..R(C) */
|
||||
builder.newBuffer();
|
||||
while ( b<=c ) {
|
||||
builder.loadLocal(pc, b++);
|
||||
builder.appendBuffer();
|
||||
for ( int k=b; k<=c; k++ )
|
||||
builder.loadLocal(pc, k);
|
||||
if ( c > b+1 ) {
|
||||
builder.tobuffer();
|
||||
for ( int k=c; --k>=b; )
|
||||
builder.concatbuffer();
|
||||
builder.tovalue();
|
||||
} else {
|
||||
builder.concatvalue();
|
||||
}
|
||||
builder.tostring();
|
||||
builder.storeLocal(pc, a);
|
||||
break;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user