JsonNode.cpp 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195
  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) //TODO: merge with AddAbility, create universal parser for all bonus properties
  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. void JsonUtils::resolveIdentifier(si32 &var, const JsonNode &node, std::string name)
  430. {
  431. const JsonNode &value = node[name];
  432. if (!value.isNull())
  433. {
  434. switch (value.getType())
  435. {
  436. case JsonNode::JsonType::DATA_INTEGER:
  437. var = static_cast<si32>(value.Integer());
  438. break;
  439. case JsonNode::JsonType::DATA_FLOAT:
  440. var = static_cast<si32>(value.Float());
  441. break;
  442. case JsonNode::JsonType::DATA_STRING:
  443. VLC->modh->identifiers.requestIdentifier(value, [&](si32 identifier)
  444. {
  445. var = identifier;
  446. });
  447. break;
  448. default:
  449. logMod->error("Error! Wrong identifier used for value of %s.", name);
  450. }
  451. }
  452. }
  453. void JsonUtils::resolveAddInfo(CAddInfo & var, const JsonNode & node)
  454. {
  455. const JsonNode & value = node["addInfo"];
  456. if (!value.isNull())
  457. {
  458. switch (value.getType())
  459. {
  460. case JsonNode::JsonType::DATA_INTEGER:
  461. var = static_cast<si32>(value.Integer());
  462. break;
  463. case JsonNode::JsonType::DATA_FLOAT:
  464. var = static_cast<si32>(value.Float());
  465. break;
  466. case JsonNode::JsonType::DATA_STRING:
  467. VLC->modh->identifiers.requestIdentifier(value, [&](si32 identifier)
  468. {
  469. var = identifier;
  470. });
  471. break;
  472. case JsonNode::JsonType::DATA_VECTOR:
  473. {
  474. const JsonVector & vec = value.Vector();
  475. var.resize(vec.size());
  476. for(int i = 0; i < vec.size(); i++)
  477. {
  478. switch(vec[i].getType())
  479. {
  480. case JsonNode::JsonType::DATA_INTEGER:
  481. var[i] = static_cast<si32>(vec[i].Integer());
  482. break;
  483. case JsonNode::JsonType::DATA_FLOAT:
  484. var[i] = static_cast<si32>(vec[i].Float());
  485. break;
  486. case JsonNode::JsonType::DATA_STRING:
  487. VLC->modh->identifiers.requestIdentifier(vec[i], [&var,i](si32 identifier)
  488. {
  489. var[i] = identifier;
  490. });
  491. break;
  492. default:
  493. logMod->error("Error! Wrong identifier used for value of addInfo[%d].", i);
  494. }
  495. }
  496. break;
  497. }
  498. default:
  499. logMod->error("Error! Wrong identifier used for value of addInfo.");
  500. }
  501. }
  502. }
  503. void JsonUtils::resolveIdentifier(const JsonNode &node, si32 &var)
  504. {
  505. switch (node.getType())
  506. {
  507. case JsonNode::JsonType::DATA_INTEGER:
  508. var = static_cast<si32>(node.Integer());
  509. break;
  510. case JsonNode::JsonType::DATA_FLOAT:
  511. var = static_cast<si32>(node.Float());
  512. break;
  513. case JsonNode::JsonType::DATA_STRING:
  514. VLC->modh->identifiers.requestIdentifier(node, [&](si32 identifier)
  515. {
  516. var = identifier;
  517. });
  518. break;
  519. default:
  520. logMod->error("Error! Wrong identifier used for identifier!");
  521. }
  522. }
  523. std::shared_ptr<ILimiter> JsonUtils::parseLimiter(const JsonNode & limiter)
  524. {
  525. switch(limiter.getType())
  526. {
  527. case JsonNode::JsonType::DATA_VECTOR:
  528. {
  529. const JsonVector & subLimiters = limiter.Vector();
  530. if(subLimiters.size() == 0)
  531. {
  532. logMod->warn("Warning: empty limiter list");
  533. return std::make_shared<AllOfLimiter>();
  534. }
  535. std::shared_ptr<AggregateLimiter> result;
  536. int offset = 1;
  537. // determine limiter type and offset for sub-limiters
  538. if(subLimiters[0].getType() == JsonNode::JsonType::DATA_STRING)
  539. {
  540. const std::string & aggregator = subLimiters[0].String();
  541. if(aggregator == AllOfLimiter::aggregator)
  542. result = std::make_shared<AllOfLimiter>();
  543. else if(aggregator == AnyOfLimiter::aggregator)
  544. result = std::make_shared<AnyOfLimiter>();
  545. else if(aggregator == NoneOfLimiter::aggregator)
  546. result = std::make_shared<NoneOfLimiter>();
  547. }
  548. if(!result)
  549. {
  550. // collapse for single limiter without explicit aggregate operator
  551. if(subLimiters.size() == 1)
  552. return parseLimiter(subLimiters[0]);
  553. // implicit aggregator must be allOf
  554. result = std::make_shared<AllOfLimiter>();
  555. offset = 0;
  556. }
  557. if(subLimiters.size() == offset)
  558. logMod->warn("Warning: empty sub-limiter list");
  559. for(int sl = offset; sl < subLimiters.size(); ++sl)
  560. result->add(parseLimiter(subLimiters[sl]));
  561. return result;
  562. }
  563. break;
  564. case JsonNode::JsonType::DATA_STRING: //pre-defined limiters
  565. return parseByMap(bonusLimiterMap, &limiter, "limiter type ");
  566. break;
  567. case JsonNode::JsonType::DATA_STRUCT: //customizable limiters
  568. {
  569. std::string limiterType = limiter["type"].String();
  570. const JsonVector & parameters = limiter["parameters"].Vector();
  571. if(limiterType == "CREATURE_TYPE_LIMITER")
  572. {
  573. std::shared_ptr<CCreatureTypeLimiter> creatureLimiter = std::make_shared<CCreatureTypeLimiter>();
  574. VLC->modh->identifiers.requestIdentifier("creature", parameters[0], [=](si32 creature)
  575. {
  576. creatureLimiter->setCreature(CreatureID(creature));
  577. });
  578. auto includeUpgrades = false;
  579. if(parameters.size() > 1)
  580. {
  581. bool success = true;
  582. includeUpgrades = parameters[1].TryBoolFromString(success);
  583. if(!success)
  584. logMod->error("Second parameter of '%s' limiter should be Bool", limiterType);
  585. }
  586. creatureLimiter->includeUpgrades = includeUpgrades;
  587. return creatureLimiter;
  588. }
  589. else if(limiterType == "HAS_ANOTHER_BONUS_LIMITER")
  590. {
  591. std::string anotherBonusType = parameters[0].String();
  592. auto it = bonusNameMap.find(anotherBonusType);
  593. if(it == bonusNameMap.end())
  594. {
  595. logMod->error("Error: invalid ability type %s.", anotherBonusType);
  596. }
  597. else
  598. {
  599. std::shared_ptr<HasAnotherBonusLimiter> bonusLimiter = std::make_shared<HasAnotherBonusLimiter>();
  600. bonusLimiter->type = it->second;
  601. if(parameters.size() > 1)
  602. {
  603. resolveIdentifier(parameters[1], bonusLimiter->subtype);
  604. bonusLimiter->isSubtypeRelevant = true;
  605. }
  606. return bonusLimiter;
  607. }
  608. }
  609. else if(limiterType == "CREATURE_ALIGNMENT_LIMITER")
  610. {
  611. int alignment = vstd::find_pos(EAlignment::names, parameters[0].String());
  612. if(alignment == -1)
  613. logMod->error("Error: invalid alignment %s.", parameters[0].String());
  614. else
  615. return std::make_shared<CreatureAlignmentLimiter>(alignment);
  616. }
  617. else if(limiterType == "CREATURE_FACTION_LIMITER")
  618. {
  619. std::shared_ptr<CreatureFactionLimiter> factionLimiter = std::make_shared<CreatureFactionLimiter>();
  620. VLC->modh->identifiers.requestIdentifier("faction", parameters[0], [=](si32 faction)
  621. {
  622. factionLimiter->faction = faction;
  623. });
  624. return factionLimiter;
  625. }
  626. else if(limiterType == "CREATURE_TERRAIN_LIMITER")
  627. {
  628. std::shared_ptr<CreatureTerrainLimiter> terrainLimiter = std::make_shared<CreatureTerrainLimiter>();
  629. if(parameters.size())
  630. {
  631. VLC->modh->identifiers.requestIdentifier("terrain", parameters[0], [=](si32 terrain)
  632. {
  633. terrainLimiter->terrainType = terrain;
  634. });
  635. }
  636. return terrainLimiter;
  637. }
  638. else
  639. {
  640. logMod->error("Error: invalid customizable limiter type %s.", limiterType);
  641. }
  642. }
  643. break;
  644. default:
  645. break;
  646. }
  647. return nullptr;
  648. }
  649. std::shared_ptr<Bonus> JsonUtils::parseBonus(const JsonNode &ability)
  650. {
  651. auto b = std::make_shared<Bonus>();
  652. if (!parseBonus(ability, b.get()))
  653. {
  654. return nullptr;
  655. }
  656. return b;
  657. }
  658. std::shared_ptr<Bonus> JsonUtils::parseBuildingBonus(const JsonNode &ability, BuildingID building, std::string description)
  659. {
  660. /* duration = Bonus::PERMANENT
  661. source = Bonus::TOWN_STRUCTURE
  662. bonusType, val, subtype - get from json
  663. */
  664. auto b = std::make_shared<Bonus>(Bonus::PERMANENT, Bonus::NONE, Bonus::TOWN_STRUCTURE, 0, building, description, -1);
  665. if(!parseBonus(ability, b.get()))
  666. return nullptr;
  667. return b;
  668. }
  669. bool JsonUtils::parseBonus(const JsonNode &ability, Bonus *b)
  670. {
  671. const JsonNode *value;
  672. std::string type = ability["type"].String();
  673. auto it = bonusNameMap.find(type);
  674. if (it == bonusNameMap.end())
  675. {
  676. logMod->error("Error: invalid ability type %s.", type);
  677. return false;
  678. }
  679. b->type = it->second;
  680. resolveIdentifier(b->subtype, ability, "subtype");
  681. b->val = static_cast<si32>(ability["val"].Float());
  682. value = &ability["valueType"];
  683. if (!value->isNull())
  684. b->valType = static_cast<Bonus::ValueType>(parseByMap(bonusValueMap, value, "value type "));
  685. b->stacking = ability["stacking"].String();
  686. resolveAddInfo(b->additionalInfo, ability);
  687. b->turnsRemain = static_cast<si32>(ability["turns"].Float());
  688. b->sid = static_cast<si32>(ability["sourceID"].Float());
  689. if(!ability["description"].isNull())
  690. b->description = ability["description"].String();
  691. value = &ability["effectRange"];
  692. if (!value->isNull())
  693. b->effectRange = static_cast<Bonus::LimitEffect>(parseByMap(bonusLimitEffect, value, "effect range "));
  694. value = &ability["duration"];
  695. if (!value->isNull())
  696. {
  697. switch (value->getType())
  698. {
  699. case JsonNode::JsonType::DATA_STRING:
  700. b->duration = (Bonus::BonusDuration)parseByMap(bonusDurationMap, value, "duration type ");
  701. break;
  702. case JsonNode::JsonType::DATA_VECTOR:
  703. {
  704. ui16 dur = 0;
  705. for (const JsonNode & d : value->Vector())
  706. {
  707. dur |= parseByMap(bonusDurationMap, &d, "duration type ");
  708. }
  709. b->duration = (Bonus::BonusDuration)dur;
  710. }
  711. break;
  712. default:
  713. logMod->error("Error! Wrong bonus duration format.");
  714. }
  715. }
  716. value = &ability["source"];
  717. if (!value->isNull())
  718. b->source = static_cast<Bonus::BonusSource>(parseByMap(bonusSourceMap, value, "source type "));
  719. value = &ability["limiters"];
  720. if (!value->isNull())
  721. b->limiter = parseLimiter(*value);
  722. value = &ability["propagator"];
  723. if (!value->isNull())
  724. b->propagator = parseByMap(bonusPropagatorMap, value, "propagator type ");
  725. value = &ability["updater"];
  726. if(!value->isNull())
  727. {
  728. const JsonNode & updaterJson = *value;
  729. switch(updaterJson.getType())
  730. {
  731. case JsonNode::JsonType::DATA_STRING:
  732. b->addUpdater(parseByMap(bonusUpdaterMap, &updaterJson, "updater type "));
  733. break;
  734. case JsonNode::JsonType::DATA_STRUCT:
  735. if(updaterJson["type"].String() == "GROWS_WITH_LEVEL")
  736. {
  737. std::shared_ptr<GrowsWithLevelUpdater> updater = std::make_shared<GrowsWithLevelUpdater>();
  738. const JsonVector param = updaterJson["parameters"].Vector();
  739. updater->valPer20 = static_cast<int>(param[0].Integer());
  740. if(param.size() > 1)
  741. updater->stepSize = static_cast<int>(param[1].Integer());
  742. b->addUpdater(updater);
  743. }
  744. else
  745. logMod->warn("Unknown updater type \"%s\"", updaterJson["type"].String());
  746. break;
  747. }
  748. }
  749. return true;
  750. }
  751. //returns first Key with value equal to given one
  752. template<class Key, class Val>
  753. Key reverseMapFirst(const Val & val, const std::map<Key, Val> & map)
  754. {
  755. for(auto it : map)
  756. {
  757. if(it.second == val)
  758. {
  759. return it.first;
  760. }
  761. }
  762. assert(0);
  763. return "";
  764. }
  765. void minimizeNode(JsonNode & node, const JsonNode & schema)
  766. {
  767. if (schema["type"].String() == "object")
  768. {
  769. std::set<std::string> foundEntries;
  770. for(auto & entry : schema["required"].Vector())
  771. {
  772. std::string name = entry.String();
  773. foundEntries.insert(name);
  774. minimizeNode(node[name], schema["properties"][name]);
  775. if (vstd::contains(node.Struct(), name) &&
  776. node[name] == schema["properties"][name]["default"])
  777. {
  778. node.Struct().erase(name);
  779. }
  780. }
  781. // erase all unhandled entries
  782. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  783. {
  784. if (!vstd::contains(foundEntries, it->first))
  785. it = node.Struct().erase(it);
  786. else
  787. it++;
  788. }
  789. }
  790. }
  791. void JsonUtils::minimize(JsonNode & node, std::string schemaName)
  792. {
  793. minimizeNode(node, getSchema(schemaName));
  794. }
  795. // FIXME: except for several lines function is identical to minimizeNode. Some way to reduce duplication?
  796. void maximizeNode(JsonNode & node, const JsonNode & schema)
  797. {
  798. // "required" entry can only be found in object/struct
  799. if (schema["type"].String() == "object")
  800. {
  801. std::set<std::string> foundEntries;
  802. // check all required entries that have default version
  803. for(auto & entry : schema["required"].Vector())
  804. {
  805. std::string name = entry.String();
  806. foundEntries.insert(name);
  807. if (node[name].isNull() &&
  808. !schema["properties"][name]["default"].isNull())
  809. {
  810. node[name] = schema["properties"][name]["default"];
  811. }
  812. maximizeNode(node[name], schema["properties"][name]);
  813. }
  814. // erase all unhandled entries
  815. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  816. {
  817. if (!vstd::contains(foundEntries, it->first))
  818. it = node.Struct().erase(it);
  819. else
  820. it++;
  821. }
  822. }
  823. }
  824. void JsonUtils::maximize(JsonNode & node, std::string schemaName)
  825. {
  826. maximizeNode(node, getSchema(schemaName));
  827. }
  828. bool JsonUtils::validate(const JsonNode &node, std::string schemaName, std::string dataName)
  829. {
  830. std::string log = Validation::check(schemaName, node);
  831. if (!log.empty())
  832. {
  833. logMod->warn("Data in %s is invalid!", dataName);
  834. logMod->warn(log);
  835. logMod->trace("%s json: %s", dataName, node.toJson(true));
  836. }
  837. return log.empty();
  838. }
  839. const JsonNode & getSchemaByName(std::string name)
  840. {
  841. // cached schemas to avoid loading json data multiple times
  842. static std::map<std::string, JsonNode> loadedSchemas;
  843. if (vstd::contains(loadedSchemas, name))
  844. return loadedSchemas[name];
  845. std::string filename = "config/schemas/" + name;
  846. if (CResourceHandler::get()->existsResource(ResourceID(filename)))
  847. {
  848. loadedSchemas[name] = JsonNode(ResourceID(filename));
  849. return loadedSchemas[name];
  850. }
  851. logMod->error("Error: missing schema with name %s!", name);
  852. assert(0);
  853. return nullNode;
  854. }
  855. const JsonNode & JsonUtils::getSchema(std::string URI)
  856. {
  857. size_t posColon = URI.find(':');
  858. size_t posHash = URI.find('#');
  859. std::string filename;
  860. if(posColon == std::string::npos)
  861. {
  862. filename = URI.substr(0, posHash);
  863. }
  864. else
  865. {
  866. std::string protocolName = URI.substr(0, posColon);
  867. filename = URI.substr(posColon + 1, posHash - posColon - 1) + ".json";
  868. if(protocolName != "vcmi")
  869. {
  870. logMod->error("Error: unsupported URI protocol for schema: %s", URI);
  871. return nullNode;
  872. }
  873. }
  874. // check if json pointer if present (section after hash in string)
  875. if(posHash == std::string::npos || posHash == URI.size() - 1)
  876. return getSchemaByName(filename);
  877. else
  878. return getSchemaByName(filename).resolvePointer(URI.substr(posHash + 1));
  879. }
  880. void JsonUtils::merge(JsonNode & dest, JsonNode & source, bool noOverride)
  881. {
  882. if (dest.getType() == JsonNode::JsonType::DATA_NULL)
  883. {
  884. std::swap(dest, source);
  885. return;
  886. }
  887. switch (source.getType())
  888. {
  889. case JsonNode::JsonType::DATA_NULL:
  890. {
  891. dest.clear();
  892. break;
  893. }
  894. case JsonNode::JsonType::DATA_BOOL:
  895. case JsonNode::JsonType::DATA_FLOAT:
  896. case JsonNode::JsonType::DATA_INTEGER:
  897. case JsonNode::JsonType::DATA_STRING:
  898. case JsonNode::JsonType::DATA_VECTOR:
  899. {
  900. std::swap(dest, source);
  901. break;
  902. }
  903. case JsonNode::JsonType::DATA_STRUCT:
  904. {
  905. if(!noOverride && vstd::contains(source.flags, "override"))
  906. {
  907. std::swap(dest, source);
  908. }
  909. else
  910. {
  911. //recursively merge all entries from struct
  912. for(auto & node : source.Struct())
  913. merge(dest[node.first], node.second, noOverride);
  914. }
  915. }
  916. }
  917. }
  918. void JsonUtils::mergeCopy(JsonNode & dest, JsonNode source, bool noOverride)
  919. {
  920. // uses copy created in stack to safely merge two nodes
  921. merge(dest, source, noOverride);
  922. }
  923. void JsonUtils::inherit(JsonNode & descendant, const JsonNode & base)
  924. {
  925. JsonNode inheritedNode(base);
  926. merge(inheritedNode, descendant, true);
  927. descendant.swap(inheritedNode);
  928. }
  929. JsonNode JsonUtils::intersect(const std::vector<JsonNode> & nodes, bool pruneEmpty)
  930. {
  931. if(nodes.size() == 0)
  932. return nullNode;
  933. JsonNode result = nodes[0];
  934. for(int i = 1; i < nodes.size(); i++)
  935. {
  936. if(result.isNull())
  937. break;
  938. result = JsonUtils::intersect(result, nodes[i], pruneEmpty);
  939. }
  940. return result;
  941. }
  942. JsonNode JsonUtils::intersect(const JsonNode & a, const JsonNode & b, bool pruneEmpty)
  943. {
  944. if(a.getType() == JsonNode::JsonType::DATA_STRUCT && b.getType() == JsonNode::JsonType::DATA_STRUCT)
  945. {
  946. // intersect individual properties
  947. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  948. for(auto property : a.Struct())
  949. {
  950. if(vstd::contains(b.Struct(), property.first))
  951. {
  952. JsonNode propertyIntersect = JsonUtils::intersect(property.second, b.Struct().find(property.first)->second);
  953. if(pruneEmpty && !propertyIntersect.containsBaseData())
  954. continue;
  955. result[property.first] = propertyIntersect;
  956. }
  957. }
  958. return result;
  959. }
  960. else
  961. {
  962. // not a struct - same or different, no middle ground
  963. if(a == b)
  964. return a;
  965. }
  966. return nullNode;
  967. }
  968. JsonNode JsonUtils::difference(const JsonNode & node, const JsonNode & base)
  969. {
  970. auto addsInfo = [](JsonNode diff) -> bool
  971. {
  972. switch(diff.getType())
  973. {
  974. case JsonNode::JsonType::DATA_NULL:
  975. return false;
  976. case JsonNode::JsonType::DATA_STRUCT:
  977. return diff.Struct().size() > 0;
  978. default:
  979. return true;
  980. }
  981. };
  982. if(node.getType() == JsonNode::JsonType::DATA_STRUCT && base.getType() == JsonNode::JsonType::DATA_STRUCT)
  983. {
  984. // subtract individual properties
  985. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  986. for(auto property : node.Struct())
  987. {
  988. if(vstd::contains(base.Struct(), property.first))
  989. {
  990. const JsonNode propertyDifference = JsonUtils::difference(property.second, base.Struct().find(property.first)->second);
  991. if(addsInfo(propertyDifference))
  992. result[property.first] = propertyDifference;
  993. }
  994. else
  995. {
  996. result[property.first] = property.second;
  997. }
  998. }
  999. return result;
  1000. }
  1001. else
  1002. {
  1003. if(node == base)
  1004. return nullNode;
  1005. }
  1006. return node;
  1007. }
  1008. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files)
  1009. {
  1010. bool isValid;
  1011. return assembleFromFiles(files, isValid);
  1012. }
  1013. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files, bool &isValid)
  1014. {
  1015. isValid = true;
  1016. JsonNode result;
  1017. for(std::string file : files)
  1018. {
  1019. bool isValidFile;
  1020. JsonNode section(ResourceID(file, EResType::TEXT), isValidFile);
  1021. merge(result, section);
  1022. isValid |= isValidFile;
  1023. }
  1024. return result;
  1025. }
  1026. JsonNode JsonUtils::assembleFromFiles(std::string filename)
  1027. {
  1028. JsonNode result;
  1029. ResourceID resID(filename, EResType::TEXT);
  1030. for(auto & loader : CResourceHandler::get()->getResourcesWithName(resID))
  1031. {
  1032. // FIXME: some way to make this code more readable
  1033. auto stream = loader->load(resID);
  1034. std::unique_ptr<ui8[]> textData(new ui8[stream->getSize()]);
  1035. stream->read(textData.get(), stream->getSize());
  1036. JsonNode section((char*)textData.get(), stream->getSize());
  1037. merge(result, section);
  1038. }
  1039. return result;
  1040. }
  1041. DLL_LINKAGE JsonNode JsonUtils::boolNode(bool value)
  1042. {
  1043. JsonNode node;
  1044. node.Bool() = value;
  1045. return node;
  1046. }
  1047. DLL_LINKAGE JsonNode JsonUtils::floatNode(double value)
  1048. {
  1049. JsonNode node;
  1050. node.Float() = value;
  1051. return node;
  1052. }
  1053. DLL_LINKAGE JsonNode JsonUtils::stringNode(std::string value)
  1054. {
  1055. JsonNode node;
  1056. node.String() = value;
  1057. return node;
  1058. }
  1059. DLL_LINKAGE JsonNode JsonUtils::intNode(si64 value)
  1060. {
  1061. JsonNode node;
  1062. node.Integer() = value;
  1063. return node;
  1064. }