CRmgTemplate.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950
  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. };
  283. handler.serializeEnum("type", type, zoneTypes);
  284. handler.serializeInt("size", size, 1);
  285. handler.serializeInt("owner", owner);
  286. handler.serializeStruct("playerTowns", playerTowns);
  287. handler.serializeStruct("neutralTowns", neutralTowns);
  288. handler.serializeBool("matchTerrainToTown", matchTerrainToTown, true);
  289. #define SERIALIZE_ZONE_LINK(fieldName) handler.serializeInt(#fieldName, fieldName, NO_ZONE);
  290. SERIALIZE_ZONE_LINK(minesLikeZone);
  291. SERIALIZE_ZONE_LINK(terrainTypeLikeZone);
  292. SERIALIZE_ZONE_LINK(treasureLikeZone);
  293. SERIALIZE_ZONE_LINK(customObjectsLikeZone);
  294. #undef SERIALIZE_ZONE_LINK
  295. if(terrainTypeLikeZone == NO_ZONE)
  296. {
  297. handler.serializeIdArray("terrainTypes", terrainTypes);
  298. handler.serializeIdArray("bannedTerrains", bannedTerrains);
  299. }
  300. handler.serializeBool("townsAreSameType", townsAreSameType, false);
  301. handler.serializeIdArray("allowedMonsters", monsterTypes);
  302. handler.serializeIdArray("bannedMonsters", bannedMonsters);
  303. handler.serializeIdArray("allowedTowns", townTypes);
  304. handler.serializeIdArray("bannedTowns", bannedTownTypes);
  305. {
  306. //TODO: add support for std::map to serializeEnum
  307. static const std::vector<std::string> zoneMonsterStrengths =
  308. {
  309. "none",
  310. "weak",
  311. "normal",
  312. "strong"
  313. };
  314. int temporaryZoneMonsterStrengthIndex = monsterStrength == EMonsterStrength::ZONE_NONE ? 0 : monsterStrength - EMonsterStrength::ZONE_WEAK + 1 ; // temporary until serializeEnum starts supporting std::map
  315. // temporaryZoneMonsterStrengthIndex = 0, 1, 2 and 3 for monsterStrength = ZONE_NONE, ZONE_WEAK, ZONE_NORMAL and ZONE_STRONG respectively
  316. handler.serializeEnum("monsters", temporaryZoneMonsterStrengthIndex, 2, zoneMonsterStrengths); // default is normal monsters
  317. switch (temporaryZoneMonsterStrengthIndex)
  318. {
  319. case 0:
  320. monsterStrength = EMonsterStrength::ZONE_NONE;
  321. break;
  322. case 1:
  323. monsterStrength = EMonsterStrength::ZONE_WEAK;
  324. break;
  325. case 2:
  326. monsterStrength = EMonsterStrength::ZONE_NORMAL;
  327. break;
  328. case 3:
  329. monsterStrength = EMonsterStrength::ZONE_STRONG;
  330. break;
  331. }
  332. }
  333. if(treasureLikeZone == NO_ZONE)
  334. {
  335. auto treasureData = handler.enterArray("treasure");
  336. treasureData.serializeStruct(treasureInfo);
  337. if (!handler.saving)
  338. {
  339. recalculateMaxTreasureValue();
  340. }
  341. }
  342. if((minesLikeZone == NO_ZONE) && (!handler.saving || !mines.empty()))
  343. {
  344. auto minesData = handler.enterStruct("mines");
  345. for(TResource idx = 0; idx < (GameConstants::RESOURCE_QUANTITY - 1); idx++)
  346. {
  347. handler.serializeInt(GameConstants::RESOURCE_NAMES[idx], mines[idx], 0);
  348. }
  349. }
  350. handler.serializeStruct("customObjects", objectConfig);
  351. }
  352. ZoneConnection::ZoneConnection():
  353. zoneA(-1),
  354. zoneB(-1),
  355. guardStrength(0),
  356. connectionType(rmg::EConnectionType::GUARDED),
  357. hasRoad(rmg::ERoadOption::ROAD_TRUE)
  358. {
  359. }
  360. TRmgTemplateZoneId ZoneConnection::getZoneA() const
  361. {
  362. return zoneA;
  363. }
  364. TRmgTemplateZoneId ZoneConnection::getZoneB() const
  365. {
  366. return zoneB;
  367. }
  368. TRmgTemplateZoneId ZoneConnection::getOtherZoneId(TRmgTemplateZoneId id) const
  369. {
  370. if (id == zoneA)
  371. {
  372. return zoneB;
  373. }
  374. else if (id == zoneB)
  375. {
  376. return zoneA;
  377. }
  378. else
  379. {
  380. throw rmgException("Zone does not belong to this connection");
  381. }
  382. }
  383. int ZoneConnection::getGuardStrength() const
  384. {
  385. return guardStrength;
  386. }
  387. rmg::EConnectionType ZoneConnection::getConnectionType() const
  388. {
  389. return connectionType;
  390. }
  391. rmg::ERoadOption ZoneConnection::getRoadOption() const
  392. {
  393. return hasRoad;
  394. }
  395. bool operator==(const ZoneConnection & l, const ZoneConnection & r)
  396. {
  397. return l.zoneA == r.zoneA && l.zoneB == r.zoneB && l.guardStrength == r.guardStrength;
  398. }
  399. void ZoneConnection::serializeJson(JsonSerializeFormat & handler)
  400. {
  401. static const std::vector<std::string> connectionTypes =
  402. {
  403. "guarded",
  404. "fictive",
  405. "repulsive",
  406. "wide",
  407. "forcePortal"
  408. };
  409. static const std::vector<std::string> roadOptions =
  410. {
  411. "true",
  412. "false",
  413. "random"
  414. };
  415. if (handler.saving)
  416. {
  417. std::string zoneNameA = std::to_string(zoneA);
  418. std::string zoneNameB = std::to_string(zoneB);
  419. handler.serializeString("a", zoneNameA);
  420. handler.serializeString("b", zoneNameB);
  421. }
  422. else
  423. {
  424. std::string zoneNameA;
  425. std::string zoneNameB;
  426. handler.serializeString("a", zoneNameA);
  427. handler.serializeString("b", zoneNameB);
  428. zoneA = std::stoi(zoneNameA);
  429. zoneB = std::stoi(zoneNameB);
  430. }
  431. handler.serializeInt("guard", guardStrength, 0);
  432. handler.serializeEnum("type", connectionType, connectionTypes);
  433. handler.serializeEnum("road", hasRoad, roadOptions);
  434. }
  435. }
  436. using namespace rmg;//todo: remove
  437. CRmgTemplate::~CRmgTemplate() = default;
  438. CRmgTemplate::CRmgTemplate()
  439. : minSize(72, 72, 2),
  440. maxSize(72, 72, 2),
  441. mapSettings(std::make_unique<JsonNode>())
  442. {
  443. }
  444. bool CRmgTemplate::matchesSize(const int3 & value) const
  445. {
  446. const int64_t square = value.x * value.y * value.z;
  447. const int64_t minSquare = minSize.x * minSize.y * minSize.z;
  448. const int64_t maxSquare = maxSize.x * maxSize.y * maxSize.z;
  449. return minSquare <= square && square <= maxSquare;
  450. }
  451. bool CRmgTemplate::isWaterContentAllowed(EWaterContent::EWaterContent waterContent) const
  452. {
  453. return waterContent == EWaterContent::RANDOM || allowedWaterContent.count(waterContent);
  454. }
  455. const std::set<EWaterContent::EWaterContent> & CRmgTemplate::getWaterContentAllowed() const
  456. {
  457. return allowedWaterContent;
  458. }
  459. void CRmgTemplate::setId(const std::string & value)
  460. {
  461. id = value;
  462. }
  463. void CRmgTemplate::setName(const std::string & value)
  464. {
  465. name = value;
  466. }
  467. const std::string & CRmgTemplate::getName() const
  468. {
  469. return name;
  470. }
  471. const std::string & CRmgTemplate::getDescription() const
  472. {
  473. return description;
  474. }
  475. const std::string & CRmgTemplate::getId() const
  476. {
  477. return id;
  478. }
  479. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getPlayers() const
  480. {
  481. return players;
  482. }
  483. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getHumanPlayers() const
  484. {
  485. return humanPlayers;
  486. }
  487. const CRmgTemplate::Zones & CRmgTemplate::getZones() const
  488. {
  489. return zones;
  490. }
  491. const std::vector<ZoneConnection> & CRmgTemplate::getConnectedZoneIds() const
  492. {
  493. return connectedZoneIds;
  494. }
  495. void CRmgTemplate::validate() const
  496. {
  497. //TODO add some validation checks, throw on failure
  498. }
  499. std::pair<int3, int3> CRmgTemplate::getMapSizes() const
  500. {
  501. return {minSize, maxSize};
  502. }
  503. void CRmgTemplate::CPlayerCountRange::addRange(int lower, int upper)
  504. {
  505. range.emplace_back(lower, upper);
  506. }
  507. void CRmgTemplate::CPlayerCountRange::addNumber(int value)
  508. {
  509. range.emplace_back(value, value);
  510. }
  511. bool CRmgTemplate::CPlayerCountRange::isInRange(int count) const
  512. {
  513. for(const auto & pair : range)
  514. {
  515. if(count >= pair.first && count <= pair.second) return true;
  516. }
  517. return false;
  518. }
  519. std::set<int> CRmgTemplate::CPlayerCountRange::getNumbers() const
  520. {
  521. std::set<int> numbers;
  522. for(const auto & pair : range)
  523. {
  524. for(int i = pair.first; i <= pair.second; ++i) numbers.insert(i);
  525. }
  526. return numbers;
  527. }
  528. std::string CRmgTemplate::CPlayerCountRange::toString() const
  529. {
  530. if(range.size() == 1)
  531. {
  532. const auto & p = range.front();
  533. if((p.first == p.second) && (p.first == 0))
  534. return "";
  535. }
  536. std::string ret;
  537. bool first = true;
  538. for(const auto & p : range)
  539. {
  540. if(!first)
  541. ret +=",";
  542. else
  543. first = false;
  544. if(p.first == p.second)
  545. {
  546. ret += std::to_string(p.first);
  547. }
  548. else
  549. {
  550. ret += boost::str(boost::format("%d-%d") % p.first % p.second);
  551. }
  552. }
  553. return ret;
  554. }
  555. void CRmgTemplate::CPlayerCountRange::fromString(const std::string & value)
  556. {
  557. range.clear();
  558. if(value.empty())
  559. {
  560. addNumber(0);
  561. }
  562. else
  563. {
  564. std::vector<std::string> commaParts;
  565. boost::split(commaParts, value, boost::is_any_of(","));
  566. for(const auto & commaPart : commaParts)
  567. {
  568. std::vector<std::string> rangeParts;
  569. boost::split(rangeParts, commaPart, boost::is_any_of("-"));
  570. if(rangeParts.size() == 2)
  571. {
  572. auto lower = boost::lexical_cast<int>(rangeParts[0]);
  573. auto upper = boost::lexical_cast<int>(rangeParts[1]);
  574. addRange(lower, upper);
  575. }
  576. else if(rangeParts.size() == 1)
  577. {
  578. auto val = boost::lexical_cast<int>(rangeParts.front());
  579. addNumber(val);
  580. }
  581. }
  582. }
  583. }
  584. int CRmgTemplate::CPlayerCountRange::maxValue() const
  585. {
  586. return *boost::max_element(getNumbers());
  587. }
  588. int CRmgTemplate::CPlayerCountRange::minValue() const
  589. {
  590. return *boost::min_element(getNumbers());
  591. }
  592. void CRmgTemplate::serializeJson(JsonSerializeFormat & handler)
  593. {
  594. handler.serializeString("name", name);
  595. handler.serializeString("description", description);
  596. serializeSize(handler, minSize, "minSize");
  597. serializeSize(handler, maxSize, "maxSize");
  598. serializePlayers(handler, players, "players");
  599. serializePlayers(handler, humanPlayers, "humans"); // TODO: Rename this parameter
  600. *mapSettings = handler.getCurrent()["settings"];
  601. {
  602. auto connectionsData = handler.enterArray("connections");
  603. connectionsData.serializeStruct(connectedZoneIds);
  604. }
  605. {
  606. boost::bimap<EWaterContent::EWaterContent, std::string> enc;
  607. enc.insert({EWaterContent::NONE, "none"});
  608. enc.insert({EWaterContent::NORMAL, "normal"});
  609. enc.insert({EWaterContent::ISLANDS, "islands"});
  610. JsonNode node;
  611. if(handler.saving)
  612. {
  613. node.setType(JsonNode::JsonType::DATA_VECTOR);
  614. for(auto wc : allowedWaterContent)
  615. {
  616. JsonNode n;
  617. n.String() = enc.left.at(wc);
  618. node.Vector().push_back(n);
  619. }
  620. }
  621. handler.serializeRaw("allowedWaterContent", node, std::nullopt);
  622. if(!handler.saving)
  623. {
  624. for(auto wc : node.Vector())
  625. {
  626. allowedWaterContent.insert(enc.right.at(std::string(wc.String())));
  627. }
  628. }
  629. }
  630. {
  631. auto zonesData = handler.enterStruct("zones");
  632. if(handler.saving)
  633. {
  634. for(auto & idAndZone : zones)
  635. {
  636. auto guard = handler.enterStruct(encodeZoneId(idAndZone.first));
  637. idAndZone.second->serializeJson(handler);
  638. }
  639. }
  640. else
  641. {
  642. for(const auto & idAndZone : zonesData->getCurrent().Struct())
  643. {
  644. auto guard = handler.enterStruct(idAndZone.first);
  645. auto zone = std::make_shared<ZoneOptions>();
  646. zone->setId(decodeZoneId(idAndZone.first));
  647. zone->serializeJson(handler);
  648. zones[zone->getId()] = zone;
  649. }
  650. }
  651. }
  652. }
  653. const JsonNode & CRmgTemplate::getMapSettings() const
  654. {
  655. return *mapSettings;
  656. }
  657. template<typename T>
  658. T CRmgTemplate::inheritZoneProperty(std::shared_ptr<rmg::ZoneOptions> zone,
  659. T (rmg::ZoneOptions::*getter)() const,
  660. void (rmg::ZoneOptions::*setter)(const T&),
  661. TRmgTemplateZoneId (rmg::ZoneOptions::*inheritFrom)() const,
  662. const std::string& propertyString,
  663. uint32_t iteration)
  664. {
  665. if (iteration >= 50)
  666. {
  667. logGlobal->error("Infinite recursion for %s detected in template %s", propertyString, name);
  668. return T();
  669. }
  670. if (((*zone).*inheritFrom)() != rmg::ZoneOptions::NO_ZONE)
  671. {
  672. iteration++;
  673. const auto otherZone = zones.at(((*zone).*inheritFrom)());
  674. T inheritedValue = inheritZoneProperty(otherZone, getter, setter, inheritFrom, propertyString, iteration);
  675. ((*zone).*setter)(inheritedValue);
  676. }
  677. return ((*zone).*getter)();
  678. }
  679. void CRmgTemplate::afterLoad()
  680. {
  681. for(auto & idAndZone : zones)
  682. {
  683. auto zone = idAndZone.second;
  684. // Inherit properties recursively
  685. inheritZoneProperty(zone,
  686. &rmg::ZoneOptions::getTerrainTypes,
  687. &rmg::ZoneOptions::setTerrainTypes,
  688. &rmg::ZoneOptions::getTerrainTypeLikeZone,
  689. "terrain types");
  690. inheritZoneProperty(zone,
  691. &rmg::ZoneOptions::getMinesInfo,
  692. &rmg::ZoneOptions::setMinesInfo,
  693. &rmg::ZoneOptions::getMinesLikeZone,
  694. "mine types");
  695. inheritZoneProperty(zone,
  696. &rmg::ZoneOptions::getTreasureInfo,
  697. &rmg::ZoneOptions::setTreasureInfo,
  698. &rmg::ZoneOptions::getTreasureLikeZone,
  699. "treasure info");
  700. inheritZoneProperty(zone,
  701. &rmg::ZoneOptions::getCustomObjects,
  702. &rmg::ZoneOptions::setCustomObjects,
  703. &rmg::ZoneOptions::getCustomObjectsLikeZone,
  704. "custom objects");
  705. //TODO: Inherit monster types as well
  706. auto monsterTypes = zone->getMonsterTypes();
  707. if (monsterTypes.empty())
  708. {
  709. zone->setMonsterTypes(VLC->townh->getAllowedFactions(false));
  710. }
  711. }
  712. for(const auto & connection : connectedZoneIds)
  713. {
  714. auto id1 = connection.getZoneA();
  715. auto id2 = connection.getZoneB();
  716. auto zone1 = zones.at(id1);
  717. auto zone2 = zones.at(id2);
  718. zone1->addConnection(connection);
  719. zone2->addConnection(connection);
  720. }
  721. if(allowedWaterContent.empty() || allowedWaterContent.count(EWaterContent::RANDOM))
  722. {
  723. allowedWaterContent.insert(EWaterContent::NONE);
  724. allowedWaterContent.insert(EWaterContent::NORMAL);
  725. allowedWaterContent.insert(EWaterContent::ISLANDS);
  726. }
  727. allowedWaterContent.erase(EWaterContent::RANDOM);
  728. }
  729. // TODO: Allow any integer size which does not match enum, as well
  730. void CRmgTemplate::serializeSize(JsonSerializeFormat & handler, int3 & value, const std::string & fieldName)
  731. {
  732. static const std::map<std::string, int3> sizeMapping =
  733. {
  734. {"s", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 1}},
  735. {"s+u", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 2}},
  736. {"m", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 1}},
  737. {"m+u", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 2}},
  738. {"l", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 1}},
  739. {"l+u", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 2}},
  740. {"xl", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 1}} ,
  741. {"xl+u", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 2}} ,
  742. {"h", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 1}},
  743. {"h+u", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 2}},
  744. {"xh", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 1}},
  745. {"xh+u", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 2}},
  746. {"g", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 1}},
  747. {"g+u", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 2}}
  748. };
  749. static const std::map<int3, std::string> sizeReverseMapping = vstd::invertMap(sizeMapping);
  750. std::string encodedValue;
  751. if(handler.saving)
  752. {
  753. auto iter = sizeReverseMapping.find(value);
  754. if(iter == sizeReverseMapping.end())
  755. encodedValue = boost::str(boost::format("%dx%dx%d") % value.x % value.y % value.z);
  756. else
  757. encodedValue = iter->second;
  758. }
  759. handler.serializeString(fieldName, encodedValue);
  760. if(!handler.saving)
  761. {
  762. auto iter = sizeMapping.find(encodedValue);
  763. if(iter == sizeMapping.end())
  764. {
  765. std::vector<std::string> parts;
  766. boost::split(parts, encodedValue, boost::is_any_of("x"));
  767. value.x = (boost::lexical_cast<int>(parts.at(0)));
  768. value.y = (boost::lexical_cast<int>(parts.at(1)));
  769. value.z = (boost::lexical_cast<int>(parts.at(2)));
  770. }
  771. else
  772. {
  773. value = iter->second;
  774. }
  775. }
  776. }
  777. void CRmgTemplate::serializePlayers(JsonSerializeFormat & handler, CPlayerCountRange & value, const std::string & fieldName)
  778. {
  779. std::string encodedValue;
  780. if(handler.saving)
  781. encodedValue = value.toString();
  782. handler.serializeString(fieldName, encodedValue);
  783. if(!handler.saving)
  784. value.fromString(encodedValue);
  785. }
  786. const std::vector<CompoundMapObjectID> & ZoneOptions::getBannedObjects() const
  787. {
  788. return objectConfig.getBannedObjects();
  789. }
  790. const std::vector<ObjectConfig::EObjectCategory> & ZoneOptions::getBannedObjectCategories() const
  791. {
  792. return objectConfig.getBannedObjectCategories();
  793. }
  794. const std::vector<ObjectInfo> & ZoneOptions::getConfiguredObjects() const
  795. {
  796. return objectConfig.getConfiguredObjects();
  797. }
  798. VCMI_LIB_NAMESPACE_END