MapFormatJson.cpp 24 KB

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