From 5eab34a1dd978eb4fd67e6e5fd22f567de84ceb4 Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Thu, 19 May 2022 21:33:16 +0300
Subject: [PATCH] Add util::flag_set for enum-based bitsets

---
 libs/util/include/psemek/util/flag_set.hpp | 53 ++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 libs/util/include/psemek/util/flag_set.hpp

diff --git a/libs/util/include/psemek/util/flag_set.hpp b/libs/util/include/psemek/util/flag_set.hpp
new file mode 100644
index 00000000..51a99259
--- /dev/null
+++ b/libs/util/include/psemek/util/flag_set.hpp
@@ -0,0 +1,53 @@
+#pragma once
+
+#include <type_traits>
+#include <cstdint>
+
+namespace psemek::util
+{
+
+	template <typename Flag, typename Repr = std::uint32_t>
+	struct flag_set
+	{
+		using flag_type = Flag;
+		using underlying_type = std::conditional_t<std::is_enum_v<Flag>, std::underlying_type_t<Flag>, Flag>;
+		using repr_type = Repr;
+
+		repr_type value = repr_type{};
+
+		flag_set() = default;
+
+		flag_set(flag_set const &) = default;
+
+		template <typename ... Flags>
+		flag_set(Flags const & ... flags)
+			: value{((1 << static_cast<underlying_type>(flags)) | ...)}
+		{}
+
+		bool is_set(Flag const & flag) const
+		{
+			return value & (1 << static_cast<underlying_type>(flag));
+		}
+
+		void set(Flag const & flag)
+		{
+			value |= (1 << static_cast<underlying_type>(flag));
+		}
+
+		void flip(Flag const & flag)
+		{
+			value ^= (1 << static_cast<underlying_type>(flag));
+		}
+
+		void unset(Flag const & flag)
+		{
+			value &= ~(1 << static_cast<underlying_type>(flag));
+		}
+
+		void clear()
+		{
+			value = {};
+		}
+	};
+
+}