2
0

JsonNode.cpp 42 KB

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