mirror of
https://github.com/araxiaonline/TrinityCore.git
synced 2026-06-16 21:20:01 -04:00
ce3787f190
- Get rid of fugly void pointers - Streamlined AI selection code, using Permissible as it should've been instead of harcoded checks on the selector code - Moved IdleMovementGenerator singleton to factory code
118 lines
4.0 KiB
C++
118 lines
4.0 KiB
C++
/*
|
|
* Copyright (C) 2008-2017 TrinityCore <http://www.trinitycore.org/>
|
|
* Copyright (C) 2005-2009 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, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "Creature.h"
|
|
#include "CreatureAISelector.h"
|
|
#include "CreatureAIFactory.h"
|
|
|
|
#include "MovementGenerator.h"
|
|
|
|
#include "GameObject.h"
|
|
#include "GameObjectAIFactory.h"
|
|
|
|
#include "ScriptMgr.h"
|
|
|
|
namespace FactorySelector
|
|
{
|
|
template <class T>
|
|
struct PermissibleOrderPred
|
|
{
|
|
public:
|
|
PermissibleOrderPred(T const* obj) : _obj(obj) { }
|
|
|
|
template <class Value>
|
|
bool operator()(Value const& left, Value const& right) const
|
|
{
|
|
Permissible<T> const* leftPermit = ASSERT_NOTNULL(dynamic_cast<Permissible<T> const*>(left.second.get()));
|
|
Permissible<T> const* rightPermit = ASSERT_NOTNULL(dynamic_cast<Permissible<T> const*>(right.second.get()));
|
|
|
|
return leftPermit->Permit(_obj) < rightPermit->Permit(_obj);
|
|
}
|
|
|
|
private:
|
|
T const* const _obj;
|
|
};
|
|
|
|
CreatureAI* SelectAI(Creature* creature)
|
|
{
|
|
CreatureAICreator const* ai_factory = nullptr;
|
|
|
|
// scriptname in db
|
|
if (!ai_factory)
|
|
if (CreatureAI* scriptedAI = sScriptMgr->GetCreatureAI(creature))
|
|
return scriptedAI;
|
|
|
|
// AIname in db
|
|
std::string const& aiName = creature->GetAIName();
|
|
if (!ai_factory && !aiName.empty())
|
|
ai_factory = sCreatureAIRegistry->GetRegistryItem(aiName);
|
|
|
|
// select by permit check
|
|
if (!ai_factory)
|
|
{
|
|
CreatureAIRegistry::RegistryMapType const& items = sCreatureAIRegistry->GetRegisteredItems();
|
|
auto itr = std::max_element(items.begin(), items.end(), PermissibleOrderPred<Creature>(creature));
|
|
if (itr != items.end())
|
|
ai_factory = itr->second.get();
|
|
}
|
|
|
|
if (!ai_factory)
|
|
ai_factory = sCreatureAIRegistry->GetRegistryItem("NullCreatureAI");
|
|
|
|
return ASSERT_NOTNULL(ai_factory)->Create(creature);
|
|
}
|
|
|
|
MovementGenerator* SelectMovementGenerator(Unit* unit)
|
|
{
|
|
MovementGeneratorType type = IDLE_MOTION_TYPE;
|
|
if (unit->GetTypeId() == TYPEID_UNIT)
|
|
type = unit->ToCreature()->GetDefaultMovementType();
|
|
|
|
MovementGeneratorCreator const* mv_factory = sMovementGeneratorRegistry->GetRegistryItem(type);
|
|
return ASSERT_NOTNULL(mv_factory)->Create(unit);
|
|
}
|
|
|
|
GameObjectAI* SelectGameObjectAI(GameObject* go)
|
|
{
|
|
GameObjectAICreator const* ai_factory = nullptr;
|
|
|
|
// scriptname in db
|
|
if (GameObjectAI* scriptedAI = sScriptMgr->GetGameObjectAI(go))
|
|
return scriptedAI;
|
|
|
|
// AIname in db
|
|
std::string const& aiName = go->GetAIName();
|
|
if (!ai_factory && !aiName.empty())
|
|
ai_factory = sGameObjectAIRegistry->GetRegistryItem(aiName);
|
|
|
|
// select by permit check
|
|
if (!ai_factory)
|
|
{
|
|
GameObjectAIRegistry::RegistryMapType const& items = sGameObjectAIRegistry->GetRegisteredItems();
|
|
auto itr = std::max_element(items.begin(), items.end(), PermissibleOrderPred<GameObject>(go));
|
|
if (itr != items.end())
|
|
ai_factory = itr->second.get();
|
|
}
|
|
|
|
if (!ai_factory)
|
|
ai_factory = sGameObjectAIRegistry->GetRegistryItem("NullGameObjectAI");
|
|
|
|
return ASSERT_NOTNULL(ai_factory)->Create(go);
|
|
}
|
|
}
|