mirror of
https://git.do.srb2.org/STJr/SRB2.git
synced 2025-02-16 17:11:33 +00:00
Merge branch 'table-len-metamethod' into 'next'
Fix __len metamethod ignored on tables See merge request STJr/SRB2!1328
This commit is contained in:
commit
bc5c65306b
3 changed files with 31 additions and 20 deletions
|
@ -31,9 +31,9 @@ void luaT_init (lua_State *L) {
|
|||
static const char *const luaT_eventname[] = { /* ORDER TM */
|
||||
"__index", "__newindex",
|
||||
"__usedindex",
|
||||
"__gc", "__mode", "__eq",
|
||||
"__gc", "__mode", "__len", "__eq",
|
||||
"__add", "__sub", "__mul", "__div", "__mod",
|
||||
"__pow", "__unm", "__len", "__lt", "__le",
|
||||
"__pow", "__unm", "__lt", "__le",
|
||||
"__concat", "__call"
|
||||
,"__strhook"
|
||||
,"__and", "__or", "__xor", "__shl", "__shr", "__not"
|
||||
|
|
|
@ -21,6 +21,7 @@ typedef enum {
|
|||
TM_USEDINDEX,
|
||||
TM_GC,
|
||||
TM_MODE,
|
||||
TM_LEN,
|
||||
TM_EQ, /* last tag method with `fast' access */
|
||||
TM_ADD,
|
||||
TM_SUB,
|
||||
|
@ -29,7 +30,6 @@ typedef enum {
|
|||
TM_MOD,
|
||||
TM_POW,
|
||||
TM_UNM,
|
||||
TM_LEN,
|
||||
TM_LT,
|
||||
TM_LE,
|
||||
TM_CONCAT,
|
||||
|
|
|
@ -311,6 +311,33 @@ void luaV_concat (lua_State *L, int total, int last) {
|
|||
}
|
||||
|
||||
|
||||
static void objlen (lua_State *L, StkId ra, TValue *rb) {
|
||||
const TValue *tm;
|
||||
switch (ttype(rb)) {
|
||||
case LUA_TTABLE: {
|
||||
Table *h = hvalue(rb);
|
||||
tm = fasttm(L, h->metatable, TM_LEN);
|
||||
if (tm) break; /* metamethod? break switch to call it */
|
||||
setnvalue(ra, cast_num(luaH_getn(h))); /* else primitive len */
|
||||
return;
|
||||
}
|
||||
case LUA_TSTRING: {
|
||||
tm = fasttm(L, G(L)->mt[LUA_TSTRING], TM_LEN);
|
||||
if (tm) break; /* metamethod? break switch to call it */
|
||||
setnvalue(ra, cast_num(tsvalue(rb)->len));
|
||||
return;
|
||||
}
|
||||
default: { /* try metamethod */
|
||||
tm = luaT_gettmbyobj(L, rb, TM_LEN);
|
||||
if (ttisnil(tm)) /* no metamethod? */
|
||||
luaG_typeerror(L, rb, "get length of");
|
||||
break;
|
||||
}
|
||||
}
|
||||
callTMres(L, ra, tm, rb, luaO_nilobject);
|
||||
}
|
||||
|
||||
|
||||
static void Arith (lua_State *L, StkId ra, TValue *rb,
|
||||
TValue *rc, TMS op) {
|
||||
TValue tempb, tempc;
|
||||
|
@ -568,23 +595,7 @@ void luaV_execute (lua_State *L, int nexeccalls) {
|
|||
continue;
|
||||
}
|
||||
case OP_LEN: {
|
||||
TValue *rb = RB(i);
|
||||
switch (ttype(rb)) {
|
||||
case LUA_TTABLE: {
|
||||
setnvalue(ra, cast_num(luaH_getn(hvalue(rb))));
|
||||
break;
|
||||
}
|
||||
case LUA_TSTRING: {
|
||||
setnvalue(ra, cast_num(tsvalue(rb)->len));
|
||||
break;
|
||||
}
|
||||
default: { /* try metamethod */
|
||||
Protect(
|
||||
if (!call_binTM(L, rb, luaO_nilobject, ra, TM_LEN))
|
||||
luaG_typeerror(L, rb, "get length of");
|
||||
)
|
||||
}
|
||||
}
|
||||
Protect(objlen(L, ra, RB(i)));
|
||||
continue;
|
||||
}
|
||||
case OP_CONCAT: {
|
||||
|
|
Loading…
Reference in a new issue