MapFormatJson.cpp 25 KB

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