ltable.h 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  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 &= ~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. /* results from get/pset */
  32. #define HOK 0
  33. #define HNOTFOUND 1
  34. #define HNOTATABLE 2
  35. #define HFIRSTNODE 3
  36. /*
  37. ** 'luaH_get*' operations set 'res' and return HOK, unless the value is
  38. ** absent. In that case, they set nothing and return HNOTFOUND.
  39. ** The 'luaH_pset*' (pre-set) operations set the given value and return
  40. ** HOK, unless the original value is absent. In that case, if the key
  41. ** is really absent, they return HNOTFOUND. Otherwise, if there is a
  42. ** slot with that key but with no value, 'luaH_pset*' return an encoding
  43. ** of where the key is (usually called 'hres'). (pset cannot set that
  44. ** value because there might be a metamethod.) If the slot is in the
  45. ** hash part, the encoding is (HFIRSTNODE + hash index); if the slot is
  46. ** in the array part, the encoding is (~array index), a negative value.
  47. ** The value HNOTATABLE is used by the fast macros to signal that the
  48. ** value being indexed is not a table.
  49. */
  50. /*
  51. ** The array part of a table is represented by an array of cells.
  52. ** Each cell is composed of (NM + 1) elements, and each element has the
  53. ** type 'ArrayCell'. In each cell, only one element has the variant
  54. ** 'tag', while the other NM elements have the variant 'value'. The
  55. ** array in the 'tag' element holds the tags of the other elements in
  56. ** that cell.
  57. */
  58. #define NM ((unsigned int)sizeof(Value))
  59. union ArrayCell {
  60. unsigned char tag[NM];
  61. Value value;
  62. };
  63. /*
  64. ** 'NMTag' defines which cell element has the tags; that could be any
  65. ** value between 0 (tags come before all values) and NM (tags come after
  66. ** all values).
  67. */
  68. #define NMTag 0
  69. /*
  70. ** Computes the concrete index that holds the tag of abstract index 'i'
  71. */
  72. #define TagIndex(i) (((i)/NM * (NM + 1u)) + NMTag)
  73. /*
  74. ** Computes the concrete index that holds the value of abstract index 'i'
  75. */
  76. #define ValueIndex(i) ((i) + (((i) + (NM - NMTag))/NM))
  77. /* Computes the address of the tag for the abstract index 'k' */
  78. #define getArrTag(t,k) (&(t)->array[TagIndex(k)].tag[(k)%NM])
  79. /* Computes the address of the value for the abstract index 'k' */
  80. #define getArrVal(t,k) (&(t)->array[ValueIndex(k)].value)
  81. /*
  82. ** Move TValues to/from arrays, using Lua indices
  83. */
  84. #define arr2obj(h,k,val) \
  85. ((val)->tt_ = *getArrTag(h,(k)-1u), (val)->value_ = *getArrVal(h,(k)-1u))
  86. #define obj2arr(h,k,val) \
  87. (*getArrTag(h,(k)-1u) = (val)->tt_, *getArrVal(h,(k)-1u) = (val)->value_)
  88. /*
  89. ** Often, we need to check the tag of a value before moving it. These
  90. ** macros also move TValues to/from arrays, but receive the precomputed
  91. ** tag value or address as an extra argument.
  92. */
  93. #define farr2val(h,k,tag,res) \
  94. ((res)->tt_ = tag, (res)->value_ = *getArrVal(h,(k)-1u))
  95. #define fval2arr(h,k,tag,val) \
  96. (*tag = (val)->tt_, *getArrVal(h,(k)-1u) = (val)->value_)
  97. LUAI_FUNC int luaH_get (Table *t, const TValue *key, TValue *res);
  98. LUAI_FUNC int luaH_getshortstr (Table *t, TString *key, TValue *res);
  99. LUAI_FUNC int luaH_getstr (Table *t, TString *key, TValue *res);
  100. LUAI_FUNC int luaH_getint (Table *t, lua_Integer key, TValue *res);
  101. /* Special get for metamethods */
  102. LUAI_FUNC const TValue *luaH_Hgetshortstr (Table *t, TString *key);
  103. LUAI_FUNC TString *luaH_getstrkey (Table *t, TString *key);
  104. LUAI_FUNC int luaH_psetint (Table *t, lua_Integer key, TValue *val);
  105. LUAI_FUNC int luaH_psetshortstr (Table *t, TString *key, TValue *val);
  106. LUAI_FUNC int luaH_psetstr (Table *t, TString *key, TValue *val);
  107. LUAI_FUNC int luaH_pset (Table *t, const TValue *key, TValue *val);
  108. LUAI_FUNC void luaH_setint (lua_State *L, Table *t, lua_Integer key,
  109. TValue *value);
  110. LUAI_FUNC void luaH_set (lua_State *L, Table *t, const TValue *key,
  111. TValue *value);
  112. LUAI_FUNC void luaH_finishset (lua_State *L, Table *t, const TValue *key,
  113. TValue *value, int hres);
  114. LUAI_FUNC Table *luaH_new (lua_State *L);
  115. LUAI_FUNC void luaH_resize (lua_State *L, Table *t, unsigned int nasize,
  116. unsigned int nhsize);
  117. LUAI_FUNC void luaH_resizearray (lua_State *L, Table *t, unsigned int nasize);
  118. LUAI_FUNC void luaH_free (lua_State *L, Table *t);
  119. LUAI_FUNC int luaH_next (lua_State *L, Table *t, StkId key);
  120. LUAI_FUNC lua_Unsigned luaH_getn (Table *t);
  121. LUAI_FUNC unsigned int luaH_realasize (const Table *t);
  122. #if defined(LUA_DEBUG)
  123. LUAI_FUNC Node *luaH_mainposition (const Table *t, const TValue *key);
  124. #endif
  125. #endif