MapFormatJson.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892
  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. //TODO: read such data like map name & size
  240. //mapHeader->version = ??? //todo: new version field
  241. //todo: multilevel map load support
  242. const JsonNode levels = header["mapLevels"];
  243. mapHeader->height = levels["surface"]["height"].Float();
  244. mapHeader->width = levels["surface"]["width"].Float();
  245. mapHeader->twoLevel = !levels["underground"].isNull();
  246. mapHeader->name = header["name"].String();
  247. mapHeader->description = header["description"].String();
  248. //todo: support arbitrary percentage
  249. static const std::map<std::string, ui8> difficultyMap =
  250. {
  251. {"", 1},
  252. {"EASY", 0},
  253. {"NORMAL", 1},
  254. {"HARD", 2},
  255. {"EXPERT", 3},
  256. {"IMPOSSIBLE", 4}
  257. };
  258. mapHeader->difficulty = difficultyMap.at(header["difficulty"].String());
  259. mapHeader->levelLimit = header["levelLimit"].Float();
  260. // std::vector<bool> allowedHeroes;
  261. // std::vector<ui16> placeholdedHeroes;
  262. readTriggeredEvents(header);
  263. readPlayerInfo(header);
  264. readTeams(header);
  265. //TODO: readHeader
  266. }
  267. void CMapLoaderJson::readPlayerInfo(const JsonNode & input)
  268. {
  269. const JsonNode & src = input["players"];
  270. int howManyTeams = 0;
  271. for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
  272. {
  273. PlayerInfo & info = map->players.at(player);
  274. const JsonNode & playerSrc = src[GameConstants::PLAYER_COLOR_NAMES[player]];
  275. if(playerSrc.isNull())
  276. {
  277. info.canComputerPlay = false;
  278. info.canHumanPlay = false;
  279. }
  280. else
  281. {
  282. readPlayerInfo(info, playerSrc);
  283. }
  284. }
  285. mapHeader->howManyTeams = howManyTeams;
  286. }
  287. void CMapLoaderJson::readPlayerInfo(PlayerInfo& info, const JsonNode& input)
  288. {
  289. //allowed factions
  290. // info.isFactionRandom =
  291. info.canComputerPlay = true;
  292. info.canHumanPlay = input["canPlay"].String() != "AIOnly";
  293. //placedHeroes
  294. //mainTown
  295. info.generateHeroAtMainTown = input["generateHeroAtMainTown"].Bool();
  296. //mainHero
  297. //mainHeroPortrait
  298. //mainCustomHeroName
  299. }
  300. void CMapLoaderJson::readTeams(const JsonNode& input)
  301. {
  302. const JsonNode & src = input["teams"];
  303. if(src.getType() != JsonNode::DATA_VECTOR)
  304. {
  305. // No alliances
  306. if(src.getType() != JsonNode::DATA_NULL)
  307. logGlobal->errorStream() << "Invalid teams field type";
  308. mapHeader->howManyTeams = 0;
  309. for(int i = 0; i < PlayerColor::PLAYER_LIMIT_I; i++)
  310. {
  311. if(mapHeader->players[i].canComputerPlay || mapHeader->players[i].canHumanPlay)
  312. {
  313. mapHeader->players[i].team = TeamID(mapHeader->howManyTeams++);
  314. }
  315. }
  316. }
  317. else
  318. {
  319. const JsonVector & srcVector = src.Vector();
  320. mapHeader->howManyTeams = srcVector.size();
  321. for(int team = 0; team < mapHeader->howManyTeams; team++)
  322. {
  323. for(const JsonNode & playerData : srcVector[team].Vector())
  324. {
  325. PlayerColor player = PlayerColor(vstd::find_pos(GameConstants::PLAYER_COLOR_NAMES, playerData.String()));
  326. if(player.isValidPlayer())
  327. {
  328. if(map->players[player.getNum()].canAnyonePlay())
  329. {
  330. map->players[player.getNum()].team = TeamID(team);
  331. }
  332. }
  333. }
  334. }
  335. for(PlayerInfo & player : map->players)
  336. {
  337. if(player.canAnyonePlay() && player.team == TeamID::NO_TEAM)
  338. player.team = TeamID(mapHeader->howManyTeams++);
  339. }
  340. }
  341. }
  342. void CMapLoaderJson::readTerrainTile(const std::string& src, TerrainTile& tile)
  343. {
  344. using namespace TerrainDetail;
  345. {//terrain type
  346. const std::string typeCode = src.substr(0,2);
  347. int rawType = vstd::find_pos(terrainCodes, typeCode);
  348. if(rawType < 0)
  349. throw new std::runtime_error("Invalid terrain type code in "+src);
  350. tile.terType = ETerrainType(rawType);
  351. }
  352. int startPos = 2; //0+typeCode fixed length
  353. {//terrain view
  354. int pos = startPos;
  355. while(isdigit(src.at(pos)))
  356. pos++;
  357. int len = pos - startPos;
  358. if(len<=0)
  359. throw new std::runtime_error("Invalid terrain view in "+src);
  360. const std::string rawCode = src.substr(startPos,len);
  361. tile.terView = atoi(rawCode.c_str());
  362. startPos+=len;
  363. }
  364. {//terrain flip
  365. int terrainFlip = vstd::find_pos(flipCodes, src.at(startPos++));
  366. if(terrainFlip < 0)
  367. throw new std::runtime_error("Invalid terrain flip in "+src);
  368. else
  369. tile.extTileFlags = terrainFlip;
  370. }
  371. if(startPos >= src.size())
  372. return;
  373. bool hasRoad = true;
  374. {//road type
  375. const std::string typeCode = src.substr(startPos,2);
  376. startPos+=2;
  377. int rawType = vstd::find_pos(roadCodes, typeCode);
  378. if(rawType < 0)
  379. {
  380. rawType = vstd::find_pos(riverCodes, typeCode);
  381. if(rawType < 0)
  382. throw new std::runtime_error("Invalid river type in "+src);
  383. else
  384. {
  385. tile.riverType = ERiverType::ERiverType(rawType);
  386. hasRoad = false;
  387. }
  388. }
  389. else
  390. tile.roadType = ERoadType::ERoadType(rawType);
  391. }
  392. if(hasRoad)
  393. {//road dir
  394. int pos = startPos;
  395. while(isdigit(src.at(pos)))
  396. pos++;
  397. int len = pos - startPos;
  398. if(len<=0)
  399. throw new std::runtime_error("Invalid road dir in "+src);
  400. const std::string rawCode = src.substr(startPos,len);
  401. tile.roadDir = atoi(rawCode.c_str());
  402. startPos+=len;
  403. }
  404. if(hasRoad)
  405. {//road flip
  406. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  407. if(flip < 0)
  408. throw new std::runtime_error("Invalid road flip in "+src);
  409. else
  410. tile.extTileFlags |= (flip<<4);
  411. }
  412. if(startPos >= src.size())
  413. return;
  414. if(hasRoad)
  415. {//river type
  416. const std::string typeCode = src.substr(startPos,2);
  417. startPos+=2;
  418. int rawType = vstd::find_pos(riverCodes, typeCode);
  419. if(rawType < 0)
  420. throw new std::runtime_error("Invalid river type in "+src);
  421. tile.riverType = ERiverType::ERiverType(rawType);
  422. }
  423. {//river dir
  424. int pos = startPos;
  425. while(isdigit(src.at(pos)))
  426. pos++;
  427. int len = pos - startPos;
  428. if(len<=0)
  429. throw new std::runtime_error("Invalid river dir in "+src);
  430. const std::string rawCode = src.substr(startPos,len);
  431. tile.riverDir = atoi(rawCode.c_str());
  432. startPos+=len;
  433. }
  434. {//river flip
  435. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  436. if(flip < 0)
  437. throw new std::runtime_error("Invalid road flip in "+src);
  438. else
  439. tile.extTileFlags |= (flip<<2);
  440. }
  441. }
  442. void CMapLoaderJson::readTerrainLevel(const JsonNode& src, const int index)
  443. {
  444. int3 pos(0,0,index);
  445. const JsonVector & rows = src.Vector();
  446. if(rows.size() != map->height)
  447. throw new std::runtime_error("Invalid terrain data");
  448. for(pos.y = 0; pos.y < map->height; pos.y++)
  449. {
  450. const JsonVector & tiles = rows[pos.y].Vector();
  451. if(tiles.size() != map->width)
  452. throw new std::runtime_error("Invalid terrain data");
  453. for(pos.x = 0; pos.x < map->width; pos.x++)
  454. readTerrainTile(tiles[pos.x].String(), map->getTile(pos));
  455. }
  456. }
  457. void CMapLoaderJson::readTerrain()
  458. {
  459. {
  460. const JsonNode surface = readJson("surface_terrain.json");
  461. readTerrainLevel(surface, 0);
  462. }
  463. if(map->twoLevel)
  464. {
  465. const JsonNode underground = readJson("underground_terrain.json");
  466. readTerrainLevel(underground, 1);
  467. }
  468. }
  469. CMapLoaderJson::MapObjectLoader::MapObjectLoader(CMapLoaderJson * _owner, const JsonMap::value_type& json):
  470. owner(_owner), instance(nullptr),handler(nullptr),id(-1), jsonKey(json.first), configuration(json.second), internalId(extractNumber(jsonKey, '_'))
  471. {
  472. }
  473. void CMapLoaderJson::MapObjectLoader::construct()
  474. {
  475. //TODO:consider move to ObjectTypeHandler
  476. //find type handler
  477. std::string typeName = configuration["type"].String(), subTypeName = configuration["subType"].String();
  478. if(typeName.empty())
  479. {
  480. logGlobal->errorStream() << "Object type missing";
  481. return;
  482. }
  483. if(subTypeName.empty())
  484. {
  485. logGlobal->errorStream() << "Object subType missing";
  486. return;
  487. }
  488. //special case for grail
  489. if(typeName == "grail")
  490. {
  491. auto & pos = owner->map->grailPos;
  492. pos.x = configuration["x"].Float();
  493. pos.y = configuration["y"].Float();
  494. pos.z = configuration["l"].Float();
  495. owner->map->grailRadious = configuration["options"]["grailRadius"].Float();
  496. }
  497. handler = VLC->objtypeh->getHandlerFor(typeName, subTypeName);
  498. instance = handler->create(ObjectTemplate());
  499. instance->id = ObjectInstanceID(owner->map->objects.size());
  500. owner->map->objects.push_back(instance);
  501. }
  502. void CMapLoaderJson::MapObjectLoader::configure()
  503. {
  504. if(nullptr == instance)
  505. return;
  506. instance->readJson(configuration);
  507. if(instance->ID == Obj::TOWN)
  508. {
  509. owner->map->towns.push_back(static_cast<CGTownInstance *>(instance));
  510. }
  511. if(instance->ID == Obj::HERO)
  512. {
  513. logGlobal->debugStream() << "Hero: " << VLC->heroh->heroes[instance->subID]->name << " at " << instance->pos;
  514. owner->map->heroesOnMap.push_back(static_cast<CGHeroInstance *>(instance));
  515. }
  516. }
  517. void CMapLoaderJson::readObjects()
  518. {
  519. LOG_TRACE(logGlobal);
  520. std::vector<std::unique_ptr<MapObjectLoader>> loaders;//todo: optimize MapObjectLoader memory layout
  521. const JsonNode data = readJson(OBJECTS_FILE_NAME);
  522. //get raw data
  523. for(const auto & p : data.Struct())
  524. loaders.push_back(vstd::make_unique<MapObjectLoader>(this, p));
  525. auto sortInfos = [](const std::unique_ptr<MapObjectLoader> & lhs, const std::unique_ptr<MapObjectLoader> & rhs) -> bool
  526. {
  527. return lhs->internalId < rhs->internalId;
  528. };
  529. boost::sort(loaders, sortInfos);//this makes instance ids consistent after save-load, needed for testing
  530. //todo: use internalId in CMap
  531. for(auto & ptr : loaders)
  532. ptr->construct();
  533. //configure objects after all objects are constructed so we may resolve internal IDs even to actual pointers OTF
  534. for(auto & ptr : loaders)
  535. ptr->configure();
  536. }
  537. ///CMapSaverJson
  538. CMapSaverJson::CMapSaverJson(CInputOutputStream * stream):
  539. CMapFormatZip(stream),
  540. saver(ioApi, "_")
  541. {
  542. }
  543. CMapSaverJson::~CMapSaverJson()
  544. {
  545. }
  546. void CMapSaverJson::addToArchive(const JsonNode& data, const std::string& filename)
  547. {
  548. std::ostringstream out;
  549. out << data;
  550. out.flush();
  551. {
  552. auto s = out.str();
  553. std::unique_ptr<COutputStream> stream = saver.addFile(filename);
  554. if (stream->write((const ui8*)s.c_str(), s.size()) != s.size())
  555. throw new std::runtime_error("CMapSaverJson::saveHeader() zip compression failed.");
  556. }
  557. }
  558. void CMapSaverJson::saveMap(const std::unique_ptr<CMap>& map)
  559. {
  560. this->map = map.get();
  561. writeHeader();
  562. writeTerrain();
  563. writeObjects();
  564. }
  565. void CMapSaverJson::writeHeader()
  566. {
  567. JsonNode header;
  568. header["versionMajor"].Float() = VERSION_MAJOR;
  569. header["versionMinor"].Float() = VERSION_MINOR;
  570. //todo: multilevel map save support
  571. JsonNode & levels = header["mapLevels"];
  572. levels["surface"]["height"].Float() = map->height;
  573. levels["surface"]["width"].Float() = map->width;
  574. levels["surface"]["index"].Float() = 0;
  575. if(map->twoLevel)
  576. {
  577. levels["underground"]["height"].Float() = map->height;
  578. levels["underground"]["width"].Float() = map->width;
  579. levels["underground"]["index"].Float() = 1;
  580. }
  581. header["name"].String() = map->name;
  582. header["description"].String() = map->description;
  583. //todo: support arbitrary percentage
  584. static const std::map<ui8, std::string> difficultyMap =
  585. {
  586. {0, "EASY"},
  587. {1, "NORMAL"},
  588. {2, "HARD"},
  589. {3, "EXPERT"},
  590. {4, "IMPOSSIBLE"}
  591. };
  592. header["difficulty"].String() = difficultyMap.at(map->difficulty);
  593. header["levelLimit"].Float() = map->levelLimit;
  594. writeTriggeredEvents(header);
  595. writePlayerInfo(header);
  596. writeTeams(header);
  597. //todo: allowedHeroes;
  598. //todo: placeholdedHeroes;
  599. addToArchive(header, HEADER_FILE_NAME);
  600. }
  601. void CMapSaverJson::writePlayerInfo(JsonNode & output)
  602. {
  603. JsonNode & dest = output["players"];
  604. dest.setType(JsonNode::DATA_STRUCT);
  605. for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
  606. {
  607. const PlayerInfo & info = map->players[player];
  608. if(info.canAnyonePlay())
  609. writePlayerInfo(info, dest[GameConstants::PLAYER_COLOR_NAMES[player]]);
  610. }
  611. }
  612. void CMapSaverJson::writePlayerInfo(const PlayerInfo & info, JsonNode & output)
  613. {
  614. //allowed factions
  615. output["canPlay"].String() = info.canHumanPlay ? "PlayerOrAI" : "AIOnly";
  616. //mainTown
  617. output["generateHeroAtMainTown"].Bool() = info.generateHeroAtMainTown;
  618. //mainHero
  619. //towns
  620. //heroes
  621. }
  622. void CMapSaverJson::writeTeams(JsonNode& output)
  623. {
  624. JsonNode & dest = output["teams"];
  625. std::vector<std::set<PlayerColor>> teamsData;
  626. teamsData.resize(map->howManyTeams);
  627. //get raw data
  628. for(int idx = 0; idx < map->players.size(); idx++)
  629. {
  630. const PlayerInfo & player = map->players.at(idx);
  631. int team = player.team.getNum();
  632. if(vstd::isbetween(team, 0, map->howManyTeams-1) && player.canAnyonePlay())
  633. teamsData.at(team).insert(PlayerColor(idx));
  634. }
  635. //just an optimization but breaks test
  636. #if 0
  637. //remove single-member teams
  638. vstd::erase_if(teamsData, [](std::set<PlayerColor> & elem) -> bool
  639. {
  640. return elem.size() <= 1;
  641. });
  642. #endif
  643. //construct output
  644. dest.setType(JsonNode::DATA_VECTOR);
  645. for(const std::set<PlayerColor> & teamData : teamsData)
  646. {
  647. JsonNode team(JsonNode::DATA_VECTOR);
  648. for(const PlayerColor & player : teamData)
  649. {
  650. JsonNode member(JsonNode::DATA_STRING);
  651. member.String() = GameConstants::PLAYER_COLOR_NAMES[player.getNum()];
  652. team.Vector().push_back(std::move(member));
  653. }
  654. dest.Vector().push_back(std::move(team));
  655. }
  656. }
  657. const std::string CMapSaverJson::writeTerrainTile(const TerrainTile & tile)
  658. {
  659. using namespace TerrainDetail;
  660. std::ostringstream out;
  661. out.setf(std::ios::dec, std::ios::basefield);
  662. out.unsetf(std::ios::showbase);
  663. out << terrainCodes.at(int(tile.terType)) << (int)tile.terView << flipCodes[tile.extTileFlags % 4];
  664. if(tile.roadType != ERoadType::NO_ROAD)
  665. {
  666. out << roadCodes.at(int(tile.roadType)) << (int)tile.roadDir << flipCodes[(tile.extTileFlags >> 4) % 4];
  667. }
  668. if(tile.riverType != ERiverType::NO_RIVER)
  669. {
  670. out << riverCodes.at(int(tile.riverType)) << (int)tile.riverDir << flipCodes[(tile.extTileFlags >> 2) % 4];
  671. }
  672. return out.str();
  673. }
  674. JsonNode CMapSaverJson::writeTerrainLevel(const int index)
  675. {
  676. JsonNode data;
  677. int3 pos(0,0,index);
  678. data.Vector().resize(map->height);
  679. for(pos.y = 0; pos.y < map->height; pos.y++)
  680. {
  681. JsonNode & row = data.Vector()[pos.y];
  682. row.Vector().resize(map->width);
  683. for(pos.x = 0; pos.x < map->width; pos.x++)
  684. row.Vector()[pos.x].String() = writeTerrainTile(map->getTile(pos));
  685. }
  686. return std::move(data);
  687. }
  688. void CMapSaverJson::writeTerrain()
  689. {
  690. //todo: multilevel map save support
  691. JsonNode surface = writeTerrainLevel(0);
  692. addToArchive(surface, "surface_terrain.json");
  693. if(map->twoLevel)
  694. {
  695. JsonNode underground = writeTerrainLevel(1);
  696. addToArchive(underground, "underground_terrain.json");
  697. }
  698. }
  699. void CMapSaverJson::writeObjects()
  700. {
  701. JsonNode data(JsonNode::DATA_STRUCT);
  702. for(const CGObjectInstance * obj : map->objects)
  703. obj->writeJson(data[obj->getStringId()]);
  704. if(map->grailPos.valid())
  705. {
  706. JsonNode grail(JsonNode::DATA_STRUCT);
  707. grail["type"].String() = "grail";
  708. grail["x"].Float() = map->grailPos.x;
  709. grail["y"].Float() = map->grailPos.y;
  710. grail["l"].Float() = map->grailPos.z;
  711. grail["options"]["radius"].Float() = map->grailRadious;
  712. std::string grailId = boost::str(boost::format("grail_%d") % map->objects.size());
  713. data[grailId] = grail;
  714. }
  715. addToArchive(data, OBJECTS_FILE_NAME);
  716. }