CRmgTemplate.cpp 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910
  1. /*
  2. * CRmgTemplate.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 <vstd/ContainerUtils.h>
  12. #include <boost/bimap.hpp>
  13. #include "CRmgTemplate.h"
  14. #include "Functions.h"
  15. #include "../TerrainHandler.h"
  16. #include "../VCMI_Lib.h"
  17. #include "../constants/StringConstants.h"
  18. #include "../entities/faction/CTownHandler.h"
  19. #include "../modding/ModScope.h"
  20. #include "../serializer/JsonSerializeFormat.h"
  21. VCMI_LIB_NAMESPACE_BEGIN
  22. namespace
  23. {
  24. si32 decodeZoneId(const std::string & json)
  25. {
  26. return boost::lexical_cast<si32>(json);
  27. }
  28. std::string encodeZoneId(si32 id)
  29. {
  30. return std::to_string(id);
  31. }
  32. }
  33. CTreasureInfo::CTreasureInfo()
  34. : min(0),
  35. max(0),
  36. density(0)
  37. {
  38. }
  39. CTreasureInfo::CTreasureInfo(ui32 imin, ui32 imax, ui16 idensity)
  40. : min(imin), max(imax), density(idensity)
  41. {
  42. }
  43. bool CTreasureInfo::operator==(const CTreasureInfo & other) const
  44. {
  45. return (min == other.min) && (max == other.max) && (density == other.density);
  46. }
  47. void CTreasureInfo::serializeJson(JsonSerializeFormat & handler)
  48. {
  49. handler.serializeInt("min", min, 0);
  50. handler.serializeInt("max", max, 0);
  51. handler.serializeInt("density", density, 0);
  52. }
  53. namespace rmg
  54. {
  55. const TRmgTemplateZoneId ZoneOptions::NO_ZONE = -1;
  56. ZoneOptions::CTownInfo::CTownInfo()
  57. : townCount(0),
  58. castleCount(0),
  59. townDensity(0),
  60. castleDensity(0)
  61. {
  62. }
  63. int ZoneOptions::CTownInfo::getTownCount() const
  64. {
  65. return townCount;
  66. }
  67. int ZoneOptions::CTownInfo::getCastleCount() const
  68. {
  69. return castleCount;
  70. }
  71. int ZoneOptions::CTownInfo::getTownDensity() const
  72. {
  73. return townDensity;
  74. }
  75. int ZoneOptions::CTownInfo::getCastleDensity() const
  76. {
  77. return castleDensity;
  78. }
  79. void ZoneOptions::CTownInfo::serializeJson(JsonSerializeFormat & handler)
  80. {
  81. handler.serializeInt("towns", townCount, 0);
  82. handler.serializeInt("castles", castleCount, 0);
  83. handler.serializeInt("townDensity", townDensity, 0);
  84. handler.serializeInt("castleDensity", castleDensity, 0);
  85. }
  86. ZoneOptions::ZoneOptions():
  87. id(0),
  88. type(ETemplateZoneType::PLAYER_START),
  89. size(1),
  90. maxTreasureValue(0),
  91. owner(std::nullopt),
  92. matchTerrainToTown(true),
  93. townsAreSameType(false),
  94. monsterStrength(EMonsterStrength::ZONE_NORMAL),
  95. minesLikeZone(NO_ZONE),
  96. terrainTypeLikeZone(NO_ZONE),
  97. treasureLikeZone(NO_ZONE)
  98. {
  99. }
  100. TRmgTemplateZoneId ZoneOptions::getId() const
  101. {
  102. return id;
  103. }
  104. void ZoneOptions::setId(TRmgTemplateZoneId value)
  105. {
  106. if(value <= 0)
  107. throw std::runtime_error(boost::str(boost::format("Zone %d id should be greater than 0.") % id));
  108. id = value;
  109. }
  110. ETemplateZoneType ZoneOptions::getType() const
  111. {
  112. return type;
  113. }
  114. void ZoneOptions::setType(ETemplateZoneType value)
  115. {
  116. type = value;
  117. }
  118. int ZoneOptions::getSize() const
  119. {
  120. return size;
  121. }
  122. void ZoneOptions::setSize(int value)
  123. {
  124. size = value;
  125. }
  126. std::optional<int> ZoneOptions::getOwner() const
  127. {
  128. return owner;
  129. }
  130. const std::set<TerrainId> ZoneOptions::getTerrainTypes() const
  131. {
  132. if (terrainTypes.empty())
  133. {
  134. return vstd::difference(getDefaultTerrainTypes(), bannedTerrains);
  135. }
  136. else
  137. {
  138. return terrainTypes;
  139. }
  140. }
  141. void ZoneOptions::setTerrainTypes(const std::set<TerrainId> & value)
  142. {
  143. terrainTypes = value;
  144. }
  145. std::set<TerrainId> ZoneOptions::getDefaultTerrainTypes() const
  146. {
  147. std::set<TerrainId> terrains;
  148. for(const auto & terrain : VLC->terrainTypeHandler->objects)
  149. {
  150. if (terrain->isLand() && terrain->isPassable())
  151. {
  152. terrains.insert(terrain->getId());
  153. }
  154. }
  155. return terrains;
  156. }
  157. std::set<FactionID> ZoneOptions::getDefaultTownTypes() const
  158. {
  159. return VLC->townh->getDefaultAllowed();
  160. }
  161. const std::set<FactionID> ZoneOptions::getTownTypes() const
  162. {
  163. if (townTypes.empty())
  164. {
  165. //Assume that all towns are allowed, unless banned
  166. return vstd::difference(getDefaultTownTypes(), bannedTownTypes);
  167. }
  168. else
  169. {
  170. return vstd::difference(townTypes, bannedTownTypes);
  171. }
  172. }
  173. void ZoneOptions::setTownTypes(const std::set<FactionID> & value)
  174. {
  175. townTypes = value;
  176. }
  177. void ZoneOptions::setMonsterTypes(const std::set<FactionID> & value)
  178. {
  179. monsterTypes = value;
  180. }
  181. const std::set<FactionID> ZoneOptions::getMonsterTypes() const
  182. {
  183. return vstd::difference(monsterTypes, bannedMonsters);
  184. }
  185. void ZoneOptions::setMinesInfo(const std::map<TResource, ui16> & value)
  186. {
  187. mines = value;
  188. }
  189. std::map<TResource, ui16> ZoneOptions::getMinesInfo() const
  190. {
  191. return mines;
  192. }
  193. void ZoneOptions::setTreasureInfo(const std::vector<CTreasureInfo> & value)
  194. {
  195. treasureInfo = value;
  196. recalculateMaxTreasureValue();
  197. }
  198. void ZoneOptions::recalculateMaxTreasureValue()
  199. {
  200. maxTreasureValue = 0;
  201. for (const auto& ti : treasureInfo)
  202. {
  203. vstd::amax(maxTreasureValue, ti.max);
  204. }
  205. }
  206. void ZoneOptions::addTreasureInfo(const CTreasureInfo & value)
  207. {
  208. treasureInfo.push_back(value);
  209. vstd::amax(maxTreasureValue, value.max);
  210. }
  211. const std::vector<CTreasureInfo> & ZoneOptions::getTreasureInfo() const
  212. {
  213. return treasureInfo;
  214. }
  215. ui32 ZoneOptions::getMaxTreasureValue() const
  216. {
  217. return maxTreasureValue;
  218. }
  219. TRmgTemplateZoneId ZoneOptions::getMinesLikeZone() const
  220. {
  221. return minesLikeZone;
  222. }
  223. TRmgTemplateZoneId ZoneOptions::getTerrainTypeLikeZone() const
  224. {
  225. return terrainTypeLikeZone;
  226. }
  227. TRmgTemplateZoneId ZoneOptions::getTreasureLikeZone() const
  228. {
  229. return treasureLikeZone;
  230. }
  231. void ZoneOptions::addConnection(const ZoneConnection & connection)
  232. {
  233. connectedZoneIds.push_back(connection.getOtherZoneId(getId()));
  234. connectionDetails.push_back(connection);
  235. }
  236. std::vector<ZoneConnection> ZoneOptions::getConnections() const
  237. {
  238. return connectionDetails;
  239. }
  240. std::vector<TRmgTemplateZoneId> ZoneOptions::getConnectedZoneIds() const
  241. {
  242. return connectedZoneIds;
  243. }
  244. bool ZoneOptions::areTownsSameType() const
  245. {
  246. return townsAreSameType;
  247. }
  248. bool ZoneOptions::isMatchTerrainToTown() const
  249. {
  250. return matchTerrainToTown;
  251. }
  252. const ZoneOptions::CTownInfo & ZoneOptions::getPlayerTowns() const
  253. {
  254. return playerTowns;
  255. }
  256. const ZoneOptions::CTownInfo & ZoneOptions::getNeutralTowns() const
  257. {
  258. return neutralTowns;
  259. }
  260. void ZoneOptions::serializeJson(JsonSerializeFormat & handler)
  261. {
  262. static const std::vector<std::string> zoneTypes =
  263. {
  264. "playerStart",
  265. "cpuStart",
  266. "treasure",
  267. "junction",
  268. "water"
  269. };
  270. handler.serializeEnum("type", type, zoneTypes);
  271. handler.serializeInt("size", size, 1);
  272. handler.serializeInt("owner", owner);
  273. handler.serializeStruct("playerTowns", playerTowns);
  274. handler.serializeStruct("neutralTowns", neutralTowns);
  275. handler.serializeBool("matchTerrainToTown", matchTerrainToTown, true);
  276. #define SERIALIZE_ZONE_LINK(fieldName) handler.serializeInt(#fieldName, fieldName, NO_ZONE);
  277. SERIALIZE_ZONE_LINK(minesLikeZone);
  278. SERIALIZE_ZONE_LINK(terrainTypeLikeZone);
  279. SERIALIZE_ZONE_LINK(treasureLikeZone);
  280. #undef SERIALIZE_ZONE_LINK
  281. if(terrainTypeLikeZone == NO_ZONE)
  282. {
  283. handler.serializeIdArray("terrainTypes", terrainTypes);
  284. handler.serializeIdArray("bannedTerrains", bannedTerrains);
  285. }
  286. handler.serializeBool("townsAreSameType", townsAreSameType, false);
  287. handler.serializeIdArray("allowedMonsters", monsterTypes);
  288. handler.serializeIdArray("bannedMonsters", bannedMonsters);
  289. handler.serializeIdArray("allowedTowns", townTypes);
  290. handler.serializeIdArray("bannedTowns", bannedTownTypes);
  291. {
  292. //TODO: add support for std::map to serializeEnum
  293. static const std::vector<std::string> zoneMonsterStrengths =
  294. {
  295. "none",
  296. "weak",
  297. "normal",
  298. "strong"
  299. };
  300. int temporaryZoneMonsterStrengthIndex = monsterStrength == EMonsterStrength::ZONE_NONE ? 0 : monsterStrength - EMonsterStrength::ZONE_WEAK + 1 ; // temporary until serializeEnum starts supporting std::map
  301. // temporaryZoneMonsterStrengthIndex = 0, 1, 2 and 3 for monsterStrength = ZONE_NONE, ZONE_WEAK, ZONE_NORMAL and ZONE_STRONG respectively
  302. handler.serializeEnum("monsters", temporaryZoneMonsterStrengthIndex, 2, zoneMonsterStrengths); // default is normal monsters
  303. switch (temporaryZoneMonsterStrengthIndex)
  304. {
  305. case 0:
  306. monsterStrength = EMonsterStrength::ZONE_NONE;
  307. break;
  308. case 1:
  309. monsterStrength = EMonsterStrength::ZONE_WEAK;
  310. break;
  311. case 2:
  312. monsterStrength = EMonsterStrength::ZONE_NORMAL;
  313. break;
  314. case 3:
  315. monsterStrength = EMonsterStrength::ZONE_STRONG;
  316. break;
  317. }
  318. }
  319. if(treasureLikeZone == NO_ZONE)
  320. {
  321. auto treasureData = handler.enterArray("treasure");
  322. treasureData.serializeStruct(treasureInfo);
  323. if (!handler.saving)
  324. {
  325. recalculateMaxTreasureValue();
  326. }
  327. }
  328. if((minesLikeZone == NO_ZONE) && (!handler.saving || !mines.empty()))
  329. {
  330. auto minesData = handler.enterStruct("mines");
  331. for(TResource idx = 0; idx < (GameConstants::RESOURCE_QUANTITY - 1); idx++)
  332. {
  333. handler.serializeInt(GameConstants::RESOURCE_NAMES[idx], mines[idx], 0);
  334. }
  335. }
  336. }
  337. ZoneConnection::ZoneConnection():
  338. zoneA(-1),
  339. zoneB(-1),
  340. guardStrength(0),
  341. connectionType(rmg::EConnectionType::GUARDED),
  342. hasRoad(rmg::ERoadOption::ROAD_TRUE)
  343. {
  344. }
  345. TRmgTemplateZoneId ZoneConnection::getZoneA() const
  346. {
  347. return zoneA;
  348. }
  349. TRmgTemplateZoneId ZoneConnection::getZoneB() const
  350. {
  351. return zoneB;
  352. }
  353. TRmgTemplateZoneId ZoneConnection::getOtherZoneId(TRmgTemplateZoneId id) const
  354. {
  355. if (id == zoneA)
  356. {
  357. return zoneB;
  358. }
  359. else if (id == zoneB)
  360. {
  361. return zoneA;
  362. }
  363. else
  364. {
  365. throw rmgException("Zone does not belong to this connection");
  366. }
  367. }
  368. int ZoneConnection::getGuardStrength() const
  369. {
  370. return guardStrength;
  371. }
  372. rmg::EConnectionType ZoneConnection::getConnectionType() const
  373. {
  374. return connectionType;
  375. }
  376. rmg::ERoadOption ZoneConnection::getRoadOption() const
  377. {
  378. return hasRoad;
  379. }
  380. bool operator==(const ZoneConnection & l, const ZoneConnection & r)
  381. {
  382. return l.zoneA == r.zoneA && l.zoneB == r.zoneB && l.guardStrength == r.guardStrength;
  383. }
  384. void ZoneConnection::serializeJson(JsonSerializeFormat & handler)
  385. {
  386. static const std::vector<std::string> connectionTypes =
  387. {
  388. "guarded",
  389. "fictive",
  390. "repulsive",
  391. "wide",
  392. "forcePortal"
  393. };
  394. static const std::vector<std::string> roadOptions =
  395. {
  396. "true",
  397. "false",
  398. "random"
  399. };
  400. if (handler.saving)
  401. {
  402. std::string zoneNameA = std::to_string(zoneA);
  403. std::string zoneNameB = std::to_string(zoneB);
  404. handler.serializeString("a", zoneNameA);
  405. handler.serializeString("b", zoneNameB);
  406. }
  407. else
  408. {
  409. std::string zoneNameA;
  410. std::string zoneNameB;
  411. handler.serializeString("a", zoneNameA);
  412. handler.serializeString("b", zoneNameB);
  413. zoneA = std::stoi(zoneNameA);
  414. zoneB = std::stoi(zoneNameB);
  415. }
  416. handler.serializeInt("guard", guardStrength, 0);
  417. handler.serializeEnum("type", connectionType, connectionTypes);
  418. handler.serializeEnum("road", hasRoad, roadOptions);
  419. }
  420. }
  421. using namespace rmg;//todo: remove
  422. CRmgTemplate::CRmgTemplate()
  423. : minSize(72, 72, 2),
  424. maxSize(72, 72, 2)
  425. {
  426. }
  427. bool CRmgTemplate::matchesSize(const int3 & value) const
  428. {
  429. const int64_t square = value.x * value.y * value.z;
  430. const int64_t minSquare = minSize.x * minSize.y * minSize.z;
  431. const int64_t maxSquare = maxSize.x * maxSize.y * maxSize.z;
  432. return minSquare <= square && square <= maxSquare;
  433. }
  434. bool CRmgTemplate::isWaterContentAllowed(EWaterContent::EWaterContent waterContent) const
  435. {
  436. return waterContent == EWaterContent::RANDOM || allowedWaterContent.count(waterContent);
  437. }
  438. const std::set<EWaterContent::EWaterContent> & CRmgTemplate::getWaterContentAllowed() const
  439. {
  440. return allowedWaterContent;
  441. }
  442. void CRmgTemplate::setId(const std::string & value)
  443. {
  444. id = value;
  445. }
  446. void CRmgTemplate::setName(const std::string & value)
  447. {
  448. name = value;
  449. }
  450. const std::string & CRmgTemplate::getName() const
  451. {
  452. return name;
  453. }
  454. const std::string & CRmgTemplate::getDescription() const
  455. {
  456. return description;
  457. }
  458. const std::string & CRmgTemplate::getId() const
  459. {
  460. return id;
  461. }
  462. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getPlayers() const
  463. {
  464. return players;
  465. }
  466. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getHumanPlayers() const
  467. {
  468. return humanPlayers;
  469. }
  470. const CRmgTemplate::Zones & CRmgTemplate::getZones() const
  471. {
  472. return zones;
  473. }
  474. const std::vector<ZoneConnection> & CRmgTemplate::getConnectedZoneIds() const
  475. {
  476. return connectedZoneIds;
  477. }
  478. void CRmgTemplate::validate() const
  479. {
  480. //TODO add some validation checks, throw on failure
  481. }
  482. std::pair<int3, int3> CRmgTemplate::getMapSizes() const
  483. {
  484. return {minSize, maxSize};
  485. }
  486. void CRmgTemplate::CPlayerCountRange::addRange(int lower, int upper)
  487. {
  488. range.emplace_back(lower, upper);
  489. }
  490. void CRmgTemplate::CPlayerCountRange::addNumber(int value)
  491. {
  492. range.emplace_back(value, value);
  493. }
  494. bool CRmgTemplate::CPlayerCountRange::isInRange(int count) const
  495. {
  496. for(const auto & pair : range)
  497. {
  498. if(count >= pair.first && count <= pair.second) return true;
  499. }
  500. return false;
  501. }
  502. std::set<int> CRmgTemplate::CPlayerCountRange::getNumbers() const
  503. {
  504. std::set<int> numbers;
  505. for(const auto & pair : range)
  506. {
  507. for(int i = pair.first; i <= pair.second; ++i) numbers.insert(i);
  508. }
  509. return numbers;
  510. }
  511. std::string CRmgTemplate::CPlayerCountRange::toString() const
  512. {
  513. if(range.size() == 1)
  514. {
  515. const auto & p = range.front();
  516. if((p.first == p.second) && (p.first == 0))
  517. return "";
  518. }
  519. std::string ret;
  520. bool first = true;
  521. for(const auto & p : range)
  522. {
  523. if(!first)
  524. ret +=",";
  525. else
  526. first = false;
  527. if(p.first == p.second)
  528. {
  529. ret += std::to_string(p.first);
  530. }
  531. else
  532. {
  533. ret += boost::str(boost::format("%d-%d") % p.first % p.second);
  534. }
  535. }
  536. return ret;
  537. }
  538. void CRmgTemplate::CPlayerCountRange::fromString(const std::string & value)
  539. {
  540. range.clear();
  541. if(value.empty())
  542. {
  543. addNumber(0);
  544. }
  545. else
  546. {
  547. std::vector<std::string> commaParts;
  548. boost::split(commaParts, value, boost::is_any_of(","));
  549. for(const auto & commaPart : commaParts)
  550. {
  551. std::vector<std::string> rangeParts;
  552. boost::split(rangeParts, commaPart, boost::is_any_of("-"));
  553. if(rangeParts.size() == 2)
  554. {
  555. auto lower = boost::lexical_cast<int>(rangeParts[0]);
  556. auto upper = boost::lexical_cast<int>(rangeParts[1]);
  557. addRange(lower, upper);
  558. }
  559. else if(rangeParts.size() == 1)
  560. {
  561. auto val = boost::lexical_cast<int>(rangeParts.front());
  562. addNumber(val);
  563. }
  564. }
  565. }
  566. }
  567. int CRmgTemplate::CPlayerCountRange::maxValue() const
  568. {
  569. return *boost::max_element(getNumbers());
  570. }
  571. int CRmgTemplate::CPlayerCountRange::minValue() const
  572. {
  573. return *boost::min_element(getNumbers());
  574. }
  575. void CRmgTemplate::serializeJson(JsonSerializeFormat & handler)
  576. {
  577. handler.serializeString("name", name);
  578. handler.serializeString("description", description);
  579. serializeSize(handler, minSize, "minSize");
  580. serializeSize(handler, maxSize, "maxSize");
  581. serializePlayers(handler, players, "players");
  582. serializePlayers(handler, humanPlayers, "humans"); // TODO: Rename this parameter
  583. {
  584. auto connectionsData = handler.enterArray("connections");
  585. connectionsData.serializeStruct(connectedZoneIds);
  586. }
  587. {
  588. boost::bimap<EWaterContent::EWaterContent, std::string> enc;
  589. enc.insert({EWaterContent::NONE, "none"});
  590. enc.insert({EWaterContent::NORMAL, "normal"});
  591. enc.insert({EWaterContent::ISLANDS, "islands"});
  592. JsonNode node;
  593. if(handler.saving)
  594. {
  595. node.setType(JsonNode::JsonType::DATA_VECTOR);
  596. for(auto wc : allowedWaterContent)
  597. {
  598. JsonNode n;
  599. n.String() = enc.left.at(wc);
  600. node.Vector().push_back(n);
  601. }
  602. }
  603. handler.serializeRaw("allowedWaterContent", node, std::nullopt);
  604. if(!handler.saving)
  605. {
  606. for(auto wc : node.Vector())
  607. {
  608. allowedWaterContent.insert(enc.right.at(std::string(wc.String())));
  609. }
  610. }
  611. }
  612. {
  613. auto zonesData = handler.enterStruct("zones");
  614. if(handler.saving)
  615. {
  616. for(auto & idAndZone : zones)
  617. {
  618. auto guard = handler.enterStruct(encodeZoneId(idAndZone.first));
  619. idAndZone.second->serializeJson(handler);
  620. }
  621. }
  622. else
  623. {
  624. for(const auto & idAndZone : zonesData->getCurrent().Struct())
  625. {
  626. auto guard = handler.enterStruct(idAndZone.first);
  627. auto zone = std::make_shared<ZoneOptions>();
  628. zone->setId(decodeZoneId(idAndZone.first));
  629. zone->serializeJson(handler);
  630. zones[zone->getId()] = zone;
  631. }
  632. }
  633. }
  634. }
  635. std::set<TerrainId> CRmgTemplate::inheritTerrainType(std::shared_ptr<ZoneOptions> zone, uint32_t iteration /* = 0 */)
  636. {
  637. if (iteration >= 50)
  638. {
  639. logGlobal->error("Infinite recursion for terrain types detected in template %s", name);
  640. return std::set<TerrainId>();
  641. }
  642. if (zone->getTerrainTypeLikeZone() != ZoneOptions::NO_ZONE)
  643. {
  644. iteration++;
  645. const auto otherZone = zones.at(zone->getTerrainTypeLikeZone());
  646. zone->setTerrainTypes(inheritTerrainType(otherZone, iteration));
  647. }
  648. //This implicitly excludes banned terrains
  649. return zone->getTerrainTypes();
  650. }
  651. std::map<TResource, ui16> CRmgTemplate::inheritMineTypes(std::shared_ptr<ZoneOptions> zone, uint32_t iteration /* = 0 */)
  652. {
  653. if (iteration >= 50)
  654. {
  655. logGlobal->error("Infinite recursion for mine types detected in template %s", name);
  656. return std::map<TResource, ui16>();
  657. }
  658. if (zone->getMinesLikeZone() != ZoneOptions::NO_ZONE)
  659. {
  660. iteration++;
  661. const auto otherZone = zones.at(zone->getMinesLikeZone());
  662. zone->setMinesInfo(inheritMineTypes(otherZone, iteration));
  663. }
  664. return zone->getMinesInfo();
  665. }
  666. std::vector<CTreasureInfo> CRmgTemplate::inheritTreasureInfo(std::shared_ptr<ZoneOptions> zone, uint32_t iteration /* = 0 */)
  667. {
  668. if (iteration >= 50)
  669. {
  670. logGlobal->error("Infinite recursion for treasures detected in template %s", name);
  671. return std::vector<CTreasureInfo>();
  672. }
  673. if (zone->getTreasureLikeZone() != ZoneOptions::NO_ZONE)
  674. {
  675. iteration++;
  676. const auto otherZone = zones.at(zone->getTreasureLikeZone());
  677. zone->setTreasureInfo(inheritTreasureInfo(otherZone, iteration));
  678. }
  679. return zone->getTreasureInfo();
  680. }
  681. void CRmgTemplate::afterLoad()
  682. {
  683. for(auto & idAndZone : zones)
  684. {
  685. auto zone = idAndZone.second;
  686. //Inherit properties recursively.
  687. inheritTerrainType(zone);
  688. inheritMineTypes(zone);
  689. inheritTreasureInfo(zone);
  690. //TODO: Inherit monster types as well
  691. auto monsterTypes = zone->getMonsterTypes();
  692. if (monsterTypes.empty())
  693. {
  694. zone->setMonsterTypes(VLC->townh->getAllowedFactions(false));
  695. }
  696. }
  697. for(const auto & connection : connectedZoneIds)
  698. {
  699. auto id1 = connection.getZoneA();
  700. auto id2 = connection.getZoneB();
  701. auto zone1 = zones.at(id1);
  702. auto zone2 = zones.at(id2);
  703. zone1->addConnection(connection);
  704. zone2->addConnection(connection);
  705. }
  706. if(allowedWaterContent.empty() || allowedWaterContent.count(EWaterContent::RANDOM))
  707. {
  708. allowedWaterContent.insert(EWaterContent::NONE);
  709. allowedWaterContent.insert(EWaterContent::NORMAL);
  710. allowedWaterContent.insert(EWaterContent::ISLANDS);
  711. }
  712. allowedWaterContent.erase(EWaterContent::RANDOM);
  713. }
  714. void CRmgTemplate::serializeSize(JsonSerializeFormat & handler, int3 & value, const std::string & fieldName)
  715. {
  716. static const std::map<std::string, int3> sizeMapping =
  717. {
  718. {"s", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 1}},
  719. {"s+u", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 2}},
  720. {"m", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 1}},
  721. {"m+u", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 2}},
  722. {"l", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 1}},
  723. {"l+u", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 2}},
  724. {"xl", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 1}} ,
  725. {"xl+u", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 2}} ,
  726. {"h", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 1}},
  727. {"h+u", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 2}},
  728. {"xh", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 1}},
  729. {"xh+u", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 2}},
  730. {"g", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 1}},
  731. {"g+u", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 2}}
  732. };
  733. static const std::map<int3, std::string> sizeReverseMapping = vstd::invertMap(sizeMapping);
  734. std::string encodedValue;
  735. if(handler.saving)
  736. {
  737. auto iter = sizeReverseMapping.find(value);
  738. if(iter == sizeReverseMapping.end())
  739. encodedValue = boost::str(boost::format("%dx%dx%d") % value.x % value.y % value.z);
  740. else
  741. encodedValue = iter->second;
  742. }
  743. handler.serializeString(fieldName, encodedValue);
  744. if(!handler.saving)
  745. {
  746. auto iter = sizeMapping.find(encodedValue);
  747. if(iter == sizeMapping.end())
  748. {
  749. std::vector<std::string> parts;
  750. boost::split(parts, encodedValue, boost::is_any_of("x"));
  751. value.x = (boost::lexical_cast<int>(parts.at(0)));
  752. value.y = (boost::lexical_cast<int>(parts.at(1)));
  753. value.z = (boost::lexical_cast<int>(parts.at(2)));
  754. }
  755. else
  756. {
  757. value = iter->second;
  758. }
  759. }
  760. }
  761. void CRmgTemplate::serializePlayers(JsonSerializeFormat & handler, CPlayerCountRange & value, const std::string & fieldName)
  762. {
  763. std::string encodedValue;
  764. if(handler.saving)
  765. encodedValue = value.toString();
  766. handler.serializeString(fieldName, encodedValue);
  767. if(!handler.saving)
  768. value.fromString(encodedValue);
  769. }
  770. VCMI_LIB_NAMESPACE_END