MapFormatJson.cpp 36 KB

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