MapFormatJson.cpp 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327
  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 "../CCreatureHandler.h"
  17. #include "../CHeroHandler.h"
  18. #include "../CTownHandler.h"
  19. #include "../VCMI_Lib.h"
  20. #include "../mapObjects/ObjectTemplate.h"
  21. #include "../mapObjects/CObjectHandler.h"
  22. #include "../mapObjects/CObjectClassesHandler.h"
  23. #include "../mapObjects/CGHeroInstance.h"
  24. #include "../mapObjects/CGTownInstance.h"
  25. #include "../spells/CSpellHandler.h"
  26. #include "../StringConstants.h"
  27. #include "../serializer/JsonDeserializer.h"
  28. #include "../serializer/JsonSerializer.h"
  29. class MapObjectResolver: public IInstanceResolver
  30. {
  31. public:
  32. MapObjectResolver(const CMapFormatJson * owner_);
  33. si32 decode (const std::string & identifier) const override;
  34. std::string encode(si32 identifier) const override;
  35. private:
  36. const CMapFormatJson * owner;
  37. };
  38. MapObjectResolver::MapObjectResolver(const CMapFormatJson * owner_):
  39. owner(owner_)
  40. {
  41. }
  42. si32 MapObjectResolver::decode(const std::string & identifier) const
  43. {
  44. //always decode as ObjectInstanceID
  45. auto it = owner->map->instanceNames.find(identifier);
  46. if(it != owner->map->instanceNames.end())
  47. {
  48. return (*it).second->id.getNum();
  49. }
  50. else
  51. {
  52. logGlobal->errorStream() << "Object not found: " << identifier;
  53. return -1;
  54. }
  55. }
  56. std::string MapObjectResolver::encode(si32 identifier) const
  57. {
  58. ObjectInstanceID id;
  59. //use h3m questIdentifiers if they are present
  60. if(owner->map->questIdentifierToId.empty())
  61. {
  62. id = ObjectInstanceID(identifier);
  63. }
  64. else
  65. {
  66. id = owner->map->questIdentifierToId[identifier];
  67. }
  68. si32 oid = id.getNum();
  69. if(oid < 0 || oid >= owner->map->objects.size())
  70. {
  71. logGlobal->errorStream() << "Cannot get object with id " << oid;
  72. return "";
  73. }
  74. return owner->map->objects[oid]->instanceName;
  75. }
  76. namespace HeaderDetail
  77. {
  78. static const ui8 difficultyDefault = 1;//normal
  79. static const std::vector<std::string> difficultyMap =
  80. {
  81. "EASY",
  82. "NORMAL",
  83. "HARD",
  84. "EXPERT",
  85. "IMPOSSIBLE"
  86. };
  87. static const std::vector<std::string> canPlayMap =
  88. {
  89. "AIOnly",
  90. "PlayerOrAI"
  91. };
  92. }
  93. namespace TriggeredEventsDetail
  94. {
  95. static const std::array<std::string, 15> conditionNames =
  96. {
  97. "haveArtifact", "haveCreatures", "haveResources", "haveBuilding",
  98. "control", "destroy", "transport", "daysPassed",
  99. "isHuman", "daysWithoutTown", "standardWin", "constValue",
  100. "have_0", "haveBuilding_0", "destroy_0"
  101. };
  102. static const std::array<std::string, 2> typeNames = { "victory", "defeat" };
  103. static EMetaclass decodeMetaclass(const std::string & source)
  104. {
  105. if(source == "")
  106. return EMetaclass::INVALID;
  107. auto rawId = vstd::find_pos(NMetaclass::names, source);
  108. if(rawId >= 0)
  109. return (EMetaclass)rawId;
  110. else
  111. return EMetaclass::INVALID;
  112. }
  113. static std::string encodeIdentifier(EMetaclass metaType, si32 type)
  114. {
  115. std::string metaclassName = NMetaclass::names[(int)metaType];
  116. std::string identifier = "";
  117. switch(metaType)
  118. {
  119. case EMetaclass::ARTIFACT:
  120. {
  121. identifier = CArtHandler::encodeArtifact(type);
  122. }
  123. break;
  124. case EMetaclass::CREATURE:
  125. {
  126. identifier = CCreatureHandler::encodeCreature(type);
  127. }
  128. break;
  129. case EMetaclass::OBJECT:
  130. {
  131. //TODO
  132. std::set<si32> subtypes = VLC->objtypeh->knownSubObjects(type);
  133. if(!subtypes.empty())
  134. {
  135. si32 subtype = *subtypes.begin();
  136. auto handler = VLC->objtypeh->getHandlerFor(type, subtype);
  137. identifier = handler->typeName;
  138. }
  139. }
  140. break;
  141. case EMetaclass::RESOURCE:
  142. {
  143. identifier = GameConstants::RESOURCE_NAMES[type];
  144. }
  145. break;
  146. default:
  147. {
  148. logGlobal->error("Unsupported metaclass %s for event condition", metaclassName);
  149. return "";
  150. }
  151. break;
  152. }
  153. return VLC->modh->makeFullIdentifier("", metaclassName, identifier);
  154. }
  155. static EventCondition JsonToCondition(const JsonNode & node)
  156. {
  157. EventCondition event;
  158. const auto & conditionName = node.Vector()[0].String();
  159. auto pos = vstd::find_pos(conditionNames, conditionName);
  160. event.condition = EventCondition::EWinLoseType(pos);
  161. if (node.Vector().size() > 1)
  162. {
  163. const JsonNode & data = node.Vector()[1];
  164. switch (event.condition)
  165. {
  166. case EventCondition::HAVE_0:
  167. case EventCondition::DESTROY_0:
  168. {
  169. //todo: support subtypes
  170. std::string fullIdentifier = data["type"].String(), metaTypeName = "", scope = "" , identifier = "";
  171. CModHandler::parseIdentifier(fullIdentifier, scope, metaTypeName, identifier);
  172. event.metaType = decodeMetaclass(metaTypeName);
  173. auto type = VLC->modh->identifiers.getIdentifier("core", fullIdentifier, false);
  174. if(type)
  175. event.objectType = type.get();
  176. event.objectInstanceName = data["object"].String();
  177. if(data["value"].isNumber())
  178. event.value = data["value"].Integer();
  179. }
  180. break;
  181. case EventCondition::HAVE_BUILDING_0:
  182. {
  183. //todo: support of new condition format HAVE_BUILDING_0
  184. }
  185. break;
  186. default:
  187. {
  188. //old format
  189. if (data["type"].getType() == JsonNode::DATA_STRING)
  190. {
  191. auto identifier = VLC->modh->identifiers.getIdentifier(data["type"]);
  192. if(identifier)
  193. event.objectType = identifier.get();
  194. else
  195. throw std::runtime_error("Identifier resolution failed in event condition");
  196. }
  197. if (data["type"].isNumber())
  198. event.objectType = data["type"].Float();
  199. if (!data["value"].isNull())
  200. event.value = data["value"].Float();
  201. }
  202. break;
  203. }
  204. if (!data["position"].isNull())
  205. {
  206. auto & position = data["position"].Vector();
  207. event.position.x = position.at(0).Float();
  208. event.position.y = position.at(1).Float();
  209. event.position.z = position.at(2).Float();
  210. }
  211. }
  212. return event;
  213. }
  214. static JsonNode ConditionToJson(const EventCondition & event)
  215. {
  216. JsonNode json;
  217. JsonVector & asVector = json.Vector();
  218. JsonNode condition;
  219. condition.String() = conditionNames.at(event.condition);
  220. asVector.push_back(condition);
  221. JsonNode data;
  222. switch (event.condition)
  223. {
  224. case EventCondition::HAVE_0:
  225. case EventCondition::DESTROY_0:
  226. {
  227. //todo: support subtypes
  228. if(event.metaType != EMetaclass::INVALID)
  229. data["type"].String() = encodeIdentifier(event.metaType, event.objectType);
  230. if(event.value > 0)
  231. data["value"].Integer() = event.value;
  232. if(event.objectInstanceName != "")
  233. data["object"].String() = event.objectInstanceName;
  234. }
  235. break;
  236. case EventCondition::HAVE_BUILDING_0:
  237. {
  238. //todo: support of new condition format HAVE_BUILDING_0
  239. }
  240. break;
  241. default:
  242. {
  243. //old format
  244. if(event.objectType != -1)
  245. data["type"].Integer() = event.objectType;
  246. if(event.value != -1)
  247. data["value"].Integer() = event.value;
  248. }
  249. break;
  250. }
  251. if(event.position != int3(-1, -1, -1))
  252. {
  253. auto & position = data["position"].Vector();
  254. position.resize(3);
  255. position[0].Float() = event.position.x;
  256. position[1].Float() = event.position.y;
  257. position[2].Float() = event.position.z;
  258. }
  259. if(!data.isNull())
  260. asVector.push_back(data);
  261. return std::move(json);
  262. }
  263. }//namespace TriggeredEventsDetail
  264. namespace TerrainDetail
  265. {
  266. static const std::array<std::string, 10> terrainCodes =
  267. {
  268. "dt", "sa", "gr", "sn", "sw", "rg", "sb", "lv", "wt", "rc"
  269. };
  270. static const std::array<std::string, 4> roadCodes =
  271. {
  272. "", "pd", "pg", "pc"
  273. };
  274. static const std::array<std::string, 5> riverCodes =
  275. {
  276. "", "rw", "ri", "rm", "rl"
  277. };
  278. static const std::array<char, 4> flipCodes =
  279. {
  280. '_', '-', '|', '+'
  281. };
  282. }
  283. ///CMapFormatJson
  284. const int CMapFormatJson::VERSION_MAJOR = 1;
  285. const int CMapFormatJson::VERSION_MINOR = 0;
  286. const std::string CMapFormatJson::HEADER_FILE_NAME = "header.json";
  287. const std::string CMapFormatJson::OBJECTS_FILE_NAME = "objects.json";
  288. CMapFormatJson::CMapFormatJson():
  289. mapObjectResolver(make_unique<MapObjectResolver>(this))
  290. {
  291. }
  292. void CMapFormatJson::serializeAllowedFactions(JsonSerializeFormat & handler, std::set<TFaction> & value)
  293. {
  294. //TODO: unify allowed factions with others - make them std::vector<bool>
  295. std::vector<bool> temp;
  296. temp.resize(VLC->townh->factions.size(), false);
  297. auto standard = VLC->townh->getDefaultAllowed();
  298. if(handler.saving)
  299. {
  300. for(auto faction : VLC->townh->factions)
  301. if(faction->town && vstd::contains(value, faction->index))
  302. temp[std::size_t(faction->index)] = true;
  303. }
  304. handler.serializeLIC("allowedFactions", &CTownHandler::decodeFaction, &CTownHandler::encodeFaction, standard, temp);
  305. if(!handler.saving)
  306. {
  307. value.clear();
  308. for (std::size_t i=0; i<temp.size(); i++)
  309. if(temp[i])
  310. value.insert(i);
  311. }
  312. }
  313. void CMapFormatJson::serializeHeader(JsonSerializeFormat & handler)
  314. {
  315. handler.serializeString("name", mapHeader->name);
  316. handler.serializeString("description", mapHeader->description);
  317. handler.serializeInt("heroLevelLimit", mapHeader->levelLimit, 0);
  318. //todo: support arbitrary percentage
  319. handler.serializeEnum("difficulty", mapHeader->difficulty, HeaderDetail::difficultyMap);
  320. serializePlayerInfo(handler);
  321. handler.serializeLIC("allowedHeroes", &CHeroHandler::decodeHero, &CHeroHandler::encodeHero, VLC->heroh->getDefaultAllowed(), mapHeader->allowedHeroes);
  322. handler.serializeString("victoryString", mapHeader->victoryMessage);
  323. handler.serializeInt("victoryIconIndex", mapHeader->victoryIconIndex);
  324. handler.serializeString("defeatString", mapHeader->defeatMessage);
  325. handler.serializeInt("defeatIconIndex", mapHeader->defeatIconIndex);
  326. }
  327. void CMapFormatJson::serializePlayerInfo(JsonSerializeFormat & handler)
  328. {
  329. auto playersData = handler.enterStruct("players");
  330. for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
  331. {
  332. PlayerInfo & info = mapHeader->players[player];
  333. if(handler.saving)
  334. {
  335. if(!info.canAnyonePlay())
  336. continue;
  337. }
  338. auto playerData = playersData.enterStruct(GameConstants::PLAYER_COLOR_NAMES[player]);
  339. if(!handler.saving)
  340. {
  341. if(playerData.get().isNull())
  342. {
  343. info.canComputerPlay = false;
  344. info.canHumanPlay = false;
  345. continue;
  346. }
  347. info.canComputerPlay = true;
  348. }
  349. serializeAllowedFactions(handler, info.allowedFactions);
  350. handler.serializeEnum("canPlay", info.canHumanPlay, HeaderDetail::canPlayMap);
  351. //saving whole structure only if position is valid
  352. if(!handler.saving || info.posOfMainTown.valid())
  353. {
  354. auto mainTown = handler.enterStruct("mainTown");
  355. handler.serializeBool("generateHero", info.generateHeroAtMainTown);
  356. handler.serializeInt("x", info.posOfMainTown.x, -1);
  357. handler.serializeInt("y", info.posOfMainTown.y, -1);
  358. handler.serializeInt("l", info.posOfMainTown.z, -1);
  359. }
  360. if(!handler.saving)
  361. {
  362. info.hasMainTown = info.posOfMainTown.valid();
  363. }
  364. handler.serializeString("mainHero", info.mainHeroInstance);//must be before "heroes"
  365. //heroes
  366. if(handler.saving)
  367. {
  368. //ignoring heroesNames and saving from actual map objects
  369. //TODO: optimize
  370. for(auto & obj : map->objects)
  371. {
  372. if((obj->ID == Obj::HERO || obj->ID == Obj::RANDOM_HERO) && obj->tempOwner == PlayerColor(player))
  373. {
  374. CGHeroInstance * hero = dynamic_cast<CGHeroInstance *>(obj.get());
  375. auto heroes = playerData.enterStruct("heroes");
  376. if(hero)
  377. {
  378. auto heroData = heroes.enterStruct(hero->instanceName);
  379. heroData->serializeString("name", hero->name);
  380. if(hero->ID == Obj::HERO)
  381. {
  382. if(hero->type)
  383. {
  384. handler.serializeString("type", hero->type->identifier);
  385. }
  386. else
  387. {
  388. auto temp = VLC->heroh->heroes[hero->subID]->identifier;
  389. handler.serializeString("type", temp);
  390. }
  391. }
  392. }
  393. }
  394. }
  395. }
  396. else
  397. {
  398. info.heroesNames.clear();
  399. auto heroes = playerData.enterStruct("heroes");
  400. for(const auto & hero : handler.getCurrent().Struct())
  401. {
  402. const JsonNode & data = hero.second;
  403. const std::string instanceName = hero.first;
  404. SHeroName hname;
  405. hname.heroId = -1;
  406. std::string rawId = data["type"].String();
  407. if(rawId != "")
  408. {
  409. hname.heroId = VLC->heroh->decodeHero(rawId);
  410. }
  411. hname.heroName = data["name"].String();
  412. if(instanceName == info.mainHeroInstance)
  413. {
  414. //this is main hero
  415. info.mainCustomHeroName = hname.heroName;
  416. info.hasRandomHero = (hname.heroId == -1);
  417. info.mainCustomHeroId = hname.heroId;
  418. info.mainCustomHeroPortrait = -1;
  419. //todo:mainHeroPortrait
  420. }
  421. info.heroesNames.push_back(hname);
  422. }
  423. }
  424. handler.serializeBool("randomFaction", info.isFactionRandom);
  425. }
  426. }
  427. void CMapFormatJson::readTeams(JsonDeserializer & handler)
  428. {
  429. auto teams = handler.enterArray("teams");
  430. const JsonNode & src = teams.get();
  431. if(src.getType() != JsonNode::DATA_VECTOR)
  432. {
  433. // No alliances
  434. if(src.getType() != JsonNode::DATA_NULL)
  435. logGlobal->errorStream() << "Invalid teams field type";
  436. mapHeader->howManyTeams = 0;
  437. for(int i = 0; i < PlayerColor::PLAYER_LIMIT_I; i++)
  438. if(mapHeader->players[i].canComputerPlay || mapHeader->players[i].canHumanPlay)
  439. mapHeader->players[i].team = TeamID(mapHeader->howManyTeams++);
  440. }
  441. else
  442. {
  443. const JsonVector & srcVector = src.Vector();
  444. mapHeader->howManyTeams = srcVector.size();
  445. for(int team = 0; team < mapHeader->howManyTeams; team++)
  446. for(const JsonNode & playerData : srcVector[team].Vector())
  447. {
  448. PlayerColor player = PlayerColor(vstd::find_pos(GameConstants::PLAYER_COLOR_NAMES, playerData.String()));
  449. if(player.isValidPlayer())
  450. if(mapHeader->players[player.getNum()].canAnyonePlay())
  451. mapHeader->players[player.getNum()].team = TeamID(team);
  452. }
  453. for(PlayerInfo & player : mapHeader->players)
  454. if(player.canAnyonePlay() && player.team == TeamID::NO_TEAM)
  455. player.team = TeamID(mapHeader->howManyTeams++);
  456. }
  457. }
  458. void CMapFormatJson::writeTeams(JsonSerializer & handler)
  459. {
  460. std::vector<std::set<PlayerColor>> teamsData;
  461. teamsData.resize(mapHeader->howManyTeams);
  462. //get raw data
  463. for(int idx = 0; idx < mapHeader->players.size(); idx++)
  464. {
  465. const PlayerInfo & player = mapHeader->players.at(idx);
  466. int team = player.team.getNum();
  467. if(vstd::iswithin(team, 0, mapHeader->howManyTeams-1) && player.canAnyonePlay())
  468. teamsData.at(team).insert(PlayerColor(idx));
  469. }
  470. //remove single-member teams
  471. vstd::erase_if(teamsData, [](std::set<PlayerColor> & elem) -> bool
  472. {
  473. return elem.size() <= 1;
  474. });
  475. if(!teamsData.empty())
  476. {
  477. auto teams = handler.enterArray("teams");
  478. JsonNode & dest = teams.get();
  479. //construct output
  480. dest.setType(JsonNode::DATA_VECTOR);
  481. for(const std::set<PlayerColor> & teamData : teamsData)
  482. {
  483. JsonNode team(JsonNode::DATA_VECTOR);
  484. for(const PlayerColor & player : teamData)
  485. {
  486. JsonNode member(JsonNode::DATA_STRING);
  487. member.String() = GameConstants::PLAYER_COLOR_NAMES[player.getNum()];
  488. team.Vector().push_back(std::move(member));
  489. }
  490. dest.Vector().push_back(std::move(team));
  491. }
  492. }
  493. }
  494. void CMapFormatJson::readTriggeredEvents(JsonDeserializer & handler)
  495. {
  496. const JsonNode & input = handler.getCurrent();
  497. mapHeader->triggeredEvents.clear();
  498. for(auto & entry : input["triggeredEvents"].Struct())
  499. {
  500. TriggeredEvent event;
  501. event.identifier = entry.first;
  502. readTriggeredEvent(event, entry.second);
  503. mapHeader->triggeredEvents.push_back(event);
  504. }
  505. }
  506. void CMapFormatJson::readTriggeredEvent(TriggeredEvent & event, const JsonNode & source)
  507. {
  508. using namespace TriggeredEventsDetail;
  509. event.onFulfill = source["message"].String();
  510. event.description = source["description"].String();
  511. event.effect.type = vstd::find_pos(typeNames, source["effect"]["type"].String());
  512. event.effect.toOtherMessage = source["effect"]["messageToSend"].String();
  513. event.trigger = EventExpression(source["condition"], JsonToCondition); // logical expression
  514. }
  515. void CMapFormatJson::writeTriggeredEvents(JsonSerializer & handler)
  516. {
  517. JsonNode & output = handler.getCurrent();
  518. JsonMap & triggeredEvents = output["triggeredEvents"].Struct();
  519. for(auto event : mapHeader->triggeredEvents)
  520. writeTriggeredEvent(event, triggeredEvents[event.identifier]);
  521. }
  522. void CMapFormatJson::writeTriggeredEvent(const TriggeredEvent & event, JsonNode & dest)
  523. {
  524. using namespace TriggeredEventsDetail;
  525. if(!event.onFulfill.empty())
  526. dest["message"].String() = event.onFulfill;
  527. if(!event.description.empty())
  528. dest["description"].String() = event.description;
  529. dest["effect"]["type"].String() = typeNames.at(size_t(event.effect.type));
  530. if(!event.effect.toOtherMessage.empty())
  531. dest["effect"]["messageToSend"].String() = event.effect.toOtherMessage;
  532. dest["condition"] = event.trigger.toJson(ConditionToJson);
  533. }
  534. void CMapFormatJson::readDisposedHeroes(JsonSerializeFormat & handler)
  535. {
  536. auto definitions = handler.enterStruct("predefinedHeroes");//DisposedHeroes are part of predefinedHeroes in VCMI map format
  537. JsonNode & data = handler.getCurrent();
  538. for(const auto & entry : data.Struct())
  539. {
  540. HeroTypeID type(VLC->heroh->decodeHero(entry.first));
  541. ui8 mask = 0;
  542. for(const JsonNode & playerData : entry.second["availableFor"].Vector())
  543. {
  544. PlayerColor player = PlayerColor(vstd::find_pos(GameConstants::PLAYER_COLOR_NAMES, playerData.String()));
  545. if(player.isValidPlayer())
  546. {
  547. mask |= 1 << player.getNum();
  548. }
  549. }
  550. if(mask != 0 && mask != GameConstants::ALL_PLAYERS && type.getNum() >= 0)
  551. {
  552. DisposedHero hero;
  553. hero.heroId = type.getNum();
  554. hero.players = mask;
  555. //name and portrait are not used
  556. map->disposedHeroes.push_back(hero);
  557. }
  558. }
  559. }
  560. void CMapFormatJson::writeDisposedHeroes(JsonSerializeFormat & handler)
  561. {
  562. if(map->disposedHeroes.empty())
  563. return;
  564. auto definitions = handler.enterStruct("predefinedHeroes");//DisposedHeroes are part of predefinedHeroes in VCMI map format
  565. JsonNode & data = handler.getCurrent();
  566. for(const DisposedHero & hero : map->disposedHeroes)
  567. {
  568. std::string type = VLC->heroh->encodeHero(hero.heroId);
  569. JsonVector & players = data[type]["availableFor"].Vector();
  570. for(int playerNum = 0; playerNum < PlayerColor::PLAYER_LIMIT_I; playerNum++)
  571. if((1 << playerNum) & hero.players)
  572. {
  573. JsonNode player(JsonNode::DATA_STRING);
  574. player.String() = GameConstants::PLAYER_COLOR_NAMES[playerNum];
  575. players.push_back(player);
  576. }
  577. }
  578. }
  579. void CMapFormatJson::serializeRumors(JsonSerializeFormat & handler)
  580. {
  581. auto rumors = handler.enterArray("rumors");
  582. rumors.serializeStruct(map->rumors);
  583. }
  584. void CMapFormatJson::serializePredefinedHeroes(JsonSerializeFormat & handler)
  585. {
  586. //todo:serializePredefinedHeroes
  587. if(handler.saving)
  588. {
  589. if(!map->predefinedHeroes.empty())
  590. {
  591. auto predefinedHeroes = handler.enterStruct("predefinedHeroes");
  592. for(auto & hero : map->predefinedHeroes)
  593. {
  594. auto predefinedHero = handler.enterStruct(hero->getHeroTypeName());
  595. hero->serializeJsonDefinition(handler);
  596. }
  597. }
  598. }
  599. else
  600. {
  601. auto predefinedHeroes = handler.enterStruct("predefinedHeroes");
  602. const JsonNode & data = handler.getCurrent();
  603. for(const auto & p : data.Struct())
  604. {
  605. auto predefinedHero = handler.enterStruct(p.first);
  606. CGHeroInstance * hero = new CGHeroInstance();
  607. hero->ID = Obj::HERO;
  608. hero->setHeroTypeName(p.first);
  609. hero->serializeJsonDefinition(handler);
  610. map->predefinedHeroes.push_back(hero);
  611. }
  612. }
  613. }
  614. void CMapFormatJson::serializeOptions(JsonSerializeFormat & handler)
  615. {
  616. serializeRumors(handler);
  617. serializePredefinedHeroes(handler);
  618. handler.serializeLIC("allowedAbilities", &CHeroHandler::decodeSkill, &CHeroHandler::encodeSkill, VLC->heroh->getDefaultAllowedAbilities(), map->allowedAbilities);
  619. handler.serializeLIC("allowedArtifacts", &CArtHandler::decodeArfifact, &CArtHandler::encodeArtifact, VLC->arth->getDefaultAllowed(), map->allowedArtifact);
  620. handler.serializeLIC("allowedSpells", &CSpellHandler::decodeSpell, &CSpellHandler::encodeSpell, VLC->spellh->getDefaultAllowed(), map->allowedSpell);
  621. //todo:events
  622. }
  623. void CMapFormatJson::readOptions(JsonDeserializer & handler)
  624. {
  625. readDisposedHeroes(handler);
  626. serializeOptions(handler);
  627. }
  628. void CMapFormatJson::writeOptions(JsonSerializer & handler)
  629. {
  630. writeDisposedHeroes(handler);
  631. serializeOptions(handler);
  632. }
  633. ///CMapPatcher
  634. CMapPatcher::CMapPatcher(JsonNode stream):
  635. CMapFormatJson(),
  636. input(stream)
  637. {
  638. //todo: update map patches and change this
  639. fileVersionMajor = 0;
  640. fileVersionMinor = 0;
  641. }
  642. void CMapPatcher::patchMapHeader(std::unique_ptr<CMapHeader> & header)
  643. {
  644. map = nullptr;
  645. mapHeader = header.get();
  646. if (!input.isNull())
  647. readPatchData();
  648. }
  649. void CMapPatcher::readPatchData()
  650. {
  651. JsonDeserializer handler(mapObjectResolver.get(), input);
  652. readTriggeredEvents(handler);
  653. }
  654. ///CMapLoaderJson
  655. CMapLoaderJson::CMapLoaderJson(CInputStream * stream):
  656. CMapFormatJson(),
  657. buffer(stream),
  658. ioApi(new CProxyROIOApi(buffer)),
  659. loader("", "_", ioApi)
  660. {
  661. }
  662. std::unique_ptr<CMap> CMapLoaderJson::loadMap()
  663. {
  664. LOG_TRACE(logGlobal);
  665. std::unique_ptr<CMap> result = std::unique_ptr<CMap>(new CMap());
  666. map = result.get();
  667. mapHeader = map;
  668. readMap();
  669. return std::move(result);
  670. }
  671. std::unique_ptr<CMapHeader> CMapLoaderJson::loadMapHeader()
  672. {
  673. LOG_TRACE(logGlobal);
  674. map = nullptr;
  675. std::unique_ptr<CMapHeader> result = std::unique_ptr<CMapHeader>(new CMapHeader());
  676. mapHeader = result.get();
  677. readHeader(false);
  678. return std::move(result);
  679. }
  680. JsonNode CMapLoaderJson::getFromArchive(const std::string & archiveFilename)
  681. {
  682. ResourceID resource(archiveFilename, EResType::TEXT);
  683. if(!loader.existsResource(resource))
  684. throw new std::runtime_error(archiveFilename+" not found");
  685. auto data = loader.load(resource)->readAll();
  686. JsonNode res(reinterpret_cast<char*>(data.first.get()), data.second);
  687. return std::move(res);
  688. }
  689. void CMapLoaderJson::readMap()
  690. {
  691. LOG_TRACE(logGlobal);
  692. readHeader(true);
  693. map->initTerrain();
  694. readTerrain();
  695. readObjects();
  696. map->calculateGuardingGreaturePositions();
  697. }
  698. void CMapLoaderJson::readHeader(const bool complete)
  699. {
  700. //do not use map field here, use only mapHeader
  701. JsonNode header = getFromArchive(HEADER_FILE_NAME);
  702. fileVersionMajor = header["versionMajor"].Float();
  703. if(fileVersionMajor != VERSION_MAJOR)
  704. {
  705. logGlobal->errorStream() << "Unsupported map format version: " << fileVersionMajor;
  706. throw std::runtime_error("Unsupported map format version");
  707. }
  708. fileVersionMinor = header["versionMinor"].Float();
  709. if(fileVersionMinor > VERSION_MINOR)
  710. {
  711. logGlobal->warnStream() << "Too new map format revision: " << fileVersionMinor << ". This map should work but some of map features may be ignored.";
  712. }
  713. JsonDeserializer handler(mapObjectResolver.get(), header);
  714. mapHeader->version = EMapFormat::VCMI;//todo: new version field
  715. //todo: multilevel map load support
  716. {
  717. auto levels = handler.enterStruct("mapLevels");
  718. {
  719. auto surface = levels.enterStruct("surface");
  720. mapHeader->height = surface.get()["height"].Float();
  721. mapHeader->width = surface.get()["width"].Float();
  722. }
  723. {
  724. auto underground = levels.enterStruct("underground");
  725. mapHeader->twoLevel = !underground.get().isNull();
  726. }
  727. }
  728. serializeHeader(handler);
  729. readTriggeredEvents(handler);
  730. readTeams(handler);
  731. //TODO: check mods
  732. if(complete)
  733. readOptions(handler);
  734. }
  735. void CMapLoaderJson::readTerrainTile(const std::string & src, TerrainTile & tile)
  736. {
  737. using namespace TerrainDetail;
  738. {//terrain type
  739. const std::string typeCode = src.substr(0, 2);
  740. int rawType = vstd::find_pos(terrainCodes, typeCode);
  741. if(rawType < 0)
  742. throw new std::runtime_error("Invalid terrain type code in "+src);
  743. tile.terType = ETerrainType(rawType);
  744. }
  745. int startPos = 2; //0+typeCode fixed length
  746. {//terrain view
  747. int pos = startPos;
  748. while(isdigit(src.at(pos)))
  749. pos++;
  750. int len = pos - startPos;
  751. if(len<=0)
  752. throw new std::runtime_error("Invalid terrain view in "+src);
  753. const std::string rawCode = src.substr(startPos, len);
  754. tile.terView = atoi(rawCode.c_str());
  755. startPos+=len;
  756. }
  757. {//terrain flip
  758. int terrainFlip = vstd::find_pos(flipCodes, src.at(startPos++));
  759. if(terrainFlip < 0)
  760. throw new std::runtime_error("Invalid terrain flip in "+src);
  761. else
  762. tile.extTileFlags = terrainFlip;
  763. }
  764. if(startPos >= src.size())
  765. return;
  766. bool hasRoad = true;
  767. {//road type
  768. const std::string typeCode = src.substr(startPos, 2);
  769. startPos+=2;
  770. int rawType = vstd::find_pos(roadCodes, typeCode);
  771. if(rawType < 0)
  772. {
  773. rawType = vstd::find_pos(riverCodes, typeCode);
  774. if(rawType < 0)
  775. throw new std::runtime_error("Invalid river type in "+src);
  776. else
  777. {
  778. tile.riverType = ERiverType::ERiverType(rawType);
  779. hasRoad = false;
  780. }
  781. }
  782. else
  783. tile.roadType = ERoadType::ERoadType(rawType);
  784. }
  785. if(hasRoad)
  786. {//road dir
  787. int pos = startPos;
  788. while(isdigit(src.at(pos)))
  789. pos++;
  790. int len = pos - startPos;
  791. if(len<=0)
  792. throw new std::runtime_error("Invalid road dir in "+src);
  793. const std::string rawCode = src.substr(startPos, len);
  794. tile.roadDir = atoi(rawCode.c_str());
  795. startPos+=len;
  796. }
  797. if(hasRoad)
  798. {//road flip
  799. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  800. if(flip < 0)
  801. throw new std::runtime_error("Invalid road flip in "+src);
  802. else
  803. tile.extTileFlags |= (flip<<4);
  804. }
  805. if(startPos >= src.size())
  806. return;
  807. if(hasRoad)
  808. {//river type
  809. const std::string typeCode = src.substr(startPos, 2);
  810. startPos+=2;
  811. int rawType = vstd::find_pos(riverCodes, typeCode);
  812. if(rawType < 0)
  813. throw new std::runtime_error("Invalid river type in "+src);
  814. tile.riverType = ERiverType::ERiverType(rawType);
  815. }
  816. {//river dir
  817. int pos = startPos;
  818. while(isdigit(src.at(pos)))
  819. pos++;
  820. int len = pos - startPos;
  821. if(len<=0)
  822. throw new std::runtime_error("Invalid river dir in "+src);
  823. const std::string rawCode = src.substr(startPos, len);
  824. tile.riverDir = atoi(rawCode.c_str());
  825. startPos+=len;
  826. }
  827. {//river flip
  828. int flip = vstd::find_pos(flipCodes, src.at(startPos++));
  829. if(flip < 0)
  830. throw new std::runtime_error("Invalid road flip in "+src);
  831. else
  832. tile.extTileFlags |= (flip<<2);
  833. }
  834. }
  835. void CMapLoaderJson::readTerrainLevel(const JsonNode & src, const int index)
  836. {
  837. int3 pos(0, 0, index);
  838. const JsonVector & rows = src.Vector();
  839. if(rows.size() != map->height)
  840. throw new std::runtime_error("Invalid terrain data");
  841. for(pos.y = 0; pos.y < map->height; pos.y++)
  842. {
  843. const JsonVector & tiles = rows[pos.y].Vector();
  844. if(tiles.size() != map->width)
  845. throw new std::runtime_error("Invalid terrain data");
  846. for(pos.x = 0; pos.x < map->width; pos.x++)
  847. readTerrainTile(tiles[pos.x].String(), map->getTile(pos));
  848. }
  849. }
  850. void CMapLoaderJson::readTerrain()
  851. {
  852. {
  853. const JsonNode surface = getFromArchive("surface_terrain.json");
  854. readTerrainLevel(surface, 0);
  855. }
  856. if(map->twoLevel)
  857. {
  858. const JsonNode underground = getFromArchive("underground_terrain.json");
  859. readTerrainLevel(underground, 1);
  860. }
  861. }
  862. CMapLoaderJson::MapObjectLoader::MapObjectLoader(CMapLoaderJson * _owner, JsonMap::value_type & json):
  863. owner(_owner), instance(nullptr), id(-1), jsonKey(json.first), configuration(json.second)
  864. {
  865. }
  866. void CMapLoaderJson::MapObjectLoader::construct()
  867. {
  868. //logGlobal->debugStream() <<"Loading: " <<jsonKey;
  869. //TODO:consider move to ObjectTypeHandler
  870. //find type handler
  871. std::string typeName = configuration["type"].String(), subtypeName = configuration["subtype"].String();
  872. if(typeName.empty())
  873. {
  874. logGlobal->errorStream() << "Object type missing";
  875. logGlobal->debugStream() << configuration;
  876. return;
  877. }
  878. int3 pos;
  879. pos.x = configuration["x"].Float();
  880. pos.y = configuration["y"].Float();
  881. pos.z = configuration["l"].Float();
  882. //special case for grail
  883. if(typeName == "grail")
  884. {
  885. owner->map->grailPos = pos;
  886. owner->map->grailRadius = configuration["options"]["grailRadius"].Float();
  887. return;
  888. }
  889. else if(subtypeName.empty())
  890. {
  891. logGlobal->errorStream() << "Object subtype missing";
  892. logGlobal->debugStream() << configuration;
  893. return;
  894. }
  895. auto handler = VLC->objtypeh->getHandlerFor(typeName, subtypeName);
  896. ObjectTemplate appearance;
  897. appearance.id = Obj(handler->type);
  898. appearance.subid = handler->subtype;
  899. appearance.readJson(configuration["template"], false);
  900. instance = handler->create(appearance);
  901. instance->id = ObjectInstanceID(owner->map->objects.size());
  902. instance->instanceName = jsonKey;
  903. instance->pos = pos;
  904. owner->map->addNewObject(instance);
  905. }
  906. void CMapLoaderJson::MapObjectLoader::configure()
  907. {
  908. if(nullptr == instance)
  909. return;
  910. JsonDeserializer handler(owner->mapObjectResolver.get(), configuration);
  911. instance->serializeJson(handler);
  912. //artifact instance serialization requires access to Map object, handle it here for now
  913. //todo: find better solution for artifact instance serialization
  914. if(auto art = dynamic_cast<CGArtifact *>(instance))
  915. {
  916. int artID = ArtifactID::NONE;
  917. int spellID = -1;
  918. if(art->ID == Obj::SPELL_SCROLL)
  919. {
  920. auto spellIdentifier = configuration["options"]["spell"].String();
  921. auto rawId = VLC->modh->identifiers.getIdentifier("core", "spell", spellIdentifier);
  922. if(rawId)
  923. spellID = rawId.get();
  924. else
  925. spellID = 0;
  926. artID = ArtifactID::SPELL_SCROLL;
  927. }
  928. else if(art->ID == Obj::ARTIFACT)
  929. {
  930. //specific artifact
  931. artID = art->subID;
  932. }
  933. art->storedArtifact = CArtifactInstance::createArtifact(owner->map, artID, spellID);
  934. }
  935. if(auto hero = dynamic_cast<CGHeroInstance *>(instance))
  936. {
  937. auto o = handler.enterStruct("options");
  938. hero->serializeJsonArtifacts(handler, "artifacts", owner->map);
  939. }
  940. }
  941. void CMapLoaderJson::readObjects()
  942. {
  943. LOG_TRACE(logGlobal);
  944. std::vector<std::unique_ptr<MapObjectLoader>> loaders;//todo: optimize MapObjectLoader memory layout
  945. JsonNode data = getFromArchive(OBJECTS_FILE_NAME);
  946. //get raw data
  947. for(auto & p : data.Struct())
  948. loaders.push_back(vstd::make_unique<MapObjectLoader>(this, p));
  949. for(auto & ptr : loaders)
  950. ptr->construct();
  951. //configure objects after all objects are constructed so we may resolve internal IDs even to actual pointers OTF
  952. for(auto & ptr : loaders)
  953. ptr->configure();
  954. std::sort(map->heroesOnMap.begin(), map->heroesOnMap.end(), [](const ConstTransitivePtr<CGHeroInstance> & a, const ConstTransitivePtr<CGHeroInstance> & b)
  955. {
  956. return a->subID < b->subID;
  957. });
  958. }
  959. ///CMapSaverJson
  960. CMapSaverJson::CMapSaverJson(CInputOutputStream * stream):
  961. CMapFormatJson(),
  962. buffer(stream),
  963. ioApi(new CProxyIOApi(buffer)),
  964. saver(ioApi, "_")
  965. {
  966. fileVersionMajor = VERSION_MAJOR;
  967. fileVersionMinor = VERSION_MINOR;
  968. }
  969. CMapSaverJson::~CMapSaverJson()
  970. {
  971. }
  972. void CMapSaverJson::addToArchive(const JsonNode & data, const std::string & filename)
  973. {
  974. std::ostringstream out;
  975. out << data;
  976. out.flush();
  977. {
  978. auto s = out.str();
  979. std::unique_ptr<COutputStream> stream = saver.addFile(filename);
  980. if (stream->write((const ui8*)s.c_str(), s.size()) != s.size())
  981. throw new std::runtime_error("CMapSaverJson::saveHeader() zip compression failed.");
  982. }
  983. }
  984. void CMapSaverJson::saveMap(const std::unique_ptr<CMap>& map)
  985. {
  986. this->map = map.get();
  987. this->mapHeader = this->map;
  988. writeHeader();
  989. writeTerrain();
  990. writeObjects();
  991. }
  992. void CMapSaverJson::writeHeader()
  993. {
  994. JsonNode header;
  995. JsonSerializer handler(mapObjectResolver.get(), header);
  996. header["versionMajor"].Float() = VERSION_MAJOR;
  997. header["versionMinor"].Float() = VERSION_MINOR;
  998. //todo: multilevel map save support
  999. JsonNode & levels = header["mapLevels"];
  1000. levels["surface"]["height"].Float() = mapHeader->height;
  1001. levels["surface"]["width"].Float() = mapHeader->width;
  1002. levels["surface"]["index"].Float() = 0;
  1003. if(mapHeader->twoLevel)
  1004. {
  1005. levels["underground"]["height"].Float() = mapHeader->height;
  1006. levels["underground"]["width"].Float() = mapHeader->width;
  1007. levels["underground"]["index"].Float() = 1;
  1008. }
  1009. serializeHeader(handler);
  1010. writeTriggeredEvents(handler);
  1011. writeTeams(handler);
  1012. writeOptions(handler);
  1013. addToArchive(header, HEADER_FILE_NAME);
  1014. }
  1015. const std::string CMapSaverJson::writeTerrainTile(const TerrainTile & tile)
  1016. {
  1017. using namespace TerrainDetail;
  1018. std::ostringstream out;
  1019. out.setf(std::ios::dec, std::ios::basefield);
  1020. out.unsetf(std::ios::showbase);
  1021. out << terrainCodes.at(int(tile.terType)) << (int)tile.terView << flipCodes[tile.extTileFlags % 4];
  1022. if(tile.roadType != ERoadType::NO_ROAD)
  1023. out << roadCodes.at(int(tile.roadType)) << (int)tile.roadDir << flipCodes[(tile.extTileFlags >> 4) % 4];
  1024. if(tile.riverType != ERiverType::NO_RIVER)
  1025. out << riverCodes.at(int(tile.riverType)) << (int)tile.riverDir << flipCodes[(tile.extTileFlags >> 2) % 4];
  1026. return out.str();
  1027. }
  1028. JsonNode CMapSaverJson::writeTerrainLevel(const int index)
  1029. {
  1030. JsonNode data;
  1031. int3 pos(0,0,index);
  1032. data.Vector().resize(map->height);
  1033. for(pos.y = 0; pos.y < map->height; pos.y++)
  1034. {
  1035. JsonNode & row = data.Vector()[pos.y];
  1036. row.Vector().resize(map->width);
  1037. for(pos.x = 0; pos.x < map->width; pos.x++)
  1038. row.Vector()[pos.x].String() = writeTerrainTile(map->getTile(pos));
  1039. }
  1040. return std::move(data);
  1041. }
  1042. void CMapSaverJson::writeTerrain()
  1043. {
  1044. //todo: multilevel map save support
  1045. JsonNode surface = writeTerrainLevel(0);
  1046. addToArchive(surface, "surface_terrain.json");
  1047. if(map->twoLevel)
  1048. {
  1049. JsonNode underground = writeTerrainLevel(1);
  1050. addToArchive(underground, "underground_terrain.json");
  1051. }
  1052. }
  1053. void CMapSaverJson::writeObjects()
  1054. {
  1055. JsonNode data(JsonNode::DATA_STRUCT);
  1056. JsonSerializer handler(mapObjectResolver.get(), data);
  1057. for(CGObjectInstance * obj : map->objects)
  1058. {
  1059. auto temp = handler.enterStruct(obj->instanceName);
  1060. obj->serializeJson(handler);
  1061. }
  1062. if(map->grailPos.valid())
  1063. {
  1064. JsonNode grail(JsonNode::DATA_STRUCT);
  1065. grail["type"].String() = "grail";
  1066. grail["x"].Float() = map->grailPos.x;
  1067. grail["y"].Float() = map->grailPos.y;
  1068. grail["l"].Float() = map->grailPos.z;
  1069. grail["options"]["radius"].Float() = map->grailRadius;
  1070. std::string grailId = boost::str(boost::format("grail_%d") % map->objects.size());
  1071. data[grailId] = grail;
  1072. }
  1073. addToArchive(data, OBJECTS_FILE_NAME);
  1074. }