From c47f0687696778cb63d4289d0f0e2f5112e3fc22 Mon Sep 17 00:00:00 2001
From: Myk Taylor <myk002@yahoo.com>
Date: Wed, 15 Mar 2023 00:28:18 -0700
Subject: [PATCH] add Lua::Push for sets

---
 library/include/LuaTools.h | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/library/include/LuaTools.h b/library/include/LuaTools.h
index 689d82260..b19fc31de 100644
--- a/library/include/LuaTools.h
+++ b/library/include/LuaTools.h
@@ -31,6 +31,7 @@ distribution.
 #include <map>
 #include <type_traits>
 #include <unordered_map>
+#include <unordered_set>
 
 #include "df/interfacest.h"
 
@@ -364,6 +365,26 @@ namespace DFHack {namespace Lua {
     DFHACK_EXPORT int PushPosXYZ(lua_State *state, const df::coord &pos);
     DFHACK_EXPORT int PushPosXY(lua_State *state, const df::coord2d &pos);
 
+    template<typename T>
+    void Push(lua_State *L, const std::set<T> &pset) {
+        lua_createtable(L, 0, pset.size());
+        for (auto &entry : pset) {
+            Lua::Push(L, entry);
+            Lua::Push(L, true);
+            lua_settable(L, -3);
+        }
+    }
+
+    template<typename T_Key, typename T_Hash>
+    void Push(lua_State *L, const std::unordered_set<T_Key, T_Hash> &pset) {
+        lua_createtable(L, 0, pset.size());
+        for (auto &entry : pset) {
+            Lua::Push(L, entry);
+            Lua::Push(L, true);
+            lua_settable(L, -3);
+        }
+    }
+
     template<typename T_Key, typename T_Value>
     void Push(lua_State *L, const std::map<T_Key, T_Value> &pmap) {
         lua_createtable(L, 0, pmap.size());