libs/lua/lua-5.3.2/ltm.c

   1 /*
   2 ** $Id: ltm.c,v 2.36 2015/11/03 15:47:30 roberto Exp $
   3 ** Tag methods
   4 ** See Copyright Notice in lua.h
   5 */
   6
   7 #define ltm_c
   8 #define LUA_CORE
   9
  10 #include "lprefix.h"
  11
  12
  13 #include <string.h>
  14
  15 #include "lua.h"
  16
  17 #include "ldebug.h"
  18 #include "ldo.h"
  19 #include "lobject.h"
  20 #include "lstate.h"
  21 #include "lstring.h"
  22 #include "ltable.h"
  23 #include "ltm.h"
  24 #include "lvm.h"
  25
  26
  27 static const char udatatypename[] = "userdata";
  28
  29 LUAI_DDEF const char *const luaT_typenames_[LUA_TOTALTAGS] = {
  30   "no value",
  31   "nil", "boolean", udatatypename, "number",
  32   "string", "table", "function", udatatypename, "thread",
  33   "proto" /* this last case is used for tests only */
  34 };
  35
  36
  37 void luaT_init (lua_State *L) {
  38   static const char *const luaT_eventname[] = {  /* ORDER TM */
  39     "__index", "__newindex",
  40     "__gc", "__mode", "__len", "__eq",
  41     "__add", "__sub", "__mul", "__mod", "__pow",
  42     "__div", "__idiv",
  43     "__band", "__bor", "__bxor", "__shl", "__shr",
  44     "__unm", "__bnot", "__lt", "__le",
  45     "__concat", "__call"
  46   };
  47   int i;
  48   for (i=0; i<TM_N; i++) {
  49     G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
  50     luaC_fix(L, obj2gco(G(L)->tmname[i]));  /* never collect these names */
  51   }
  52 }
  53
  54
  55 /*
  56 ** function to be used with macro "fasttm": optimized for absence of
  57 ** tag methods
  58 */
  59 const TValue *luaT_gettm (Table *events, TMS event, TString *ename) {
  60   const TValue *tm = luaH_getshortstr(events, ename);
  61   lua_assert(event <= TM_EQ);
  62   if (ttisnil(tm)) {  /* no tag method? */
  63     events->flags |= cast_byte(1u<<event);  /* cache this fact */
  64     return NULL;
  65   }
  66   else return tm;
  67 }
  68
  69
  70 const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o, TMS event) {
  71   Table *mt;
  72   switch (ttnov(o)) {
  73     case LUA_TTABLE:
  74       mt = hvalue(o)->metatable;
  75       break;
  76     case LUA_TUSERDATA:
  77       mt = uvalue(o)->metatable;
  78       break;
  79     default:
  80       mt = G(L)->mt[ttnov(o)];
  81   }
  82   return (mt ? luaH_getshortstr(mt, G(L)->tmname[event]) : luaO_nilobject);
  83 }
  84
  85
  86 void luaT_callTM (lua_State *L, const TValue *f, const TValue *p1,
  87                   const TValue *p2, TValue *p3, int hasres) {
  88   ptrdiff_t result = savestack(L, p3);
  89   StkId func = L->top;
  90   setobj2s(L, func, f);  /* push function (assume EXTRA_STACK) */
  91   setobj2s(L, func + 1, p1);  /* 1st argument */
  92   setobj2s(L, func + 2, p2);  /* 2nd argument */
  93   L->top += 3;
  94   if (!hasres)  /* no result? 'p3' is third argument */
  95     setobj2s(L, L->top++, p3);  /* 3rd argument */
  96   /* metamethod may yield only when called from Lua code */
  97   if (isLua(L->ci))
  98     luaD_call(L, func, hasres);
  99   else
 100     luaD_callnoyield(L, func, hasres);
 101   if (hasres) {  /* if has result, move it to its place */
 102     p3 = restorestack(L, result);
 103     setobjs2s(L, p3, --L->top);
 104   }
 105 }
 106
 107
 108 int luaT_callbinTM (lua_State *L, const TValue *p1, const TValue *p2,
 109                     StkId res, TMS event) {
 110   const TValue *tm = luaT_gettmbyobj(L, p1, event);  /* try first operand */
 111   if (ttisnil(tm))
 112     tm = luaT_gettmbyobj(L, p2, event);  /* try second operand */
 113   if (ttisnil(tm)) return 0;
 114   luaT_callTM(L, tm, p1, p2, res, 1);
 115   return 1;
 116 }
 117
 118
 119 void luaT_trybinTM (lua_State *L, const TValue *p1, const TValue *p2,
 120                     StkId res, TMS event) {
 121   if (!luaT_callbinTM(L, p1, p2, res, event)) {
 122     switch (event) {
 123       case TM_CONCAT:
 124         luaG_concaterror(L, p1, p2);
 125       /* call never returns, but to avoid warnings: *//* FALLTHROUGH */
 126       case TM_BAND: case TM_BOR: case TM_BXOR:
 127       case TM_SHL: case TM_SHR: case TM_BNOT: {
 128         lua_Number dummy;
 129         if (tonumber(p1, &dummy) && tonumber(p2, &dummy))
 130           luaG_tointerror(L, p1, p2);
 131         else
 132           luaG_opinterror(L, p1, p2, "perform bitwise operation on");
 133       }
 134       /* calls never return, but to avoid warnings: *//* FALLTHROUGH */
 135       default:
 136         luaG_opinterror(L, p1, p2, "perform arithmetic on");
 137     }
 138   }
 139 }
 140
 141
 142 int luaT_callorderTM (lua_State *L, const TValue *p1, const TValue *p2,
 143                       TMS event) {
 144   if (!luaT_callbinTM(L, p1, p2, L->top, event))
 145     return -1;  /* no metamethod */
 146   else
 147     return !l_isfalse(L->top);
 148 }
 149