PriorityEvaluator.cpp 50 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576
  1. /*
  2. * PriorityEvaluator.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include <limits>
  12. #include "Nullkiller.h"
  13. #include "../../../lib/mapObjectConstructors/AObjectTypeHandler.h"
  14. #include "../../../lib/mapObjectConstructors/CObjectClassesHandler.h"
  15. #include "../../../lib/mapObjectConstructors/CBankInstanceConstructor.h"
  16. #include "../../../lib/mapObjects/MapObjects.h"
  17. #include "../../../lib/CCreatureHandler.h"
  18. #include "../../../lib/VCMI_Lib.h"
  19. #include "../../../lib/StartInfo.h"
  20. #include "../../../CCallback.h"
  21. #include "../../../lib/filesystem/Filesystem.h"
  22. #include "../Goals/ExecuteHeroChain.h"
  23. #include "../Goals/BuildThis.h"
  24. #include "../Goals/StayAtTown.h"
  25. #include "../Goals/ExchangeSwapTownHeroes.h"
  26. #include "../Goals/DismissHero.h"
  27. #include "../Markers/UnlockCluster.h"
  28. #include "../Markers/HeroExchange.h"
  29. #include "../Markers/ArmyUpgrade.h"
  30. #include "../Markers/DefendTown.h"
  31. namespace NKAI
  32. {
  33. #define MIN_AI_STRENGTH (0.5f) //lower when combat AI gets smarter
  34. #define UNGUARDED_OBJECT (100.0f) //we consider unguarded objects 100 times weaker than us
  35. const float MIN_CRITICAL_VALUE = 2.0f;
  36. EvaluationContext::EvaluationContext(const Nullkiller* ai)
  37. : movementCost(0.0),
  38. manaCost(0),
  39. danger(0),
  40. closestWayRatio(1),
  41. movementCostByRole(),
  42. skillReward(0),
  43. goldReward(0),
  44. goldCost(0),
  45. armyReward(0),
  46. armyLossPersentage(0),
  47. heroRole(HeroRole::SCOUT),
  48. turn(0),
  49. strategicalValue(0),
  50. conquestValue(0),
  51. evaluator(ai),
  52. enemyHeroDangerRatio(0),
  53. threat(0),
  54. armyGrowth(0),
  55. armyInvolvement(0),
  56. defenseValue(0),
  57. isDefend(false),
  58. threatTurns(INT_MAX),
  59. involvesSailing(false),
  60. isTradeBuilding(false),
  61. isExchange(false)
  62. {
  63. }
  64. void EvaluationContext::addNonCriticalStrategicalValue(float value)
  65. {
  66. vstd::amax(strategicalValue, std::min(value, MIN_CRITICAL_VALUE));
  67. }
  68. PriorityEvaluator::~PriorityEvaluator()
  69. {
  70. delete engine;
  71. }
  72. void PriorityEvaluator::initVisitTile()
  73. {
  74. auto file = CResourceHandler::get()->load(ResourcePath("config/ai/nkai/object-priorities.txt"))->readAll();
  75. std::string str = std::string((char *)file.first.get(), file.second);
  76. engine = fl::FllImporter().fromString(str);
  77. armyLossPersentageVariable = engine->getInputVariable("armyLoss");
  78. armyGrowthVariable = engine->getInputVariable("armyGrowth");
  79. heroRoleVariable = engine->getInputVariable("heroRole");
  80. dangerVariable = engine->getInputVariable("danger");
  81. turnVariable = engine->getInputVariable("turn");
  82. mainTurnDistanceVariable = engine->getInputVariable("mainTurnDistance");
  83. scoutTurnDistanceVariable = engine->getInputVariable("scoutTurnDistance");
  84. goldRewardVariable = engine->getInputVariable("goldReward");
  85. armyRewardVariable = engine->getInputVariable("armyReward");
  86. skillRewardVariable = engine->getInputVariable("skillReward");
  87. rewardTypeVariable = engine->getInputVariable("rewardType");
  88. closestHeroRatioVariable = engine->getInputVariable("closestHeroRatio");
  89. strategicalValueVariable = engine->getInputVariable("strategicalValue");
  90. goldPressureVariable = engine->getInputVariable("goldPressure");
  91. goldCostVariable = engine->getInputVariable("goldCost");
  92. fearVariable = engine->getInputVariable("fear");
  93. value = engine->getOutputVariable("Value");
  94. }
  95. bool isAnotherAi(const CGObjectInstance * obj, const CPlayerSpecificInfoCallback & cb)
  96. {
  97. return obj->getOwner().isValidPlayer()
  98. && cb.getStartInfo()->getIthPlayersSettings(obj->getOwner()).isControlledByAI();
  99. }
  100. int32_t estimateTownIncome(CCallback * cb, const CGObjectInstance * target, const CGHeroInstance * hero)
  101. {
  102. auto relations = cb->getPlayerRelations(hero->tempOwner, target->tempOwner);
  103. if(relations != PlayerRelations::ENEMIES)
  104. return 0; // if we already own it, no additional reward will be received by just visiting it
  105. auto booster = isAnotherAi(target, *cb) ? 1 : 2;
  106. auto town = cb->getTown(target->id);
  107. auto fortLevel = town->fortLevel();
  108. if(town->hasCapitol())
  109. return booster * 2000;
  110. // probably well developed town will have city hall
  111. if(fortLevel == CGTownInstance::CASTLE) return booster * 750;
  112. return booster * (town->hasFort() && town->tempOwner != PlayerColor::NEUTRAL ? booster * 500 : 250);
  113. }
  114. TResources getCreatureBankResources(const CGObjectInstance * target, const CGHeroInstance * hero)
  115. {
  116. //Fixme: unused variable hero
  117. auto objectInfo = target->getObjectHandler()->getObjectInfo(target->appearance);
  118. CBankInfo * bankInfo = dynamic_cast<CBankInfo *>(objectInfo.get());
  119. auto resources = bankInfo->getPossibleResourcesReward();
  120. TResources result = TResources();
  121. int sum = 0;
  122. for(auto & reward : resources)
  123. {
  124. result += reward.data * reward.chance;
  125. sum += reward.chance;
  126. }
  127. return sum > 1 ? result / sum : result;
  128. }
  129. int32_t getResourcesGoldReward(const TResources & res)
  130. {
  131. int32_t result = 0;
  132. for(auto r : GameResID::ALL_RESOURCES())
  133. {
  134. if(res[r] > 0)
  135. result += r == EGameResID::GOLD ? res[r] : res[r] * 100;
  136. }
  137. return result;
  138. }
  139. uint64_t getCreatureBankArmyReward(const CGObjectInstance * target, const CGHeroInstance * hero)
  140. {
  141. auto objectInfo = target->getObjectHandler()->getObjectInfo(target->appearance);
  142. CBankInfo * bankInfo = dynamic_cast<CBankInfo *>(objectInfo.get());
  143. auto creatures = bankInfo->getPossibleCreaturesReward(target->cb);
  144. uint64_t result = 0;
  145. const auto& slots = hero->Slots();
  146. ui64 weakestStackPower = 0;
  147. int duplicatingSlots = getDuplicatingSlots(hero);
  148. if (slots.size() >= GameConstants::ARMY_SIZE)
  149. {
  150. //No free slot, we might discard our weakest stack
  151. weakestStackPower = std::numeric_limits<ui64>().max();
  152. for (const auto & stack : slots)
  153. {
  154. vstd::amin(weakestStackPower, stack.second->getPower());
  155. }
  156. }
  157. for (auto c : creatures)
  158. {
  159. //Only if hero has slot for this creature in the army
  160. auto ccre = dynamic_cast<const CCreature*>(c.data.type);
  161. if (hero->getSlotFor(ccre).validSlot() || duplicatingSlots > 0)
  162. {
  163. result += (c.data.type->getAIValue() * c.data.count) * c.chance;
  164. }
  165. /*else
  166. {
  167. //we will need to discard the weakest stack
  168. result += (c.data.type->getAIValue() * c.data.count - weakestStackPower) * c.chance;
  169. }*/
  170. }
  171. result /= 100; //divide by total chance
  172. return result;
  173. }
  174. uint64_t getDwellingArmyValue(CCallback * cb, const CGObjectInstance * target, bool checkGold)
  175. {
  176. auto dwelling = dynamic_cast<const CGDwelling *>(target);
  177. uint64_t score = 0;
  178. for(auto & creLevel : dwelling->creatures)
  179. {
  180. if(creLevel.first && creLevel.second.size())
  181. {
  182. auto creature = creLevel.second.back().toCreature();
  183. auto creaturesAreFree = creature->getLevel() == 1;
  184. if(!creaturesAreFree && checkGold && !cb->getResourceAmount().canAfford(creature->getFullRecruitCost() * creLevel.first))
  185. continue;
  186. score += creature->getAIValue() * creLevel.first;
  187. }
  188. }
  189. return score;
  190. }
  191. uint64_t getDwellingArmyGrowth(CCallback * cb, const CGObjectInstance * target, PlayerColor myColor)
  192. {
  193. auto dwelling = dynamic_cast<const CGDwelling *>(target);
  194. uint64_t score = 0;
  195. if(dwelling->getOwner() == myColor)
  196. return 0;
  197. for(auto & creLevel : dwelling->creatures)
  198. {
  199. if(creLevel.second.size())
  200. {
  201. auto creature = creLevel.second.back().toCreature();
  202. score += creature->getAIValue() * creature->getGrowth();
  203. }
  204. }
  205. return score;
  206. }
  207. int getDwellingArmyCost(const CGObjectInstance * target)
  208. {
  209. auto dwelling = dynamic_cast<const CGDwelling *>(target);
  210. int cost = 0;
  211. for(auto & creLevel : dwelling->creatures)
  212. {
  213. if(creLevel.first && creLevel.second.size())
  214. {
  215. auto creature = creLevel.second.back().toCreature();
  216. auto creaturesAreFree = creature->getLevel() == 1;
  217. if(!creaturesAreFree)
  218. cost += creature->getFullRecruitCost().marketValue() * creLevel.first;
  219. }
  220. }
  221. return cost;
  222. }
  223. static uint64_t evaluateArtifactArmyValue(const CArtifact * art)
  224. {
  225. if(art->getId() == ArtifactID::SPELL_SCROLL)
  226. return 1500;
  227. auto statsValue =
  228. 10 * art->valOfBonuses(BonusType::MOVEMENT, BonusCustomSubtype::heroMovementLand)
  229. + 1200 * art->valOfBonuses(BonusType::STACKS_SPEED)
  230. + 700 * art->valOfBonuses(BonusType::MORALE)
  231. + 700 * art->valOfBonuses(BonusType::PRIMARY_SKILL, BonusSubtypeID(PrimarySkill::ATTACK))
  232. + 700 * art->valOfBonuses(BonusType::PRIMARY_SKILL, BonusSubtypeID(PrimarySkill::DEFENSE))
  233. + 700 * art->valOfBonuses(BonusType::PRIMARY_SKILL, BonusSubtypeID(PrimarySkill::KNOWLEDGE))
  234. + 700 * art->valOfBonuses(BonusType::PRIMARY_SKILL, BonusSubtypeID(PrimarySkill::SPELL_POWER))
  235. + 500 * art->valOfBonuses(BonusType::LUCK);
  236. auto classValue = 0;
  237. switch(art->aClass)
  238. {
  239. case CArtifact::EartClass::ART_MINOR:
  240. classValue = 1000;
  241. break;
  242. case CArtifact::EartClass::ART_MAJOR:
  243. classValue = 3000;
  244. break;
  245. case CArtifact::EartClass::ART_RELIC:
  246. case CArtifact::EartClass::ART_SPECIAL:
  247. classValue = 8000;
  248. break;
  249. }
  250. return statsValue > classValue ? statsValue : classValue;
  251. }
  252. uint64_t RewardEvaluator::getArmyReward(
  253. const CGObjectInstance * target,
  254. const CGHeroInstance * hero,
  255. const CCreatureSet * army,
  256. bool checkGold) const
  257. {
  258. const float enemyArmyEliminationRewardRatio = 0.5f;
  259. auto relations = ai->cb->getPlayerRelations(target->tempOwner, ai->playerID);
  260. if(!target)
  261. return 0;
  262. switch(target->ID)
  263. {
  264. case Obj::HILL_FORT:
  265. return ai->armyManager->calculateCreaturesUpgrade(army, target, ai->cb->getResourceAmount()).upgradeValue;
  266. case Obj::CREATURE_BANK:
  267. return getCreatureBankArmyReward(target, hero);
  268. case Obj::CREATURE_GENERATOR1:
  269. case Obj::CREATURE_GENERATOR2:
  270. case Obj::CREATURE_GENERATOR3:
  271. case Obj::CREATURE_GENERATOR4:
  272. return getDwellingArmyValue(ai->cb.get(), target, checkGold);
  273. case Obj::CRYPT:
  274. case Obj::SHIPWRECK:
  275. case Obj::SHIPWRECK_SURVIVOR:
  276. case Obj::WARRIORS_TOMB:
  277. return 1000;
  278. case Obj::ARTIFACT:
  279. return evaluateArtifactArmyValue(dynamic_cast<const CGArtifact *>(target)->storedArtifact->artType);
  280. case Obj::DRAGON_UTOPIA:
  281. return 10000;
  282. case Obj::HERO:
  283. return relations == PlayerRelations::ENEMIES
  284. ? enemyArmyEliminationRewardRatio * dynamic_cast<const CGHeroInstance *>(target)->getArmyStrength()
  285. : 0;
  286. case Obj::PANDORAS_BOX:
  287. return 5000;
  288. case Obj::MAGIC_WELL:
  289. case Obj::MAGIC_SPRING:
  290. return getManaRecoveryArmyReward(hero);
  291. default:
  292. break;
  293. }
  294. auto rewardable = dynamic_cast<const Rewardable::Interface *>(target);
  295. if(rewardable)
  296. {
  297. auto totalValue = 0;
  298. for(int index : rewardable->getAvailableRewards(hero, Rewardable::EEventType::EVENT_FIRST_VISIT))
  299. {
  300. auto & info = rewardable->configuration.info[index];
  301. auto rewardValue = 0;
  302. if(!info.reward.artifacts.empty())
  303. {
  304. for(auto artID : info.reward.artifacts)
  305. {
  306. const auto * art = dynamic_cast<const CArtifact *>(VLC->artifacts()->getById(artID));
  307. rewardValue += evaluateArtifactArmyValue(art);
  308. }
  309. }
  310. if(!info.reward.creatures.empty())
  311. {
  312. for(const auto & stackInfo : info.reward.creatures)
  313. {
  314. rewardValue += stackInfo.getType()->getAIValue() * stackInfo.getCount();
  315. }
  316. }
  317. totalValue += rewardValue > 0 ? rewardValue / (info.reward.artifacts.size() + info.reward.creatures.size()) : 0;
  318. }
  319. return totalValue;
  320. }
  321. return 0;
  322. }
  323. uint64_t RewardEvaluator::getArmyGrowth(
  324. const CGObjectInstance * target,
  325. const CGHeroInstance * hero,
  326. const CCreatureSet * army) const
  327. {
  328. if(!target)
  329. return 0;
  330. auto relations = ai->cb->getPlayerRelations(target->tempOwner, hero->tempOwner);
  331. if(relations != PlayerRelations::ENEMIES)
  332. return 0;
  333. switch(target->ID)
  334. {
  335. case Obj::TOWN:
  336. {
  337. auto town = dynamic_cast<const CGTownInstance *>(target);
  338. auto fortLevel = town->fortLevel();
  339. auto neutral = !town->getOwner().isValidPlayer();
  340. auto booster = isAnotherAi(town, *ai->cb) || neutral ? 1 : 2;
  341. if(fortLevel < CGTownInstance::CITADEL)
  342. return town->hasFort() ? booster * 500 : 0;
  343. else
  344. return booster * (fortLevel == CGTownInstance::CASTLE ? 5000 : 2000);
  345. }
  346. case Obj::CREATURE_GENERATOR1:
  347. case Obj::CREATURE_GENERATOR2:
  348. case Obj::CREATURE_GENERATOR3:
  349. case Obj::CREATURE_GENERATOR4:
  350. return getDwellingArmyGrowth(ai->cb.get(), target, hero->getOwner());
  351. case Obj::ARTIFACT:
  352. // it is not supported now because hero will not sit in town on 7th day but later parts of legion may be counted as army growth as well.
  353. return 0;
  354. default:
  355. return 0;
  356. }
  357. }
  358. int RewardEvaluator::getGoldCost(const CGObjectInstance * target, const CGHeroInstance * hero, const CCreatureSet * army) const
  359. {
  360. if(!target)
  361. return 0;
  362. if(auto * m = dynamic_cast<const IMarket *>(target))
  363. {
  364. if(m->allowsTrade(EMarketMode::RESOURCE_SKILL))
  365. return 2000;
  366. }
  367. switch(target->ID)
  368. {
  369. case Obj::HILL_FORT:
  370. return ai->armyManager->calculateCreaturesUpgrade(army, target, ai->cb->getResourceAmount()).upgradeCost[EGameResID::GOLD];
  371. case Obj::SCHOOL_OF_MAGIC:
  372. case Obj::SCHOOL_OF_WAR:
  373. return 1000;
  374. case Obj::CREATURE_GENERATOR1:
  375. case Obj::CREATURE_GENERATOR2:
  376. case Obj::CREATURE_GENERATOR3:
  377. case Obj::CREATURE_GENERATOR4:
  378. return getDwellingArmyCost(target);
  379. default:
  380. return 0;
  381. }
  382. }
  383. float RewardEvaluator::getEnemyHeroStrategicalValue(const CGHeroInstance * enemy) const
  384. {
  385. auto objectsUnderTreat = ai->dangerHitMap->getOneTurnAccessibleObjects(enemy);
  386. float objectValue = 0;
  387. for(auto obj : objectsUnderTreat)
  388. {
  389. vstd::amax(objectValue, getStrategicalValue(obj));
  390. }
  391. /*
  392. 1. If an enemy hero can attack nearby object, it's not useful to capture the object on our own.
  393. Killing the hero is almost as important (0.9) as capturing the object itself.
  394. 2. The formula quickly approaches 1.0 as hero level increases,
  395. but higher level always means higher value and the minimal value for level 1 hero is 0.5
  396. */
  397. return std::min(1.5f, objectValue * 0.9f + (1.5f - (1.5f / (1 + enemy->level))));
  398. }
  399. float RewardEvaluator::getResourceRequirementStrength(int resType) const
  400. {
  401. TResources requiredResources = ai->buildAnalyzer->getResourcesRequiredNow();
  402. TResources dailyIncome = ai->buildAnalyzer->getDailyIncome();
  403. if(requiredResources[resType] == 0)
  404. return 0;
  405. if(dailyIncome[resType] == 0)
  406. return 1.0f;
  407. float ratio = (float)requiredResources[resType] / dailyIncome[resType] / 2;
  408. return std::min(ratio, 1.0f);
  409. }
  410. float RewardEvaluator::getTotalResourceRequirementStrength(int resType) const
  411. {
  412. TResources requiredResources = ai->buildAnalyzer->getTotalResourcesRequired();
  413. TResources dailyIncome = ai->buildAnalyzer->getDailyIncome();
  414. if(requiredResources[resType] == 0)
  415. return 0;
  416. float ratio = dailyIncome[resType] == 0
  417. ? (float)requiredResources[resType] / 10.0f
  418. : (float)requiredResources[resType] / dailyIncome[resType] / 20.0f;
  419. return std::min(ratio, 2.0f);
  420. }
  421. uint64_t RewardEvaluator::townArmyGrowth(const CGTownInstance * town) const
  422. {
  423. uint64_t result = 0;
  424. for(auto creatureInfo : town->creatures)
  425. {
  426. if(creatureInfo.second.empty())
  427. continue;
  428. auto creature = creatureInfo.second.back().toCreature();
  429. result += creature->getAIValue() * town->getGrowthInfo(creature->getLevel() - 1).totalGrowth();
  430. }
  431. return result;
  432. }
  433. uint64_t RewardEvaluator::getManaRecoveryArmyReward(const CGHeroInstance * hero) const
  434. {
  435. return ai->heroManager->getMagicStrength(hero) * 10000 * (1.0f - std::sqrt(static_cast<float>(hero->mana) / hero->manaLimit()));
  436. }
  437. float RewardEvaluator::getResourceRequirementStrength(const TResources & res) const
  438. {
  439. float sum = 0.0f;
  440. for(TResources::nziterator it(res); it.valid(); it++)
  441. {
  442. //Evaluate resources used for construction. Gold is evaluated separately.
  443. if(it->resType != EGameResID::GOLD)
  444. {
  445. sum += 0.1f * it->resVal * getResourceRequirementStrength(it->resType)
  446. + 0.05f * it->resVal * getTotalResourceRequirementStrength(it->resType);
  447. }
  448. }
  449. return sum;
  450. }
  451. float RewardEvaluator::getStrategicalValue(const CGObjectInstance * target, const CGHeroInstance * hero) const
  452. {
  453. if(!target)
  454. return 0;
  455. switch(target->ID)
  456. {
  457. case Obj::MINE:
  458. {
  459. auto mine = dynamic_cast<const CGMine *>(target);
  460. return mine->producedResource == EGameResID::GOLD
  461. ? 0.5f
  462. : 0.4f * getTotalResourceRequirementStrength(mine->producedResource) + 0.1f * getResourceRequirementStrength(mine->producedResource);
  463. }
  464. case Obj::RESOURCE:
  465. {
  466. auto resource = dynamic_cast<const CGResource *>(target);
  467. TResources res;
  468. res[resource->resourceID()] = resource->amount;
  469. return getResourceRequirementStrength(res);
  470. }
  471. case Obj::CREATURE_BANK:
  472. {
  473. auto resourceReward = getCreatureBankResources(target, nullptr);
  474. return getResourceRequirementStrength(resourceReward);
  475. }
  476. case Obj::TOWN:
  477. {
  478. if(ai->buildAnalyzer->getDevelopmentInfo().empty())
  479. return 10.0f;
  480. auto town = dynamic_cast<const CGTownInstance *>(target);
  481. if(town->getOwner() == ai->playerID)
  482. {
  483. auto armyIncome = townArmyGrowth(town);
  484. auto dailyIncome = town->dailyIncome()[EGameResID::GOLD];
  485. return std::min(1.0f, std::sqrt(armyIncome / 40000.0f)) + std::min(0.3f, dailyIncome / 10000.0f);
  486. }
  487. auto fortLevel = town->fortLevel();
  488. auto booster = isAnotherAi(town, *ai->cb) ? 0.4f : 1.0f;
  489. if(town->hasCapitol())
  490. return booster * 1.5;
  491. if(fortLevel < CGTownInstance::CITADEL)
  492. return booster * (town->hasFort() ? 1.0 : 0.8);
  493. else
  494. return booster * (fortLevel == CGTownInstance::CASTLE ? 1.4 : 1.2);
  495. }
  496. case Obj::HERO:
  497. return ai->cb->getPlayerRelations(target->tempOwner, ai->playerID) == PlayerRelations::ENEMIES
  498. ? getEnemyHeroStrategicalValue(dynamic_cast<const CGHeroInstance *>(target))
  499. : 0;
  500. case Obj::KEYMASTER:
  501. return 0.6f;
  502. default:
  503. break;
  504. }
  505. auto rewardable = dynamic_cast<const Rewardable::Interface *>(target);
  506. if(rewardable && hero)
  507. {
  508. auto resourceReward = 0.0f;
  509. for(int index : rewardable->getAvailableRewards(hero, Rewardable::EEventType::EVENT_FIRST_VISIT))
  510. {
  511. resourceReward += getResourceRequirementStrength(rewardable->configuration.info[index].reward.resources);
  512. }
  513. return resourceReward;
  514. }
  515. return 0;
  516. }
  517. float RewardEvaluator::getConquestValue(const CGObjectInstance* target) const
  518. {
  519. if (!target)
  520. return 0;
  521. if (target->getOwner() == ai->playerID)
  522. return 0;
  523. switch (target->ID)
  524. {
  525. case Obj::TOWN:
  526. {
  527. if (ai->buildAnalyzer->getDevelopmentInfo().empty())
  528. return 10.0f;
  529. auto town = dynamic_cast<const CGTownInstance*>(target);
  530. if (town->getOwner() == ai->playerID)
  531. {
  532. auto armyIncome = townArmyGrowth(town);
  533. auto dailyIncome = town->dailyIncome()[EGameResID::GOLD];
  534. return std::min(1.0f, std::sqrt(armyIncome / 40000.0f)) + std::min(0.3f, dailyIncome / 10000.0f);
  535. }
  536. auto fortLevel = town->fortLevel();
  537. auto booster = 1.0f;
  538. if (town->hasCapitol())
  539. return booster * 1.5;
  540. if (fortLevel < CGTownInstance::CITADEL)
  541. return booster * (town->hasFort() ? 1.0 : 0.8);
  542. else
  543. return booster * (fortLevel == CGTownInstance::CASTLE ? 1.4 : 1.2);
  544. }
  545. case Obj::HERO:
  546. return ai->cb->getPlayerRelations(target->tempOwner, ai->playerID) == PlayerRelations::ENEMIES
  547. ? getEnemyHeroStrategicalValue(dynamic_cast<const CGHeroInstance*>(target))
  548. : 0;
  549. case Obj::KEYMASTER:
  550. return 0.6f;
  551. default:
  552. return 0;
  553. }
  554. }
  555. float RewardEvaluator::evaluateWitchHutSkillScore(const CGObjectInstance * hut, const CGHeroInstance * hero, HeroRole role) const
  556. {
  557. auto rewardable = dynamic_cast<const CRewardableObject *>(hut);
  558. assert(rewardable);
  559. auto skill = SecondarySkill(*rewardable->configuration.getVariable("secondarySkill", "gainedSkill"));
  560. if(!hut->wasVisited(hero->tempOwner))
  561. return role == HeroRole::SCOUT ? 2 : 0;
  562. if(hero->getSecSkillLevel(skill) != MasteryLevel::NONE
  563. || hero->secSkills.size() >= GameConstants::SKILL_PER_HERO)
  564. return 0;
  565. auto score = ai->heroManager->evaluateSecSkill(skill, hero);
  566. return score >= 2 ? (role == HeroRole::MAIN ? 10 : 4) : score;
  567. }
  568. float RewardEvaluator::getSkillReward(const CGObjectInstance * target, const CGHeroInstance * hero, HeroRole role) const
  569. {
  570. const float enemyHeroEliminationSkillRewardRatio = 0.5f;
  571. if(!target)
  572. return 0;
  573. switch(target->ID)
  574. {
  575. case Obj::STAR_AXIS:
  576. case Obj::SCHOLAR:
  577. case Obj::SCHOOL_OF_MAGIC:
  578. case Obj::SCHOOL_OF_WAR:
  579. case Obj::GARDEN_OF_REVELATION:
  580. case Obj::MARLETTO_TOWER:
  581. case Obj::MERCENARY_CAMP:
  582. case Obj::TREE_OF_KNOWLEDGE:
  583. return 1;
  584. case Obj::LEARNING_STONE:
  585. return 1.0f / std::sqrt(hero->level);
  586. case Obj::ARENA:
  587. return 2;
  588. case Obj::SHRINE_OF_MAGIC_INCANTATION:
  589. return 0.25f;
  590. case Obj::SHRINE_OF_MAGIC_GESTURE:
  591. return 1.0f;
  592. case Obj::SHRINE_OF_MAGIC_THOUGHT:
  593. return 2.0f;
  594. case Obj::LIBRARY_OF_ENLIGHTENMENT:
  595. return 8;
  596. case Obj::WITCH_HUT:
  597. return evaluateWitchHutSkillScore(target, hero, role);
  598. case Obj::PANDORAS_BOX:
  599. //Can contains experience, spells, or skills (only on custom maps)
  600. return 2.5f;
  601. case Obj::PYRAMID:
  602. return 6.0f;
  603. case Obj::HERO:
  604. return ai->cb->getPlayerRelations(target->tempOwner, ai->playerID) == PlayerRelations::ENEMIES
  605. ? enemyHeroEliminationSkillRewardRatio * dynamic_cast<const CGHeroInstance *>(target)->level
  606. : 0;
  607. default:
  608. break;
  609. }
  610. auto rewardable = dynamic_cast<const Rewardable::Interface *>(target);
  611. if(rewardable)
  612. {
  613. auto totalValue = 0.0f;
  614. for(int index : rewardable->getAvailableRewards(hero, Rewardable::EEventType::EVENT_FIRST_VISIT))
  615. {
  616. auto & info = rewardable->configuration.info[index];
  617. auto rewardValue = 0.0f;
  618. if(!info.reward.spells.empty())
  619. {
  620. for(auto spellID : info.reward.spells)
  621. {
  622. const spells::Spell * spell = VLC->spells()->getById(spellID);
  623. if(hero->canLearnSpell(spell) && !hero->spellbookContainsSpell(spellID))
  624. {
  625. rewardValue += std::sqrt(spell->getLevel()) / 4.0f;
  626. }
  627. }
  628. totalValue += rewardValue / info.reward.spells.size();
  629. }
  630. if(!info.reward.primary.empty())
  631. {
  632. for(auto value : info.reward.primary)
  633. {
  634. totalValue += value;
  635. }
  636. }
  637. }
  638. return totalValue;
  639. }
  640. return 0;
  641. }
  642. const HitMapInfo & RewardEvaluator::getEnemyHeroDanger(const int3 & tile, uint8_t turn) const
  643. {
  644. auto & treatNode = ai->dangerHitMap->getTileThreat(tile);
  645. if(treatNode.maximumDanger.danger == 0)
  646. return HitMapInfo::NoThreat;
  647. if(treatNode.maximumDanger.turn <= turn)
  648. return treatNode.maximumDanger;
  649. return treatNode.fastestDanger.turn <= turn ? treatNode.fastestDanger : HitMapInfo::NoThreat;
  650. }
  651. int32_t getArmyCost(const CArmedInstance * army)
  652. {
  653. int32_t value = 0;
  654. for(auto stack : army->Slots())
  655. {
  656. value += stack.second->getCreatureID().toCreature()->getFullRecruitCost().marketValue() * stack.second->count;
  657. }
  658. return value;
  659. }
  660. int32_t RewardEvaluator::getGoldReward(const CGObjectInstance * target, const CGHeroInstance * hero) const
  661. {
  662. if(!target)
  663. return 0;
  664. auto relations = ai->cb->getPlayerRelations(target->tempOwner, hero->tempOwner);
  665. const int dailyIncomeMultiplier = 5;
  666. const float enemyArmyEliminationGoldRewardRatio = 0.2f;
  667. const int32_t heroEliminationBonus = GameConstants::HERO_GOLD_COST / 2;
  668. switch(target->ID)
  669. {
  670. case Obj::RESOURCE:
  671. {
  672. auto * res = dynamic_cast<const CGResource*>(target);
  673. return res && res->resourceID() == GameResID::GOLD ? 600 : 100;
  674. }
  675. case Obj::TREASURE_CHEST:
  676. return 1500;
  677. case Obj::WATER_WHEEL:
  678. return 1000;
  679. case Obj::TOWN:
  680. return dailyIncomeMultiplier * estimateTownIncome(ai->cb.get(), target, hero);
  681. case Obj::MINE:
  682. case Obj::ABANDONED_MINE:
  683. {
  684. auto * mine = dynamic_cast<const CGMine*>(target);
  685. return dailyIncomeMultiplier * (mine->producedResource == GameResID::GOLD ? 1000 : 75);
  686. }
  687. case Obj::MYSTICAL_GARDEN:
  688. case Obj::WINDMILL:
  689. return 100;
  690. case Obj::CAMPFIRE:
  691. return 800;
  692. case Obj::WAGON:
  693. return 100;
  694. case Obj::CREATURE_BANK:
  695. return getResourcesGoldReward(getCreatureBankResources(target, hero));
  696. case Obj::CRYPT:
  697. case Obj::DERELICT_SHIP:
  698. return 3000;
  699. case Obj::DRAGON_UTOPIA:
  700. return 10000;
  701. case Obj::SEA_CHEST:
  702. return 1500;
  703. case Obj::PANDORAS_BOX:
  704. return 2500;
  705. case Obj::PRISON:
  706. //Objectively saves us 2500 to hire hero
  707. return GameConstants::HERO_GOLD_COST;
  708. case Obj::HERO:
  709. return relations == PlayerRelations::ENEMIES
  710. ? heroEliminationBonus + enemyArmyEliminationGoldRewardRatio * getArmyCost(dynamic_cast<const CGHeroInstance *>(target))
  711. : 0;
  712. default:
  713. break;
  714. }
  715. auto rewardable = dynamic_cast<const Rewardable::Interface *>(target);
  716. if(rewardable)
  717. {
  718. auto goldReward = 0;
  719. for(int index : rewardable->getAvailableRewards(hero, Rewardable::EEventType::EVENT_FIRST_VISIT))
  720. {
  721. auto & info = rewardable->configuration.info[index];
  722. goldReward += getResourcesGoldReward(info.reward.resources);
  723. }
  724. return goldReward;
  725. }
  726. return 0;
  727. }
  728. class HeroExchangeEvaluator : public IEvaluationContextBuilder
  729. {
  730. public:
  731. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  732. {
  733. if(task->goalType != Goals::HERO_EXCHANGE)
  734. return;
  735. Goals::HeroExchange & heroExchange = dynamic_cast<Goals::HeroExchange &>(*task);
  736. uint64_t armyStrength = heroExchange.getReinforcementArmyStrength(evaluationContext.evaluator.ai);
  737. evaluationContext.addNonCriticalStrategicalValue(2.0f * armyStrength / (float)heroExchange.hero->getArmyStrength());
  738. evaluationContext.conquestValue += 2.0f * armyStrength / (float)heroExchange.hero->getArmyStrength();
  739. evaluationContext.heroRole = evaluationContext.evaluator.ai->heroManager->getHeroRole(heroExchange.hero);
  740. evaluationContext.isExchange = true;
  741. }
  742. };
  743. class ArmyUpgradeEvaluator : public IEvaluationContextBuilder
  744. {
  745. public:
  746. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  747. {
  748. if(task->goalType != Goals::ARMY_UPGRADE)
  749. return;
  750. Goals::ArmyUpgrade & armyUpgrade = dynamic_cast<Goals::ArmyUpgrade &>(*task);
  751. uint64_t upgradeValue = armyUpgrade.getUpgradeValue();
  752. evaluationContext.armyReward += upgradeValue;
  753. evaluationContext.addNonCriticalStrategicalValue(upgradeValue / (float)armyUpgrade.hero->getArmyStrength());
  754. }
  755. };
  756. class ExplorePointEvaluator : public IEvaluationContextBuilder
  757. {
  758. public:
  759. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  760. {
  761. if(task->goalType != Goals::EXPLORATION_POINT)
  762. return;
  763. int tilesDiscovered = task->value;
  764. evaluationContext.addNonCriticalStrategicalValue(0.03f * tilesDiscovered);
  765. }
  766. };
  767. class StayAtTownManaRecoveryEvaluator : public IEvaluationContextBuilder
  768. {
  769. public:
  770. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  771. {
  772. if(task->goalType != Goals::STAY_AT_TOWN)
  773. return;
  774. Goals::StayAtTown & stayAtTown = dynamic_cast<Goals::StayAtTown &>(*task);
  775. evaluationContext.armyReward += evaluationContext.evaluator.getManaRecoveryArmyReward(stayAtTown.getHero());
  776. evaluationContext.movementCostByRole[evaluationContext.heroRole] += stayAtTown.getMovementWasted();
  777. evaluationContext.movementCost += stayAtTown.getMovementWasted();
  778. }
  779. };
  780. void addTileDanger(EvaluationContext & evaluationContext, const int3 & tile, uint8_t turn, uint64_t ourStrength)
  781. {
  782. HitMapInfo enemyDanger = evaluationContext.evaluator.getEnemyHeroDanger(tile, turn);
  783. if(enemyDanger.danger)
  784. {
  785. auto dangerRatio = enemyDanger.danger / (double)ourStrength;
  786. vstd::amax(evaluationContext.enemyHeroDangerRatio, dangerRatio);
  787. vstd::amax(evaluationContext.threat, enemyDanger.threat);
  788. }
  789. }
  790. class DefendTownEvaluator : public IEvaluationContextBuilder
  791. {
  792. public:
  793. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  794. {
  795. if(task->goalType != Goals::DEFEND_TOWN)
  796. return;
  797. Goals::DefendTown & defendTown = dynamic_cast<Goals::DefendTown &>(*task);
  798. const CGTownInstance * town = defendTown.town;
  799. auto & treat = defendTown.getTreat();
  800. auto strategicalValue = evaluationContext.evaluator.getStrategicalValue(town);
  801. float multiplier = 1;
  802. if(treat.turn < defendTown.getTurn())
  803. multiplier /= 1 + (defendTown.getTurn() - treat.turn);
  804. multiplier /= 1.0f + treat.turn / 5.0f;
  805. if(defendTown.getTurn() > 0 && defendTown.isCounterAttack())
  806. {
  807. auto ourSpeed = defendTown.hero->movementPointsLimit(true);
  808. auto enemySpeed = treat.hero.get(evaluationContext.evaluator.ai->cb.get())->movementPointsLimit(true);
  809. if(enemySpeed > ourSpeed) multiplier *= 0.7f;
  810. }
  811. auto dailyIncome = town->dailyIncome()[EGameResID::GOLD];
  812. auto armyGrowth = evaluationContext.evaluator.townArmyGrowth(town);
  813. evaluationContext.armyGrowth += armyGrowth * multiplier;
  814. evaluationContext.goldReward += dailyIncome * 5 * multiplier;
  815. if(evaluationContext.evaluator.ai->buildAnalyzer->getDevelopmentInfo().size() == 1)
  816. vstd::amax(evaluationContext.strategicalValue, 2.5f * multiplier * strategicalValue);
  817. else
  818. evaluationContext.addNonCriticalStrategicalValue(1.7f * multiplier * strategicalValue);
  819. evaluationContext.defenseValue = town->fortLevel();
  820. evaluationContext.isDefend = true;
  821. evaluationContext.threatTurns = treat.turn;
  822. vstd::amax(evaluationContext.danger, defendTown.getTreat().danger);
  823. addTileDanger(evaluationContext, town->visitablePos(), defendTown.getTurn(), defendTown.getDefenceStrength());
  824. }
  825. };
  826. class ExecuteHeroChainEvaluationContextBuilder : public IEvaluationContextBuilder
  827. {
  828. private:
  829. const Nullkiller * ai;
  830. public:
  831. ExecuteHeroChainEvaluationContextBuilder(const Nullkiller * ai) : ai(ai) {}
  832. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  833. {
  834. if(task->goalType != Goals::EXECUTE_HERO_CHAIN)
  835. return;
  836. Goals::ExecuteHeroChain & chain = dynamic_cast<Goals::ExecuteHeroChain &>(*task);
  837. const AIPath & path = chain.getPath();
  838. vstd::amax(evaluationContext.danger, path.getTotalDanger());
  839. evaluationContext.movementCost += path.movementCost();
  840. evaluationContext.closestWayRatio = chain.closestWayRatio;
  841. std::map<const CGHeroInstance *, float> costsPerHero;
  842. for(auto & node : path.nodes)
  843. {
  844. vstd::amax(costsPerHero[node.targetHero], node.cost);
  845. if (node.layer == EPathfindingLayer::SAIL)
  846. evaluationContext.involvesSailing = true;
  847. }
  848. for(auto pair : costsPerHero)
  849. {
  850. auto role = evaluationContext.evaluator.ai->heroManager->getHeroRole(pair.first);
  851. evaluationContext.movementCostByRole[role] += pair.second;
  852. }
  853. auto hero = task->hero;
  854. bool checkGold = evaluationContext.danger == 0;
  855. auto army = path.heroArmy;
  856. const CGObjectInstance * target = ai->cb->getObj((ObjectInstanceID)task->objid, false);
  857. auto heroRole = evaluationContext.evaluator.ai->heroManager->getHeroRole(hero);
  858. if(heroRole == HeroRole::MAIN)
  859. evaluationContext.heroRole = heroRole;
  860. if (target)
  861. {
  862. evaluationContext.goldReward += evaluationContext.evaluator.getGoldReward(target, hero);
  863. evaluationContext.armyReward += evaluationContext.evaluator.getArmyReward(target, hero, army, checkGold);
  864. evaluationContext.armyGrowth += evaluationContext.evaluator.getArmyGrowth(target, hero, army);
  865. evaluationContext.skillReward += evaluationContext.evaluator.getSkillReward(target, hero, heroRole);
  866. evaluationContext.addNonCriticalStrategicalValue(evaluationContext.evaluator.getStrategicalValue(target));
  867. evaluationContext.conquestValue += evaluationContext.evaluator.getConquestValue(target);
  868. evaluationContext.goldCost += evaluationContext.evaluator.getGoldCost(target, hero, army);
  869. evaluationContext.armyInvolvement += army->getArmyCost();
  870. }
  871. vstd::amax(evaluationContext.armyLossPersentage, path.getTotalArmyLoss() / (double)path.getHeroStrength());
  872. addTileDanger(evaluationContext, path.targetTile(), path.turn(), path.getHeroStrength());
  873. vstd::amax(evaluationContext.turn, path.turn());
  874. }
  875. };
  876. class ClusterEvaluationContextBuilder : public IEvaluationContextBuilder
  877. {
  878. public:
  879. ClusterEvaluationContextBuilder(const Nullkiller * ai) {}
  880. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  881. {
  882. if(task->goalType != Goals::UNLOCK_CLUSTER)
  883. return;
  884. Goals::UnlockCluster & clusterGoal = dynamic_cast<Goals::UnlockCluster &>(*task);
  885. std::shared_ptr<ObjectCluster> cluster = clusterGoal.getCluster();
  886. auto hero = clusterGoal.hero;
  887. auto role = evaluationContext.evaluator.ai->heroManager->getHeroRole(hero);
  888. std::vector<std::pair<ObjectInstanceID, ClusterObjectInfo>> objects(cluster->objects.begin(), cluster->objects.end());
  889. std::sort(objects.begin(), objects.end(), [](std::pair<ObjectInstanceID, ClusterObjectInfo> o1, std::pair<ObjectInstanceID, ClusterObjectInfo> o2) -> bool
  890. {
  891. return o1.second.priority > o2.second.priority;
  892. });
  893. int boost = 1;
  894. for(auto & objInfo : objects)
  895. {
  896. auto target = evaluationContext.evaluator.ai->cb->getObj(objInfo.first);
  897. bool checkGold = objInfo.second.danger == 0;
  898. auto army = hero;
  899. evaluationContext.goldReward += evaluationContext.evaluator.getGoldReward(target, hero) / boost;
  900. evaluationContext.armyReward += evaluationContext.evaluator.getArmyReward(target, hero, army, checkGold) / boost;
  901. evaluationContext.skillReward += evaluationContext.evaluator.getSkillReward(target, hero, role) / boost;
  902. evaluationContext.addNonCriticalStrategicalValue(evaluationContext.evaluator.getStrategicalValue(target) / boost);
  903. evaluationContext.conquestValue += evaluationContext.evaluator.getConquestValue(target);
  904. evaluationContext.goldCost += evaluationContext.evaluator.getGoldCost(target, hero, army) / boost;
  905. evaluationContext.movementCostByRole[role] += objInfo.second.movementCost / boost;
  906. evaluationContext.movementCost += objInfo.second.movementCost / boost;
  907. vstd::amax(evaluationContext.turn, objInfo.second.turn / boost);
  908. boost <<= 1;
  909. if(boost > 8)
  910. break;
  911. }
  912. }
  913. };
  914. class ExchangeSwapTownHeroesContextBuilder : public IEvaluationContextBuilder
  915. {
  916. public:
  917. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  918. {
  919. if(task->goalType != Goals::EXCHANGE_SWAP_TOWN_HEROES)
  920. return;
  921. Goals::ExchangeSwapTownHeroes & swapCommand = dynamic_cast<Goals::ExchangeSwapTownHeroes &>(*task);
  922. const CGHeroInstance * garrisonHero = swapCommand.getGarrisonHero();
  923. logAi->trace("buildEvaluationContext ExchangeSwapTownHeroesContextBuilder %s affected objects: %d", swapCommand.toString(), swapCommand.getAffectedObjects().size());
  924. for (auto obj : swapCommand.getAffectedObjects())
  925. {
  926. logAi->trace("affected object: %s", evaluationContext.evaluator.ai->cb->getObj(obj)->getObjectName());
  927. }
  928. if (garrisonHero)
  929. logAi->debug("with %s and %d", garrisonHero->getNameTranslated(), int(swapCommand.getLockingReason()));
  930. if(garrisonHero && swapCommand.getLockingReason() == HeroLockedReason::DEFENCE)
  931. {
  932. auto defenderRole = evaluationContext.evaluator.ai->heroManager->getHeroRole(garrisonHero);
  933. auto mpLeft = garrisonHero->movementPointsRemaining() / (float)garrisonHero->movementPointsLimit(true);
  934. evaluationContext.movementCost += mpLeft;
  935. evaluationContext.movementCostByRole[defenderRole] += mpLeft;
  936. evaluationContext.heroRole = defenderRole;
  937. evaluationContext.isDefend = true;
  938. evaluationContext.armyInvolvement = garrisonHero->getArmyStrength();
  939. logAi->debug("evaluationContext.isDefend: %d", evaluationContext.isDefend);
  940. }
  941. }
  942. };
  943. class DismissHeroContextBuilder : public IEvaluationContextBuilder
  944. {
  945. private:
  946. const Nullkiller * ai;
  947. public:
  948. DismissHeroContextBuilder(const Nullkiller * ai) : ai(ai) {}
  949. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  950. {
  951. if(task->goalType != Goals::DISMISS_HERO)
  952. return;
  953. Goals::DismissHero & dismissCommand = dynamic_cast<Goals::DismissHero &>(*task);
  954. const CGHeroInstance * dismissedHero = dismissCommand.getHero();
  955. auto role = ai->heroManager->getHeroRole(dismissedHero);
  956. auto mpLeft = dismissedHero->movementPointsRemaining();
  957. evaluationContext.movementCost += mpLeft;
  958. evaluationContext.movementCostByRole[role] += mpLeft;
  959. evaluationContext.goldCost += GameConstants::HERO_GOLD_COST + getArmyCost(dismissedHero);
  960. }
  961. };
  962. class BuildThisEvaluationContextBuilder : public IEvaluationContextBuilder
  963. {
  964. public:
  965. void buildEvaluationContext(EvaluationContext & evaluationContext, Goals::TSubgoal task) const override
  966. {
  967. if(task->goalType != Goals::BUILD_STRUCTURE)
  968. return;
  969. Goals::BuildThis & buildThis = dynamic_cast<Goals::BuildThis &>(*task);
  970. auto & bi = buildThis.buildingInfo;
  971. evaluationContext.goldReward += 7 * bi.dailyIncome[EGameResID::GOLD] / 2; // 7 day income but half we already have
  972. evaluationContext.heroRole = HeroRole::MAIN;
  973. evaluationContext.movementCostByRole[evaluationContext.heroRole] += bi.prerequisitesCount;
  974. int32_t cost = bi.buildCost[EGameResID::GOLD];
  975. evaluationContext.goldCost += cost;
  976. evaluationContext.closestWayRatio = 1;
  977. evaluationContext.buildingCost += bi.buildCostWithPrerequisites;
  978. if (bi.id == BuildingID::MARKETPLACE || bi.dailyIncome[EGameResID::WOOD] > 0)
  979. evaluationContext.isTradeBuilding = true;
  980. logAi->trace("Building costs for %s : %s MarketValue: %d",bi.toString(), evaluationContext.buildingCost.toString(), evaluationContext.buildingCost.marketValue());
  981. if(bi.creatureID != CreatureID::NONE)
  982. {
  983. evaluationContext.addNonCriticalStrategicalValue(buildThis.townInfo.armyStrength / 50000.0);
  984. if(bi.baseCreatureID == bi.creatureID)
  985. {
  986. evaluationContext.addNonCriticalStrategicalValue((0.5f + 0.1f * bi.creatureLevel) / (float)bi.prerequisitesCount);
  987. evaluationContext.armyReward += bi.armyStrength;
  988. }
  989. else
  990. {
  991. auto potentialUpgradeValue = evaluationContext.evaluator.getUpgradeArmyReward(buildThis.town, bi);
  992. evaluationContext.addNonCriticalStrategicalValue(potentialUpgradeValue / 10000.0f / (float)bi.prerequisitesCount);
  993. evaluationContext.armyReward += potentialUpgradeValue / (float)bi.prerequisitesCount;
  994. }
  995. }
  996. else if(bi.id == BuildingID::CITADEL || bi.id == BuildingID::CASTLE)
  997. {
  998. evaluationContext.addNonCriticalStrategicalValue(buildThis.town->creatures.size() * 0.2f);
  999. evaluationContext.armyReward += buildThis.townInfo.armyStrength / 2;
  1000. }
  1001. else if(bi.id >= BuildingID::MAGES_GUILD_1 && bi.id <= BuildingID::MAGES_GUILD_5)
  1002. {
  1003. evaluationContext.skillReward += 2 * (bi.id - BuildingID::MAGES_GUILD_1);
  1004. for (auto hero : evaluationContext.evaluator.ai->cb->getHeroesInfo())
  1005. {
  1006. evaluationContext.armyInvolvement += hero->getArmyCost();
  1007. }
  1008. }
  1009. int sameTownBonus = 0;
  1010. for (auto town : evaluationContext.evaluator.ai->cb->getTownsInfo())
  1011. {
  1012. if (buildThis.town->getFaction() == town->getFaction())
  1013. sameTownBonus += town->getTownLevel();
  1014. }
  1015. evaluationContext.armyReward *= sameTownBonus;
  1016. if(evaluationContext.goldReward)
  1017. {
  1018. auto goldPressure = evaluationContext.evaluator.ai->buildAnalyzer->getGoldPressure();
  1019. evaluationContext.addNonCriticalStrategicalValue(evaluationContext.goldReward * goldPressure / 3500.0f / bi.prerequisitesCount);
  1020. }
  1021. if(bi.notEnoughRes && bi.prerequisitesCount == 1)
  1022. {
  1023. evaluationContext.strategicalValue /= 3;
  1024. evaluationContext.movementCostByRole[evaluationContext.heroRole] += 5;
  1025. evaluationContext.turn += 5;
  1026. }
  1027. }
  1028. };
  1029. uint64_t RewardEvaluator::getUpgradeArmyReward(const CGTownInstance * town, const BuildingInfo & bi) const
  1030. {
  1031. if(ai->buildAnalyzer->hasAnyBuilding(town->getFaction(), bi.id))
  1032. return 0;
  1033. auto creaturesToUpgrade = ai->armyManager->getTotalCreaturesAvailable(bi.baseCreatureID);
  1034. auto upgradedPower = ai->armyManager->evaluateStackPower(bi.creatureID.toCreature(), creaturesToUpgrade.count);
  1035. return upgradedPower - creaturesToUpgrade.power;
  1036. }
  1037. PriorityEvaluator::PriorityEvaluator(const Nullkiller * ai)
  1038. :ai(ai)
  1039. {
  1040. initVisitTile();
  1041. evaluationContextBuilders.push_back(std::make_shared<ExecuteHeroChainEvaluationContextBuilder>(ai));
  1042. evaluationContextBuilders.push_back(std::make_shared<BuildThisEvaluationContextBuilder>());
  1043. evaluationContextBuilders.push_back(std::make_shared<ClusterEvaluationContextBuilder>(ai));
  1044. evaluationContextBuilders.push_back(std::make_shared<HeroExchangeEvaluator>());
  1045. evaluationContextBuilders.push_back(std::make_shared<ArmyUpgradeEvaluator>());
  1046. evaluationContextBuilders.push_back(std::make_shared<DefendTownEvaluator>());
  1047. evaluationContextBuilders.push_back(std::make_shared<ExchangeSwapTownHeroesContextBuilder>());
  1048. evaluationContextBuilders.push_back(std::make_shared<DismissHeroContextBuilder>(ai));
  1049. evaluationContextBuilders.push_back(std::make_shared<StayAtTownManaRecoveryEvaluator>());
  1050. evaluationContextBuilders.push_back(std::make_shared<ExplorePointEvaluator>());
  1051. }
  1052. EvaluationContext PriorityEvaluator::buildEvaluationContext(Goals::TSubgoal goal) const
  1053. {
  1054. Goals::TGoalVec parts;
  1055. EvaluationContext context(ai);
  1056. if(goal->goalType == Goals::COMPOSITION)
  1057. {
  1058. parts = goal->decompose(ai);
  1059. }
  1060. else
  1061. {
  1062. parts.push_back(goal);
  1063. }
  1064. for(auto subgoal : parts)
  1065. {
  1066. context.goldCost += subgoal->goldCost;
  1067. context.buildingCost += subgoal->buildingCost;
  1068. for(auto builder : evaluationContextBuilders)
  1069. {
  1070. builder->buildEvaluationContext(context, subgoal);
  1071. }
  1072. }
  1073. return context;
  1074. }
  1075. float PriorityEvaluator::evaluate(Goals::TSubgoal task, int priorityTier)
  1076. {
  1077. auto evaluationContext = buildEvaluationContext(task);
  1078. int rewardType = (evaluationContext.goldReward > 0 ? 1 : 0)
  1079. + (evaluationContext.armyReward > 0 ? 1 : 0)
  1080. + (evaluationContext.skillReward > 0 ? 1 : 0)
  1081. + (evaluationContext.strategicalValue > 0 ? 1 : 0);
  1082. float goldRewardPerTurn = evaluationContext.goldReward / std::log2f(2 + evaluationContext.movementCost * 10);
  1083. double result = 0;
  1084. if (ai->settings->isUseFuzzy())
  1085. {
  1086. float fuzzyResult = 0;
  1087. try
  1088. {
  1089. armyLossPersentageVariable->setValue(evaluationContext.armyLossPersentage);
  1090. heroRoleVariable->setValue(evaluationContext.heroRole);
  1091. mainTurnDistanceVariable->setValue(evaluationContext.movementCostByRole[HeroRole::MAIN]);
  1092. scoutTurnDistanceVariable->setValue(evaluationContext.movementCostByRole[HeroRole::SCOUT]);
  1093. goldRewardVariable->setValue(goldRewardPerTurn);
  1094. armyRewardVariable->setValue(evaluationContext.armyReward);
  1095. armyGrowthVariable->setValue(evaluationContext.armyGrowth);
  1096. skillRewardVariable->setValue(evaluationContext.skillReward);
  1097. dangerVariable->setValue(evaluationContext.danger);
  1098. rewardTypeVariable->setValue(rewardType);
  1099. closestHeroRatioVariable->setValue(evaluationContext.closestWayRatio);
  1100. strategicalValueVariable->setValue(evaluationContext.strategicalValue);
  1101. goldPressureVariable->setValue(ai->buildAnalyzer->getGoldPressure());
  1102. goldCostVariable->setValue(evaluationContext.goldCost / ((float)ai->getFreeResources()[EGameResID::GOLD] + (float)ai->buildAnalyzer->getDailyIncome()[EGameResID::GOLD] + 1.0f));
  1103. turnVariable->setValue(evaluationContext.turn);
  1104. fearVariable->setValue(evaluationContext.enemyHeroDangerRatio);
  1105. engine->process();
  1106. fuzzyResult = value->getValue();
  1107. }
  1108. catch (fl::Exception& fe)
  1109. {
  1110. logAi->error("evaluate VisitTile: %s", fe.getWhat());
  1111. }
  1112. result = fuzzyResult;
  1113. }
  1114. else
  1115. {
  1116. float score = 0;
  1117. float maxWillingToLose = ai->cb->getTownsInfo().empty() ? 1 : 0.25;
  1118. #if NKAI_TRACE_LEVEL >= 2
  1119. logAi->trace("BEFORE: priorityTier %d, Evaluated %s, loss: %f, turn: %d, turns main: %f, scout: %f, gold: %f, cost: %d, army gain: %f, army growth: %f skill: %f danger: %d, threatTurns: %d, threat: %d, role: %s, strategical value: %f, conquest value: %f cwr: %f, fear: %f, isDefend: %d, fuzzy: %f",
  1120. priorityTier,
  1121. task->toString(),
  1122. evaluationContext.armyLossPersentage,
  1123. (int)evaluationContext.turn,
  1124. evaluationContext.movementCostByRole[HeroRole::MAIN],
  1125. evaluationContext.movementCostByRole[HeroRole::SCOUT],
  1126. goldRewardPerTurn,
  1127. evaluationContext.goldCost,
  1128. evaluationContext.armyReward,
  1129. evaluationContext.armyGrowth,
  1130. evaluationContext.skillReward,
  1131. evaluationContext.danger,
  1132. evaluationContext.threatTurns,
  1133. evaluationContext.threat,
  1134. evaluationContext.heroRole == HeroRole::MAIN ? "main" : "scout",
  1135. evaluationContext.strategicalValue,
  1136. evaluationContext.conquestValue,
  1137. evaluationContext.closestWayRatio,
  1138. evaluationContext.enemyHeroDangerRatio,
  1139. evaluationContext.isDefend,
  1140. fuzzyResult);
  1141. #endif
  1142. switch (priorityTier)
  1143. {
  1144. case PriorityTier::INSTAKILL: //Take towns / kill heroes in immediate reach
  1145. {
  1146. if (evaluationContext.turn > 0)
  1147. return 0;
  1148. if(evaluationContext.conquestValue > 0)
  1149. score = 1000;
  1150. if (score == 0 || (evaluationContext.enemyHeroDangerRatio > 1 && (evaluationContext.turn > 0 || evaluationContext.isExchange) && !ai->cb->getTownsInfo().empty()))
  1151. return 0;
  1152. if (maxWillingToLose - evaluationContext.armyLossPersentage < 0)
  1153. return 0;
  1154. score *= evaluationContext.closestWayRatio;
  1155. if (evaluationContext.movementCost > 0)
  1156. score /= evaluationContext.movementCost;
  1157. break;
  1158. }
  1159. case PriorityTier::INSTADEFEND: //Defend immediately threatened towns
  1160. {
  1161. if (evaluationContext.isDefend && evaluationContext.threatTurns == 0 && evaluationContext.turn == 0)
  1162. score = evaluationContext.armyInvolvement;
  1163. score *= evaluationContext.closestWayRatio;
  1164. break;
  1165. }
  1166. case PriorityTier::KILL: //Take towns / kill heroes that are further away
  1167. {
  1168. if (evaluationContext.conquestValue > 0)
  1169. score = 1000;
  1170. if (score == 0 || (evaluationContext.enemyHeroDangerRatio > 1 && (evaluationContext.turn > 0 || evaluationContext.isExchange) && !ai->cb->getTownsInfo().empty()))
  1171. return 0;
  1172. if (maxWillingToLose - evaluationContext.armyLossPersentage < 0)
  1173. return 0;
  1174. score *= evaluationContext.closestWayRatio;
  1175. if (evaluationContext.movementCost > 0)
  1176. score /= evaluationContext.movementCost;
  1177. break;
  1178. }
  1179. case PriorityTier::GATHER: //Collect unguarded stuff
  1180. {
  1181. if (evaluationContext.enemyHeroDangerRatio > 1)
  1182. return 0;
  1183. if (evaluationContext.isDefend)
  1184. return 0;
  1185. if (evaluationContext.armyLossPersentage > 0)
  1186. return 0;
  1187. if (evaluationContext.involvesSailing && evaluationContext.movementCostByRole[HeroRole::MAIN] > 0)
  1188. return 0;
  1189. if (evaluationContext.buildingCost.marketValue() > 0)
  1190. return 0;
  1191. if (evaluationContext.closestWayRatio < 1)
  1192. return 0;
  1193. score += evaluationContext.strategicalValue * 1000;
  1194. score += evaluationContext.goldReward;
  1195. score += evaluationContext.skillReward * evaluationContext.armyInvolvement * (1 - evaluationContext.armyLossPersentage) * 0.05;
  1196. score += evaluationContext.armyReward;
  1197. score += evaluationContext.armyGrowth;
  1198. if (score <= 0)
  1199. return 0;
  1200. else
  1201. score = 1000;
  1202. score *= evaluationContext.closestWayRatio;
  1203. if (evaluationContext.threat > evaluationContext.armyInvolvement && !evaluationContext.isDefend)
  1204. score *= evaluationContext.armyInvolvement / evaluationContext.threat;
  1205. if (evaluationContext.movementCost > 0)
  1206. score /= evaluationContext.movementCost;
  1207. break;
  1208. }
  1209. case PriorityTier::HUNTER_GATHER: //Collect guarded stuff
  1210. {
  1211. if (evaluationContext.enemyHeroDangerRatio > 1 && !evaluationContext.isDefend)
  1212. return 0;
  1213. if (evaluationContext.buildingCost.marketValue() > 0)
  1214. return 0;
  1215. if (evaluationContext.isDefend && (evaluationContext.enemyHeroDangerRatio < 1 || evaluationContext.threatTurns > 0 || evaluationContext.turn > 0))
  1216. return 0;
  1217. score += evaluationContext.strategicalValue * 1000;
  1218. score += evaluationContext.goldReward;
  1219. score += evaluationContext.skillReward * evaluationContext.armyInvolvement * (1 - evaluationContext.armyLossPersentage) * 0.05;
  1220. score += evaluationContext.armyReward;
  1221. score += evaluationContext.armyGrowth;
  1222. score -= evaluationContext.goldCost;
  1223. score -= evaluationContext.armyInvolvement * evaluationContext.armyLossPersentage;
  1224. if (score > 0)
  1225. {
  1226. score *= evaluationContext.closestWayRatio;
  1227. if (evaluationContext.enemyHeroDangerRatio > 1)
  1228. score /= evaluationContext.enemyHeroDangerRatio;
  1229. if (evaluationContext.movementCost > 0)
  1230. score /= evaluationContext.movementCost;
  1231. score *= (maxWillingToLose - evaluationContext.armyLossPersentage);
  1232. }
  1233. break;
  1234. }
  1235. case PriorityTier::DEFEND: //Defend whatever if nothing else is to do
  1236. {
  1237. if (evaluationContext.enemyHeroDangerRatio > 1 && evaluationContext.isExchange)
  1238. return 0;
  1239. if (evaluationContext.isDefend)
  1240. score = evaluationContext.armyInvolvement;
  1241. score *= evaluationContext.closestWayRatio;
  1242. score /= (evaluationContext.turn + 1);
  1243. break;
  1244. }
  1245. case PriorityTier::BUILDINGS: //For buildings and buying army
  1246. {
  1247. if (maxWillingToLose - evaluationContext.armyLossPersentage < 0)
  1248. return 0;
  1249. //If we already have locked resources, we don't look at other buildings
  1250. if (ai->getLockedResources().marketValue() > 0)
  1251. return 0;
  1252. score += evaluationContext.conquestValue * 1000;
  1253. score += evaluationContext.strategicalValue * 1000;
  1254. score += evaluationContext.goldReward;
  1255. score += evaluationContext.skillReward * evaluationContext.armyInvolvement * (1 - evaluationContext.armyLossPersentage) * 0.05;
  1256. score += evaluationContext.armyReward;
  1257. score += evaluationContext.armyGrowth;
  1258. if (evaluationContext.buildingCost.marketValue() > 0)
  1259. {
  1260. if (!evaluationContext.isTradeBuilding && ai->getFreeResources()[EGameResID::WOOD] - evaluationContext.buildingCost[EGameResID::WOOD] < 5 && ai->buildAnalyzer->getDailyIncome()[EGameResID::WOOD] < 1)
  1261. {
  1262. logAi->trace("Should make sure to build market-place instead of %s", task->toString());
  1263. for (auto town : ai->cb->getTownsInfo())
  1264. {
  1265. if (!town->hasBuiltSomeTradeBuilding())
  1266. return 0;
  1267. }
  1268. }
  1269. score += 1000;
  1270. auto resourcesAvailable = evaluationContext.evaluator.ai->getFreeResources();
  1271. auto income = ai->buildAnalyzer->getDailyIncome();
  1272. score /= evaluationContext.buildingCost.marketValue();
  1273. if (resourcesAvailable.canAfford(evaluationContext.buildingCost))
  1274. {
  1275. TResources needed = evaluationContext.buildingCost - resourcesAvailable;
  1276. needed.positive();
  1277. int turnsTo = needed.div(income);
  1278. if (turnsTo == INT_MAX)
  1279. return 0;
  1280. else
  1281. score /= turnsTo;
  1282. }
  1283. }
  1284. else
  1285. {
  1286. if (evaluationContext.enemyHeroDangerRatio > 1 && !evaluationContext.isDefend && evaluationContext.conquestValue == 0)
  1287. return 0;
  1288. }
  1289. break;
  1290. }
  1291. }
  1292. result = score;
  1293. //TODO: Figure out the root cause for why evaluationContext.closestWayRatio has become -nan(ind).
  1294. if (std::isnan(result))
  1295. return 0;
  1296. }
  1297. #if NKAI_TRACE_LEVEL >= 2
  1298. logAi->trace("priorityTier %d, Evaluated %s, loss: %f, turn: %d, turns main: %f, scout: %f, gold: %f, cost: %d, army gain: %f, army growth: %f skill: %f danger: %d, threatTurns: %d, threat: %d, role: %s, strategical value: %f, conquest value: %f cwr: %f, fear: %f, fuzzy: %f, result %f",
  1299. priorityTier,
  1300. task->toString(),
  1301. evaluationContext.armyLossPersentage,
  1302. (int)evaluationContext.turn,
  1303. evaluationContext.movementCostByRole[HeroRole::MAIN],
  1304. evaluationContext.movementCostByRole[HeroRole::SCOUT],
  1305. goldRewardPerTurn,
  1306. evaluationContext.goldCost,
  1307. evaluationContext.armyReward,
  1308. evaluationContext.armyGrowth,
  1309. evaluationContext.skillReward,
  1310. evaluationContext.danger,
  1311. evaluationContext.threatTurns,
  1312. evaluationContext.threat,
  1313. evaluationContext.heroRole == HeroRole::MAIN ? "main" : "scout",
  1314. evaluationContext.strategicalValue,
  1315. evaluationContext.conquestValue,
  1316. evaluationContext.closestWayRatio,
  1317. evaluationContext.enemyHeroDangerRatio,
  1318. fuzzyResult,
  1319. result);
  1320. #endif
  1321. return result;
  1322. }
  1323. }