aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/server/game/Entities/Creature/Creature.h1
-rw-r--r--src/server/game/Entities/Unit/Unit.cpp66
-rw-r--r--src/server/game/Entities/Unit/Unit.h5
-rw-r--r--src/server/game/Spells/Spell.cpp49
4 files changed, 75 insertions, 46 deletions
diff --git a/src/server/game/Entities/Creature/Creature.h b/src/server/game/Entities/Creature/Creature.h
index 618175be89c..3fb716d85d8 100644
--- a/src/server/game/Entities/Creature/Creature.h
+++ b/src/server/game/Entities/Creature/Creature.h
@@ -101,6 +101,7 @@ class TC_GAME_API Creature : public Unit, public GridObject<Creature>, public Ma
bool CanSwim() const override { return (GetCreatureTemplate()->InhabitType & INHABIT_WATER) != 0 || IsPet(); }
bool CanFly() const override { return (GetCreatureTemplate()->InhabitType & INHABIT_AIR) != 0; }
bool IsDungeonBoss() const { return (GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_DUNGEON_BOSS) != 0; }
+ bool IsAffectedByDiminishingReturns() const override { return Unit::IsAffectedByDiminishingReturns() || (GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_ALL_DIMINISH) != 0; }
void SetReactState(ReactStates st) { m_reactState = st; }
ReactStates GetReactState() const { return m_reactState; }
diff --git a/src/server/game/Entities/Unit/Unit.cpp b/src/server/game/Entities/Unit/Unit.cpp
index 261ddcd9979..2e28c5bef97 100644
--- a/src/server/game/Entities/Unit/Unit.cpp
+++ b/src/server/game/Entities/Unit/Unit.cpp
@@ -4633,17 +4633,6 @@ bool Unit::HasNegativeAuraWithInterruptFlag(InterruptFlags flag, ObjectGuid guid
template TC_GAME_API bool Unit::HasNegativeAuraWithInterruptFlag(SpellAuraInterruptFlags flag, ObjectGuid guid) const;
template TC_GAME_API bool Unit::HasNegativeAuraWithInterruptFlag(SpellAuraInterruptFlags2 flag, ObjectGuid guid) const;
-bool Unit::HasNegativeAuraWithAttribute(uint32 flag, ObjectGuid guid) const
-{
- for (AuraApplicationMap::const_iterator iter = m_appliedAuras.begin(); iter != m_appliedAuras.end(); ++iter)
- {
- Aura const* aura = iter->second->GetBase();
- if (!iter->second->IsPositive() && aura->GetSpellInfo()->Attributes & flag && (!guid || aura->GetCasterGUID() == guid))
- return true;
- }
- return false;
-}
-
bool Unit::HasAuraWithMechanic(uint32 mechanicMask) const
{
for (AuraApplicationMap::const_iterator iter = m_appliedAuras.begin(); iter != m_appliedAuras.end(); ++iter)
@@ -4661,6 +4650,26 @@ bool Unit::HasAuraWithMechanic(uint32 mechanicMask) const
return false;
}
+bool Unit::HasStrongerAuraWithDR(SpellInfo const* auraSpellInfo, Unit* caster) const
+{
+ DiminishingGroup diminishGroup = auraSpellInfo->GetDiminishingReturnsGroupForSpell();
+ DiminishingLevels level = GetDiminishing(diminishGroup);
+ for (auto itr = m_appliedAuras.begin(); itr != m_appliedAuras.end(); ++itr)
+ {
+ SpellInfo const* spellInfo = itr->second->GetBase()->GetSpellInfo();
+ if (spellInfo->GetDiminishingReturnsGroupForSpell() != diminishGroup)
+ continue;
+
+ int32 existingDuration = itr->second->GetBase()->GetMaxDuration();
+ int32 newDuration = auraSpellInfo->GetMaxDuration();
+ ApplyDiminishingToDuration(auraSpellInfo, newDuration, caster, level);
+ if (newDuration > 0 && newDuration < existingDuration)
+ return true;
+ }
+
+ return false;
+}
+
AuraEffect* Unit::IsScriptOverriden(SpellInfo const* spell, int32 script) const
{
AuraEffectList const& auras = GetAuraEffectsByType(SPELL_AURA_OVERRIDE_CLASS_SCRIPTS);
@@ -9403,31 +9412,28 @@ void Unit::ModSpellDurationTime(SpellInfo const* spellInfo, int32 & duration, Sp
duration = int32(float(duration) * m_modAttackSpeedPct[RANGED_ATTACK]);
}
-DiminishingLevels Unit::GetDiminishing(DiminishingGroup group)
+DiminishingLevels Unit::GetDiminishing(DiminishingGroup group) const
{
- DiminishingReturn& diminish = m_Diminishing[group];
+ DiminishingReturn const& diminish = m_Diminishing[group];
if (!diminish.hitCount)
return DIMINISHING_LEVEL_1;
- // If last spell was cast more than 18 seconds ago - reset the count.
+ // If last spell was cast more than 18 seconds ago - reset level.
if (!diminish.stack && GetMSTimeDiffToNow(diminish.hitTime) > 18 * IN_MILLISECONDS)
- {
- diminish.hitCount = DIMINISHING_LEVEL_1;
return DIMINISHING_LEVEL_1;
- }
return DiminishingLevels(diminish.hitCount);
}
void Unit::IncrDiminishing(SpellInfo const* auraSpellInfo)
{
- DiminishingGroup const group = auraSpellInfo->GetDiminishingReturnsGroupForSpell();
- DiminishingLevels const maxLevel = auraSpellInfo->GetDiminishingReturnsMaxLevel();
+ DiminishingGroup group = auraSpellInfo->GetDiminishingReturnsGroupForSpell();
+ uint32 currentLevel = GetDiminishing(group);
+ uint32 const maxLevel = auraSpellInfo->GetDiminishingReturnsMaxLevel();
- // Checking for existing in the table
DiminishingReturn& diminish = m_Diminishing[group];
- if (static_cast<int32>(diminish.hitCount) < maxLevel)
- ++diminish.hitCount;
+ if (currentLevel < maxLevel)
+ diminish.hitCount = currentLevel + 1;
}
bool Unit::ApplyDiminishingToDuration(SpellInfo const* auraSpellInfo, int32& duration, Unit* caster, DiminishingLevels previousLevel) const
@@ -9447,9 +9453,7 @@ bool Unit::ApplyDiminishingToDuration(SpellInfo const* auraSpellInfo, int32& dur
Unit const* target = targetOwner ? targetOwner : this;
Unit const* source = casterOwner ? casterOwner : caster;
- if ((target->GetTypeId() == TYPEID_PLAYER
- || (target->ToCreature()->GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_ALL_DIMINISH))
- && source->GetTypeId() == TYPEID_PLAYER)
+ if (target->IsAffectedByDiminishingReturns() && source->GetTypeId() == TYPEID_PLAYER)
duration = limitDuration;
}
@@ -9475,9 +9479,9 @@ bool Unit::ApplyDiminishingToDuration(SpellInfo const* auraSpellInfo, int32& dur
}
case DIMINISHING_AOE_KNOCKBACK:
{
- if ((auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_PLAYER && (((targetOwner ? targetOwner : this)->ToPlayer())
- || (ToCreature() && (ToCreature()->GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_ALL_DIMINISH))))
- || auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_ALL)
+ if (auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_ALL ||
+ (auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_PLAYER &&
+ (targetOwner ? targetOwner->IsAffectedByDiminishingReturns() : IsAffectedByDiminishingReturns())))
{
DiminishingLevels diminish = previousLevel;
switch (diminish)
@@ -9491,9 +9495,9 @@ bool Unit::ApplyDiminishingToDuration(SpellInfo const* auraSpellInfo, int32& dur
}
default:
{
- if ((auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_PLAYER && (((targetOwner ? targetOwner : this)->ToPlayer())
- || (ToCreature() && (ToCreature()->GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_ALL_DIMINISH))))
- || auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_ALL)
+ if (auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_ALL ||
+ (auraSpellInfo->GetDiminishingReturnsGroupType() == DRTYPE_PLAYER &&
+ (targetOwner ? targetOwner->IsAffectedByDiminishingReturns() : IsAffectedByDiminishingReturns())))
{
DiminishingLevels diminish = previousLevel;
switch (diminish)
diff --git a/src/server/game/Entities/Unit/Unit.h b/src/server/game/Entities/Unit/Unit.h
index 79561280d05..0a1f3c72686 100644
--- a/src/server/game/Entities/Unit/Unit.h
+++ b/src/server/game/Entities/Unit/Unit.h
@@ -952,7 +952,8 @@ class TC_GAME_API Unit : public WorldObject
void SendCombatLogMessage(WorldPackets::CombatLog::CombatLogServerPacket* combatLog) const;
- DiminishingLevels GetDiminishing(DiminishingGroup group);
+ virtual bool IsAffectedByDiminishingReturns() const { return (GetCharmerOrOwnerPlayerOrPlayerItself() != nullptr); }
+ DiminishingLevels GetDiminishing(DiminishingGroup group) const;
void IncrDiminishing(SpellInfo const* auraSpellInfo);
bool ApplyDiminishingToDuration(SpellInfo const* auraSpellInfo, int32& duration, Unit* caster, DiminishingLevels previousLevel) const;
void ApplyDiminishingAura(DiminishingGroup group, bool apply);
@@ -1598,8 +1599,8 @@ class TC_GAME_API Unit : public WorldObject
bool HasAuraTypeWithValue(AuraType auratype, int32 value) const;
template <typename InterruptFlags>
bool HasNegativeAuraWithInterruptFlag(InterruptFlags flag, ObjectGuid guid = ObjectGuid::Empty) const;
- bool HasNegativeAuraWithAttribute(uint32 flag, ObjectGuid guid = ObjectGuid::Empty) const;
bool HasAuraWithMechanic(uint32 mechanicMask) const;
+ bool HasStrongerAuraWithDR(SpellInfo const* auraSpellInfo, Unit* caster) const;
AuraEffect* IsScriptOverriden(SpellInfo const* spell, int32 script) const;
uint32 GetDiseasesByCaster(ObjectGuid casterGUID, bool remove = false);
diff --git a/src/server/game/Spells/Spell.cpp b/src/server/game/Spells/Spell.cpp
index 754e25011d7..a9a5c34f38a 100644
--- a/src/server/game/Spells/Spell.cpp
+++ b/src/server/game/Spells/Spell.cpp
@@ -2650,9 +2650,7 @@ SpellMissInfo Spell::DoSpellHitOnUnit(Unit* unit, uint32 effectMask)
diminishLevel = unit->GetDiminishing(diminishGroup);
DiminishingReturnsType type = m_spellInfo->GetDiminishingReturnsGroupType();
// Increase Diminishing on unit, current informations for actually casts will use values above
- if ((type == DRTYPE_PLAYER &&
- (unit->GetCharmerOrOwnerPlayerOrPlayerItself() || (unit->GetTypeId() == TYPEID_UNIT && unit->ToCreature()->GetCreatureTemplate()->flags_extra & CREATURE_FLAG_EXTRA_ALL_DIMINISH))) ||
- type == DRTYPE_ALL)
+ if (type == DRTYPE_ALL || (type == DRTYPE_PLAYER && unit->IsAffectedByDiminishingReturns()))
unit->IncrDiminishing(m_spellInfo);
}
@@ -3203,11 +3201,9 @@ void Spell::_cast(bool skipCheck)
// skip check if done already (for instant cast spells for example)
if (!skipCheck)
{
- uint32 param1 = 0, param2 = 0;
- SpellCastResult castResult = CheckCast(false, &param1, &param2);
- if (castResult != SPELL_CAST_OK)
+ auto cleanupSpell = [this, modOwner](SpellCastResult res, uint32* p1 = nullptr, uint32* p2 = nullptr)
{
- SendCastResult(castResult, &param1, &param2);
+ SendCastResult(res, p1, p2);
SendInterrupted(0);
if (modOwner)
@@ -3215,6 +3211,13 @@ void Spell::_cast(bool skipCheck)
finish(false);
SetExecutedCurrently(false);
+ };
+
+ uint32 param1 = 0, param2 = 0;
+ SpellCastResult castResult = CheckCast(false, &param1, &param2);
+ if (castResult != SPELL_CAST_OK)
+ {
+ cleanupSpell(castResult, &param1, &param2);
return;
}
@@ -3230,14 +3233,34 @@ void Spell::_cast(bool skipCheck)
{
// Spell will be cast after completing the trade. Silently ignore at this place
my_trade->SetSpell(m_spellInfo->Id, m_CastItem);
- SendCastResult(SPELL_FAILED_DONT_REPORT);
- SendInterrupted(0);
+ cleanupSpell(SPELL_FAILED_DONT_REPORT);
+ return;
+ }
+ }
+ }
+ }
- modOwner->SetSpellModTakingSpell(this, false);
+ // check diminishing returns (again, only after finish cast bar, tested on retail)
+ if (Unit* target = m_targets.GetUnitTarget())
+ {
+ uint8 aura_effmask = 0;
+ for (uint8 i = 0; i < MAX_SPELL_EFFECTS; ++i)
+ if (m_spellInfo->GetEffect(i)->IsUnitOwnedAuraEffect())
+ aura_effmask |= 1 << i;
- finish(false);
- SetExecutedCurrently(false);
- return;
+ if (aura_effmask)
+ {
+ if (DiminishingGroup diminishGroup = m_spellInfo->GetDiminishingReturnsGroupForSpell())
+ {
+ DiminishingReturnsType type = m_spellInfo->GetDiminishingReturnsGroupType();
+ if (type == DRTYPE_ALL || (type == DRTYPE_PLAYER && target->IsAffectedByDiminishingReturns()))
+ {
+ Unit* caster = m_originalCaster ? m_originalCaster : m_caster;
+ if (target->HasStrongerAuraWithDR(m_spellInfo, caster))
+ {
+ cleanupSpell(SPELL_FAILED_AURA_BOUNCED);
+ return;
+ }
}
}
}