a small optimization

2 files changed, 23 insertions, 6 deletions

diff --git a/ltm.h b/ltm.h
index cbd6feee..9c0f99d3 100644
--- a/ltm.h
+++ b/ltm.h
@@ -1,5 +1,5 @@
 /*
-** $Id: ltm.h,v 1.19 2000/12/26 18:46:09 roberto Exp roberto $
+** $Id: ltm.h,v 1.20 2001/01/19 13:20:30 roberto Exp roberto $
 ** Tag methods
 ** See Copyright Notice in lua.h
 */
@@ -35,6 +35,21 @@ typedef enum {
 } TMS;
 
 
+
+/*
+** masks for allowable tag methods
+*/
+#define HAS_TM_GETGLOBAL(L,t)   (1<<(t) & ((1<<LUA_TUSERDATA) | \
+                                           (1<<LUA_TTABLE) | \
+                                           (1<<LUA_TNIL)))
+
+#define HAS_TM_SETGLOBAL(L,t)   (1<<(t) & ((1<<LUA_TUSERDATA) | \
+                                           (1<<LUA_TTABLE) | \
+                                           (1<<LUA_TNIL) | \
+                                           (1<<LUA_TFUNCTION)))
+
+
+
 struct TM {
   Closure *method[TM_N];
   TString *collected;  /* list of garbage-collected udata with this tag */
diff --git a/lvm.c b/lvm.c
index db4ba548..9146a08c 100644
--- a/lvm.c
+++ b/lvm.c
@@ -1,5 +1,5 @@
 /*
-** $Id: lvm.c,v 1.155 2001/01/19 13:20:30 roberto Exp roberto $
+** $Id: lvm.c,v 1.156 2001/01/24 15:45:33 roberto Exp roberto $
 ** Lua virtual machine
 ** See Copyright Notice in lua.h
 */
@@ -174,8 +174,9 @@ void luaV_settable (lua_State *L, StkId t, StkId key) {
 
 const TObject *luaV_getglobal (lua_State *L, TString *s) {
   const TObject *value = luaH_getstr(L->gt, s);
-  Closure *tm = luaT_gettmbyObj(G(L), value, TM_GETGLOBAL);
-  if (tm == NULL)  /* is there a tag method? */
+  Closure *tm;
+  if (!HAS_TM_GETGLOBAL(L, ttype(value)) ||  /* is there a tag method? */
+      (tm = luaT_gettmbyObj(G(L), value, TM_GETGLOBAL)) == NULL)
     return value;  /* default behavior */
   else {  /* tag method */
     luaD_checkstack(L, 3);
@@ -191,8 +192,9 @@ const TObject *luaV_getglobal (lua_State *L, TString *s) {
 
 void luaV_setglobal (lua_State *L, TString *s) {
   TObject *oldvalue = luaH_setstr(L, L->gt, s);
-  Closure *tm = luaT_gettmbyObj(G(L), oldvalue, TM_SETGLOBAL);
-  if (tm == NULL) {  /* no tag methods? */
+  Closure *tm;
+  if (!HAS_TM_SETGLOBAL(L, ttype(oldvalue)) ||  /* no tag methods? */
+     (tm = luaT_gettmbyObj(G(L), oldvalue, TM_SETGLOBAL)) == NULL) {
     setobj(oldvalue, L->top - 1);  /* raw set */
   }
   else {  /* call tag method */