mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-24 22:14:36 +02:00
111 lines
2.6 KiB
C++
111 lines
2.6 KiB
C++
/*
|
|
* AIPathfinder.cpp, part of VCMI engine
|
|
*
|
|
* Authors: listed in file AUTHORS in main folder
|
|
*
|
|
* License: GNU General Public License v2.0 or later
|
|
* Full text of license available in license.txt file, in main folder
|
|
*
|
|
*/
|
|
#include "StdInc.h"
|
|
#include "AIPathfinder.h"
|
|
#include "AIPathfinderConfig.h"
|
|
#include "../../../CCallback.h"
|
|
#include "../../../lib/mapping/CMapDefines.h"
|
|
|
|
std::vector<std::shared_ptr<AINodeStorage>> AIPathfinder::storagePool;
|
|
std::map<HeroPtr, std::shared_ptr<AINodeStorage>> AIPathfinder::storageMap;
|
|
|
|
AIPathfinder::AIPathfinder(CPlayerSpecificInfoCallback * cb, VCAI * ai)
|
|
:cb(cb), ai(ai)
|
|
{
|
|
}
|
|
|
|
void AIPathfinder::init()
|
|
{
|
|
storagePool.clear();
|
|
storageMap.clear();
|
|
}
|
|
|
|
bool AIPathfinder::isTileAccessible(const HeroPtr & hero, const int3 & tile) const
|
|
{
|
|
std::shared_ptr<const AINodeStorage> nodeStorage = getStorage(hero);
|
|
|
|
return nodeStorage->isTileAccessible(tile, EPathfindingLayer::LAND)
|
|
|| nodeStorage->isTileAccessible(tile, EPathfindingLayer::SAIL);
|
|
}
|
|
|
|
std::vector<AIPath> AIPathfinder::getPathInfo(const HeroPtr & hero, const int3 & tile) const
|
|
{
|
|
std::shared_ptr<const AINodeStorage> nodeStorage = getStorage(hero);
|
|
|
|
const TerrainTile * tileInfo = cb->getTile(tile, false);
|
|
|
|
if(!tileInfo)
|
|
{
|
|
return std::vector<AIPath>();
|
|
}
|
|
|
|
return nodeStorage->getChainInfo(tile, !tileInfo->isWater());
|
|
}
|
|
|
|
void AIPathfinder::updatePaths(std::vector<HeroPtr> heroes)
|
|
{
|
|
storageMap.clear();
|
|
|
|
auto calculatePaths = [&](const CGHeroInstance * hero, std::shared_ptr<AIPathfinding::AIPathfinderConfig> config)
|
|
{
|
|
logAi->debug("Recalculate paths for %s", hero->getNameTranslated());
|
|
|
|
cb->calculatePaths(config);
|
|
};
|
|
|
|
std::vector<CThreadHelper::Task> calculationTasks;
|
|
|
|
for(HeroPtr hero : heroes)
|
|
{
|
|
std::shared_ptr<AINodeStorage> nodeStorage;
|
|
|
|
if(storageMap.size() < storagePool.size())
|
|
{
|
|
nodeStorage = storagePool.at(storageMap.size());
|
|
}
|
|
else
|
|
{
|
|
nodeStorage = std::make_shared<AINodeStorage>(cb->getMapSize());
|
|
storagePool.push_back(nodeStorage);
|
|
}
|
|
|
|
storageMap[hero] = nodeStorage;
|
|
nodeStorage->setHero(hero, ai);
|
|
|
|
auto config = std::make_shared<AIPathfinding::AIPathfinderConfig>(cb, ai, nodeStorage);
|
|
|
|
calculationTasks.push_back(std::bind(calculatePaths, hero.get(), config));
|
|
}
|
|
|
|
int threadsCount = std::min(
|
|
boost::thread::hardware_concurrency(),
|
|
(uint32_t)calculationTasks.size());
|
|
|
|
if(threadsCount <= 1)
|
|
{
|
|
for(auto task : calculationTasks)
|
|
{
|
|
task();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
CThreadHelper helper(&calculationTasks, threadsCount);
|
|
|
|
helper.run();
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<const AINodeStorage> AIPathfinder::getStorage(const HeroPtr & hero) const
|
|
{
|
|
return storageMap.at(hero);
|
|
}
|
|
|