JsonNode.cpp 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635
  1. /*
  2. * JsonNode.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 "JsonNode.h"
  12. #include "ScopeGuard.h"
  13. #include "bonuses/BonusParams.h"
  14. #include "bonuses/Bonus.h"
  15. #include "bonuses/Limiters.h"
  16. #include "bonuses/Propagators.h"
  17. #include "bonuses/Updaters.h"
  18. #include "filesystem/Filesystem.h"
  19. #include "modding/IdentifierStorage.h"
  20. #include "VCMI_Lib.h" //for identifier resolution
  21. #include "CGeneralTextHandler.h"
  22. #include "JsonDetail.h"
  23. #include "constants/StringConstants.h"
  24. #include "battle/BattleHex.h"
  25. namespace
  26. {
  27. // to avoid duplicating const and non-const code
  28. template<typename Node>
  29. Node & resolvePointer(Node & in, const std::string & pointer)
  30. {
  31. if(pointer.empty())
  32. return in;
  33. assert(pointer[0] == '/');
  34. size_t splitPos = pointer.find('/', 1);
  35. std::string entry = pointer.substr(1, splitPos - 1);
  36. std::string remainer = splitPos == std::string::npos ? "" : pointer.substr(splitPos);
  37. if(in.getType() == VCMI_LIB_WRAP_NAMESPACE(JsonNode)::JsonType::DATA_VECTOR)
  38. {
  39. if(entry.find_first_not_of("0123456789") != std::string::npos) // non-numbers in string
  40. throw std::runtime_error("Invalid Json pointer");
  41. if(entry.size() > 1 && entry[0] == '0') // leading zeros are not allowed
  42. throw std::runtime_error("Invalid Json pointer");
  43. auto index = boost::lexical_cast<size_t>(entry);
  44. if (in.Vector().size() > index)
  45. return in.Vector()[index].resolvePointer(remainer);
  46. }
  47. return in[entry].resolvePointer(remainer);
  48. }
  49. }
  50. VCMI_LIB_NAMESPACE_BEGIN
  51. using namespace JsonDetail;
  52. class LibClasses;
  53. class CModHandler;
  54. static const JsonNode nullNode;
  55. JsonNode::JsonNode(JsonType Type)
  56. {
  57. setType(Type);
  58. }
  59. JsonNode::JsonNode(const char *data, size_t datasize)
  60. {
  61. JsonParser parser(data, datasize);
  62. *this = parser.parse("<unknown>");
  63. }
  64. JsonNode::JsonNode(const JsonPath & fileURI)
  65. {
  66. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  67. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  68. *this = parser.parse(fileURI.getName());
  69. }
  70. JsonNode::JsonNode(const std::string & idx, const JsonPath & fileURI)
  71. {
  72. auto file = CResourceHandler::get(idx)->load(fileURI)->readAll();
  73. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  74. *this = parser.parse(fileURI.getName());
  75. }
  76. JsonNode::JsonNode(const JsonPath & fileURI, bool &isValidSyntax)
  77. {
  78. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  79. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  80. *this = parser.parse(fileURI.getName());
  81. isValidSyntax = parser.isValid();
  82. }
  83. bool JsonNode::operator == (const JsonNode &other) const
  84. {
  85. return data == other.data;
  86. }
  87. bool JsonNode::operator != (const JsonNode &other) const
  88. {
  89. return !(*this == other);
  90. }
  91. JsonNode::JsonType JsonNode::getType() const
  92. {
  93. return static_cast<JsonType>(data.index());
  94. }
  95. void JsonNode::setMeta(const std::string & metadata, bool recursive)
  96. {
  97. meta = metadata;
  98. if (recursive)
  99. {
  100. switch (getType())
  101. {
  102. break; case JsonType::DATA_VECTOR:
  103. {
  104. for(auto & node : Vector())
  105. {
  106. node.setMeta(metadata);
  107. }
  108. }
  109. break; case JsonType::DATA_STRUCT:
  110. {
  111. for(auto & node : Struct())
  112. {
  113. node.second.setMeta(metadata);
  114. }
  115. }
  116. }
  117. }
  118. }
  119. void JsonNode::setType(JsonType Type)
  120. {
  121. if (getType() == Type)
  122. return;
  123. //float<->int conversion
  124. if(getType() == JsonType::DATA_FLOAT && Type == JsonType::DATA_INTEGER)
  125. {
  126. si64 converted = static_cast<si64>(std::get<double>(data));
  127. data = JsonData(converted);
  128. return;
  129. }
  130. else if(getType() == JsonType::DATA_INTEGER && Type == JsonType::DATA_FLOAT)
  131. {
  132. double converted = static_cast<double>(std::get<si64>(data));
  133. data = JsonData(converted);
  134. return;
  135. }
  136. //Set new node type
  137. switch(Type)
  138. {
  139. break; case JsonType::DATA_NULL: data = JsonData();
  140. break; case JsonType::DATA_BOOL: data = JsonData(false);
  141. break; case JsonType::DATA_FLOAT: data = JsonData(static_cast<double>(0.0));
  142. break; case JsonType::DATA_STRING: data = JsonData(std::string());
  143. break; case JsonType::DATA_VECTOR: data = JsonData(JsonVector());
  144. break; case JsonType::DATA_STRUCT: data = JsonData(JsonMap());
  145. break; case JsonType::DATA_INTEGER: data = JsonData(static_cast<si64>(0));
  146. }
  147. }
  148. bool JsonNode::isNull() const
  149. {
  150. return getType() == JsonType::DATA_NULL;
  151. }
  152. bool JsonNode::isNumber() const
  153. {
  154. return getType() == JsonType::DATA_INTEGER || getType() == JsonType::DATA_FLOAT;
  155. }
  156. bool JsonNode::isString() const
  157. {
  158. return getType() == JsonType::DATA_STRING;
  159. }
  160. bool JsonNode::isVector() const
  161. {
  162. return getType() == JsonType::DATA_VECTOR;
  163. }
  164. bool JsonNode::isStruct() const
  165. {
  166. return getType() == JsonType::DATA_STRUCT;
  167. }
  168. bool JsonNode::containsBaseData() const
  169. {
  170. switch(getType())
  171. {
  172. case JsonType::DATA_NULL:
  173. return false;
  174. case JsonType::DATA_STRUCT:
  175. for(const auto & elem : Struct())
  176. {
  177. if(elem.second.containsBaseData())
  178. return true;
  179. }
  180. return false;
  181. default:
  182. //other types (including vector) cannot be extended via merge
  183. return true;
  184. }
  185. }
  186. bool JsonNode::isCompact() const
  187. {
  188. switch(getType())
  189. {
  190. case JsonType::DATA_VECTOR:
  191. for(const JsonNode & elem : Vector())
  192. {
  193. if(!elem.isCompact())
  194. return false;
  195. }
  196. return true;
  197. case JsonType::DATA_STRUCT:
  198. {
  199. auto propertyCount = Struct().size();
  200. if(propertyCount == 0)
  201. return true;
  202. else if(propertyCount == 1)
  203. return Struct().begin()->second.isCompact();
  204. }
  205. return false;
  206. default:
  207. return true;
  208. }
  209. }
  210. bool JsonNode::TryBoolFromString(bool & success) const
  211. {
  212. success = true;
  213. if(getType() == JsonNode::JsonType::DATA_BOOL)
  214. return Bool();
  215. success = getType() == JsonNode::JsonType::DATA_STRING;
  216. if(success)
  217. {
  218. auto boolParamStr = String();
  219. boost::algorithm::trim(boolParamStr);
  220. boost::algorithm::to_lower(boolParamStr);
  221. success = boolParamStr == "true";
  222. if(success)
  223. return true;
  224. success = boolParamStr == "false";
  225. }
  226. return false;
  227. }
  228. void JsonNode::clear()
  229. {
  230. setType(JsonType::DATA_NULL);
  231. }
  232. bool & JsonNode::Bool()
  233. {
  234. setType(JsonType::DATA_BOOL);
  235. return std::get<bool>(data);
  236. }
  237. double & JsonNode::Float()
  238. {
  239. setType(JsonType::DATA_FLOAT);
  240. return std::get<double>(data);
  241. }
  242. si64 & JsonNode::Integer()
  243. {
  244. setType(JsonType::DATA_INTEGER);
  245. return std::get<si64>(data);
  246. }
  247. std::string & JsonNode::String()
  248. {
  249. setType(JsonType::DATA_STRING);
  250. return std::get<std::string>(data);
  251. }
  252. JsonVector & JsonNode::Vector()
  253. {
  254. setType(JsonType::DATA_VECTOR);
  255. return std::get<JsonVector>(data);
  256. }
  257. JsonMap & JsonNode::Struct()
  258. {
  259. setType(JsonType::DATA_STRUCT);
  260. return std::get<JsonMap>(data);
  261. }
  262. const bool boolDefault = false;
  263. bool JsonNode::Bool() const
  264. {
  265. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_BOOL);
  266. if (getType() == JsonType::DATA_BOOL)
  267. return std::get<bool>(data);
  268. return boolDefault;
  269. }
  270. const double floatDefault = 0;
  271. double JsonNode::Float() const
  272. {
  273. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_INTEGER || getType() == JsonType::DATA_FLOAT);
  274. if(getType() == JsonType::DATA_FLOAT)
  275. return std::get<double>(data);
  276. if(getType() == JsonType::DATA_INTEGER)
  277. return static_cast<double>(std::get<si64>(data));
  278. return floatDefault;
  279. }
  280. const si64 integerDefault = 0;
  281. si64 JsonNode::Integer() const
  282. {
  283. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_INTEGER || getType() == JsonType::DATA_FLOAT);
  284. if(getType() == JsonType::DATA_INTEGER)
  285. return std::get<si64>(data);
  286. if(getType() == JsonType::DATA_FLOAT)
  287. return static_cast<si64>(std::get<double>(data));
  288. return integerDefault;
  289. }
  290. const std::string stringDefault = std::string();
  291. const std::string & JsonNode::String() const
  292. {
  293. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_STRING);
  294. if (getType() == JsonType::DATA_STRING)
  295. return std::get<std::string>(data);
  296. return stringDefault;
  297. }
  298. const JsonVector vectorDefault = JsonVector();
  299. const JsonVector & JsonNode::Vector() const
  300. {
  301. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_VECTOR);
  302. if (getType() == JsonType::DATA_VECTOR)
  303. return std::get<JsonVector>(data);
  304. return vectorDefault;
  305. }
  306. const JsonMap mapDefault = JsonMap();
  307. const JsonMap & JsonNode::Struct() const
  308. {
  309. assert(getType() == JsonType::DATA_NULL || getType() == JsonType::DATA_STRUCT);
  310. if (getType() == JsonType::DATA_STRUCT)
  311. return std::get<JsonMap>(data);
  312. return mapDefault;
  313. }
  314. JsonNode & JsonNode::operator[](const std::string & child)
  315. {
  316. return Struct()[child];
  317. }
  318. const JsonNode & JsonNode::operator[](const std::string & child) const
  319. {
  320. auto it = Struct().find(child);
  321. if (it != Struct().end())
  322. return it->second;
  323. return nullNode;
  324. }
  325. JsonNode & JsonNode::operator[](size_t child)
  326. {
  327. if (child >= Vector().size() )
  328. Vector().resize(child + 1);
  329. return Vector()[child];
  330. }
  331. const JsonNode & JsonNode::operator[](size_t child) const
  332. {
  333. if (child < Vector().size() )
  334. return Vector()[child];
  335. return nullNode;
  336. }
  337. const JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer) const
  338. {
  339. return ::resolvePointer(*this, jsonPointer);
  340. }
  341. JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer)
  342. {
  343. return ::resolvePointer(*this, jsonPointer);
  344. }
  345. std::string JsonNode::toJson(bool compact) const
  346. {
  347. std::ostringstream out;
  348. JsonWriter writer(out, compact);
  349. writer.writeNode(*this);
  350. return out.str();
  351. }
  352. ///JsonUtils
  353. static void loadBonusSubtype(BonusSubtypeID & subtype, BonusType type, const JsonNode & node)
  354. {
  355. if (node.isNull())
  356. {
  357. subtype = BonusSubtypeID();
  358. return;
  359. }
  360. if (node.isNumber()) // Compatibility code for 1.3 or older
  361. {
  362. logMod->warn("Bonus subtype must be string! (%s)", node.meta);
  363. subtype = BonusCustomSubtype(node.Integer());
  364. return;
  365. }
  366. if (!node.isString())
  367. {
  368. logMod->warn("Bonus subtype must be string! (%s)", node.meta);
  369. subtype = BonusSubtypeID();
  370. return;
  371. }
  372. switch (type)
  373. {
  374. case BonusType::MAGIC_SCHOOL_SKILL:
  375. case BonusType::SPELL_DAMAGE:
  376. case BonusType::SPELLS_OF_SCHOOL:
  377. case BonusType::SPELL_DAMAGE_REDUCTION:
  378. case BonusType::SPELL_SCHOOL_IMMUNITY:
  379. case BonusType::NEGATIVE_EFFECTS_IMMUNITY:
  380. {
  381. VLC->identifiers()->requestIdentifier( "spellSchool", node, [&subtype](int32_t identifier)
  382. {
  383. subtype = SpellSchool(identifier);
  384. });
  385. break;
  386. }
  387. case BonusType::NO_TERRAIN_PENALTY:
  388. {
  389. VLC->identifiers()->requestIdentifier( "terrain", node, [&subtype](int32_t identifier)
  390. {
  391. subtype = TerrainId(identifier);
  392. });
  393. break;
  394. }
  395. case BonusType::PRIMARY_SKILL:
  396. {
  397. VLC->identifiers()->requestIdentifier( "primarySkill", node, [&subtype](int32_t identifier)
  398. {
  399. subtype = PrimarySkill(identifier);
  400. });
  401. break;
  402. }
  403. case BonusType::IMPROVED_NECROMANCY:
  404. case BonusType::HERO_GRANTS_ATTACKS:
  405. case BonusType::BONUS_DAMAGE_CHANCE:
  406. case BonusType::BONUS_DAMAGE_PERCENTAGE:
  407. case BonusType::SPECIAL_UPGRADE:
  408. case BonusType::HATE:
  409. case BonusType::SUMMON_GUARDIANS:
  410. case BonusType::MANUAL_CONTROL:
  411. {
  412. VLC->identifiers()->requestIdentifier( "creature", node, [&subtype](int32_t identifier)
  413. {
  414. subtype = CreatureID(identifier);
  415. });
  416. break;
  417. }
  418. case BonusType::SPELL_IMMUNITY:
  419. case BonusType::SPELL_DURATION:
  420. case BonusType::SPECIAL_ADD_VALUE_ENCHANT:
  421. case BonusType::SPECIAL_FIXED_VALUE_ENCHANT:
  422. case BonusType::SPECIAL_PECULIAR_ENCHANT:
  423. case BonusType::SPECIAL_SPELL_LEV:
  424. case BonusType::SPECIFIC_SPELL_DAMAGE:
  425. case BonusType::SPELL:
  426. case BonusType::OPENING_BATTLE_SPELL:
  427. case BonusType::SPELL_LIKE_ATTACK:
  428. case BonusType::CATAPULT:
  429. case BonusType::CATAPULT_EXTRA_SHOTS:
  430. case BonusType::HEALER:
  431. case BonusType::SPELLCASTER:
  432. case BonusType::ENCHANTER:
  433. case BonusType::SPELL_AFTER_ATTACK:
  434. case BonusType::SPELL_BEFORE_ATTACK:
  435. case BonusType::SPECIFIC_SPELL_POWER:
  436. case BonusType::ENCHANTED:
  437. case BonusType::MORE_DAMAGE_FROM_SPELL:
  438. case BonusType::NOT_ACTIVE:
  439. {
  440. VLC->identifiers()->requestIdentifier( "spell", node, [&subtype](int32_t identifier)
  441. {
  442. subtype = SpellID(identifier);
  443. });
  444. break;
  445. }
  446. case BonusType::GENERATE_RESOURCE:
  447. {
  448. VLC->identifiers()->requestIdentifier( "resource", node, [&subtype](int32_t identifier)
  449. {
  450. subtype = GameResID(identifier);
  451. });
  452. break;
  453. }
  454. case BonusType::MOVEMENT:
  455. case BonusType::WATER_WALKING:
  456. case BonusType::FLYING_MOVEMENT:
  457. case BonusType::NEGATE_ALL_NATURAL_IMMUNITIES:
  458. case BonusType::CREATURE_DAMAGE:
  459. case BonusType::FLYING:
  460. case BonusType::GENERAL_DAMAGE_REDUCTION:
  461. case BonusType::PERCENTAGE_DAMAGE_BOOST:
  462. case BonusType::SOUL_STEAL:
  463. case BonusType::TRANSMUTATION:
  464. case BonusType::DESTRUCTION:
  465. case BonusType::DEATH_STARE:
  466. case BonusType::REBIRTH:
  467. case BonusType::VISIONS:
  468. case BonusType::SPELLS_OF_LEVEL: // spell level
  469. case BonusType::CREATURE_GROWTH: // creature level
  470. {
  471. VLC->identifiers()->requestIdentifier( "bonusSubtype", node, [&subtype](int32_t identifier)
  472. {
  473. subtype = BonusCustomSubtype(identifier);
  474. });
  475. break;
  476. }
  477. default:
  478. for(const auto & i : bonusNameMap)
  479. if(i.second == type)
  480. logMod->warn("Bonus type %s does not supports subtypes!", i.first );
  481. subtype = BonusSubtypeID();
  482. }
  483. }
  484. static void loadBonusSourceInstance(BonusSourceID & sourceInstance, BonusSource sourceType, const JsonNode & node)
  485. {
  486. if (node.isNull())
  487. {
  488. sourceInstance = BonusCustomSource();
  489. return;
  490. }
  491. if (node.isNumber()) // Compatibility code for 1.3 or older
  492. {
  493. logMod->warn("Bonus source must be string!");
  494. sourceInstance = BonusCustomSource(node.Integer());
  495. return;
  496. }
  497. if (!node.isString())
  498. {
  499. logMod->warn("Bonus source must be string!");
  500. sourceInstance = BonusCustomSource();
  501. return;
  502. }
  503. switch (sourceType)
  504. {
  505. case BonusSource::ARTIFACT:
  506. case BonusSource::ARTIFACT_INSTANCE:
  507. {
  508. VLC->identifiers()->requestIdentifier( "artifact", node, [&sourceInstance](int32_t identifier)
  509. {
  510. sourceInstance = ArtifactID(identifier);
  511. });
  512. break;
  513. }
  514. case BonusSource::OBJECT_TYPE:
  515. {
  516. VLC->identifiers()->requestIdentifier( "object", node, [&sourceInstance](int32_t identifier)
  517. {
  518. sourceInstance = Obj(identifier);
  519. });
  520. break;
  521. }
  522. case BonusSource::OBJECT_INSTANCE:
  523. case BonusSource::HERO_BASE_SKILL:
  524. sourceInstance = ObjectInstanceID(ObjectInstanceID::decode(node.String()));
  525. break;
  526. case BonusSource::CREATURE_ABILITY:
  527. {
  528. VLC->identifiers()->requestIdentifier( "creature", node, [&sourceInstance](int32_t identifier)
  529. {
  530. sourceInstance = CreatureID(identifier);
  531. });
  532. break;
  533. }
  534. case BonusSource::TERRAIN_OVERLAY:
  535. {
  536. VLC->identifiers()->requestIdentifier( "spell", node, [&sourceInstance](int32_t identifier)
  537. {
  538. sourceInstance = BattleField(identifier);
  539. });
  540. break;
  541. }
  542. case BonusSource::SPELL_EFFECT:
  543. {
  544. VLC->identifiers()->requestIdentifier( "spell", node, [&sourceInstance](int32_t identifier)
  545. {
  546. sourceInstance = SpellID(identifier);
  547. });
  548. break;
  549. }
  550. case BonusSource::TOWN_STRUCTURE:
  551. assert(0); // TODO
  552. sourceInstance = BuildingTypeUniqueID();
  553. break;
  554. case BonusSource::SECONDARY_SKILL:
  555. {
  556. VLC->identifiers()->requestIdentifier( "secondarySkill", node, [&sourceInstance](int32_t identifier)
  557. {
  558. sourceInstance = SecondarySkill(identifier);
  559. });
  560. break;
  561. }
  562. case BonusSource::HERO_SPECIAL:
  563. {
  564. VLC->identifiers()->requestIdentifier( "hero", node, [&sourceInstance](int32_t identifier)
  565. {
  566. sourceInstance = HeroTypeID(identifier);
  567. });
  568. break;
  569. }
  570. case BonusSource::CAMPAIGN_BONUS:
  571. sourceInstance = CampaignScenarioID(CampaignScenarioID::decode(node.String()));
  572. break;
  573. case BonusSource::ARMY:
  574. case BonusSource::STACK_EXPERIENCE:
  575. case BonusSource::COMMANDER:
  576. case BonusSource::GLOBAL:
  577. case BonusSource::TERRAIN_NATIVE:
  578. case BonusSource::OTHER:
  579. default:
  580. sourceInstance = BonusSourceID();
  581. break;
  582. }
  583. }
  584. std::shared_ptr<Bonus> JsonUtils::parseBonus(const JsonVector & ability_vec)
  585. {
  586. auto b = std::make_shared<Bonus>();
  587. std::string type = ability_vec[0].String();
  588. auto it = bonusNameMap.find(type);
  589. if (it == bonusNameMap.end())
  590. {
  591. logMod->error("Error: invalid ability type %s.", type);
  592. return b;
  593. }
  594. b->type = it->second;
  595. b->val = static_cast<si32>(ability_vec[1].Float());
  596. loadBonusSubtype(b->subtype, b->type, ability_vec[2]);
  597. b->additionalInfo = static_cast<si32>(ability_vec[3].Float());
  598. b->duration = BonusDuration::PERMANENT; //TODO: handle flags (as integer)
  599. b->turnsRemain = 0;
  600. return b;
  601. }
  602. template <typename T>
  603. const T parseByMap(const std::map<std::string, T> & map, const JsonNode * val, const std::string & err)
  604. {
  605. static T defaultValue = T();
  606. if (!val->isNull())
  607. {
  608. const std::string & type = val->String();
  609. auto it = map.find(type);
  610. if (it == map.end())
  611. {
  612. logMod->error("Error: invalid %s%s.", err, type);
  613. return defaultValue;
  614. }
  615. else
  616. {
  617. return it->second;
  618. }
  619. }
  620. else
  621. return defaultValue;
  622. }
  623. template <typename T>
  624. const T parseByMapN(const std::map<std::string, T> & map, const JsonNode * val, const std::string & err)
  625. {
  626. if(val->isNumber())
  627. return static_cast<T>(val->Integer());
  628. else
  629. return parseByMap<T>(map, val, err);
  630. }
  631. void JsonUtils::resolveAddInfo(CAddInfo & var, const JsonNode & node)
  632. {
  633. const JsonNode & value = node["addInfo"];
  634. if (!value.isNull())
  635. {
  636. switch (value.getType())
  637. {
  638. case JsonNode::JsonType::DATA_INTEGER:
  639. var = static_cast<si32>(value.Integer());
  640. break;
  641. case JsonNode::JsonType::DATA_FLOAT:
  642. var = static_cast<si32>(value.Float());
  643. break;
  644. case JsonNode::JsonType::DATA_STRING:
  645. VLC->identifiers()->requestIdentifier(value, [&](si32 identifier)
  646. {
  647. var = identifier;
  648. });
  649. break;
  650. case JsonNode::JsonType::DATA_VECTOR:
  651. {
  652. const JsonVector & vec = value.Vector();
  653. var.resize(vec.size());
  654. for(int i = 0; i < vec.size(); i++)
  655. {
  656. switch(vec[i].getType())
  657. {
  658. case JsonNode::JsonType::DATA_INTEGER:
  659. var[i] = static_cast<si32>(vec[i].Integer());
  660. break;
  661. case JsonNode::JsonType::DATA_FLOAT:
  662. var[i] = static_cast<si32>(vec[i].Float());
  663. break;
  664. case JsonNode::JsonType::DATA_STRING:
  665. VLC->identifiers()->requestIdentifier(vec[i], [&var,i](si32 identifier)
  666. {
  667. var[i] = identifier;
  668. });
  669. break;
  670. default:
  671. logMod->error("Error! Wrong identifier used for value of addInfo[%d].", i);
  672. }
  673. }
  674. break;
  675. }
  676. default:
  677. logMod->error("Error! Wrong identifier used for value of addInfo.");
  678. }
  679. }
  680. }
  681. std::shared_ptr<ILimiter> JsonUtils::parseLimiter(const JsonNode & limiter)
  682. {
  683. switch(limiter.getType())
  684. {
  685. case JsonNode::JsonType::DATA_VECTOR:
  686. {
  687. const JsonVector & subLimiters = limiter.Vector();
  688. if(subLimiters.empty())
  689. {
  690. logMod->warn("Warning: empty limiter list");
  691. return std::make_shared<AllOfLimiter>();
  692. }
  693. std::shared_ptr<AggregateLimiter> result;
  694. int offset = 1;
  695. // determine limiter type and offset for sub-limiters
  696. if(subLimiters[0].getType() == JsonNode::JsonType::DATA_STRING)
  697. {
  698. const std::string & aggregator = subLimiters[0].String();
  699. if(aggregator == AllOfLimiter::aggregator)
  700. result = std::make_shared<AllOfLimiter>();
  701. else if(aggregator == AnyOfLimiter::aggregator)
  702. result = std::make_shared<AnyOfLimiter>();
  703. else if(aggregator == NoneOfLimiter::aggregator)
  704. result = std::make_shared<NoneOfLimiter>();
  705. }
  706. if(!result)
  707. {
  708. // collapse for single limiter without explicit aggregate operator
  709. if(subLimiters.size() == 1)
  710. return parseLimiter(subLimiters[0]);
  711. // implicit aggregator must be allOf
  712. result = std::make_shared<AllOfLimiter>();
  713. offset = 0;
  714. }
  715. if(subLimiters.size() == offset)
  716. logMod->warn("Warning: empty sub-limiter list");
  717. for(int sl = offset; sl < subLimiters.size(); ++sl)
  718. result->add(parseLimiter(subLimiters[sl]));
  719. return result;
  720. }
  721. break;
  722. case JsonNode::JsonType::DATA_STRING: //pre-defined limiters
  723. return parseByMap(bonusLimiterMap, &limiter, "limiter type ");
  724. break;
  725. case JsonNode::JsonType::DATA_STRUCT: //customizable limiters
  726. {
  727. std::string limiterType = limiter["type"].String();
  728. const JsonVector & parameters = limiter["parameters"].Vector();
  729. if(limiterType == "CREATURE_TYPE_LIMITER")
  730. {
  731. std::shared_ptr<CCreatureTypeLimiter> creatureLimiter = std::make_shared<CCreatureTypeLimiter>();
  732. VLC->identifiers()->requestIdentifier("creature", parameters[0], [=](si32 creature)
  733. {
  734. creatureLimiter->setCreature(CreatureID(creature));
  735. });
  736. auto includeUpgrades = false;
  737. if(parameters.size() > 1)
  738. {
  739. bool success = true;
  740. includeUpgrades = parameters[1].TryBoolFromString(success);
  741. if(!success)
  742. logMod->error("Second parameter of '%s' limiter should be Bool", limiterType);
  743. }
  744. creatureLimiter->includeUpgrades = includeUpgrades;
  745. return creatureLimiter;
  746. }
  747. else if(limiterType == "HAS_ANOTHER_BONUS_LIMITER")
  748. {
  749. std::string anotherBonusType = parameters[0].String();
  750. auto it = bonusNameMap.find(anotherBonusType);
  751. if(it == bonusNameMap.end())
  752. {
  753. logMod->error("Error: invalid ability type %s.", anotherBonusType);
  754. }
  755. else
  756. {
  757. std::shared_ptr<HasAnotherBonusLimiter> bonusLimiter = std::make_shared<HasAnotherBonusLimiter>();
  758. bonusLimiter->type = it->second;
  759. auto findSource = [&](const JsonNode & parameter)
  760. {
  761. if(parameter.getType() == JsonNode::JsonType::DATA_STRUCT)
  762. {
  763. auto sourceIt = bonusSourceMap.find(parameter["type"].String());
  764. if(sourceIt != bonusSourceMap.end())
  765. {
  766. bonusLimiter->source = sourceIt->second;
  767. bonusLimiter->isSourceRelevant = true;
  768. if(!parameter["id"].isNull()) {
  769. loadBonusSourceInstance(bonusLimiter->sid, bonusLimiter->source, parameter["id"]);
  770. bonusLimiter->isSourceIDRelevant = true;
  771. }
  772. }
  773. }
  774. return false;
  775. };
  776. if(parameters.size() > 1)
  777. {
  778. if(findSource(parameters[1]) && parameters.size() == 2)
  779. return bonusLimiter;
  780. else
  781. {
  782. loadBonusSubtype(bonusLimiter->subtype, bonusLimiter->type, parameters[1]);
  783. bonusLimiter->isSubtypeRelevant = true;
  784. if(parameters.size() > 2)
  785. findSource(parameters[2]);
  786. }
  787. }
  788. return bonusLimiter;
  789. }
  790. }
  791. else if(limiterType == "CREATURE_ALIGNMENT_LIMITER")
  792. {
  793. int alignment = vstd::find_pos(GameConstants::ALIGNMENT_NAMES, parameters[0].String());
  794. if(alignment == -1)
  795. logMod->error("Error: invalid alignment %s.", parameters[0].String());
  796. else
  797. return std::make_shared<CreatureAlignmentLimiter>(static_cast<EAlignment>(alignment));
  798. }
  799. else if(limiterType == "FACTION_LIMITER" || limiterType == "CREATURE_FACTION_LIMITER") //Second name is deprecated, 1.2 compat
  800. {
  801. std::shared_ptr<FactionLimiter> factionLimiter = std::make_shared<FactionLimiter>();
  802. VLC->identifiers()->requestIdentifier("faction", parameters[0], [=](si32 faction)
  803. {
  804. factionLimiter->faction = FactionID(faction);
  805. });
  806. return factionLimiter;
  807. }
  808. else if(limiterType == "CREATURE_LEVEL_LIMITER")
  809. {
  810. auto levelLimiter = std::make_shared<CreatureLevelLimiter>();
  811. if(!parameters.empty()) //If parameters is empty, level limiter works as CREATURES_ONLY limiter
  812. {
  813. levelLimiter->minLevel = parameters[0].Integer();
  814. if(parameters[1].isNumber())
  815. levelLimiter->maxLevel = parameters[1].Integer();
  816. }
  817. return levelLimiter;
  818. }
  819. else if(limiterType == "CREATURE_TERRAIN_LIMITER")
  820. {
  821. std::shared_ptr<CreatureTerrainLimiter> terrainLimiter = std::make_shared<CreatureTerrainLimiter>();
  822. if(!parameters.empty())
  823. {
  824. VLC->identifiers()->requestIdentifier("terrain", parameters[0], [=](si32 terrain)
  825. {
  826. //TODO: support limiters
  827. //terrainLimiter->terrainType = terrain;
  828. });
  829. }
  830. return terrainLimiter;
  831. }
  832. else if(limiterType == "UNIT_ON_HEXES") {
  833. auto hexLimiter = std::make_shared<UnitOnHexLimiter>();
  834. if(!parameters.empty())
  835. {
  836. for (const auto & parameter: parameters){
  837. if(parameter.isNumber())
  838. hexLimiter->applicableHexes.insert(BattleHex(parameter.Integer()));
  839. }
  840. }
  841. return hexLimiter;
  842. }
  843. else
  844. {
  845. logMod->error("Error: invalid customizable limiter type %s.", limiterType);
  846. }
  847. }
  848. break;
  849. default:
  850. break;
  851. }
  852. return nullptr;
  853. }
  854. std::shared_ptr<Bonus> JsonUtils::parseBonus(const JsonNode &ability)
  855. {
  856. auto b = std::make_shared<Bonus>();
  857. if (!parseBonus(ability, b.get()))
  858. {
  859. // caller code can not handle this case and presumes that returned bonus is always valid
  860. logGlobal->error("Failed to parse bonus! Json config was %S ", ability.toJson());
  861. b->type = BonusType::NONE;
  862. return b;
  863. }
  864. return b;
  865. }
  866. std::shared_ptr<Bonus> JsonUtils::parseBuildingBonus(const JsonNode & ability, const FactionID & faction, const BuildingID & building, const std::string & description)
  867. {
  868. /* duration = BonusDuration::PERMANENT
  869. source = BonusSource::TOWN_STRUCTURE
  870. bonusType, val, subtype - get from json
  871. */
  872. auto b = std::make_shared<Bonus>(BonusDuration::PERMANENT, BonusType::NONE, BonusSource::TOWN_STRUCTURE, 0, BuildingTypeUniqueID(faction, building), description);
  873. if(!parseBonus(ability, b.get()))
  874. return nullptr;
  875. return b;
  876. }
  877. static BonusParams convertDeprecatedBonus(const JsonNode &ability)
  878. {
  879. if(vstd::contains(deprecatedBonusSet, ability["type"].String()))
  880. {
  881. logMod->warn("There is deprecated bonus found:\n%s\nTrying to convert...", ability.toJson());
  882. auto params = BonusParams(ability["type"].String(),
  883. ability["subtype"].isString() ? ability["subtype"].String() : "",
  884. ability["subtype"].isNumber() ? ability["subtype"].Integer() : -1);
  885. if(params.isConverted)
  886. {
  887. if(ability["type"].String() == "SECONDARY_SKILL_PREMY" && bonusValueMap.find(ability["valueType"].String())->second == BonusValueType::PERCENT_TO_BASE) //assume secondary skill special
  888. {
  889. params.valueType = BonusValueType::PERCENT_TO_TARGET_TYPE;
  890. params.targetType = BonusSource::SECONDARY_SKILL;
  891. }
  892. logMod->warn("Please, use this bonus:\n%s\nConverted successfully!", params.toJson().toJson());
  893. return params;
  894. }
  895. else
  896. logMod->error("Cannot convert bonus!\n%s", ability.toJson());
  897. }
  898. BonusParams ret;
  899. ret.isConverted = false;
  900. return ret;
  901. }
  902. static TUpdaterPtr parseUpdater(const JsonNode & updaterJson)
  903. {
  904. switch(updaterJson.getType())
  905. {
  906. case JsonNode::JsonType::DATA_STRING:
  907. return parseByMap(bonusUpdaterMap, &updaterJson, "updater type ");
  908. break;
  909. case JsonNode::JsonType::DATA_STRUCT:
  910. if(updaterJson["type"].String() == "GROWS_WITH_LEVEL")
  911. {
  912. std::shared_ptr<GrowsWithLevelUpdater> updater = std::make_shared<GrowsWithLevelUpdater>();
  913. const JsonVector param = updaterJson["parameters"].Vector();
  914. updater->valPer20 = static_cast<int>(param[0].Integer());
  915. if(param.size() > 1)
  916. updater->stepSize = static_cast<int>(param[1].Integer());
  917. return updater;
  918. }
  919. else if (updaterJson["type"].String() == "ARMY_MOVEMENT")
  920. {
  921. std::shared_ptr<ArmyMovementUpdater> updater = std::make_shared<ArmyMovementUpdater>();
  922. if(updaterJson["parameters"].isVector())
  923. {
  924. const auto & param = updaterJson["parameters"].Vector();
  925. if(param.size() < 4)
  926. logMod->warn("Invalid ARMY_MOVEMENT parameters, using default!");
  927. else
  928. {
  929. updater->base = static_cast<si32>(param.at(0).Integer());
  930. updater->divider = static_cast<si32>(param.at(1).Integer());
  931. updater->multiplier = static_cast<si32>(param.at(2).Integer());
  932. updater->max = static_cast<si32>(param.at(3).Integer());
  933. }
  934. return updater;
  935. }
  936. }
  937. else
  938. logMod->warn("Unknown updater type \"%s\"", updaterJson["type"].String());
  939. break;
  940. }
  941. return nullptr;
  942. }
  943. bool JsonUtils::parseBonus(const JsonNode &ability, Bonus *b)
  944. {
  945. const JsonNode * value = nullptr;
  946. std::string type = ability["type"].String();
  947. auto it = bonusNameMap.find(type);
  948. auto params = std::make_unique<BonusParams>(false);
  949. if (it == bonusNameMap.end())
  950. {
  951. params = std::make_unique<BonusParams>(convertDeprecatedBonus(ability));
  952. if(!params->isConverted)
  953. {
  954. logMod->error("Error: invalid ability type %s.", type);
  955. return false;
  956. }
  957. b->type = params->type;
  958. b->val = params->val.value_or(0);
  959. b->valType = params->valueType.value_or(BonusValueType::ADDITIVE_VALUE);
  960. if(params->targetType)
  961. b->targetSourceType = params->targetType.value();
  962. }
  963. else
  964. b->type = it->second;
  965. loadBonusSubtype(b->subtype, b->type, params->isConverted ? params->toJson()["subtype"] : ability["subtype"]);
  966. if(!params->isConverted)
  967. {
  968. b->val = static_cast<si32>(ability["val"].Float());
  969. value = &ability["valueType"];
  970. if (!value->isNull())
  971. b->valType = static_cast<BonusValueType>(parseByMapN(bonusValueMap, value, "value type "));
  972. }
  973. b->stacking = ability["stacking"].String();
  974. resolveAddInfo(b->additionalInfo, ability);
  975. b->turnsRemain = static_cast<si32>(ability["turns"].Float());
  976. if(!ability["description"].isNull())
  977. {
  978. if (ability["description"].isString())
  979. b->description = ability["description"].String();
  980. if (ability["description"].isNumber())
  981. b->description = VLC->generaltexth->translate("core.arraytxt", ability["description"].Integer());
  982. }
  983. value = &ability["effectRange"];
  984. if (!value->isNull())
  985. b->effectRange = static_cast<BonusLimitEffect>(parseByMapN(bonusLimitEffect, value, "effect range "));
  986. value = &ability["duration"];
  987. if (!value->isNull())
  988. {
  989. switch (value->getType())
  990. {
  991. case JsonNode::JsonType::DATA_STRING:
  992. b->duration = parseByMap(bonusDurationMap, value, "duration type ");
  993. break;
  994. case JsonNode::JsonType::DATA_VECTOR:
  995. {
  996. BonusDuration::Type dur = 0;
  997. for (const JsonNode & d : value->Vector())
  998. dur |= parseByMapN(bonusDurationMap, &d, "duration type ");
  999. b->duration = dur;
  1000. }
  1001. break;
  1002. default:
  1003. logMod->error("Error! Wrong bonus duration format.");
  1004. }
  1005. }
  1006. value = &ability["sourceType"];
  1007. if (!value->isNull())
  1008. b->source = static_cast<BonusSource>(parseByMap(bonusSourceMap, value, "source type "));
  1009. if (!ability["sourceID"].isNull())
  1010. loadBonusSourceInstance(b->sid, b->source, ability["sourceID"]);
  1011. value = &ability["targetSourceType"];
  1012. if (!value->isNull())
  1013. b->targetSourceType = static_cast<BonusSource>(parseByMap(bonusSourceMap, value, "target type "));
  1014. value = &ability["limiters"];
  1015. if (!value->isNull())
  1016. b->limiter = parseLimiter(*value);
  1017. value = &ability["propagator"];
  1018. if (!value->isNull())
  1019. {
  1020. //ALL_CREATURES old propagator compatibility
  1021. if(value->String() == "ALL_CREATURES")
  1022. {
  1023. logMod->warn("ALL_CREATURES propagator is deprecated. Use GLOBAL_EFFECT propagator with CREATURES_ONLY limiter");
  1024. b->addLimiter(std::make_shared<CreatureLevelLimiter>());
  1025. b->propagator = bonusPropagatorMap.at("GLOBAL_EFFECT");
  1026. }
  1027. else
  1028. b->propagator = parseByMap(bonusPropagatorMap, value, "propagator type ");
  1029. }
  1030. value = &ability["updater"];
  1031. if(!value->isNull())
  1032. b->addUpdater(parseUpdater(*value));
  1033. value = &ability["propagationUpdater"];
  1034. if(!value->isNull())
  1035. b->propagationUpdater = parseUpdater(*value);
  1036. return true;
  1037. }
  1038. CSelector JsonUtils::parseSelector(const JsonNode & ability)
  1039. {
  1040. CSelector ret = Selector::all;
  1041. // Recursive parsers for anyOf, allOf, noneOf
  1042. const auto * value = &ability["allOf"];
  1043. if(value->isVector())
  1044. {
  1045. for(const auto & andN : value->Vector())
  1046. ret = ret.And(parseSelector(andN));
  1047. }
  1048. value = &ability["anyOf"];
  1049. if(value->isVector())
  1050. {
  1051. CSelector base = Selector::none;
  1052. for(const auto & andN : value->Vector())
  1053. base = base.Or(parseSelector(andN));
  1054. ret = ret.And(base);
  1055. }
  1056. value = &ability["noneOf"];
  1057. if(value->isVector())
  1058. {
  1059. CSelector base = Selector::none;
  1060. for(const auto & andN : value->Vector())
  1061. base = base.Or(parseSelector(andN));
  1062. ret = ret.And(base.Not());
  1063. }
  1064. BonusType type = BonusType::NONE;
  1065. // Actual selector parser
  1066. value = &ability["type"];
  1067. if(value->isString())
  1068. {
  1069. auto it = bonusNameMap.find(value->String());
  1070. if(it != bonusNameMap.end())
  1071. {
  1072. type = it->second;
  1073. ret = ret.And(Selector::type()(it->second));
  1074. }
  1075. }
  1076. value = &ability["subtype"];
  1077. if(!value->isNull() && type != BonusType::NONE)
  1078. {
  1079. BonusSubtypeID subtype;
  1080. loadBonusSubtype(subtype, type, ability);
  1081. ret = ret.And(Selector::subtype()(subtype));
  1082. }
  1083. value = &ability["sourceType"];
  1084. std::optional<BonusSource> src = std::nullopt; //Fixes for GCC false maybe-uninitialized
  1085. std::optional<BonusSourceID> id = std::nullopt;
  1086. if(value->isString())
  1087. {
  1088. auto it = bonusSourceMap.find(value->String());
  1089. if(it != bonusSourceMap.end())
  1090. src = it->second;
  1091. }
  1092. value = &ability["sourceID"];
  1093. if(!value->isNull() && src.has_value())
  1094. {
  1095. loadBonusSourceInstance(*id, *src, ability);
  1096. }
  1097. if(src && id)
  1098. ret = ret.And(Selector::source(*src, *id));
  1099. else if(src)
  1100. ret = ret.And(Selector::sourceTypeSel(*src));
  1101. value = &ability["targetSourceType"];
  1102. if(value->isString())
  1103. {
  1104. auto it = bonusSourceMap.find(value->String());
  1105. if(it != bonusSourceMap.end())
  1106. ret = ret.And(Selector::targetSourceType()(it->second));
  1107. }
  1108. value = &ability["valueType"];
  1109. if(value->isString())
  1110. {
  1111. auto it = bonusValueMap.find(value->String());
  1112. if(it != bonusValueMap.end())
  1113. ret = ret.And(Selector::valueType(it->second));
  1114. }
  1115. CAddInfo info;
  1116. value = &ability["addInfo"];
  1117. if(!value->isNull())
  1118. {
  1119. resolveAddInfo(info, ability["addInfo"]);
  1120. ret = ret.And(Selector::info()(info));
  1121. }
  1122. value = &ability["effectRange"];
  1123. if(value->isString())
  1124. {
  1125. auto it = bonusLimitEffect.find(value->String());
  1126. if(it != bonusLimitEffect.end())
  1127. ret = ret.And(Selector::effectRange()(it->second));
  1128. }
  1129. value = &ability["lastsTurns"];
  1130. if(value->isNumber())
  1131. ret = ret.And(Selector::turns(value->Integer()));
  1132. value = &ability["lastsDays"];
  1133. if(value->isNumber())
  1134. ret = ret.And(Selector::days(value->Integer()));
  1135. return ret;
  1136. }
  1137. //returns first Key with value equal to given one
  1138. template<class Key, class Val>
  1139. Key reverseMapFirst(const Val & val, const std::map<Key, Val> & map)
  1140. {
  1141. for(auto it : map)
  1142. {
  1143. if(it.second == val)
  1144. {
  1145. return it.first;
  1146. }
  1147. }
  1148. assert(0);
  1149. return "";
  1150. }
  1151. static JsonNode getDefaultValue(const JsonNode & schema, std::string fieldName)
  1152. {
  1153. const JsonNode & fieldProps = schema["properties"][fieldName];
  1154. #if defined(VCMI_IOS)
  1155. if (!fieldProps["defaultIOS"].isNull())
  1156. return fieldProps["defaultIOS"];
  1157. #elif defined(VCMI_ANDROID)
  1158. if (!fieldProps["defaultAndroid"].isNull())
  1159. return fieldProps["defaultAndroid"];
  1160. #elif !defined(VCMI_MOBILE)
  1161. if (!fieldProps["defaultDesktop"].isNull())
  1162. return fieldProps["defaultDesktop"];
  1163. #endif
  1164. return fieldProps["default"];
  1165. }
  1166. static void eraseOptionalNodes(JsonNode & node, const JsonNode & schema)
  1167. {
  1168. assert(schema["type"].String() == "object");
  1169. std::set<std::string> foundEntries;
  1170. for(const auto & entry : schema["required"].Vector())
  1171. foundEntries.insert(entry.String());
  1172. vstd::erase_if(node.Struct(), [&](const auto & node){
  1173. return !vstd::contains(foundEntries, node.first);
  1174. });
  1175. }
  1176. static void minimizeNode(JsonNode & node, const JsonNode & schema)
  1177. {
  1178. if (schema["type"].String() != "object")
  1179. return;
  1180. for(const auto & entry : schema["required"].Vector())
  1181. {
  1182. const std::string & name = entry.String();
  1183. minimizeNode(node[name], schema["properties"][name]);
  1184. if (vstd::contains(node.Struct(), name) && node[name] == getDefaultValue(schema, name))
  1185. node.Struct().erase(name);
  1186. }
  1187. eraseOptionalNodes(node, schema);
  1188. }
  1189. static void maximizeNode(JsonNode & node, const JsonNode & schema)
  1190. {
  1191. // "required" entry can only be found in object/struct
  1192. if (schema["type"].String() != "object")
  1193. return;
  1194. // check all required entries that have default version
  1195. for(const auto & entry : schema["required"].Vector())
  1196. {
  1197. const std::string & name = entry.String();
  1198. if (node[name].isNull() && !getDefaultValue(schema, name).isNull())
  1199. node[name] = getDefaultValue(schema, name);
  1200. maximizeNode(node[name], schema["properties"][name]);
  1201. }
  1202. eraseOptionalNodes(node, schema);
  1203. }
  1204. void JsonUtils::minimize(JsonNode & node, const std::string & schemaName)
  1205. {
  1206. minimizeNode(node, getSchema(schemaName));
  1207. }
  1208. void JsonUtils::maximize(JsonNode & node, const std::string & schemaName)
  1209. {
  1210. maximizeNode(node, getSchema(schemaName));
  1211. }
  1212. bool JsonUtils::validate(const JsonNode & node, const std::string & schemaName, const std::string & dataName)
  1213. {
  1214. std::string log = Validation::check(schemaName, node);
  1215. if (!log.empty())
  1216. {
  1217. logMod->warn("Data in %s is invalid!", dataName);
  1218. logMod->warn(log);
  1219. logMod->trace("%s json: %s", dataName, node.toJson(true));
  1220. }
  1221. return log.empty();
  1222. }
  1223. const JsonNode & getSchemaByName(const std::string & name)
  1224. {
  1225. // cached schemas to avoid loading json data multiple times
  1226. static std::map<std::string, JsonNode> loadedSchemas;
  1227. if (vstd::contains(loadedSchemas, name))
  1228. return loadedSchemas[name];
  1229. auto filename = JsonPath::builtin("config/schemas/" + name);
  1230. if (CResourceHandler::get()->existsResource(filename))
  1231. {
  1232. loadedSchemas[name] = JsonNode(filename);
  1233. return loadedSchemas[name];
  1234. }
  1235. logMod->error("Error: missing schema with name %s!", name);
  1236. assert(0);
  1237. return nullNode;
  1238. }
  1239. const JsonNode & JsonUtils::getSchema(const std::string & URI)
  1240. {
  1241. size_t posColon = URI.find(':');
  1242. size_t posHash = URI.find('#');
  1243. std::string filename;
  1244. if(posColon == std::string::npos)
  1245. {
  1246. filename = URI.substr(0, posHash);
  1247. }
  1248. else
  1249. {
  1250. std::string protocolName = URI.substr(0, posColon);
  1251. filename = URI.substr(posColon + 1, posHash - posColon - 1) + ".json";
  1252. if(protocolName != "vcmi")
  1253. {
  1254. logMod->error("Error: unsupported URI protocol for schema: %s", URI);
  1255. return nullNode;
  1256. }
  1257. }
  1258. // check if json pointer if present (section after hash in string)
  1259. if(posHash == std::string::npos || posHash == URI.size() - 1)
  1260. {
  1261. auto const & result = getSchemaByName(filename);
  1262. if (result.isNull())
  1263. logMod->error("Error: missing schema %s", URI);
  1264. return result;
  1265. }
  1266. else
  1267. {
  1268. auto const & result = getSchemaByName(filename).resolvePointer(URI.substr(posHash + 1));
  1269. if (result.isNull())
  1270. logMod->error("Error: missing schema %s", URI);
  1271. return result;
  1272. }
  1273. }
  1274. void JsonUtils::merge(JsonNode & dest, JsonNode & source, bool ignoreOverride, bool copyMeta)
  1275. {
  1276. if (dest.getType() == JsonNode::JsonType::DATA_NULL)
  1277. {
  1278. std::swap(dest, source);
  1279. return;
  1280. }
  1281. switch (source.getType())
  1282. {
  1283. case JsonNode::JsonType::DATA_NULL:
  1284. {
  1285. dest.clear();
  1286. break;
  1287. }
  1288. case JsonNode::JsonType::DATA_BOOL:
  1289. case JsonNode::JsonType::DATA_FLOAT:
  1290. case JsonNode::JsonType::DATA_INTEGER:
  1291. case JsonNode::JsonType::DATA_STRING:
  1292. case JsonNode::JsonType::DATA_VECTOR:
  1293. {
  1294. std::swap(dest, source);
  1295. break;
  1296. }
  1297. case JsonNode::JsonType::DATA_STRUCT:
  1298. {
  1299. if(!ignoreOverride && vstd::contains(source.flags, "override"))
  1300. {
  1301. std::swap(dest, source);
  1302. }
  1303. else
  1304. {
  1305. if (copyMeta)
  1306. dest.meta = source.meta;
  1307. //recursively merge all entries from struct
  1308. for(auto & node : source.Struct())
  1309. merge(dest[node.first], node.second, ignoreOverride);
  1310. }
  1311. }
  1312. }
  1313. }
  1314. void JsonUtils::mergeCopy(JsonNode & dest, JsonNode source, bool ignoreOverride, bool copyMeta)
  1315. {
  1316. // uses copy created in stack to safely merge two nodes
  1317. merge(dest, source, ignoreOverride, copyMeta);
  1318. }
  1319. void JsonUtils::inherit(JsonNode & descendant, const JsonNode & base)
  1320. {
  1321. JsonNode inheritedNode(base);
  1322. merge(inheritedNode, descendant, true, true);
  1323. std::swap(descendant, inheritedNode);
  1324. }
  1325. JsonNode JsonUtils::intersect(const std::vector<JsonNode> & nodes, bool pruneEmpty)
  1326. {
  1327. if(nodes.empty())
  1328. return nullNode;
  1329. JsonNode result = nodes[0];
  1330. for(int i = 1; i < nodes.size(); i++)
  1331. {
  1332. if(result.isNull())
  1333. break;
  1334. result = JsonUtils::intersect(result, nodes[i], pruneEmpty);
  1335. }
  1336. return result;
  1337. }
  1338. JsonNode JsonUtils::intersect(const JsonNode & a, const JsonNode & b, bool pruneEmpty)
  1339. {
  1340. if(a.getType() == JsonNode::JsonType::DATA_STRUCT && b.getType() == JsonNode::JsonType::DATA_STRUCT)
  1341. {
  1342. // intersect individual properties
  1343. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  1344. for(const auto & property : a.Struct())
  1345. {
  1346. if(vstd::contains(b.Struct(), property.first))
  1347. {
  1348. JsonNode propertyIntersect = JsonUtils::intersect(property.second, b.Struct().find(property.first)->second);
  1349. if(pruneEmpty && !propertyIntersect.containsBaseData())
  1350. continue;
  1351. result[property.first] = propertyIntersect;
  1352. }
  1353. }
  1354. return result;
  1355. }
  1356. else
  1357. {
  1358. // not a struct - same or different, no middle ground
  1359. if(a == b)
  1360. return a;
  1361. }
  1362. return nullNode;
  1363. }
  1364. JsonNode JsonUtils::difference(const JsonNode & node, const JsonNode & base)
  1365. {
  1366. auto addsInfo = [](JsonNode diff) -> bool
  1367. {
  1368. switch(diff.getType())
  1369. {
  1370. case JsonNode::JsonType::DATA_NULL:
  1371. return false;
  1372. case JsonNode::JsonType::DATA_STRUCT:
  1373. return !diff.Struct().empty();
  1374. default:
  1375. return true;
  1376. }
  1377. };
  1378. if(node.getType() == JsonNode::JsonType::DATA_STRUCT && base.getType() == JsonNode::JsonType::DATA_STRUCT)
  1379. {
  1380. // subtract individual properties
  1381. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  1382. for(const auto & property : node.Struct())
  1383. {
  1384. if(vstd::contains(base.Struct(), property.first))
  1385. {
  1386. const JsonNode propertyDifference = JsonUtils::difference(property.second, base.Struct().find(property.first)->second);
  1387. if(addsInfo(propertyDifference))
  1388. result[property.first] = propertyDifference;
  1389. }
  1390. else
  1391. {
  1392. result[property.first] = property.second;
  1393. }
  1394. }
  1395. return result;
  1396. }
  1397. else
  1398. {
  1399. if(node == base)
  1400. return nullNode;
  1401. }
  1402. return node;
  1403. }
  1404. JsonNode JsonUtils::assembleFromFiles(const std::vector<std::string> & files)
  1405. {
  1406. bool isValid = false;
  1407. return assembleFromFiles(files, isValid);
  1408. }
  1409. JsonNode JsonUtils::assembleFromFiles(const std::vector<std::string> & files, bool & isValid)
  1410. {
  1411. isValid = true;
  1412. JsonNode result;
  1413. for(const auto & file : files)
  1414. {
  1415. bool isValidFile = false;
  1416. JsonNode section(JsonPath::builtinTODO(file), isValidFile);
  1417. merge(result, section);
  1418. isValid |= isValidFile;
  1419. }
  1420. return result;
  1421. }
  1422. JsonNode JsonUtils::assembleFromFiles(const std::string & filename)
  1423. {
  1424. JsonNode result;
  1425. JsonPath resID = JsonPath::builtinTODO(filename);
  1426. for(auto & loader : CResourceHandler::get()->getResourcesWithName(resID))
  1427. {
  1428. // FIXME: some way to make this code more readable
  1429. auto stream = loader->load(resID);
  1430. std::unique_ptr<ui8[]> textData(new ui8[stream->getSize()]);
  1431. stream->read(textData.get(), stream->getSize());
  1432. JsonNode section(reinterpret_cast<char *>(textData.get()), stream->getSize());
  1433. merge(result, section);
  1434. }
  1435. return result;
  1436. }
  1437. DLL_LINKAGE JsonNode JsonUtils::boolNode(bool value)
  1438. {
  1439. JsonNode node;
  1440. node.Bool() = value;
  1441. return node;
  1442. }
  1443. DLL_LINKAGE JsonNode JsonUtils::floatNode(double value)
  1444. {
  1445. JsonNode node;
  1446. node.Float() = value;
  1447. return node;
  1448. }
  1449. DLL_LINKAGE JsonNode JsonUtils::stringNode(const std::string & value)
  1450. {
  1451. JsonNode node;
  1452. node.String() = value;
  1453. return node;
  1454. }
  1455. DLL_LINKAGE JsonNode JsonUtils::intNode(si64 value)
  1456. {
  1457. JsonNode node;
  1458. node.Integer() = value;
  1459. return node;
  1460. }
  1461. VCMI_LIB_NAMESPACE_END