1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#pragma once
#include "compat.h"
#include "macros_and_utils.h"
#include <bit>
class UniqueKey
{
private:
uintptr_t m_storage;
public:
char const* m_debugName{ nullptr };
constexpr explicit UniqueKey(uint64_t val_, char const* debugName_ = nullptr)
#if LUAJIT_FLAVOR() == 64 // building against LuaJIT headers for 64 bits, light userdata is restricted to 47 significant bits, because LuaJIT uses the other bits for internal optimizations
: m_storage{ static_cast<uintptr_t>(val_ & 0x7fffffffffffull) }
#else // LUAJIT_FLAVOR()
: m_storage{ static_cast<uintptr_t>(val_) }
#endif // LUAJIT_FLAVOR()
, m_debugName{ debugName_ }
{
}
constexpr UniqueKey(UniqueKey const& rhs_) = default;
constexpr bool operator!=(UniqueKey const& rhs_) const
{
return m_storage != rhs_.m_storage;
}
constexpr bool operator==(UniqueKey const& rhs_) const
{
return m_storage == rhs_.m_storage;
}
void pushKey(lua_State* const L) const
{
lua_pushlightuserdata(L, std::bit_cast<void*>(m_storage));
}
bool equals(lua_State* const L, int i) const
{
return lua_touserdata(L, i) == std::bit_cast<void*>(m_storage);
}
void pushValue(lua_State* const L) const
{
pushKey(L);
lua_rawget(L, LUA_REGISTRYINDEX);
}
template <typename OP>
void setValue(lua_State* L, OP operation_) const
{
// Note we can't check stack consistency because operation is not always a push (could be insert, replace, whatever)
pushKey(L); // ... key
operation_(L); // ... key value
lua_rawset(L, LUA_REGISTRYINDEX); // ...
}
template <typename T>
T* readLightUserDataValue(lua_State* const L) const
{
STACK_GROW(L, 1);
STACK_CHECK_START_REL(L, 0);
pushValue(L);
T* const value{ lua_tolightuserdata<T>(L, -1) }; // lightuserdata/nil
lua_pop(L, 1);
STACK_CHECK(L, 0);
return value;
}
bool readBoolValue(lua_State* const L) const
{
STACK_GROW(L, 1);
STACK_CHECK_START_REL(L, 0);
pushValue(L);
bool const value{ lua_toboolean(L, -1) ? true : false}; // bool/nil
lua_pop(L, 1);
STACK_CHECK(L, 0);
return value;
}
};
|
