Core/Utils: Extract template base class out of flag128

1 files changed, 146 insertions, 0 deletions

diff --git a/src/common/Utilities/FlagsArray.h b/src/common/Utilities/FlagsArray.h
new file mode 100644
index 00000000000..cfdb066c321
--- /dev/null
+++ b/src/common/Utilities/FlagsArray.h
@@ -0,0 +1,146 @@
+/*
+ * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef FlagsArray_h__
+#define FlagsArray_h__
+
+#include "Define.h"
+#include <array>
+
+template<typename T, size_t N>
+class FlagsArray
+{
+public:
+    constexpr FlagsArray operator&(FlagsArray const& right) const
+    {
+        FlagsArray temp = *this;
+        temp &= right;
+        return temp;
+    }
+
+    constexpr FlagsArray& operator&=(FlagsArray const& right)
+    {
+        for (size_t i = 0; i < N; ++i)
+            _storage[i] &= right[i];
+
+        return *this;
+    }
+
+    constexpr FlagsArray operator|(FlagsArray const& right) const
+    {
+        FlagsArray temp = *this;
+        temp |= right;
+        return temp;
+    }
+
+    constexpr FlagsArray& operator|=(FlagsArray const& right)
+    {
+        for (size_t i = 0; i < N; ++i)
+            _storage[i] |= right[i];
+
+        return *this;
+    }
+
+    constexpr FlagsArray operator~() const
+    {
+        FlagsArray temp = *this;
+        for (size_t i = 0; i < N; ++i)
+            temp._storage[i] = ~_storage[i];
+
+        return temp;
+    }
+
+    constexpr FlagsArray operator^(FlagsArray const& right) const
+    {
+        FlagsArray temp = *this;
+        temp ^= right;
+        return temp;
+    }
+
+    constexpr FlagsArray& operator^=(FlagsArray const& right)
+    {
+        for (size_t i = 0; i < N; ++i)
+            _storage[i] ^= right[i];
+
+        return *this;
+    }
+
+    constexpr operator bool() const
+    {
+        for (size_t i = 0; i < N; ++i)
+            if (_storage[i])
+                return true;
+
+        return false;
+    }
+
+    constexpr bool operator !() const
+    {
+        return !(bool(*this));
+    }
+
+    constexpr T& operator[](size_t i)
+    {
+        return _storage[i];
+    }
+
+    constexpr T const& operator [](size_t i) const
+    {
+        return _storage[i];
+    }
+
+    constexpr FlagsArray& operator=(std::array<T, N> const& right)
+    {
+        _storage = right;
+        return *this;
+    }
+
+protected:
+    std::array<T, N> _storage = { };
+};
+
+class flag128 : public FlagsArray<uint32, 4>
+{
+public:
+    constexpr flag128(uint32 p1 = 0, uint32 p2 = 0, uint32 p3 = 0, uint32 p4 = 0)
+    {
+        _storage[0] = p1;
+        _storage[1] = p2;
+        _storage[2] = p3;
+        _storage[3] = p4;
+    }
+
+    constexpr bool IsEqual(uint32 p1 = 0, uint32 p2 = 0, uint32 p3 = 0, uint32 p4 = 0) const
+    {
+        return (_storage[0] == p1 && _storage[1] == p2 && _storage[2] == p3 && _storage[3] == p4);
+    }
+
+    constexpr bool HasFlag(uint32 p1 = 0, uint32 p2 = 0, uint32 p3 = 0, uint32 p4 = 0) const
+    {
+        return (_storage[0] & p1 || _storage[1] & p2 || _storage[2] & p3 || _storage[3] & p4);
+    }
+
+    constexpr void Set(uint32 p1 = 0, uint32 p2 = 0, uint32 p3 = 0, uint32 p4 = 0)
+    {
+        _storage[0] = p1;
+        _storage[1] = p2;
+        _storage[2] = p3;
+        _storage[3] = p4;
+    }
+};
+
+#endif // FlagsArray_h__