CRmgTemplate.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939
  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. };
  408. static const std::vector<std::string> roadOptions =
  409. {
  410. "true",
  411. "false",
  412. "random"
  413. };
  414. if (handler.saving)
  415. {
  416. std::string zoneNameA = std::to_string(zoneA);
  417. std::string zoneNameB = std::to_string(zoneB);
  418. handler.serializeString("a", zoneNameA);
  419. handler.serializeString("b", zoneNameB);
  420. }
  421. else
  422. {
  423. std::string zoneNameA;
  424. std::string zoneNameB;
  425. handler.serializeString("a", zoneNameA);
  426. handler.serializeString("b", zoneNameB);
  427. zoneA = std::stoi(zoneNameA);
  428. zoneB = std::stoi(zoneNameB);
  429. }
  430. handler.serializeInt("guard", guardStrength, 0);
  431. handler.serializeEnum("type", connectionType, connectionTypes);
  432. handler.serializeEnum("road", hasRoad, roadOptions);
  433. }
  434. }
  435. using namespace rmg;//todo: remove
  436. CRmgTemplate::CRmgTemplate()
  437. : minSize(72, 72, 2),
  438. maxSize(72, 72, 2)
  439. {
  440. }
  441. bool CRmgTemplate::matchesSize(const int3 & value) const
  442. {
  443. const int64_t square = value.x * value.y * value.z;
  444. const int64_t minSquare = minSize.x * minSize.y * minSize.z;
  445. const int64_t maxSquare = maxSize.x * maxSize.y * maxSize.z;
  446. return minSquare <= square && square <= maxSquare;
  447. }
  448. bool CRmgTemplate::isWaterContentAllowed(EWaterContent::EWaterContent waterContent) const
  449. {
  450. return waterContent == EWaterContent::RANDOM || allowedWaterContent.count(waterContent);
  451. }
  452. const std::set<EWaterContent::EWaterContent> & CRmgTemplate::getWaterContentAllowed() const
  453. {
  454. return allowedWaterContent;
  455. }
  456. void CRmgTemplate::setId(const std::string & value)
  457. {
  458. id = value;
  459. }
  460. void CRmgTemplate::setName(const std::string & value)
  461. {
  462. name = value;
  463. }
  464. const std::string & CRmgTemplate::getName() const
  465. {
  466. return name;
  467. }
  468. const std::string & CRmgTemplate::getDescription() const
  469. {
  470. return description;
  471. }
  472. const std::string & CRmgTemplate::getId() const
  473. {
  474. return id;
  475. }
  476. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getPlayers() const
  477. {
  478. return players;
  479. }
  480. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getHumanPlayers() const
  481. {
  482. return humanPlayers;
  483. }
  484. const CRmgTemplate::Zones & CRmgTemplate::getZones() const
  485. {
  486. return zones;
  487. }
  488. const std::vector<ZoneConnection> & CRmgTemplate::getConnectedZoneIds() const
  489. {
  490. return connectedZoneIds;
  491. }
  492. void CRmgTemplate::validate() const
  493. {
  494. //TODO add some validation checks, throw on failure
  495. }
  496. std::pair<int3, int3> CRmgTemplate::getMapSizes() const
  497. {
  498. return {minSize, maxSize};
  499. }
  500. void CRmgTemplate::CPlayerCountRange::addRange(int lower, int upper)
  501. {
  502. range.emplace_back(lower, upper);
  503. }
  504. void CRmgTemplate::CPlayerCountRange::addNumber(int value)
  505. {
  506. range.emplace_back(value, value);
  507. }
  508. bool CRmgTemplate::CPlayerCountRange::isInRange(int count) const
  509. {
  510. for(const auto & pair : range)
  511. {
  512. if(count >= pair.first && count <= pair.second) return true;
  513. }
  514. return false;
  515. }
  516. std::set<int> CRmgTemplate::CPlayerCountRange::getNumbers() const
  517. {
  518. std::set<int> numbers;
  519. for(const auto & pair : range)
  520. {
  521. for(int i = pair.first; i <= pair.second; ++i) numbers.insert(i);
  522. }
  523. return numbers;
  524. }
  525. std::string CRmgTemplate::CPlayerCountRange::toString() const
  526. {
  527. if(range.size() == 1)
  528. {
  529. const auto & p = range.front();
  530. if((p.first == p.second) && (p.first == 0))
  531. return "";
  532. }
  533. std::string ret;
  534. bool first = true;
  535. for(const auto & p : range)
  536. {
  537. if(!first)
  538. ret +=",";
  539. else
  540. first = false;
  541. if(p.first == p.second)
  542. {
  543. ret += std::to_string(p.first);
  544. }
  545. else
  546. {
  547. ret += boost::str(boost::format("%d-%d") % p.first % p.second);
  548. }
  549. }
  550. return ret;
  551. }
  552. void CRmgTemplate::CPlayerCountRange::fromString(const std::string & value)
  553. {
  554. range.clear();
  555. if(value.empty())
  556. {
  557. addNumber(0);
  558. }
  559. else
  560. {
  561. std::vector<std::string> commaParts;
  562. boost::split(commaParts, value, boost::is_any_of(","));
  563. for(const auto & commaPart : commaParts)
  564. {
  565. std::vector<std::string> rangeParts;
  566. boost::split(rangeParts, commaPart, boost::is_any_of("-"));
  567. if(rangeParts.size() == 2)
  568. {
  569. auto lower = boost::lexical_cast<int>(rangeParts[0]);
  570. auto upper = boost::lexical_cast<int>(rangeParts[1]);
  571. addRange(lower, upper);
  572. }
  573. else if(rangeParts.size() == 1)
  574. {
  575. auto val = boost::lexical_cast<int>(rangeParts.front());
  576. addNumber(val);
  577. }
  578. }
  579. }
  580. }
  581. int CRmgTemplate::CPlayerCountRange::maxValue() const
  582. {
  583. return *boost::max_element(getNumbers());
  584. }
  585. int CRmgTemplate::CPlayerCountRange::minValue() const
  586. {
  587. return *boost::min_element(getNumbers());
  588. }
  589. void CRmgTemplate::serializeJson(JsonSerializeFormat & handler)
  590. {
  591. handler.serializeString("name", name);
  592. handler.serializeString("description", description);
  593. serializeSize(handler, minSize, "minSize");
  594. serializeSize(handler, maxSize, "maxSize");
  595. serializePlayers(handler, players, "players");
  596. serializePlayers(handler, humanPlayers, "humans"); // TODO: Rename this parameter
  597. {
  598. auto connectionsData = handler.enterArray("connections");
  599. connectionsData.serializeStruct(connectedZoneIds);
  600. }
  601. {
  602. boost::bimap<EWaterContent::EWaterContent, std::string> enc;
  603. enc.insert({EWaterContent::NONE, "none"});
  604. enc.insert({EWaterContent::NORMAL, "normal"});
  605. enc.insert({EWaterContent::ISLANDS, "islands"});
  606. JsonNode node;
  607. if(handler.saving)
  608. {
  609. node.setType(JsonNode::JsonType::DATA_VECTOR);
  610. for(auto wc : allowedWaterContent)
  611. {
  612. JsonNode n;
  613. n.String() = enc.left.at(wc);
  614. node.Vector().push_back(n);
  615. }
  616. }
  617. handler.serializeRaw("allowedWaterContent", node, std::nullopt);
  618. if(!handler.saving)
  619. {
  620. for(auto wc : node.Vector())
  621. {
  622. allowedWaterContent.insert(enc.right.at(std::string(wc.String())));
  623. }
  624. }
  625. }
  626. {
  627. auto zonesData = handler.enterStruct("zones");
  628. if(handler.saving)
  629. {
  630. for(auto & idAndZone : zones)
  631. {
  632. auto guard = handler.enterStruct(encodeZoneId(idAndZone.first));
  633. idAndZone.second->serializeJson(handler);
  634. }
  635. }
  636. else
  637. {
  638. for(const auto & idAndZone : zonesData->getCurrent().Struct())
  639. {
  640. auto guard = handler.enterStruct(idAndZone.first);
  641. auto zone = std::make_shared<ZoneOptions>();
  642. zone->setId(decodeZoneId(idAndZone.first));
  643. zone->serializeJson(handler);
  644. zones[zone->getId()] = zone;
  645. }
  646. }
  647. }
  648. }
  649. template<typename T>
  650. T CRmgTemplate::inheritZoneProperty(std::shared_ptr<rmg::ZoneOptions> zone,
  651. T (rmg::ZoneOptions::*getter)() const,
  652. void (rmg::ZoneOptions::*setter)(const T&),
  653. TRmgTemplateZoneId (rmg::ZoneOptions::*inheritFrom)() const,
  654. const std::string& propertyString,
  655. uint32_t iteration)
  656. {
  657. if (iteration >= 50)
  658. {
  659. logGlobal->error("Infinite recursion for %s detected in template %s", propertyString, name);
  660. return T();
  661. }
  662. if (((*zone).*inheritFrom)() != rmg::ZoneOptions::NO_ZONE)
  663. {
  664. iteration++;
  665. const auto otherZone = zones.at(((*zone).*inheritFrom)());
  666. T inheritedValue = inheritZoneProperty(otherZone, getter, setter, inheritFrom, propertyString, iteration);
  667. ((*zone).*setter)(inheritedValue);
  668. }
  669. return ((*zone).*getter)();
  670. }
  671. void CRmgTemplate::afterLoad()
  672. {
  673. for(auto & idAndZone : zones)
  674. {
  675. auto zone = idAndZone.second;
  676. // Inherit properties recursively
  677. inheritZoneProperty(zone,
  678. &rmg::ZoneOptions::getTerrainTypes,
  679. &rmg::ZoneOptions::setTerrainTypes,
  680. &rmg::ZoneOptions::getTerrainTypeLikeZone,
  681. "terrain types");
  682. inheritZoneProperty(zone,
  683. &rmg::ZoneOptions::getMinesInfo,
  684. &rmg::ZoneOptions::setMinesInfo,
  685. &rmg::ZoneOptions::getMinesLikeZone,
  686. "mine types");
  687. inheritZoneProperty(zone,
  688. &rmg::ZoneOptions::getTreasureInfo,
  689. &rmg::ZoneOptions::setTreasureInfo,
  690. &rmg::ZoneOptions::getTreasureLikeZone,
  691. "treasure info");
  692. inheritZoneProperty(zone,
  693. &rmg::ZoneOptions::getCustomObjects,
  694. &rmg::ZoneOptions::setCustomObjects,
  695. &rmg::ZoneOptions::getCustomObjectsLikeZone,
  696. "custom objects");
  697. //TODO: Inherit monster types as well
  698. auto monsterTypes = zone->getMonsterTypes();
  699. if (monsterTypes.empty())
  700. {
  701. zone->setMonsterTypes(VLC->townh->getAllowedFactions(false));
  702. }
  703. }
  704. for(const auto & connection : connectedZoneIds)
  705. {
  706. auto id1 = connection.getZoneA();
  707. auto id2 = connection.getZoneB();
  708. auto zone1 = zones.at(id1);
  709. auto zone2 = zones.at(id2);
  710. zone1->addConnection(connection);
  711. zone2->addConnection(connection);
  712. }
  713. if(allowedWaterContent.empty() || allowedWaterContent.count(EWaterContent::RANDOM))
  714. {
  715. allowedWaterContent.insert(EWaterContent::NONE);
  716. allowedWaterContent.insert(EWaterContent::NORMAL);
  717. allowedWaterContent.insert(EWaterContent::ISLANDS);
  718. }
  719. allowedWaterContent.erase(EWaterContent::RANDOM);
  720. }
  721. // TODO: Allow any integer size which does not match enum, as well
  722. void CRmgTemplate::serializeSize(JsonSerializeFormat & handler, int3 & value, const std::string & fieldName)
  723. {
  724. static const std::map<std::string, int3> sizeMapping =
  725. {
  726. {"s", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 1}},
  727. {"s+u", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 2}},
  728. {"m", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 1}},
  729. {"m+u", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 2}},
  730. {"l", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 1}},
  731. {"l+u", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 2}},
  732. {"xl", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 1}} ,
  733. {"xl+u", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 2}} ,
  734. {"h", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 1}},
  735. {"h+u", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 2}},
  736. {"xh", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 1}},
  737. {"xh+u", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 2}},
  738. {"g", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 1}},
  739. {"g+u", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 2}}
  740. };
  741. static const std::map<int3, std::string> sizeReverseMapping = vstd::invertMap(sizeMapping);
  742. std::string encodedValue;
  743. if(handler.saving)
  744. {
  745. auto iter = sizeReverseMapping.find(value);
  746. if(iter == sizeReverseMapping.end())
  747. encodedValue = boost::str(boost::format("%dx%dx%d") % value.x % value.y % value.z);
  748. else
  749. encodedValue = iter->second;
  750. }
  751. handler.serializeString(fieldName, encodedValue);
  752. if(!handler.saving)
  753. {
  754. auto iter = sizeMapping.find(encodedValue);
  755. if(iter == sizeMapping.end())
  756. {
  757. std::vector<std::string> parts;
  758. boost::split(parts, encodedValue, boost::is_any_of("x"));
  759. value.x = (boost::lexical_cast<int>(parts.at(0)));
  760. value.y = (boost::lexical_cast<int>(parts.at(1)));
  761. value.z = (boost::lexical_cast<int>(parts.at(2)));
  762. }
  763. else
  764. {
  765. value = iter->second;
  766. }
  767. }
  768. }
  769. void CRmgTemplate::serializePlayers(JsonSerializeFormat & handler, CPlayerCountRange & value, const std::string & fieldName)
  770. {
  771. std::string encodedValue;
  772. if(handler.saving)
  773. encodedValue = value.toString();
  774. handler.serializeString(fieldName, encodedValue);
  775. if(!handler.saving)
  776. value.fromString(encodedValue);
  777. }
  778. const std::vector<CompoundMapObjectID> & ZoneOptions::getBannedObjects() const
  779. {
  780. return objectConfig.getBannedObjects();
  781. }
  782. const std::vector<ObjectConfig::EObjectCategory> & ZoneOptions::getBannedObjectCategories() const
  783. {
  784. return objectConfig.getBannedObjectCategories();
  785. }
  786. const std::vector<ObjectInfo> & ZoneOptions::getConfiguredObjects() const
  787. {
  788. return objectConfig.getConfiguredObjects();
  789. }
  790. VCMI_LIB_NAMESPACE_END