static void forlist (LexState *ls, TString *indexname)
{
/* forlist -> NAME {,NAME} IN explist1 forbody */
FuncState *fs = GetCurrentFuncState( ls );
expdesc e;
int nvars = 0;
int base = fs->freereg;
/* create control variables */
new_localvarliteral(ls, "(for generator)", nvars++);
new_localvarliteral(ls, "(for state)", nvars++);
new_localvarliteral(ls, "(for control)", nvars++);
/* create declared variables */
new_localvar(ls, indexname, nvars++);
while (testnext(ls, ','))
{
new_localvar(ls, str_checkname(ls), nvars++);
}
checknext(ls, TK_IN);
int line = ls->linenumber;
adjust_assign(ls, 3, explist1(ls, &e), &e);
luaK_checkstack(fs, 3); /* extra space to call generator */
forbody(ls, base, line, nvars - 3, 0);
}
static void forlist (LexState *ls, TString *indexname) {
/* forlist -> NAME {,NAME} IN explist1 DO body */
FuncState *fs = ls->fs;
expdesc e;
int nvars = 0;
int line;
int base = fs->freereg;
new_localvarstr(ls, "(for generator)", nvars++);
new_localvarstr(ls, "(for state)", nvars++);
new_localvar(ls, indexname, nvars++);
while (testnext(ls, ','))
new_localvar(ls, str_checkname(ls), nvars++);
check(ls, TK_IN);
line = ls->linenumber;
adjust_assign(ls, nvars, explist1(ls, &e), &e);
luaK_checkstack(fs, 3); /* extra space to call generator */
luaK_codeAsBx(fs, OP_TFORPREP, base, NO_JUMP);
forbody(ls, base, line, nvars, 0);
}
static void new_localvar (LexState *ls, TString *name) {
luaK_checkstack(ls->fs, 1);
registerlocalvar(ls, name);
}
/*
** Reserve 'n' registers in register stack
*/
void luaK_reserveregs (FuncState *fs, int n) {
luaK_checkstack(fs, n);
fs->freereg += n;
}
void luaK_reserveregs (FuncState *fs, int n) {
luaK_checkstack(fs, n);
LUAI_ERRORCHECK()
fs->freereg += n;
}
void ICACHE_FLASH_ATTR luaK_reserveregs (FuncState *fs, int n) {
luaK_checkstack(fs, n);
fs->freereg += n;
}
void FuncState::luaK_reserveregs (/*FuncState *fs,*/ int n) {
luaK_checkstack(n);
free_reg += n;
}
