MapFormatJson.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903
  1. /*
  2. * MapFormatJson.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 "MapFormatJson.h"
  12. #include "../filesystem/CInputStream.h"
  13. #include "../filesystem/COutputStream.h"
  14. #include "CMap.h"
  15. #include "../CModHandler.h"
  16. #include "../CHeroHandler.h"
  17. #include "../CTownHandler.h"
  18. #include "../VCMI_Lib.h"
  19. #include "../mapObjects/ObjectTemplate.h"
  20. #include "../mapObjects/CObjectHandler.h"
  21. #include "../mapObjects/CObjectClassesHandler.h"
  22. #include "../mapObjects/CGHeroInstance.h"
  23. #include "../mapObjects/CGTownInstance.h"
  24. #include "../StringConstants.h"
  25. namespace TriggeredEventsDetail
  26. {
  27. static const std::array<std::string, 12> conditionNames =
  28. {
  29. "haveArtifact", "haveCreatures", "haveResources", "haveBuilding",
  30. "control", "destroy", "transport", "daysPassed",
  31. "isHuman", "daysWithoutTown", "standardWin", "constValue"
  32. };
  33. static const std::array<std::string, 2> typeNames = { "victory", "defeat" };
  34. static EventCondition JsonToCondition(const JsonNode & node)
  35. {
  36. //todo: support of new condition format
  37. EventCondition event;
  38. const auto & conditionName = node.Vector()[0].String();
  39. auto pos = vstd::find_pos(conditionNames, conditionName);
  40. event.condition = EventCondition::EWinLoseType(pos));
  41. if (node.Vector().size() > 1)
  42. {
  43. const JsonNode & data = node.Vector()[1];
  44. if (data["type"].getType() == JsonNode::DATA_STRING)
  45. {
  46. auto identifier = VLC->modh->identifiers.getIdentifier(data["type"]);
  47. if(identifier)
  48. event.objectType = identifier.get();
  49. else
  50. throw std::runtime_error("Identifier resolution failed in event condition");
  51. }
  52. if (data["type"].getType() == JsonNode::DATA_FLOAT)
  53. event.objectType = data["type"].Float();
  54. if (!data["value"].isNull())
  55. event.value = data["value"].Float();
  56. if (!data["position"].isNull())
  57. {
  58. auto & position = data["position"].Vector();
  59. event.position.x = position.at(0).Float();
  60. event.position.y = position.at(1).Float();
  61. event.position.z = position.at(2).Float();
  62. }
  63. }
  64. return event;
  65. }
  66. static JsonNode ConditionToJson(const EventCondition& event)
  67. {
  68. JsonNode json;
  69. JsonVector& asVector = json.Vector();
  70. JsonNode condition;
  71. condition.String() = conditionNames.at(event.condition);
  72. asVector.push_back(condition);
  73. JsonNode data;
  74. //todo: save identifier
  75. if(event.objectType != -1)
  76. data["type"].Float() = event.objectType;
  77. if(event.value != -1)
  78. data["value"].Float() = event.value;
  79. if(event.position != int3(-1,-1,-1))
  80. {
  81. auto & position = data["position"].Vector();
  82. position.resize(3);
  83. position[0].Float() = event.position.x;
  84. position[1].Float() = event.position.y;
  85. position[2].Float() = event.position.z;
  86. }
  87. if(!data.isNull())
  88. asVector.push_back(data);
  89. return std::move(json);
  90. }
  91. }//namespace TriggeredEventsDetail
  92. namespace TerrainDetail
  93. {
  94. static const std::array<std::string, 10> terrainCodes =
  95. {
  96. "dt", "sa", "gr", "sn", "sw", "rg", "sb", "lv", "wt", "rc"
  97. };
  98. static const std::array<std::string, 4> roadCodes =
  99. {
  100. "", "pd", "pg", "pc"
  101. };
  102. static const std::array<std::string, 5> riverCodes =
  103. {
  104. "", "rw", "ri", "rm", "rl"
  105. };
  106. static const std::array<char, 10> flipCodes =
  107. {
  108. '_', '-', '|', '+'
  109. };
  110. }
  111. static std::string tailString(const std::string & input, char separator)
  112. {
  113. std::string ret;
  114. size_t splitPos = input.find(separator);
  115. if (splitPos != std::string::npos)
  116. ret = input.substr(splitPos + 1);
  117. return ret;
  118. }
  119. static si32 extractNumber(const std::string & input, char separator)
  120. {
  121. std::string tmp = tailString(input, separator);
  122. return atoi(tmp.c_str());
  123. }
  124. ///CMapFormatJson
  125. const int CMapFormatJson::VERSION_MAJOR = 1;
  126. const int CMapFormatJson::VERSION_MINOR = 0;
  127. const std::string CMapFormatJson::HEADER_FILE_NAME = "header.json";
  128. const std::string CMapFormatJson::OBJECTS_FILE_NAME = "objects.json";
  129. void CMapFormatJson::readTriggeredEvents(const JsonNode & input)
  130. {
  131. mapHeader->victoryMessage = input["victoryString"].String();
  132. mapHeader->victoryIconIndex = input["victoryIconIndex"].Float();
  133. mapHeader->defeatMessage = input["defeatString"].String();
  134. mapHeader->defeatIconIndex = input["defeatIconIndex"].Float();
  135. mapHeader->triggeredEvents.clear();
  136. for (auto & entry : input["triggeredEvents"].Struct())
  137. {
  138. TriggeredEvent event;
  139. event.identifier = entry.first;
  140. readTriggeredEvent(event, entry.second);
  141. mapHeader->triggeredEvents.push_back(event);
  142. }
  143. }
  144. void CMapFormatJson::readTriggeredEvent(TriggeredEvent & event, const JsonNode & source)
  145. {
  146. using namespace TriggeredEventsDetail;
  147. event.onFulfill = source["message"].String();
  148. event.description = source["description"].String();
  149. event.effect.type = vstd::find_pos(typeNames, source["effect"]["type"].String());
  150. event.effect.toOtherMessage = source["effect"]["messageToSend"].String();
  151. event.trigger = EventExpression(source["condition"], JsonToCondition); // logical expression
  152. }
  153. void CMapFormatJson::writeTriggeredEvents(JsonNode& output)
  154. {
  155. output["victoryString"].String() = map->victoryMessage;
  156. output["victoryIconIndex"].Float() = map->victoryIconIndex;
  157. output["defeatString"].String() = map->defeatMessage;
  158. output["defeatIconIndex"].Float() = map->defeatIconIndex;
  159. JsonMap & triggeredEvents = output["triggeredEvents"].Struct();
  160. for(auto event : map->triggeredEvents)
  161. writeTriggeredEvent(event, triggeredEvents[event.identifier]);
  162. }
  163. void CMapFormatJson::writeTriggeredEvent(const TriggeredEvent& event, JsonNode& dest)
  164. {
  165. using namespace TriggeredEventsDetail;
  166. dest["message"].String() = event.onFulfill;
  167. dest["description"].String() = event.description;
  168. dest["effect"]["type"].String() = typeNames.at(size_t(event.effect.type));
  169. dest["effect"]["messageToSend"].String() = event.effect.toOtherMessage;
  170. dest["condition"] = event.trigger.toJson(ConditionToJson);
  171. }
  172. ///CMapPatcher
  173. CMapPatcher::CMapPatcher(JsonNode stream):
  174. input(stream)
  175. {
  176. }
  177. void CMapPatcher::patchMapHeader(std::unique_ptr<CMapHeader> & header)
  178. {
  179. header.swap(mapHeader);
  180. if (!input.isNull())
  181. readPatchData();
  182. header.swap(mapHeader);
  183. }
  184. void CMapPatcher::readPatchData()
  185. {
  186. readTriggeredEvents(input);
  187. }
  188. ///CMapLoaderJson
  189. CMapLoaderJson::CMapLoaderJson(CInputStream * stream):
  190. buffer(stream),
  191. ioApi(new CProxyROIOApi(buffer)),
  192. loader("", "_", ioApi)
  193. {
  194. }
  195. si32 CMapLoaderJson::getIdentifier(const std::string& type, const std::string& name)
  196. {
  197. boost::optional<si32> res = VLC->modh->identifiers.getIdentifier("core", type, name, false);
  198. if(!res)
  199. {
  200. throw new std::runtime_error("Map load failed. Identifier not resolved.");
  201. }
  202. return res.get();
  203. }
  204. std::unique_ptr<CMap> CMapLoaderJson::loadMap()
  205. {
  206. LOG_TRACE(logGlobal);
  207. map = new CMap();
  208. mapHeader = std::unique_ptr<CMapHeader>(dynamic_cast<CMapHeader *>(map));
  209. readMap();
  210. return std::unique_ptr<CMap>(dynamic_cast<CMap *>(mapHeader.release()));
  211. }
  212. std::unique_ptr<CMapHeader> CMapLoaderJson::loadMapHeader()
  213. {
  214. LOG_TRACE(logGlobal);
  215. mapHeader.reset(new CMapHeader);
  216. readHeader();
  217. return std::move(mapHeader);
  218. }
  219. const JsonNode CMapLoaderJson::readJson(const std::string & archiveFilename)
  220. {
  221. ResourceID resource(archiveFilename, EResType::TEXT);
  222. if(!loader.existsResource(resource))
  223. throw new std::runtime_error(archiveFilename+" not found");
  224. auto data = loader.load(resource)->readAll();
  225. JsonNode res(reinterpret_cast<char*>(data.first.get()), data.second);
  226. return std::move(res);
  227. }
  228. void CMapLoaderJson::readMap()
  229. {
  230. LOG_TRACE(logGlobal);
  231. readHeader();
  232. map->initTerrain();
  233. readTerrain();
  234. readObjects();
  235. // Calculate blocked / visitable positions
  236. for(auto & elem : map->objects)
  237. {
  238. map->addBlockVisTiles(elem);
  239. }
  240. map->calculateGuardingGreaturePositions();
  241. }
  242. void CMapLoaderJson::readHeader()
  243. {
  244. //do not use map field here, use only mapHeader
  245. const JsonNode header = readJson(HEADER_FILE_NAME);
  246. mapHeader->version = EMapFormat::VCMI;//todo: new version field
  247. //todo: multilevel map load support
  248. const JsonNode levels = header["mapLevels"];
  249. mapHeader->height = levels["surface"]["height"].Float();
  250. mapHeader->width = levels["surface"]["width"].Float();
  251. mapHeader->twoLevel = !levels["underground"].isNull();
  252. mapHeader->name = header["name"].String();
  253. mapHeader->description = header["description"].String();
  254. //todo: support arbitrary percentage
  255. static const std::map<std::string, ui8> difficultyMap =
  256. {
  257. {"", 1},
  258. {"EASY", 0},
  259. {"NORMAL", 1},
  260. {"HARD", 2},
  261. {"EXPERT", 3},
  262. {"IMPOSSIBLE", 4}
  263. };
  264. mapHeader->difficulty = difficultyMap.at(header["difficulty"].String());
  265. mapHeader->levelLimit = header["levelLimit"].Float();
  266. // std::vector<bool> allowedHeroes;
  267. // std::vector<ui16> placeholdedHeroes;
  268. readTriggeredEvents(header);
  269. readPlayerInfo(header);
  270. readTeams(header);
  271. //TODO: readHeader
  272. }
  273. void CMapLoaderJson::readPlayerInfo(const JsonNode & input)
  274. {
  275. const JsonNode & src = input["players"];
  276. int howManyTeams = 0;
  277. for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
  278. {
  279. PlayerInfo & info = mapHeader->players.at(player);
  280. const JsonNode & playerSrc = src[GameConstants::PLAYER_COLOR_NAMES[player]];
  281. if(playerSrc.isNull())
  282. {
  283. info.canComputerPlay = false;
  284. info.canHumanPlay = false;
  285. }
  286. else
  287. {
  288. readPlayerInfo(info, playerSrc);
  289. }
  290. }
  291. mapHeader->howManyTeams = howManyTeams;
  292. }
  293. void CMapLoaderJson::readPlayerInfo(PlayerInfo& info, const JsonNode& input)
  294. {
  295. //allowed factions
  296. // info.isFactionRandom =
  297. info.canComputerPlay = true;
  298. info.canHumanPlay = input["canPlay"].String() != "AIOnly";
  299. //placedHeroes
  300. //mainTown
  301. info.generateHeroAtMainTown = input["generateHeroAtMainTown"].Bool();
  302. //mainHero
  303. //mainHeroPortrait
  304. //mainCustomHeroName
  305. }
  306. void CMapLoaderJson::readTeams(const JsonNode& input)
  307. {
  308. const JsonNode & src = input["teams"];
  309. if(src.getType() != JsonNode::DATA_VECTOR)
  310. {
  311. // No alliances
  312. if(src.getType() != JsonNode::DATA_NULL)
  313. logGlobal->errorStream() << "Invalid teams field type";
  314. mapHeader->howManyTeams = 0;
  315. for(int i = 0; i < PlayerColor::PLAYER_LIMIT_I; i++)
  316. {
  317. if(mapHeader->players[i].canComputerPlay || mapHeader->players[i].canHumanPlay)
  318. {
  319. mapHeader->players[i].team = TeamID(mapHeader->howManyTeams++);
  320. }
  321. }
  322. }
  323. else
  324. {
  325. const JsonVector & srcVector = src.Vector();
  326. mapHeader->howManyTeams = srcVector.size();
  327. for(int team = 0; team < mapHeader->howManyTeams; team++)
  328. {
  329. for(const JsonNode & playerData : srcVector[team].Vector())
  330. {
  331. PlayerColor player = PlayerColor(vstd::find_pos(GameConstants::PLAYER_COLOR_NAMES, playerData.String()));
  332. if(player.isValidPlayer())
  333. {
  334. if(mapHeader->players[player.getNum()].canAnyonePlay())
  335. {
  336. mapHeader->players[player.getNum()].team = TeamID(team);
  337. }
  338. }
  339. }
  340. }
  341. for(PlayerInfo & player : mapHeader->players)
  342. {
  343. if(player.canAnyonePlay() && player.team == TeamID::NO_TEAM)
  344. player.team = TeamID(mapHeader->howManyTeams++);
  345. }
  346. }
  347. }
  348. void CMapLoaderJson::readTerrainTile(const std::string& src, TerrainTile& tile)
  349. {
  350. using namespace TerrainDetail;
  351. {//terrain type
  352. const std::string typeCode = src.substr(0,2);
  353. int rawType = vstd::find_pos(terrainCodes, typeCode);
  354. if(rawType < 0)
  355. throw new std::runtime_error("Invalid terrain type code in "+src);
  356. tile.terType = ETerrainType(rawType);
  357. }
  358. int startPos = 2; //0+typeCode fixed length
  359. {//terrain view
  360. int pos = startPos;
  361. while(isdigit(src.at(pos)))
  362. pos++;
  363. int len = pos - startPos;
  364. if(len<=0)
  365. throw new std::runtime_error("Invalid terrain view in "+src);
  366. const std::string rawCode = src.substr(startPos,len);
  367. tile.terView = atoi(rawCode.c_str());
  368. startPos+=len;
  369. }
  370. {//terrain flip
  371. int terrainFlip = vstd::find_pos(flipCodes, src.at(startPos++));
  372. if(terrainFlip < 0)
  373. throw new std::runtime_error("Invalid terrain flip in "+src);
  374. else
  375. tile.extTileFlags = terrainFlip;
  376. }
  377. if(startPos >= src.size())
  378. return;
  379. bool hasRoad = true;
  380. {//road type
  381. const std::string typeCode = src.substr(startPos,2);
  382. startPos+=2;
  383. int rawType = vstd::find_pos(roadCodes, typeCode);
  384. if(rawType < 0)
  385. {
  386. rawType = vstd::find_pos(riverCodes, typeCode);
  387. if(rawType < 0)
  388. throw new std::runtime_error("Invalid river type in "+src);
  389. else
  390. {
  391. tile.riverType = ERiverType::ERiverType(rawType);
  392. hasRoad = false;
  393. }
  394. }
  395. else
  396. tile.roadType = ERoadType::ERoadType(rawType);
  397. }
  398. if(hasRoad)
  399. {//road dir
  400. int pos = startPos;
  401. while(isdigit(src.at(pos)))
  402. pos++;
  403. int len = pos - startPos;
  404. if(len<=0)
  405. throw new std::runtime_error("Invalid road dir in "+src);
  406. const std::string rawCode = src.substr(startPos,len);
  407. tile.roadDir = atoi(rawCode.c_str());
  408. startPos+=len;
  409. }
  410. if(hasRoad)
  411. {//road flip
  412. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  413. if(flip < 0)
  414. throw new std::runtime_error("Invalid road flip in "+src);
  415. else
  416. tile.extTileFlags |= (flip<<4);
  417. }
  418. if(startPos >= src.size())
  419. return;
  420. if(hasRoad)
  421. {//river type
  422. const std::string typeCode = src.substr(startPos,2);
  423. startPos+=2;
  424. int rawType = vstd::find_pos(riverCodes, typeCode);
  425. if(rawType < 0)
  426. throw new std::runtime_error("Invalid river type in "+src);
  427. tile.riverType = ERiverType::ERiverType(rawType);
  428. }
  429. {//river dir
  430. int pos = startPos;
  431. while(isdigit(src.at(pos)))
  432. pos++;
  433. int len = pos - startPos;
  434. if(len<=0)
  435. throw new std::runtime_error("Invalid river dir in "+src);
  436. const std::string rawCode = src.substr(startPos,len);
  437. tile.riverDir = atoi(rawCode.c_str());
  438. startPos+=len;
  439. }
  440. {//river flip
  441. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  442. if(flip < 0)
  443. throw new std::runtime_error("Invalid road flip in "+src);
  444. else
  445. tile.extTileFlags |= (flip<<2);
  446. }
  447. }
  448. void CMapLoaderJson::readTerrainLevel(const JsonNode& src, const int index)
  449. {
  450. int3 pos(0,0,index);
  451. const JsonVector & rows = src.Vector();
  452. if(rows.size() != map->height)
  453. throw new std::runtime_error("Invalid terrain data");
  454. for(pos.y = 0; pos.y < map->height; pos.y++)
  455. {
  456. const JsonVector & tiles = rows[pos.y].Vector();
  457. if(tiles.size() != map->width)
  458. throw new std::runtime_error("Invalid terrain data");
  459. for(pos.x = 0; pos.x < map->width; pos.x++)
  460. readTerrainTile(tiles[pos.x].String(), map->getTile(pos));
  461. }
  462. }
  463. void CMapLoaderJson::readTerrain()
  464. {
  465. {
  466. const JsonNode surface = readJson("surface_terrain.json");
  467. readTerrainLevel(surface, 0);
  468. }
  469. if(map->twoLevel)
  470. {
  471. const JsonNode underground = readJson("underground_terrain.json");
  472. readTerrainLevel(underground, 1);
  473. }
  474. }
  475. CMapLoaderJson::MapObjectLoader::MapObjectLoader(CMapLoaderJson * _owner, const JsonMap::value_type& json):
  476. owner(_owner), instance(nullptr),handler(nullptr),id(-1), jsonKey(json.first), configuration(json.second), internalId(extractNumber(jsonKey, '_'))
  477. {
  478. }
  479. void CMapLoaderJson::MapObjectLoader::construct()
  480. {
  481. //TODO:consider move to ObjectTypeHandler
  482. //find type handler
  483. std::string typeName = configuration["type"].String(), subTypeName = configuration["subType"].String();
  484. if(typeName.empty())
  485. {
  486. logGlobal->errorStream() << "Object type missing";
  487. logGlobal->traceStream() << configuration;
  488. return;
  489. }
  490. if(subTypeName.empty())
  491. {
  492. logGlobal->errorStream() << "Object subType missing";
  493. logGlobal->traceStream() << configuration;
  494. return;
  495. }
  496. //special case for grail
  497. if(typeName == "grail")
  498. {
  499. auto & pos = owner->map->grailPos;
  500. pos.x = configuration["x"].Float();
  501. pos.y = configuration["y"].Float();
  502. pos.z = configuration["l"].Float();
  503. owner->map->grailRadius = configuration["options"]["grailRadius"].Float();
  504. }
  505. handler = VLC->objtypeh->getHandlerFor(typeName, subTypeName);
  506. instance = handler->create(ObjectTemplate());
  507. instance->id = ObjectInstanceID(owner->map->objects.size());
  508. owner->map->objects.push_back(instance);
  509. }
  510. void CMapLoaderJson::MapObjectLoader::configure()
  511. {
  512. if(nullptr == instance)
  513. return;
  514. instance->readJson(configuration);
  515. if(instance->ID == Obj::TOWN)
  516. {
  517. owner->map->towns.push_back(static_cast<CGTownInstance *>(instance));
  518. }
  519. if(instance->ID == Obj::HERO)
  520. {
  521. logGlobal->debugStream() << "Hero: " << VLC->heroh->heroes[instance->subID]->name << " at " << instance->pos;
  522. owner->map->heroesOnMap.push_back(static_cast<CGHeroInstance *>(instance));
  523. }
  524. }
  525. void CMapLoaderJson::readObjects()
  526. {
  527. LOG_TRACE(logGlobal);
  528. std::vector<std::unique_ptr<MapObjectLoader>> loaders;//todo: optimize MapObjectLoader memory layout
  529. const JsonNode data = readJson(OBJECTS_FILE_NAME);
  530. //get raw data
  531. for(const auto & p : data.Struct())
  532. loaders.push_back(vstd::make_unique<MapObjectLoader>(this, p));
  533. auto sortInfos = [](const std::unique_ptr<MapObjectLoader> & lhs, const std::unique_ptr<MapObjectLoader> & rhs) -> bool
  534. {
  535. return lhs->internalId < rhs->internalId;
  536. };
  537. boost::sort(loaders, sortInfos);//this makes instance ids consistent after save-load, needed for testing
  538. //todo: use internalId in CMap
  539. for(auto & ptr : loaders)
  540. ptr->construct();
  541. //configure objects after all objects are constructed so we may resolve internal IDs even to actual pointers OTF
  542. for(auto & ptr : loaders)
  543. ptr->configure();
  544. }
  545. ///CMapSaverJson
  546. CMapSaverJson::CMapSaverJson(CInputOutputStream * stream):
  547. buffer(stream),
  548. ioApi(new CProxyIOApi(buffer)),
  549. saver(ioApi, "_")
  550. {
  551. }
  552. CMapSaverJson::~CMapSaverJson()
  553. {
  554. }
  555. void CMapSaverJson::addToArchive(const JsonNode& data, const std::string& filename)
  556. {
  557. std::ostringstream out;
  558. out << data;
  559. out.flush();
  560. {
  561. auto s = out.str();
  562. std::unique_ptr<COutputStream> stream = saver.addFile(filename);
  563. if (stream->write((const ui8*)s.c_str(), s.size()) != s.size())
  564. throw new std::runtime_error("CMapSaverJson::saveHeader() zip compression failed.");
  565. }
  566. }
  567. void CMapSaverJson::saveMap(const std::unique_ptr<CMap>& map)
  568. {
  569. this->map = map.get();
  570. writeHeader();
  571. writeTerrain();
  572. writeObjects();
  573. }
  574. void CMapSaverJson::writeHeader()
  575. {
  576. JsonNode header;
  577. header["versionMajor"].Float() = VERSION_MAJOR;
  578. header["versionMinor"].Float() = VERSION_MINOR;
  579. //todo: multilevel map save support
  580. JsonNode & levels = header["mapLevels"];
  581. levels["surface"]["height"].Float() = map->height;
  582. levels["surface"]["width"].Float() = map->width;
  583. levels["surface"]["index"].Float() = 0;
  584. if(map->twoLevel)
  585. {
  586. levels["underground"]["height"].Float() = map->height;
  587. levels["underground"]["width"].Float() = map->width;
  588. levels["underground"]["index"].Float() = 1;
  589. }
  590. header["name"].String() = map->name;
  591. header["description"].String() = map->description;
  592. //todo: support arbitrary percentage
  593. static const std::map<ui8, std::string> difficultyMap =
  594. {
  595. {0, "EASY"},
  596. {1, "NORMAL"},
  597. {2, "HARD"},
  598. {3, "EXPERT"},
  599. {4, "IMPOSSIBLE"}
  600. };
  601. header["difficulty"].String() = difficultyMap.at(map->difficulty);
  602. header["levelLimit"].Float() = map->levelLimit;
  603. writeTriggeredEvents(header);
  604. writePlayerInfo(header);
  605. writeTeams(header);
  606. //todo: allowedHeroes;
  607. //todo: placeholdedHeroes;
  608. addToArchive(header, HEADER_FILE_NAME);
  609. }
  610. void CMapSaverJson::writePlayerInfo(JsonNode & output)
  611. {
  612. JsonNode & dest = output["players"];
  613. dest.setType(JsonNode::DATA_STRUCT);
  614. for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
  615. {
  616. const PlayerInfo & info = map->players[player];
  617. if(info.canAnyonePlay())
  618. writePlayerInfo(info, dest[GameConstants::PLAYER_COLOR_NAMES[player]]);
  619. }
  620. }
  621. void CMapSaverJson::writePlayerInfo(const PlayerInfo & info, JsonNode & output)
  622. {
  623. //allowed factions
  624. output["canPlay"].String() = info.canHumanPlay ? "PlayerOrAI" : "AIOnly";
  625. //mainTown
  626. output["generateHeroAtMainTown"].Bool() = info.generateHeroAtMainTown;
  627. //mainHero
  628. //towns
  629. //heroes
  630. }
  631. void CMapSaverJson::writeTeams(JsonNode& output)
  632. {
  633. JsonNode & dest = output["teams"];
  634. std::vector<std::set<PlayerColor>> teamsData;
  635. teamsData.resize(map->howManyTeams);
  636. //get raw data
  637. for(int idx = 0; idx < map->players.size(); idx++)
  638. {
  639. const PlayerInfo & player = map->players.at(idx);
  640. int team = player.team.getNum();
  641. if(vstd::isbetween(team, 0, map->howManyTeams-1) && player.canAnyonePlay())
  642. teamsData.at(team).insert(PlayerColor(idx));
  643. }
  644. //just an optimization but breaks test
  645. #if 0
  646. //remove single-member teams
  647. vstd::erase_if(teamsData, [](std::set<PlayerColor> & elem) -> bool
  648. {
  649. return elem.size() <= 1;
  650. });
  651. #endif
  652. //construct output
  653. dest.setType(JsonNode::DATA_VECTOR);
  654. for(const std::set<PlayerColor> & teamData : teamsData)
  655. {
  656. JsonNode team(JsonNode::DATA_VECTOR);
  657. for(const PlayerColor & player : teamData)
  658. {
  659. JsonNode member(JsonNode::DATA_STRING);
  660. member.String() = GameConstants::PLAYER_COLOR_NAMES[player.getNum()];
  661. team.Vector().push_back(std::move(member));
  662. }
  663. dest.Vector().push_back(std::move(team));
  664. }
  665. }
  666. const std::string CMapSaverJson::writeTerrainTile(const TerrainTile & tile)
  667. {
  668. using namespace TerrainDetail;
  669. std::ostringstream out;
  670. out.setf(std::ios::dec, std::ios::basefield);
  671. out.unsetf(std::ios::showbase);
  672. out << terrainCodes.at(int(tile.terType)) << (int)tile.terView << flipCodes[tile.extTileFlags % 4];
  673. if(tile.roadType != ERoadType::NO_ROAD)
  674. {
  675. out << roadCodes.at(int(tile.roadType)) << (int)tile.roadDir << flipCodes[(tile.extTileFlags >> 4) % 4];
  676. }
  677. if(tile.riverType != ERiverType::NO_RIVER)
  678. {
  679. out << riverCodes.at(int(tile.riverType)) << (int)tile.riverDir << flipCodes[(tile.extTileFlags >> 2) % 4];
  680. }
  681. return out.str();
  682. }
  683. JsonNode CMapSaverJson::writeTerrainLevel(const int index)
  684. {
  685. JsonNode data;
  686. int3 pos(0,0,index);
  687. data.Vector().resize(map->height);
  688. for(pos.y = 0; pos.y < map->height; pos.y++)
  689. {
  690. JsonNode & row = data.Vector()[pos.y];
  691. row.Vector().resize(map->width);
  692. for(pos.x = 0; pos.x < map->width; pos.x++)
  693. row.Vector()[pos.x].String() = writeTerrainTile(map->getTile(pos));
  694. }
  695. return std::move(data);
  696. }
  697. void CMapSaverJson::writeTerrain()
  698. {
  699. //todo: multilevel map save support
  700. JsonNode surface = writeTerrainLevel(0);
  701. addToArchive(surface, "surface_terrain.json");
  702. if(map->twoLevel)
  703. {
  704. JsonNode underground = writeTerrainLevel(1);
  705. addToArchive(underground, "underground_terrain.json");
  706. }
  707. }
  708. void CMapSaverJson::writeObjects()
  709. {
  710. JsonNode data(JsonNode::DATA_STRUCT);
  711. for(const CGObjectInstance * obj : map->objects)
  712. obj->writeJson(data[obj->getStringId()]);
  713. if(map->grailPos.valid())
  714. {
  715. JsonNode grail(JsonNode::DATA_STRUCT);
  716. grail["type"].String() = "grail";
  717. grail["x"].Float() = map->grailPos.x;
  718. grail["y"].Float() = map->grailPos.y;
  719. grail["l"].Float() = map->grailPos.z;
  720. grail["options"]["radius"].Float() = map->grailRadius;
  721. std::string grailId = boost::str(boost::format("grail_%d") % map->objects.size());
  722. data[grailId] = grail;
  723. }
  724. addToArchive(data, OBJECTS_FILE_NAME);
  725. }