src/game/TargetedMovementGenerator.h

   1 /*
   2  * Copyright (C) 2005-2010 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 #ifndef MANGOS_TARGETEDMOVEMENTGENERATOR_H
  20 #define MANGOS_TARGETEDMOVEMENTGENERATOR_H
  21
  22 #include "MovementGenerator.h"
  23 #include "DestinationHolder.h"
  24 #include "Traveller.h"
  25 #include "FollowerReference.h"
  26
  27 class MANGOS_DLL_SPEC TargetedMovementGeneratorBase
  28 {
  29     public:
  30         TargetedMovementGeneratorBase(Unit &target) { i_target.link(&target, this); }
  31         void stopFollowing() { }
  32     protected:
  33         FollowerReference i_target;
  34 };
  35
  36 template<class T, typename D>
  37 class MANGOS_DLL_SPEC TargetedMovementGeneratorMedium
  38 : public MovementGeneratorMedium< T, D >, public TargetedMovementGeneratorBase
  39 {
  40     protected:
  41         TargetedMovementGeneratorMedium()
  42             : TargetedMovementGeneratorBase(), i_offset(0), i_angle(0), i_recalculateTravel(false) {}
  43         TargetedMovementGeneratorMedium(Unit &target)
  44             : TargetedMovementGeneratorBase(target), i_offset(0), i_angle(0), i_recalculateTravel(false) {}
  45         TargetedMovementGeneratorMedium(Unit &target, float offset, float angle)
  46             : TargetedMovementGeneratorBase(target), i_offset(offset), i_angle(angle), i_recalculateTravel(false) {}
  47         ~TargetedMovementGeneratorMedium() {}
  48
  49     public:
  50         bool Update(T &, const uint32 &);
  51
  52         Unit* GetTarget() const { return i_target.getTarget(); }
  53
  54         bool GetDestination(float &x, float &y, float &z) const
  55         {
  56             if(!i_destinationHolder.HasDestination()) return false;
  57             i_destinationHolder.GetDestination(x,y,z);
  58             return true;
  59         }
  60
  61         void unitSpeedChanged() { i_recalculateTravel=true; }
  62         void UpdateFinalDistance(float fDistance);
  63
  64     protected:
  65         void _setTargetLocation(T &);
  66
  67         float i_offset;
  68         float i_angle;
  69         DestinationHolder< Traveller<T> > i_destinationHolder;
  70         bool i_recalculateTravel;
  71 };
  72
  73 template<class T>
  74 class MANGOS_DLL_SPEC ChaseMovementGenerator : public TargetedMovementGeneratorMedium<T, ChaseMovementGenerator<T> >
  75 {
  76     public:
  77         ChaseMovementGenerator(Unit &target)
  78             : TargetedMovementGeneratorMedium<T, ChaseMovementGenerator<T> >(target) {}
  79         ChaseMovementGenerator(Unit &target, float offset, float angle)
  80             : TargetedMovementGeneratorMedium<T, ChaseMovementGenerator<T> >(target, offset, angle) {}
  81         ~ChaseMovementGenerator() {}
  82
  83         MovementGeneratorType GetMovementGeneratorType() const { return CHASE_MOTION_TYPE; }
  84
  85         void Initialize(T &);
  86         void Finalize(T &);
  87         void Interrupt(T &);
  88         void Reset(T &);
  89
  90         static void _clearUnitStateMove(T &u) { u.clearUnitState(UNIT_STAT_CHASE_MOVE); }
  91         static void _addUnitStateMove(T &u)  { u.addUnitState(UNIT_STAT_CHASE_MOVE); }
  92         bool _lostTarget(T &u) const { return u.getVictim() != this->GetTarget(); }
  93         void _reachTarget(T &);
  94 };
  95
  96 template<class T>
  97 class MANGOS_DLL_SPEC FollowMovementGenerator : public TargetedMovementGeneratorMedium<T, FollowMovementGenerator<T> >
  98 {
  99     public:
 100         FollowMovementGenerator(Unit &target)
 101             : TargetedMovementGeneratorMedium<T, FollowMovementGenerator<T> >(target){}
 102         FollowMovementGenerator(Unit &target, float offset, float angle)
 103             : TargetedMovementGeneratorMedium<T, FollowMovementGenerator<T> >(target, offset, angle) {}
 104         ~FollowMovementGenerator() {}
 105
 106         MovementGeneratorType GetMovementGeneratorType() const { return FOLLOW_MOTION_TYPE; }
 107
 108         void Initialize(T &);
 109         void Finalize(T &);
 110         void Interrupt(T &);
 111         void Reset(T &);
 112
 113         static void _clearUnitStateMove(T &u) { u.clearUnitState(UNIT_STAT_FOLLOW_MOVE); }
 114         static void _addUnitStateMove(T &u)  { u.addUnitState(UNIT_STAT_FOLLOW_MOVE); }
 115         bool _lostTarget(T &) const { return false; }
 116         void _reachTarget(T &) {}
 117     private:
 118         void _updateWalkMode(T &u);
 119         void _updateSpeed(T &u);
 120 };
 121
 122 #endif