mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-20 20:23:03 +02:00
84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
/*
|
|
* BuyArmyBehavior.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 "BuyArmyBehavior.h"
|
|
#include "../AIGateway.h"
|
|
#include "../AIUtility.h"
|
|
#include "../Goals/BuyArmy.h"
|
|
#include "../Engine/Nullkiller.h"
|
|
|
|
namespace NKAI
|
|
{
|
|
|
|
using namespace Goals;
|
|
|
|
std::string BuyArmyBehavior::toString() const
|
|
{
|
|
return "Buy army";
|
|
}
|
|
|
|
Goals::TGoalVec BuyArmyBehavior::decompose(const Nullkiller * ai) const
|
|
{
|
|
Goals::TGoalVec tasks;
|
|
|
|
auto heroes = cb->getHeroesInfo();
|
|
|
|
if(heroes.empty())
|
|
{
|
|
return tasks;
|
|
}
|
|
|
|
for(auto town : cb->getTownsInfo())
|
|
{
|
|
//If we can recruit a hero that comes with more army than he costs, we are better off spending our gold on them
|
|
if (ai->heroManager->canRecruitHero(town))
|
|
{
|
|
auto availableHeroes = ai->cb->getAvailableHeroes(town);
|
|
for (auto hero : availableHeroes)
|
|
{
|
|
if (hero->getArmyCost() > GameConstants::HERO_GOLD_COST)
|
|
return tasks;
|
|
}
|
|
}
|
|
|
|
if (ai->buildAnalyzer->isGoldPressureHigh() && !town->hasBuilt(BuildingID::CITY_HALL) && cb->canBuildStructure(town, BuildingID::CITY_HALL) != EBuildingState::FORBIDDEN)
|
|
{
|
|
return tasks;
|
|
}
|
|
|
|
auto townArmyAvailableToBuy = ai->armyManager->getArmyAvailableToBuyAsCCreatureSet(
|
|
town,
|
|
ai->getFreeResources());
|
|
|
|
for(const CGHeroInstance * targetHero : heroes)
|
|
{
|
|
if(ai->heroManager->getHeroRole(targetHero) == HeroRole::MAIN)
|
|
{
|
|
auto reinforcement = ai->armyManager->howManyReinforcementsCanGet(
|
|
targetHero,
|
|
targetHero,
|
|
&*townArmyAvailableToBuy);
|
|
|
|
if(reinforcement)
|
|
vstd::amin(reinforcement, ai->armyManager->howManyReinforcementsCanBuy(town->getUpperArmy(), town));
|
|
|
|
if(reinforcement)
|
|
{
|
|
tasks.push_back(Goals::sptr(Goals::BuyArmy(town, reinforcement).setpriority(5)));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return tasks;
|
|
}
|
|
|
|
}
|