liba 0.1.15

#include "a.h"

void *lua_alloc(lua_State *L, void const *ptr_, size_t siz)
{
    void *ud = (void *)(a_uptr)ptr_; // NOLINT
    void *ptr = (void *)(a_uptr)ptr_; // NOLINT
    return lua_getallocf(L, &ud)(ud, ptr, 0, siz);
}

void lua_registry_get(lua_State *L, int (*fn)(lua_State *))
{
    lua_rawgetp(L, LUA_REGISTRYINDEX, (void *)(a_uptr)fn); // NOLINT
}

void lua_registry_set(lua_State *L, int (*fn)(lua_State *))
{
    lua_rawsetp(L, LUA_REGISTRYINDEX, (void *)(a_uptr)fn); // NOLINT
}

void lua_fun_set(lua_State *L, int idx, char const *name, lua_CFunction func)
{
    /* table[name]=func */
    lua_pushstring(L, name);
    lua_pushcclosure(L, func, 0);
    lua_rawset(L, idx < 0 ? idx - 2 : idx);
}

void lua_fun_reg(lua_State *L, int idx, lua_fun const *tab, size_t len)
{
    for (idx = idx < 0 ? idx - 2 : idx; len--; ++tab)
    {
        /* table[name]=func */
        lua_pushstring(L, tab->name);
        lua_pushcclosure(L, tab->func, 0);
        lua_rawset(L, idx);
    }
}

void lua_str_set(lua_State *L, int idx, char const *name, char const *data)
{
    /* table[name]=data */
    lua_pushstring(L, name);
    lua_pushstring(L, data);
    lua_rawset(L, idx < 0 ? idx - 2 : idx);
}

char const *lua_str_get(lua_State *L, int idx, char const *name)
{
    /* data=table[name] */
    lua_pushstring(L, name);
    lua_rawget(L, idx < 0 ? idx - 1 : idx);
    char const *s = lua_tostring(L, -1);
    lua_pop(L, 1);
    return s;
}

void lua_str_reg(lua_State *L, int idx, lua_str const *tab, size_t len)
{
    for (idx = idx < 0 ? idx - 2 : idx; len--; ++tab)
    {
        /* table[name]=data */
        lua_pushstring(L, tab->name);
        lua_pushstring(L, tab->data);
        lua_rawset(L, idx);
    }
}

void lua_int_set(lua_State *L, int idx, char const *name, LUA_INT data)
{
    /* table[name]=data */
    lua_pushstring(L, name);
    lua_pushinteger(L, (lua_Integer)data);
    lua_rawset(L, idx < 0 ? idx - 2 : idx);
}

LUA_INT lua_int_get(lua_State *L, int idx, char const *name)
{
    /* data=table[name] */
    lua_pushstring(L, name);
    lua_rawget(L, idx < 0 ? idx - 1 : idx);
    LUA_INT x = (LUA_INT)lua_tointeger(L, -1);
    lua_pop(L, 1);
    return x;
}

void lua_int_reg(lua_State *L, int idx, lua_int const *tab, size_t len)
{
    for (idx = idx < 0 ? idx - 2 : idx; len--; ++tab)
    {
        /* table[name]=data */
        lua_pushstring(L, tab->name);
        lua_pushinteger(L, (lua_Integer)tab->data);
        lua_rawset(L, idx);
    }
}

void lua_num_set(lua_State *L, int idx, char const *name, LUA_NUM data)
{
    /* table[name]=data */
    lua_pushstring(L, name);
    lua_pushnumber(L, (lua_Number)data);
    lua_rawset(L, idx < 0 ? idx - 2 : idx);
}

LUA_NUM lua_num_get(lua_State *L, int idx, char const *name)
{
    /* data=table[name] */
    lua_pushstring(L, name);
    lua_rawget(L, idx < 0 ? idx - 1 : idx);
    LUA_NUM x = (LUA_NUM)lua_tonumber(L, -1);
    lua_pop(L, 1);
    return x;
}

void lua_num_reg(lua_State *L, int idx, lua_num const *tab, size_t len)
{
    for (idx = idx < 0 ? idx - 2 : idx; len--; ++tab)
    {
        /* table[name]=data */
        lua_pushstring(L, tab->name);
        lua_pushnumber(L, (lua_Number)tab->data);
        lua_rawset(L, idx);
    }
}

void lua_array_str_get(lua_State *L, int idx, char const **ptr, unsigned int num)
{
    size_t len = (size_t)lua_rawlen(L, idx);
    if (num > len) { num = (unsigned int)len; }
    for (unsigned int i = num; i--; num = i)
    {
        lua_rawgeti(L, idx, (int)num);
        ptr[i] = lua_tostring(L, -1);
        lua_pop(L, 1);
    }
}

void lua_array_str_set(lua_State *L, int idx, char const *const *ptr, unsigned int num)
{
    idx = idx < 0 ? idx - 1 : idx;
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushstring(L, ptr[i]);
        lua_rawseti(L, idx, (int)num);
    }
}

void lua_array_str_new(lua_State *L, char const *const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushstring(L, ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

void lua_array_int_get(lua_State *L, int idx, LUA_INT *ptr, unsigned int num)
{
    size_t len = (size_t)lua_rawlen(L, idx);
    if (num > len) { num = (unsigned int)len; }
    for (unsigned int i = num; i--; num = i)
    {
        lua_rawgeti(L, idx, (int)num);
        ptr[i] = (LUA_INT)lua_tointeger(L, -1);
        lua_pop(L, 1);
    }
}

void lua_array_int_set(lua_State *L, int idx, LUA_INT const *ptr, unsigned int num)
{
    idx = idx < 0 ? idx - 1 : idx;
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushinteger(L, (lua_Integer)ptr[i]);
        lua_rawseti(L, idx, (int)num);
    }
}

void lua_array_int_new(lua_State *L, LUA_INT const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushinteger(L, (lua_Integer)ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

void lua_array_num_get(lua_State *L, int idx, LUA_NUM *ptr, unsigned int num)
{
    size_t len = (size_t)lua_rawlen(L, idx);
    if (num > len) { num = (unsigned int)len; }
    for (unsigned int i = num; i--; num = i)
    {
        lua_rawgeti(L, idx, (int)num);
        ptr[i] = (LUA_NUM)lua_tonumber(L, -1);
        lua_pop(L, 1);
    }
}

void lua_array_num_set(lua_State *L, int idx, LUA_NUM const *ptr, unsigned int num)
{
    idx = idx < 0 ? idx - 1 : idx;
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushnumber(L, (lua_Number)ptr[i]);
        lua_rawseti(L, idx, (int)num);
    }
}

void lua_array_num_new(lua_State *L, LUA_NUM const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_pushnumber(L, (lua_Number)ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

size_t lua_table_num_len(lua_State *L, int idx);
size_t lua_table_num_len(lua_State *L, int idx) // NOLINT(misc-no-recursion)
{
    size_t num = 0;
    size_t n = (size_t)lua_rawlen(L, idx);
    for (unsigned int i = 0; i++ != n;)
    {
        lua_rawgeti(L, idx, (int)i);
        int e = lua_type(L, -1);
        if (e == LUA_TNUMBER) { ++num; }
        else if (e == LUA_TTABLE)
        {
            num += lua_table_num_len(L, -1);
        }
        lua_pop(L, 1);
    }
    return num;
}

LUA_NUM *lua_table_num_ptr(lua_State *L, int idx, LUA_NUM *ptr);
LUA_NUM *lua_table_num_ptr(lua_State *L, int idx, LUA_NUM *ptr) // NOLINT(misc-no-recursion)
{
    size_t n = (size_t)lua_rawlen(L, idx);
    for (unsigned int i = 0; i++ != n;)
    {
        lua_rawgeti(L, idx, (int)i);
        int e = lua_type(L, -1);
        if (e == LUA_TNUMBER)
        {
            *ptr++ = (LUA_NUM)lua_tonumber(L, -1);
        }
        else if (e == LUA_TTABLE)
        {
            ptr = lua_table_num_ptr(L, -1, ptr);
        }
        lua_pop(L, 1);
    }
    return ptr;
}

LUA_NUM *lua_table_num_get(lua_State *L, int idx, LUA_NUM const *ptr, size_t *num)
{
    LUA_NUM *ret = (LUA_NUM *)(intptr_t)ptr; // NOLINT(performance-no-int-to-ptr)
    if (lua_type(L, idx) == LUA_TTABLE)
    {
        size_t num_ = 0;
        num = num ? num : &num_;
        *num = lua_table_num_len(L, idx);
        if (*num)
        {
            ret = (LUA_NUM *)lua_alloc(L, ret, sizeof(LUA_NUM) * *num);
            lua_table_num_ptr(L, idx, ret);
        }
    }
    return ret;
}

#include <stdio.h>
#include <stdlib.h>
#include <inttypes.h>
#if defined(_MSC_VER) && (_MSC_VER < 1900)
#define snprintf sprintf_s
#endif /* _MSC_VER */

void lua_u8_new(lua_State *L, a_u8 value)
{
    lua_pushinteger(L, (lua_Integer)value);
}

a_u8 lua_u8_get(lua_State *L, int idx)
{
    a_u8 value = 0;
    switch (lua_type(L, idx))
    {
    default: A_FALLTHROUGH;
    case LUA_TNUMBER: value = (a_u8)luaL_checkinteger(L, idx); break;
    case LUA_TSTRING: value = (a_u8)strtoul(lua_tostring(L, idx), NULL, 0);
    }
    return value;
}

void lua_u16_new(lua_State *L, a_u16 value)
{
    lua_pushinteger(L, (lua_Integer)value);
}

a_u16 lua_u16_get(lua_State *L, int idx)
{
    a_u16 value = 0;
    switch (lua_type(L, idx))
    {
    default: A_FALLTHROUGH;
    case LUA_TNUMBER: value = (a_u16)luaL_checkinteger(L, idx); break;
    case LUA_TSTRING: value = (a_u16)strtoul(lua_tostring(L, idx), NULL, 0);
    }
    return value;
}

void lua_u32_new(lua_State *L, a_u32 value)
{
    char buf[8 + 3];
    (void)snprintf(buf, sizeof(buf), "0x%08" PRIX32, value);
    lua_pushstring(L, buf);
}

a_u32 lua_u32_get(lua_State *L, int idx)
{
    a_u32 value = 0;
    switch (lua_type(L, idx))
    {
    default: A_FALLTHROUGH;
    case LUA_TNUMBER: value = (a_u32)luaL_checkinteger(L, idx); break;
    case LUA_TSTRING: value = (a_u32)strtoul(lua_tostring(L, idx), NULL, 0);
    }
    return value;
}

void lua_u64_new(lua_State *L, a_u64 value)
{
    char buf[16 + 3];
    (void)snprintf(buf, sizeof(buf), "0x%016" PRIX64, value);
    lua_pushstring(L, buf);
}

a_u64 lua_u64_get(lua_State *L, int idx)
{
    a_u64 value = 0;
    switch (lua_type(L, idx))
    {
    default: A_FALLTHROUGH;
    case LUA_TNUMBER: value = (a_u64)luaL_checkinteger(L, idx); break;
#if ULONG_MAX > A_U32_MAX
    case LUA_TSTRING: value = (a_u64)strtoul(lua_tostring(L, idx), NULL, 0);
#else /* >long */
    case LUA_TSTRING: value = (a_u64)strtoull(lua_tostring(L, idx), NULL, 0);
#endif /* long */
    }
    return value;
}

void lua_array_u8_new(lua_State *L, a_u8 const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_u8_new(L, ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

void lua_array_u16_new(lua_State *L, a_u16 const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_u16_new(L, ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

void lua_array_u32_new(lua_State *L, a_u32 const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_u32_new(L, ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

void lua_array_u64_new(lua_State *L, a_u64 const *ptr, unsigned int num)
{
    lua_createtable(L, (int)num, 0);
    for (unsigned int i = num; i--; num = i)
    {
        lua_u64_new(L, ptr[i]);
        lua_rawseti(L, -2, (int)num);
    }
}

#if defined(LUA_STACK)
void lua_stack_view(lua_State *L, unsigned int line)
{
    int const n = lua_gettop(L);
    printf("%u:", line);
    for (int i = 0; i++ != n;)
    {
        putchar(' ');
        switch (lua_type(L, i))
        {
        case LUA_TBOOLEAN:
            printf("%s", lua_toboolean(L, i) ? "true" : "false");
            break;
        case LUA_TLIGHTUSERDATA:
            printf("ptr:%p", lua_topointer(L, i));
            break;
        case LUA_TNUMBER:
            printf("%g", lua_tonumber(L, i));
            break;
        case LUA_TSTRING:
            printf("\"%s\"", lua_tostring(L, i));
            break;
        case LUA_TTABLE:
            printf("tab:%p", lua_topointer(L, i));
            break;
        case LUA_TFUNCTION:
        {
            union
            {
                int (*fn)(lua_State *);
                void *p;
            } func = {lua_tocfunction(L, i)};
            printf("func:%p", func.p);
            break;
        }
        case LUA_TUSERDATA:
            printf("data:%p", lua_touserdata(L, i));
            break;
        case LUA_TTHREAD:
            printf("thrd:%p", (void *)lua_tothread(L, i));
            break;
        default: printf("%s", "nil");
        }
    }
    putchar('\n');
}
#endif /* LUA_STACK */