CTownHandler.cpp 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266
  1. /*
  2. * CTownHandler.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 "CTownHandler.h"
  12. #include "VCMI_Lib.h"
  13. #include "CGeneralTextHandler.h"
  14. #include "JsonNode.h"
  15. #include "constants/StringConstants.h"
  16. #include "CCreatureHandler.h"
  17. #include "CHeroHandler.h"
  18. #include "CArtHandler.h"
  19. #include "GameSettings.h"
  20. #include "TerrainHandler.h"
  21. #include "spells/CSpellHandler.h"
  22. #include "filesystem/Filesystem.h"
  23. #include "bonuses/Bonus.h"
  24. #include "bonuses/Propagators.h"
  25. #include "ResourceSet.h"
  26. #include "mapObjectConstructors/AObjectTypeHandler.h"
  27. #include "mapObjectConstructors/CObjectClassesHandler.h"
  28. #include "modding/IdentifierStorage.h"
  29. #include "modding/ModScope.h"
  30. VCMI_LIB_NAMESPACE_BEGIN
  31. const int NAMES_PER_TOWN=16; // number of town names per faction in H3 files. Json can define any number
  32. const std::map<std::string, CBuilding::EBuildMode> CBuilding::MODES =
  33. {
  34. { "normal", CBuilding::BUILD_NORMAL },
  35. { "auto", CBuilding::BUILD_AUTO },
  36. { "special", CBuilding::BUILD_SPECIAL },
  37. { "grail", CBuilding::BUILD_GRAIL }
  38. };
  39. const std::map<std::string, CBuilding::ETowerHeight> CBuilding::TOWER_TYPES =
  40. {
  41. { "low", CBuilding::HEIGHT_LOW },
  42. { "average", CBuilding::HEIGHT_AVERAGE },
  43. { "high", CBuilding::HEIGHT_HIGH },
  44. { "skyship", CBuilding::HEIGHT_SKYSHIP }
  45. };
  46. std::string CBuilding::getJsonKey() const
  47. {
  48. return modScope + ':' + identifier;;
  49. }
  50. std::string CBuilding::getNameTranslated() const
  51. {
  52. return VLC->generaltexth->translate(getNameTextID());
  53. }
  54. std::string CBuilding::getDescriptionTranslated() const
  55. {
  56. return VLC->generaltexth->translate(getDescriptionTextID());
  57. }
  58. std::string CBuilding::getBaseTextID() const
  59. {
  60. return TextIdentifier("building", modScope, town->faction->identifier, identifier).get();
  61. }
  62. std::string CBuilding::getNameTextID() const
  63. {
  64. return TextIdentifier(getBaseTextID(), "name").get();
  65. }
  66. std::string CBuilding::getDescriptionTextID() const
  67. {
  68. return TextIdentifier(getBaseTextID(), "description").get();
  69. }
  70. BuildingID CBuilding::getBase() const
  71. {
  72. const CBuilding * build = this;
  73. while (build->upgrade != BuildingID::NONE)
  74. {
  75. build = build->town->buildings.at(build->upgrade);
  76. }
  77. return build->bid;
  78. }
  79. si32 CBuilding::getDistance(const BuildingID & buildID) const
  80. {
  81. const CBuilding * build = town->buildings.at(buildID);
  82. int distance = 0;
  83. while (build->upgrade != BuildingID::NONE && build != this)
  84. {
  85. build = build->town->buildings.at(build->upgrade);
  86. distance++;
  87. }
  88. if (build == this)
  89. return distance;
  90. return -1;
  91. }
  92. void CBuilding::addNewBonus(const std::shared_ptr<Bonus> & b, BonusList & bonusList) const
  93. {
  94. bonusList.push_back(b);
  95. }
  96. CFaction::~CFaction()
  97. {
  98. if (town)
  99. {
  100. delete town;
  101. town = nullptr;
  102. }
  103. }
  104. int32_t CFaction::getIndex() const
  105. {
  106. return index;
  107. }
  108. int32_t CFaction::getIconIndex() const
  109. {
  110. return index; //???
  111. }
  112. std::string CFaction::getJsonKey() const
  113. {
  114. return modScope + ':' + identifier;;
  115. }
  116. void CFaction::registerIcons(const IconRegistar & cb) const
  117. {
  118. if(town)
  119. {
  120. auto & info = town->clientInfo;
  121. cb(info.icons[0][0], 0, "ITPT", info.iconLarge[0][0]);
  122. cb(info.icons[0][1], 0, "ITPT", info.iconLarge[0][1]);
  123. cb(info.icons[1][0], 0, "ITPT", info.iconLarge[1][0]);
  124. cb(info.icons[1][1], 0, "ITPT", info.iconLarge[1][1]);
  125. cb(info.icons[0][0] + 2, 0, "ITPA", info.iconSmall[0][0]);
  126. cb(info.icons[0][1] + 2, 0, "ITPA", info.iconSmall[0][1]);
  127. cb(info.icons[1][0] + 2, 0, "ITPA", info.iconSmall[1][0]);
  128. cb(info.icons[1][1] + 2, 0, "ITPA", info.iconSmall[1][1]);
  129. cb(index, 1, "CPRSMALL", info.towerIconSmall);
  130. cb(index, 1, "TWCRPORT", info.towerIconLarge);
  131. }
  132. }
  133. std::string CFaction::getNameTranslated() const
  134. {
  135. return VLC->generaltexth->translate(getNameTextID());
  136. }
  137. std::string CFaction::getNameTextID() const
  138. {
  139. return TextIdentifier("faction", modScope, identifier, "name").get();
  140. }
  141. FactionID CFaction::getId() const
  142. {
  143. return FactionID(index);
  144. }
  145. FactionID CFaction::getFaction() const
  146. {
  147. return FactionID(index);
  148. }
  149. bool CFaction::hasTown() const
  150. {
  151. return town != nullptr;
  152. }
  153. EAlignment CFaction::getAlignment() const
  154. {
  155. return alignment;
  156. }
  157. BoatId CFaction::getBoatType() const
  158. {
  159. return boatType;
  160. }
  161. TerrainId CFaction::getNativeTerrain() const
  162. {
  163. return nativeTerrain;
  164. }
  165. void CFaction::updateFrom(const JsonNode & data)
  166. {
  167. }
  168. void CFaction::serializeJson(JsonSerializeFormat & handler)
  169. {
  170. }
  171. CTown::CTown()
  172. : faction(nullptr), mageLevel(0), primaryRes(0), moatAbility(SpellID::NONE), defaultTavernChance(0)
  173. {
  174. }
  175. CTown::~CTown()
  176. {
  177. for(auto & build : buildings)
  178. build.second.dellNull();
  179. for(auto & str : clientInfo.structures)
  180. str.dellNull();
  181. }
  182. std::string CTown::getRandomNameTranslated(size_t index) const
  183. {
  184. return VLC->generaltexth->translate(getRandomNameTextID(index));
  185. }
  186. std::string CTown::getRandomNameTextID(size_t index) const
  187. {
  188. return TextIdentifier("faction", faction->modScope, faction->identifier, "randomName", index).get();
  189. }
  190. size_t CTown::getRandomNamesCount() const
  191. {
  192. return namesCount;
  193. }
  194. std::string CTown::getBuildingScope() const
  195. {
  196. if(faction == nullptr)
  197. //no faction == random faction
  198. return "building";
  199. else
  200. return "building." + faction->getJsonKey();
  201. }
  202. std::set<si32> CTown::getAllBuildings() const
  203. {
  204. std::set<si32> res;
  205. for(const auto & b : buildings)
  206. {
  207. res.insert(b.first.num);
  208. }
  209. return res;
  210. }
  211. const CBuilding * CTown::getSpecialBuilding(BuildingSubID::EBuildingSubID subID) const
  212. {
  213. for(const auto & kvp : buildings)
  214. {
  215. if(kvp.second->subId == subID)
  216. return buildings.at(kvp.first);
  217. }
  218. return nullptr;
  219. }
  220. BuildingID CTown::getBuildingType(BuildingSubID::EBuildingSubID subID) const
  221. {
  222. const auto * building = getSpecialBuilding(subID);
  223. return building == nullptr ? BuildingID::NONE : building->bid.num;
  224. }
  225. std::string CTown::getGreeting(BuildingSubID::EBuildingSubID subID) const
  226. {
  227. return CTownHandler::getMappedValue<const std::string, BuildingSubID::EBuildingSubID>(subID, std::string(), specialMessages, false);
  228. }
  229. void CTown::setGreeting(BuildingSubID::EBuildingSubID subID, const std::string & message) const
  230. {
  231. specialMessages.insert(std::pair<BuildingSubID::EBuildingSubID, const std::string>(subID, message));
  232. }
  233. CTownHandler::CTownHandler():
  234. randomTown(new CTown()),
  235. randomFaction(new CFaction())
  236. {
  237. randomFaction->town = randomTown;
  238. randomTown->faction = randomFaction;
  239. randomFaction->identifier = "random";
  240. randomFaction->modScope = "core";
  241. }
  242. CTownHandler::~CTownHandler()
  243. {
  244. delete randomTown;
  245. }
  246. JsonNode readBuilding(CLegacyConfigParser & parser)
  247. {
  248. JsonNode ret;
  249. JsonNode & cost = ret["cost"];
  250. //note: this code will try to parse mithril as well but wil always return 0 for it
  251. for(const std::string & resID : GameConstants::RESOURCE_NAMES)
  252. cost[resID].Float() = parser.readNumber();
  253. cost.Struct().erase("mithril"); // erase mithril to avoid confusing validator
  254. parser.endLine();
  255. return ret;
  256. }
  257. TPropagatorPtr & CTownHandler::emptyPropagator()
  258. {
  259. static TPropagatorPtr emptyProp(nullptr);
  260. return emptyProp;
  261. }
  262. std::vector<JsonNode> CTownHandler::loadLegacyData()
  263. {
  264. size_t dataSize = VLC->settings()->getInteger(EGameSettings::TEXTS_FACTION);
  265. std::vector<JsonNode> dest(dataSize);
  266. objects.resize(dataSize);
  267. auto getBuild = [&](size_t town, size_t building) -> JsonNode &
  268. {
  269. return dest[town]["town"]["buildings"][EBuildingType::names[building]];
  270. };
  271. CLegacyConfigParser parser(TextPath::builtin("DATA/BUILDING.TXT"));
  272. parser.endLine(); // header
  273. parser.endLine();
  274. //Unique buildings
  275. for (size_t town=0; town<dataSize; town++)
  276. {
  277. parser.endLine(); //header
  278. parser.endLine();
  279. int buildID = 17;
  280. do
  281. {
  282. getBuild(town, buildID) = readBuilding(parser);
  283. buildID++;
  284. }
  285. while (!parser.isNextEntryEmpty());
  286. }
  287. // Common buildings
  288. parser.endLine(); // header
  289. parser.endLine();
  290. parser.endLine();
  291. int buildID = 0;
  292. do
  293. {
  294. JsonNode building = readBuilding(parser);
  295. for (size_t town=0; town<dataSize; town++)
  296. getBuild(town, buildID) = building;
  297. buildID++;
  298. }
  299. while (!parser.isNextEntryEmpty());
  300. parser.endLine(); //header
  301. parser.endLine();
  302. //Dwellings
  303. for (size_t town=0; town<dataSize; town++)
  304. {
  305. parser.endLine(); //header
  306. parser.endLine();
  307. for (size_t i=0; i<14; i++)
  308. {
  309. getBuild(town, 30+i) = readBuilding(parser);
  310. }
  311. }
  312. {
  313. CLegacyConfigParser parser(TextPath::builtin("DATA/BLDGNEUT.TXT"));
  314. for(int building=0; building<15; building++)
  315. {
  316. std::string name = parser.readString();
  317. std::string descr = parser.readString();
  318. parser.endLine();
  319. for(int j=0; j<dataSize; j++)
  320. {
  321. getBuild(j, building)["name"].String() = name;
  322. getBuild(j, building)["description"].String() = descr;
  323. }
  324. }
  325. parser.endLine(); // silo
  326. parser.endLine(); // blacksmith //unused entries
  327. parser.endLine(); // moat
  328. //shipyard with the ship
  329. std::string name = parser.readString();
  330. std::string descr = parser.readString();
  331. parser.endLine();
  332. for(int town=0; town<dataSize; town++)
  333. {
  334. getBuild(town, 20)["name"].String() = name;
  335. getBuild(town, 20)["description"].String() = descr;
  336. }
  337. //blacksmith
  338. for(int town=0; town<dataSize; town++)
  339. {
  340. getBuild(town, 16)["name"].String() = parser.readString();
  341. getBuild(town, 16)["description"].String() = parser.readString();
  342. parser.endLine();
  343. }
  344. }
  345. {
  346. CLegacyConfigParser parser(TextPath::builtin("DATA/BLDGSPEC.TXT"));
  347. for(int town=0; town<dataSize; town++)
  348. {
  349. for(int build=0; build<9; build++)
  350. {
  351. getBuild(town, 17 + build)["name"].String() = parser.readString();
  352. getBuild(town, 17 + build)["description"].String() = parser.readString();
  353. parser.endLine();
  354. }
  355. getBuild(town, 26)["name"].String() = parser.readString(); // Grail
  356. getBuild(town, 26)["description"].String() = parser.readString();
  357. parser.endLine();
  358. getBuild(town, 15)["name"].String() = parser.readString(); // Resource silo
  359. getBuild(town, 15)["description"].String() = parser.readString();
  360. parser.endLine();
  361. }
  362. }
  363. {
  364. CLegacyConfigParser parser(TextPath::builtin("DATA/DWELLING.TXT"));
  365. for(int town=0; town<dataSize; town++)
  366. {
  367. for(int build=0; build<14; build++)
  368. {
  369. getBuild(town, 30 + build)["name"].String() = parser.readString();
  370. getBuild(town, 30 + build)["description"].String() = parser.readString();
  371. parser.endLine();
  372. }
  373. }
  374. }
  375. {
  376. CLegacyConfigParser typeParser(TextPath::builtin("DATA/TOWNTYPE.TXT"));
  377. CLegacyConfigParser nameParser(TextPath::builtin("DATA/TOWNNAME.TXT"));
  378. size_t townID=0;
  379. do
  380. {
  381. dest[townID]["name"].String() = typeParser.readString();
  382. for (int i=0; i<NAMES_PER_TOWN; i++)
  383. {
  384. JsonNode name;
  385. name.String() = nameParser.readString();
  386. dest[townID]["town"]["names"].Vector().push_back(name);
  387. nameParser.endLine();
  388. }
  389. townID++;
  390. }
  391. while (typeParser.endLine());
  392. }
  393. return dest;
  394. }
  395. void CTownHandler::loadBuildingRequirements(CBuilding * building, const JsonNode & source, std::vector<BuildingRequirementsHelper> & bidsToLoad) const
  396. {
  397. if (source.isNull())
  398. return;
  399. BuildingRequirementsHelper hlp;
  400. hlp.building = building;
  401. hlp.town = building->town;
  402. hlp.json = source;
  403. bidsToLoad.push_back(hlp);
  404. }
  405. template<typename R, typename K>
  406. R CTownHandler::getMappedValue(const K key, const R defval, const std::map<K, R> & map, bool required)
  407. {
  408. auto it = map.find(key);
  409. if(it != map.end())
  410. return it->second;
  411. if(required)
  412. logMod->warn("Warning: Property: '%s' is unknown. Correct the typo or update VCMI.", key);
  413. return defval;
  414. }
  415. template<typename R>
  416. R CTownHandler::getMappedValue(const JsonNode & node, const R defval, const std::map<std::string, R> & map, bool required)
  417. {
  418. if(!node.isNull() && node.getType() == JsonNode::JsonType::DATA_STRING)
  419. return getMappedValue<R, std::string>(node.String(), defval, map, required);
  420. return defval;
  421. }
  422. void CTownHandler::addBonusesForVanilaBuilding(CBuilding * building) const
  423. {
  424. std::shared_ptr<Bonus> b;
  425. static TPropagatorPtr playerPropagator = std::make_shared<CPropagatorNodeType>(CBonusSystemNode::ENodeTypes::PLAYER);
  426. if(building->bid == BuildingID::TAVERN)
  427. {
  428. b = createBonus(building, BonusType::MORALE, +1);
  429. }
  430. switch(building->subId)
  431. {
  432. case BuildingSubID::BROTHERHOOD_OF_SWORD:
  433. b = createBonus(building, BonusType::MORALE, +2);
  434. building->overrideBids.insert(BuildingID::TAVERN);
  435. break;
  436. case BuildingSubID::FOUNTAIN_OF_FORTUNE:
  437. b = createBonus(building, BonusType::LUCK, +2);
  438. break;
  439. case BuildingSubID::SPELL_POWER_GARRISON_BONUS:
  440. b = createBonus(building, BonusType::PRIMARY_SKILL, +2, static_cast<int>(PrimarySkill::SPELL_POWER));
  441. break;
  442. case BuildingSubID::ATTACK_GARRISON_BONUS:
  443. b = createBonus(building, BonusType::PRIMARY_SKILL, +2, static_cast<int>(PrimarySkill::ATTACK));
  444. break;
  445. case BuildingSubID::DEFENSE_GARRISON_BONUS:
  446. b = createBonus(building, BonusType::PRIMARY_SKILL, +2, static_cast<int>(PrimarySkill::DEFENSE));
  447. break;
  448. case BuildingSubID::LIGHTHOUSE:
  449. b = createBonus(building, BonusType::MOVEMENT, +500, playerPropagator, 0);
  450. break;
  451. }
  452. if(b)
  453. building->addNewBonus(b, building->buildingBonuses);
  454. }
  455. std::shared_ptr<Bonus> CTownHandler::createBonus(CBuilding * build, BonusType type, int val, int subtype) const
  456. {
  457. return createBonus(build, type, val, emptyPropagator(), subtype);
  458. }
  459. std::shared_ptr<Bonus> CTownHandler::createBonus(CBuilding * build, BonusType type, int val, TPropagatorPtr & prop, int subtype) const
  460. {
  461. std::ostringstream descr;
  462. descr << build->getNameTranslated();
  463. return createBonusImpl(build->bid, type, val, prop, descr.str(), subtype);
  464. }
  465. std::shared_ptr<Bonus> CTownHandler::createBonusImpl(const BuildingID & building,
  466. BonusType type,
  467. int val,
  468. TPropagatorPtr & prop,
  469. const std::string & description,
  470. int subtype) const
  471. {
  472. auto b = std::make_shared<Bonus>(BonusDuration::PERMANENT, type, BonusSource::TOWN_STRUCTURE, val, building, description, subtype);
  473. if(prop)
  474. b->addPropagator(prop);
  475. return b;
  476. }
  477. void CTownHandler::loadSpecialBuildingBonuses(const JsonNode & source, BonusList & bonusList, CBuilding * building)
  478. {
  479. for(const auto & b : source.Vector())
  480. {
  481. auto bonus = JsonUtils::parseBuildingBonus(b, building->bid, building->getNameTranslated());
  482. if(bonus == nullptr)
  483. continue;
  484. bonus->sid = Bonus::getSid32(building->town->faction->getIndex(), building->bid);
  485. //JsonUtils::parseBuildingBonus produces UNKNOWN type propagator instead of empty.
  486. if(bonus->propagator != nullptr
  487. && bonus->propagator->getPropagatorType() == CBonusSystemNode::ENodeTypes::UNKNOWN)
  488. bonus->addPropagator(emptyPropagator());
  489. building->addNewBonus(bonus, bonusList);
  490. }
  491. }
  492. void CTownHandler::loadBuilding(CTown * town, const std::string & stringID, const JsonNode & source)
  493. {
  494. assert(stringID.find(':') == std::string::npos);
  495. assert(!source.meta.empty());
  496. auto * ret = new CBuilding();
  497. ret->bid = getMappedValue<BuildingID, std::string>(stringID, BuildingID::NONE, MappedKeys::BUILDING_NAMES_TO_TYPES, false);
  498. ret->subId = BuildingSubID::NONE;
  499. if(ret->bid == BuildingID::NONE && !source["id"].isNull())
  500. {
  501. // FIXME: A lot of false-positives with no clear way to handle them in mods
  502. //logMod->warn("Building %s: id field is deprecated", stringID);
  503. ret->bid = source["id"].isNull() ? BuildingID(BuildingID::NONE) : BuildingID(source["id"].Float());
  504. }
  505. if (ret->bid == BuildingID::NONE)
  506. logMod->error("Building '%s' isn't recognized and won't work properly. Correct the typo or update VCMI.", stringID);
  507. ret->mode = ret->bid == BuildingID::GRAIL
  508. ? CBuilding::BUILD_GRAIL
  509. : getMappedValue<CBuilding::EBuildMode>(source["mode"], CBuilding::BUILD_NORMAL, CBuilding::MODES);
  510. ret->height = getMappedValue<CBuilding::ETowerHeight>(source["height"], CBuilding::HEIGHT_NO_TOWER, CBuilding::TOWER_TYPES);
  511. ret->identifier = stringID;
  512. ret->modScope = source.meta;
  513. ret->town = town;
  514. VLC->generaltexth->registerString(source.meta, ret->getNameTextID(), source["name"].String());
  515. VLC->generaltexth->registerString(source.meta, ret->getDescriptionTextID(), source["description"].String());
  516. ret->resources = TResources(source["cost"]);
  517. ret->produce = TResources(source["produce"]);
  518. if(ret->bid == BuildingID::TAVERN)
  519. addBonusesForVanilaBuilding(ret);
  520. else if(ret->bid.IsSpecialOrGrail())
  521. {
  522. loadSpecialBuildingBonuses(source["bonuses"], ret->buildingBonuses, ret);
  523. if(ret->buildingBonuses.empty())
  524. {
  525. ret->subId = getMappedValue<BuildingSubID::EBuildingSubID>(source["type"], BuildingSubID::NONE, MappedKeys::SPECIAL_BUILDINGS);
  526. addBonusesForVanilaBuilding(ret);
  527. }
  528. loadSpecialBuildingBonuses(source["onVisitBonuses"], ret->onVisitBonuses, ret);
  529. if(!ret->onVisitBonuses.empty())
  530. {
  531. if(ret->subId == BuildingSubID::NONE)
  532. ret->subId = BuildingSubID::CUSTOM_VISITING_BONUS;
  533. for(auto & bonus : ret->onVisitBonuses)
  534. bonus->sid = Bonus::getSid32(ret->town->faction->getIndex(), ret->bid);
  535. }
  536. if(source["type"].String() == "configurable" && ret->subId == BuildingSubID::NONE)
  537. {
  538. ret->subId = BuildingSubID::CUSTOM_VISITING_REWARD;
  539. ret->rewardableObjectInfo.init(source, ret->getBaseTextID());
  540. }
  541. }
  542. //MODS COMPATIBILITY FOR 0.96
  543. if(!ret->produce.nonZero())
  544. {
  545. switch (ret->bid) {
  546. break; case BuildingID::VILLAGE_HALL: ret->produce[EGameResID::GOLD] = 500;
  547. break; case BuildingID::TOWN_HALL : ret->produce[EGameResID::GOLD] = 1000;
  548. break; case BuildingID::CITY_HALL : ret->produce[EGameResID::GOLD] = 2000;
  549. break; case BuildingID::CAPITOL : ret->produce[EGameResID::GOLD] = 4000;
  550. break; case BuildingID::GRAIL : ret->produce[EGameResID::GOLD] = 5000;
  551. break; case BuildingID::RESOURCE_SILO :
  552. {
  553. switch (ret->town->primaryRes.toEnum())
  554. {
  555. case EGameResID::GOLD:
  556. ret->produce[ret->town->primaryRes] = 500;
  557. break;
  558. case EGameResID::WOOD_AND_ORE:
  559. ret->produce[EGameResID::WOOD] = 1;
  560. ret->produce[EGameResID::ORE] = 1;
  561. break;
  562. default:
  563. ret->produce[ret->town->primaryRes] = 1;
  564. break;
  565. }
  566. }
  567. }
  568. }
  569. loadBuildingRequirements(ret, source["requires"], requirementsToLoad);
  570. if(ret->bid.IsSpecialOrGrail())
  571. loadBuildingRequirements(ret, source["overrides"], overriddenBidsToLoad);
  572. if (!source["upgrades"].isNull())
  573. {
  574. // building id and upgrades can't be the same
  575. if(stringID == source["upgrades"].String())
  576. {
  577. throw std::runtime_error(boost::str(boost::format("Building with ID '%s' of town '%s' can't be an upgrade of the same building.") %
  578. stringID % ret->town->faction->getNameTranslated()));
  579. }
  580. VLC->identifiers()->requestIdentifier(ret->town->getBuildingScope(), source["upgrades"], [=](si32 identifier)
  581. {
  582. ret->upgrade = BuildingID(identifier);
  583. });
  584. }
  585. else
  586. ret->upgrade = BuildingID::NONE;
  587. ret->town->buildings[ret->bid] = ret;
  588. registerObject(source.meta, ret->town->getBuildingScope(), ret->identifier, ret->bid);
  589. }
  590. void CTownHandler::loadBuildings(CTown * town, const JsonNode & source)
  591. {
  592. if(source.isStruct())
  593. {
  594. for(const auto & node : source.Struct())
  595. {
  596. if (!node.second.isNull())
  597. loadBuilding(town, node.first, node.second);
  598. }
  599. }
  600. }
  601. void CTownHandler::loadStructure(CTown &town, const std::string & stringID, const JsonNode & source) const
  602. {
  603. auto * ret = new CStructure();
  604. ret->building = nullptr;
  605. ret->buildable = nullptr;
  606. VLC->identifiers()->tryRequestIdentifier( source.meta, "building." + town.faction->getJsonKey(), stringID, [=, &town](si32 identifier) mutable
  607. {
  608. ret->building = town.buildings[BuildingID(identifier)];
  609. });
  610. if (source["builds"].isNull())
  611. {
  612. VLC->identifiers()->tryRequestIdentifier( source.meta, "building." + town.faction->getJsonKey(), stringID, [=, &town](si32 identifier) mutable
  613. {
  614. ret->building = town.buildings[BuildingID(identifier)];
  615. });
  616. }
  617. else
  618. {
  619. VLC->identifiers()->requestIdentifier("building." + town.faction->getJsonKey(), source["builds"], [=, &town](si32 identifier) mutable
  620. {
  621. ret->buildable = town.buildings[BuildingID(identifier)];
  622. });
  623. }
  624. ret->identifier = stringID;
  625. ret->pos.x = static_cast<si32>(source["x"].Float());
  626. ret->pos.y = static_cast<si32>(source["y"].Float());
  627. ret->pos.z = static_cast<si32>(source["z"].Float());
  628. ret->hiddenUpgrade = source["hidden"].Bool();
  629. ret->defName = AnimationPath::fromJson(source["animation"]);
  630. ret->borderName = ImagePath::fromJson(source["border"]);
  631. ret->areaName = ImagePath::fromJson(source["area"]);
  632. town.clientInfo.structures.emplace_back(ret);
  633. }
  634. void CTownHandler::loadStructures(CTown &town, const JsonNode & source) const
  635. {
  636. for(const auto & node : source.Struct())
  637. {
  638. if (!node.second.isNull())
  639. loadStructure(town, node.first, node.second);
  640. }
  641. }
  642. void CTownHandler::loadTownHall(CTown &town, const JsonNode & source) const
  643. {
  644. auto & dstSlots = town.clientInfo.hallSlots;
  645. const auto & srcSlots = source.Vector();
  646. dstSlots.resize(srcSlots.size());
  647. for(size_t i=0; i<dstSlots.size(); i++)
  648. {
  649. auto & dstRow = dstSlots[i];
  650. const auto & srcRow = srcSlots[i].Vector();
  651. dstRow.resize(srcRow.size());
  652. for(size_t j=0; j < dstRow.size(); j++)
  653. {
  654. auto & dstBox = dstRow[j];
  655. const auto & srcBox = srcRow[j].Vector();
  656. dstBox.resize(srcBox.size());
  657. for(size_t k=0; k<dstBox.size(); k++)
  658. {
  659. auto & dst = dstBox[k];
  660. const auto & src = srcBox[k];
  661. VLC->identifiers()->requestIdentifier("building." + town.faction->getJsonKey(), src, [&](si32 identifier)
  662. {
  663. dst = BuildingID(identifier);
  664. });
  665. }
  666. }
  667. }
  668. }
  669. Point JsonToPoint(const JsonNode & node)
  670. {
  671. if(!node.isStruct())
  672. return Point::makeInvalid();
  673. Point ret;
  674. ret.x = static_cast<si32>(node["x"].Float());
  675. ret.y = static_cast<si32>(node["y"].Float());
  676. return ret;
  677. }
  678. void CTownHandler::loadSiegeScreen(CTown &town, const JsonNode & source) const
  679. {
  680. town.clientInfo.siegePrefix = source["imagePrefix"].String();
  681. town.clientInfo.towerIconSmall = source["towerIconSmall"].String();
  682. town.clientInfo.towerIconLarge = source["towerIconLarge"].String();
  683. VLC->identifiers()->requestIdentifier("creature", source["shooter"], [&town](si32 creature)
  684. {
  685. auto crId = CreatureID(creature);
  686. if((*VLC->creh)[crId]->animation.missleFrameAngles.empty())
  687. logMod->error("Mod '%s' error: Creature '%s' on the Archer's tower is not a shooter. Mod should be fixed. Siege will not work properly!"
  688. , town.faction->getNameTranslated()
  689. , (*VLC->creh)[crId]->getNameSingularTranslated());
  690. town.clientInfo.siegeShooter = crId;
  691. });
  692. auto & pos = town.clientInfo.siegePositions;
  693. pos.resize(21);
  694. pos[8] = JsonToPoint(source["towers"]["top"]["tower"]);
  695. pos[17] = JsonToPoint(source["towers"]["top"]["battlement"]);
  696. pos[20] = JsonToPoint(source["towers"]["top"]["creature"]);
  697. pos[2] = JsonToPoint(source["towers"]["keep"]["tower"]);
  698. pos[15] = JsonToPoint(source["towers"]["keep"]["battlement"]);
  699. pos[18] = JsonToPoint(source["towers"]["keep"]["creature"]);
  700. pos[3] = JsonToPoint(source["towers"]["bottom"]["tower"]);
  701. pos[16] = JsonToPoint(source["towers"]["bottom"]["battlement"]);
  702. pos[19] = JsonToPoint(source["towers"]["bottom"]["creature"]);
  703. pos[9] = JsonToPoint(source["gate"]["gate"]);
  704. pos[10] = JsonToPoint(source["gate"]["arch"]);
  705. pos[7] = JsonToPoint(source["walls"]["upper"]);
  706. pos[6] = JsonToPoint(source["walls"]["upperMid"]);
  707. pos[5] = JsonToPoint(source["walls"]["bottomMid"]);
  708. pos[4] = JsonToPoint(source["walls"]["bottom"]);
  709. pos[13] = JsonToPoint(source["moat"]["moat"]);
  710. pos[14] = JsonToPoint(source["moat"]["bank"]);
  711. pos[11] = JsonToPoint(source["static"]["bottom"]);
  712. pos[12] = JsonToPoint(source["static"]["top"]);
  713. pos[1] = JsonToPoint(source["static"]["background"]);
  714. }
  715. static void readIcon(JsonNode source, std::string & small, std::string & large)
  716. {
  717. if (source.getType() == JsonNode::JsonType::DATA_STRUCT) // don't crash on old format
  718. {
  719. small = source["small"].String();
  720. large = source["large"].String();
  721. }
  722. }
  723. void CTownHandler::loadClientData(CTown &town, const JsonNode & source) const
  724. {
  725. CTown::ClientInfo & info = town.clientInfo;
  726. readIcon(source["icons"]["village"]["normal"], info.iconSmall[0][0], info.iconLarge[0][0]);
  727. readIcon(source["icons"]["village"]["built"], info.iconSmall[0][1], info.iconLarge[0][1]);
  728. readIcon(source["icons"]["fort"]["normal"], info.iconSmall[1][0], info.iconLarge[1][0]);
  729. readIcon(source["icons"]["fort"]["built"], info.iconSmall[1][1], info.iconLarge[1][1]);
  730. info.hallBackground = ImagePath::fromJson(source["hallBackground"]);
  731. info.musicTheme = source["musicTheme"].String();
  732. info.townBackground = ImagePath::fromJson(source["townBackground"]);
  733. info.guildWindow = ImagePath::fromJson(source["guildWindow"]);
  734. info.buildingsIcons = AnimationPath::fromJson(source["buildingsIcons"]);
  735. info.guildBackground = ImagePath::fromJson(source["guildBackground"]);
  736. info.tavernVideo = VideoPath::fromJson(source["tavernVideo"]);
  737. loadTownHall(town, source["hallSlots"]);
  738. loadStructures(town, source["structures"]);
  739. loadSiegeScreen(town, source["siege"]);
  740. }
  741. void CTownHandler::loadTown(CTown * town, const JsonNode & source)
  742. {
  743. const auto * resIter = boost::find(GameConstants::RESOURCE_NAMES, source["primaryResource"].String());
  744. if(resIter == std::end(GameConstants::RESOURCE_NAMES))
  745. town->primaryRes = GameResID(EGameResID::WOOD_AND_ORE); //Wood + Ore
  746. else
  747. town->primaryRes = GameResID(resIter - std::begin(GameConstants::RESOURCE_NAMES));
  748. warMachinesToLoad[town] = source["warMachine"];
  749. town->mageLevel = static_cast<ui32>(source["mageGuild"].Float());
  750. town->namesCount = 0;
  751. for(const auto & name : source["names"].Vector())
  752. {
  753. VLC->generaltexth->registerString(town->faction->modScope, town->getRandomNameTextID(town->namesCount), name.String());
  754. town->namesCount += 1;
  755. }
  756. if (!source["moatAbility"].isNull()) // VCMI 1.2 compatibility code
  757. {
  758. VLC->identifiers()->requestIdentifier( "spell", source["moatAbility"], [=](si32 ability)
  759. {
  760. town->moatAbility = SpellID(ability);
  761. });
  762. }
  763. else
  764. {
  765. VLC->identifiers()->requestIdentifier( source.meta, "spell", "castleMoat", [=](si32 ability)
  766. {
  767. town->moatAbility = SpellID(ability);
  768. });
  769. }
  770. // Horde building creature level
  771. for(const JsonNode &node : source["horde"].Vector())
  772. town->hordeLvl[static_cast<int>(town->hordeLvl.size())] = static_cast<int>(node.Float());
  773. // town needs to have exactly 2 horde entries. Validation will take care of 2+ entries
  774. // but anything below 2 must be handled here
  775. for (size_t i=source["horde"].Vector().size(); i<2; i++)
  776. town->hordeLvl[static_cast<int>(i)] = -1;
  777. const JsonVector & creatures = source["creatures"].Vector();
  778. town->creatures.resize(creatures.size());
  779. for (size_t i=0; i< creatures.size(); i++)
  780. {
  781. const JsonVector & level = creatures[i].Vector();
  782. town->creatures[i].resize(level.size());
  783. for (size_t j=0; j<level.size(); j++)
  784. {
  785. VLC->identifiers()->requestIdentifier("creature", level[j], [=](si32 creature)
  786. {
  787. town->creatures[i][j] = CreatureID(creature);
  788. });
  789. }
  790. }
  791. town->defaultTavernChance = static_cast<ui32>(source["defaultTavern"].Float());
  792. /// set chance of specific hero class to appear in this town
  793. for(const auto & node : source["tavern"].Struct())
  794. {
  795. int chance = static_cast<int>(node.second.Float());
  796. VLC->identifiers()->requestIdentifier(node.second.meta, "heroClass",node.first, [=](si32 classID)
  797. {
  798. VLC->heroh->classes[HeroClassID(classID)]->selectionProbability[town->faction->getId()] = chance;
  799. });
  800. }
  801. for(const auto & node : source["guildSpells"].Struct())
  802. {
  803. int chance = static_cast<int>(node.second.Float());
  804. VLC->identifiers()->requestIdentifier(node.second.meta, "spell", node.first, [=](si32 spellID)
  805. {
  806. VLC->spellh->objects.at(spellID)->probabilities[town->faction->getId()] = chance;
  807. });
  808. }
  809. for(const JsonNode & d : source["adventureMap"]["dwellings"].Vector())
  810. {
  811. town->dwellings.push_back(d["graphics"].String());
  812. town->dwellingNames.push_back(d["name"].String());
  813. }
  814. loadBuildings(town, source["buildings"]);
  815. loadClientData(*town, source);
  816. }
  817. void CTownHandler::loadPuzzle(CFaction &faction, const JsonNode &source) const
  818. {
  819. faction.puzzleMap.reserve(GameConstants::PUZZLE_MAP_PIECES);
  820. std::string prefix = source["prefix"].String();
  821. for(const JsonNode &piece : source["pieces"].Vector())
  822. {
  823. size_t index = faction.puzzleMap.size();
  824. SPuzzleInfo spi;
  825. spi.x = static_cast<si16>(piece["x"].Float());
  826. spi.y = static_cast<si16>(piece["y"].Float());
  827. spi.whenUncovered = static_cast<ui16>(piece["index"].Float());
  828. spi.number = static_cast<ui16>(index);
  829. // filename calculation
  830. std::ostringstream suffix;
  831. suffix << std::setfill('0') << std::setw(2) << index;
  832. spi.filename = ImagePath::builtinTODO(prefix + suffix.str());
  833. faction.puzzleMap.push_back(spi);
  834. }
  835. assert(faction.puzzleMap.size() == GameConstants::PUZZLE_MAP_PIECES);
  836. }
  837. CFaction * CTownHandler::loadFromJson(const std::string & scope, const JsonNode & source, const std::string & identifier, size_t index)
  838. {
  839. assert(identifier.find(':') == std::string::npos);
  840. auto * faction = new CFaction();
  841. faction->index = static_cast<FactionID>(index);
  842. faction->modScope = scope;
  843. faction->identifier = identifier;
  844. VLC->generaltexth->registerString(scope, faction->getNameTextID(), source["name"].String());
  845. faction->creatureBg120 = ImagePath::fromJson(source["creatureBackground"]["120px"]);
  846. faction->creatureBg130 = ImagePath::fromJson(source["creatureBackground"]["130px"]);
  847. faction->boatType = BoatId::CASTLE; //Do not crash
  848. if (!source["boat"].isNull())
  849. {
  850. VLC->identifiers()->requestIdentifier("core:boat", source["boat"], [=](int32_t boatTypeID)
  851. {
  852. faction->boatType = BoatId(boatTypeID);
  853. });
  854. }
  855. int alignment = vstd::find_pos(GameConstants::ALIGNMENT_NAMES, source["alignment"].String());
  856. if (alignment == -1)
  857. faction->alignment = EAlignment::NEUTRAL;
  858. else
  859. faction->alignment = static_cast<EAlignment>(alignment);
  860. auto preferUndergound = source["preferUndergroundPlacement"];
  861. faction->preferUndergroundPlacement = preferUndergound.isNull() ? false : preferUndergound.Bool();
  862. // NOTE: semi-workaround - normally, towns are supposed to have native terrains.
  863. // Towns without one are exceptions. So, vcmi requires nativeTerrain to be defined
  864. // But allows it to be defined with explicit value of "none" if town should not have native terrain
  865. // This is better than allowing such terrain-less towns silently, leading to issues with RMG
  866. faction->nativeTerrain = ETerrainId::NONE;
  867. if ( !source["nativeTerrain"].isNull() && source["nativeTerrain"].String() != "none")
  868. {
  869. VLC->identifiers()->requestIdentifier("terrain", source["nativeTerrain"], [=](int32_t index){
  870. faction->nativeTerrain = TerrainId(index);
  871. auto const & terrain = VLC->terrainTypeHandler->getById(faction->nativeTerrain);
  872. if (!terrain->isSurface() && !terrain->isUnderground())
  873. logMod->warn("Faction %s has terrain %s as native, but terrain is not suitable for either surface or subterranean layers!", faction->getJsonKey(), terrain->getJsonKey());
  874. });
  875. }
  876. if (!source["town"].isNull())
  877. {
  878. faction->town = new CTown();
  879. faction->town->faction = faction;
  880. loadTown(faction->town, source["town"]);
  881. }
  882. else
  883. faction->town = nullptr;
  884. if (!source["puzzleMap"].isNull())
  885. loadPuzzle(*faction, source["puzzleMap"]);
  886. return faction;
  887. }
  888. void CTownHandler::loadObject(std::string scope, std::string name, const JsonNode & data)
  889. {
  890. auto * object = loadFromJson(scope, data, name, objects.size());
  891. objects.emplace_back(object);
  892. if (object->town)
  893. {
  894. auto & info = object->town->clientInfo;
  895. info.icons[0][0] = 8 + object->index * 4 + 0;
  896. info.icons[0][1] = 8 + object->index * 4 + 1;
  897. info.icons[1][0] = 8 + object->index * 4 + 2;
  898. info.icons[1][1] = 8 + object->index * 4 + 3;
  899. VLC->identifiers()->requestIdentifier(scope, "object", "town", [=](si32 index)
  900. {
  901. // register town once objects are loaded
  902. JsonNode config = data["town"]["mapObject"];
  903. config["faction"].String() = name;
  904. config["faction"].meta = scope;
  905. if (config.meta.empty())// MODS COMPATIBILITY FOR 0.96
  906. config.meta = scope;
  907. VLC->objtypeh->loadSubObject(object->identifier, config, index, object->index);
  908. // MODS COMPATIBILITY FOR 0.96
  909. const auto & advMap = data["town"]["adventureMap"];
  910. if (!advMap.isNull())
  911. {
  912. logMod->warn("Outdated town mod. Will try to generate valid templates out of fort");
  913. JsonNode config;
  914. config["animation"] = advMap["castle"];
  915. VLC->objtypeh->getHandlerFor(index, object->index)->addTemplate(config);
  916. }
  917. });
  918. }
  919. registerObject(scope, "faction", name, object->index);
  920. }
  921. void CTownHandler::loadObject(std::string scope, std::string name, const JsonNode & data, size_t index)
  922. {
  923. auto * object = loadFromJson(scope, data, name, index);
  924. if (objects.size() > index)
  925. assert(objects[index] == nullptr); // ensure that this id was not loaded before
  926. else
  927. objects.resize(index + 1);
  928. objects[index] = object;
  929. if (object->town)
  930. {
  931. auto & info = object->town->clientInfo;
  932. info.icons[0][0] = (GameConstants::F_NUMBER + object->index) * 2 + 0;
  933. info.icons[0][1] = (GameConstants::F_NUMBER + object->index) * 2 + 1;
  934. info.icons[1][0] = object->index * 2 + 0;
  935. info.icons[1][1] = object->index * 2 + 1;
  936. VLC->identifiers()->requestIdentifier(scope, "object", "town", [=](si32 index)
  937. {
  938. // register town once objects are loaded
  939. JsonNode config = data["town"]["mapObject"];
  940. config["faction"].String() = name;
  941. config["faction"].meta = scope;
  942. VLC->objtypeh->loadSubObject(object->identifier, config, index, object->index);
  943. });
  944. }
  945. registerObject(scope, "faction", name, object->index);
  946. }
  947. void CTownHandler::loadRandomFaction()
  948. {
  949. JsonNode randomFactionJson(JsonPath::builtin("config/factions/random.json"));
  950. randomFactionJson.setMeta(ModScope::scopeBuiltin(), true);
  951. loadBuildings(randomTown, randomFactionJson["random"]["town"]["buildings"]);
  952. }
  953. void CTownHandler::loadCustom()
  954. {
  955. loadRandomFaction();
  956. }
  957. void CTownHandler::afterLoadFinalization()
  958. {
  959. initializeRequirements();
  960. initializeOverridden();
  961. initializeWarMachines();
  962. }
  963. void CTownHandler::initializeRequirements()
  964. {
  965. // must be done separately after all ID's are known
  966. for (auto & requirement : requirementsToLoad)
  967. {
  968. requirement.building->requirements = CBuilding::TRequired(requirement.json, [&](const JsonNode & node) -> BuildingID
  969. {
  970. if (node.Vector().size() > 1)
  971. {
  972. logMod->error("Unexpected length of town buildings requirements: %d", node.Vector().size());
  973. logMod->error("Entry contains: ");
  974. logMod->error(node.toJson());
  975. }
  976. auto index = VLC->identifiers()->getIdentifier(requirement.town->getBuildingScope(), node[0]);
  977. if (!index.has_value())
  978. {
  979. logMod->error("Unknown building in town buildings: %s", node[0].String());
  980. return BuildingID::NONE;
  981. }
  982. return BuildingID(index.value());
  983. });
  984. }
  985. requirementsToLoad.clear();
  986. }
  987. void CTownHandler::initializeOverridden()
  988. {
  989. for(auto & bidHelper : overriddenBidsToLoad)
  990. {
  991. auto jsonNode = bidHelper.json;
  992. auto scope = bidHelper.town->getBuildingScope();
  993. for(const auto & b : jsonNode.Vector())
  994. {
  995. auto bid = BuildingID(VLC->identifiers()->getIdentifier(scope, b).value());
  996. bidHelper.building->overrideBids.insert(bid);
  997. }
  998. }
  999. overriddenBidsToLoad.clear();
  1000. }
  1001. void CTownHandler::initializeWarMachines()
  1002. {
  1003. // must be done separately after all objects are loaded
  1004. for(auto & p : warMachinesToLoad)
  1005. {
  1006. CTown * t = p.first;
  1007. JsonNode creatureKey = p.second;
  1008. auto ret = VLC->identifiers()->getIdentifier("creature", creatureKey, false);
  1009. if(ret)
  1010. {
  1011. const CCreature * creature = CreatureID(*ret).toCreature();
  1012. t->warMachine = creature->warMachine;
  1013. }
  1014. }
  1015. warMachinesToLoad.clear();
  1016. }
  1017. std::vector<bool> CTownHandler::getDefaultAllowed() const
  1018. {
  1019. std::vector<bool> allowedFactions;
  1020. allowedFactions.reserve(objects.size());
  1021. for(auto town : objects)
  1022. {
  1023. allowedFactions.push_back(town->town != nullptr);
  1024. }
  1025. return allowedFactions;
  1026. }
  1027. std::set<FactionID> CTownHandler::getAllowedFactions(bool withTown) const
  1028. {
  1029. std::set<FactionID> allowedFactions;
  1030. std::vector<bool> allowed;
  1031. if (withTown)
  1032. allowed = getDefaultAllowed();
  1033. else
  1034. allowed.resize( objects.size(), true);
  1035. for (size_t i=0; i<allowed.size(); i++)
  1036. if (allowed[i])
  1037. allowedFactions.insert(static_cast<FactionID>(i));
  1038. return allowedFactions;
  1039. }
  1040. const std::vector<std::string> & CTownHandler::getTypeNames() const
  1041. {
  1042. static const std::vector<std::string> typeNames = { "faction", "town" };
  1043. return typeNames;
  1044. }
  1045. VCMI_LIB_NAMESPACE_END