JsonNode.cpp 31 KB

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