diff options
Diffstat (limited to 'src/game/ObjectAccessor.h')
-rw-r--r-- | src/game/ObjectAccessor.h | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/src/game/ObjectAccessor.h b/src/game/ObjectAccessor.h index fb465e7d2ba..655b44b98f2 100644 --- a/src/game/ObjectAccessor.h +++ b/src/game/ObjectAccessor.h @@ -17,18 +17,24 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ + #ifndef TRINITY_OBJECTACCESSOR_H #define TRINITY_OBJECTACCESSOR_H + #include "Platform/Define.h" #include "Policies/Singleton.h" #include <ace/Thread_Mutex.h> #include "Utilities/UnorderedMap.h" #include "Policies/ThreadingModel.h" + #include "UpdateData.h" + #include "GridDefines.h" #include "Object.h" #include "Player.h" + #include <set> + class Creature; class Corpse; class Unit; @@ -36,97 +42,124 @@ class GameObject; class DynamicObject; class WorldObject; class Map; + template <class T> class HashMapHolder { public: + typedef UNORDERED_MAP< uint64, T* > MapType; typedef ACE_Thread_Mutex LockType; typedef MaNGOS::GeneralLock<LockType > Guard; + static void Insert(T* o) { m_objectMap[o->GetGUID()] = o; } + static void Remove(T* o) { Guard guard(i_lock); m_objectMap.erase(o->GetGUID()); } + static T* Find(uint64 guid) { typename MapType::iterator itr = m_objectMap.find(guid); return (itr != m_objectMap.end()) ? itr->second : NULL; } + static MapType& GetContainer() { return m_objectMap; } + static LockType* GetLock() { return &i_lock; } private: + //Non instanceable only static HashMapHolder() {} + static LockType i_lock; static MapType m_objectMap; }; + class MANGOS_DLL_DECL ObjectAccessor : public MaNGOS::Singleton<ObjectAccessor, MaNGOS::ClassLevelLockable<ObjectAccessor, ACE_Thread_Mutex> > { + friend class Trinity::OperatorNew<ObjectAccessor>; ObjectAccessor(); ~ObjectAccessor(); ObjectAccessor(const ObjectAccessor &); ObjectAccessor& operator=(const ObjectAccessor &); + public: typedef UNORDERED_MAP<uint64, Corpse* > Player2CorpsesMapType; typedef UNORDERED_MAP<Player*, UpdateData>::value_type UpdateDataValueType; + template<class T> static T* GetObjectInWorld(uint64 guid, T* /*fake*/) { return HashMapHolder<T>::Find(guid); } + static Unit* GetObjectInWorld(uint64 guid, Unit* /*fake*/) { if(!guid) return NULL; + if (IS_PLAYER_GUID(guid)) { Unit * u = (Unit*)HashMapHolder<Player>::Find(guid); if(!u || !u->IsInWorld()) return NULL; + return u; } + if(IS_PET_GUID(guid)) return (Unit*)HashMapHolder<Pet>::Find(guid); + return (Unit*)HashMapHolder<Creature>::Find(guid); } + static Unit* GetUnitInOrOutOfWorld(uint64 guid, Unit* /*fake*/) { if(!guid) return NULL; + if (IS_PLAYER_GUID(guid)) { Unit * u = (Unit*)HashMapHolder<Player>::Find(guid); if(!u) return NULL; + return u; } // Other object types than player are unloaded while out of world return GetObjectInWorld(guid, ((Unit*)NULL)); } + template<class T> static T* GetObjectInWorld(uint32 mapid, float x, float y, uint64 guid, T* /*fake*/) { T* obj = HashMapHolder<T>::Find(guid); if(!obj || obj->GetMapId() != mapid) return NULL; + CellPair p = Trinity::ComputeCellPair(x,y); if(p.x_coord >= TOTAL_NUMBER_OF_CELLS_PER_MAP || p.y_coord >= TOTAL_NUMBER_OF_CELLS_PER_MAP ) { sLog.outError("ObjectAccessor::GetObjectInWorld: invalid coordinates supplied X:%f Y:%f grid cell [%u:%u]", x, y, p.x_coord, p.y_coord); return NULL; } + CellPair q = Trinity::ComputeCellPair(obj->GetPositionX(),obj->GetPositionY()); if(q.x_coord >= TOTAL_NUMBER_OF_CELLS_PER_MAP || q.y_coord >= TOTAL_NUMBER_OF_CELLS_PER_MAP ) { sLog.outError("ObjectAccessor::GetObjecInWorld: object (GUID: %u TypeId: %u) has invalid coordinates X:%f Y:%f grid cell [%u:%u]", obj->GetGUIDLow(), obj->GetTypeId(), obj->GetPositionX(), obj->GetPositionY(), q.x_coord, q.y_coord); return NULL; } + int32 dx = int32(p.x_coord) - int32(q.x_coord); int32 dy = int32(p.y_coord) - int32(q.y_coord); + if (dx > -2 && dx < 2 && dy > -2 && dy < 2) return obj; else return NULL; } + static WorldObject* GetWorldObject(WorldObject const &, uint64); static Object* GetObjectByTypeMask(WorldObject const &, uint64, uint32 typemask); static Creature* GetCreatureOrPetOrVehicle(WorldObject const &, uint64); @@ -137,52 +170,67 @@ class MANGOS_DLL_DECL ObjectAccessor : public MaNGOS::Singleton<ObjectAccessor, static Corpse* GetCorpse(WorldObject const &u, uint64 guid); static Pet* GetPet(uint64 guid); static Player* FindPlayer(uint64); + Player* FindPlayerByName(const char *name) ; + HashMapHolder<Player>::MapType& GetPlayers() { return HashMapHolder<Player>::GetContainer(); } + HashMapHolder<Creature>::MapType& GetCreatures() { return HashMapHolder<Creature>::GetContainer(); } + HashMapHolder<GameObject>::MapType& GetGameObjects() { return HashMapHolder<GameObject>::GetContainer(); } + template<class T> void AddObject(T *object) { HashMapHolder<T>::Insert(object); } + template<class T> void RemoveObject(T *object) { HashMapHolder<T>::Remove(object); } + void RemoveObject(Player *pl) { HashMapHolder<Player>::Remove(pl); + Guard guard(i_updateGuard); i_objects.erase((Object *)pl); } + void SaveAllPlayers(); + void AddUpdateObject(Object *obj) { Guard guard(i_updateGuard); i_objects.insert(obj); } + void RemoveUpdateObject(Object *obj) { Guard guard(i_updateGuard); i_objects.erase( obj ); } + void Update(uint32 diff); + Corpse* GetCorpseForPlayerGUID(uint64 guid); void RemoveCorpse(Corpse *corpse); void AddCorpse(Corpse *corpse); void AddCorpsesToGrid(GridPair const& gridpair,GridType& grid,Map* map); Corpse* ConvertCorpseForPlayer(uint64 player_guid, bool insignia = false); + static void UpdateObject(Object* obj, Player* exceptPlayer); static void _buildUpdateObject(Object* obj, UpdateDataMapType &); + static void UpdateObjectVisibility(WorldObject* obj); //static void UpdateVisibilityForPlayer(Player* player); private: @@ -198,10 +246,13 @@ class MANGOS_DLL_DECL ObjectAccessor : public MaNGOS::Singleton<ObjectAccessor, void BuildPacket(Player* plr); template<class SKIP> void Visit(GridRefManager<SKIP> &) {} }; + friend struct WorldObjectChangeAccumulator; Player2CorpsesMapType i_player2corpse; + typedef ACE_Thread_Mutex LockType; typedef MaNGOS::GeneralLock<LockType > Guard; + static void _buildChangeObjectForPlayer(WorldObject *, UpdateDataMapType &); static void _buildPacket(Player *, Object *, UpdateDataMapType &); void _update(void); |