CRmgTemplate.cpp 22 KB

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