comparison src/luan/interp/GenericForStmt.java @ 19:a7c13c6017f7

add GenericForStmt git-svn-id: https://luan-java.googlecode.com/svn/trunk@20 21e917c8-12df-6dd8-5cb6-c86387c605b9
author fschmidt@gmail.com <fschmidt@gmail.com@21e917c8-12df-6dd8-5cb6-c86387c605b9>
date Sun, 02 Dec 2012 10:08:24 +0000
parents
children d85510d92eee
comparison
equal deleted inserted replaced
18:3971113699b8 19:a7c13c6017f7
1 package luan.interp;
2
3 import luan.Lua;
4 import luan.LuaState;
5 import luan.LuaException;
6 import luan.LuaFunction;
7
8
9 final class GenericForStmt implements Stmt {
10 private final int iVars;
11 private final int nVars;
12 private final Expr iterExpr;
13 private final Stmt block;
14
15 GenericForStmt(int iVars,int nVars,Expr iterExpr,Stmt block) {
16 this.iVars = iVars;
17 this.nVars = nVars;
18 this.iterExpr = iterExpr;
19 this.block = block;
20 }
21
22 @Override public void eval(LuaState lua) throws LuaException {
23 LuaFunction iter = Lua.checkFunction( iterExpr.eval(lua) );
24 Object[] stack = lua.stack();
25 try {
26 while(true) {
27 Object[] vals = iter.call(lua);
28 if( vals.length==0 || vals[0]==null )
29 break;
30 for( int i=0; i<nVars; i++ ) {
31 stack[iVars+i] = i < vals.length ? vals[i] : null;
32 }
33 block.eval(lua);
34 }
35 } finally {
36 for( int i=iVars; i<iVars+nVars; i++ ) {
37 stack[i] = null;
38 }
39 }
40 }
41
42 }