CGTownInstance.cpp 33 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262
  1. /*
  2. * CGTownInstance.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 "CGTownInstance.h"
  12. #include "TownBuildingInstance.h"
  13. #include "../spells/CSpellHandler.h"
  14. #include "../bonuses/Bonus.h"
  15. #include "../battle/IBattleInfoCallback.h"
  16. #include "../battle/BattleLayout.h"
  17. #include "../CConfigHandler.h"
  18. #include "../texts/CGeneralTextHandler.h"
  19. #include "../IGameCallback.h"
  20. #include "../gameState/CGameState.h"
  21. #include "../mapping/CMap.h"
  22. #include "../CPlayerState.h"
  23. #include "../StartInfo.h"
  24. #include "../TerrainHandler.h"
  25. #include "../entities/building/CBuilding.h"
  26. #include "../entities/faction/CTownHandler.h"
  27. #include "../mapObjectConstructors/AObjectTypeHandler.h"
  28. #include "../mapObjectConstructors/CObjectClassesHandler.h"
  29. #include "../mapObjects/CGHeroInstance.h"
  30. #include "../modding/ModScope.h"
  31. #include "../networkPacks/StackLocation.h"
  32. #include "../networkPacks/PacksForClient.h"
  33. #include "../networkPacks/PacksForClientBattle.h"
  34. #include "../serializer/JsonSerializeFormat.h"
  35. #include <vstd/RNG.h>
  36. VCMI_LIB_NAMESPACE_BEGIN
  37. int CGTownInstance::getSightRadius() const //returns sight distance
  38. {
  39. auto ret = CBuilding::HEIGHT_NO_TOWER;
  40. for(const auto & bid : builtBuildings)
  41. {
  42. auto height = getTown()->buildings.at(bid)->height;
  43. if(ret < height)
  44. ret = height;
  45. }
  46. return ret;
  47. }
  48. void CGTownInstance::setPropertyDer(ObjProperty what, ObjPropertyID identifier)
  49. {
  50. ///this is freakin' overcomplicated solution
  51. switch (what)
  52. {
  53. case ObjProperty::STRUCTURE_ADD_VISITING_HERO:
  54. rewardableBuildings.at(identifier.getNum())->setProperty(ObjProperty::VISITORS, visitingHero->id);
  55. break;
  56. case ObjProperty::STRUCTURE_CLEAR_VISITORS:
  57. rewardableBuildings.at(identifier.getNum())->setProperty(ObjProperty::STRUCTURE_CLEAR_VISITORS, NumericID(0));
  58. break;
  59. case ObjProperty::STRUCTURE_ADD_GARRISONED_HERO: //add garrisoned hero to visitors
  60. rewardableBuildings.at(identifier.getNum())->setProperty(ObjProperty::VISITORS, garrisonHero->id);
  61. break;
  62. case ObjProperty::BONUS_VALUE_FIRST:
  63. bonusValue.first = identifier.getNum();
  64. break;
  65. case ObjProperty::BONUS_VALUE_SECOND:
  66. bonusValue.second = identifier.getNum();
  67. break;
  68. }
  69. }
  70. CGTownInstance::EFortLevel CGTownInstance::fortLevel() const //0 - none, 1 - fort, 2 - citadel, 3 - castle
  71. {
  72. if (hasBuilt(BuildingID::CASTLE))
  73. return CASTLE;
  74. if (hasBuilt(BuildingID::CITADEL))
  75. return CITADEL;
  76. if (hasBuilt(BuildingID::FORT))
  77. return FORT;
  78. return NONE;
  79. }
  80. int CGTownInstance::hallLevel() const // -1 - none, 0 - village, 1 - town, 2 - city, 3 - capitol
  81. {
  82. if (hasBuilt(BuildingID::CAPITOL))
  83. return 3;
  84. if (hasBuilt(BuildingID::CITY_HALL))
  85. return 2;
  86. if (hasBuilt(BuildingID::TOWN_HALL))
  87. return 1;
  88. if (hasBuilt(BuildingID::VILLAGE_HALL))
  89. return 0;
  90. return -1;
  91. }
  92. int CGTownInstance::mageGuildLevel() const
  93. {
  94. if (hasBuilt(BuildingID::MAGES_GUILD_5))
  95. return 5;
  96. if (hasBuilt(BuildingID::MAGES_GUILD_4))
  97. return 4;
  98. if (hasBuilt(BuildingID::MAGES_GUILD_3))
  99. return 3;
  100. if (hasBuilt(BuildingID::MAGES_GUILD_2))
  101. return 2;
  102. if (hasBuilt(BuildingID::MAGES_GUILD_1))
  103. return 1;
  104. return 0;
  105. }
  106. int CGTownInstance::getHordeLevel(const int & HID) const//HID - 0 or 1; returns creature level or -1 if that horde structure is not present
  107. {
  108. return getTown()->hordeLvl.at(HID);
  109. }
  110. int CGTownInstance::creatureGrowth(const int & level) const
  111. {
  112. return getGrowthInfo(level).totalGrowth();
  113. }
  114. GrowthInfo CGTownInstance::getGrowthInfo(int level) const
  115. {
  116. GrowthInfo ret;
  117. if (level<0 || level >=getTown()->creatures.size())
  118. return ret;
  119. if (creatures[level].second.empty())
  120. return ret; //no dwelling
  121. const Creature *creature = creatures[level].second.back().toEntity(VLC);
  122. const int base = creature->getGrowth();
  123. int castleBonus = 0;
  124. if(tempOwner.isValidPlayer())
  125. {
  126. auto * playerSettings = cb->getPlayerSettings(tempOwner);
  127. ret.handicapPercentage = playerSettings->handicap.percentGrowth;
  128. }
  129. else
  130. ret.handicapPercentage = 100;
  131. ret.entries.emplace_back(VLC->generaltexth->allTexts[590], base); // \n\nBasic growth %d"
  132. if (hasBuilt(BuildingID::CASTLE))
  133. ret.entries.emplace_back(subID, BuildingID::CASTLE, castleBonus = base);
  134. else if (hasBuilt(BuildingID::CITADEL))
  135. ret.entries.emplace_back(subID, BuildingID::CITADEL, castleBonus = base / 2);
  136. if(getTown()->hordeLvl.at(0) == level)//horde 1
  137. if(hasBuilt(BuildingID::HORDE_1))
  138. ret.entries.emplace_back(subID, BuildingID::HORDE_1, creature->getHorde());
  139. if(getTown()->hordeLvl.at(1) == level)//horde 2
  140. if(hasBuilt(BuildingID::HORDE_2))
  141. ret.entries.emplace_back(subID, BuildingID::HORDE_2, creature->getHorde());
  142. //statue-of-legion-like bonus: % to base+castle
  143. TConstBonusListPtr bonuses2 = getBonuses(Selector::type()(BonusType::CREATURE_GROWTH_PERCENT));
  144. for(const auto & b : *bonuses2)
  145. {
  146. const auto growth = b->val * (base + castleBonus) / 100;
  147. if (growth)
  148. {
  149. ret.entries.emplace_back(growth, b->Description(cb, growth));
  150. }
  151. }
  152. //other *-of-legion-like bonuses (%d to growth cumulative with grail)
  153. // Note: bonus uses 1-based levels (Pikeman is level 1), town list uses 0-based (Pikeman in 0-th creatures entry)
  154. TConstBonusListPtr bonuses = getBonuses(Selector::typeSubtype(BonusType::CREATURE_GROWTH, BonusCustomSubtype::creatureLevel(level+1)));
  155. for(const auto & b : *bonuses)
  156. ret.entries.emplace_back(b->val, b->Description(cb));
  157. int dwellingBonus = 0;
  158. if(const PlayerState *p = cb->getPlayerState(tempOwner, false))
  159. {
  160. dwellingBonus = getDwellingBonus(creatures[level].second, p->getOwnedObjects());
  161. }
  162. if(dwellingBonus)
  163. ret.entries.emplace_back(VLC->generaltexth->allTexts[591], dwellingBonus); // \nExternal dwellings %+d
  164. if(hasBuilt(BuildingID::GRAIL)) //grail - +50% to ALL (so far added) growth
  165. ret.entries.emplace_back(subID, BuildingID::GRAIL, ret.totalGrowth() / 2);
  166. return ret;
  167. }
  168. int CGTownInstance::getDwellingBonus(const std::vector<CreatureID>& creatureIds, const std::vector<const CGObjectInstance * >& dwellings) const
  169. {
  170. int totalBonus = 0;
  171. for (const auto& dwelling : dwellings)
  172. {
  173. const auto & dwellingCreatures = dwelling->asOwnable()->providedCreatures();
  174. bool hasMatch = false;
  175. for (const auto& creature : dwellingCreatures)
  176. hasMatch = vstd::contains(creatureIds, creature);
  177. if (hasMatch)
  178. totalBonus += 1;
  179. }
  180. return totalBonus;
  181. }
  182. TResources CGTownInstance::dailyIncome() const
  183. {
  184. TResources ret;
  185. for(const auto & p : getTown()->buildings)
  186. {
  187. BuildingID buildingUpgrade;
  188. for(const auto & p2 : getTown()->buildings)
  189. {
  190. if (p2.second->upgrade == p.first)
  191. {
  192. buildingUpgrade = p2.first;
  193. }
  194. }
  195. if (!hasBuilt(buildingUpgrade)&&(hasBuilt(p.first)))
  196. {
  197. ret += p.second->produce;
  198. }
  199. }
  200. if (!getOwner().isValidPlayer())
  201. return ret;
  202. const auto & playerSettings = cb->getPlayerSettings(getOwner());
  203. ret.applyHandicap(playerSettings->handicap.percentIncome);
  204. return ret;
  205. }
  206. std::vector<CreatureID> CGTownInstance::providedCreatures() const
  207. {
  208. return {};
  209. }
  210. bool CGTownInstance::hasFort() const
  211. {
  212. return hasBuilt(BuildingID::FORT);
  213. }
  214. bool CGTownInstance::hasCapitol() const
  215. {
  216. return hasBuilt(BuildingID::CAPITOL);
  217. }
  218. TownFortifications CGTownInstance::fortificationsLevel() const
  219. {
  220. auto result = getTown()->fortifications;
  221. for (auto const & buildingID : builtBuildings)
  222. result += getTown()->buildings.at(buildingID)->fortifications;
  223. if (result.wallsHealth == 0)
  224. return TownFortifications();
  225. return result;
  226. }
  227. CGTownInstance::CGTownInstance(IGameCallback *cb):
  228. CGDwelling(cb),
  229. built(0),
  230. destroyed(0),
  231. identifier(0),
  232. alignmentToPlayer(PlayerColor::NEUTRAL),
  233. spellResearchCounterDay(0),
  234. spellResearchAcceptedCounter(0),
  235. spellResearchAllowed(true)
  236. {
  237. this->setNodeType(CBonusSystemNode::TOWN);
  238. }
  239. CGTownInstance::~CGTownInstance()
  240. {
  241. for (auto & elem : rewardableBuildings)
  242. delete elem.second;
  243. }
  244. int CGTownInstance::spellsAtLevel(int level, bool checkGuild) const
  245. {
  246. if(checkGuild && mageGuildLevel() < level)
  247. return 0;
  248. int ret = 6 - level; //how many spells are available at this level
  249. if (hasBuilt(BuildingSubID::LIBRARY))
  250. ret++;
  251. return ret;
  252. }
  253. bool CGTownInstance::needsLastStack() const
  254. {
  255. return garrisonHero != nullptr;
  256. }
  257. void CGTownInstance::setOwner(const PlayerColor & player) const
  258. {
  259. removeCapitols(player);
  260. cb->setOwner(this, player);
  261. }
  262. void CGTownInstance::onHeroVisit(const CGHeroInstance * h) const
  263. {
  264. if(cb->gameState()->getPlayerRelations( getOwner(), h->getOwner() ) == PlayerRelations::ENEMIES)
  265. {
  266. if(armedGarrison() || visitingHero)
  267. {
  268. const CGHeroInstance * defendingHero = visitingHero ? visitingHero : garrisonHero;
  269. const CArmedInstance * defendingArmy = defendingHero ? (CArmedInstance *)defendingHero : this;
  270. const bool isBattleOutside = isBattleOutsideTown(defendingHero);
  271. if(!isBattleOutside && visitingHero && defendingHero == visitingHero)
  272. {
  273. //we have two approaches to merge armies: mergeGarrisonOnSiege() and used in the CGameHandler::garrisonSwap(ObjectInstanceID tid)
  274. auto * nodeSiege = defendingHero->whereShouldBeAttachedOnSiege(isBattleOutside);
  275. if(nodeSiege == (CBonusSystemNode *)this)
  276. cb->swapGarrisonOnSiege(this->id);
  277. const_cast<CGHeroInstance *>(defendingHero)->inTownGarrison = false; //hack to return visitor from garrison after battle
  278. }
  279. cb->startBattle(h, defendingArmy, getSightCenter(), h, defendingHero, BattleLayout::createDefaultLayout(cb, h, defendingArmy), (isBattleOutside ? nullptr : this));
  280. }
  281. else
  282. {
  283. auto heroColor = h->getOwner();
  284. onTownCaptured(heroColor);
  285. if(cb->gameState()->getPlayerStatus(heroColor) == EPlayerStatus::WINNER)
  286. {
  287. return; //we just won game, we do not need to perform any extra actions
  288. //TODO: check how does H3 behave, visiting town on victory can affect campaigns (spells learned, +1 stat building visited)
  289. }
  290. cb->heroVisitCastle(this, h);
  291. }
  292. }
  293. else
  294. {
  295. assert(h->visitablePos() == this->visitablePos());
  296. bool commander_recover = h->commander && !h->commander->alive;
  297. if (commander_recover) // rise commander from dead
  298. {
  299. SetCommanderProperty scp;
  300. scp.heroid = h->id;
  301. scp.which = SetCommanderProperty::ALIVE;
  302. scp.amount = 1;
  303. cb->sendAndApply(scp);
  304. }
  305. cb->heroVisitCastle(this, h);
  306. // TODO(vmarkovtsev): implement payment for rising the commander
  307. if (commander_recover) // info window about commander
  308. {
  309. InfoWindow iw;
  310. iw.player = h->tempOwner;
  311. iw.text.appendRawString(h->commander->getName());
  312. iw.components.emplace_back(ComponentType::CREATURE, h->commander->getId(), h->commander->getCount());
  313. cb->showInfoDialog(&iw);
  314. }
  315. }
  316. }
  317. void CGTownInstance::onHeroLeave(const CGHeroInstance * h) const
  318. {
  319. //FIXME: find out why this issue appears on random maps
  320. if(visitingHero == h)
  321. {
  322. cb->stopHeroVisitCastle(this, h);
  323. logGlobal->trace("%s correctly left town %s", h->getNameTranslated(), getNameTranslated());
  324. }
  325. else
  326. logGlobal->warn("Warning, %s tries to leave the town %s but hero is not inside.", h->getNameTranslated(), getNameTranslated());
  327. }
  328. std::string CGTownInstance::getObjectName() const
  329. {
  330. return getNameTranslated() + ", " + getTown()->faction->getNameTranslated();
  331. }
  332. bool CGTownInstance::townEnvisagesBuilding(BuildingSubID::EBuildingSubID subId) const
  333. {
  334. return getTown()->getBuildingType(subId) != BuildingID::NONE;
  335. }
  336. void CGTownInstance::initializeConfigurableBuildings(vstd::RNG & rand)
  337. {
  338. for(const auto & kvp : getTown()->buildings)
  339. {
  340. if(!kvp.second->rewardableObjectInfo.getParameters().isNull())
  341. rewardableBuildings[kvp.first] = new TownRewardableBuildingInstance(this, kvp.second->bid, rand);
  342. }
  343. }
  344. DamageRange CGTownInstance::getTowerDamageRange() const
  345. {
  346. // http://heroes.thelazy.net/wiki/Arrow_tower
  347. // base damage, irregardless of town level
  348. static constexpr int baseDamage = 6;
  349. // extra damage, for each building in town
  350. static constexpr int extraDamage = 1;
  351. const int minDamage = baseDamage + extraDamage * getTownLevel();
  352. return {
  353. minDamage,
  354. minDamage * 2
  355. };
  356. }
  357. DamageRange CGTownInstance::getKeepDamageRange() const
  358. {
  359. // http://heroes.thelazy.net/wiki/Arrow_tower
  360. // base damage, irregardless of town level
  361. static constexpr int baseDamage = 10;
  362. // extra damage, for each building in town
  363. static constexpr int extraDamage = 2;
  364. const int minDamage = baseDamage + extraDamage * getTownLevel();
  365. return {
  366. minDamage,
  367. minDamage * 2
  368. };
  369. }
  370. FactionID CGTownInstance::randomizeFaction(vstd::RNG & rand)
  371. {
  372. if(getOwner().isValidPlayer())
  373. return cb->gameState()->scenarioOps->getIthPlayersSettings(getOwner()).castle;
  374. if(alignmentToPlayer.isValidPlayer())
  375. return cb->gameState()->scenarioOps->getIthPlayersSettings(alignmentToPlayer).castle;
  376. std::vector<FactionID> potentialPicks;
  377. for (FactionID faction(0); faction < FactionID(VLC->townh->size()); ++faction)
  378. if (VLC->factions()->getById(faction)->hasTown())
  379. potentialPicks.push_back(faction);
  380. assert(!potentialPicks.empty());
  381. return *RandomGeneratorUtil::nextItem(potentialPicks, rand);
  382. }
  383. void CGTownInstance::pickRandomObject(vstd::RNG & rand)
  384. {
  385. assert(ID == MapObjectID::TOWN || ID == MapObjectID::RANDOM_TOWN);
  386. if (ID == MapObjectID::RANDOM_TOWN)
  387. {
  388. ID = MapObjectID::TOWN;
  389. subID = randomizeFaction(rand);
  390. }
  391. assert(ID == Obj::TOWN); // just in case
  392. setType(ID, subID);
  393. randomizeArmy(getFactionID());
  394. updateAppearance();
  395. }
  396. void CGTownInstance::initObj(vstd::RNG & rand) ///initialize town structures
  397. {
  398. blockVisit = true;
  399. if(townEnvisagesBuilding(BuildingSubID::PORTAL_OF_SUMMONING)) //Dungeon for example
  400. creatures.resize(getTown()->creatures.size() + 1);
  401. else
  402. creatures.resize(getTown()->creatures.size());
  403. for (int level = 0; level < getTown()->creatures.size(); level++)
  404. {
  405. BuildingID buildID = BuildingID(BuildingID::getDwellingFromLevel(level, 0));
  406. int upgradeNum = 0;
  407. for (; getTown()->buildings.count(buildID); upgradeNum++, BuildingID::advanceDwelling(buildID))
  408. {
  409. if (hasBuilt(buildID) && getTown()->creatures.at(level).size() > upgradeNum)
  410. creatures[level].second.push_back(getTown()->creatures[level][upgradeNum]);
  411. }
  412. }
  413. initializeConfigurableBuildings(rand);
  414. initializeNeutralTownGarrison(rand);
  415. recreateBuildingsBonuses();
  416. updateAppearance();
  417. }
  418. void CGTownInstance::initializeNeutralTownGarrison(vstd::RNG & rand)
  419. {
  420. struct RandomGuardsInfo{
  421. int tier;
  422. int chance;
  423. int min;
  424. int max;
  425. };
  426. constexpr std::array<RandomGuardsInfo, 4> randomGuards = {
  427. RandomGuardsInfo{ 0, 33, 8, 15 },
  428. RandomGuardsInfo{ 1, 33, 5, 7 },
  429. RandomGuardsInfo{ 2, 20, 3, 5 },
  430. RandomGuardsInfo{ 3, 14, 1, 3 },
  431. };
  432. // Only neutral towns may get initial garrison
  433. if (getOwner().isValidPlayer())
  434. return;
  435. // Only towns with garrison not set in map editor may get initial garrison
  436. // FIXME: H3 editor allow explicitly empty garrison, but vcmi loses this flag on load
  437. if (stacksCount() > 0)
  438. return;
  439. for (auto const & guard : randomGuards)
  440. {
  441. if (rand.nextInt(99) >= guard.chance)
  442. continue;
  443. CreatureID guardID = getTown()->creatures[guard.tier].at(0);
  444. int guardSize = rand.nextInt(guard.min, guard.max);
  445. putStack(getFreeSlot(), new CStackInstance(guardID, guardSize));
  446. }
  447. }
  448. void CGTownInstance::newTurn(vstd::RNG & rand) const
  449. {
  450. for(const auto & building : rewardableBuildings)
  451. building.second->newTurn(rand);
  452. if(hasBuilt(BuildingSubID::BANK) && bonusValue.second > 0)
  453. {
  454. TResources res;
  455. res[EGameResID::GOLD] = -500;
  456. cb->giveResources(getOwner(), res);
  457. cb->setObjPropertyValue(id, ObjProperty::BONUS_VALUE_SECOND, bonusValue.second - 500);
  458. }
  459. }
  460. bool CGTownInstance::passableFor(PlayerColor color) const
  461. {
  462. if (!armedGarrison())//empty castle - anyone can visit
  463. return true;
  464. if ( tempOwner == PlayerColor::NEUTRAL )//neutral guarded - no one can visit
  465. return false;
  466. return cb->getPlayerRelations(tempOwner, color) != PlayerRelations::ENEMIES;
  467. }
  468. void CGTownInstance::getOutOffsets( std::vector<int3> &offsets ) const
  469. {
  470. offsets = {int3(-1,2,0), int3(+1,2,0)};
  471. }
  472. CGTownInstance::EGeneratorState CGTownInstance::shipyardStatus() const
  473. {
  474. if (!hasBuilt(BuildingID::SHIPYARD))
  475. return EGeneratorState::UNKNOWN;
  476. return IShipyard::shipyardStatus();
  477. }
  478. const IObjectInterface * CGTownInstance::getObject() const
  479. {
  480. return this;
  481. }
  482. void CGTownInstance::mergeGarrisonOnSiege() const
  483. {
  484. auto getWeakestStackSlot = [&](ui64 powerLimit)
  485. {
  486. std::vector<SlotID> weakSlots;
  487. auto stacksList = visitingHero->stacks;
  488. std::pair<SlotID, CStackInstance *> pair;
  489. while(!stacksList.empty())
  490. {
  491. pair = *vstd::minElementByFun(stacksList, [&](const std::pair<SlotID, CStackInstance *> & elem) { return elem.second->getPower(); });
  492. if(powerLimit > pair.second->getPower() &&
  493. (weakSlots.empty() || pair.second->getPower() == visitingHero->getStack(weakSlots.front()).getPower()))
  494. {
  495. weakSlots.push_back(pair.first);
  496. stacksList.erase(pair.first);
  497. }
  498. else
  499. break;
  500. }
  501. if(!weakSlots.empty())
  502. return *std::max_element(weakSlots.begin(), weakSlots.end());
  503. return SlotID();
  504. };
  505. auto count = static_cast<int>(stacks.size());
  506. for(int i = 0; i < count; i++)
  507. {
  508. auto pair = *vstd::maxElementByFun(stacks, [&](const std::pair<SlotID, CStackInstance *> & elem)
  509. {
  510. ui64 power = elem.second->getPower();
  511. auto dst = visitingHero->getSlotFor(elem.second->getCreatureID());
  512. if(dst.validSlot() && visitingHero->hasStackAtSlot(dst))
  513. power += visitingHero->getStack(dst).getPower();
  514. return power;
  515. });
  516. auto dst = visitingHero->getSlotFor(pair.second->getCreatureID());
  517. if(dst.validSlot())
  518. cb->moveStack(StackLocation(this, pair.first), StackLocation(visitingHero, dst), -1);
  519. else
  520. {
  521. dst = getWeakestStackSlot(static_cast<int>(pair.second->getPower()));
  522. if(dst.validSlot())
  523. cb->swapStacks(StackLocation(this, pair.first), StackLocation(visitingHero, dst));
  524. }
  525. }
  526. }
  527. void CGTownInstance::removeCapitols(const PlayerColor & owner) const
  528. {
  529. if (hasCapitol()) // search if there's an older capitol
  530. {
  531. PlayerState* state = cb->gameState()->getPlayerState(owner); //get all towns owned by player
  532. for (const auto & otherTown : state->getTowns())
  533. {
  534. if (otherTown != this && otherTown->hasCapitol())
  535. {
  536. RazeStructures rs;
  537. rs.tid = id;
  538. rs.bid.insert(BuildingID::CAPITOL);
  539. rs.destroyed = destroyed;
  540. cb->sendAndApply(rs);
  541. return;
  542. }
  543. }
  544. }
  545. }
  546. void CGTownInstance::clearArmy() const
  547. {
  548. while(!stacks.empty())
  549. {
  550. cb->eraseStack(StackLocation(this, stacks.begin()->first));
  551. }
  552. }
  553. BoatId CGTownInstance::getBoatType() const
  554. {
  555. return getTown()->faction->boatType;
  556. }
  557. int CGTownInstance::getMarketEfficiency() const
  558. {
  559. if(!hasBuiltSomeTradeBuilding())
  560. return 0;
  561. const PlayerState *p = cb->getPlayerState(tempOwner);
  562. assert(p);
  563. int marketCount = 0;
  564. for(const CGTownInstance *t : p->getTowns())
  565. if(t->hasBuiltSomeTradeBuilding())
  566. marketCount++;
  567. return marketCount;
  568. }
  569. std::vector<TradeItemBuy> CGTownInstance::availableItemsIds(EMarketMode mode) const
  570. {
  571. if(mode == EMarketMode::RESOURCE_ARTIFACT)
  572. {
  573. std::vector<TradeItemBuy> ret;
  574. for(const ArtifactID a : cb->gameState()->map->townMerchantArtifacts)
  575. ret.push_back(a);
  576. return ret;
  577. }
  578. else if ( mode == EMarketMode::RESOURCE_SKILL )
  579. {
  580. return cb->gameState()->map->townUniversitySkills;
  581. }
  582. else
  583. return IMarket::availableItemsIds(mode);
  584. }
  585. ObjectInstanceID CGTownInstance::getObjInstanceID() const
  586. {
  587. return id;
  588. }
  589. void CGTownInstance::updateAppearance()
  590. {
  591. auto terrain = cb->gameState()->getTile(visitablePos())->getTerrainID();
  592. //FIXME: not the best way to do this
  593. auto app = getObjectHandler()->getOverride(terrain, this);
  594. if (app)
  595. appearance = app;
  596. }
  597. std::string CGTownInstance::nodeName() const
  598. {
  599. return "Town (" + getTown()->faction->getNameTranslated() + ") of " + getNameTranslated();
  600. }
  601. void CGTownInstance::deserializationFix()
  602. {
  603. attachTo(townAndVis);
  604. //Hero is already handled by CGameState::attachArmedObjects
  605. // if(visitingHero)
  606. // visitingHero->attachTo(&townAndVis);
  607. // if(garrisonHero)
  608. // garrisonHero->attachTo(this);
  609. }
  610. void CGTownInstance::updateMoraleBonusFromArmy()
  611. {
  612. auto b = getExportedBonusList().getFirst(Selector::sourceType()(BonusSource::ARMY).And(Selector::type()(BonusType::MORALE)));
  613. if(!b)
  614. {
  615. b = std::make_shared<Bonus>(BonusDuration::PERMANENT, BonusType::MORALE, BonusSource::ARMY, 0, BonusSourceID());
  616. addNewBonus(b);
  617. }
  618. if (garrisonHero)
  619. {
  620. b->val = 0;
  621. CBonusSystemNode::treeHasChanged();
  622. }
  623. else
  624. CArmedInstance::updateMoraleBonusFromArmy();
  625. }
  626. void CGTownInstance::recreateBuildingsBonuses()
  627. {
  628. BonusList bl;
  629. getExportedBonusList().getBonuses(bl, Selector::sourceType()(BonusSource::TOWN_STRUCTURE));
  630. for(const auto & b : bl)
  631. removeBonus(b);
  632. for(const auto & bid : builtBuildings)
  633. {
  634. bool bonusesReplacedByUpgrade = false;
  635. for(const auto & upgradeID : builtBuildings)
  636. {
  637. const auto & upgrade = getTown()->buildings.at(upgradeID);
  638. if (upgrade->getBase() == bid && upgrade->upgradeReplacesBonuses)
  639. bonusesReplacedByUpgrade = true;
  640. }
  641. // bonuses from this building are disabled and replaced by bonuses from an upgrade
  642. if (bonusesReplacedByUpgrade)
  643. continue;
  644. auto building = getTown()->buildings.at(bid);
  645. if(building->buildingBonuses.empty())
  646. continue;
  647. for(auto & bonus : building->buildingBonuses)
  648. addNewBonus(bonus);
  649. }
  650. }
  651. void CGTownInstance::setVisitingHero(CGHeroInstance *h)
  652. {
  653. if(visitingHero.get() == h)
  654. return;
  655. if(h)
  656. {
  657. PlayerState *p = cb->gameState()->getPlayerState(h->tempOwner);
  658. assert(p);
  659. h->detachFrom(*p);
  660. h->attachTo(townAndVis);
  661. visitingHero = h;
  662. h->visitedTown = this;
  663. h->inTownGarrison = false;
  664. }
  665. else
  666. {
  667. PlayerState *p = cb->gameState()->getPlayerState(visitingHero->tempOwner);
  668. visitingHero->visitedTown = nullptr;
  669. visitingHero->detachFrom(townAndVis);
  670. visitingHero->attachTo(*p);
  671. visitingHero = nullptr;
  672. }
  673. }
  674. void CGTownInstance::setGarrisonedHero(CGHeroInstance *h)
  675. {
  676. if(garrisonHero.get() == h)
  677. return;
  678. if(h)
  679. {
  680. PlayerState *p = cb->gameState()->getPlayerState(h->tempOwner);
  681. assert(p);
  682. h->detachFrom(*p);
  683. h->attachTo(*this);
  684. garrisonHero = h;
  685. h->visitedTown = this;
  686. h->inTownGarrison = true;
  687. }
  688. else
  689. {
  690. PlayerState *p = cb->gameState()->getPlayerState(garrisonHero->tempOwner);
  691. garrisonHero->visitedTown = nullptr;
  692. garrisonHero->inTownGarrison = false;
  693. garrisonHero->detachFrom(*this);
  694. garrisonHero->attachTo(*p);
  695. garrisonHero = nullptr;
  696. }
  697. updateMoraleBonusFromArmy(); //avoid giving morale bonus for same army twice
  698. }
  699. bool CGTownInstance::armedGarrison() const
  700. {
  701. return !stacks.empty() || garrisonHero;
  702. }
  703. int CGTownInstance::getTownLevel() const
  704. {
  705. // count all buildings that are not upgrades
  706. int level = 0;
  707. for(const auto & bid : builtBuildings)
  708. {
  709. if(getTown()->buildings.at(bid)->upgrade == BuildingID::NONE)
  710. level++;
  711. }
  712. return level;
  713. }
  714. CBonusSystemNode & CGTownInstance::whatShouldBeAttached()
  715. {
  716. return townAndVis;
  717. }
  718. std::string CGTownInstance::getNameTranslated() const
  719. {
  720. return VLC->generaltexth->translate(nameTextId);
  721. }
  722. std::string CGTownInstance::getNameTextID() const
  723. {
  724. return nameTextId;
  725. }
  726. void CGTownInstance::setNameTextId( const std::string & newName )
  727. {
  728. nameTextId = newName;
  729. }
  730. const CArmedInstance * CGTownInstance::getUpperArmy() const
  731. {
  732. if(garrisonHero)
  733. return garrisonHero;
  734. return this;
  735. }
  736. bool CGTownInstance::hasBuiltSomeTradeBuilding() const
  737. {
  738. return availableModes().empty() ? false : true;
  739. }
  740. bool CGTownInstance::hasBuilt(BuildingSubID::EBuildingSubID buildingID) const
  741. {
  742. for(const auto & bid : builtBuildings)
  743. {
  744. if(getTown()->buildings.at(bid)->subId == buildingID)
  745. return true;
  746. }
  747. return false;
  748. }
  749. bool CGTownInstance::hasBuilt(const BuildingID & buildingID) const
  750. {
  751. return vstd::contains(builtBuildings, buildingID);
  752. }
  753. bool CGTownInstance::hasBuilt(const BuildingID & buildingID, FactionID townID) const
  754. {
  755. if (townID == getTown()->faction->getId() || townID == FactionID::ANY)
  756. return hasBuilt(buildingID);
  757. return false;
  758. }
  759. void CGTownInstance::addBuilding(const BuildingID & buildingID)
  760. {
  761. if(buildingID == BuildingID::NONE)
  762. return;
  763. builtBuildings.insert(buildingID);
  764. }
  765. std::set<EMarketMode> CGTownInstance::availableModes() const
  766. {
  767. std::set<EMarketMode> result;
  768. for (const auto & buildingID : builtBuildings)
  769. {
  770. const auto * buildingPtr = getTown()->buildings.at(buildingID).get();
  771. result.insert(buildingPtr->marketModes.begin(), buildingPtr->marketModes.end());
  772. }
  773. return result;
  774. }
  775. void CGTownInstance::removeBuilding(const BuildingID & buildingID)
  776. {
  777. if(!vstd::contains(builtBuildings, buildingID))
  778. return;
  779. builtBuildings.erase(buildingID);
  780. }
  781. void CGTownInstance::removeAllBuildings()
  782. {
  783. builtBuildings.clear();
  784. }
  785. std::set<BuildingID> CGTownInstance::getBuildings() const
  786. {
  787. return builtBuildings;
  788. }
  789. TResources CGTownInstance::getBuildingCost(const BuildingID & buildingID) const
  790. {
  791. if (vstd::contains(getTown()->buildings, buildingID))
  792. return getTown()->buildings.at(buildingID)->resources;
  793. else
  794. {
  795. logGlobal->error("Town %s at %s has no possible building %d!", getNameTranslated(), anchorPos().toString(), buildingID.toEnum());
  796. return TResources();
  797. }
  798. }
  799. CBuilding::TRequired CGTownInstance::genBuildingRequirements(const BuildingID & buildID, bool deep) const
  800. {
  801. const CBuilding * building = getTown()->buildings.at(buildID);
  802. //TODO: find better solution to prevent infinite loops
  803. std::set<BuildingID> processed;
  804. std::function<CBuilding::TRequired::Variant(const BuildingID &)> dependTest =
  805. [&](const BuildingID & id) -> CBuilding::TRequired::Variant
  806. {
  807. if (getTown()->buildings.count(id) == 0)
  808. {
  809. logMod->error("Invalid building ID %d in building dependencies!", id.getNum());
  810. return CBuilding::TRequired::OperatorAll();
  811. }
  812. const CBuilding * build = getTown()->buildings.at(id);
  813. CBuilding::TRequired::OperatorAll requirements;
  814. if (!hasBuilt(id))
  815. {
  816. if (deep)
  817. requirements.expressions.emplace_back(id);
  818. else
  819. return id;
  820. }
  821. if(!vstd::contains(processed, id))
  822. {
  823. processed.insert(id);
  824. if (build->upgrade != BuildingID::NONE)
  825. requirements.expressions.push_back(dependTest(build->upgrade));
  826. requirements.expressions.push_back(build->requirements.morph(dependTest));
  827. }
  828. return requirements;
  829. };
  830. CBuilding::TRequired::OperatorAll requirements;
  831. if (building->upgrade != BuildingID::NONE)
  832. {
  833. const CBuilding * upgr = getTown()->buildings.at(building->upgrade);
  834. requirements.expressions.push_back(dependTest(upgr->bid));
  835. processed.clear();
  836. }
  837. requirements.expressions.push_back(building->requirements.morph(dependTest));
  838. CBuilding::TRequired::Variant variant(requirements);
  839. CBuilding::TRequired ret(variant);
  840. ret.minimize();
  841. return ret;
  842. }
  843. void CGTownInstance::addHeroToStructureVisitors(const CGHeroInstance *h, si64 structureInstanceID ) const
  844. {
  845. if(visitingHero == h)
  846. cb->setObjPropertyValue(id, ObjProperty::STRUCTURE_ADD_VISITING_HERO, structureInstanceID); //add to visitors
  847. else if(garrisonHero == h)
  848. cb->setObjPropertyValue(id, ObjProperty::STRUCTURE_ADD_GARRISONED_HERO, structureInstanceID); //then it must be garrisoned hero
  849. else
  850. {
  851. //should never ever happen
  852. logGlobal->error("Cannot add hero %s to visitors of structure # %d", h->getNameTranslated(), structureInstanceID);
  853. throw std::runtime_error("unexpected hero in CGTownInstance::addHeroToStructureVisitors");
  854. }
  855. }
  856. void CGTownInstance::battleFinished(const CGHeroInstance * hero, const BattleResult & result) const
  857. {
  858. if(result.winner == BattleSide::ATTACKER)
  859. {
  860. clearArmy();
  861. onTownCaptured(hero->getOwner());
  862. }
  863. }
  864. void CGTownInstance::onTownCaptured(const PlayerColor & winner) const
  865. {
  866. setOwner(winner);
  867. cb->changeFogOfWar(getSightCenter(), getSightRadius(), winner, ETileVisibility::REVEALED);
  868. }
  869. void CGTownInstance::afterAddToMap(CMap * map)
  870. {
  871. map->towns.emplace_back(this);
  872. }
  873. void CGTownInstance::afterRemoveFromMap(CMap * map)
  874. {
  875. vstd::erase_if_present(map->towns, this);
  876. }
  877. void CGTownInstance::serializeJsonOptions(JsonSerializeFormat & handler)
  878. {
  879. CGObjectInstance::serializeJsonOwner(handler);
  880. if(!handler.saving)
  881. handler.serializeEnum("tightFormation", formation, NArmyFormation::names); //for old format
  882. CArmedInstance::serializeJsonOptions(handler);
  883. handler.serializeString("name", nameTextId);
  884. {
  885. auto decodeBuilding = [this](const std::string & identifier) -> si32
  886. {
  887. auto rawId = VLC->identifiers()->getIdentifier(ModScope::scopeMap(), getTown()->getBuildingScope(), identifier);
  888. if(rawId)
  889. return rawId.value();
  890. else
  891. return -1;
  892. };
  893. auto encodeBuilding = [this](si32 index) -> std::string
  894. {
  895. return getTown()->buildings.at(BuildingID(index))->getJsonKey();
  896. };
  897. const std::set<si32> standard = getTown()->getAllBuildings();//by default all buildings are allowed
  898. JsonSerializeFormat::LICSet buildingsLIC(standard, decodeBuilding, encodeBuilding);
  899. if(handler.saving)
  900. {
  901. bool customBuildings = false;
  902. boost::logic::tribool hasFort(false);
  903. for(const BuildingID & id : forbiddenBuildings)
  904. {
  905. buildingsLIC.none.insert(id.getNum());
  906. customBuildings = true;
  907. }
  908. for(const BuildingID & id : builtBuildings)
  909. {
  910. if(id == BuildingID::DEFAULT)
  911. continue;
  912. const CBuilding * building = getTown()->buildings.at(id);
  913. if(building->mode == CBuilding::BUILD_AUTO)
  914. continue;
  915. if(id == BuildingID::FORT)
  916. hasFort = true;
  917. buildingsLIC.all.insert(id.getNum());
  918. customBuildings = true;
  919. }
  920. if(customBuildings)
  921. handler.serializeLIC("buildings", buildingsLIC);
  922. else
  923. handler.serializeBool("hasFort",hasFort);
  924. }
  925. else
  926. {
  927. handler.serializeLIC("buildings", buildingsLIC);
  928. addBuilding(BuildingID::VILLAGE_HALL);
  929. if(buildingsLIC.none.empty() && buildingsLIC.all.empty())
  930. {
  931. addBuilding(BuildingID::DEFAULT);
  932. bool hasFort = false;
  933. handler.serializeBool("hasFort",hasFort);
  934. if(hasFort)
  935. addBuilding(BuildingID::FORT);
  936. }
  937. else
  938. {
  939. for(const si32 item : buildingsLIC.none)
  940. forbiddenBuildings.insert(BuildingID(item));
  941. for(const si32 item : buildingsLIC.all)
  942. addBuilding(BuildingID(item));
  943. }
  944. }
  945. }
  946. {
  947. handler.serializeIdArray( "possibleSpells", possibleSpells);
  948. handler.serializeIdArray( "obligatorySpells", obligatorySpells);
  949. }
  950. {
  951. auto eventsHandler = handler.enterArray("events");
  952. eventsHandler.syncSize(events, JsonNode::JsonType::DATA_VECTOR);
  953. eventsHandler.serializeStruct(events);
  954. }
  955. }
  956. const CFaction * CGTownInstance::getFaction() const
  957. {
  958. return getFactionID().toFaction();
  959. }
  960. const CTown * CGTownInstance::getTown() const
  961. {
  962. if(ID == Obj::RANDOM_TOWN)
  963. return VLC->townh->randomTown;
  964. return getFaction()->town;
  965. }
  966. FactionID CGTownInstance::getFactionID() const
  967. {
  968. return FactionID(subID.getNum());
  969. }
  970. TerrainId CGTownInstance::getNativeTerrain() const
  971. {
  972. return getTown()->faction->getNativeTerrain();
  973. }
  974. ArtifactID CGTownInstance::getWarMachineInBuilding(BuildingID building) const
  975. {
  976. if (builtBuildings.count(building) == 0)
  977. return ArtifactID::NONE;
  978. if (building == BuildingID::BLACKSMITH && getTown()->warMachineDeprecated.hasValue())
  979. return getTown()->warMachineDeprecated.toCreature()->warMachine;
  980. return getTown()->buildings.at(building)->warMachine;
  981. }
  982. bool CGTownInstance::isWarMachineAvailable(ArtifactID warMachine) const
  983. {
  984. for (auto const & buildingID : builtBuildings)
  985. if (getTown()->buildings.at(buildingID)->warMachine == warMachine)
  986. return true;
  987. if (builtBuildings.count(BuildingID::BLACKSMITH) &&
  988. getTown()->warMachineDeprecated.hasValue() &&
  989. getTown()->warMachineDeprecated.toCreature()->warMachine == warMachine)
  990. return true;
  991. return false;
  992. }
  993. GrowthInfo::Entry::Entry(const std::string &format, int _count)
  994. : count(_count)
  995. {
  996. MetaString formatter;
  997. formatter.appendRawString(format);
  998. formatter.replacePositiveNumber(count);
  999. description = formatter.toString();
  1000. }
  1001. GrowthInfo::Entry::Entry(int subID, const BuildingID & building, int _count): count(_count)
  1002. {
  1003. MetaString formatter;
  1004. formatter.appendRawString("%s %+d");
  1005. formatter.replaceRawString(FactionID(subID).toFaction()->town->buildings.at(building)->getNameTranslated());
  1006. formatter.replacePositiveNumber(count);
  1007. description = formatter.toString();
  1008. }
  1009. GrowthInfo::Entry::Entry(int _count, std::string fullDescription):
  1010. count(_count),
  1011. description(std::move(fullDescription))
  1012. {
  1013. }
  1014. CTownAndVisitingHero::CTownAndVisitingHero()
  1015. {
  1016. setNodeType(TOWN_AND_VISITOR);
  1017. }
  1018. int GrowthInfo::totalGrowth() const
  1019. {
  1020. int ret = 0;
  1021. for(const Entry &entry : entries)
  1022. ret += entry.count;
  1023. // always round up income - we don't want buildings to always produce zero if handicap in use
  1024. return vstd::divideAndCeil(ret * handicapPercentage, 100);
  1025. }
  1026. void CGTownInstance::fillUpgradeInfo(UpgradeInfo & info, const CStackInstance &stack) const
  1027. {
  1028. for(const CGTownInstance::TCreaturesSet::value_type & dwelling : creatures)
  1029. {
  1030. if (vstd::contains(dwelling.second, stack.getId())) //Dwelling with our creature
  1031. {
  1032. for(const auto & upgrID : dwelling.second)
  1033. {
  1034. if(vstd::contains(stack.getCreature()->upgrades, upgrID)) //possible upgrade
  1035. {
  1036. info.newID.push_back(upgrID);
  1037. info.cost.push_back(upgrID.toCreature()->getFullRecruitCost() - stack.getType()->getFullRecruitCost());
  1038. }
  1039. }
  1040. }
  1041. }
  1042. }
  1043. void CGTownInstance::postDeserialize()
  1044. {
  1045. setNodeType(CBonusSystemNode::TOWN);
  1046. for(auto & building : rewardableBuildings)
  1047. building.second->town = this;
  1048. }
  1049. std::map<BuildingID, TownRewardableBuildingInstance*> CGTownInstance::convertOldBuildings(std::vector<TownRewardableBuildingInstance*> oldVector)
  1050. {
  1051. std::map<BuildingID, TownRewardableBuildingInstance*> result;
  1052. for(auto & building : oldVector)
  1053. {
  1054. result[building->getBuildingType()] = new TownRewardableBuildingInstance(*building);
  1055. delete building;
  1056. }
  1057. return result;
  1058. }
  1059. VCMI_LIB_NAMESPACE_END