Spell: Implement GetDamageForEffect for scripting purposes

cmangos · Jan 12, 2024 · a8fc3ac · a8fc3ac
1 parent a98131b
commit a8fc3ac
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/game/Spells/Spell.h b/src/game/Spells/Spell.h
@@ -861,9 +861,10 @@ class Spell
         // OnInit use only
         void SetEffectSkipMask(uint32 mask) { m_effectSkipMask = mask; }
         // OnHit use only
-        uint32 GetTotalTargetDamage() { return m_damage; }
-        uint32 GetTotalTargetAbsorb() { return m_absorb; }
+        uint32 GetTotalTargetDamage() const { return m_damage; }
+        uint32 GetTotalTargetAbsorb() const { return m_absorb; }
         void SetTotalTargetValueModifier(float modifier);
+        int32 GetDamageForEffect(SpellEffectIndex effIdx) const { return m_damagePerEffect[effIdx]; }
         // script initialization hook only setters - use only if dynamic - else use appropriate helper
         void SetMaxAffectedTargets(uint32 newValue) { m_affectedTargetCount = newValue; }
         void SetChainTargetsCount(SpellEffectIndex effIdx, uint32 newValue) { m_chainTargetCount[effIdx] = newValue; }