2
0

JsonNode.cpp 31 KB

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