CRmgTemplate.cpp 22 KB

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