MapFormatJson.cpp 23 KB

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