mirror of
https://github.com/mangosfour/server.git
synced 2025-12-28 13:37:13 +00:00
84 lines
2.4 KiB
C++
84 lines
2.4 KiB
C++
/*
|
|
* Copyright (C) 2005-2008 MaNGOS <http://getmangos.com/>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#ifndef MANGOS_HATEMATRIX_H
|
|
#define MANGOS_HATEMATRIX_H
|
|
|
|
#include "Utilities/HashMap.h"
|
|
#include <cassert>
|
|
|
|
class Unit;
|
|
|
|
struct MANGOS_DLL_DECL HateMatrix
|
|
{
|
|
typedef hash_map<Unit *, uint32> HateMatrixMapType;
|
|
|
|
inline uint32 operator[](Unit *unit) const
|
|
{
|
|
HateMatrixMapType::const_iterator iter = i_hateValues.find(unit);
|
|
return (iter == i_hateValues.end() ? 0 : iter->second);
|
|
}
|
|
|
|
inline uint32& operator[](Unit *unit)
|
|
{
|
|
HateMatrixMapType::iterator iter = i_hateValues.find(unit);
|
|
if( iter == i_hateValues.end() )
|
|
{
|
|
std::pair<HateMatrixMapType::iterator, bool> p = i_hateValues.insert( HateMatrixMapType::value_type(unit, 0) );
|
|
assert(p.second);
|
|
iter = p.first;
|
|
}
|
|
|
|
return iter->second;
|
|
}
|
|
|
|
inline void ClearMatrix(void) { i_hateValues.clear(); }
|
|
|
|
inline void RemoveValue(Unit *unit)
|
|
{
|
|
HateMatrixMapType::iterator iter = i_hateValues.find(unit);
|
|
if( iter != i_hateValues.end() )
|
|
i_hateValues.erase( iter );
|
|
}
|
|
|
|
inline void AddValue(Unit *unit, uint32 val)
|
|
{
|
|
(*this)[unit] += val;
|
|
}
|
|
|
|
private:
|
|
HateMatrixMapType i_hateValues;
|
|
};
|
|
|
|
struct HateBinder
|
|
{
|
|
static uint32 si_noHateValue;
|
|
uint32 &i_hateValue;
|
|
Unit *i_unit;
|
|
HateBinder(uint32 &val, Unit *u) : i_hateValue(val), i_unit(u) {}
|
|
HateBinder() : i_hateValue(si_noHateValue), i_unit(NULL) {}
|
|
HateBinder(const HateBinder &obj) : i_hateValue(obj.i_hateValue), i_unit(obj.i_unit) {}
|
|
|
|
HateBinder& operator=(const HateBinder &obj)
|
|
{
|
|
i_hateValue = obj.i_hateValue;
|
|
i_unit = obj.i_unit;
|
|
return *this;
|
|
}
|
|
};
|
|
#endif
|