*wintergrasp: allow players to click titan relic to change the attacker/defender teams. Now only some of npc and portal will be changed.

--HG--
branch : trunk

4 files changed, 120 insertions, 35 deletions

diff --git a/src/game/Creature.cpp b/src/game/Creature.cpp
index 76bc87cf470..c695993ce26 100644
--- a/src/game/Creature.cpp
+++ b/src/game/Creature.cpp
@@ -1464,7 +1464,7 @@ bool Creature::CreateFromProto(uint32 guidlow, uint32 Entry, uint32 team, const
         return false;
     }
 
-    m_originalEntry = Entry;
+    SetOriginalEntry(Entry);
 
     if(isVehicle())
         Object::_Create(guidlow, Entry, HIGHGUID_VEHICLE);
diff --git a/src/game/Creature.h b/src/game/Creature.h
index 76072f124d9..e4ced6ff457 100644
--- a/src/game/Creature.h
+++ b/src/game/Creature.h
@@ -716,6 +716,8 @@ class TRINITY_DLL_SPEC Creature : public Unit
         bool IsDamageEnoughForLootingAndReward() { return m_unDamageByPlayers >= (uint32)(GetMaxHealth() / 2); }
         void AddDamageByPlayers(uint32 unDamage) { m_unDamageByPlayers += unDamage; }
         void ResetDamageByPlayers() { m_unDamageByPlayers = 0; }
+
+        void SetOriginalEntry(uint32 entry) { m_originalEntry = entry; }
         
     protected:
         bool CreateFromProto(uint32 guidlow,uint32 Entry,uint32 team, const CreatureData *data = NULL);
diff --git a/src/game/Wintergrasp.cpp b/src/game/Wintergrasp.cpp
index 02cc9d00a03..abcee7c3812 100644
--- a/src/game/Wintergrasp.cpp
+++ b/src/game/Wintergrasp.cpp
@@ -21,6 +21,33 @@
 #include "Vehicle.h"
 //#include "GameEventMgr.h"
 #include "ObjectMgr.h"
+#include "World.h"
+
+typedef uint32 TeamPair[2];
+
+const TeamPair CreatureEntryPair[] =
+{
+    {30739, 30740},
+    {30400, 30499},
+    {0,0}
+};
+
+const TeamPair GODisplayPair[] =
+{
+    {5651, 5652},
+    {8256, 8257},
+    {0,0}
+};
+
+void LoadTeamPair(TeamPairMap &pairMap, const TeamPair *pair)
+{
+    while((*pair)[0])
+    {
+        pairMap[(*pair)[0]] = (*pair)[1];
+        pairMap[(*pair)[1]] = (*pair)[0];
+        ++pair;
+    }
+}
 
 bool OPvPWintergrasp::SetupOutdoorPvP()
 {
@@ -38,66 +65,106 @@ bool OPvPWintergrasp::SetupOutdoorPvP()
     //gameeventmgr.StartInternalEvent(GameEventWintergraspDefender[m_defender]);
 
     //Titan Relic eventid = 19982
-    objmgr.AddGameObject(192829, 571, 5444.6, 2840.8, 420.43, 0);
+    objmgr.AddGameObject(192829, 571, 5440, 2840.8, 420.43, 0);
 
+    LoadTeamPair(m_goDisplayPair, GODisplayPair);
+    LoadTeamPair(m_creEntryPair, CreatureEntryPair);
 
     return true;
 }
 
+void OPvPWintergrasp::ProcessEvent(WorldObject *obj, uint32 eventId)
+{
+    if(eventId == 19982)
+        ChangeDefender();
+}
+
+void OPvPWintergrasp::ChangeDefender()
+{
+    m_defender = OTHER_TEAM(m_defender);
+    if(m_defender == TEAM_ALLIANCE)
+        sWorld.SendZoneText(ZONE_WINTERGRASP, "Alliance has taken over the fortress!");
+    else
+        sWorld.SendZoneText(ZONE_WINTERGRASP, "Horde has taken over the fortress!");
+    UpdateAllWorldObject();
+}
+
 uint32 OPvPWintergrasp::GetCreatureEntry(uint32 guidlow, uint32 entry)
 {
     if(m_defender == TEAM_ALLIANCE)
     {
-        switch(entry)
-        {
-            case 30739: return 30740;
-            case 30740: return 30739;
-            case 30400: return 30499;
-            case 30499: return 30400;
-        }
+        TeamPairMap::const_iterator itr = m_creEntryPair.find(entry);
+        if(itr != m_creEntryPair.end())
+            return itr->second;
     }
     return entry;
 }
 
-/*
-uint32 OPvPWintergrasp::GetGameObjectEntry(uint32 guidlow, uint32 entry)
+void OPvPWintergrasp::OnCreatureCreate(Creature *creature, bool add)
 {
-    if(m_defender == TEAM_ALLIANCE)
+    if(m_creEntryPair.find(creature->GetEntry()) != m_creEntryPair.end())
     {
-        GameObjectInfo const* goInfo = objmgr.GetGameObjectInfo(entry);
-        if(!goInfo)
-            return 0;
-        switch(goInfo->displayId)
-        {
-            case 5651: return 192289;
-            case 5652: return 192288;
-            case 8256: return 192502;
-            case 8257: return 192501;
-        }
+        if(add) m_creatures.insert(creature);
+        else m_creatures.erase(creature);
     }
-    return entry;
 }
-*/
 
 void OPvPWintergrasp::OnGameObjectCreate(GameObject *go, bool add)
 {
     OutdoorPvP::OnGameObjectCreate(go, add);
-    if(m_defender == TEAM_ALLIANCE)
+
+    if(UpdateGameObjectInfo(go))
     {
-        switch(go->GetEntry())
-        {
-            case 190763: go->SetUInt32Value(GAMEOBJECT_FACTION, WintergraspFaction[m_defender]); break;
-        }
+        if(add) m_gobjects.insert(go);
+        else m_gobjects.erase(go);
+    }
+}
 
-        // Note: this is only for test, still need db support
-        switch(go->GetGOInfo()->displayId)
+void OPvPWintergrasp::UpdateAllWorldObject()
+{
+    for(GameObjectSet::iterator itr = m_gobjects.begin(); itr != m_gobjects.end(); ++itr)
+        UpdateGameObjectInfo(*itr);
+    for(CreatureSet::iterator itr = m_creatures.begin(); itr != m_creatures.end(); ++itr)
+        UpdateCreatureInfo(*itr);
+}
+
+bool OPvPWintergrasp::UpdateCreatureInfo(Creature *creature)
+{
+    TeamPairMap::const_iterator itr = m_creEntryPair.find(creature->GetCreatureData()->id);
+    if(itr != m_creEntryPair.end())
+    {
+        uint32 entry = m_defender == TEAM_ALLIANCE ? itr->second : itr->first;
+        if(entry != creature->GetEntry())
         {
-            case 5651: go->SetUInt32Value(GAMEOBJECT_DISPLAYID, 5652); break;
-            case 5652: go->SetUInt32Value(GAMEOBJECT_DISPLAYID, 5651); break;
-            case 8256: go->SetUInt32Value(GAMEOBJECT_DISPLAYID, 8257); break;
-            case 8257: go->SetUInt32Value(GAMEOBJECT_DISPLAYID, 8256); break;
+            creature->SetOriginalEntry(entry);
+            creature->setDeathState(DEAD);
+            creature->Respawn();
         }
     }
+
+    return false;
+}
+
+bool OPvPWintergrasp::UpdateGameObjectInfo(GameObject *go)
+{
+    switch(go->GetEntry())
+    {
+        // Defender's Portal
+        case 190763:
+            go->SetUInt32Value(GAMEOBJECT_FACTION, WintergraspFaction[m_defender]);
+            return true;
+    }
+
+    // Note: this is only for test, still need db support
+    TeamPairMap::const_iterator itr = m_goDisplayPair.find(go->GetGOInfo()->displayId);
+    if(itr != m_goDisplayPair.end())
+    {
+        go->SetUInt32Value(GAMEOBJECT_DISPLAYID, m_defender == TEAM_ALLIANCE ?
+            itr->second : itr->first);
+        return true;
+    }
+
+    return false;
 }
 
 void OPvPWintergrasp::HandlePlayerEnterZone(Player * plr, uint32 zone)
diff --git a/src/game/Wintergrasp.h b/src/game/Wintergrasp.h
index cce9eb55d7e..d9d8756a972 100644
--- a/src/game/Wintergrasp.h
+++ b/src/game/Wintergrasp.h
@@ -55,11 +55,15 @@ struct BuildingState
     DamageState damageState;
 };
 
+typedef std::map<uint32, uint32> TeamPairMap;
+
 class OPvPWintergrasp : public OutdoorPvP
 {
     protected:
         typedef std::list<const AreaPOIEntry *> AreaPOIList;
         typedef std::map<uint32, BuildingState *> BuildingStateMap;
+        typedef std::set<Creature*> CreatureSet;
+        typedef std::set<GameObject*> GameObjectSet;
     public:
         explicit OPvPWintergrasp() : m_tenacityStack(0) {}
         bool SetupOutdoorPvP();
@@ -67,8 +71,11 @@ class OPvPWintergrasp : public OutdoorPvP
         uint32 GetCreatureEntry(uint32 guidlow, uint32 entry);
         //uint32 GetGameObjectEntry(uint32 guidlow, uint32 entry);
 
+        void OnCreatureCreate(Creature *creature, bool add);
         void OnGameObjectCreate(GameObject *go, bool add);
 
+        void ProcessEvent(WorldObject *obj, uint32 eventId);
+
         void HandlePlayerEnterZone(Player *plr, uint32 zone);
         void HandlePlayerLeaveZone(Player *plr, uint32 zone);
         void HandleKill(Player *killer, Unit *victim);
@@ -77,8 +84,17 @@ class OPvPWintergrasp : public OutdoorPvP
         int32 m_tenacityStack;
         AreaPOIList areaPOIs;
         BuildingStateMap buildingStates;
+        CreatureSet m_creatures;
+        GameObjectSet m_gobjects;
+
+        TeamPairMap m_creEntryPair, m_goDisplayPair;
+
+        void ChangeDefender();
 
         void UpdateTenacityStack();
+        bool UpdateCreatureInfo(Creature *creature);
+        void UpdateAllWorldObject();
+        bool UpdateGameObjectInfo(GameObject *go);
 };
 
 #endif