lvm.h 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. /*
  2. ** $Id: lvm.h $
  3. ** Lua virtual machine
  4. ** See Copyright Notice in lua.h
  5. */
  6. #ifndef lvm_h
  7. #define lvm_h
  8. #include "ldo.h"
  9. #include "lobject.h"
  10. #include "ltm.h"
  11. #if !defined(LUA_NOCVTN2S)
  12. #define cvt2str(o) ttisnumber(o)
  13. #else
  14. #define cvt2str(o) 0 /* no conversion from numbers to strings */
  15. #endif
  16. #if !defined(LUA_NOCVTS2N)
  17. #define cvt2num(o) ttisstring(o)
  18. #else
  19. #define cvt2num(o) 0 /* no conversion from strings to numbers */
  20. #endif
  21. /*
  22. ** You can define LUA_FLOORN2I if you want to convert floats to integers
  23. ** by flooring them (instead of raising an error if they are not
  24. ** integral values)
  25. */
  26. #if !defined(LUA_FLOORN2I)
  27. #define LUA_FLOORN2I F2Ieq
  28. #endif
  29. /*
  30. ** Rounding modes for float->integer coercion
  31. */
  32. typedef enum {
  33. F2Ieq, /* no rounding; accepts only integral values */
  34. F2Ifloor, /* takes the floor of the number */
  35. F2Iceil /* takes the ceil of the number */
  36. } F2Imod;
  37. /* convert an object to a float (including string coercion) */
  38. #define tonumber(o,n) \
  39. (ttisfloat(o) ? (*(n) = fltvalue(o), 1) : luaV_tonumber_(o,n))
  40. /* convert an object to a float (without string coercion) */
  41. #define tonumberns(o,n) \
  42. (ttisfloat(o) ? ((n) = fltvalue(o), 1) : \
  43. (ttisinteger(o) ? ((n) = cast_num(ivalue(o)), 1) : 0))
  44. /* convert an object to an integer (including string coercion) */
  45. #define tointeger(o,i) \
  46. (l_likely(ttisinteger(o)) ? (*(i) = ivalue(o), 1) \
  47. : luaV_tointeger(o,i,LUA_FLOORN2I))
  48. /* convert an object to an integer (without string coercion) */
  49. #define tointegerns(o,i) \
  50. (l_likely(ttisinteger(o)) ? (*(i) = ivalue(o), 1) \
  51. : luaV_tointegerns(o,i,LUA_FLOORN2I))
  52. #define intop(op,v1,v2) l_castU2S(l_castS2U(v1) op l_castS2U(v2))
  53. #define luaV_rawequalobj(t1,t2) luaV_equalobj(NULL,t1,t2)
  54. /*
  55. ** fast track for 'gettable': if 't' is a table and 't[k]' is present,
  56. ** return 1 with 'slot' pointing to 't[k]' (position of final result).
  57. ** Otherwise, return 0 (meaning it will have to check metamethod)
  58. ** with 'slot' pointing to an empty 't[k]' (if 't' is a table) or NULL
  59. ** (otherwise). 'f' is the raw get function to use.
  60. */
  61. #define luaV_fastget(L,t,k,slot,f) \
  62. (!ttistable(t) \
  63. ? (slot = NULL, 0) /* not a table; 'slot' is NULL and result is 0 */ \
  64. : (slot = f(hvalue(t), k), /* else, do raw access */ \
  65. !isempty(slot))) /* result not empty? */
  66. #define luaV_fastget1(t,k,res,f, aux) \
  67. (aux = (!ttistable(t) ? HNOTATABLE : f(hvalue(t), k, res)))
  68. /*
  69. ** Special case of 'luaV_fastget' for integers, inlining the fast case
  70. ** of 'luaH_getint'.
  71. */
  72. #define luaV_fastgeti(L,t,k,slot) \
  73. (!ttistable(t) \
  74. ? (slot = NULL, 0) /* not a table; 'slot' is NULL and result is 0 */ \
  75. : (slot = (l_castS2U(k) - 1u < hvalue(t)->alimit) \
  76. ? &hvalue(t)->array[k - 1] : luaH_getint(hvalue(t), k), \
  77. !isempty(slot))) /* result not empty? */
  78. #define luaV_fastgeti1(t,k,res,aux) \
  79. if (!ttistable(t)) aux = HNOTATABLE; \
  80. else { Table *h = hvalue(t); lua_Unsigned u = l_castS2U(k); \
  81. if ((u - 1u < h->alimit)) { \
  82. int tag = *getArrTag(h,u); \
  83. if (tagisempty(tag)) aux = HNOTFOUND; \
  84. else { arr2val(h, u, tag, res); aux = HOK; }} \
  85. else { aux = luaH_getint1(h, u, res); }}
  86. #define luaV_fastset1(t,k,val,aux,f) \
  87. (aux = (!ttistable(t) ? HNOTATABLE : f(hvalue(t), k, val)))
  88. #define luaV_fastseti1(t,k,val,aux) \
  89. if (!ttistable(t)) aux = HNOTATABLE; \
  90. else { Table *h = hvalue(t); lua_Unsigned u = l_castS2U(k); \
  91. if ((u - 1u < h->alimit)) { \
  92. lu_byte *tag = getArrTag(h,u); \
  93. if (tagisempty(*tag)) aux = ~cast_int(u); \
  94. else { val2arr(h, u, tag, val); aux = HOK; }} \
  95. else { aux = luaH_setint1(h, u, val); }}
  96. /*
  97. ** Finish a fast set operation (when fast get succeeds). In that case,
  98. ** 'slot' points to the place to put the value.
  99. */
  100. #define luaV_finishfastset(L,t,slot,v) \
  101. { setobj2t(L, cast(TValue *,slot), v); \
  102. luaC_barrierback(L, gcvalue(t), v); }
  103. #define luaV_finishfastset1(L,t,v) luaC_barrierback(L, gcvalue(t), v)
  104. /*
  105. ** Shift right is the same as shift left with a negative 'y'
  106. */
  107. #define luaV_shiftr(x,y) luaV_shiftl(x,intop(-, 0, y))
  108. LUAI_FUNC int luaV_equalobj (lua_State *L, const TValue *t1, const TValue *t2);
  109. LUAI_FUNC int luaV_lessthan (lua_State *L, const TValue *l, const TValue *r);
  110. LUAI_FUNC int luaV_lessequal (lua_State *L, const TValue *l, const TValue *r);
  111. LUAI_FUNC int luaV_tonumber_ (const TValue *obj, lua_Number *n);
  112. LUAI_FUNC int luaV_tointeger (const TValue *obj, lua_Integer *p, F2Imod mode);
  113. LUAI_FUNC int luaV_tointegerns (const TValue *obj, lua_Integer *p,
  114. F2Imod mode);
  115. LUAI_FUNC int luaV_flttointeger (lua_Number n, lua_Integer *p, F2Imod mode);
  116. LUAI_FUNC void luaV_finishget1 (lua_State *L, const TValue *t, TValue *key,
  117. StkId val, int aux);
  118. LUAI_FUNC void luaV_finishset1 (lua_State *L, const TValue *t, TValue *key,
  119. TValue *val, int aux);
  120. LUAI_FUNC void luaV_finishOp (lua_State *L);
  121. LUAI_FUNC void luaV_execute (lua_State *L, CallInfo *ci);
  122. LUAI_FUNC void luaV_concat (lua_State *L, int total);
  123. LUAI_FUNC lua_Integer luaV_idiv (lua_State *L, lua_Integer x, lua_Integer y);
  124. LUAI_FUNC lua_Integer luaV_mod (lua_State *L, lua_Integer x, lua_Integer y);
  125. LUAI_FUNC lua_Number luaV_modf (lua_State *L, lua_Number x, lua_Number y);
  126. LUAI_FUNC lua_Integer luaV_shiftl (lua_Integer x, lua_Integer y);
  127. LUAI_FUNC void luaV_objlen (lua_State *L, StkId ra, const TValue *rb);
  128. #endif