Core/Spells: Load entire proc flags array from db2

1 files changed, 25 insertions, 0 deletions

diff --git a/src/server/game/Spells/SpellDefines.h b/src/server/game/Spells/SpellDefines.h
index e9e683d36bc..f651339e1f8 100644
--- a/src/server/game/Spells/SpellDefines.h
+++ b/src/server/game/Spells/SpellDefines.h
@@ -19,6 +19,7 @@
 #define TRINITY_SPELLDEFINES_H
 
 #include "Define.h"
+#include "FlagsArray.h"
 #include "EnumFlag.h"
 #include "ObjectGuid.h"
 #include "Optional.h"
@@ -34,6 +35,8 @@ class Spell;
 class Unit;
 class WorldObject;
 enum Difficulty : uint8;
+enum ProcFlags : uint32;
+enum ProcFlags2 : int32;
 
 namespace UF
 {
@@ -484,4 +487,26 @@ struct SpellCastVisual
     operator WorldPackets::Spells::SpellCastVisual() const;
 };
 
+class ProcFlagsInit : public FlagsArray<int32, 2>
+{
+public:
+    constexpr ProcFlagsInit(ProcFlags procFlags = {}, ProcFlags2 procFlags2 = {})
+    {
+        _storage[0] = int32(procFlags);
+        _storage[1] = int32(procFlags2);
+    }
+
+    constexpr ProcFlagsInit& operator|=(ProcFlags procFlags)
+    {
+        _storage[0] |= int32(procFlags);
+        return *this;
+    }
+
+    constexpr ProcFlagsInit& operator|=(ProcFlags2 procFlags2)
+    {
+        _storage[1] |= int32(procFlags2);
+        return *this;
+    }
+};
+
 #endif