Mercurial Hosting > luan
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 } |