src/game/PetAI.cpp

   1 /*
   2  * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include "PetAI.h"
  20 #include "Errors.h"
  21 #include "Pet.h"
  22 #include "Player.h"
  23 #include "Database/DBCStores.h"
  24 #include "Spell.h"
  25 #include "ObjectAccessor.h"
  26 #include "SpellMgr.h"
  27 #include "Creature.h"
  28 #include "World.h"
  29 #include "Util.h"
  30
  31 int PetAI::Permissible(const Creature *creature)
  32 {
  33     if( creature->isPet())
  34         return PERMIT_BASE_SPECIAL;
  35
  36     return PERMIT_BASE_NO;
  37 }
  38
  39 PetAI::PetAI(Creature &c) : i_pet(c), i_tracker(TIME_INTERVAL_LOOK), inCombat(false)
  40 {
  41     m_AllySet.clear();
  42     UpdateAllies();
  43 }
  44
  45 void PetAI::MoveInLineOfSight(Unit *u)
  46 {
  47     if( !i_pet.getVictim() && i_pet.GetCharmInfo() &&
  48         i_pet.GetCharmInfo()->HasReactState(REACT_AGGRESSIVE) &&
  49         u->isTargetableForAttack() && i_pet.IsHostileTo( u ) &&
  50         u->isInAccessablePlaceFor(&i_pet))
  51     {
  52         float attackRadius = i_pet.GetAttackDistance(u);
  53         if(i_pet.IsWithinDistInMap(u, attackRadius) && i_pet.GetDistanceZ(u) <= CREATURE_Z_ATTACK_RANGE)
  54         {
  55             if(i_pet.IsWithinLOSInMap(u))
  56             {
  57                 AttackStart(u);
  58                 u->RemoveSpellsCausingAura(SPELL_AURA_MOD_STEALTH);
  59             }
  60         }
  61     }
  62 }
  63
  64 void PetAI::AttackStart(Unit *u)
  65 {
  66     if( inCombat || !u || (i_pet.isPet() && ((Pet&)i_pet).getPetType() == MINI_PET) )
  67         return;
  68
  69     if(i_pet.Attack(u,true))
  70     {
  71         i_pet.clearUnitState(UNIT_STAT_FOLLOW);
  72         // TMGs call CreatureRelocation which via MoveInLineOfSight can call this function
  73         // thus with the following clear the original TMG gets invalidated and crash, doh
  74         // hope it doesn't start to leak memory without this :-/
  75         //i_pet->Clear();
  76         i_pet.GetMotionMaster()->MoveChase(u);
  77         inCombat = true;
  78     }
  79 }
  80
  81 void PetAI::EnterEvadeMode()
  82 {
  83 }
  84
  85 bool PetAI::IsVisible(Unit *pl) const
  86 {
  87     return _isVisible(pl);
  88 }
  89
  90 bool PetAI::_needToStop() const
  91 {
  92     // This is needed for charmed creatures, as once their target was reset other effects can trigger threat
  93     if(i_pet.isCharmed() && i_pet.getVictim() == i_pet.GetCharmer())
  94         return true;
  95
  96     return !i_pet.getVictim()->isTargetableForAttack();
  97 }
  98
  99 void PetAI::_stopAttack()
 100 {
 101     inCombat = false;
 102     if( !i_pet.isAlive() )
 103     {
 104         DEBUG_LOG("Creature stoped attacking cuz his dead [guid=%u]", i_pet.GetGUIDLow());
 105         i_pet.StopMoving();
 106         i_pet.GetMotionMaster()->Clear();
 107         i_pet.GetMotionMaster()->MoveIdle();
 108         i_pet.CombatStop();
 109         i_pet.getHostilRefManager().deleteReferences();
 110
 111         return;
 112     }
 113
 114     Unit* owner = i_pet.GetCharmerOrOwner();
 115
 116     if(owner && i_pet.GetCharmInfo() && i_pet.GetCharmInfo()->HasCommandState(COMMAND_FOLLOW))
 117     {
 118         i_pet.GetMotionMaster()->MoveFollow(owner,PET_FOLLOW_DIST,PET_FOLLOW_ANGLE);
 119     }
 120     else
 121     {
 122         i_pet.clearUnitState(UNIT_STAT_FOLLOW);
 123         i_pet.GetMotionMaster()->Clear();
 124         i_pet.GetMotionMaster()->MoveIdle();
 125     }
 126     i_pet.AttackStop();
 127 }
 128
 129 void PetAI::UpdateAI(const uint32 diff)
 130 {
 131     if (!i_pet.isAlive())
 132         return;
 133
 134     Unit* owner = i_pet.GetCharmerOrOwner();
 135
 136     if(m_updateAlliesTimer <= diff)
 137         // UpdateAllies self set update timer
 138         UpdateAllies();
 139     else
 140         m_updateAlliesTimer -= diff;
 141
 142     if (inCombat && !i_pet.getVictim())
 143         _stopAttack();
 144
 145     // i_pet.getVictim() can't be used for check in case stop fighting, i_pet.getVictim() clear at Unit death etc.
 146     if( i_pet.getVictim() )
 147     {
 148         if( _needToStop() )
 149         {
 150             DEBUG_LOG("Pet AI stoped attacking [guid=%u]", i_pet.GetGUIDLow());
 151             _stopAttack();
 152             return;
 153         }
 154         else if( i_pet.IsStopped() || i_pet.IsWithinDistInMap(i_pet.getVictim(), ATTACK_DISTANCE))
 155         {
 156             // required to be stopped cases
 157             if ( i_pet.IsStopped() && i_pet.IsNonMeleeSpellCasted(false) )
 158             {
 159                 if( i_pet.hasUnitState(UNIT_STAT_FOLLOW) )
 160                     i_pet.InterruptNonMeleeSpells(false);
 161                 else
 162                     return;
 163             }
 164             // not required to be stopped case
 165             else if( i_pet.isAttackReady() && i_pet.canReachWithAttack(i_pet.getVictim()) )
 166             {
 167                 i_pet.AttackerStateUpdate(i_pet.getVictim());
 168
 169                 i_pet.resetAttackTimer();
 170
 171                 if ( !i_pet.getVictim() )
 172                     return;
 173
 174                 //if pet misses its target, it will also be the first in threat list
 175                 i_pet.getVictim()->AddThreat(&i_pet,0.0f);
 176
 177                 if( _needToStop() )
 178                     _stopAttack();
 179             }
 180         }
 181     }
 182     else if(owner && i_pet.GetCharmInfo())
 183     {
 184         if(owner->isInCombat() && !(i_pet.GetCharmInfo()->HasReactState(REACT_PASSIVE) || i_pet.GetCharmInfo()->HasCommandState(COMMAND_STAY)))
 185         {
 186             AttackStart(owner->getAttackerForHelper());
 187         }
 188         else if(i_pet.GetCharmInfo()->HasCommandState(COMMAND_FOLLOW))
 189         {
 190             if (!i_pet.hasUnitState(UNIT_STAT_FOLLOW) )
 191             {
 192                 i_pet.GetMotionMaster()->MoveFollow(owner,PET_FOLLOW_DIST,PET_FOLLOW_ANGLE);
 193             }
 194         }
 195     }
 196
 197     if (i_pet.GetGlobalCooldown() == 0 && !i_pet.IsNonMeleeSpellCasted(false))
 198     {
 199         //Autocast
 200         for (uint8 i = 0; i < i_pet.GetPetAutoSpellSize(); i++)
 201         {
 202             uint32 spellID = i_pet.GetPetAutoSpellOnPos(i);
 203             if (!spellID)
 204                 continue;
 205
 206             SpellEntry const *spellInfo = sSpellStore.LookupEntry(spellID);
 207             if (!spellInfo)
 208                 continue;
 209
 210             // ignore some combinations of combat state and combat/noncombat spells
 211             if (!inCombat)
 212             {
 213                 if (!IsPositiveSpell(spellInfo->Id))
 214                     continue;
 215             }
 216             else
 217             {
 218                 if (IsNonCombatSpell(spellInfo))
 219                     continue;
 220             }
 221
 222             Spell *spell = new Spell(&i_pet, spellInfo, false, 0);
 223
 224             if(inCombat && !i_pet.hasUnitState(UNIT_STAT_FOLLOW) && spell->CanAutoCast(i_pet.getVictim()))
 225             {
 226                 m_targetSpellStore.push_back(std::make_pair<Unit*, Spell*>(i_pet.getVictim(), spell));
 227                 continue;
 228             }
 229             else
 230             {
 231                 bool spellUsed = false;
 232                 for(std::set<uint64>::iterator tar = m_AllySet.begin(); tar != m_AllySet.end(); ++tar)
 233                 {
 234                     Unit* Target = ObjectAccessor::GetUnit(i_pet,*tar);
 235
 236                     //only buff targets that are in combat, unless the spell can only be cast while out of combat
 237                     if(!Target)
 238                         continue;
 239
 240                     if(spell->CanAutoCast(Target))
 241                     {
 242                         m_targetSpellStore.push_back(std::make_pair<Unit*, Spell*>(Target, spell));
 243                         spellUsed = true;
 244                         break;
 245                     }
 246                 }
 247                 if (!spellUsed)
 248                     delete spell;
 249             }
 250         }
 251
 252         //found units to cast on to
 253         if(!m_targetSpellStore.empty())
 254         {
 255             uint32 index = urand(0, m_targetSpellStore.size() - 1);
 256
 257             Spell* spell  = m_targetSpellStore[index].second;
 258             Unit*  target = m_targetSpellStore[index].first;
 259
 260             m_targetSpellStore.erase(m_targetSpellStore.begin() + index);
 261
 262             SpellCastTargets targets;
 263             targets.setUnitTarget( target );
 264
 265             if( !i_pet.HasInArc(M_PI, target) )
 266             {
 267                 i_pet.SetInFront(target);
 268                 if( target->GetTypeId() == TYPEID_PLAYER )
 269                     i_pet.SendUpdateToPlayer( (Player*)target );
 270
 271                 if(owner && owner->GetTypeId() == TYPEID_PLAYER)
 272                     i_pet.SendUpdateToPlayer( (Player*)owner );
 273             }
 274
 275             i_pet.AddCreatureSpellCooldown(spell->m_spellInfo->Id);
 276             if(i_pet.isPet())
 277                 ((Pet*)&i_pet)->CheckLearning(spell->m_spellInfo->Id);
 278
 279             spell->prepare(&targets);
 280         }
 281         while (!m_targetSpellStore.empty())
 282         {
 283             delete m_targetSpellStore.begin()->second;
 284             m_targetSpellStore.erase(m_targetSpellStore.begin());
 285         }
 286     }
 287 }
 288
 289 bool PetAI::_isVisible(Unit *u) const
 290 {
 291     //return false;                                           //( ((Creature*)&i_pet)->GetDistanceSq(u) * 1.0<= sWorld.getConfig(CONFIG_SIGHT_GUARDER) && !u->m_stealth && u->isAlive());
 292     return i_pet.GetDistance(u) < sWorld.getConfig(CONFIG_SIGHT_GUARDER)
 293         && u->isVisibleForOrDetect(&i_pet,true);
 294 }
 295
 296 void PetAI::UpdateAllies()
 297 {
 298     Unit* owner = i_pet.GetCharmerOrOwner();
 299     Group *pGroup = NULL;
 300
 301     m_updateAlliesTimer = 10000;                            //update friendly targets every 10 seconds, lesser checks increase performance
 302
 303     if(!owner)
 304         return;
 305     else if(owner->GetTypeId() == TYPEID_PLAYER)
 306         pGroup = ((Player*)owner)->GetGroup();
 307
 308     //only pet and owner/not in group->ok
 309     if(m_AllySet.size() == 2 && !pGroup)
 310         return;
 311     //owner is in group; group members filled in already (no raid -> subgroupcount = whole count)
 312     if(pGroup && !pGroup->isRaidGroup() && m_AllySet.size() == (pGroup->GetMembersCount() + 2))
 313         return;
 314
 315     m_AllySet.clear();
 316     m_AllySet.insert(i_pet.GetGUID());
 317     if(pGroup)                                              //add group
 318     {
 319         for(GroupReference *itr = pGroup->GetFirstMember(); itr != NULL; itr = itr->next())
 320         {
 321             Player* Target = itr->getSource();
 322             if(!Target || !pGroup->SameSubGroup((Player*)owner, Target))
 323                 continue;
 324
 325             if(Target->GetGUID() == owner->GetGUID())
 326                 continue;
 327
 328             m_AllySet.insert(Target->GetGUID());
 329         }
 330     }
 331     else                                                    //remove group
 332         m_AllySet.insert(owner->GetGUID());
 333 }
 334
 335 void PetAI::AttackedBy(Unit *attacker)
 336 {
 337     //when attacked, fight back in case 1)no victim already AND 2)not set to passive AND 3)not set to stay, unless can it can reach attacker with melee attack anyway
 338     if(!i_pet.getVictim() && i_pet.GetCharmInfo() && !i_pet.GetCharmInfo()->HasReactState(REACT_PASSIVE) &&
 339         (!i_pet.GetCharmInfo()->HasCommandState(COMMAND_STAY) || i_pet.canReachWithAttack(attacker)))
 340         AttackStart(attacker);
 341 }