xref: /freebsd/contrib/lua/src/lvm.h (revision a9490b81)
18e3e3a7aSWarner Losh /*
20495ed39SKyle Evans ** $Id: lvm.h $
38e3e3a7aSWarner Losh ** Lua virtual machine
48e3e3a7aSWarner Losh ** See Copyright Notice in lua.h
58e3e3a7aSWarner Losh */
68e3e3a7aSWarner Losh 
78e3e3a7aSWarner Losh #ifndef lvm_h
88e3e3a7aSWarner Losh #define lvm_h
98e3e3a7aSWarner Losh 
108e3e3a7aSWarner Losh 
118e3e3a7aSWarner Losh #include "ldo.h"
128e3e3a7aSWarner Losh #include "lobject.h"
138e3e3a7aSWarner Losh #include "ltm.h"
148e3e3a7aSWarner Losh 
158e3e3a7aSWarner Losh 
168e3e3a7aSWarner Losh #if !defined(LUA_NOCVTN2S)
178e3e3a7aSWarner Losh #define cvt2str(o)	ttisnumber(o)
188e3e3a7aSWarner Losh #else
198e3e3a7aSWarner Losh #define cvt2str(o)	0	/* no conversion from numbers to strings */
208e3e3a7aSWarner Losh #endif
218e3e3a7aSWarner Losh 
228e3e3a7aSWarner Losh 
238e3e3a7aSWarner Losh #if !defined(LUA_NOCVTS2N)
248e3e3a7aSWarner Losh #define cvt2num(o)	ttisstring(o)
258e3e3a7aSWarner Losh #else
268e3e3a7aSWarner Losh #define cvt2num(o)	0	/* no conversion from strings to numbers */
278e3e3a7aSWarner Losh #endif
288e3e3a7aSWarner Losh 
298e3e3a7aSWarner Losh 
308e3e3a7aSWarner Losh /*
318e3e3a7aSWarner Losh ** You can define LUA_FLOORN2I if you want to convert floats to integers
328e3e3a7aSWarner Losh ** by flooring them (instead of raising an error if they are not
338e3e3a7aSWarner Losh ** integral values)
348e3e3a7aSWarner Losh */
358e3e3a7aSWarner Losh #if !defined(LUA_FLOORN2I)
360495ed39SKyle Evans #define LUA_FLOORN2I		F2Ieq
378e3e3a7aSWarner Losh #endif
388e3e3a7aSWarner Losh 
398e3e3a7aSWarner Losh 
400495ed39SKyle Evans /*
410495ed39SKyle Evans ** Rounding modes for float->integer coercion
420495ed39SKyle Evans  */
430495ed39SKyle Evans typedef enum {
440495ed39SKyle Evans   F2Ieq,     /* no rounding; accepts only integral values */
450495ed39SKyle Evans   F2Ifloor,  /* takes the floor of the number */
460495ed39SKyle Evans   F2Iceil    /* takes the ceil of the number */
470495ed39SKyle Evans } F2Imod;
480495ed39SKyle Evans 
490495ed39SKyle Evans 
500495ed39SKyle Evans /* convert an object to a float (including string coercion) */
518e3e3a7aSWarner Losh #define tonumber(o,n) \
528e3e3a7aSWarner Losh 	(ttisfloat(o) ? (*(n) = fltvalue(o), 1) : luaV_tonumber_(o,n))
538e3e3a7aSWarner Losh 
540495ed39SKyle Evans 
550495ed39SKyle Evans /* convert an object to a float (without string coercion) */
560495ed39SKyle Evans #define tonumberns(o,n) \
570495ed39SKyle Evans 	(ttisfloat(o) ? ((n) = fltvalue(o), 1) : \
580495ed39SKyle Evans 	(ttisinteger(o) ? ((n) = cast_num(ivalue(o)), 1) : 0))
590495ed39SKyle Evans 
600495ed39SKyle Evans 
610495ed39SKyle Evans /* convert an object to an integer (including string coercion) */
628e3e3a7aSWarner Losh #define tointeger(o,i) \
638c784bb8SWarner Losh   (l_likely(ttisinteger(o)) ? (*(i) = ivalue(o), 1) \
648c784bb8SWarner Losh                           : luaV_tointeger(o,i,LUA_FLOORN2I))
658e3e3a7aSWarner Losh 
660495ed39SKyle Evans 
670495ed39SKyle Evans /* convert an object to an integer (without string coercion) */
680495ed39SKyle Evans #define tointegerns(o,i) \
698c784bb8SWarner Losh   (l_likely(ttisinteger(o)) ? (*(i) = ivalue(o), 1) \
708c784bb8SWarner Losh                           : luaV_tointegerns(o,i,LUA_FLOORN2I))
710495ed39SKyle Evans 
720495ed39SKyle Evans 
738e3e3a7aSWarner Losh #define intop(op,v1,v2) l_castU2S(l_castS2U(v1) op l_castS2U(v2))
748e3e3a7aSWarner Losh 
758e3e3a7aSWarner Losh #define luaV_rawequalobj(t1,t2)		luaV_equalobj(NULL,t1,t2)
768e3e3a7aSWarner Losh 
778e3e3a7aSWarner Losh 
788e3e3a7aSWarner Losh /*
790495ed39SKyle Evans ** fast track for 'gettable': if 't' is a table and 't[k]' is present,
800495ed39SKyle Evans ** return 1 with 'slot' pointing to 't[k]' (position of final result).
810495ed39SKyle Evans ** Otherwise, return 0 (meaning it will have to check metamethod)
820495ed39SKyle Evans ** with 'slot' pointing to an empty 't[k]' (if 't' is a table) or NULL
830495ed39SKyle Evans ** (otherwise). 'f' is the raw get function to use.
848e3e3a7aSWarner Losh */
858e3e3a7aSWarner Losh #define luaV_fastget(L,t,k,slot,f) \
868e3e3a7aSWarner Losh   (!ttistable(t)  \
878e3e3a7aSWarner Losh    ? (slot = NULL, 0)  /* not a table; 'slot' is NULL and result is 0 */  \
888e3e3a7aSWarner Losh    : (slot = f(hvalue(t), k),  /* else, do raw access */  \
890495ed39SKyle Evans       !isempty(slot)))  /* result not empty? */
908e3e3a7aSWarner Losh 
918e3e3a7aSWarner Losh 
928e3e3a7aSWarner Losh /*
930495ed39SKyle Evans ** Special case of 'luaV_fastget' for integers, inlining the fast case
940495ed39SKyle Evans ** of 'luaH_getint'.
958e3e3a7aSWarner Losh */
960495ed39SKyle Evans #define luaV_fastgeti(L,t,k,slot) \
978e3e3a7aSWarner Losh   (!ttistable(t)  \
980495ed39SKyle Evans    ? (slot = NULL, 0)  /* not a table; 'slot' is NULL and result is 0 */  \
990495ed39SKyle Evans    : (slot = (l_castS2U(k) - 1u < hvalue(t)->alimit) \
1000495ed39SKyle Evans               ? &hvalue(t)->array[k - 1] : luaH_getint(hvalue(t), k), \
1010495ed39SKyle Evans       !isempty(slot)))  /* result not empty? */
1028e3e3a7aSWarner Losh 
1038e3e3a7aSWarner Losh 
1040495ed39SKyle Evans /*
1050495ed39SKyle Evans ** Finish a fast set operation (when fast get succeeds). In that case,
1060495ed39SKyle Evans ** 'slot' points to the place to put the value.
1070495ed39SKyle Evans */
1080495ed39SKyle Evans #define luaV_finishfastset(L,t,slot,v) \
1090495ed39SKyle Evans     { setobj2t(L, cast(TValue *,slot), v); \
1100495ed39SKyle Evans       luaC_barrierback(L, gcvalue(t), v); }
1110495ed39SKyle Evans 
1128e3e3a7aSWarner Losh 
113*a9490b81SWarner Losh /*
114*a9490b81SWarner Losh ** Shift right is the same as shift left with a negative 'y'
115*a9490b81SWarner Losh */
116*a9490b81SWarner Losh #define luaV_shiftr(x,y)	luaV_shiftl(x,intop(-, 0, y))
117*a9490b81SWarner Losh 
1188e3e3a7aSWarner Losh 
1198e3e3a7aSWarner Losh 
1208e3e3a7aSWarner Losh LUAI_FUNC int luaV_equalobj (lua_State *L, const TValue *t1, const TValue *t2);
1218e3e3a7aSWarner Losh LUAI_FUNC int luaV_lessthan (lua_State *L, const TValue *l, const TValue *r);
1228e3e3a7aSWarner Losh LUAI_FUNC int luaV_lessequal (lua_State *L, const TValue *l, const TValue *r);
1238e3e3a7aSWarner Losh LUAI_FUNC int luaV_tonumber_ (const TValue *obj, lua_Number *n);
1240495ed39SKyle Evans LUAI_FUNC int luaV_tointeger (const TValue *obj, lua_Integer *p, F2Imod mode);
1250495ed39SKyle Evans LUAI_FUNC int luaV_tointegerns (const TValue *obj, lua_Integer *p,
1260495ed39SKyle Evans                                 F2Imod mode);
1270495ed39SKyle Evans LUAI_FUNC int luaV_flttointeger (lua_Number n, lua_Integer *p, F2Imod mode);
1288e3e3a7aSWarner Losh LUAI_FUNC void luaV_finishget (lua_State *L, const TValue *t, TValue *key,
1298e3e3a7aSWarner Losh                                StkId val, const TValue *slot);
1308e3e3a7aSWarner Losh LUAI_FUNC void luaV_finishset (lua_State *L, const TValue *t, TValue *key,
1310495ed39SKyle Evans                                TValue *val, const TValue *slot);
1328e3e3a7aSWarner Losh LUAI_FUNC void luaV_finishOp (lua_State *L);
1330495ed39SKyle Evans LUAI_FUNC void luaV_execute (lua_State *L, CallInfo *ci);
1348e3e3a7aSWarner Losh LUAI_FUNC void luaV_concat (lua_State *L, int total);
1350495ed39SKyle Evans LUAI_FUNC lua_Integer luaV_idiv (lua_State *L, lua_Integer x, lua_Integer y);
1368e3e3a7aSWarner Losh LUAI_FUNC lua_Integer luaV_mod (lua_State *L, lua_Integer x, lua_Integer y);
1370495ed39SKyle Evans LUAI_FUNC lua_Number luaV_modf (lua_State *L, lua_Number x, lua_Number y);
1388e3e3a7aSWarner Losh LUAI_FUNC lua_Integer luaV_shiftl (lua_Integer x, lua_Integer y);
1398e3e3a7aSWarner Losh LUAI_FUNC void luaV_objlen (lua_State *L, StkId ra, const TValue *rb);
1408e3e3a7aSWarner Losh 
1418e3e3a7aSWarner Losh #endif
142