CRmgTemplate.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140
  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 "../entities/ResourceTypeHandler.h"
  20. #include "../modding/ModScope.h"
  21. #include "../serializer/JsonSerializeFormat.h"
  22. #include <boost/lexical_cast.hpp>
  23. VCMI_LIB_NAMESPACE_BEGIN
  24. namespace
  25. {
  26. si32 decodeZoneId(const std::string & json)
  27. {
  28. return boost::lexical_cast<si32>(json);
  29. }
  30. std::string encodeZoneId(si32 id)
  31. {
  32. return std::to_string(id);
  33. }
  34. }
  35. CTreasureInfo::CTreasureInfo()
  36. : min(0),
  37. max(0),
  38. density(0)
  39. {
  40. }
  41. CTreasureInfo::CTreasureInfo(ui32 imin, ui32 imax, ui16 idensity)
  42. : min(imin), max(imax), density(idensity)
  43. {
  44. }
  45. bool CTreasureInfo::operator==(const CTreasureInfo & other) const
  46. {
  47. return (min == other.min) && (max == other.max) && (density == other.density);
  48. }
  49. void CTreasureInfo::serializeJson(JsonSerializeFormat & handler)
  50. {
  51. handler.serializeInt("min", min, 0);
  52. handler.serializeInt("max", max, 0);
  53. handler.serializeInt("density", density, 0);
  54. }
  55. namespace rmg
  56. {
  57. const TRmgTemplateZoneId ZoneOptions::NO_ZONE = -1;
  58. ZoneOptions::CTownInfo::CTownInfo()
  59. : townCount(0),
  60. castleCount(0),
  61. townDensity(0),
  62. castleDensity(0)
  63. {
  64. }
  65. int ZoneOptions::CTownInfo::getTownCount() const
  66. {
  67. return townCount;
  68. }
  69. int ZoneOptions::CTownInfo::getCastleCount() const
  70. {
  71. return castleCount;
  72. }
  73. int ZoneOptions::CTownInfo::getTownDensity() const
  74. {
  75. return townDensity;
  76. }
  77. int ZoneOptions::CTownInfo::getCastleDensity() const
  78. {
  79. return castleDensity;
  80. }
  81. void ZoneOptions::CTownInfo::serializeJson(JsonSerializeFormat & handler)
  82. {
  83. handler.serializeInt("towns", townCount, 0);
  84. handler.serializeInt("castles", castleCount, 0);
  85. handler.serializeInt("townDensity", townDensity, 0);
  86. handler.serializeInt("castleDensity", castleDensity, 0);
  87. handler.serializeInt("townTypesLikeZone", townTypesLikeZone, NO_ZONE);
  88. handler.serializeInt("townTypesNotLikeZone", townTypesNotLikeZone, NO_ZONE);
  89. handler.serializeInt("townTypesRelatedToZoneTerrain", townTypesRelatedToZoneTerrain, NO_ZONE);
  90. }
  91. ZoneOptions::CTownHints::CTownHints()
  92. : likeZone(NO_ZONE),
  93. relatedToZoneTerrain(NO_ZONE)
  94. {
  95. }
  96. void ZoneOptions::CTownHints::serializeJson(JsonSerializeFormat & handler)
  97. {
  98. handler.serializeInt("likeZone", likeZone, NO_ZONE);
  99. auto node = handler.getCurrent();
  100. if (node["notLikeZone"].isVector())
  101. {
  102. // TODO: Utility to serialize vector of ints?
  103. auto notLikeZoneData = handler.enterArray("notLikeZone");
  104. notLikeZone.resize(notLikeZoneData.size());
  105. for (size_t i = 0; i < notLikeZoneData.size(); ++i)
  106. {
  107. notLikeZoneData.serializeInt(i, notLikeZone[i]);
  108. }
  109. }
  110. else
  111. {
  112. notLikeZone.resize(1);
  113. handler.serializeInt("notLikeZone", notLikeZone[0], NO_ZONE);
  114. }
  115. handler.serializeInt("relatedToZoneTerrain", relatedToZoneTerrain, NO_ZONE);
  116. }
  117. ZoneOptions::ZoneOptions():
  118. id(0),
  119. type(ETemplateZoneType::PLAYER_START),
  120. size(1),
  121. maxTreasureValue(0),
  122. owner(PlayerColor(0)),
  123. matchTerrainToTown(true),
  124. townsAreSameType(false),
  125. monsterStrength(EMonsterStrength::ZONE_NORMAL),
  126. townsLikeZone(NO_ZONE),
  127. minesLikeZone(NO_ZONE),
  128. terrainTypeLikeZone(NO_ZONE),
  129. treasureLikeZone(NO_ZONE),
  130. customObjectsLikeZone(NO_ZONE),
  131. visiblePosition(Point(0, 0)),
  132. visibleSize(1.0)
  133. {
  134. }
  135. TRmgTemplateZoneId ZoneOptions::getId() const
  136. {
  137. return id;
  138. }
  139. void ZoneOptions::setId(TRmgTemplateZoneId value)
  140. {
  141. if(value <= 0)
  142. throw std::runtime_error(boost::str(boost::format("Zone %d id should be greater than 0.") % id));
  143. id = value;
  144. }
  145. ETemplateZoneType ZoneOptions::getType() const
  146. {
  147. return type;
  148. }
  149. void ZoneOptions::setType(ETemplateZoneType value)
  150. {
  151. type = value;
  152. }
  153. int ZoneOptions::getSize() const
  154. {
  155. return size;
  156. }
  157. void ZoneOptions::setSize(int value)
  158. {
  159. size = value;
  160. }
  161. std::optional<int> ZoneOptions::getOwner() const
  162. {
  163. return owner;
  164. }
  165. std::set<TerrainId> ZoneOptions::getTerrainTypes() const
  166. {
  167. if (terrainTypes.empty())
  168. {
  169. return vstd::difference(getDefaultTerrainTypes(), bannedTerrains);
  170. }
  171. else
  172. {
  173. return terrainTypes;
  174. }
  175. }
  176. void ZoneOptions::setTerrainTypes(const std::set<TerrainId> & value)
  177. {
  178. terrainTypes = value;
  179. }
  180. std::set<TerrainId> ZoneOptions::getDefaultTerrainTypes() const
  181. {
  182. std::set<TerrainId> terrains;
  183. for(const auto & terrain : LIBRARY->terrainTypeHandler->objects)
  184. {
  185. if (terrain->isLand() && terrain->isPassable())
  186. {
  187. terrains.insert(terrain->getId());
  188. }
  189. }
  190. return terrains;
  191. }
  192. std::set<FactionID> ZoneOptions::getDefaultTownTypes() const
  193. {
  194. return LIBRARY->townh->getDefaultAllowed();
  195. }
  196. std::set<FactionID> ZoneOptions::getTownTypes() const
  197. {
  198. if (townTypes.empty())
  199. {
  200. //Assume that all towns are allowed, unless banned
  201. return vstd::difference(getDefaultTownTypes(), bannedTownTypes);
  202. }
  203. else
  204. {
  205. return vstd::difference(townTypes, bannedTownTypes);
  206. }
  207. }
  208. void ZoneOptions::setTownTypes(const std::set<FactionID> & value)
  209. {
  210. townTypes = value;
  211. }
  212. void ZoneOptions::setMonsterTypes(const std::set<FactionID> & value)
  213. {
  214. monsterTypes = value;
  215. }
  216. std::set<FactionID> ZoneOptions::getMonsterTypes() const
  217. {
  218. return vstd::difference(monsterTypes, bannedMonsters);
  219. }
  220. void ZoneOptions::setMinesInfo(const std::map<TResource, ui16> & value)
  221. {
  222. mines = value;
  223. }
  224. std::map<TResource, ui16> ZoneOptions::getMinesInfo() const
  225. {
  226. return mines;
  227. }
  228. void ZoneOptions::setTreasureInfo(const std::vector<CTreasureInfo> & value)
  229. {
  230. treasureInfo = value;
  231. recalculateMaxTreasureValue();
  232. }
  233. void ZoneOptions::recalculateMaxTreasureValue()
  234. {
  235. maxTreasureValue = 0;
  236. for (const auto& ti : treasureInfo)
  237. {
  238. vstd::amax(maxTreasureValue, ti.max);
  239. }
  240. }
  241. void ZoneOptions::addTreasureInfo(const CTreasureInfo & value)
  242. {
  243. treasureInfo.push_back(value);
  244. vstd::amax(maxTreasureValue, value.max);
  245. }
  246. std::vector<CTreasureInfo> ZoneOptions::getTreasureInfo() const
  247. {
  248. return treasureInfo;
  249. }
  250. ui32 ZoneOptions::getMaxTreasureValue() const
  251. {
  252. return maxTreasureValue;
  253. }
  254. TRmgTemplateZoneId ZoneOptions::getMinesLikeZone() const
  255. {
  256. return minesLikeZone;
  257. }
  258. TRmgTemplateZoneId ZoneOptions::getTerrainTypeLikeZone() const
  259. {
  260. return terrainTypeLikeZone;
  261. }
  262. TRmgTemplateZoneId ZoneOptions::getTreasureLikeZone() const
  263. {
  264. return treasureLikeZone;
  265. }
  266. ObjectConfig ZoneOptions::getCustomObjects() const
  267. {
  268. return objectConfig;
  269. }
  270. void ZoneOptions::setCustomObjects(const ObjectConfig & value)
  271. {
  272. objectConfig = value;
  273. }
  274. TRmgTemplateZoneId ZoneOptions::getCustomObjectsLikeZone() const
  275. {
  276. return customObjectsLikeZone;
  277. }
  278. TRmgTemplateZoneId ZoneOptions::getTownsLikeZone() const
  279. {
  280. return townsLikeZone;
  281. }
  282. Point ZoneOptions::getVisiblePosition() const
  283. {
  284. return visiblePosition;
  285. }
  286. void ZoneOptions::setVisiblePosition(Point value)
  287. {
  288. visiblePosition = value;
  289. }
  290. float ZoneOptions::getVisibleSize() const
  291. {
  292. return visibleSize;
  293. }
  294. void ZoneOptions::setVisibleSize(float value)
  295. {
  296. visibleSize = value;
  297. }
  298. void ZoneOptions::addConnection(const ZoneConnection & connection)
  299. {
  300. connectedZoneIds.push_back(connection.getOtherZoneId(getId()));
  301. connectionDetails.push_back(connection);
  302. }
  303. std::vector<ZoneConnection> ZoneOptions::getConnections() const
  304. {
  305. return connectionDetails;
  306. }
  307. std::vector<ZoneConnection>& ZoneOptions::getConnectionsRef()
  308. {
  309. return connectionDetails;
  310. }
  311. void ZoneOptions::setRoadOption(int connectionId, rmg::ERoadOption roadOption)
  312. {
  313. for(auto & connection : connectionDetails)
  314. {
  315. if(connection.getId() == connectionId)
  316. {
  317. connection.setRoadOption(roadOption);
  318. logGlobal->info("Set road option for connection %d between zones %d and %d to %s",
  319. connectionId, connection.getZoneA(), connection.getZoneB(),
  320. roadOption == rmg::ERoadOption::ROAD_TRUE ? "true" :
  321. roadOption == rmg::ERoadOption::ROAD_FALSE ? "false" : "random");
  322. return;
  323. }
  324. }
  325. logGlobal->warn("Failed to find connection with ID %d in zone %d", connectionId, id);
  326. }
  327. std::vector<TRmgTemplateZoneId> ZoneOptions::getConnectedZoneIds() const
  328. {
  329. return connectedZoneIds;
  330. }
  331. bool ZoneOptions::areTownsSameType() const
  332. {
  333. return townsAreSameType;
  334. }
  335. bool ZoneOptions::isMatchTerrainToTown() const
  336. {
  337. return matchTerrainToTown;
  338. }
  339. void ZoneOptions::setMatchTerrainToTown(bool value)
  340. {
  341. matchTerrainToTown = value;
  342. }
  343. const ZoneOptions::CTownInfo & ZoneOptions::getPlayerTowns() const
  344. {
  345. return playerTowns;
  346. }
  347. void ZoneOptions::setPlayerTowns(const CTownInfo & value)
  348. {
  349. playerTowns = value;
  350. }
  351. const ZoneOptions::CTownInfo & ZoneOptions::getNeutralTowns() const
  352. {
  353. return neutralTowns;
  354. }
  355. void ZoneOptions::setNeutralTowns(const CTownInfo & value)
  356. {
  357. neutralTowns = value;
  358. }
  359. const std::vector<ZoneOptions::CTownHints> & ZoneOptions::getTownHints() const
  360. {
  361. return townHints;
  362. }
  363. void ZoneOptions::setTownHints(const std::vector<CTownHints> & value)
  364. {
  365. townHints = value;
  366. }
  367. std::set<FactionID> ZoneOptions::getBannedTownTypes() const
  368. {
  369. return bannedTownTypes;
  370. }
  371. void ZoneOptions::setBannedTownTypes(const std::set<FactionID> & value)
  372. {
  373. bannedTownTypes = value;
  374. }
  375. void ZoneOptions::serializeJson(JsonSerializeFormat & handler)
  376. {
  377. static const std::vector<std::string> zoneTypes =
  378. {
  379. "playerStart",
  380. "cpuStart",
  381. "treasure",
  382. "junction",
  383. "water",
  384. "sealed"
  385. };
  386. handler.serializeEnum("type", type, zoneTypes);
  387. handler.serializeInt("size", size, 1);
  388. handler.serializeInt("owner", owner);
  389. handler.serializeStruct("playerTowns", playerTowns);
  390. handler.serializeStruct("neutralTowns", neutralTowns);
  391. handler.serializeBool("matchTerrainToTown", matchTerrainToTown, true);
  392. #define SERIALIZE_ZONE_LINK(fieldName) handler.serializeInt(#fieldName, fieldName, NO_ZONE);
  393. SERIALIZE_ZONE_LINK(townsLikeZone);
  394. SERIALIZE_ZONE_LINK(minesLikeZone);
  395. SERIALIZE_ZONE_LINK(terrainTypeLikeZone);
  396. SERIALIZE_ZONE_LINK(treasureLikeZone);
  397. SERIALIZE_ZONE_LINK(customObjectsLikeZone);
  398. #undef SERIALIZE_ZONE_LINK
  399. if(terrainTypeLikeZone == NO_ZONE)
  400. {
  401. handler.serializeIdArray("terrainTypes", terrainTypes);
  402. handler.serializeIdArray("bannedTerrains", bannedTerrains);
  403. }
  404. handler.serializeBool("townsAreSameType", townsAreSameType, false);
  405. handler.serializeIdArray("allowedMonsters", monsterTypes);
  406. handler.serializeIdArray("bannedMonsters", bannedMonsters);
  407. handler.serializeIdArray("allowedTowns", townTypes);
  408. handler.serializeIdArray("bannedTowns", bannedTownTypes);
  409. handler.enterArray("townHints").serializeStruct(townHints);
  410. {
  411. //TODO: add support for std::map to serializeEnum
  412. static const std::vector<std::string> zoneMonsterStrengths =
  413. {
  414. "none",
  415. "weak",
  416. "normal",
  417. "strong"
  418. };
  419. int temporaryZoneMonsterStrengthIndex = monsterStrength == EMonsterStrength::ZONE_NONE ? 0 : monsterStrength - EMonsterStrength::ZONE_WEAK + 1 ; // temporary until serializeEnum starts supporting std::map
  420. // temporaryZoneMonsterStrengthIndex = 0, 1, 2 and 3 for monsterStrength = ZONE_NONE, ZONE_WEAK, ZONE_NORMAL and ZONE_STRONG respectively
  421. handler.serializeEnum("monsters", temporaryZoneMonsterStrengthIndex, 2, zoneMonsterStrengths); // default is normal monsters
  422. switch (temporaryZoneMonsterStrengthIndex)
  423. {
  424. case 0:
  425. monsterStrength = EMonsterStrength::ZONE_NONE;
  426. break;
  427. case 1:
  428. monsterStrength = EMonsterStrength::ZONE_WEAK;
  429. break;
  430. case 2:
  431. monsterStrength = EMonsterStrength::ZONE_NORMAL;
  432. break;
  433. case 3:
  434. monsterStrength = EMonsterStrength::ZONE_STRONG;
  435. break;
  436. }
  437. }
  438. if(treasureLikeZone == NO_ZONE)
  439. {
  440. auto treasureData = handler.enterArray("treasure");
  441. treasureData.serializeStruct(treasureInfo);
  442. if (!handler.saving)
  443. {
  444. recalculateMaxTreasureValue();
  445. }
  446. }
  447. if((minesLikeZone == NO_ZONE) && (!handler.saving || !mines.empty()))
  448. {
  449. auto minesData = handler.enterStruct("mines");
  450. for(auto & idx : LIBRARY->resourceTypeHandler->getAllObjects())
  451. if(idx != GameResID::MITHRIL)
  452. handler.serializeInt(LIBRARY->resourceTypeHandler->getById(idx)->getJsonKey(), mines[idx], 0);
  453. }
  454. handler.serializeStruct("customObjects", objectConfig);
  455. handler.serializeInt("visiblePositionX", visiblePosition.x);
  456. handler.serializeInt("visiblePositionY", visiblePosition.y);
  457. handler.serializeFloat("visibleSize", visibleSize);
  458. if(!handler.saving && visibleSize < 0.01)
  459. visibleSize = 1.0;
  460. }
  461. ZoneConnection::ZoneConnection():
  462. id(-1),
  463. zoneA(-1),
  464. zoneB(-1),
  465. guardStrength(0),
  466. connectionType(rmg::EConnectionType::GUARDED),
  467. hasRoad(rmg::ERoadOption::ROAD_RANDOM)
  468. {
  469. }
  470. int ZoneConnection::getId() const
  471. {
  472. return id;
  473. }
  474. void ZoneConnection::setId(int id)
  475. {
  476. this->id = id;
  477. }
  478. TRmgTemplateZoneId ZoneConnection::getZoneA() const
  479. {
  480. return zoneA;
  481. }
  482. TRmgTemplateZoneId ZoneConnection::getZoneB() const
  483. {
  484. return zoneB;
  485. }
  486. TRmgTemplateZoneId ZoneConnection::getOtherZoneId(TRmgTemplateZoneId id) const
  487. {
  488. if (id == zoneA)
  489. {
  490. return zoneB;
  491. }
  492. else if (id == zoneB)
  493. {
  494. return zoneA;
  495. }
  496. else
  497. {
  498. throw rmgException("Zone does not belong to this connection");
  499. }
  500. }
  501. int ZoneConnection::getGuardStrength() const
  502. {
  503. return guardStrength;
  504. }
  505. rmg::EConnectionType ZoneConnection::getConnectionType() const
  506. {
  507. return connectionType;
  508. }
  509. rmg::ERoadOption ZoneConnection::getRoadOption() const
  510. {
  511. return hasRoad;
  512. }
  513. void ZoneConnection::setRoadOption(rmg::ERoadOption roadOption)
  514. {
  515. hasRoad = roadOption;
  516. }
  517. bool operator==(const ZoneConnection & l, const ZoneConnection & r)
  518. {
  519. return l.id == r.id;
  520. }
  521. bool operator<(const ZoneConnection & l, const ZoneConnection & r)
  522. {
  523. return l.id < r.id;
  524. }
  525. void ZoneConnection::serializeJson(JsonSerializeFormat & handler)
  526. {
  527. static const std::vector<std::string> connectionTypes =
  528. {
  529. "guarded",
  530. "fictive",
  531. "repulsive",
  532. "wide",
  533. "forcePortal"
  534. };
  535. static const std::vector<std::string> roadOptions =
  536. {
  537. "random",
  538. "true",
  539. "false"
  540. };
  541. if (handler.saving)
  542. {
  543. std::string zoneNameA = std::to_string(zoneA);
  544. std::string zoneNameB = std::to_string(zoneB);
  545. handler.serializeString("a", zoneNameA);
  546. handler.serializeString("b", zoneNameB);
  547. }
  548. else
  549. {
  550. std::string zoneNameA;
  551. std::string zoneNameB;
  552. handler.serializeString("a", zoneNameA);
  553. handler.serializeString("b", zoneNameB);
  554. zoneA = std::stoi(zoneNameA);
  555. zoneB = std::stoi(zoneNameB);
  556. }
  557. handler.serializeInt("guard", guardStrength, 0);
  558. handler.serializeEnum("type", connectionType, connectionTypes);
  559. handler.serializeEnum("road", hasRoad, roadOptions);
  560. }
  561. }
  562. using namespace rmg;//todo: remove
  563. CRmgTemplate::~CRmgTemplate() = default;
  564. CRmgTemplate::CRmgTemplate()
  565. : minSize(72, 72, 2),
  566. maxSize(72, 72, 2),
  567. mapSettings(std::make_unique<JsonNode>())
  568. {
  569. }
  570. bool CRmgTemplate::matchesSize(const int3 & value) const
  571. {
  572. const int64_t square = value.x * value.y * value.z;
  573. const int64_t minSquare = minSize.x * minSize.y * minSize.z;
  574. const int64_t maxSquare = maxSize.x * maxSize.y * maxSize.z;
  575. return minSquare <= square && square <= maxSquare;
  576. }
  577. bool CRmgTemplate::isWaterContentAllowed(EWaterContent::EWaterContent waterContent) const
  578. {
  579. return waterContent == EWaterContent::RANDOM || allowedWaterContent.count(waterContent);
  580. }
  581. const std::set<EWaterContent::EWaterContent> & CRmgTemplate::getWaterContentAllowed() const
  582. {
  583. return allowedWaterContent;
  584. }
  585. void CRmgTemplate::setId(const std::string & value)
  586. {
  587. id = value;
  588. }
  589. void CRmgTemplate::setName(const std::string & value)
  590. {
  591. name = value;
  592. }
  593. const std::string & CRmgTemplate::getName() const
  594. {
  595. return name;
  596. }
  597. const std::string & CRmgTemplate::getDescription() const
  598. {
  599. return description;
  600. }
  601. const std::string & CRmgTemplate::getId() const
  602. {
  603. return id;
  604. }
  605. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getPlayers() const
  606. {
  607. return players;
  608. }
  609. const CRmgTemplate::CPlayerCountRange & CRmgTemplate::getHumanPlayers() const
  610. {
  611. return humanPlayers;
  612. }
  613. const CRmgTemplate::Zones & CRmgTemplate::getZones() const
  614. {
  615. return zones;
  616. }
  617. const std::vector<ZoneConnection> & CRmgTemplate::getConnectedZoneIds() const
  618. {
  619. return connections;
  620. }
  621. void CRmgTemplate::validate() const
  622. {
  623. //TODO add some validation checks, throw on failure
  624. }
  625. std::pair<int3, int3> CRmgTemplate::getMapSizes() const
  626. {
  627. return {minSize, maxSize};
  628. }
  629. void CRmgTemplate::CPlayerCountRange::addRange(int lower, int upper)
  630. {
  631. range.emplace_back(lower, upper);
  632. }
  633. void CRmgTemplate::CPlayerCountRange::addNumber(int value)
  634. {
  635. range.emplace_back(value, value);
  636. }
  637. bool CRmgTemplate::CPlayerCountRange::isInRange(int count) const
  638. {
  639. for(const auto & pair : range)
  640. {
  641. if(count >= pair.first && count <= pair.second) return true;
  642. }
  643. return false;
  644. }
  645. std::set<int> CRmgTemplate::CPlayerCountRange::getNumbers() const
  646. {
  647. std::set<int> numbers;
  648. for(const auto & pair : range)
  649. {
  650. for(int i = pair.first; i <= pair.second; ++i) numbers.insert(i);
  651. }
  652. return numbers;
  653. }
  654. std::string CRmgTemplate::CPlayerCountRange::toString() const
  655. {
  656. if(range.size() == 1)
  657. {
  658. const auto & p = range.front();
  659. if((p.first == p.second) && (p.first == 0))
  660. return "";
  661. }
  662. std::string ret;
  663. bool first = true;
  664. for(const auto & p : range)
  665. {
  666. if(!first)
  667. ret +=",";
  668. else
  669. first = false;
  670. if(p.first == p.second)
  671. {
  672. ret += std::to_string(p.first);
  673. }
  674. else
  675. {
  676. ret += boost::str(boost::format("%d-%d") % p.first % p.second);
  677. }
  678. }
  679. return ret;
  680. }
  681. void CRmgTemplate::CPlayerCountRange::fromString(const std::string & value)
  682. {
  683. range.clear();
  684. if(value.empty())
  685. {
  686. addNumber(0);
  687. }
  688. else
  689. {
  690. std::vector<std::string> commaParts;
  691. boost::split(commaParts, value, boost::is_any_of(","));
  692. for(const auto & commaPart : commaParts)
  693. {
  694. std::vector<std::string> rangeParts;
  695. boost::split(rangeParts, commaPart, boost::is_any_of("-"));
  696. if(rangeParts.size() == 2)
  697. {
  698. auto lower = boost::lexical_cast<int>(rangeParts[0]);
  699. auto upper = boost::lexical_cast<int>(rangeParts[1]);
  700. addRange(lower, upper);
  701. }
  702. else if(rangeParts.size() == 1)
  703. {
  704. auto val = boost::lexical_cast<int>(rangeParts.front());
  705. addNumber(val);
  706. }
  707. }
  708. }
  709. }
  710. int CRmgTemplate::CPlayerCountRange::maxValue() const
  711. {
  712. return *boost::max_element(getNumbers());
  713. }
  714. int CRmgTemplate::CPlayerCountRange::minValue() const
  715. {
  716. return *boost::min_element(getNumbers());
  717. }
  718. void CRmgTemplate::serializeJson(JsonSerializeFormat & handler)
  719. {
  720. handler.serializeString("name", name);
  721. handler.serializeString("description", description);
  722. serializeSize(handler, minSize, "minSize");
  723. serializeSize(handler, maxSize, "maxSize");
  724. serializePlayers(handler, players, "players");
  725. serializePlayers(handler, humanPlayers, "humans"); // TODO: Rename this parameter
  726. handler.serializeIdArray("bannedSpells", bannedSpells);
  727. handler.serializeIdArray("bannedArtifacts", bannedArtifacts);
  728. handler.serializeIdArray("bannedSkills", bannedSkills);
  729. handler.serializeIdArray("bannedHeroes", bannedHeroes);
  730. *mapSettings = handler.getCurrent()["settings"];
  731. {
  732. auto connectionsData = handler.enterArray("connections");
  733. connectionsData.serializeStruct(connections);
  734. if(!handler.saving)
  735. {
  736. for(size_t i = 0; i < connections.size(); ++i)
  737. {
  738. connections[i].setId(i);
  739. }
  740. }
  741. }
  742. {
  743. boost::bimap<EWaterContent::EWaterContent, std::string> enc;
  744. enc.insert({EWaterContent::NONE, "none"});
  745. enc.insert({EWaterContent::NORMAL, "normal"});
  746. enc.insert({EWaterContent::ISLANDS, "islands"});
  747. JsonNode node;
  748. if(handler.saving)
  749. {
  750. node.setType(JsonNode::JsonType::DATA_VECTOR);
  751. for(auto wc : allowedWaterContent)
  752. {
  753. JsonNode n;
  754. n.String() = enc.left.at(wc);
  755. node.Vector().push_back(n);
  756. }
  757. }
  758. handler.serializeRaw("allowedWaterContent", node, std::nullopt);
  759. if(!handler.saving)
  760. {
  761. for(auto wc : node.Vector())
  762. {
  763. allowedWaterContent.insert(enc.right.at(std::string(wc.String())));
  764. }
  765. }
  766. }
  767. {
  768. auto zonesData = handler.enterStruct("zones");
  769. if(handler.saving)
  770. {
  771. for(auto & idAndZone : zones)
  772. {
  773. auto guard = handler.enterStruct(encodeZoneId(idAndZone.first));
  774. idAndZone.second->serializeJson(handler);
  775. }
  776. }
  777. else
  778. {
  779. for(const auto & idAndZone : zonesData->getCurrent().Struct())
  780. {
  781. auto guard = handler.enterStruct(idAndZone.first);
  782. auto zone = std::make_shared<ZoneOptions>();
  783. zone->setId(decodeZoneId(idAndZone.first));
  784. zone->serializeJson(handler);
  785. zones[zone->getId()] = zone;
  786. }
  787. }
  788. }
  789. }
  790. const JsonNode & CRmgTemplate::getMapSettings() const
  791. {
  792. return *mapSettings;
  793. }
  794. template<typename T>
  795. T CRmgTemplate::inheritZoneProperty(std::shared_ptr<rmg::ZoneOptions> zone,
  796. T (rmg::ZoneOptions::*getter)() const,
  797. void (rmg::ZoneOptions::*setter)(const T&),
  798. TRmgTemplateZoneId (rmg::ZoneOptions::*inheritFrom)() const,
  799. const std::string& propertyString,
  800. uint32_t iteration)
  801. {
  802. if (iteration >= 50)
  803. {
  804. logGlobal->error("Infinite recursion for %s detected in template %s", propertyString, name);
  805. return T();
  806. }
  807. if (((*zone).*inheritFrom)() != rmg::ZoneOptions::NO_ZONE)
  808. {
  809. iteration++;
  810. const auto otherZone = zones.at(((*zone).*inheritFrom)());
  811. T inheritedValue = inheritZoneProperty(otherZone, getter, setter, inheritFrom, propertyString, iteration);
  812. ((*zone).*setter)(inheritedValue);
  813. }
  814. return ((*zone).*getter)();
  815. }
  816. void CRmgTemplate::afterLoad()
  817. {
  818. for(auto & idAndZone : zones)
  819. {
  820. auto zone = idAndZone.second;
  821. // Inherit properties recursively
  822. inheritTownProperties(zone);
  823. inheritZoneProperty(zone,
  824. &rmg::ZoneOptions::getTerrainTypes,
  825. &rmg::ZoneOptions::setTerrainTypes,
  826. &rmg::ZoneOptions::getTerrainTypeLikeZone,
  827. "terrain types");
  828. inheritZoneProperty(zone,
  829. &rmg::ZoneOptions::getMinesInfo,
  830. &rmg::ZoneOptions::setMinesInfo,
  831. &rmg::ZoneOptions::getMinesLikeZone,
  832. "mine types");
  833. inheritZoneProperty(zone,
  834. &rmg::ZoneOptions::getTreasureInfo,
  835. &rmg::ZoneOptions::setTreasureInfo,
  836. &rmg::ZoneOptions::getTreasureLikeZone,
  837. "treasure info");
  838. inheritZoneProperty(zone,
  839. &rmg::ZoneOptions::getCustomObjects,
  840. &rmg::ZoneOptions::setCustomObjects,
  841. &rmg::ZoneOptions::getCustomObjectsLikeZone,
  842. "custom objects");
  843. //TODO: Inherit monster types as well
  844. auto monsterTypes = zone->getMonsterTypes();
  845. if (monsterTypes.empty())
  846. {
  847. zone->setMonsterTypes(LIBRARY->townh->getAllowedFactions(false));
  848. }
  849. }
  850. for(const auto & connection : connections)
  851. {
  852. auto id1 = connection.getZoneA();
  853. auto id2 = connection.getZoneB();
  854. auto zone1 = zones.at(id1);
  855. auto zone2 = zones.at(id2);
  856. zone1->addConnection(connection);
  857. zone2->addConnection(connection);
  858. }
  859. if(allowedWaterContent.empty() || allowedWaterContent.count(EWaterContent::RANDOM))
  860. {
  861. allowedWaterContent.insert(EWaterContent::NONE);
  862. allowedWaterContent.insert(EWaterContent::NORMAL);
  863. allowedWaterContent.insert(EWaterContent::ISLANDS);
  864. }
  865. allowedWaterContent.erase(EWaterContent::RANDOM);
  866. }
  867. void CRmgTemplate::inheritTownProperties(std::shared_ptr<rmg::ZoneOptions> zone, uint32_t iteration)
  868. {
  869. if (iteration >= 50)
  870. {
  871. logGlobal->error("Infinite recursion for town properties detected in template %s", name);
  872. return;
  873. }
  874. if (zone->getTownsLikeZone() != rmg::ZoneOptions::NO_ZONE)
  875. {
  876. const auto otherZone = zones.at(zone->getTownsLikeZone());
  877. // Recursively inherit from the source zone first
  878. inheritTownProperties(otherZone, iteration + 1);
  879. // Now copy all town-related properties from the source zone
  880. zone->setPlayerTowns(otherZone->getPlayerTowns());
  881. zone->setNeutralTowns(otherZone->getNeutralTowns());
  882. zone->setMatchTerrainToTown(otherZone->isMatchTerrainToTown());
  883. zone->setTownHints(otherZone->getTownHints());
  884. zone->setTownTypes(otherZone->getTownTypes());
  885. zone->setBannedTownTypes(otherZone->getBannedTownTypes());
  886. }
  887. }
  888. // TODO: Allow any integer size which does not match enum, as well
  889. void CRmgTemplate::serializeSize(JsonSerializeFormat & handler, int3 & value, const std::string & fieldName)
  890. {
  891. static const std::map<std::string, int3> sizeMapping =
  892. {
  893. {"s", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 1}},
  894. {"s+u", {CMapHeader::MAP_SIZE_SMALL, CMapHeader::MAP_SIZE_SMALL, 2}},
  895. {"m", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 1}},
  896. {"m+u", {CMapHeader::MAP_SIZE_MIDDLE, CMapHeader::MAP_SIZE_MIDDLE, 2}},
  897. {"l", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 1}},
  898. {"l+u", {CMapHeader::MAP_SIZE_LARGE, CMapHeader::MAP_SIZE_LARGE, 2}},
  899. {"xl", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 1}} ,
  900. {"xl+u", {CMapHeader::MAP_SIZE_XLARGE, CMapHeader::MAP_SIZE_XLARGE, 2}} ,
  901. {"h", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 1}},
  902. {"h+u", {CMapHeader::MAP_SIZE_HUGE, CMapHeader::MAP_SIZE_HUGE, 2}},
  903. {"xh", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 1}},
  904. {"xh+u", {CMapHeader::MAP_SIZE_XHUGE, CMapHeader::MAP_SIZE_XHUGE, 2}},
  905. {"g", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 1}},
  906. {"g+u", {CMapHeader::MAP_SIZE_GIANT, CMapHeader::MAP_SIZE_GIANT, 2}}
  907. };
  908. static const std::map<int3, std::string> sizeReverseMapping = vstd::invertMap(sizeMapping);
  909. std::string encodedValue;
  910. if(handler.saving)
  911. {
  912. auto iter = sizeReverseMapping.find(value);
  913. if(iter == sizeReverseMapping.end())
  914. encodedValue = boost::str(boost::format("%dx%dx%d") % value.x % value.y % value.z);
  915. else
  916. encodedValue = iter->second;
  917. }
  918. handler.serializeString(fieldName, encodedValue);
  919. if(!handler.saving)
  920. {
  921. auto iter = sizeMapping.find(encodedValue);
  922. if(iter == sizeMapping.end())
  923. {
  924. std::vector<std::string> parts;
  925. boost::split(parts, encodedValue, boost::is_any_of("x"));
  926. value.x = (boost::lexical_cast<int>(parts.at(0)));
  927. value.y = (boost::lexical_cast<int>(parts.at(1)));
  928. value.z = (boost::lexical_cast<int>(parts.at(2)));
  929. }
  930. else
  931. {
  932. value = iter->second;
  933. }
  934. }
  935. }
  936. void CRmgTemplate::serializePlayers(JsonSerializeFormat & handler, CPlayerCountRange & value, const std::string & fieldName)
  937. {
  938. std::string encodedValue;
  939. if(handler.saving)
  940. encodedValue = value.toString();
  941. handler.serializeString(fieldName, encodedValue);
  942. if(!handler.saving)
  943. value.fromString(encodedValue);
  944. }
  945. const std::vector<CompoundMapObjectID> & ZoneOptions::getBannedObjects() const
  946. {
  947. return objectConfig.getBannedObjects();
  948. }
  949. const std::vector<ObjectConfig::EObjectCategory> & ZoneOptions::getBannedObjectCategories() const
  950. {
  951. return objectConfig.getBannedObjectCategories();
  952. }
  953. const std::vector<ObjectInfo> & ZoneOptions::getConfiguredObjects() const
  954. {
  955. return objectConfig.getConfiguredObjects();
  956. }
  957. VCMI_LIB_NAMESPACE_END