mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-24 22:14:36 +02:00
3dd4fa2528
Final goal (of multiple PR's) is to remove all remaining pointers from serializeable game state, and replace them with either identifiers or with shared/unique pointers. CGTownInstance::town and CGHeroInstance::type members have been removed. Now this data is computed dynamically using subID member. VLC entity of a town can now be accessed via following methods: - getFactionID() returns ID of a faction - getFaction() returns pointer to a faction - getTown() returns pointer to a town VLC entity of a hero can now be accessed via following methods: - getHeroTypeID() returns ID of a hero - getHeroClassID() returns ID of a hero class - getHeroType() returns pointer to a hero - getHeroClass() returns pointer to a hero class
163 lines
4.1 KiB
C++
163 lines
4.1 KiB
C++
/*
|
|
* GatherTroops.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 "Goals.h"
|
|
#include "../VCAI.h"
|
|
#include "../AIUtility.h"
|
|
#include "../AIhelper.h"
|
|
#include "../FuzzyHelper.h"
|
|
#include "../ResourceManager.h"
|
|
#include "../BuildingManager.h"
|
|
#include "../../../lib/mapObjects/CGTownInstance.h"
|
|
#include "../../../lib/constants/StringConstants.h"
|
|
|
|
using namespace Goals;
|
|
|
|
bool GatherTroops::operator==(const GatherTroops & other) const
|
|
{
|
|
return objid == other.objid;
|
|
}
|
|
|
|
int GatherTroops::getCreaturesCount(const CArmedInstance * army)
|
|
{
|
|
int count = 0;
|
|
|
|
for(auto stack : army->Slots())
|
|
{
|
|
if(objid == stack.second->getCreatureID().num)
|
|
{
|
|
count += stack.second->count;
|
|
}
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
TSubgoal GatherTroops::whatToDoToAchieve()
|
|
{
|
|
logAi->trace("Entering GatherTroops::whatToDoToAchieve");
|
|
|
|
auto heroes = cb->getHeroesInfo(true);
|
|
|
|
for(auto hero : heroes)
|
|
{
|
|
if(getCreaturesCount(hero) >= this->value)
|
|
{
|
|
logAi->trace("Completing GATHER_TROOPS by hero %s", hero->getNameTranslated());
|
|
|
|
throw goalFulfilledException(sptr(*this));
|
|
}
|
|
}
|
|
|
|
TGoalVec solutions = getAllPossibleSubgoals();
|
|
|
|
if(solutions.empty())
|
|
return sptr(Explore());
|
|
|
|
return fh->chooseSolution(solutions);
|
|
}
|
|
|
|
|
|
TGoalVec GatherTroops::getAllPossibleSubgoals()
|
|
{
|
|
TGoalVec solutions;
|
|
|
|
for(const CGTownInstance * t : cb->getTownsInfo())
|
|
{
|
|
int count = getCreaturesCount(t->getUpperArmy());
|
|
|
|
if(count >= this->value)
|
|
{
|
|
if(t->visitingHero)
|
|
{
|
|
solutions.push_back(sptr(VisitObj(t->id.getNum()).sethero(t->visitingHero.get())));
|
|
}
|
|
else
|
|
{
|
|
vstd::concatenate(solutions, ai->ah->howToVisitObj(t));
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
auto creature = VLC->creatures()->getByIndex(objid);
|
|
if(t->getFactionID() == creature->getFactionID()) //TODO: how to force AI to build unupgraded creatures? :O
|
|
{
|
|
auto tryFindCreature = [&]() -> std::optional<std::vector<CreatureID>>
|
|
{
|
|
if(vstd::isValidIndex(t->getTown()->creatures, creature->getLevel() - 1))
|
|
{
|
|
auto itr = t->getTown()->creatures.begin();
|
|
std::advance(itr, creature->getLevel() - 1);
|
|
return make_optional(*itr);
|
|
}
|
|
return std::nullopt;
|
|
};
|
|
|
|
auto creatures = tryFindCreature();
|
|
if(!creatures)
|
|
continue;
|
|
|
|
int upgradeNumber = vstd::find_pos(*creatures, creature->getId());
|
|
if(upgradeNumber < 0)
|
|
continue;
|
|
|
|
BuildingID bid(BuildingID::DWELL_FIRST + creature->getLevel() - 1 + upgradeNumber * t->getTown()->creatures.size());
|
|
if(t->hasBuilt(bid) && ai->ah->freeResources().canAfford(creature->getFullRecruitCost())) //this assumes only creatures with dwellings are assigned to faction
|
|
{
|
|
solutions.push_back(sptr(BuyArmy(t, creature->getAIValue() * this->value).setobjid(objid)));
|
|
}
|
|
/*else //disable random building requests for now - this code needs to know a lot of town/resource context to do more good than harm
|
|
{
|
|
return sptr(BuildThis(bid, t).setpriority(priority));
|
|
}*/
|
|
}
|
|
}
|
|
|
|
for(auto obj : ai->visitableObjs)
|
|
{
|
|
auto d = dynamic_cast<const CGDwelling *>(obj);
|
|
|
|
if(!d || obj->ID == Obj::TOWN)
|
|
continue;
|
|
|
|
for(auto creature : d->creatures)
|
|
{
|
|
if(creature.first) //there are more than 0 creatures avaliabe
|
|
{
|
|
for(auto type : creature.second)
|
|
{
|
|
if(type.getNum() == objid && ai->ah->freeResources().canAfford(VLC->creatures()->getById(type)->getFullRecruitCost()))
|
|
vstd::concatenate(solutions, ai->ah->howToVisitObj(obj));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
CreatureID creID = CreatureID(objid);
|
|
|
|
vstd::erase_if(solutions, [&](TSubgoal goal)->bool
|
|
{
|
|
return goal->hero && !goal->hero->getSlotFor(creID).validSlot() && !goal->hero->getFreeSlot().validSlot();
|
|
});
|
|
|
|
return solutions;
|
|
//TODO: exchange troops between heroes
|
|
}
|
|
|
|
bool GatherTroops::fulfillsMe(TSubgoal goal)
|
|
{
|
|
if (!hero || hero == goal->hero) //we got army for desired hero or any hero
|
|
if (goal->objid == objid) //same creature type //TODO: consider upgrades?
|
|
if (goal->value >= value) //notify every time we get resources?
|
|
return true;
|
|
return false;
|
|
}
|