Bruce/lua-immutable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added __ipairs

Browse Source
This commit is contained in:
Bruce Hill 2018-02-09 17:10:14 -08:00
parent 0f15ee34d7
commit cd479f05fd
1 changed files with 44 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
limmutable.c
View File

@ -252,11 +252,18 @@ static int Ltostring(lua_State *L)
} else {
needs_comma = 1;
}
// Stack: [mt, buckets, bucket, inst_table, fields, i, fieldname]
lua_getglobal(L, "tostring");
lua_insert(L, -2);
lua_call(L, 1, 1);
luaL_addvalue(&b);
// Stack: [mt, buckets, bucket, inst_table, fields, i]
luaL_addstring(&b, "=");
lua_rawgeti(L, -3, lua_tonumber(L, -1));
// Stack: [mt, buckets, bucket, inst_table, fields, i, value]
lua_getglobal(L, "tostring");
lua_insert(L, -2);
lua_call(L, 1, 1);
luaL_addvalue(&b);
// Stack: [mt, buckets, bucket, inst_table, fields, i]
}
@ -265,10 +272,47 @@ static int Ltostring(lua_State *L)
return 1;
}
static int Lnexti(lua_State *L)
{
lua_getmetatable(L, 1);
// Stack: [mt]
lua_getfield(L, -1, "__instances");
// Stack: [mt, buckets]
lua_rawgeti(L, -1, *((lua_Integer*)lua_touserdata(L, 1)));
// Stack: [mt, buckets, bucket]
lua_pushvalue(L, 1);
// Stack: [mt, buckets, bucket, inst_udata]
lua_rawget(L, -2);
// Stack: [mt, buckets, bucket, inst_table]
lua_getfield(L, -4, "__fields");
// Stack: [mt, buckets, bucket, inst_table, fields]
lua_pushvalue(L, 2);
// Stack: [mt, buckets, bucket, inst_table, fields, i]
if (lua_next(L, -2) == 0) {
return 0;
}
// Stack: [mt, buckets, bucket, inst_table, fields, i2, next_fieldname]
lua_rawgeti(L, -4, lua_tonumber(L, -2));
// Stack: [mt, buckets, bucket, inst_table, fields, i2, next_fieldname, value]
return 3;
}
static int Lipairs(lua_State *L)
{
lua_pushcfunction(L, Lnexti);
// Stack: [Lnexti]
lua_pushvalue(L, 1);
// Stack: [Lnexti, inst_udata]
lua_pushnil(L);
// Stack: [Lnexti, inst_udata, nil]
return 3;
}
static const luaL_Reg R[] =
{
{ "__len", Llen},
{ "__index", Lindex},
{ "__ipairs", Lipairs},
{ "new", Lcreate_instance},
{ NULL, NULL}
};