ltm.c
上传用户:dzyhzl
上传日期:2019-04-29
资源大小:56270k
文件大小:4k
源码类别:

模拟服务器

开发平台:

C/C++

  1. /*
  2. ** $Id: ltm.c,v 1.56 2000/10/31 13:10:24 roberto Exp $
  3. ** Tag methods
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <stdio.h>
  7. #include <string.h>
  8. #include "lua.h"
  9. #include "ldo.h"
  10. #include "lmem.h"
  11. #include "lobject.h"
  12. #include "lstate.h"
  13. #include "ltm.h"
  14. const char *const luaT_eventname[] = {  /* ORDER TM */
  15.   "gettable", "settable", "index", "getglobal", "setglobal", "add", "sub",
  16.   "mul", "div", "pow", "unm", "lt", "concat", "gc", "function",
  17.   "le", "gt", "ge",  /* deprecated options!! */
  18.   NULL
  19. };
  20. static int findevent (const char *name) {
  21.   int i;
  22.   for (i=0; luaT_eventname[i]; i++)
  23.     if (strcmp(luaT_eventname[i], name) == 0)
  24.       return i;
  25.   return -1;  /* name not found */
  26. }
  27. static int luaI_checkevent (lua_State *L, const char *name, int t) {
  28.   int e = findevent(name);
  29.   if (e >= TM_N)
  30.     luaO_verror(L, "event `%.50s' is deprecated", name);
  31.   if (e == TM_GC && t == LUA_TTABLE)
  32.     luaO_verror(L, "event `gc' for tables is deprecated");
  33.   if (e < 0)
  34.     luaO_verror(L, "`%.50s' is not a valid event name", name);
  35.   return e;
  36. }
  37. /* events in LUA_TNIL are all allowed, since this is used as a
  38. *  'placeholder' for "default" fallbacks
  39. */
  40. /* ORDER LUA_T, ORDER TM */
  41. static const char luaT_validevents[NUM_TAGS][TM_N] = {
  42.   {1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1},  /* LUA_TUSERDATA */
  43.   {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1},  /* LUA_TNIL */
  44.   {1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1},  /* LUA_TNUMBER */
  45.   {1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},  /* LUA_TSTRING */
  46.   {0, 0, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1},  /* LUA_TTABLE */
  47.   {1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0}   /* LUA_TFUNCTION */
  48. };
  49. int luaT_validevent (int t, int e) {  /* ORDER LUA_T */
  50.   return (t >= NUM_TAGS) ?  1 : luaT_validevents[t][e];
  51. }
  52. static void init_entry (lua_State *L, int tag) {
  53.   int i;
  54.   for (i=0; i<TM_N; i++)
  55.     luaT_gettm(L, tag, i) = NULL;
  56.   L->TMtable[tag].collected = NULL;
  57. }
  58. void luaT_init (lua_State *L) {
  59.   int t;
  60.   luaM_growvector(L, L->TMtable, 0, NUM_TAGS, struct TM, "", MAX_INT);
  61.   L->nblocks += NUM_TAGS*sizeof(struct TM);
  62.   L->last_tag = NUM_TAGS-1;
  63.   for (t=0; t<=L->last_tag; t++)
  64.     init_entry(L, t);
  65. }
  66. LUA_API int lua_newtag (lua_State *L) {
  67.   luaM_growvector(L, L->TMtable, L->last_tag, 1, struct TM,
  68.                   "tag table overflow", MAX_INT);
  69.   L->nblocks += sizeof(struct TM);
  70.   L->last_tag++;
  71.   init_entry(L, L->last_tag);
  72.   return L->last_tag;
  73. }
  74. static void checktag (lua_State *L, int tag) {
  75.   if (!(0 <= tag && tag <= L->last_tag))
  76.     luaO_verror(L, "%d is not a valid tag", tag);
  77. }
  78. void luaT_realtag (lua_State *L, int tag) {
  79.   if (!validtag(tag))
  80.     luaO_verror(L, "tag %d was not created by `newtag'", tag);
  81. }
  82. LUA_API int lua_copytagmethods (lua_State *L, int tagto, int tagfrom) {
  83.   int e;
  84.   checktag(L, tagto);
  85.   checktag(L, tagfrom);
  86.   for (e=0; e<TM_N; e++) {
  87.     if (luaT_validevent(tagto, e))
  88.       luaT_gettm(L, tagto, e) = luaT_gettm(L, tagfrom, e);
  89.   }
  90.   return tagto;
  91. }
  92. int luaT_tag (const TObject *o) {
  93.   int t = ttype(o);
  94.   switch (t) {
  95.     case LUA_TUSERDATA: return tsvalue(o)->u.d.tag;
  96.     case LUA_TTABLE:    return hvalue(o)->htag;
  97.     default:            return t;
  98.   }
  99. }
  100. LUA_API void lua_gettagmethod (lua_State *L, int t, const char *event) {
  101.   int e;
  102.   e = luaI_checkevent(L, event, t);
  103.   checktag(L, t);
  104.   if (luaT_validevent(t, e) && luaT_gettm(L, t, e)) {
  105.     clvalue(L->top) = luaT_gettm(L, t, e);
  106.     ttype(L->top) = LUA_TFUNCTION;
  107.   }
  108.   else
  109.     ttype(L->top) = LUA_TNIL;
  110.   incr_top;
  111. }
  112. LUA_API void lua_settagmethod (lua_State *L, int t, const char *event) {
  113.   int e = luaI_checkevent(L, event, t);
  114.   checktag(L, t);
  115.   if (!luaT_validevent(t, e))
  116.     luaO_verror(L, "cannot change `%.20s' tag method for type `%.20s'%.20s",
  117.                 luaT_eventname[e], luaO_typenames[t],
  118.                 (t == LUA_TTABLE || t == LUA_TUSERDATA) ?
  119.                    " with default tag" : "");
  120.   switch (ttype(L->top - 1)) {
  121.     case LUA_TNIL:
  122.       luaT_gettm(L, t, e) = NULL;
  123.       break;
  124.     case LUA_TFUNCTION:
  125.       luaT_gettm(L, t, e) = clvalue(L->top - 1);
  126.       break;
  127.     default:
  128.       lua_error(L, "tag method must be a function (or nil)");
  129.   }
  130.   L->top--;
  131. }