JsonNode.cpp 39 KB

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