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

模拟服务器

开发平台:

C/C++

  1. /*
  2. ** $Id: lapi.c,v 1.110 2000/10/30 12:50:09 roberto Exp $
  3. ** Lua API
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <string.h>
  7. #include "lua.h"
  8. #include "lapi.h"
  9. #include "ldo.h"
  10. #include "lfunc.h"
  11. #include "lgc.h"
  12. #include "lmem.h"
  13. #include "lobject.h"
  14. #include "lstate.h"
  15. #include "lstring.h"
  16. #include "ltable.h"
  17. #include "ltm.h"
  18. #include "lvm.h"
  19. const char lua_ident[] = "$Lua: " LUA_VERSION " " LUA_COPYRIGHT " $n"
  20.                                "$Authors: " LUA_AUTHORS " $";
  21. #define Index(L,i) ((i) >= 0 ? (L->Cbase+((i)-1)) : (L->top+(i)))
  22. #define api_incr_top(L) incr_top
  23. TObject *luaA_index (lua_State *L, int index) {
  24.   return Index(L, index);
  25. }
  26. static TObject *luaA_indexAcceptable (lua_State *L, int index) {
  27.   if (index >= 0) {
  28.     TObject *o = L->Cbase+(index-1);
  29.     if (o >= L->top) return NULL;
  30.     else return o;
  31.   }
  32.   else return L->top+index;
  33. }
  34. void luaA_pushobject (lua_State *L, const TObject *o) {
  35.   *L->top = *o;
  36.   incr_top;
  37. }
  38. LUA_API int lua_stackspace (lua_State *L) {
  39.   return (L->stack_last - L->top);
  40. }
  41. /*
  42. ** basic stack manipulation
  43. */
  44. LUA_API int lua_gettop (lua_State *L) {
  45.   return (L->top - L->Cbase);
  46. }
  47. /*!*****************************************************************************
  48. // Function : void lua_gettopindex
  49. // Purpose : 
  50. // Return : LUA_API  
  51. // Argumant : lua_State *L
  52. // Argumant : int * pindex
  53. // Comments :
  54. // Author : Romandou
  55. *****************************************************************************/
  56. LUA_API  void lua_gettopindex(lua_State *L , int * pindex)
  57. {
  58. (*pindex) = L->top - L->Cbase;
  59. }
  60. LUA_API void lua_settop (lua_State *L, int index) {
  61.   if (index >= 0)
  62.     luaD_adjusttop(L, L->Cbase, index);
  63.   else
  64.     L->top = L->top+index+1;  /* index is negative */
  65. }
  66. LUA_API void lua_remove (lua_State *L, int index) {
  67.   StkId p = luaA_index(L, index);
  68.   while (++p < L->top) *(p-1) = *p;
  69.   L->top--;
  70. }
  71. LUA_API void lua_insert (lua_State *L, int index) {
  72.   StkId p = luaA_index(L, index);
  73.   StkId q;
  74.   for (q = L->top; q>p; q--)
  75.     *q = *(q-1);
  76.   *p = *L->top;
  77. }
  78. LUA_API void lua_pushvalue (lua_State *L, int index) {
  79.   *L->top = *luaA_index(L, index);
  80.   api_incr_top(L);
  81. }
  82. /*
  83. ** access functions (stack -> C)
  84. */
  85. LUA_API int lua_type (lua_State *L, int index) {
  86.   StkId o = luaA_indexAcceptable(L, index);
  87.   return (o == NULL) ? LUA_TNONE : ttype(o);
  88. }
  89. LUA_API const char *lua_typename (lua_State *L, int t) {
  90.   UNUSED(L);
  91.   return (t == LUA_TNONE) ? "no value" : luaO_typenames[t];
  92. }
  93. LUA_API int lua_iscfunction (lua_State *L, int index) {
  94.   StkId o = luaA_indexAcceptable(L, index);
  95.   return (o == NULL) ? 0 : iscfunction(o);
  96. }
  97. LUA_API int lua_isnumber (lua_State *L, int index) {
  98.   TObject *o = luaA_indexAcceptable(L, index);
  99.   return (o == NULL) ? 0 : (tonumber(o) == 0);
  100. }
  101. LUA_API int lua_isstring (lua_State *L, int index) {
  102.   int t = lua_type(L, index);
  103.   return (t == LUA_TSTRING || t == LUA_TNUMBER);
  104. }
  105. LUA_API int lua_tag (lua_State *L, int index) {
  106.   StkId o = luaA_indexAcceptable(L, index);
  107.   return (o == NULL) ? LUA_NOTAG : luaT_tag(o);
  108. }
  109. LUA_API int lua_equal (lua_State *L, int index1, int index2) {
  110.   StkId o1 = luaA_indexAcceptable(L, index1);
  111.   StkId o2 = luaA_indexAcceptable(L, index2);
  112.   if (o1 == NULL || o2 == NULL) return 0;  /* index out-of-range */
  113.   else return luaO_equalObj(o1, o2);
  114. }
  115. LUA_API int lua_lessthan (lua_State *L, int index1, int index2) {
  116.   StkId o1 = luaA_indexAcceptable(L, index1);
  117.   StkId o2 = luaA_indexAcceptable(L, index2);
  118.   if (o1 == NULL || o2 == NULL) return 0;  /* index out-of-range */
  119.   else return luaV_lessthan(L, o1, o2, L->top);
  120. }
  121. LUA_API double lua_tonumber (lua_State *L, int index) {
  122.   StkId o = luaA_indexAcceptable(L, index);
  123.   return (o == NULL || tonumber(o)) ? 0 : nvalue(o);
  124. }
  125. LUA_API const char *lua_tostring (lua_State *L, int index) {
  126.   StkId o = luaA_indexAcceptable(L, index);
  127.   return (o == NULL || tostring(L, o)) ? NULL : svalue(o);
  128. }
  129. LUA_API size_t lua_strlen (lua_State *L, int index) {
  130.   StkId o = luaA_indexAcceptable(L, index);
  131.   return (o == NULL || tostring(L, o)) ? 0 : tsvalue(o)->len;
  132. }
  133. LUA_API lua_CFunction lua_tocfunction (lua_State *L, int index) {
  134.   StkId o = luaA_indexAcceptable(L, index);
  135.   return (o == NULL || !iscfunction(o)) ? NULL : clvalue(o)->f.c;
  136. }
  137. LUA_API void *lua_touserdata (lua_State *L, int index) {
  138.   StkId o = luaA_indexAcceptable(L, index);
  139.   return (o == NULL || ttype(o) != LUA_TUSERDATA) ? NULL :
  140.                                                     tsvalue(o)->u.d.value;
  141. }
  142. LUA_API const void *lua_topointer (lua_State *L, int index) {
  143.   StkId o = luaA_indexAcceptable(L, index);
  144.   if (o == NULL) return NULL;
  145.   switch (ttype(o)) {
  146.     case LUA_TTABLE: 
  147.       return hvalue(o);
  148.     case LUA_TFUNCTION:
  149.       return clvalue(o);
  150.     default: return NULL;
  151.   }
  152. }
  153. /*
  154. ** push functions (C -> stack)
  155. */
  156. LUA_API void lua_pushnil (lua_State *L) {
  157.   ttype(L->top) = LUA_TNIL;
  158.   api_incr_top(L);
  159. }
  160. LUA_API void lua_pushnumber (lua_State *L, double n) {
  161.   nvalue(L->top) = n;
  162.   ttype(L->top) = LUA_TNUMBER;
  163.   api_incr_top(L);
  164. }
  165. LUA_API void lua_pushlstring (lua_State *L, const char *s, size_t len) {
  166.   tsvalue(L->top) = luaS_newlstr(L, s, len);
  167.   ttype(L->top) = LUA_TSTRING;
  168.   api_incr_top(L);
  169. }
  170. LUA_API void lua_pushstring (lua_State *L, const char *s) {
  171.   if (s == NULL)
  172.     lua_pushnil(L);
  173.   else
  174.     lua_pushlstring(L, s, strlen(s));
  175. }
  176. LUA_API void lua_pushcclosure (lua_State *L, lua_CFunction fn, int n) {
  177.   luaV_Cclosure(L, fn, n);
  178. }
  179. LUA_API void lua_pushusertag (lua_State *L, void *u, int tag) {
  180.   /* ORDER LUA_T */
  181.   if (!(tag == LUA_ANYTAG || tag == LUA_TUSERDATA || validtag(tag)))
  182.     luaO_verror(L, "invalid tag for a userdata (%d)", tag);
  183.   tsvalue(L->top) = luaS_createudata(L, u, tag);
  184.   ttype(L->top) = LUA_TUSERDATA;
  185.   api_incr_top(L);
  186. }
  187. /*
  188. ** get functions (Lua -> stack)
  189. */
  190. LUA_API void lua_getglobal (lua_State *L, const char *name) {
  191.   StkId top = L->top;
  192.   *top = *luaV_getglobal(L, luaS_new(L, name));
  193.   L->top = top;
  194.   api_incr_top(L);
  195. }
  196. LUA_API void lua_gettable (lua_State *L, int index) {
  197.   StkId t = Index(L, index);
  198.   StkId top = L->top;
  199.   *(top-1) = *luaV_gettable(L, t);
  200.   L->top = top;  /* tag method may change top */
  201. }
  202. LUA_API void lua_rawget (lua_State *L, int index) {
  203.   StkId t = Index(L, index);
  204.   LUA_ASSERT(ttype(t) == LUA_TTABLE, "table expected");
  205.   *(L->top - 1) = *luaH_get(L, hvalue(t), L->top - 1);
  206. }
  207. LUA_API void lua_rawgeti (lua_State *L, int index, int n) {
  208.   StkId o = Index(L, index);
  209.   LUA_ASSERT(ttype(o) == LUA_TTABLE, "table expected");
  210.   *L->top = *luaH_getnum(hvalue(o), n);
  211.   api_incr_top(L);
  212. }
  213. LUA_API void lua_getglobals (lua_State *L) {
  214.   hvalue(L->top) = L->gt;
  215.   ttype(L->top) = LUA_TTABLE;
  216.   api_incr_top(L);
  217. }
  218. LUA_API int lua_getref (lua_State *L, int ref) {
  219.   if (ref == LUA_REFNIL)
  220.     ttype(L->top) = LUA_TNIL;
  221.   else if (0 <= ref && ref < L->refSize &&
  222.           (L->refArray[ref].st == LOCK || L->refArray[ref].st == HOLD))
  223.     *L->top = L->refArray[ref].o;
  224.   else
  225.     return 0;
  226.   api_incr_top(L);
  227.   return 1;
  228. }
  229. LUA_API void lua_newtable (lua_State *L) {
  230.   hvalue(L->top) = luaH_new(L, 0);
  231.   ttype(L->top) = LUA_TTABLE;
  232.   api_incr_top(L);
  233. }
  234. /*
  235. ** set functions (stack -> Lua)
  236. */
  237. LUA_API void lua_setglobal (lua_State *L, const char *name) {
  238.   StkId top = L->top;
  239.   luaV_setglobal(L, luaS_new(L, name));
  240.   L->top = top-1;  /* remove element from the top */
  241. }
  242. LUA_API void lua_settable (lua_State *L, int index) {
  243.   StkId t = Index(L, index);
  244.   StkId top = L->top;
  245.   luaV_settable(L, t, top-2);
  246.   L->top = top-2;  /* pop index and value */
  247. }
  248. LUA_API void lua_rawset (lua_State *L, int index) {
  249.   StkId t = Index(L, index);
  250.   LUA_ASSERT(ttype(t) == LUA_TTABLE, "table expected");
  251.   *luaH_set(L, hvalue(t), L->top-2) = *(L->top-1);
  252.   L->top -= 2;
  253. }
  254. LUA_API void lua_rawseti (lua_State *L, int index, int n) {
  255.   StkId o = Index(L, index);
  256.   LUA_ASSERT(ttype(o) == LUA_TTABLE, "table expected");
  257.   *luaH_setint(L, hvalue(o), n) = *(L->top-1);
  258.   L->top--;
  259. }
  260. LUA_API void lua_setglobals (lua_State *L) {
  261.   StkId newtable = --L->top;
  262.   LUA_ASSERT(ttype(newtable) == LUA_TTABLE, "table expected");
  263.   L->gt = hvalue(newtable);
  264. }
  265. LUA_API int lua_ref (lua_State *L,  int lock) {
  266.   int ref;
  267.   if (ttype(L->top-1) == LUA_TNIL)
  268.     ref = LUA_REFNIL;
  269.   else {
  270.     if (L->refFree != NONEXT) {  /* is there a free place? */
  271.       ref = L->refFree;
  272.       L->refFree = L->refArray[ref].st;
  273.     }
  274.     else {  /* no more free places */
  275.       luaM_growvector(L, L->refArray, L->refSize, 1, struct Ref,
  276.                       "reference table overflow", MAX_INT);
  277.       L->nblocks += sizeof(struct Ref);
  278.       ref = L->refSize++;
  279.     }
  280.     L->refArray[ref].o = *(L->top-1);
  281.     L->refArray[ref].st = lock ? LOCK : HOLD;
  282.   }
  283.   L->top--;
  284.   return ref;
  285. }
  286. /*
  287. ** "do" functions (run Lua code)
  288. ** (most of them are in ldo.c)
  289. */
  290. LUA_API void lua_rawcall (lua_State *L, int nargs, int nresults) {
  291.   luaD_call(L, L->top-(nargs+1), nresults);
  292. }
  293. /*
  294. ** Garbage-collection functions
  295. */
  296. /* GC values are expressed in Kbytes: #bytes/2^10 */
  297. #define GCscale(x) ((int)((x)>>10))
  298. #define GCunscale(x) ((unsigned long)(x)<<10)
  299. LUA_API int lua_getgcthreshold (lua_State *L) {
  300.   return GCscale(L->GCthreshold);
  301. }
  302. LUA_API int lua_getgccount (lua_State *L) {
  303.   return GCscale(L->nblocks);
  304. }
  305. LUA_API void lua_setgcthreshold (lua_State *L, int newthreshold) {
  306.   if (newthreshold > GCscale(ULONG_MAX))
  307.     L->GCthreshold = ULONG_MAX;
  308.   else
  309.     L->GCthreshold = GCunscale(newthreshold);
  310.   luaC_checkGC(L);
  311. }
  312. /*
  313. ** miscellaneous functions
  314. */
  315. LUA_API void lua_settag (lua_State *L, int tag) {
  316.   luaT_realtag(L, tag);
  317.   switch (ttype(L->top-1)) {
  318.     case LUA_TTABLE:
  319.       hvalue(L->top-1)->htag = tag;
  320.       break;
  321.     case LUA_TUSERDATA:
  322.       tsvalue(L->top-1)->u.d.tag = tag;
  323.       break;
  324.     default:
  325.       luaO_verror(L, "cannot change the tag of a %.20s",
  326.                   luaO_typename(L->top-1));
  327.   }
  328. }
  329. LUA_API void lua_unref (lua_State *L, int ref) {
  330.   if (ref >= 0) {
  331.     LUA_ASSERT(ref < L->refSize && L->refArray[ref].st < 0, "invalid ref");
  332.     L->refArray[ref].st = L->refFree;
  333.     L->refFree = ref;
  334.   }
  335. }
  336. LUA_API int lua_next (lua_State *L, int index) {
  337.   StkId t = luaA_index(L, index);
  338.   Node *n;
  339.   LUA_ASSERT(ttype(t) == LUA_TTABLE, "table expected");
  340.   n = luaH_next(L, hvalue(t), luaA_index(L, -1));
  341.   if (n) {
  342.     *(L->top-1) = *key(n);
  343.     *L->top = *val(n);
  344.     api_incr_top(L);
  345.     return 1;
  346.   }
  347.   else {  /* no more elements */
  348.     L->top -= 1;  /* remove key */
  349.     return 0;
  350.   }
  351. }
  352. LUA_API int lua_getn (lua_State *L, int index) {
  353.   Hash *h = hvalue(luaA_index(L, index));
  354.   const TObject *value = luaH_getstr(h, luaS_new(L, "n"));  /* value = h.n */
  355.   if (ttype(value) == LUA_TNUMBER)
  356.     return (int)nvalue(value);
  357.   else {
  358.     Number max = 0;
  359.     int i = h->size;
  360.     Node *n = h->node;
  361.     while (i--) {
  362.       if (ttype(key(n)) == LUA_TNUMBER &&
  363.           ttype(val(n)) != LUA_TNIL &&
  364.           nvalue(key(n)) > max)
  365.         max = nvalue(key(n));
  366.       n++;
  367.     }
  368.     return (int)max;
  369.   }
  370. }
  371. LUA_API void lua_concat (lua_State *L, int n) {
  372.   StkId top = L->top;
  373.   luaV_strconc(L, n, top);
  374.   L->top = top-(n-1);
  375.   luaC_checkGC(L);
  376. }
  377. LUA_API void *lua_newuserdata (lua_State *L, size_t size) {
  378.   TString *ts = luaS_newudata(L, size, NULL);
  379.   tsvalue(L->top) = ts;
  380.   ttype(L->top) = LUA_TUSERDATA;
  381.   api_incr_top(L);
  382.   return ts->u.d.value;
  383. }