ltable.h 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. /*
  2. ** $Id: ltable.h $
  3. ** Lua tables (hash)
  4. ** See Copyright Notice in lua.h
  5. */
  6. #ifndef ltable_h
  7. #define ltable_h
  8. #include "lobject.h"
  9. #define gnode(t,i) (&(t)->node[i])
  10. #define gval(n) (&(n)->i_val)
  11. #define gnext(n) ((n)->u.next)
  12. /*
  13. ** Clear all bits of fast-access metamethods, which means that the table
  14. ** may have any of these metamethods. (First access that fails after the
  15. ** clearing will set the bit again.)
  16. */
  17. #define invalidateTMcache(t) ((t)->flags &= cast_byte(~maskflags))
  18. /*
  19. ** Bit BITDUMMY set in 'flags' means the table is using the dummy node
  20. ** for its hash part.
  21. */
  22. #define BITDUMMY (1 << 6)
  23. #define NOTBITDUMMY cast_byte(~BITDUMMY)
  24. #define isdummy(t) ((t)->flags & BITDUMMY)
  25. #define setnodummy(t) ((t)->flags &= NOTBITDUMMY)
  26. #define setdummy(t) ((t)->flags |= BITDUMMY)
  27. /* allocated size for hash nodes */
  28. #define allocsizenode(t) (isdummy(t) ? 0 : sizenode(t))
  29. /* returns the Node, given the value of a table entry */
  30. #define nodefromval(v) cast(Node *, (v))
  31. #define luaH_fastgeti(t,k,res,tag) \
  32. { Table *h = t; lua_Unsigned u = l_castS2U(k) - 1u; \
  33. if ((u < h->alimit)) { \
  34. tag = *getArrTag(h, u); \
  35. if (!tagisempty(tag)) { farr2val(h, u, tag, res); }} \
  36. else { tag = luaH_getint(h, (k), res); }}
  37. #define luaH_fastseti(t,k,val,hres) \
  38. { Table *h = t; lua_Unsigned u = l_castS2U(k) - 1u; \
  39. if ((u < h->alimit)) { \
  40. lu_byte *tag = getArrTag(h, u); \
  41. if (tagisempty(*tag)) hres = ~cast_int(u); \
  42. else { fval2arr(h, u, tag, val); hres = HOK; }} \
  43. else { hres = luaH_psetint(h, k, val); }}
  44. /* results from pset */
  45. #define HOK 0
  46. #define HNOTFOUND 1
  47. #define HNOTATABLE 2
  48. #define HFIRSTNODE 3
  49. /*
  50. ** 'luaH_get*' operations set 'res', unless the value is absent, and
  51. ** return the tag of the result.
  52. ** The 'luaH_pset*' (pre-set) operations set the given value and return
  53. ** HOK, unless the original value is absent. In that case, if the key
  54. ** is really absent, they return HNOTFOUND. Otherwise, if there is a
  55. ** slot with that key but with no value, 'luaH_pset*' return an encoding
  56. ** of where the key is (usually called 'hres'). (pset cannot set that
  57. ** value because there might be a metamethod.) If the slot is in the
  58. ** hash part, the encoding is (HFIRSTNODE + hash index); if the slot is
  59. ** in the array part, the encoding is (~array index), a negative value.
  60. ** The value HNOTATABLE is used by the fast macros to signal that the
  61. ** value being indexed is not a table.
  62. ** (The size for the array part is limited by the maximum power of two
  63. ** that fits in an unsigned integer; that is INT_MAX+1. So, the C-index
  64. ** ranges from 0, which encodes to -1, to INT_MAX, which encodes to
  65. ** INT_MIN. The size of the hash part is limited by the maximum power of
  66. ** two that fits in a signed integer; that is (INT_MAX+1)/2. So, it is
  67. ** safe to add HFIRSTNODE to any index there.)
  68. */
  69. /*
  70. ** The array part of a table is represented by an inverted array of
  71. ** values followed by an array of tags, to avoid wasting space with
  72. ** padding. The 'array' pointer points to the junction of the two
  73. ** arrays, so that values are indexed with negative indices and tags
  74. ** with non-negative indices.
  75. Values Tags
  76. --------------------------------------------------------
  77. ... | Value 1 | Value 0 |0|1|...
  78. --------------------------------------------------------
  79. ^ t->array
  80. ** All accesses to 't->array' should be through the macros 'getArrTag'
  81. ** and 'getArrVal'.
  82. */
  83. /* Computes the address of the tag for the abstract C-index 'k' */
  84. #define getArrTag(t,k) (cast(lu_byte*, (t)->array) + (k))
  85. /* Computes the address of the value for the abstract C-index 'k' */
  86. #define getArrVal(t,k) ((t)->array - 1 - (k))
  87. /*
  88. ** Move TValues to/from arrays, using C indices
  89. */
  90. #define arr2obj(h,k,val) \
  91. ((val)->tt_ = *getArrTag(h,(k)), (val)->value_ = *getArrVal(h,(k)))
  92. #define obj2arr(h,k,val) \
  93. (*getArrTag(h,(k)) = (val)->tt_, *getArrVal(h,(k)) = (val)->value_)
  94. /*
  95. ** Often, we need to check the tag of a value before moving it. The
  96. ** following macros also move TValues to/from arrays, but receive the
  97. ** precomputed tag value or address as an extra argument.
  98. */
  99. #define farr2val(h,k,tag,res) \
  100. ((res)->tt_ = tag, (res)->value_ = *getArrVal(h,(k)))
  101. #define fval2arr(h,k,tag,val) \
  102. (*tag = (val)->tt_, *getArrVal(h,(k)) = (val)->value_)
  103. LUAI_FUNC lu_byte luaH_get (Table *t, const TValue *key, TValue *res);
  104. LUAI_FUNC lu_byte luaH_getshortstr (Table *t, TString *key, TValue *res);
  105. LUAI_FUNC lu_byte luaH_getstr (Table *t, TString *key, TValue *res);
  106. LUAI_FUNC lu_byte luaH_getint (Table *t, lua_Integer key, TValue *res);
  107. /* Special get for metamethods */
  108. LUAI_FUNC const TValue *luaH_Hgetshortstr (Table *t, TString *key);
  109. LUAI_FUNC TString *luaH_getstrkey (Table *t, TString *key);
  110. LUAI_FUNC int luaH_psetint (Table *t, lua_Integer key, TValue *val);
  111. LUAI_FUNC int luaH_psetshortstr (Table *t, TString *key, TValue *val);
  112. LUAI_FUNC int luaH_psetstr (Table *t, TString *key, TValue *val);
  113. LUAI_FUNC int luaH_pset (Table *t, const TValue *key, TValue *val);
  114. LUAI_FUNC void luaH_setint (lua_State *L, Table *t, lua_Integer key,
  115. TValue *value);
  116. LUAI_FUNC void luaH_set (lua_State *L, Table *t, const TValue *key,
  117. TValue *value);
  118. LUAI_FUNC void luaH_finishset (lua_State *L, Table *t, const TValue *key,
  119. TValue *value, int hres);
  120. LUAI_FUNC Table *luaH_new (lua_State *L);
  121. LUAI_FUNC void luaH_resize (lua_State *L, Table *t, unsigned nasize,
  122. unsigned nhsize);
  123. LUAI_FUNC void luaH_resizearray (lua_State *L, Table *t, unsigned nasize);
  124. LUAI_FUNC lu_mem luaH_size (Table *t);
  125. LUAI_FUNC void luaH_free (lua_State *L, Table *t);
  126. LUAI_FUNC int luaH_next (lua_State *L, Table *t, StkId key);
  127. LUAI_FUNC lua_Unsigned luaH_getn (Table *t);
  128. LUAI_FUNC unsigned luaH_realasize (const Table *t);
  129. #if defined(LUA_DEBUG)
  130. LUAI_FUNC Node *luaH_mainposition (const Table *t, const TValue *key);
  131. #endif
  132. #endif