GitHub/lua
1
0
Fork 0
mirror of https://github.com/lua/lua.git synced 2025-01-28 06:03:00 +08:00
Code Issues Projects Releases Wiki Activity

`nexti' returns correct indices

Browse Source
This commit is contained in:
Roberto Ierusalimschy 2002-05-13 10:10:58 -03:00
parent c18fe57e54
commit 8da6fe62d8
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lbaselib.c
View File

@ -1,5 +1,5 @@
/*
** $Id: lbaselib.c,v 1.71 2002/05/02 17:12:27 roberto Exp roberto $
** $Id: lbaselib.c,v 1.72 2002/05/06 19:05:10 roberto Exp roberto $
** Basic library
** See Copyright Notice in lua.h
*/
@ -198,14 +198,15 @@ static int luaB_next (lua_State *L) {
static int luaB_nexti (lua_State *L) {
lua_Number i = lua_tonumber(L, 2);
luaL_check_type(L, 1, LUA_TTABLE);
if (i == 0) { /* `for' start? */
if (i == 0 && lua_isnull(L, 2)) { /* `for' start? */
lua_getglobal(L, "nexti"); /* return generator, */
lua_pushvalue(L, 1); /* state, */
lua_pushnumber(L, 1); /* and initial value */
lua_pushnumber(L, 0); /* and initial value */
return 3;
}
else { /* `for' step */
lua_pushnumber(L, i+1); /* next value */
i++; /* next value */
lua_pushnumber(L, i);
lua_rawgeti(L, 1, i);
return (lua_isnil(L, -1)) ? 0 : 2;
}