JsonNode.cpp 30 KB

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