skin this cat

luakiwi/luakiwi-int.h

@@ -0,0 +1,84 @@
+#ifndef LUAKIWI_INT_H_
+#define LUAKIWI_INT_H_
+
+#include "luacompat.h"
+
+#define ARR_COUNT(x) ((int)(sizeof(x) / sizeof((x)[0])))
+
+#if defined(__GNUC__) && !defined(LKIWI_NO_BUILTIN)
+   #define l_likely(x) (__builtin_expect(((x) != 0), 1))
+   #define l_unlikely(x) (__builtin_expect(((x) != 0), 0))
+#else
+   #define l_likely(x) (x)
+   #define l_unlikely(x) (x)
+#endif
+
+#ifndef LKIWI_NO_STALLOC
+   #if defined(_MSC_VER)
+      #include <malloc.h>
+
+static inline void* stalloc(size_t sz) {
+   void* p = _malloca(sz);
+   if (!p) {
+      luaL_error(L, "out of memory");
+   }
+   return p;
+}
+
+      #define stfree(ptr) _freea(ptr)
+   #else
+      #include <alloca.h>
+      #define stalloc alloca
+      #define stfree(ptr)
+   #endif
+#else
+   #include <stdlib.h>
+
+static inline void* stalloc(size_t sz) {
+   void* p = malloc(sz);
+   if (l_unlikely(!p)) {
+      luaL_error(L, "out of memory");
+   }
+   return p;
+}
+
+   #define stfree(ptr) free(ptr)
+#endif
+
+// Lua 5.1 compatibility for missing lua_arith.
+static void compat_arith_unm(lua_State* L) {
+#if defined(LUA_VERSION_NUM) && LUA_VERSION_NUM == 501
+   lua_Number n = lua_tonumber(L, -1);
+   if (n != 0 || lua_isnumber(L, -1)) {
+      lua_pop(L, 1);
+      lua_pushnumber(L, -n);
+   } else {
+      if (!luaL_callmeta(L, -1, "__unm"))
+         luaL_error(L, "attempt to perform arithmetic on a %s value", luaL_typename(L, -1));
+      lua_replace(L, -2);
+   }
+#else
+   lua_arith(L, LUA_OPUNM);
+#endif
+}
+
+// This version supports placeholders.
+static void setfuncs(lua_State* L, const luaL_Reg* l, int nup) {
+   luaL_checkstack(L, nup, "too many upvalues");
+   for (; l->name != NULL; l++) { /* fill the table with given functions */
+      if (l->func == NULL) /* place holder? */
+         lua_pushboolean(L, 0);
+      else {
+         int i;
+         for (i = 0; i < nup; i++) /* copy upvalues to the top */
+            lua_pushvalue(L, -nup);
+         lua_pushcclosure(L, l->func, nup); /* closure with those upvalues */
+      }
+      lua_setfield(L, -(nup + 2), l->name);
+   }
+   lua_pop(L, nup); /* remove upvalues */
+}
+
+#define newlib(L, l) (lua_newtable((L)), setfuncs((L), (l), 0))
+
+#endif  // LUAKIWI_INT_H_