初始化内容

third/LuaBridge3/Source/LuaBridge/FlatMap.h

@@ -0,0 +1,89 @@
+// https://github.com/kunitoki/LuaBridge3
+// Copyright 2020, kunitoki
+// SPDX-License-Identifier: MIT
+
+#pragma once
+
+#include "detail/Stack.h"
+
+#if LUABRIDGE_HAS_CXX23_FLAT_CONTAINERS
+
+#include <flat_map>
+
+namespace luabridge {
+
+//=================================================================================================
+/**
+ * @brief Stack specialization for `std::flat_map`.
+ */
+template <class K, class V, class Compare, class KeyContainer, class MappedContainer>
+struct Stack<std::flat_map<K, V, Compare, KeyContainer, MappedContainer>>
+{
+    using Type = std::flat_map<K, V, Compare, KeyContainer, MappedContainer>;
+
+    [[nodiscard]] static Result push(lua_State* L, const Type& map)
+    {
+#if LUABRIDGE_SAFE_STACK_CHECKS
+        if (! lua_checkstack(L, 3))
+            return makeErrorCode(ErrorCode::LuaStackOverflow);
+#endif
+
+        StackRestore stackRestore(L);
+
+        lua_createtable(L, 0, static_cast<int>(map.size()));
+
+        for (auto it = map.begin(); it != map.end(); ++it)
+        {
+            auto result = Stack<K>::push(L, it->first);
+            if (! result)
+                return result;
+
+            result = Stack<V>::push(L, it->second);
+            if (! result)
+                return result;
+
+            lua_settable(L, -3);
+        }
+
+        stackRestore.reset();
+        return {};
+    }
+
+    [[nodiscard]] static TypeResult<Type> get(lua_State* L, int index)
+    {
+        if (!lua_istable(L, index))
+            return makeErrorCode(ErrorCode::InvalidTypeCast);
+
+        const StackRestore stackRestore(L);
+
+        Type map;
+
+        int absIndex = lua_absindex(L, index);
+        lua_pushnil(L);
+
+        while (lua_next(L, absIndex) != 0)
+        {
+            auto value = Stack<V>::get(L, -1);
+            if (! value)
+                return makeErrorCode(ErrorCode::InvalidTypeCast);
+
+            auto key = Stack<K>::get(L, -2);
+            if (! key)
+                return makeErrorCode(ErrorCode::InvalidTypeCast);
+
+            map.emplace(*key, *value);
+            lua_pop(L, 1);
+        }
+
+        return map;
+    }
+
+    [[nodiscard]] static bool isInstance(lua_State* L, int index)
+    {
+        return lua_istable(L, index);
+    }
+};
+
+} // namespace luabridge
+
+#endif // LUABRIDGE_HAS_CXX23_FLAT_CONTAINERS