JsonNode.cpp 29 KB

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