JsonNode.cpp 31 KB

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