JsonNode.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149
  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 "HeroBonus.h"
  14. #include "filesystem/Filesystem.h"
  15. #include "VCMI_Lib.h" //for identifier resolution
  16. #include "CModHandler.h"
  17. #include "CGeneralTextHandler.h"
  18. #include "JsonDetail.h"
  19. #include "StringConstants.h"
  20. using namespace JsonDetail;
  21. class LibClasses;
  22. class CModHandler;
  23. static const JsonNode nullNode;
  24. JsonNode::JsonNode(JsonType Type):
  25. type(JsonType::DATA_NULL)
  26. {
  27. setType(Type);
  28. }
  29. JsonNode::JsonNode(const char *data, size_t datasize):
  30. type(JsonType::DATA_NULL)
  31. {
  32. JsonParser parser(data, datasize);
  33. *this = parser.parse("<unknown>");
  34. }
  35. JsonNode::JsonNode(ResourceID && fileURI):
  36. type(JsonType::DATA_NULL)
  37. {
  38. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  39. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  40. *this = parser.parse(fileURI.getName());
  41. }
  42. JsonNode::JsonNode(const ResourceID & fileURI):
  43. type(JsonType::DATA_NULL)
  44. {
  45. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  46. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  47. *this = parser.parse(fileURI.getName());
  48. }
  49. JsonNode::JsonNode(ResourceID && fileURI, bool &isValidSyntax):
  50. type(JsonType::DATA_NULL)
  51. {
  52. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  53. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  54. *this = parser.parse(fileURI.getName());
  55. isValidSyntax = parser.isValid();
  56. }
  57. JsonNode::JsonNode(const JsonNode &copy):
  58. type(JsonType::DATA_NULL),
  59. meta(copy.meta),
  60. flags(copy.flags)
  61. {
  62. setType(copy.getType());
  63. switch(type)
  64. {
  65. break; case JsonType::DATA_NULL:
  66. break; case JsonType::DATA_BOOL: Bool() = copy.Bool();
  67. break; case JsonType::DATA_FLOAT: Float() = copy.Float();
  68. break; case JsonType::DATA_STRING: String() = copy.String();
  69. break; case JsonType::DATA_VECTOR: Vector() = copy.Vector();
  70. break; case JsonType::DATA_STRUCT: Struct() = copy.Struct();
  71. break; case JsonType::DATA_INTEGER:Integer() = copy.Integer();
  72. }
  73. }
  74. JsonNode::~JsonNode()
  75. {
  76. setType(JsonType::DATA_NULL);
  77. }
  78. void JsonNode::swap(JsonNode &b)
  79. {
  80. using std::swap;
  81. swap(meta, b.meta);
  82. swap(data, b.data);
  83. swap(type, b.type);
  84. swap(flags, b.flags);
  85. }
  86. JsonNode & JsonNode::operator =(JsonNode node)
  87. {
  88. swap(node);
  89. return *this;
  90. }
  91. bool JsonNode::operator == (const JsonNode &other) const
  92. {
  93. if (getType() == other.getType())
  94. {
  95. switch(type)
  96. {
  97. case JsonType::DATA_NULL: return true;
  98. case JsonType::DATA_BOOL: return Bool() == other.Bool();
  99. case JsonType::DATA_FLOAT: return Float() == other.Float();
  100. case JsonType::DATA_STRING: return String() == other.String();
  101. case JsonType::DATA_VECTOR: return Vector() == other.Vector();
  102. case JsonType::DATA_STRUCT: return Struct() == other.Struct();
  103. case JsonType::DATA_INTEGER:return Integer()== other.Integer();
  104. }
  105. }
  106. return false;
  107. }
  108. bool JsonNode::operator != (const JsonNode &other) const
  109. {
  110. return !(*this == other);
  111. }
  112. JsonNode::JsonType JsonNode::getType() const
  113. {
  114. return type;
  115. }
  116. void JsonNode::setMeta(std::string metadata, bool recursive)
  117. {
  118. meta = metadata;
  119. if (recursive)
  120. {
  121. switch (type)
  122. {
  123. break; case JsonType::DATA_VECTOR:
  124. {
  125. for(auto & node : Vector())
  126. {
  127. node.setMeta(metadata);
  128. }
  129. }
  130. break; case JsonType::DATA_STRUCT:
  131. {
  132. for(auto & node : Struct())
  133. {
  134. node.second.setMeta(metadata);
  135. }
  136. }
  137. }
  138. }
  139. }
  140. void JsonNode::setType(JsonType Type)
  141. {
  142. if (type == Type)
  143. return;
  144. //float<->int conversion
  145. if(type == JsonType::DATA_FLOAT && Type == JsonType::DATA_INTEGER)
  146. {
  147. si64 converted = data.Float;
  148. type = Type;
  149. data.Integer = converted;
  150. return;
  151. }
  152. else if(type == JsonType::DATA_INTEGER && Type == JsonType::DATA_FLOAT)
  153. {
  154. double converted = data.Integer;
  155. type = Type;
  156. data.Float = converted;
  157. return;
  158. }
  159. //Reset node to nullptr
  160. if (Type != JsonType::DATA_NULL)
  161. setType(JsonType::DATA_NULL);
  162. switch (type)
  163. {
  164. break; case JsonType::DATA_STRING: delete data.String;
  165. break; case JsonType::DATA_VECTOR: delete data.Vector;
  166. break; case JsonType::DATA_STRUCT: delete data.Struct;
  167. break; default:
  168. break;
  169. }
  170. //Set new node type
  171. type = Type;
  172. switch(type)
  173. {
  174. break; case JsonType::DATA_NULL:
  175. break; case JsonType::DATA_BOOL: data.Bool = false;
  176. break; case JsonType::DATA_FLOAT: data.Float = 0;
  177. break; case JsonType::DATA_STRING: data.String = new std::string();
  178. break; case JsonType::DATA_VECTOR: data.Vector = new JsonVector();
  179. break; case JsonType::DATA_STRUCT: data.Struct = new JsonMap();
  180. break; case JsonType::DATA_INTEGER: data.Integer = 0;
  181. }
  182. }
  183. bool JsonNode::isNull() const
  184. {
  185. return type == JsonType::DATA_NULL;
  186. }
  187. bool JsonNode::isNumber() const
  188. {
  189. return type == JsonType::DATA_INTEGER || type == JsonType::DATA_FLOAT;
  190. }
  191. bool JsonNode::containsBaseData() const
  192. {
  193. switch(type)
  194. {
  195. case JsonType::DATA_NULL:
  196. return false;
  197. case JsonType::DATA_STRUCT:
  198. for(auto elem : *data.Struct)
  199. {
  200. if(elem.second.containsBaseData())
  201. return true;
  202. }
  203. return false;
  204. default:
  205. //other types (including vector) cannot be extended via merge
  206. return true;
  207. }
  208. }
  209. bool JsonNode::isCompact() const
  210. {
  211. switch(type)
  212. {
  213. case JsonType::DATA_VECTOR:
  214. for(JsonNode & elem : *data.Vector)
  215. {
  216. if(!elem.isCompact())
  217. return false;
  218. }
  219. return true;
  220. case JsonType::DATA_STRUCT:
  221. {
  222. int propertyCount = data.Struct->size();
  223. if(propertyCount == 0)
  224. return true;
  225. else if(propertyCount == 1)
  226. return data.Struct->begin()->second.isCompact();
  227. }
  228. return false;
  229. default:
  230. return true;
  231. }
  232. }
  233. void JsonNode::clear()
  234. {
  235. setType(JsonType::DATA_NULL);
  236. }
  237. bool & JsonNode::Bool()
  238. {
  239. setType(JsonType::DATA_BOOL);
  240. return data.Bool;
  241. }
  242. double & JsonNode::Float()
  243. {
  244. setType(JsonType::DATA_FLOAT);
  245. return data.Float;
  246. }
  247. si64 & JsonNode::Integer()
  248. {
  249. setType(JsonType::DATA_INTEGER);
  250. return data.Integer;
  251. }
  252. std::string & JsonNode::String()
  253. {
  254. setType(JsonType::DATA_STRING);
  255. return *data.String;
  256. }
  257. JsonVector & JsonNode::Vector()
  258. {
  259. setType(JsonType::DATA_VECTOR);
  260. return *data.Vector;
  261. }
  262. JsonMap & JsonNode::Struct()
  263. {
  264. setType(JsonType::DATA_STRUCT);
  265. return *data.Struct;
  266. }
  267. const bool boolDefault = false;
  268. bool JsonNode::Bool() const
  269. {
  270. if (type == JsonType::DATA_NULL)
  271. return boolDefault;
  272. assert(type == JsonType::DATA_BOOL);
  273. return data.Bool;
  274. }
  275. const double floatDefault = 0;
  276. double JsonNode::Float() const
  277. {
  278. if(type == JsonType::DATA_NULL)
  279. return floatDefault;
  280. else if(type == JsonType::DATA_INTEGER)
  281. return data.Integer;
  282. assert(type == JsonType::DATA_FLOAT);
  283. return data.Float;
  284. }
  285. const si64 integetDefault = 0;
  286. si64 JsonNode::Integer() const
  287. {
  288. if(type == JsonType::DATA_NULL)
  289. return integetDefault;
  290. else if(type == JsonType::DATA_FLOAT)
  291. return data.Float;
  292. assert(type == JsonType::DATA_INTEGER);
  293. return data.Integer;
  294. }
  295. const std::string stringDefault = std::string();
  296. const std::string & JsonNode::String() const
  297. {
  298. if (type == JsonType::DATA_NULL)
  299. return stringDefault;
  300. assert(type == JsonType::DATA_STRING);
  301. return *data.String;
  302. }
  303. const JsonVector vectorDefault = JsonVector();
  304. const JsonVector & JsonNode::Vector() const
  305. {
  306. if (type == JsonType::DATA_NULL)
  307. return vectorDefault;
  308. assert(type == JsonType::DATA_VECTOR);
  309. return *data.Vector;
  310. }
  311. const JsonMap mapDefault = JsonMap();
  312. const JsonMap & JsonNode::Struct() const
  313. {
  314. if (type == JsonType::DATA_NULL)
  315. return mapDefault;
  316. assert(type == JsonType::DATA_STRUCT);
  317. return *data.Struct;
  318. }
  319. JsonNode & JsonNode::operator[](std::string child)
  320. {
  321. return Struct()[child];
  322. }
  323. const JsonNode & JsonNode::operator[](std::string child) const
  324. {
  325. auto it = Struct().find(child);
  326. if (it != Struct().end())
  327. return it->second;
  328. return nullNode;
  329. }
  330. // to avoid duplicating const and non-const code
  331. template<typename Node>
  332. Node & resolvePointer(Node & in, const std::string & pointer)
  333. {
  334. if (pointer.empty())
  335. return in;
  336. assert(pointer[0] == '/');
  337. size_t splitPos = pointer.find('/', 1);
  338. std::string entry = pointer.substr(1, splitPos -1);
  339. std::string remainer = splitPos == std::string::npos ? "" : pointer.substr(splitPos);
  340. if (in.getType() == JsonNode::JsonType::DATA_VECTOR)
  341. {
  342. if (entry.find_first_not_of("0123456789") != std::string::npos) // non-numbers in string
  343. throw std::runtime_error("Invalid Json pointer");
  344. if (entry.size() > 1 && entry[0] == '0') // leading zeros are not allowed
  345. throw std::runtime_error("Invalid Json pointer");
  346. size_t index = boost::lexical_cast<size_t>(entry);
  347. if (in.Vector().size() > index)
  348. return in.Vector()[index].resolvePointer(remainer);
  349. }
  350. return in[entry].resolvePointer(remainer);
  351. }
  352. const JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer) const
  353. {
  354. return ::resolvePointer(*this, jsonPointer);
  355. }
  356. JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer)
  357. {
  358. return ::resolvePointer(*this, jsonPointer);
  359. }
  360. std::string JsonNode::toJson(bool compact) const
  361. {
  362. std::ostringstream out;
  363. JsonWriter writer(out, compact);
  364. writer.writeNode(*this);
  365. return out.str();
  366. }
  367. ///JsonUtils
  368. void JsonUtils::parseTypedBonusShort(const JsonVector& source, std::shared_ptr<Bonus> dest)
  369. {
  370. dest->val = source[1].Float();
  371. resolveIdentifier(source[2],dest->subtype);
  372. dest->additionalInfo = source[3].Float();
  373. dest->duration = Bonus::PERMANENT; //TODO: handle flags (as integer)
  374. dest->turnsRemain = 0;
  375. }
  376. std::shared_ptr<Bonus> JsonUtils::parseBonus (const JsonVector &ability_vec) //TODO: merge with AddAbility, create universal parser for all bonus properties
  377. {
  378. auto b = std::make_shared<Bonus>();
  379. std::string type = ability_vec[0].String();
  380. auto it = bonusNameMap.find(type);
  381. if (it == bonusNameMap.end())
  382. {
  383. logMod->error("Error: invalid ability type %s.", type);
  384. return b;
  385. }
  386. b->type = it->second;
  387. parseTypedBonusShort(ability_vec, b);
  388. return b;
  389. }
  390. template <typename T>
  391. const T & parseByMap(const std::map<std::string, T> & map, const JsonNode * val, std::string err)
  392. {
  393. static T defaultValue = T();
  394. if (!val->isNull())
  395. {
  396. std::string type = val->String();
  397. auto it = map.find(type);
  398. if (it == map.end())
  399. {
  400. logMod->error("Error: invalid %s%s.", err, type);
  401. return defaultValue;
  402. }
  403. else
  404. {
  405. return it->second;
  406. }
  407. }
  408. else
  409. return defaultValue;
  410. }
  411. void JsonUtils::resolveIdentifier(si32 &var, const JsonNode &node, std::string name)
  412. {
  413. const JsonNode &value = node[name];
  414. if (!value.isNull())
  415. {
  416. switch (value.getType())
  417. {
  418. case JsonNode::JsonType::DATA_INTEGER:
  419. var = value.Integer();
  420. break;
  421. case JsonNode::JsonType::DATA_FLOAT:
  422. var = value.Float();
  423. break;
  424. case JsonNode::JsonType::DATA_STRING:
  425. VLC->modh->identifiers.requestIdentifier(value, [&](si32 identifier)
  426. {
  427. var = identifier;
  428. });
  429. break;
  430. default:
  431. logMod->error("Error! Wrong identifier used for value of %s.", name);
  432. }
  433. }
  434. }
  435. void JsonUtils::resolveAddInfo(CAddInfo & var, const JsonNode & node)
  436. {
  437. const JsonNode & value = node["addInfo"];
  438. if (!value.isNull())
  439. {
  440. switch (value.getType())
  441. {
  442. case JsonNode::JsonType::DATA_INTEGER:
  443. var = value.Integer();
  444. break;
  445. case JsonNode::JsonType::DATA_FLOAT:
  446. var = value.Float();
  447. break;
  448. case JsonNode::JsonType::DATA_STRING:
  449. VLC->modh->identifiers.requestIdentifier(value, [&](si32 identifier)
  450. {
  451. var = identifier;
  452. });
  453. break;
  454. case JsonNode::JsonType::DATA_VECTOR:
  455. {
  456. const JsonVector & vec = value.Vector();
  457. var.resize(vec.size());
  458. for(int i = 0; i < vec.size(); i++)
  459. {
  460. switch(vec[i].getType())
  461. {
  462. case JsonNode::JsonType::DATA_INTEGER:
  463. var[i] = vec[i].Integer();
  464. break;
  465. case JsonNode::JsonType::DATA_FLOAT:
  466. var[i] = vec[i].Float();
  467. break;
  468. case JsonNode::JsonType::DATA_STRING:
  469. VLC->modh->identifiers.requestIdentifier(vec[i], [&var,i](si32 identifier)
  470. {
  471. var[i] = identifier;
  472. });
  473. break;
  474. default:
  475. logMod->error("Error! Wrong identifier used for value of addInfo[%d].", i);
  476. }
  477. }
  478. break;
  479. }
  480. default:
  481. logMod->error("Error! Wrong identifier used for value of addInfo.");
  482. }
  483. }
  484. }
  485. void JsonUtils::resolveIdentifier(const JsonNode &node, si32 &var)
  486. {
  487. switch (node.getType())
  488. {
  489. case JsonNode::JsonType::DATA_INTEGER:
  490. var = node.Integer();
  491. break;
  492. case JsonNode::JsonType::DATA_FLOAT:
  493. var = node.Float();
  494. break;
  495. case JsonNode::JsonType::DATA_STRING:
  496. VLC->modh->identifiers.requestIdentifier(node, [&](si32 identifier)
  497. {
  498. var = identifier;
  499. });
  500. break;
  501. default:
  502. logMod->error("Error! Wrong identifier used for identifier!");
  503. }
  504. }
  505. std::shared_ptr<ILimiter> JsonUtils::parseLimiter(const JsonNode & limiter)
  506. {
  507. switch(limiter.getType())
  508. {
  509. case JsonNode::JsonType::DATA_VECTOR:
  510. {
  511. const JsonVector & subLimiters = limiter.Vector();
  512. if(subLimiters.size() == 0)
  513. {
  514. logMod->warn("Warning: empty limiter list");
  515. return std::make_shared<AllOfLimiter>();
  516. }
  517. std::shared_ptr<AggregateLimiter> result;
  518. int offset = 1;
  519. // determine limiter type and offset for sub-limiters
  520. if(subLimiters[0].getType() == JsonNode::JsonType::DATA_STRING)
  521. {
  522. const std::string & aggregator = subLimiters[0].String();
  523. if(aggregator == AllOfLimiter::aggregator)
  524. result = std::make_shared<AllOfLimiter>();
  525. else if(aggregator == AnyOfLimiter::aggregator)
  526. result = std::make_shared<AnyOfLimiter>();
  527. else if(aggregator == NoneOfLimiter::aggregator)
  528. result = std::make_shared<NoneOfLimiter>();
  529. }
  530. if(!result)
  531. {
  532. // collapse for single limiter without explicit aggregate operator
  533. if(subLimiters.size() == 1)
  534. return parseLimiter(subLimiters[0]);
  535. // implicit aggregator must be allOf
  536. result = std::make_shared<AllOfLimiter>();
  537. offset = 0;
  538. }
  539. if(subLimiters.size() == offset)
  540. logMod->warn("Warning: empty sub-limiter list");
  541. for(int sl = offset; sl < subLimiters.size(); ++sl)
  542. result->add(parseLimiter(subLimiters[sl]));
  543. return result;
  544. }
  545. break;
  546. case JsonNode::JsonType::DATA_STRING: //pre-defined limiters
  547. return parseByMap(bonusLimiterMap, &limiter, "limiter type ");
  548. break;
  549. case JsonNode::JsonType::DATA_STRUCT: //customizable limiters
  550. {
  551. std::string limiterType = limiter["type"].String();
  552. const JsonVector & parameters = limiter["parameters"].Vector();
  553. if(limiterType == "CREATURE_TYPE_LIMITER")
  554. {
  555. std::shared_ptr<CCreatureTypeLimiter> creatureLimiter = std::make_shared<CCreatureTypeLimiter>();
  556. VLC->modh->identifiers.requestIdentifier("creature", parameters[0], [=](si32 creature)
  557. {
  558. creatureLimiter->setCreature(CreatureID(creature));
  559. });
  560. creatureLimiter->includeUpgrades = parameters.size() > 1 ? parameters[1].Bool() : false;
  561. return creatureLimiter;
  562. }
  563. else if(limiterType == "HAS_ANOTHER_BONUS_LIMITER")
  564. {
  565. std::string anotherBonusType = parameters[0].String();
  566. auto it = bonusNameMap.find(anotherBonusType);
  567. if(it == bonusNameMap.end())
  568. {
  569. logMod->error("Error: invalid ability type %s.", anotherBonusType);
  570. }
  571. else
  572. {
  573. std::shared_ptr<HasAnotherBonusLimiter> bonusLimiter = std::make_shared<HasAnotherBonusLimiter>();
  574. bonusLimiter->type = it->second;
  575. if(parameters.size() > 1)
  576. {
  577. resolveIdentifier(parameters[1], bonusLimiter->subtype);
  578. bonusLimiter->isSubtypeRelevant = true;
  579. }
  580. return bonusLimiter;
  581. }
  582. }
  583. else if(limiterType == "CREATURE_ALIGNMENT_LIMITER")
  584. {
  585. int alignment = vstd::find_pos(EAlignment::names, parameters[0].String());
  586. if(alignment == -1)
  587. logMod->error("Error: invalid alignment %s.", parameters[0].String());
  588. else
  589. return std::make_shared<CreatureAlignmentLimiter>(alignment);
  590. }
  591. else if(limiterType == "CREATURE_FACTION_LIMITER")
  592. {
  593. std::shared_ptr<CreatureFactionLimiter> factionLimiter = std::make_shared<CreatureFactionLimiter>();
  594. VLC->modh->identifiers.requestIdentifier("faction", parameters[0], [=](si32 faction)
  595. {
  596. factionLimiter->faction = faction;
  597. });
  598. return factionLimiter;
  599. }
  600. else if(limiterType == "CREATURE_TERRAIN_LIMITER")
  601. {
  602. std::shared_ptr<CreatureTerrainLimiter> terrainLimiter = std::make_shared<CreatureTerrainLimiter>();
  603. if(parameters.size())
  604. {
  605. VLC->modh->identifiers.requestIdentifier("terrain", parameters[0], [=](si32 terrain)
  606. {
  607. terrainLimiter->terrainType = terrain;
  608. });
  609. }
  610. return terrainLimiter;
  611. }
  612. else
  613. {
  614. logMod->error("Error: invalid customizable limiter type %s.", limiterType);
  615. }
  616. }
  617. break;
  618. default:
  619. break;
  620. }
  621. return nullptr;
  622. }
  623. std::shared_ptr<Bonus> JsonUtils::parseBonus(const JsonNode &ability)
  624. {
  625. auto b = std::make_shared<Bonus>();
  626. if (!parseBonus(ability, b.get()))
  627. {
  628. return nullptr;
  629. }
  630. return b;
  631. }
  632. bool JsonUtils::parseBonus(const JsonNode &ability, Bonus *b)
  633. {
  634. const JsonNode *value;
  635. std::string type = ability["type"].String();
  636. auto it = bonusNameMap.find(type);
  637. if (it == bonusNameMap.end())
  638. {
  639. logMod->error("Error: invalid ability type %s.", type);
  640. return false;
  641. }
  642. b->type = it->second;
  643. resolveIdentifier(b->subtype, ability, "subtype");
  644. b->val = ability["val"].Float();
  645. value = &ability["valueType"];
  646. if (!value->isNull())
  647. b->valType = static_cast<Bonus::ValueType>(parseByMap(bonusValueMap, value, "value type "));
  648. b->stacking = ability["stacking"].String();
  649. resolveAddInfo(b->additionalInfo, ability);
  650. b->turnsRemain = ability["turns"].Float();
  651. b->sid = ability["sourceID"].Float();
  652. b->description = ability["description"].String();
  653. value = &ability["effectRange"];
  654. if (!value->isNull())
  655. b->effectRange = static_cast<Bonus::LimitEffect>(parseByMap(bonusLimitEffect, value, "effect range "));
  656. value = &ability["duration"];
  657. if (!value->isNull())
  658. {
  659. switch (value->getType())
  660. {
  661. case JsonNode::JsonType::DATA_STRING:
  662. b->duration = parseByMap(bonusDurationMap, value, "duration type ");
  663. break;
  664. case JsonNode::JsonType::DATA_VECTOR:
  665. {
  666. ui16 dur = 0;
  667. for (const JsonNode & d : value->Vector())
  668. {
  669. dur |= parseByMap(bonusDurationMap, &d, "duration type ");
  670. }
  671. b->duration = dur;
  672. }
  673. break;
  674. default:
  675. logMod->error("Error! Wrong bonus duration format.");
  676. }
  677. }
  678. value = &ability["source"];
  679. if (!value->isNull())
  680. b->source = static_cast<Bonus::BonusSource>(parseByMap(bonusSourceMap, value, "source type "));
  681. value = &ability["limiters"];
  682. if (!value->isNull())
  683. b->limiter = parseLimiter(*value);
  684. value = &ability["propagator"];
  685. if (!value->isNull())
  686. b->propagator = parseByMap(bonusPropagatorMap, value, "propagator type ");
  687. value = &ability["updater"];
  688. if(!value->isNull())
  689. {
  690. const JsonNode & updaterJson = *value;
  691. switch(updaterJson.getType())
  692. {
  693. case JsonNode::JsonType::DATA_STRING:
  694. b->addUpdater(parseByMap(bonusUpdaterMap, &updaterJson, "updater type "));
  695. break;
  696. case JsonNode::JsonType::DATA_STRUCT:
  697. if(updaterJson["type"].String() == "GROWS_WITH_LEVEL")
  698. {
  699. std::shared_ptr<GrowsWithLevelUpdater> updater = std::make_shared<GrowsWithLevelUpdater>();
  700. const JsonVector param = updaterJson["parameters"].Vector();
  701. updater->valPer20 = param[0].Integer();
  702. if(param.size() > 1)
  703. updater->stepSize = param[1].Integer();
  704. b->addUpdater(updater);
  705. }
  706. else
  707. logMod->warn("Unknown updater type \"%s\"", updaterJson["type"].String());
  708. break;
  709. }
  710. }
  711. return true;
  712. }
  713. //returns first Key with value equal to given one
  714. template<class Key, class Val>
  715. Key reverseMapFirst(const Val & val, const std::map<Key, Val> & map)
  716. {
  717. for(auto it : map)
  718. {
  719. if(it.second == val)
  720. {
  721. return it.first;
  722. }
  723. }
  724. assert(0);
  725. return "";
  726. }
  727. void minimizeNode(JsonNode & node, const JsonNode & schema)
  728. {
  729. if (schema["type"].String() == "object")
  730. {
  731. std::set<std::string> foundEntries;
  732. for(auto & entry : schema["required"].Vector())
  733. {
  734. std::string name = entry.String();
  735. foundEntries.insert(name);
  736. minimizeNode(node[name], schema["properties"][name]);
  737. if (vstd::contains(node.Struct(), name) &&
  738. node[name] == schema["properties"][name]["default"])
  739. {
  740. node.Struct().erase(name);
  741. }
  742. }
  743. // erase all unhandled entries
  744. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  745. {
  746. if (!vstd::contains(foundEntries, it->first))
  747. it = node.Struct().erase(it);
  748. else
  749. it++;
  750. }
  751. }
  752. }
  753. void JsonUtils::minimize(JsonNode & node, std::string schemaName)
  754. {
  755. minimizeNode(node, getSchema(schemaName));
  756. }
  757. // FIXME: except for several lines function is identical to minimizeNode. Some way to reduce duplication?
  758. void maximizeNode(JsonNode & node, const JsonNode & schema)
  759. {
  760. // "required" entry can only be found in object/struct
  761. if (schema["type"].String() == "object")
  762. {
  763. std::set<std::string> foundEntries;
  764. // check all required entries that have default version
  765. for(auto & entry : schema["required"].Vector())
  766. {
  767. std::string name = entry.String();
  768. foundEntries.insert(name);
  769. if (node[name].isNull() &&
  770. !schema["properties"][name]["default"].isNull())
  771. {
  772. node[name] = schema["properties"][name]["default"];
  773. }
  774. maximizeNode(node[name], schema["properties"][name]);
  775. }
  776. // erase all unhandled entries
  777. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  778. {
  779. if (!vstd::contains(foundEntries, it->first))
  780. it = node.Struct().erase(it);
  781. else
  782. it++;
  783. }
  784. }
  785. }
  786. void JsonUtils::maximize(JsonNode & node, std::string schemaName)
  787. {
  788. maximizeNode(node, getSchema(schemaName));
  789. }
  790. bool JsonUtils::validate(const JsonNode &node, std::string schemaName, std::string dataName)
  791. {
  792. std::string log = Validation::check(schemaName, node);
  793. if (!log.empty())
  794. {
  795. logMod->warn("Data in %s is invalid!", dataName);
  796. logMod->warn(log);
  797. logMod->trace("%s json: %s", dataName, node.toJson(true));
  798. }
  799. return log.empty();
  800. }
  801. const JsonNode & getSchemaByName(std::string name)
  802. {
  803. // cached schemas to avoid loading json data multiple times
  804. static std::map<std::string, JsonNode> loadedSchemas;
  805. if (vstd::contains(loadedSchemas, name))
  806. return loadedSchemas[name];
  807. std::string filename = "config/schemas/" + name;
  808. if (CResourceHandler::get()->existsResource(ResourceID(filename)))
  809. {
  810. loadedSchemas[name] = JsonNode(ResourceID(filename));
  811. return loadedSchemas[name];
  812. }
  813. logMod->error("Error: missing schema with name %s!", name);
  814. assert(0);
  815. return nullNode;
  816. }
  817. const JsonNode & JsonUtils::getSchema(std::string URI)
  818. {
  819. size_t posColon = URI.find(':');
  820. size_t posHash = URI.find('#');
  821. std::string filename;
  822. if(posColon == std::string::npos)
  823. {
  824. filename = URI.substr(0, posHash);
  825. }
  826. else
  827. {
  828. std::string protocolName = URI.substr(0, posColon);
  829. filename = URI.substr(posColon + 1, posHash - posColon - 1) + ".json";
  830. if(protocolName != "vcmi")
  831. {
  832. logMod->error("Error: unsupported URI protocol for schema: %s", URI);
  833. return nullNode;
  834. }
  835. }
  836. // check if json pointer if present (section after hash in string)
  837. if(posHash == std::string::npos || posHash == URI.size() - 1)
  838. return getSchemaByName(filename);
  839. else
  840. return getSchemaByName(filename).resolvePointer(URI.substr(posHash + 1));
  841. }
  842. void JsonUtils::merge(JsonNode & dest, JsonNode & source, bool noOverride)
  843. {
  844. if (dest.getType() == JsonNode::JsonType::DATA_NULL)
  845. {
  846. std::swap(dest, source);
  847. return;
  848. }
  849. switch (source.getType())
  850. {
  851. case JsonNode::JsonType::DATA_NULL:
  852. {
  853. dest.clear();
  854. break;
  855. }
  856. case JsonNode::JsonType::DATA_BOOL:
  857. case JsonNode::JsonType::DATA_FLOAT:
  858. case JsonNode::JsonType::DATA_INTEGER:
  859. case JsonNode::JsonType::DATA_STRING:
  860. case JsonNode::JsonType::DATA_VECTOR:
  861. {
  862. std::swap(dest, source);
  863. break;
  864. }
  865. case JsonNode::JsonType::DATA_STRUCT:
  866. {
  867. if(!noOverride && vstd::contains(source.flags, "override"))
  868. {
  869. std::swap(dest, source);
  870. }
  871. else
  872. {
  873. //recursively merge all entries from struct
  874. for(auto & node : source.Struct())
  875. merge(dest[node.first], node.second, noOverride);
  876. }
  877. }
  878. }
  879. }
  880. void JsonUtils::mergeCopy(JsonNode & dest, JsonNode source, bool noOverride)
  881. {
  882. // uses copy created in stack to safely merge two nodes
  883. merge(dest, source, noOverride);
  884. }
  885. void JsonUtils::inherit(JsonNode & descendant, const JsonNode & base)
  886. {
  887. JsonNode inheritedNode(base);
  888. merge(inheritedNode, descendant, true);
  889. descendant.swap(inheritedNode);
  890. }
  891. JsonNode JsonUtils::intersect(const std::vector<JsonNode> & nodes, bool pruneEmpty)
  892. {
  893. if(nodes.size() == 0)
  894. return nullNode;
  895. JsonNode result = nodes[0];
  896. for(int i = 1; i < nodes.size(); i++)
  897. {
  898. if(result.isNull())
  899. break;
  900. result = JsonUtils::intersect(result, nodes[i], pruneEmpty);
  901. }
  902. return result;
  903. }
  904. JsonNode JsonUtils::intersect(const JsonNode & a, const JsonNode & b, bool pruneEmpty)
  905. {
  906. if(a.getType() == JsonNode::JsonType::DATA_STRUCT && b.getType() == JsonNode::JsonType::DATA_STRUCT)
  907. {
  908. // intersect individual properties
  909. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  910. for(auto property : a.Struct())
  911. {
  912. if(vstd::contains(b.Struct(), property.first))
  913. {
  914. JsonNode propertyIntersect = JsonUtils::intersect(property.second, b.Struct().find(property.first)->second);
  915. if(pruneEmpty && !propertyIntersect.containsBaseData())
  916. continue;
  917. result[property.first] = propertyIntersect;
  918. }
  919. }
  920. return result;
  921. }
  922. else
  923. {
  924. // not a struct - same or different, no middle ground
  925. if(a == b)
  926. return a;
  927. }
  928. return nullNode;
  929. }
  930. JsonNode JsonUtils::difference(const JsonNode & node, const JsonNode & base)
  931. {
  932. auto addsInfo = [](JsonNode diff) -> bool
  933. {
  934. switch(diff.getType())
  935. {
  936. case JsonNode::JsonType::DATA_NULL:
  937. return false;
  938. case JsonNode::JsonType::DATA_STRUCT:
  939. return diff.Struct().size() > 0;
  940. default:
  941. return true;
  942. }
  943. };
  944. if(node.getType() == JsonNode::JsonType::DATA_STRUCT && base.getType() == JsonNode::JsonType::DATA_STRUCT)
  945. {
  946. // subtract individual properties
  947. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  948. for(auto property : node.Struct())
  949. {
  950. if(vstd::contains(base.Struct(), property.first))
  951. {
  952. const JsonNode propertyDifference = JsonUtils::difference(property.second, base.Struct().find(property.first)->second);
  953. if(addsInfo(propertyDifference))
  954. result[property.first] = propertyDifference;
  955. }
  956. else
  957. {
  958. result[property.first] = property.second;
  959. }
  960. }
  961. return result;
  962. }
  963. else
  964. {
  965. if(node == base)
  966. return nullNode;
  967. }
  968. return node;
  969. }
  970. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files)
  971. {
  972. bool isValid;
  973. return assembleFromFiles(files, isValid);
  974. }
  975. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files, bool &isValid)
  976. {
  977. isValid = true;
  978. JsonNode result;
  979. for(std::string file : files)
  980. {
  981. bool isValidFile;
  982. JsonNode section(ResourceID(file, EResType::TEXT), isValidFile);
  983. merge(result, section);
  984. isValid |= isValidFile;
  985. }
  986. return result;
  987. }
  988. JsonNode JsonUtils::assembleFromFiles(std::string filename)
  989. {
  990. JsonNode result;
  991. ResourceID resID(filename, EResType::TEXT);
  992. for(auto & loader : CResourceHandler::get()->getResourcesWithName(resID))
  993. {
  994. // FIXME: some way to make this code more readable
  995. auto stream = loader->load(resID);
  996. std::unique_ptr<ui8[]> textData(new ui8[stream->getSize()]);
  997. stream->read(textData.get(), stream->getSize());
  998. JsonNode section((char*)textData.get(), stream->getSize());
  999. merge(result, section);
  1000. }
  1001. return result;
  1002. }
  1003. DLL_LINKAGE JsonNode JsonUtils::boolNode(bool value)
  1004. {
  1005. JsonNode node;
  1006. node.Bool() = value;
  1007. return node;
  1008. }
  1009. DLL_LINKAGE JsonNode JsonUtils::floatNode(double value)
  1010. {
  1011. JsonNode node;
  1012. node.Float() = value;
  1013. return node;
  1014. }
  1015. DLL_LINKAGE JsonNode JsonUtils::stringNode(std::string value)
  1016. {
  1017. JsonNode node;
  1018. node.String() = value;
  1019. return node;
  1020. }
  1021. DLL_LINKAGE JsonNode JsonUtils::intNode(si64 value)
  1022. {
  1023. JsonNode node;
  1024. node.Integer() = value;
  1025. return node;
  1026. }