165 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
		
		
			
		
	
	
			165 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| 
								 | 
							
								/*
							 | 
						||
| 
								 | 
							
								** $Id: ltm.c,v 2.38 2016/12/22 13:08:50 roberto Exp $
							 | 
						||
| 
								 | 
							
								** Tag methods
							 | 
						||
| 
								 | 
							
								** See Copyright Notice in lua.h
							 | 
						||
| 
								 | 
							
								*/
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define ltm_c
							 | 
						||
| 
								 | 
							
								#define LUA_CORE
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#include "ltm.h"
							 | 
						||
| 
								 | 
							
								#include "lprefix.h"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#include <string.h>
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#include "lua.h"
							 | 
						||
| 
								 | 
							
								#include "ldebug.h"
							 | 
						||
| 
								 | 
							
								#include "ldo.h"
							 | 
						||
| 
								 | 
							
								#include "lobject.h"
							 | 
						||
| 
								 | 
							
								#include "lstate.h"
							 | 
						||
| 
								 | 
							
								#include "lstring.h"
							 | 
						||
| 
								 | 
							
								#include "ltable.h"
							 | 
						||
| 
								 | 
							
								#include "lvm.h"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								namespace NS_SLUA {
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static const char udatatypename[] = "userdata";
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								LUAI_DDEF const char *const luaT_typenames_[LUA_TOTALTAGS] = {
							 | 
						||
| 
								 | 
							
								  "no value",
							 | 
						||
| 
								 | 
							
								  "nil", "boolean", udatatypename, "number",
							 | 
						||
| 
								 | 
							
								  "string", "table", "function", udatatypename, "thread",
							 | 
						||
| 
								 | 
							
								  "proto" /* this last case is used for tests only */
							 | 
						||
| 
								 | 
							
								};
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								void luaT_init (lua_State *L) {
							 | 
						||
| 
								 | 
							
								  static const char *const luaT_eventname[] = {  /* ORDER TM */
							 | 
						||
| 
								 | 
							
								    "__index", "__newindex",
							 | 
						||
| 
								 | 
							
								    "__gc", "__mode", "__len", "__eq",
							 | 
						||
| 
								 | 
							
								    "__add", "__sub", "__mul", "__mod", "__pow",
							 | 
						||
| 
								 | 
							
								    "__div", "__idiv",
							 | 
						||
| 
								 | 
							
								    "__band", "__bor", "__bxor", "__shl", "__shr",
							 | 
						||
| 
								 | 
							
								    "__unm", "__bnot", "__lt", "__le",
							 | 
						||
| 
								 | 
							
								    "__concat", "__call"
							 | 
						||
| 
								 | 
							
								  };
							 | 
						||
| 
								 | 
							
								  int i;
							 | 
						||
| 
								 | 
							
								  for (i=0; i<TM_N; i++) {
							 | 
						||
| 
								 | 
							
								    G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
							 | 
						||
| 
								 | 
							
								    luaC_fix(L, obj2gco(G(L)->tmname[i]));  /* never collect these names */
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/*
							 | 
						||
| 
								 | 
							
								** function to be used with macro "fasttm": optimized for absence of
							 | 
						||
| 
								 | 
							
								** tag methods
							 | 
						||
| 
								 | 
							
								*/
							 | 
						||
| 
								 | 
							
								const TValue *luaT_gettm (Table *events, TMS event, TString *ename) {
							 | 
						||
| 
								 | 
							
								  const TValue *tm = luaH_getshortstr(events, ename);
							 | 
						||
| 
								 | 
							
								  lua_assert(event <= TM_EQ);
							 | 
						||
| 
								 | 
							
								  if (ttisnil(tm)) {  /* no tag method? */
							 | 
						||
| 
								 | 
							
								    events->flags |= cast_byte(1u<<event);  /* cache this fact */
							 | 
						||
| 
								 | 
							
								    return NULL;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  else return tm;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o, TMS event) {
							 | 
						||
| 
								 | 
							
								  Table *mt;
							 | 
						||
| 
								 | 
							
								  switch (ttnov(o)) {
							 | 
						||
| 
								 | 
							
								    case LUA_TTABLE:
							 | 
						||
| 
								 | 
							
								      mt = hvalue(o)->metatable;
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    case LUA_TUSERDATA:
							 | 
						||
| 
								 | 
							
								      mt = uvalue(o)->metatable;
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    default:
							 | 
						||
| 
								 | 
							
								      mt = G(L)->mt[ttnov(o)];
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  return (mt ? luaH_getshortstr(mt, G(L)->tmname[event]) : luaO_nilobject);
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/*
							 | 
						||
| 
								 | 
							
								** Return the name of the type of an object. For tables and userdata
							 | 
						||
| 
								 | 
							
								** with metatable, use their '__name' metafield, if present.
							 | 
						||
| 
								 | 
							
								*/
							 | 
						||
| 
								 | 
							
								const char *luaT_objtypename (lua_State *L, const TValue *o) {
							 | 
						||
| 
								 | 
							
								  Table *mt;
							 | 
						||
| 
								 | 
							
								  if ((ttistable(o) && (mt = hvalue(o)->metatable) != NULL) ||
							 | 
						||
| 
								 | 
							
								      (ttisfulluserdata(o) && (mt = uvalue(o)->metatable) != NULL)) {
							 | 
						||
| 
								 | 
							
								    const TValue *name = luaH_getshortstr(mt, luaS_new(L, "__name"));
							 | 
						||
| 
								 | 
							
								    if (ttisstring(name))  /* is '__name' a string? */
							 | 
						||
| 
								 | 
							
								      return getstr(tsvalue(name));  /* use it as type name */
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  return ttypename(ttnov(o));  /* else use standard type name */
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								void luaT_callTM (lua_State *L, const TValue *f, const TValue *p1,
							 | 
						||
| 
								 | 
							
								                  const TValue *p2, TValue *p3, int hasres) {
							 | 
						||
| 
								 | 
							
								  ptrdiff_t result = savestack(L, p3);
							 | 
						||
| 
								 | 
							
								  StkId func = L->top;
							 | 
						||
| 
								 | 
							
								  setobj2s(L, func, f);  /* push function (assume EXTRA_STACK) */
							 | 
						||
| 
								 | 
							
								  setobj2s(L, func + 1, p1);  /* 1st argument */
							 | 
						||
| 
								 | 
							
								  setobj2s(L, func + 2, p2);  /* 2nd argument */
							 | 
						||
| 
								 | 
							
								  L->top += 3;
							 | 
						||
| 
								 | 
							
								  if (!hasres)  /* no result? 'p3' is third argument */
							 | 
						||
| 
								 | 
							
								    setobj2s(L, L->top++, p3);  /* 3rd argument */
							 | 
						||
| 
								 | 
							
								  /* metamethod may yield only when called from Lua code */
							 | 
						||
| 
								 | 
							
								  if (isLua(L->ci))
							 | 
						||
| 
								 | 
							
								    luaD_call(L, func, hasres);
							 | 
						||
| 
								 | 
							
								  else
							 | 
						||
| 
								 | 
							
								    luaD_callnoyield(L, func, hasres);
							 | 
						||
| 
								 | 
							
								  if (hasres) {  /* if has result, move it to its place */
							 | 
						||
| 
								 | 
							
								    p3 = restorestack(L, result);
							 | 
						||
| 
								 | 
							
								    setobjs2s(L, p3, --L->top);
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								int luaT_callbinTM (lua_State *L, const TValue *p1, const TValue *p2,
							 | 
						||
| 
								 | 
							
								                    StkId res, TMS event) {
							 | 
						||
| 
								 | 
							
								  const TValue *tm = luaT_gettmbyobj(L, p1, event);  /* try first operand */
							 | 
						||
| 
								 | 
							
								  if (ttisnil(tm))
							 | 
						||
| 
								 | 
							
								    tm = luaT_gettmbyobj(L, p2, event);  /* try second operand */
							 | 
						||
| 
								 | 
							
								  if (ttisnil(tm)) return 0;
							 | 
						||
| 
								 | 
							
								  luaT_callTM(L, tm, p1, p2, res, 1);
							 | 
						||
| 
								 | 
							
								  return 1;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								void luaT_trybinTM (lua_State *L, const TValue *p1, const TValue *p2,
							 | 
						||
| 
								 | 
							
								                    StkId res, TMS event) {
							 | 
						||
| 
								 | 
							
								  if (!luaT_callbinTM(L, p1, p2, res, event)) {
							 | 
						||
| 
								 | 
							
								    switch (event) {
							 | 
						||
| 
								 | 
							
								      case TM_CONCAT:
							 | 
						||
| 
								 | 
							
								        luaG_concaterror(L, p1, p2);
							 | 
						||
| 
								 | 
							
								      /* call never returns, but to avoid warnings: *//* FALLTHROUGH */
							 | 
						||
| 
								 | 
							
								      case TM_BAND: case TM_BOR: case TM_BXOR:
							 | 
						||
| 
								 | 
							
								      case TM_SHL: case TM_SHR: case TM_BNOT: {
							 | 
						||
| 
								 | 
							
								        lua_Number dummy;
							 | 
						||
| 
								 | 
							
								        if (tonumber(p1, &dummy) && tonumber(p2, &dummy))
							 | 
						||
| 
								 | 
							
								          luaG_tointerror(L, p1, p2);
							 | 
						||
| 
								 | 
							
								        else
							 | 
						||
| 
								 | 
							
								          luaG_opinterror(L, p1, p2, "perform bitwise operation on");
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								      /* calls never return, but to avoid warnings: *//* FALLTHROUGH */
							 | 
						||
| 
								 | 
							
								      default:
							 | 
						||
| 
								 | 
							
								        luaG_opinterror(L, p1, p2, "perform arithmetic on");
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								int luaT_callorderTM (lua_State *L, const TValue *p1, const TValue *p2,
							 | 
						||
| 
								 | 
							
								                      TMS event) {
							 | 
						||
| 
								 | 
							
								  if (!luaT_callbinTM(L, p1, p2, L->top, event))
							 | 
						||
| 
								 | 
							
								    return -1;  /* no metamethod */
							 | 
						||
| 
								 | 
							
								  else
							 | 
						||
| 
								 | 
							
								    return !l_isfalse(L->top);
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								} // end NS_SLUA
							 |