| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798 | 
							- /*
 
-  * HeroBonus.cpp, part of VCMI engine
 
-  *
 
-  * Authors: listed in file AUTHORS in main folder
 
-  *
 
-  * License: GNU General Public License v2.0 or later
 
-  * Full text of license available in license.txt file, in main folder
 
-  *
 
-  */
 
- #include "StdInc.h"
 
- #include "HeroBonus.h"
 
- #include "VCMI_Lib.h"
 
- #include "spells/CSpellHandler.h"
 
- #include "CCreatureHandler.h"
 
- #include "CCreatureSet.h"
 
- #include "CHeroHandler.h"
 
- #include "CTownHandler.h"
 
- #include "CGeneralTextHandler.h"
 
- #include "CSkillHandler.h"
 
- #include "CStack.h"
 
- #include "CArtHandler.h"
 
- #include "CModHandler.h"
 
- #include "TerrainHandler.h"
 
- #include "StringConstants.h"
 
- #include "battle/BattleInfo.h"
 
- VCMI_LIB_NAMESPACE_BEGIN
 
- #define FOREACH_PARENT(pname) 	TNodes lparents; getParents(lparents); for(CBonusSystemNode *pname : lparents)
 
- #define FOREACH_RED_CHILD(pname) 	TNodes lchildren; getRedChildren(lchildren); for(CBonusSystemNode *pname : lchildren)
 
- #define BONUS_NAME(x) { #x, Bonus::x },
 
- 	const std::map<std::string, Bonus::BonusType> bonusNameMap = {
 
- 		BONUS_LIST
 
- 	};
 
- #undef BONUS_NAME
 
- #define BONUS_VALUE(x) { #x, Bonus::x },
 
- 	const std::map<std::string, Bonus::ValueType> bonusValueMap = { BONUS_VALUE_LIST };
 
- #undef BONUS_VALUE
 
- #define BONUS_SOURCE(x) { #x, Bonus::x },
 
- 	const std::map<std::string, Bonus::BonusSource> bonusSourceMap = { BONUS_SOURCE_LIST };
 
- #undef BONUS_SOURCE
 
- #define BONUS_ITEM(x) { #x, Bonus::x },
 
- const std::map<std::string, ui16> bonusDurationMap =
 
- {
 
- 	BONUS_ITEM(PERMANENT)
 
- 	BONUS_ITEM(ONE_BATTLE)
 
- 	BONUS_ITEM(ONE_DAY)
 
- 	BONUS_ITEM(ONE_WEEK)
 
- 	BONUS_ITEM(N_TURNS)
 
- 	BONUS_ITEM(N_DAYS)
 
- 	BONUS_ITEM(UNTIL_BEING_ATTACKED)
 
- 	BONUS_ITEM(UNTIL_ATTACK)
 
- 	BONUS_ITEM(STACK_GETS_TURN)
 
- 	BONUS_ITEM(COMMANDER_KILLED)
 
- 	{ "UNITL_BEING_ATTACKED", Bonus::UNTIL_BEING_ATTACKED }//typo, but used in some mods
 
- };
 
- const std::map<std::string, Bonus::LimitEffect> bonusLimitEffect =
 
- {
 
- 	BONUS_ITEM(NO_LIMIT)
 
- 	BONUS_ITEM(ONLY_DISTANCE_FIGHT)
 
- 	BONUS_ITEM(ONLY_MELEE_FIGHT)
 
- };
 
- const std::map<std::string, TLimiterPtr> bonusLimiterMap =
 
- {
 
- 	{"SHOOTER_ONLY", std::make_shared<HasAnotherBonusLimiter>(Bonus::SHOOTER)},
 
- 	{"DRAGON_NATURE", std::make_shared<HasAnotherBonusLimiter>(Bonus::DRAGON_NATURE)},
 
- 	{"IS_UNDEAD", std::make_shared<HasAnotherBonusLimiter>(Bonus::UNDEAD)},
 
- 	{"CREATURE_NATIVE_TERRAIN", std::make_shared<CreatureTerrainLimiter>()},
 
- 	{"CREATURE_FACTION", std::make_shared<CreatureFactionLimiter>()},
 
- 	{"OPPOSITE_SIDE", std::make_shared<OppositeSideLimiter>()}
 
- };
 
- const std::map<std::string, TPropagatorPtr> bonusPropagatorMap =
 
- {
 
- 	{"BATTLE_WIDE", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::BATTLE)},
 
- 	{"VISITED_TOWN_AND_VISITOR", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::TOWN_AND_VISITOR)},
 
- 	{"PLAYER_PROPAGATOR", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::PLAYER)},
 
- 	{"HERO", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::HERO)},
 
- 	{"TEAM_PROPAGATOR", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::TEAM)}, //untested
 
- 	{"GLOBAL_EFFECT", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::GLOBAL_EFFECTS)},
 
- 	{"ALL_CREATURES", std::make_shared<CPropagatorNodeType>(CBonusSystemNode::ALL_CREATURES)}
 
- }; //untested
 
- const std::map<std::string, TUpdaterPtr> bonusUpdaterMap =
 
- {
 
- 	{"TIMES_HERO_LEVEL", std::make_shared<TimesHeroLevelUpdater>()},
 
- 	{"TIMES_STACK_LEVEL", std::make_shared<TimesStackLevelUpdater>()},
 
- 	{"ARMY_MOVEMENT", std::make_shared<ArmyMovementUpdater>()},
 
- 	{"BONUS_OWNER_UPDATER", std::make_shared<OwnerUpdater>()}
 
- };
 
- const std::set<std::string> deprecatedBonusSet = {
 
- 	"SECONDARY_SKILL_PREMY",
 
- 	"SECONDARY_SKILL_VAL2",
 
- 	"MAXED_SPELL",
 
- 	"LAND_MOVEMENT",
 
- 	"SEA_MOVEMENT",
 
- 	"SIGHT_RADIOUS",
 
- 	"NO_TYPE",
 
- 	"SPECIAL_SECONDARY_SKILL",
 
- 	"FULL_HP_REGENERATION",
 
- 	"KING1",
 
- 	"KING2",
 
- 	"KING3",
 
- 	"BLOCK_MORALE",
 
- 	"BLOCK_LUCK",
 
- 	"SELF_MORALE",
 
- 	"SELF_LUCK"
 
- };
 
- ///CBonusProxy
 
- CBonusProxy::CBonusProxy(const IBonusBearer * Target, CSelector Selector):
 
- 	bonusListCachedLast(0),
 
- 	target(Target),
 
- 	selector(std::move(Selector)),
 
- 	currentBonusListIndex(0)
 
- {
 
- }
 
- CBonusProxy::CBonusProxy(const CBonusProxy & other):
 
- 	bonusListCachedLast(other.bonusListCachedLast),
 
- 	target(other.target),
 
- 	selector(other.selector),
 
- 	currentBonusListIndex(other.currentBonusListIndex)
 
- {
 
- 	bonusList[currentBonusListIndex] = other.bonusList[currentBonusListIndex];
 
- }
 
- CBonusProxy::CBonusProxy(CBonusProxy && other) noexcept:
 
- 	bonusListCachedLast(0),
 
- 	target(other.target),
 
- 	currentBonusListIndex(0)
 
- {
 
- 	std::swap(bonusListCachedLast, other.bonusListCachedLast);
 
- 	std::swap(selector, other.selector);
 
- 	std::swap(bonusList, other.bonusList);
 
- 	std::swap(currentBonusListIndex, other.currentBonusListIndex);
 
- }
 
- CBonusProxy & CBonusProxy::operator=(const CBonusProxy & other)
 
- {
 
- 	boost::lock_guard<boost::mutex> lock(swapGuard);
 
- 	selector = other.selector;
 
- 	swapBonusList(other.bonusList[other.currentBonusListIndex]);
 
- 	bonusListCachedLast = other.bonusListCachedLast;
 
- 	return *this;
 
- }
 
- CBonusProxy & CBonusProxy::operator=(CBonusProxy && other) noexcept
 
- {
 
- 	std::swap(bonusListCachedLast, other.bonusListCachedLast);
 
- 	std::swap(selector, other.selector);
 
- 	std::swap(bonusList, other.bonusList);
 
- 	std::swap(currentBonusListIndex, other.currentBonusListIndex);
 
- 	return *this;
 
- }
 
- void CBonusProxy::swapBonusList(TConstBonusListPtr other) const
 
- {
 
- 	// The idea here is to avoid changing active bonusList while it can be read by a different thread.
 
- 	// Because such use of shared ptr is not thread safe
 
- 	// So to avoid this we change the second offline instance and swap active index
 
- 	auto newCurrent = 1 - currentBonusListIndex;
 
- 	bonusList[newCurrent] = std::move(other);
 
- 	currentBonusListIndex = newCurrent;
 
- }
 
- TConstBonusListPtr CBonusProxy::getBonusList() const
 
- {
 
- 	auto needUpdateBonusList = [&]() -> bool
 
- 	{
 
- 		return target->getTreeVersion() != bonusListCachedLast || !bonusList[currentBonusListIndex];
 
- 	};
 
- 	// avoid locking if everything is up-to-date
 
- 	if(needUpdateBonusList())
 
- 	{
 
- 		boost::lock_guard<boost::mutex>lock(swapGuard);
 
- 		if(needUpdateBonusList())
 
- 		{
 
- 			//TODO: support limiters
 
- 			swapBonusList(target->getAllBonuses(selector, Selector::all));
 
- 			bonusListCachedLast = target->getTreeVersion();
 
- 		}
 
- 	}
 
- 	return bonusList[currentBonusListIndex];
 
- }
 
- const BonusList * CBonusProxy::operator->() const
 
- {
 
- 	return getBonusList().get();
 
- }
 
- CTotalsProxy::CTotalsProxy(const IBonusBearer * Target, CSelector Selector, int InitialValue):
 
- 	CBonusProxy(Target, std::move(Selector)),
 
- 	initialValue(InitialValue),
 
- 	meleeCachedLast(0),
 
- 	meleeValue(0),
 
- 	rangedCachedLast(0),
 
- 	rangedValue(0)
 
- {
 
- }
 
- CTotalsProxy::CTotalsProxy(const CTotalsProxy & other)
 
- 	: CBonusProxy(other),
 
- 	initialValue(other.initialValue),
 
- 	meleeCachedLast(other.meleeCachedLast),
 
- 	meleeValue(other.meleeValue),
 
- 	rangedCachedLast(other.rangedCachedLast),
 
- 	rangedValue(other.rangedValue)
 
- {
 
- }
 
- int CTotalsProxy::getValue() const
 
- {
 
- 	const auto treeVersion = target->getTreeVersion();
 
- 	if(treeVersion != valueCachedLast)
 
- 	{
 
- 		auto bonuses = getBonusList();
 
- 		value = initialValue + bonuses->totalValue();
 
- 		valueCachedLast = treeVersion;
 
- 	}
 
- 	return value;
 
- }
 
- int CTotalsProxy::getValueAndList(TConstBonusListPtr & outBonusList) const
 
- {
 
- 	const auto treeVersion = target->getTreeVersion();
 
- 	outBonusList = getBonusList();
 
- 	if(treeVersion != valueCachedLast)
 
- 	{
 
- 		value = initialValue + outBonusList->totalValue();
 
- 		valueCachedLast = treeVersion;
 
- 	}
 
- 	return value;
 
- }
 
- int CTotalsProxy::getMeleeValue() const
 
- {
 
- 	static const auto limit = Selector::effectRange()(Bonus::NO_LIMIT).Or(Selector::effectRange()(Bonus::ONLY_MELEE_FIGHT));
 
- 	const auto treeVersion = target->getTreeVersion();
 
- 	if(treeVersion != meleeCachedLast)
 
- 	{
 
- 		auto bonuses = target->getBonuses(selector, limit);
 
- 		meleeValue = initialValue + bonuses->totalValue();
 
- 		meleeCachedLast = treeVersion;
 
- 	}
 
- 	return meleeValue;
 
- }
 
- int CTotalsProxy::getRangedValue() const
 
- {
 
- 	static const auto limit = Selector::effectRange()(Bonus::NO_LIMIT).Or(Selector::effectRange()(Bonus::ONLY_DISTANCE_FIGHT));
 
- 	const auto treeVersion = target->getTreeVersion();
 
- 	if(treeVersion != rangedCachedLast)
 
- 	{
 
- 		auto bonuses = target->getBonuses(selector, limit);
 
- 		rangedValue = initialValue + bonuses->totalValue();
 
- 		rangedCachedLast = treeVersion;
 
- 	}
 
- 	return rangedValue;
 
- }
 
- ///CCheckProxy
 
- CCheckProxy::CCheckProxy(const IBonusBearer * Target, CSelector Selector):
 
- 	target(Target),
 
- 	selector(std::move(Selector)),
 
- 	cachedLast(0),
 
- 	hasBonus(false)
 
- {
 
- }
 
- //This constructor should be placed here to avoid side effects
 
- CCheckProxy::CCheckProxy(const CCheckProxy & other) = default;
 
- bool CCheckProxy::getHasBonus() const
 
- {
 
- 	const auto treeVersion = target->getTreeVersion();
 
- 	if(treeVersion != cachedLast)
 
- 	{
 
- 		hasBonus = target->hasBonus(selector);
 
- 		cachedLast = treeVersion;
 
- 	}
 
- 	return hasBonus;
 
- }
 
- //This constructor should be placed here to avoid side effects
 
- CAddInfo::CAddInfo() = default;
 
- CAddInfo::CAddInfo(si32 value)
 
- {
 
- 	if(value != CAddInfo::NONE)
 
- 		push_back(value);
 
- }
 
- bool CAddInfo::operator==(si32 value) const
 
- {
 
- 	switch(size())
 
- 	{
 
- 	case 0:
 
- 		return value == CAddInfo::NONE;
 
- 	case 1:
 
- 		return operator[](0) == value;
 
- 	default:
 
- 		return false;
 
- 	}
 
- }
 
- bool CAddInfo::operator!=(si32 value) const
 
- {
 
- 	return !operator==(value);
 
- }
 
- si32 & CAddInfo::operator[](size_type pos)
 
- {
 
- 	if(pos >= size())
 
- 		resize(pos + 1, CAddInfo::NONE);
 
- 	return vector::operator[](pos);
 
- }
 
- si32 CAddInfo::operator[](size_type pos) const
 
- {
 
- 	return pos < size() ? vector::operator[](pos) : CAddInfo::NONE;
 
- }
 
- std::string CAddInfo::toString() const
 
- {
 
- 	return toJsonNode().toJson(true);
 
- }
 
- JsonNode CAddInfo::toJsonNode() const
 
- {
 
- 	if(size() < 2)
 
- 	{
 
- 		return JsonUtils::intNode(operator[](0));
 
- 	}
 
- 	else
 
- 	{
 
- 		JsonNode node(JsonNode::JsonType::DATA_VECTOR);
 
- 		for(si32 value : *this)
 
- 			node.Vector().push_back(JsonUtils::intNode(value));
 
- 		return node;
 
- 	}
 
- }
 
- std::atomic<int32_t> CBonusSystemNode::treeChanged(1);
 
- const bool CBonusSystemNode::cachingEnabled = true;
 
- BonusList::BonusList(bool BelongsToTree) : belongsToTree(BelongsToTree)
 
- {
 
- }
 
- BonusList::BonusList(const BonusList & bonusList): belongsToTree(false)
 
- {
 
- 	bonuses.resize(bonusList.size());
 
- 	std::copy(bonusList.begin(), bonusList.end(), bonuses.begin());
 
- }
 
- BonusList::BonusList(BonusList && other) noexcept: belongsToTree(false)
 
- {
 
- 	std::swap(belongsToTree, other.belongsToTree);
 
- 	std::swap(bonuses, other.bonuses);
 
- }
 
- BonusList& BonusList::operator=(const BonusList &bonusList)
 
- {
 
- 	bonuses.resize(bonusList.size());
 
- 	std::copy(bonusList.begin(), bonusList.end(), bonuses.begin());
 
- 	belongsToTree = false;
 
- 	return *this;
 
- }
 
- void BonusList::changed() const
 
- {
 
-     if(belongsToTree)
 
- 		CBonusSystemNode::treeHasChanged();
 
- }
 
- void BonusList::stackBonuses()
 
- {
 
- 	boost::sort(bonuses, [](const std::shared_ptr<Bonus> & b1, const std::shared_ptr<Bonus> & b2) -> bool
 
- 	{
 
- 		if(b1 == b2)
 
- 			return false;
 
- #define COMPARE_ATT(ATT) if(b1->ATT != b2->ATT) return b1->ATT < b2->ATT
 
- 		COMPARE_ATT(stacking);
 
- 		COMPARE_ATT(type);
 
- 		COMPARE_ATT(subtype);
 
- 		COMPARE_ATT(valType);
 
- #undef COMPARE_ATT
 
- 		return b1->val > b2->val;
 
- 	});
 
- 	// remove non-stacking
 
- 	size_t next = 1;
 
- 	while(next < bonuses.size())
 
- 	{
 
- 		bool remove = false;
 
- 		std::shared_ptr<Bonus> last = bonuses[next-1];
 
- 		std::shared_ptr<Bonus> current = bonuses[next];
 
- 		if(current->stacking.empty())
 
- 			remove = current == last;
 
- 		else if(current->stacking == "ALWAYS")
 
- 			remove = false;
 
- 		else
 
- 			remove = current->stacking == last->stacking
 
- 				&& current->type == last->type
 
- 				&& current->subtype == last->subtype
 
- 				&& current->valType == last->valType;
 
- 		if(remove)
 
- 			bonuses.erase(bonuses.begin() + next);
 
- 		else
 
- 			next++;
 
- 	}
 
- }
 
- int BonusList::totalValue() const
 
- {
 
- 	struct BonusCollection
 
- 	{
 
- 		int base = 0;
 
- 		int percentToBase = 0;
 
- 		int percentToAll = 0;
 
- 		int additive = 0;
 
- 		int percentToSource;
 
- 		int indepMin = std::numeric_limits<int>::max();
 
- 		int indepMax = std::numeric_limits<int>::min();
 
- 	};
 
- 	auto percent = [](int base, int percent) -> int {return (base * (100 + percent)) / 100; };
 
- 	std::array <BonusCollection, Bonus::BonusSource::NUM_BONUS_SOURCE> sources = {};
 
- 	BonusCollection any;
 
- 	bool hasIndepMax = false;
 
- 	bool hasIndepMin = false;
 
- 	for(const std::shared_ptr<Bonus> & b : bonuses)
 
- 	{
 
- 		switch(b->valType)
 
- 		{
 
- 		case Bonus::BASE_NUMBER:
 
- 			sources[b->source].base += b->val;
 
- 			break;
 
- 		case Bonus::PERCENT_TO_ALL:
 
- 			sources[b->source].percentToAll += b->val;
 
- 			break;
 
- 		case Bonus::PERCENT_TO_BASE:
 
- 			sources[b->source].percentToBase += b->val;
 
- 			break;
 
- 		case Bonus::PERCENT_TO_SOURCE:
 
- 			sources[b->source].percentToSource += b->val;
 
- 			break;
 
- 		case Bonus::PERCENT_TO_TARGET_TYPE:
 
- 			sources[b->targetSourceType].percentToSource += b->val;
 
- 			break;
 
- 		case Bonus::ADDITIVE_VALUE:
 
- 			sources[b->source].additive += b->val;
 
- 			break;
 
- 		case Bonus::INDEPENDENT_MAX:
 
- 			hasIndepMax = true;
 
- 			vstd::amax(sources[b->source].indepMax, b->val);
 
- 			break;
 
- 		case Bonus::INDEPENDENT_MIN:
 
- 			hasIndepMin = true;
 
- 			vstd::amin(sources[b->source].indepMin, b->val);
 
- 			break;
 
- 		}
 
- 	}
 
- 	for(auto src : sources)
 
- 	{
 
- 		any.base += percent(src.base ,src.percentToSource);
 
- 		any.percentToBase += percent(src.percentToBase, src.percentToSource);
 
- 		any.percentToAll += percent(src.percentToAll, src.percentToSource);
 
- 		any.additive += percent(src.additive, src.percentToSource);
 
- 		if(hasIndepMin)
 
- 			vstd::amin(any.indepMin, percent(src.indepMin, src.percentToSource));
 
- 		if(hasIndepMax)
 
- 			vstd::amax(any.indepMax, percent(src.indepMin, src.percentToSource));
 
- 	}
 
- 	any.base = percent(any.base, any.percentToBase);
 
- 	any.base += any.additive;
 
- 	auto valFirst = percent(any.base ,any.percentToAll);
 
- 	if(hasIndepMin && hasIndepMax)
 
- 		assert(any.indepMin < any.indepMax);
 
- 	const int notIndepBonuses = static_cast<int>(std::count_if(bonuses.cbegin(), bonuses.cend(), [](const std::shared_ptr<Bonus>& b)
 
- 	{
 
- 		return b->valType != Bonus::INDEPENDENT_MAX && b->valType != Bonus::INDEPENDENT_MIN;
 
- 	}));
 
- 	if (hasIndepMax)
 
- 	{
 
- 		if(notIndepBonuses)
 
- 			vstd::amax(valFirst, any.indepMax);
 
- 		else
 
- 			valFirst = any.indepMax;
 
- 	}
 
- 	if (hasIndepMin)
 
- 	{
 
- 		if(notIndepBonuses)
 
- 			vstd::amin(valFirst, any.indepMin);
 
- 		else
 
- 			valFirst = any.indepMin;
 
- 	}
 
- 	return valFirst;
 
- }
 
- std::shared_ptr<Bonus> BonusList::getFirst(const CSelector &select)
 
- {
 
- 	for (auto & b : bonuses)
 
- 	{
 
- 		if(select(b.get()))
 
- 			return b;
 
- 	}
 
- 	return nullptr;
 
- }
 
- std::shared_ptr<const Bonus> BonusList::getFirst(const CSelector &selector) const
 
- {
 
- 	for(const auto & b : bonuses)
 
- 	{
 
- 		if(selector(b.get()))
 
- 			return b;
 
- 	}
 
- 	return nullptr;
 
- }
 
- void BonusList::getBonuses(BonusList & out, const CSelector &selector, const CSelector &limit) const
 
- {
 
- 	out.reserve(bonuses.size());
 
- 	for(const auto & b : bonuses)
 
- 	{
 
- 		//add matching bonuses that matches limit predicate or have NO_LIMIT if no given predicate
 
- 		auto noFightLimit = b->effectRange == Bonus::NO_LIMIT;
 
- 		if(selector(b.get()) && ((!limit && noFightLimit) || ((bool)limit && limit(b.get()))))
 
- 			out.push_back(b);
 
- 	}
 
- }
 
- void BonusList::getAllBonuses(BonusList &out) const
 
- {
 
- 	for(const auto & b : bonuses)
 
- 		out.push_back(b);
 
- }
 
- int BonusList::valOfBonuses(const CSelector &select) const
 
- {
 
- 	BonusList ret;
 
- 	CSelector limit = nullptr;
 
- 	getBonuses(ret, select, limit);
 
- 	return ret.totalValue();
 
- }
 
- JsonNode BonusList::toJsonNode() const
 
- {
 
- 	JsonNode node(JsonNode::JsonType::DATA_VECTOR);
 
- 	for(const std::shared_ptr<Bonus> & b : bonuses)
 
- 		node.Vector().push_back(b->toJsonNode());
 
- 	return node;
 
- }
 
- void BonusList::push_back(const std::shared_ptr<Bonus> & x)
 
- {
 
- 	bonuses.push_back(x);
 
- 	changed();
 
- }
 
- BonusList::TInternalContainer::iterator BonusList::erase(const int position)
 
- {
 
- 	changed();
 
- 	return bonuses.erase(bonuses.begin() + position);
 
- }
 
- void BonusList::clear()
 
- {
 
- 	bonuses.clear();
 
- 	changed();
 
- }
 
- std::vector<BonusList *>::size_type BonusList::operator-=(const std::shared_ptr<Bonus> & i)
 
- {
 
- 	auto itr = std::find(bonuses.begin(), bonuses.end(), i);
 
- 	if(itr == bonuses.end())
 
- 		return false;
 
- 	bonuses.erase(itr);
 
- 	changed();
 
- 	return true;
 
- }
 
- void BonusList::resize(BonusList::TInternalContainer::size_type sz, const std::shared_ptr<Bonus> & c)
 
- {
 
- 	bonuses.resize(sz, c);
 
- 	changed();
 
- }
 
- void BonusList::reserve(TInternalContainer::size_type sz)
 
- {
 
- 	bonuses.reserve(sz);
 
- }
 
- void BonusList::insert(BonusList::TInternalContainer::iterator position, BonusList::TInternalContainer::size_type n, const std::shared_ptr<Bonus> & x)
 
- {
 
- 	bonuses.insert(position, n, x);
 
- 	changed();
 
- }
 
- CSelector IBonusBearer::anaffectedByMoraleSelector =
 
- Selector::type()(Bonus::NON_LIVING)
 
- .Or(Selector::type()(Bonus::UNDEAD))
 
- .Or(Selector::type()(Bonus::SIEGE_WEAPON))
 
- .Or(Selector::type()(Bonus::NO_MORALE));
 
- CSelector IBonusBearer::moraleSelector = Selector::type()(Bonus::MORALE);
 
- CSelector IBonusBearer::luckSelector = Selector::type()(Bonus::LUCK);
 
- IBonusBearer::IBonusBearer()
 
- 	:anaffectedByMorale(this, anaffectedByMoraleSelector),
 
- 	moraleValue(this, moraleSelector, 0),
 
- 	luckValue(this, luckSelector, 0)
 
- {
 
- }
 
- int IBonusBearer::valOfBonuses(Bonus::BonusType type, const CSelector &selector) const
 
- {
 
- 	return valOfBonuses(Selector::type()(type).And(selector));
 
- }
 
- int IBonusBearer::valOfBonuses(Bonus::BonusType type, int subtype) const
 
- {
 
- 	//This part is performance-critical
 
- 	std::string cachingStr = "type_" + std::to_string(static_cast<int>(type)) + "_" + std::to_string(subtype);
 
- 	CSelector s = Selector::type()(type);
 
- 	if(subtype != -1)
 
- 		s = s.And(Selector::subtype()(subtype));
 
- 	return valOfBonuses(s, cachingStr);
 
- }
 
- int IBonusBearer::valOfBonuses(const CSelector &selector, const std::string &cachingStr) const
 
- {
 
- 	CSelector limit = nullptr;
 
- 	TConstBonusListPtr hlp = getAllBonuses(selector, limit, nullptr, cachingStr);
 
- 	return hlp->totalValue();
 
- }
 
- bool IBonusBearer::hasBonus(const CSelector &selector, const std::string &cachingStr) const
 
- {
 
- 	//TODO: We don't need to count all bonuses and could break on first matching
 
- 	return getBonuses(selector, cachingStr)->size() > 0;
 
- }
 
- bool IBonusBearer::hasBonus(const CSelector &selector, const CSelector &limit, const std::string &cachingStr) const
 
- {
 
- 	return getBonuses(selector, limit, cachingStr)->size() > 0;
 
- }
 
- bool IBonusBearer::hasBonusOfType(Bonus::BonusType type, int subtype) const
 
- {
 
- 	//This part is performance-ciritcal
 
- 	std::string cachingStr = "type_" + std::to_string(static_cast<int>(type)) + "_" + std::to_string(subtype);
 
- 	CSelector s = Selector::type()(type);
 
- 	if(subtype != -1)
 
- 		s = s.And(Selector::subtype()(subtype));
 
- 	return hasBonus(s, cachingStr);
 
- }
 
- TConstBonusListPtr IBonusBearer::getBonuses(const CSelector &selector, const std::string &cachingStr) const
 
- {
 
- 	return getAllBonuses(selector, nullptr, nullptr, cachingStr);
 
- }
 
- TConstBonusListPtr IBonusBearer::getBonuses(const CSelector &selector, const CSelector &limit, const std::string &cachingStr) const
 
- {
 
- 	return getAllBonuses(selector, limit, nullptr, cachingStr);
 
- }
 
- bool IBonusBearer::hasBonusFrom(Bonus::BonusSource source, ui32 sourceID) const
 
- {
 
- 	boost::format fmt("source_%did_%d");
 
- 	fmt % static_cast<int>(source) % sourceID;
 
- 	return hasBonus(Selector::source(source,sourceID), fmt.str());
 
- }
 
- int IBonusBearer::MoraleVal() const
 
- {
 
- 	if(anaffectedByMorale.getHasBonus())
 
- 		return 0;
 
- 	int ret = moraleValue.getValue();
 
- 	return vstd::abetween(ret, -3, +3);
 
- }
 
- int IBonusBearer::LuckVal() const
 
- {
 
- 	if(hasBonusOfType(Bonus::NO_LUCK))
 
- 		return 0;
 
- 	int ret = luckValue.getValue();
 
- 	return vstd::abetween(ret, -3, +3);
 
- }
 
- int IBonusBearer::MoraleValAndBonusList(TConstBonusListPtr & bonusList) const
 
- {
 
- 	if(anaffectedByMorale.getHasBonus())
 
- 	{
 
- 		if(!bonusList->empty())
 
- 			bonusList = std::make_shared<const BonusList>();
 
- 		return 0;
 
- 	}
 
- 	int ret = moraleValue.getValueAndList(bonusList);
 
- 	return vstd::abetween(ret, -3, +3);
 
- }
 
- int IBonusBearer::LuckValAndBonusList(TConstBonusListPtr & bonusList) const
 
- {
 
- 	if(hasBonusOfType(Bonus::NO_LUCK))
 
- 	{
 
- 		if(!bonusList->empty())
 
- 			bonusList = std::make_shared<const BonusList>();
 
- 		return 0;
 
- 	}
 
- 	int ret = luckValue.getValueAndList(bonusList);
 
- 	return vstd::abetween(ret, -3, +3);
 
- }
 
- ui32 IBonusBearer::MaxHealth() const
 
- {
 
- 	const std::string cachingStr = "type_STACK_HEALTH";
 
- 	static const auto selector = Selector::type()(Bonus::STACK_HEALTH);
 
- 	auto value = valOfBonuses(selector, cachingStr);
 
- 	return std::max(1, value); //never 0
 
- }
 
- int IBonusBearer::getAttack(bool ranged) const
 
- {
 
- 	const std::string cachingStr = "type_PRIMARY_SKILLs_ATTACK";
 
- 	static const auto selector = Selector::typeSubtype(Bonus::PRIMARY_SKILL, PrimarySkill::ATTACK);
 
- 	return getBonuses(selector, nullptr, cachingStr)->totalValue();
 
- }
 
- int IBonusBearer::getDefense(bool ranged) const
 
- {
 
- 	const std::string cachingStr = "type_PRIMARY_SKILLs_DEFENSE";
 
- 	static const auto selector = Selector::typeSubtype(Bonus::PRIMARY_SKILL, PrimarySkill::DEFENSE);
 
- 	return getBonuses(selector, nullptr, cachingStr)->totalValue();
 
- }
 
- int IBonusBearer::getMinDamage(bool ranged) const
 
- {
 
- 	const std::string cachingStr = "type_CREATURE_DAMAGEs_0Otype_CREATURE_DAMAGEs_1";
 
- 	static const auto selector = Selector::typeSubtype(Bonus::CREATURE_DAMAGE, 0).Or(Selector::typeSubtype(Bonus::CREATURE_DAMAGE, 1));
 
- 	return valOfBonuses(selector, cachingStr);
 
- }
 
- int IBonusBearer::getMaxDamage(bool ranged) const
 
- {
 
- 	const std::string cachingStr = "type_CREATURE_DAMAGEs_0Otype_CREATURE_DAMAGEs_2";
 
- 	static const auto selector = Selector::typeSubtype(Bonus::CREATURE_DAMAGE, 0).Or(Selector::typeSubtype(Bonus::CREATURE_DAMAGE, 2));
 
- 	return valOfBonuses(selector, cachingStr);
 
- }
 
- si32 IBonusBearer::manaLimit() const
 
- {
 
- 	return 0;
 
- }
 
- int IBonusBearer::getPrimSkillLevel(PrimarySkill::PrimarySkill id) const
 
- {
 
- 	static const CSelector selectorAllSkills = Selector::type()(Bonus::PRIMARY_SKILL);
 
- 	static const std::string keyAllSkills = "type_PRIMARY_SKILL";
 
- 	auto allSkills = getBonuses(selectorAllSkills, keyAllSkills);
 
- 	auto ret = allSkills->valOfBonuses(Selector::subtype()(id));
 
- 	auto minSkillValue = (id == PrimarySkill::SPELL_POWER || id == PrimarySkill::KNOWLEDGE) ? 1 : 0;
 
- 	vstd::amax(ret, minSkillValue); //otherwise, some artifacts may cause negative skill value effect
 
- 	return ret; //sp=0 works in old saves
 
- }
 
- si32 IBonusBearer::magicResistance() const
 
- {
 
- 	return valOfBonuses(Bonus::MAGIC_RESISTANCE);
 
- }
 
- ui32 IBonusBearer::Speed(int turn, bool useBind) const
 
- {
 
- 	//war machines cannot move
 
- 	if(hasBonus(Selector::type()(Bonus::SIEGE_WEAPON).And(Selector::turns(turn))))
 
- 	{
 
- 		return 0;
 
- 	}
 
- 	//bind effect check - doesn't influence stack initiative
 
- 	if(useBind && hasBonus(Selector::type()(Bonus::BIND_EFFECT).And(Selector::turns(turn))))
 
- 	{
 
- 		return 0;
 
- 	}
 
- 	return valOfBonuses(Selector::type()(Bonus::STACKS_SPEED).And(Selector::turns(turn)));
 
- }
 
- bool IBonusBearer::isLiving() const //TODO: theoreticaly there exists "LIVING" bonus in stack experience documentation
 
- {
 
- 	static const std::string cachingStr = "IBonusBearer::isLiving";
 
- 	static const CSelector selector = Selector::type()(Bonus::UNDEAD)
 
- 		.Or(Selector::type()(Bonus::NON_LIVING))
 
- 		.Or(Selector::type()(Bonus::GARGOYLE))
 
- 		.Or(Selector::type()(Bonus::SIEGE_WEAPON));
 
- 	return !hasBonus(selector, cachingStr);
 
- }
 
- std::shared_ptr<const Bonus> IBonusBearer::getBonus(const CSelector &selector) const
 
- {
 
- 	auto bonuses = getAllBonuses(selector, Selector::all);
 
- 	return bonuses->getFirst(Selector::all);
 
- }
 
- const CStack * retrieveStackBattle(const CBonusSystemNode * node)
 
- {
 
- 	switch(node->getNodeType())
 
- 	{
 
- 	case CBonusSystemNode::STACK_BATTLE:
 
- 		return dynamic_cast<const CStack *>(node);
 
- 	default:
 
- 		return nullptr;
 
- 	}
 
- }
 
- const CStackInstance * retrieveStackInstance(const CBonusSystemNode * node)
 
- {
 
- 	switch(node->getNodeType())
 
- 	{
 
- 	case CBonusSystemNode::STACK_INSTANCE:
 
- 		return (dynamic_cast<const CStackInstance *>(node));
 
- 	case CBonusSystemNode::STACK_BATTLE:
 
- 		return (dynamic_cast<const CStack *>(node))->base;
 
- 	default:
 
- 		return nullptr;
 
- 	}
 
- }
 
- PlayerColor CBonusSystemNode::retrieveNodeOwner(const CBonusSystemNode * node)
 
- {
 
- 	return node ? node->getOwner() : PlayerColor::CANNOT_DETERMINE;
 
- }
 
- std::shared_ptr<Bonus> CBonusSystemNode::getBonusLocalFirst(const CSelector & selector)
 
- {
 
- 	auto ret = bonuses.getFirst(selector);
 
- 	if(ret)
 
- 		return ret;
 
- 	FOREACH_PARENT(pname)
 
- 	{
 
- 		ret = pname->getBonusLocalFirst(selector);
 
- 		if (ret)
 
- 			return ret;
 
- 	}
 
- 	return nullptr;
 
- }
 
- std::shared_ptr<const Bonus> CBonusSystemNode::getBonusLocalFirst(const CSelector & selector) const
 
- {
 
- 	return (const_cast<CBonusSystemNode*>(this))->getBonusLocalFirst(selector);
 
- }
 
- void CBonusSystemNode::getParents(TCNodes & out) const /*retrieves list of parent nodes (nodes to inherit bonuses from) */
 
- {
 
- 	for(const auto & elem : parents)
 
- 	{
 
- 		const CBonusSystemNode *parent = elem;
 
- 		out.insert(parent);
 
- 	}
 
- }
 
- void CBonusSystemNode::getParents(TNodes &out)
 
- {
 
- 	for (auto & elem : parents)
 
- 	{
 
- 		const CBonusSystemNode *parent = elem;
 
- 		out.insert(const_cast<CBonusSystemNode*>(parent));
 
- 	}
 
- }
 
- void CBonusSystemNode::getAllParents(TCNodes & out) const //retrieves list of parent nodes (nodes to inherit bonuses from)
 
- {
 
- 	for(auto * parent : parents)
 
- 	{
 
- 		out.insert(parent);
 
- 		parent->getAllParents(out);
 
- 	}
 
- }
 
- void CBonusSystemNode::getAllBonusesRec(BonusList &out, const CSelector & selector) const
 
- {
 
- 	//out has been reserved sufficient capacity at getAllBonuses() call
 
- 	BonusList beforeUpdate;
 
- 	TCNodes lparents;
 
- 	getAllParents(lparents);
 
- 	if(!lparents.empty())
 
- 	{
 
- 		//estimate on how many bonuses are missing yet - must be positive
 
- 		beforeUpdate.reserve(std::max(out.capacity() - out.size(), bonuses.size()));
 
- 	}
 
- 	else
 
- 	{
 
- 		beforeUpdate.reserve(bonuses.size()); //at most all local bonuses
 
- 	}
 
- 	for(const auto * parent : lparents)
 
- 	{
 
- 		parent->getAllBonusesRec(beforeUpdate, selector);
 
- 	}
 
- 	bonuses.getAllBonuses(beforeUpdate);
 
- 	for(const auto & b : beforeUpdate)
 
- 	{
 
- 		//We should not run updaters on non-selected bonuses
 
- 		auto updated = selector(b.get()) && b->updater
 
- 			? getUpdatedBonus(b, b->updater)
 
- 			: b;
 
- 		//do not add bonus with updater
 
- 		bool bonusExists = false;
 
- 		for(const auto & bonus : out)
 
- 		{
 
- 			if (bonus == updated)
 
- 				bonusExists = true;
 
- 			if (bonus->updater && bonus->updater == updated->updater)
 
- 				bonusExists = true;
 
- 		}
 
- 		if (!bonusExists)
 
- 			out.push_back(updated);
 
- 	}
 
- }
 
- TConstBonusListPtr CBonusSystemNode::getAllBonuses(const CSelector &selector, const CSelector &limit, const CBonusSystemNode *root, const std::string &cachingStr) const
 
- {
 
- 	bool limitOnUs = (!root || root == this); //caching won't work when we want to limit bonuses against an external node
 
- 	if (CBonusSystemNode::cachingEnabled && limitOnUs)
 
- 	{
 
- 		// Exclusive access for one thread
 
- 		boost::lock_guard<boost::mutex> lock(sync);
 
- 		// If the bonus system tree changes(state of a single node or the relations to each other) then
 
- 		// cache all bonus objects. Selector objects doesn't matter.
 
- 		if (cachedLast != treeChanged)
 
- 		{
 
- 			BonusList allBonuses;
 
- 			allBonuses.reserve(cachedBonuses.capacity()); //we assume we'll get about the same number of bonuses
 
- 			cachedBonuses.clear();
 
- 			cachedRequests.clear();
 
- 			getAllBonusesRec(allBonuses, Selector::all);
 
- 			limitBonuses(allBonuses, cachedBonuses);
 
- 			cachedBonuses.stackBonuses();
 
- 			cachedLast = treeChanged;
 
- 		}
 
- 		// If a bonus system request comes with a caching string then look up in the map if there are any
 
- 		// pre-calculated bonus results. Limiters can't be cached so they have to be calculated.
 
- 		if(!cachingStr.empty())
 
- 		{
 
- 			auto it = cachedRequests.find(cachingStr);
 
- 			if(it != cachedRequests.end())
 
- 			{
 
- 				//Cached list contains bonuses for our query with applied limiters
 
- 				return it->second;
 
- 			}
 
- 		}
 
- 		//We still don't have the bonuses (didn't returned them from cache)
 
- 		//Perform bonus selection
 
- 		auto ret = std::make_shared<BonusList>();
 
- 		cachedBonuses.getBonuses(*ret, selector, limit);
 
- 		// Save the results in the cache
 
- 		if(!cachingStr.empty())
 
- 			cachedRequests[cachingStr] = ret;
 
- 		return ret;
 
- 	}
 
- 	else
 
- 	{
 
- 		return getAllBonusesWithoutCaching(selector, limit, root);
 
- 	}
 
- }
 
- TConstBonusListPtr CBonusSystemNode::getAllBonusesWithoutCaching(const CSelector &selector, const CSelector &limit, const CBonusSystemNode *root) const
 
- {
 
- 	auto ret = std::make_shared<BonusList>();
 
- 	// Get bonus results without caching enabled.
 
- 	BonusList beforeLimiting;
 
- 	BonusList afterLimiting;
 
- 	getAllBonusesRec(beforeLimiting, selector);
 
- 	if(!root || root == this)
 
- 	{
 
- 		limitBonuses(beforeLimiting, afterLimiting);
 
- 	}
 
- 	else if(root)
 
- 	{
 
- 		//We want to limit our query against an external node. We get all its bonuses,
 
- 		// add the ones we're considering and see if they're cut out by limiters
 
- 		BonusList rootBonuses;
 
- 		BonusList limitedRootBonuses;
 
- 		getAllBonusesRec(rootBonuses, selector);
 
- 		for(const auto & b : beforeLimiting)
 
- 			rootBonuses.push_back(b);
 
- 		root->limitBonuses(rootBonuses, limitedRootBonuses);
 
- 		for(const auto & b : beforeLimiting)
 
- 			if(vstd::contains(limitedRootBonuses, b))
 
- 				afterLimiting.push_back(b);
 
- 	}
 
- 	afterLimiting.getBonuses(*ret, selector, limit);
 
- 	ret->stackBonuses();
 
- 	return ret;
 
- }
 
- std::shared_ptr<Bonus> CBonusSystemNode::getUpdatedBonus(const std::shared_ptr<Bonus> & b, const TUpdaterPtr & updater) const
 
- {
 
- 	assert(updater);
 
- 	return updater->createUpdatedBonus(b, * this);
 
- }
 
- CBonusSystemNode::CBonusSystemNode()
 
- 	:CBonusSystemNode(false)
 
- {
 
- }
 
- CBonusSystemNode::CBonusSystemNode(bool isHypotetic):
 
- 	bonuses(true),
 
- 	exportedBonuses(true),
 
- 	nodeType(UNKNOWN),
 
- 	cachedLast(0),
 
- 	isHypotheticNode(isHypotetic)
 
- {
 
- }
 
- CBonusSystemNode::CBonusSystemNode(ENodeTypes NodeType):
 
- 	bonuses(true),
 
- 	exportedBonuses(true),
 
- 	nodeType(NodeType),
 
- 	cachedLast(0),
 
- 	isHypotheticNode(false)
 
- {
 
- }
 
- CBonusSystemNode::CBonusSystemNode(CBonusSystemNode && other) noexcept:
 
- 	bonuses(std::move(other.bonuses)),
 
- 	exportedBonuses(std::move(other.exportedBonuses)),
 
- 	nodeType(other.nodeType),
 
- 	description(other.description),
 
- 	cachedLast(0),
 
- 	isHypotheticNode(other.isHypotheticNode)
 
- {
 
- 	std::swap(parents, other.parents);
 
- 	std::swap(children, other.children);
 
- 	//fixing bonus tree without recalculation
 
- 	if(!isHypothetic())
 
- 	{
 
- 		for(CBonusSystemNode * n : parents)
 
- 		{
 
- 			n->children -= &other;
 
- 			n->children.push_back(this);
 
- 		}
 
- 	}
 
- 	for(CBonusSystemNode * n : children)
 
- 	{
 
- 		n->parents -= &other;
 
- 		n->parents.push_back(this);
 
- 	}
 
- 	//cache ignored
 
- 	//cachedBonuses
 
- 	//cachedRequests
 
- }
 
- CBonusSystemNode::~CBonusSystemNode()
 
- {
 
- 	detachFromAll();
 
- 	if(!children.empty())
 
- 	{
 
- 		while(!children.empty())
 
- 			children.front()->detachFrom(*this);
 
- 	}
 
- }
 
- void CBonusSystemNode::attachTo(CBonusSystemNode & parent)
 
- {
 
- 	assert(!vstd::contains(parents, &parent));
 
- 	parents.push_back(&parent);
 
- 	if(!isHypothetic())
 
- 	{
 
- 		if(parent.actsAsBonusSourceOnly())
 
- 			parent.newRedDescendant(*this);
 
- 		else
 
- 			newRedDescendant(parent);
 
- 		parent.newChildAttached(*this);
 
- 	}
 
- 	CBonusSystemNode::treeHasChanged();
 
- }
 
- void CBonusSystemNode::detachFrom(CBonusSystemNode & parent)
 
- {
 
- 	assert(vstd::contains(parents, &parent));
 
- 	if(!isHypothetic())
 
- 	{
 
- 		if(parent.actsAsBonusSourceOnly())
 
- 			parent.removedRedDescendant(*this);
 
- 		else
 
- 			removedRedDescendant(parent);
 
- 	}
 
- 	if (vstd::contains(parents, &parent))
 
- 	{
 
- 		parents -= &parent;
 
- 	}
 
- 	else
 
- 	{
 
- 		logBonus->error("Error on Detach. Node %s (nodeType=%d) has not parent %s (nodeType=%d)"
 
- 			, nodeShortInfo(), nodeType, parent.nodeShortInfo(), parent.nodeType);
 
- 	}
 
- 	if(!isHypothetic())
 
- 	{
 
- 		parent.childDetached(*this);
 
- 	}
 
- 	CBonusSystemNode::treeHasChanged();
 
- }
 
- void CBonusSystemNode::removeBonusesRecursive(const CSelector & s)
 
- {
 
- 	removeBonuses(s);
 
- 	for(CBonusSystemNode * child : children)
 
- 		child->removeBonusesRecursive(s);
 
- }
 
- void CBonusSystemNode::reduceBonusDurations(const CSelector &s)
 
- {
 
- 	BonusList bl;
 
- 	exportedBonuses.getBonuses(bl, s, Selector::all);
 
- 	for(const auto & b : bl)
 
- 	{
 
- 		b->turnsRemain--;
 
- 		if(b->turnsRemain <= 0)
 
- 			removeBonus(b);
 
- 	}
 
- 	for(CBonusSystemNode *child : children)
 
- 		child->reduceBonusDurations(s);
 
- }
 
- void CBonusSystemNode::addNewBonus(const std::shared_ptr<Bonus>& b)
 
- {
 
- 	//turnsRemain shouldn't be zero for following durations
 
- 	if(Bonus::NTurns(b.get()) || Bonus::NDays(b.get()) || Bonus::OneWeek(b.get()))
 
- 	{
 
- 		assert(b->turnsRemain);
 
- 	}
 
- 	assert(!vstd::contains(exportedBonuses, b));
 
- 	exportedBonuses.push_back(b);
 
- 	exportBonus(b);
 
- 	CBonusSystemNode::treeHasChanged();
 
- }
 
- void CBonusSystemNode::accumulateBonus(const std::shared_ptr<Bonus>& b)
 
- {
 
- 	auto bonus = exportedBonuses.getFirst(Selector::typeSubtype(b->type, b->subtype)); //only local bonuses are interesting //TODO: what about value type?
 
- 	if(bonus)
 
- 		bonus->val += b->val;
 
- 	else
 
- 		addNewBonus(std::make_shared<Bonus>(*b)); //duplicate needed, original may get destroyed
 
- }
 
- void CBonusSystemNode::removeBonus(const std::shared_ptr<Bonus>& b)
 
- {
 
- 	exportedBonuses -= b;
 
- 	if(b->propagator)
 
- 		unpropagateBonus(b);
 
- 	else
 
- 		bonuses -= b;
 
- 	CBonusSystemNode::treeHasChanged();
 
- }
 
- void CBonusSystemNode::removeBonuses(const CSelector & selector)
 
- {
 
- 	BonusList toRemove;
 
- 	exportedBonuses.getBonuses(toRemove, selector, Selector::all);
 
- 	for(const auto & bonus : toRemove)
 
- 		removeBonus(bonus);
 
- }
 
- bool CBonusSystemNode::actsAsBonusSourceOnly() const
 
- {
 
- 	switch(nodeType)
 
- 	{
 
- 	case CREATURE:
 
- 	case ARTIFACT:
 
- 	case ARTIFACT_INSTANCE:
 
- 		return true;
 
- 	default:
 
- 		return false;
 
- 	}
 
- }
 
- void CBonusSystemNode::propagateBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & source)
 
- {
 
- 	if(b->propagator->shouldBeAttached(this))
 
- 	{
 
- 		auto propagated = b->propagationUpdater 
 
- 			? source.getUpdatedBonus(b, b->propagationUpdater)
 
- 			: b;
 
- 		bonuses.push_back(propagated);
 
- 		logBonus->trace("#$# %s #propagated to# %s",  propagated->Description(), nodeName());
 
- 	}
 
- 	FOREACH_RED_CHILD(child)
 
- 		child->propagateBonus(b, source);
 
- }
 
- void CBonusSystemNode::unpropagateBonus(const std::shared_ptr<Bonus> & b)
 
- {
 
- 	if(b->propagator->shouldBeAttached(this))
 
- 	{
 
- 		bonuses -= b;
 
- 		logBonus->trace("#$# %s #is no longer propagated to# %s",  b->Description(), nodeName());
 
- 	}
 
- 	FOREACH_RED_CHILD(child)
 
- 		child->unpropagateBonus(b);
 
- }
 
- void CBonusSystemNode::newChildAttached(CBonusSystemNode & child)
 
- {
 
- 	assert(!vstd::contains(children, &child));
 
- 	children.push_back(&child);
 
- }
 
- void CBonusSystemNode::childDetached(CBonusSystemNode & child)
 
- {
 
- 	if(vstd::contains(children, &child))
 
- 		children -= &child;
 
- 	else
 
- 	{
 
- 		logBonus->error("Error on Detach. Node %s (nodeType=%d) is not a child of %s (nodeType=%d)"
 
- 			, child.nodeShortInfo(), child.nodeType, nodeShortInfo(), nodeType);
 
- 	}
 
- }
 
- void CBonusSystemNode::detachFromAll()
 
- {
 
- 	while(!parents.empty())
 
- 		detachFrom(*parents.front());
 
- }
 
- bool CBonusSystemNode::isIndependentNode() const
 
- {
 
- 	return parents.empty() && children.empty();
 
- }
 
- std::string CBonusSystemNode::nodeName() const
 
- {
 
- 	return !description.empty()
 
- 		? description
 
- 		: std::string("Bonus system node of type ") + typeid(*this).name();
 
- }
 
- std::string CBonusSystemNode::nodeShortInfo() const
 
- {
 
- 	std::ostringstream str;
 
- 	str << "'" << typeid(* this).name() << "'";
 
- 	description.length() > 0 
 
- 		? str << " (" << description << ")"
 
- 		: str << " (no description)";
 
- 	return str.str();
 
- }
 
- void CBonusSystemNode::deserializationFix()
 
- {
 
- 	exportBonuses();
 
- }
 
- void CBonusSystemNode::getRedParents(TNodes & out)
 
- {
 
- 	FOREACH_PARENT(pname)
 
- 	{
 
- 		if(pname->actsAsBonusSourceOnly())
 
- 		{
 
- 			out.insert(pname);
 
- 		}
 
- 	}
 
- 	if(!actsAsBonusSourceOnly())
 
- 	{
 
- 		for(CBonusSystemNode *child : children)
 
- 		{
 
- 			out.insert(child);
 
- 		}
 
- 	}
 
- }
 
- void CBonusSystemNode::getRedChildren(TNodes &out)
 
- {
 
- 	FOREACH_PARENT(pname)
 
- 	{
 
- 		if(!pname->actsAsBonusSourceOnly())
 
- 		{
 
- 			out.insert(pname);
 
- 		}
 
- 	}
 
- 	if(actsAsBonusSourceOnly())
 
- 	{
 
- 		for(CBonusSystemNode *child : children)
 
- 		{
 
- 			out.insert(child);
 
- 		}
 
- 	}
 
- }
 
- void CBonusSystemNode::newRedDescendant(CBonusSystemNode & descendant)
 
- {
 
- 	for(const auto & b : exportedBonuses)
 
- 	{
 
- 		if(b->propagator)
 
- 			descendant.propagateBonus(b, *this);
 
- 	}
 
- 	TNodes redParents;
 
- 	getRedAncestors(redParents); //get all red parents recursively
 
- 	for(auto * parent : redParents)
 
- 	{
 
- 		for(const auto & b : parent->exportedBonuses)
 
- 		{
 
- 			if(b->propagator)
 
- 				descendant.propagateBonus(b, *this);
 
- 		}
 
- 	}
 
- }
 
- void CBonusSystemNode::removedRedDescendant(CBonusSystemNode & descendant)
 
- {
 
- 	for(const auto & b : exportedBonuses)
 
- 		if(b->propagator)
 
- 			descendant.unpropagateBonus(b);
 
- 	TNodes redParents;
 
- 	getRedAncestors(redParents); //get all red parents recursively
 
- 	for(auto * parent : redParents)
 
- 	{
 
- 		for(const auto & b : parent->exportedBonuses)
 
- 			if(b->propagator)
 
- 				descendant.unpropagateBonus(b);
 
- 	}
 
- }
 
- void CBonusSystemNode::getRedAncestors(TNodes &out)
 
- {
 
- 	getRedParents(out);
 
- 	TNodes redParents; 
 
- 	getRedParents(redParents);
 
- 	for(CBonusSystemNode * parent : redParents)
 
- 		parent->getRedAncestors(out);
 
- }
 
- void CBonusSystemNode::getRedDescendants(TNodes &out)
 
- {
 
- 	getRedChildren(out);
 
- 	FOREACH_RED_CHILD(c)
 
- 		c->getRedChildren(out);
 
- }
 
- void CBonusSystemNode::exportBonus(const std::shared_ptr<Bonus> & b)
 
- {
 
- 	if(b->propagator)
 
- 		propagateBonus(b, *this);
 
- 	else
 
- 		bonuses.push_back(b);
 
- 	CBonusSystemNode::treeHasChanged();
 
- }
 
- void CBonusSystemNode::exportBonuses()
 
- {
 
- 	for(const auto & b : exportedBonuses)
 
- 		exportBonus(b);
 
- }
 
- CBonusSystemNode::ENodeTypes CBonusSystemNode::getNodeType() const
 
- {
 
- 	return nodeType;
 
- }
 
- const BonusList& CBonusSystemNode::getBonusList() const
 
- {
 
- 	return bonuses;
 
- }
 
- const TNodesVector& CBonusSystemNode::getParentNodes() const
 
- {
 
- 	return parents;
 
- }
 
- const TNodesVector& CBonusSystemNode::getChildrenNodes() const
 
- {
 
- 	return children;
 
- }
 
- void CBonusSystemNode::setNodeType(CBonusSystemNode::ENodeTypes type)
 
- {
 
- 	nodeType = type;
 
- }
 
- BonusList & CBonusSystemNode::getExportedBonusList()
 
- {
 
- 	return exportedBonuses;
 
- }
 
- const BonusList & CBonusSystemNode::getExportedBonusList() const
 
- {
 
- 	return exportedBonuses;
 
- }
 
- const std::string& CBonusSystemNode::getDescription() const
 
- {
 
- 	return description;
 
- }
 
- void CBonusSystemNode::setDescription(const std::string &description)
 
- {
 
- 	this->description = description;
 
- }
 
- void CBonusSystemNode::limitBonuses(const BonusList &allBonuses, BonusList &out) const
 
- {
 
- 	assert(&allBonuses != &out); //todo should it work in-place?
 
- 	BonusList undecided = allBonuses;
 
- 	BonusList & accepted = out;
 
- 	while(true)
 
- 	{
 
- 		int undecidedCount = static_cast<int>(undecided.size());
 
- 		for(int i = 0; i < undecided.size(); i++)
 
- 		{
 
- 			auto b = undecided[i];
 
- 			BonusLimitationContext context = {b, *this, out, undecided};
 
- 			int decision = b->limiter ? b->limiter->limit(context) : ILimiter::ACCEPT; //bonuses without limiters will be accepted by default
 
- 			if(decision == ILimiter::DISCARD)
 
- 			{
 
- 				undecided.erase(i);
 
- 				i--; continue;
 
- 			}
 
- 			else if(decision == ILimiter::ACCEPT)
 
- 			{
 
- 				accepted.push_back(b);
 
- 				undecided.erase(i);
 
- 				i--; continue;
 
- 			}
 
- 			else
 
- 				assert(decision == ILimiter::NOT_SURE);
 
- 		}
 
- 		if(undecided.size() == undecidedCount) //we haven't moved a single bonus -> limiters reached a stable state
 
- 			return;
 
- 	}
 
- }
 
- TBonusListPtr CBonusSystemNode::limitBonuses(const BonusList &allBonuses) const
 
- {
 
- 	auto ret = std::make_shared<BonusList>();
 
- 	limitBonuses(allBonuses, *ret);
 
- 	return ret;
 
- }
 
- void CBonusSystemNode::treeHasChanged()
 
- {
 
- 	treeChanged++;
 
- }
 
- int64_t CBonusSystemNode::getTreeVersion() const
 
- {
 
- 	int64_t ret = treeChanged;
 
- 	return ret << 32;
 
- }
 
- int NBonus::valOf(const CBonusSystemNode *obj, Bonus::BonusType type, int subtype)
 
- {
 
- 	if(obj)
 
- 		return obj->valOfBonuses(type, subtype);
 
- 	return 0;
 
- }
 
- bool NBonus::hasOfType(const CBonusSystemNode *obj, Bonus::BonusType type, int subtype)
 
- {
 
- 	if(obj)
 
- 		return obj->hasBonusOfType(type, subtype);
 
- 	return false;
 
- }
 
- std::string Bonus::Description(boost::optional<si32> customValue) const
 
- {
 
- 	std::ostringstream str;
 
- 	if(description.empty())
 
- 	{
 
- 		if(stacking.empty() || stacking == "ALWAYS")
 
- 		{
 
- 			switch(source)
 
- 			{
 
- 			case ARTIFACT:
 
- 				str << ArtifactID(sid).toArtifact(VLC->artifacts())->getNameTranslated();
 
- 				break;
 
- 			case SPELL_EFFECT:
 
- 				str << SpellID(sid).toSpell(VLC->spells())->getNameTranslated();
 
- 				break;
 
- 			case CREATURE_ABILITY:
 
- 				str << VLC->creh->objects[sid]->getNamePluralTranslated();
 
- 				break;
 
- 			case SECONDARY_SKILL:
 
- 				str << VLC->skillh->getByIndex(sid)->getNameTranslated();
 
- 				break;
 
- 			case HERO_SPECIAL:
 
- 				str << VLC->heroh->objects[sid]->getNameTranslated();
 
- 				break;
 
- 			default:
 
- 				//todo: handle all possible sources
 
- 				str << "Unknown";
 
- 				break;
 
- 			}
 
- 		}
 
- 		else
 
- 			str << stacking;
 
- 	}
 
- 	else
 
- 	{
 
- 		str << description;
 
- 	}
 
- 	if(auto value = customValue.value_or(val))
 
- 		str << " " << std::showpos << value;
 
- 	return str.str();
 
- }
 
- JsonNode subtypeToJson(Bonus::BonusType type, int subtype)
 
- {
 
- 	switch(type)
 
- 	{
 
- 	case Bonus::PRIMARY_SKILL:
 
- 		return JsonUtils::stringNode("primSkill." + PrimarySkill::names[subtype]);
 
- 	case Bonus::SPECIAL_SPELL_LEV:
 
- 	case Bonus::SPECIFIC_SPELL_DAMAGE:
 
- 	case Bonus::SPELL:
 
- 	case Bonus::SPECIAL_PECULIAR_ENCHANT:
 
- 	case Bonus::SPECIAL_ADD_VALUE_ENCHANT:
 
- 	case Bonus::SPECIAL_FIXED_VALUE_ENCHANT:
 
- 		return JsonUtils::stringNode(CModHandler::makeFullIdentifier("", "spell", SpellID::encode(subtype)));
 
- 	case Bonus::IMPROVED_NECROMANCY:
 
- 	case Bonus::SPECIAL_UPGRADE:
 
- 		return JsonUtils::stringNode(CModHandler::makeFullIdentifier("", "creature", CreatureID::encode(subtype)));
 
- 	case Bonus::GENERATE_RESOURCE:
 
- 		return JsonUtils::stringNode("resource." + GameConstants::RESOURCE_NAMES[subtype]);
 
- 	default:
 
- 		return JsonUtils::intNode(subtype);
 
- 	}
 
- }
 
- JsonNode additionalInfoToJson(Bonus::BonusType type, CAddInfo addInfo)
 
- {
 
- 	switch(type)
 
- 	{
 
- 	case Bonus::SPECIAL_UPGRADE:
 
- 		return JsonUtils::stringNode(CModHandler::makeFullIdentifier("", "creature", CreatureID::encode(addInfo[0])));
 
- 	default:
 
- 		return addInfo.toJsonNode();
 
- 	}
 
- }
 
- JsonNode durationToJson(ui16 duration)
 
- {
 
- 	std::vector<std::string> durationNames;
 
- 	for(ui16 durBit = 1; durBit; durBit = durBit << 1)
 
- 	{
 
- 		if(duration & durBit)
 
- 			durationNames.push_back(vstd::findKey(bonusDurationMap, durBit));
 
- 	}
 
- 	if(durationNames.size() == 1)
 
- 	{
 
- 		return JsonUtils::stringNode(durationNames[0]);
 
- 	}
 
- 	else
 
- 	{
 
- 		JsonNode node(JsonNode::JsonType::DATA_VECTOR);
 
- 		for(const std::string & dur : durationNames)
 
- 			node.Vector().push_back(JsonUtils::stringNode(dur));
 
- 		return node;
 
- 	}
 
- }
 
- JsonNode Bonus::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	// only add values that might reasonably be found in config files
 
- 	root["type"].String() = vstd::findKey(bonusNameMap, type);
 
- 	if(subtype != -1)
 
- 		root["subtype"] = subtypeToJson(type, subtype);
 
- 	if(additionalInfo != CAddInfo::NONE)
 
- 		root["addInfo"] = additionalInfoToJson(type, additionalInfo);
 
- 	if(duration != 0)
 
- 	{
 
- 		JsonNode durationVec(JsonNode::JsonType::DATA_VECTOR);
 
- 		for(const auto & kv : bonusDurationMap)
 
- 		{
 
- 			if(duration & kv.second)
 
- 				durationVec.Vector().push_back(JsonUtils::stringNode(kv.first));
 
- 		}
 
- 		root["duration"] = durationVec;
 
- 	}
 
- 	if(turnsRemain != 0)
 
- 		root["turns"].Integer() = turnsRemain;
 
- 	if(source != OTHER)
 
- 		root["sourceType"].String() = vstd::findKey(bonusSourceMap, source);
 
- 	if(targetSourceType != OTHER)
 
- 		root["targetSourceType"].String() = vstd::findKey(bonusSourceMap, targetSourceType);
 
- 	if(sid != 0)
 
- 		root["sourceID"].Integer() = sid;
 
- 	if(val != 0)
 
- 		root["val"].Integer() = val;
 
- 	if(valType != ADDITIVE_VALUE)
 
- 		root["valueType"].String() = vstd::findKey(bonusValueMap, valType);
 
- 	if(!stacking.empty())
 
- 		root["stacking"].String() = stacking;
 
- 	if(!description.empty())
 
- 		root["description"].String() = description;
 
- 	if(effectRange != NO_LIMIT)
 
- 		root["effectRange"].String() = vstd::findKey(bonusLimitEffect, effectRange);
 
- 	if(duration != PERMANENT)
 
- 		root["duration"] = durationToJson(duration);
 
- 	if(turnsRemain)
 
- 		root["turns"].Integer() = turnsRemain;
 
- 	if(limiter)
 
- 		root["limiters"] = limiter->toJsonNode();
 
- 	if(updater)
 
- 		root["updater"] = updater->toJsonNode();
 
- 	if(propagator)
 
- 		root["propagator"].String() = vstd::findKey(bonusPropagatorMap, propagator);
 
- 	return root;
 
- }
 
- std::string Bonus::nameForBonus() const
 
- {
 
- 	switch(type)
 
- 	{
 
- 	case Bonus::PRIMARY_SKILL:
 
- 		return PrimarySkill::names[subtype];
 
- 	case Bonus::SPECIAL_SPELL_LEV:
 
- 	case Bonus::SPECIFIC_SPELL_DAMAGE:
 
- 	case Bonus::SPELL:
 
- 	case Bonus::SPECIAL_PECULIAR_ENCHANT:
 
- 	case Bonus::SPECIAL_ADD_VALUE_ENCHANT:
 
- 	case Bonus::SPECIAL_FIXED_VALUE_ENCHANT:
 
- 		return (*VLC->spellh)[static_cast<SpellID::ESpellID>(subtype)]->identifier;
 
- 	case Bonus::SPECIAL_UPGRADE:
 
- 		return CreatureID::encode(subtype) + "2" + CreatureID::encode(additionalInfo[0]);
 
- 	case Bonus::GENERATE_RESOURCE:
 
- 		return GameConstants::RESOURCE_NAMES[subtype];
 
- 	case Bonus::STACKS_SPEED:
 
- 		return "speed";
 
- 	default:
 
- 		return vstd::findKey(bonusNameMap, type);
 
- 	}
 
- }
 
- BonusParams::BonusParams(std::string deprecatedTypeStr, std::string deprecatedSubtypeStr, int deprecatedSubtype):
 
- 	isConverted(true)
 
- {
 
- 	if(deprecatedTypeStr == "SECONDARY_SKILL_PREMY" || deprecatedTypeStr == "SPECIAL_SECONDARY_SKILL")
 
- 	{
 
- 		if(deprecatedSubtype == SecondarySkill::PATHFINDING || deprecatedSubtypeStr == "skill.pathfinding")
 
- 			type = Bonus::ROUGH_TERRAIN_DISCOUNT;
 
- 		else if(deprecatedSubtype == SecondarySkill::DIPLOMACY || deprecatedSubtypeStr == "skill.diplomacy")
 
- 			type = Bonus::WANDERING_CREATURES_JOIN_BONUS;
 
- 		else if(deprecatedSubtype == SecondarySkill::WISDOM || deprecatedSubtypeStr == "skill.wisdom")
 
- 			type = Bonus::MAX_LEARNABLE_SPELL_LEVEL;
 
- 		else if(deprecatedSubtype == SecondarySkill::MYSTICISM || deprecatedSubtypeStr == "skill.mysticism")
 
- 			type = Bonus::MANA_REGENERATION;
 
- 		else if(deprecatedSubtype == SecondarySkill::NECROMANCY || deprecatedSubtypeStr == "skill.necromancy")
 
- 			type = Bonus::UNDEAD_RAISE_PERCENTAGE;
 
- 		else if(deprecatedSubtype == SecondarySkill::LEARNING || deprecatedSubtypeStr == "skill.learning")
 
- 			type = Bonus::HERO_EXPERIENCE_GAIN_PERCENT;
 
- 		else if(deprecatedSubtype == SecondarySkill::RESISTANCE || deprecatedSubtypeStr == "skill.resistance")
 
- 			type = Bonus::MAGIC_RESISTANCE;
 
- 		else if(deprecatedSubtype == SecondarySkill::EAGLE_EYE || deprecatedSubtypeStr == "skill.eagleEye")
 
- 			type = Bonus::LEARN_BATTLE_SPELL_CHANCE;
 
- 		else if(deprecatedSubtype == SecondarySkill::INTELLIGENCE || deprecatedSubtypeStr == "skill.intelligence")
 
- 		{
 
- 			type = Bonus::MANA_PER_KNOWLEDGE;
 
- 			valueType = Bonus::PERCENT_TO_BASE;
 
- 			valueTypeRelevant = true;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::SORCERY || deprecatedSubtypeStr == "skill.sorcery")
 
- 			type = Bonus::SPELL_DAMAGE;
 
- 		else if(deprecatedSubtype == SecondarySkill::SCHOLAR || deprecatedSubtypeStr == "skill.scholar")
 
- 			type = Bonus::LEARN_MEETING_SPELL_LIMIT;
 
- 		else if(deprecatedSubtype == SecondarySkill::ARCHERY|| deprecatedSubtypeStr == "skill.archery")
 
- 		{
 
- 			subtype = 1;
 
- 			subtypeRelevant = true;
 
- 			type = Bonus::PERCENTAGE_DAMAGE_BOOST;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::OFFENCE || deprecatedSubtypeStr == "skill.offence")
 
- 		{
 
- 			subtype = 0;
 
- 			subtypeRelevant = true;
 
- 			type = Bonus::PERCENTAGE_DAMAGE_BOOST;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::ARMORER || deprecatedSubtypeStr == "skill.armorer")
 
- 		{
 
- 			subtype = -1;
 
- 			subtypeRelevant = true;
 
- 			type = Bonus::GENERAL_DAMAGE_REDUCTION;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::NAVIGATION || deprecatedSubtypeStr == "skill.navigation")
 
- 		{
 
- 			subtype = 0;
 
- 			subtypeRelevant = true;
 
- 			valueType = Bonus::PERCENT_TO_BASE;
 
- 			valueTypeRelevant = true;
 
- 			type = Bonus::MOVEMENT;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::LOGISTICS || deprecatedSubtypeStr == "skill.logistics")
 
- 		{
 
- 			subtype = 0;
 
- 			subtypeRelevant = true;
 
- 			valueType = Bonus::PERCENT_TO_BASE;
 
- 			valueTypeRelevant = true;
 
- 			type = Bonus::MOVEMENT;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::ESTATES || deprecatedSubtypeStr == "skill.estates")
 
- 		{
 
- 			type = Bonus::GENERATE_RESOURCE;
 
- 			subtype = Res::GOLD;
 
- 			subtypeRelevant = true;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::AIR_MAGIC || deprecatedSubtypeStr == "skill.airMagic")
 
- 		{
 
- 			type = Bonus::MAGIC_SCHOOL_SKILL;
 
- 			subtypeRelevant = true;
 
- 			subtype = 4;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::WATER_MAGIC || deprecatedSubtypeStr == "skill.waterMagic")
 
- 		{
 
- 			type = Bonus::MAGIC_SCHOOL_SKILL;
 
- 			subtypeRelevant = true;
 
- 			subtype = 1;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::FIRE_MAGIC || deprecatedSubtypeStr == "skill.fireMagic")
 
- 		{
 
- 			type = Bonus::MAGIC_SCHOOL_SKILL;
 
- 			subtypeRelevant = true;
 
- 			subtype = 2;
 
- 		}
 
- 		else if(deprecatedSubtype == SecondarySkill::EARTH_MAGIC || deprecatedSubtypeStr == "skill.earthMagic")
 
- 		{
 
- 			type = Bonus::MAGIC_SCHOOL_SKILL;
 
- 			subtypeRelevant = true;
 
- 			subtype = 8;
 
- 		}
 
- 		else if (deprecatedSubtype == SecondarySkill::ARTILLERY || deprecatedSubtypeStr == "skill.artillery")
 
- 		{
 
- 			type = Bonus::BONUS_DAMAGE_CHANCE;
 
- 			subtypeRelevant = true;
 
- 			subtypeStr = "core:creature.ballista";
 
- 		}
 
- 		else if (deprecatedSubtype == SecondarySkill::FIRST_AID || deprecatedSubtypeStr == "skill.firstAid")
 
- 		{
 
- 			type = Bonus::SPECIFIC_SPELL_POWER;
 
- 			subtypeRelevant = true;
 
- 			subtypeStr = "core:spell.firstAid";
 
- 		}
 
- 		else if (deprecatedSubtype == SecondarySkill::BALLISTICS || deprecatedSubtypeStr == "skill.ballistics")
 
- 		{
 
- 			type = Bonus::CATAPULT_EXTRA_SHOTS;
 
- 			subtypeRelevant = true;
 
- 			subtypeStr = "core:spell.catapultShot";
 
- 		}
 
- 		else
 
- 			isConverted = false;
 
- 	}
 
- 	else if (deprecatedTypeStr == "SECONDARY_SKILL_VAL2")
 
- 	{
 
- 		if(deprecatedSubtype == SecondarySkill::EAGLE_EYE || deprecatedSubtypeStr == "skill.eagleEye")
 
- 			type = Bonus::LEARN_BATTLE_SPELL_LEVEL_LIMIT;
 
- 		else if (deprecatedSubtype == SecondarySkill::ARTILLERY || deprecatedSubtypeStr == "skill.artillery")
 
- 		{
 
- 			type = Bonus::HERO_GRANTS_ATTACKS;
 
- 			subtypeRelevant = true;
 
- 			subtypeStr = "core:creature.ballista";
 
- 		}
 
- 		else
 
- 			isConverted = false;
 
- 	}
 
- 	else if (deprecatedTypeStr == "SEA_MOVEMENT")
 
- 	{
 
- 		subtype = 0;
 
- 		subtypeRelevant = true;
 
- 		valueType = Bonus::ADDITIVE_VALUE;
 
- 		valueTypeRelevant = true;
 
- 		type = Bonus::MOVEMENT;
 
- 	}
 
- 	else if (deprecatedTypeStr == "LAND_MOVEMENT")
 
- 	{
 
- 		subtype = 1;
 
- 		subtypeRelevant = true;
 
- 		valueType = Bonus::ADDITIVE_VALUE;
 
- 		valueTypeRelevant = true;
 
- 		type = Bonus::MOVEMENT;
 
- 	}
 
- 	else if (deprecatedTypeStr == "MAXED_SPELL")
 
- 	{
 
- 		type = Bonus::SPELL;
 
- 		subtypeStr = deprecatedSubtypeStr;
 
- 		subtypeRelevant = true;
 
- 		valueType = Bonus::INDEPENDENT_MAX;
 
- 		valueTypeRelevant = true;
 
- 		val = 3;
 
- 		valRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "FULL_HP_REGENERATION")
 
- 	{
 
- 		type = Bonus::HP_REGENERATION;
 
- 		val = 100000; //very high value to always chose stack health
 
- 		valRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "KING1")
 
- 	{
 
- 		type = Bonus::KING;
 
- 		val = 0;
 
- 		valRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "KING2")
 
- 	{
 
- 		type = Bonus::KING;
 
- 		val = 2;
 
- 		valRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "KING3")
 
- 	{
 
- 		type = Bonus::KING;
 
- 		val = 3;
 
- 		valRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "SIGHT_RADIOUS")
 
- 		type = Bonus::SIGHT_RADIUS;
 
- 	else if (deprecatedTypeStr == "SELF_MORALE")
 
- 	{
 
- 		type = Bonus::MORALE;
 
- 		val = 1;
 
- 		valRelevant = true;
 
- 		valueType = Bonus::INDEPENDENT_MAX;
 
- 		valueTypeRelevant = true;
 
- 	}
 
- 	else if (deprecatedTypeStr == "SELF_LUCK")
 
- 	{
 
- 		type = Bonus::LUCK;
 
- 		val = 1;
 
- 		valRelevant = true;
 
- 		valueType = Bonus::INDEPENDENT_MAX;
 
- 		valueTypeRelevant = true;
 
- 	}
 
- 	else
 
- 		isConverted = false;
 
- }
 
- const JsonNode & BonusParams::toJson()
 
- {
 
- 	assert(isConverted);
 
- 	if(ret.isNull())
 
- 	{
 
- 		ret["type"].String() = vstd::findKey(bonusNameMap, type);
 
- 		if(subtypeRelevant && !subtypeStr.empty())
 
- 			ret["subtype"].String() = subtypeStr;
 
- 		else if(subtypeRelevant)
 
- 			ret["subtype"].Integer() = subtype;
 
- 		if(valueTypeRelevant)
 
- 			ret["valueType"].String() = vstd::findKey(bonusValueMap, valueType);
 
- 		if(valRelevant)
 
- 			ret["val"].Float() = val;
 
- 		if(targetTypeRelevant)
 
- 			ret["targetSourceType"].String() = vstd::findKey(bonusSourceMap, targetType);
 
- 		jsonCreated = true;
 
- 	}
 
- 	return ret;
 
- };
 
- CSelector BonusParams::toSelector()
 
- {
 
- 	assert(isConverted);
 
- 	if(subtypeRelevant && !subtypeStr.empty())
 
- 		JsonUtils::resolveIdentifier(subtype, toJson(), "subtype");
 
- 	auto ret = Selector::type()(type);
 
- 	if(subtypeRelevant)
 
- 		ret = ret.And(Selector::subtype()(subtype));
 
- 	if(valueTypeRelevant)
 
- 		ret = ret.And(Selector::valueType(valueType));
 
- 	if(targetTypeRelevant)
 
- 		ret = ret.And(Selector::targetSourceType()(targetType));
 
- 	return ret;
 
- }
 
- Bonus::Bonus(Bonus::BonusDuration Duration, BonusType Type, BonusSource Src, si32 Val, ui32 ID, std::string Desc, si32 Subtype):
 
- 	duration(static_cast<ui16>(Duration)),
 
- 	type(Type),
 
- 	subtype(Subtype),
 
- 	source(Src),
 
- 	val(Val),
 
- 	sid(ID),
 
- 	description(std::move(Desc))
 
- {
 
- 	boost::algorithm::trim(description);
 
- 	targetSourceType = OTHER;
 
- }
 
- Bonus::Bonus(Bonus::BonusDuration Duration, BonusType Type, BonusSource Src, si32 Val, ui32 ID, si32 Subtype, ValueType ValType):
 
- 	duration(static_cast<ui16>(Duration)),
 
- 	type(Type),
 
- 	subtype(Subtype),
 
- 	source(Src),
 
- 	val(Val),
 
- 	sid(ID),
 
- 	valType(ValType)
 
- {
 
- 	turnsRemain = 0;
 
- 	effectRange = NO_LIMIT;
 
- 	targetSourceType = OTHER;
 
- }
 
- std::shared_ptr<Bonus> Bonus::addPropagator(const TPropagatorPtr & Propagator)
 
- {
 
- 	propagator = Propagator;
 
- 	return this->shared_from_this();
 
- }
 
- namespace Selector
 
- {
 
- 	DLL_LINKAGE CSelectFieldEqual<Bonus::BonusType> & type()
 
- 	{
 
- 		static CSelectFieldEqual<Bonus::BonusType> stype(&Bonus::type);
 
- 		return stype;
 
- 	}
 
- 	DLL_LINKAGE CSelectFieldEqual<TBonusSubtype> & subtype()
 
- 	{
 
- 		static CSelectFieldEqual<TBonusSubtype> ssubtype(&Bonus::subtype);
 
- 		return ssubtype;
 
- 	}
 
- 	DLL_LINKAGE CSelectFieldEqual<CAddInfo> & info()
 
- 	{
 
- 		static CSelectFieldEqual<CAddInfo> sinfo(&Bonus::additionalInfo);
 
- 		return sinfo;
 
- 	}
 
- 	DLL_LINKAGE CSelectFieldEqual<Bonus::BonusSource> & sourceType()
 
- 	{
 
- 		static CSelectFieldEqual<Bonus::BonusSource> ssourceType(&Bonus::source);
 
- 		return ssourceType;
 
- 	}
 
- 	DLL_LINKAGE CSelectFieldEqual<Bonus::BonusSource> & targetSourceType()
 
- 	{
 
- 		static CSelectFieldEqual<Bonus::BonusSource> ssourceType(&Bonus::targetSourceType);
 
- 		return ssourceType;
 
- 	}
 
- 	DLL_LINKAGE CSelectFieldEqual<Bonus::LimitEffect> & effectRange()
 
- 	{
 
- 		static CSelectFieldEqual<Bonus::LimitEffect> seffectRange(&Bonus::effectRange);
 
- 		return seffectRange;
 
- 	}
 
- 	DLL_LINKAGE CWillLastTurns turns;
 
- 	DLL_LINKAGE CWillLastDays days;
 
- 	CSelector DLL_LINKAGE typeSubtype(Bonus::BonusType Type, TBonusSubtype Subtype)
 
- 	{
 
- 		return type()(Type).And(subtype()(Subtype));
 
- 	}
 
- 	CSelector DLL_LINKAGE typeSubtypeInfo(Bonus::BonusType type, TBonusSubtype subtype, const CAddInfo & info)
 
- 	{
 
- 		return CSelectFieldEqual<Bonus::BonusType>(&Bonus::type)(type)
 
- 			.And(CSelectFieldEqual<TBonusSubtype>(&Bonus::subtype)(subtype))
 
- 			.And(CSelectFieldEqual<CAddInfo>(&Bonus::additionalInfo)(info));
 
- 	}
 
- 	CSelector DLL_LINKAGE source(Bonus::BonusSource source, ui32 sourceID)
 
- 	{
 
- 		return CSelectFieldEqual<Bonus::BonusSource>(&Bonus::source)(source)
 
- 			.And(CSelectFieldEqual<ui32>(&Bonus::sid)(sourceID));
 
- 	}
 
- 	CSelector DLL_LINKAGE sourceTypeSel(Bonus::BonusSource source)
 
- 	{
 
- 		return CSelectFieldEqual<Bonus::BonusSource>(&Bonus::source)(source);
 
- 	}
 
- 	CSelector DLL_LINKAGE valueType(Bonus::ValueType valType)
 
- 	{
 
- 		return CSelectFieldEqual<Bonus::ValueType>(&Bonus::valType)(valType);
 
- 	}
 
- 	DLL_LINKAGE CSelector all([](const Bonus * b){return true;});
 
- 	DLL_LINKAGE CSelector none([](const Bonus * b){return false;});
 
- 	bool DLL_LINKAGE matchesType(const CSelector &sel, Bonus::BonusType type)
 
- 	{
 
- 		Bonus dummy;
 
- 		dummy.type = type;
 
- 		return sel(&dummy);
 
- 	}
 
- 	bool DLL_LINKAGE matchesTypeSubtype(const CSelector &sel, Bonus::BonusType type, TBonusSubtype subtype)
 
- 	{
 
- 		Bonus dummy;
 
- 		dummy.type = type;
 
- 		dummy.subtype = subtype;
 
- 		return sel(&dummy);
 
- 	}
 
- }
 
- const CCreature * retrieveCreature(const CBonusSystemNode *node)
 
- {
 
- 	switch(node->getNodeType())
 
- 	{
 
- 	case CBonusSystemNode::CREATURE:
 
- 		return (dynamic_cast<const CCreature *>(node));
 
- 	case CBonusSystemNode::STACK_BATTLE:
 
- 		return (dynamic_cast<const CStack *>(node))->type;
 
- 	default:
 
- 		const CStackInstance * csi = retrieveStackInstance(node);
 
- 		if(csi)
 
- 			return csi->type;
 
- 		return nullptr;
 
- 	}
 
- }
 
- DLL_LINKAGE std::ostream & operator<<(std::ostream &out, const BonusList &bonusList)
 
- {
 
- 	for (ui32 i = 0; i < bonusList.size(); i++)
 
- 	{
 
- 		const auto & b = bonusList[i];
 
- 		out << "Bonus " << i << "\n" << *b << std::endl;
 
- 	}
 
- 	return out;
 
- }
 
- DLL_LINKAGE std::ostream & operator<<(std::ostream &out, const Bonus &bonus)
 
- {
 
- 	for(const auto & i : bonusNameMap)
 
- 	if(i.second == bonus.type)
 
- 		out << "\tType: " << i.first << " \t";
 
- #define printField(field) out << "\t" #field ": " << (int)bonus.field << "\n"
 
- 	printField(val);
 
- 	printField(subtype);
 
- 	printField(duration);
 
- 	printField(source);
 
- 	printField(sid);
 
- 	if(bonus.additionalInfo != CAddInfo::NONE)
 
- 		out << "\taddInfo: " << bonus.additionalInfo.toString() << "\n";
 
- 	printField(turnsRemain);
 
- 	printField(valType);
 
- 	if(!bonus.stacking.empty())
 
- 		out << "\tstacking: \"" << bonus.stacking << "\"\n";
 
- 	printField(effectRange);
 
- #undef printField
 
- 	if(bonus.limiter)
 
- 		out << "\tLimiter: " << bonus.limiter->toString() << "\n";
 
- 	if(bonus.updater)
 
- 		out << "\tUpdater: " << bonus.updater->toString() << "\n";
 
- 	return out;
 
- }
 
- std::shared_ptr<Bonus> Bonus::addLimiter(const TLimiterPtr & Limiter)
 
- {
 
- 	if (limiter)
 
- 	{
 
- 		//If we already have limiter list, retrieve it
 
- 		auto limiterList = std::dynamic_pointer_cast<AllOfLimiter>(limiter);
 
- 		if(!limiterList)
 
- 		{
 
- 			//Create a new limiter list with old limiter and the new one will be pushed later
 
- 			limiterList = std::make_shared<AllOfLimiter>();
 
- 			limiterList->add(limiter);
 
- 			limiter = limiterList;
 
- 		}
 
- 		limiterList->add(Limiter);
 
- 	}
 
- 	else
 
- 	{
 
- 		limiter = Limiter;
 
- 	}
 
- 	return this->shared_from_this();
 
- }
 
- int ILimiter::limit(const BonusLimitationContext &context) const /*return true to drop the bonus */
 
- {
 
- 	return false;
 
- }
 
- std::string ILimiter::toString() const
 
- {
 
- 	return typeid(*this).name();
 
- }
 
- JsonNode ILimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = toString();
 
- 	return root;
 
- }
 
- int CCreatureTypeLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CCreature *c = retrieveCreature(&context.node);
 
- 	if(!c)
 
- 		return true;
 
- 	return c->getId() != creature->getId() && (!includeUpgrades || !creature->isMyUpgrade(c));
 
- 	//drop bonus if it's not our creature and (we don`t check upgrades or its not our upgrade)
 
- }
 
- CCreatureTypeLimiter::CCreatureTypeLimiter(const CCreature & creature_, bool IncludeUpgrades)
 
- 	: creature(&creature_), includeUpgrades(IncludeUpgrades)
 
- {
 
- }
 
- void CCreatureTypeLimiter::setCreature(const CreatureID & id)
 
- {
 
- 	creature = VLC->creh->objects[id];
 
- }
 
- std::string CCreatureTypeLimiter::toString() const
 
- {
 
- 	boost::format fmt("CCreatureTypeLimiter(creature=%s, includeUpgrades=%s)");
 
- 	fmt % creature->getJsonKey() % (includeUpgrades ? "true" : "false");
 
- 	return fmt.str();
 
- }
 
- JsonNode CCreatureTypeLimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "CREATURE_TYPE_LIMITER";
 
- 	root["parameters"].Vector().push_back(JsonUtils::stringNode(creature->getJsonKey()));
 
- 	root["parameters"].Vector().push_back(JsonUtils::boolNode(includeUpgrades));
 
- 	return root;
 
- }
 
- HasAnotherBonusLimiter::HasAnotherBonusLimiter( Bonus::BonusType bonus )
 
- 	: type(bonus), subtype(0), isSubtypeRelevant(false), isSourceRelevant(false), isSourceIDRelevant(false)
 
- {
 
- }
 
- HasAnotherBonusLimiter::HasAnotherBonusLimiter( Bonus::BonusType bonus, TBonusSubtype _subtype )
 
- 	: type(bonus), subtype(_subtype), isSubtypeRelevant(true), isSourceRelevant(false), isSourceIDRelevant(false)
 
- {
 
- }
 
- HasAnotherBonusLimiter::HasAnotherBonusLimiter(Bonus::BonusType bonus, Bonus::BonusSource src)
 
- 	: type(bonus), source(src), isSubtypeRelevant(false), isSourceRelevant(true), isSourceIDRelevant(false)
 
- {
 
- }
 
- HasAnotherBonusLimiter::HasAnotherBonusLimiter(Bonus::BonusType bonus, TBonusSubtype _subtype, Bonus::BonusSource src)
 
- 	: type(bonus), subtype(_subtype), isSubtypeRelevant(true), source(src), isSourceRelevant(true), isSourceIDRelevant(false)
 
- {
 
- }
 
- int HasAnotherBonusLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	//TODO: proper selector config with parsing of JSON
 
- 	auto mySelector = Selector::type()(type);
 
- 	if(isSubtypeRelevant)
 
- 		mySelector = mySelector.And(Selector::subtype()(subtype));
 
- 	if(isSourceRelevant && isSourceIDRelevant)
 
- 		mySelector = mySelector.And(Selector::source(source, sid));
 
- 	else if (isSourceRelevant)
 
- 		mySelector = mySelector.And(Selector::sourceTypeSel(source));
 
- 	//if we have a bonus of required type accepted, limiter should accept also this bonus
 
- 	if(context.alreadyAccepted.getFirst(mySelector))
 
- 		return ACCEPT;
 
- 	//if there are no matching bonuses pending, we can (and must) reject right away
 
- 	if(!context.stillUndecided.getFirst(mySelector))
 
- 		return DISCARD;
 
- 	//do not accept for now but it may change if more bonuses gets included
 
- 	return NOT_SURE;
 
- }
 
- std::string HasAnotherBonusLimiter::toString() const
 
- {
 
- 	std::string typeName = vstd::findKey(bonusNameMap, type);
 
- 	if(isSubtypeRelevant)
 
- 	{
 
- 		boost::format fmt("HasAnotherBonusLimiter(type=%s, subtype=%d)");
 
- 		fmt % typeName % subtype;
 
- 		return fmt.str();
 
- 	}
 
- 	else
 
- 	{
 
- 		boost::format fmt("HasAnotherBonusLimiter(type=%s)");
 
- 		fmt % typeName;
 
- 		return fmt.str();
 
- 	}
 
- }
 
- JsonNode HasAnotherBonusLimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	std::string typeName = vstd::findKey(bonusNameMap, type);
 
- 	auto sourceTypeName = vstd::findKey(bonusSourceMap, source);
 
- 	root["type"].String() = "HAS_ANOTHER_BONUS_LIMITER";
 
- 	root["parameters"].Vector().push_back(JsonUtils::stringNode(typeName));
 
- 	if(isSubtypeRelevant)
 
- 		root["parameters"].Vector().push_back(JsonUtils::intNode(subtype));
 
- 	if(isSourceRelevant)
 
- 		root["parameters"].Vector().push_back(JsonUtils::stringNode(sourceTypeName));
 
- 	return root;
 
- }
 
- bool IPropagator::shouldBeAttached(CBonusSystemNode *dest)
 
- {
 
- 	return false;
 
- }
 
- CBonusSystemNode::ENodeTypes IPropagator::getPropagatorType() const
 
- {
 
- 	return CBonusSystemNode::ENodeTypes::NONE;
 
- }
 
- CPropagatorNodeType::CPropagatorNodeType()
 
- 	:nodeType(CBonusSystemNode::ENodeTypes::UNKNOWN)
 
- {
 
- }
 
- CPropagatorNodeType::CPropagatorNodeType(CBonusSystemNode::ENodeTypes NodeType)
 
- 	: nodeType(NodeType)
 
- {
 
- }
 
- CBonusSystemNode::ENodeTypes CPropagatorNodeType::getPropagatorType() const
 
- {
 
- 	return nodeType;
 
- }
 
- bool CPropagatorNodeType::shouldBeAttached(CBonusSystemNode *dest)
 
- {
 
- 	return nodeType == dest->getNodeType();
 
- }
 
- CreatureTerrainLimiter::CreatureTerrainLimiter()
 
- 	: terrainType(ETerrainId::NATIVE_TERRAIN)
 
- {
 
- }
 
- CreatureTerrainLimiter::CreatureTerrainLimiter(TerrainId terrain):
 
- 	terrainType(terrain)
 
- {
 
- }
 
- int CreatureTerrainLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CStack *stack = retrieveStackBattle(&context.node);
 
- 	if(stack)
 
- 	{
 
- 		if (terrainType == ETerrainId::NATIVE_TERRAIN)//terrainType not specified = native
 
- 		{
 
- 			return !stack->isOnNativeTerrain();
 
- 		}
 
- 		else
 
- 		{
 
- 			return !stack->isOnTerrain(terrainType);
 
- 		}
 
- 	}
 
- 	return true;
 
- 	//TODO neutral creatues
 
- }
 
- std::string CreatureTerrainLimiter::toString() const
 
- {
 
- 	boost::format fmt("CreatureTerrainLimiter(terrainType=%s)");
 
- 	auto terrainName = VLC->terrainTypeHandler->getById(terrainType)->getJsonKey();
 
- 	fmt % (terrainType == ETerrainId::NATIVE_TERRAIN ? "native" : terrainName);
 
- 	return fmt.str();
 
- }
 
- JsonNode CreatureTerrainLimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "CREATURE_TERRAIN_LIMITER";
 
- 	auto terrainName = VLC->terrainTypeHandler->getById(terrainType)->getJsonKey();
 
- 	root["parameters"].Vector().push_back(JsonUtils::stringNode(terrainName));
 
- 	return root;
 
- }
 
- CreatureFactionLimiter::CreatureFactionLimiter(TFaction creatureFaction)
 
- 	: faction(creatureFaction)
 
- {
 
- }
 
- CreatureFactionLimiter::CreatureFactionLimiter():
 
- 	faction(static_cast<TFaction>(-1))
 
- {
 
- }
 
- int CreatureFactionLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CCreature *c = retrieveCreature(&context.node);
 
- 	return !c || c->faction != faction; //drop bonus for non-creatures or non-native residents
 
- }
 
- std::string CreatureFactionLimiter::toString() const
 
- {
 
- 	boost::format fmt("CreatureFactionLimiter(faction=%s)");
 
- 	fmt % VLC->factions()->getByIndex(faction)->getJsonKey();
 
- 	return fmt.str();
 
- }
 
- JsonNode CreatureFactionLimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "CREATURE_FACTION_LIMITER";
 
- 	root["parameters"].Vector().push_back(JsonUtils::stringNode(VLC->factions()->getByIndex(faction)->getJsonKey()));
 
- 	return root;
 
- }
 
- CreatureAlignmentLimiter::CreatureAlignmentLimiter()
 
- 	: alignment(-1)
 
- {
 
- }
 
- CreatureAlignmentLimiter::CreatureAlignmentLimiter(si8 Alignment)
 
- 	: alignment(Alignment)
 
- {
 
- }
 
- int CreatureAlignmentLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CCreature *c = retrieveCreature(&context.node);
 
- 	if(!c)
 
- 		return true;
 
- 	switch(alignment)
 
- 	{
 
- 	case EAlignment::GOOD:
 
- 		return !c->isGood(); //if not good -> return true (drop bonus)
 
- 	case EAlignment::NEUTRAL:
 
- 		return c->isEvil() || c->isGood();
 
- 	case EAlignment::EVIL:
 
- 		return !c->isEvil();
 
- 	default:
 
- 		logBonus->warn("Warning: illegal alignment in limiter!");
 
- 		return true;
 
- 	}
 
- }
 
- std::string CreatureAlignmentLimiter::toString() const
 
- {
 
- 	boost::format fmt("CreatureAlignmentLimiter(alignment=%s)");
 
- 	fmt % EAlignment::names[alignment];
 
- 	return fmt.str();
 
- }
 
- JsonNode CreatureAlignmentLimiter::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "CREATURE_ALIGNMENT_LIMITER";
 
- 	root["parameters"].Vector().push_back(JsonUtils::stringNode(EAlignment::names[alignment]));
 
- 	return root;
 
- }
 
- RankRangeLimiter::RankRangeLimiter(ui8 Min, ui8 Max)
 
- 	:minRank(Min), maxRank(Max)
 
- {
 
- }
 
- RankRangeLimiter::RankRangeLimiter()
 
- {
 
- 	minRank = maxRank = -1;
 
- }
 
- int RankRangeLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CStackInstance * csi = retrieveStackInstance(&context.node);
 
- 	if(csi)
 
- 	{
 
- 		if (csi->getNodeType() == CBonusSystemNode::COMMANDER) //no stack exp bonuses for commander creatures
 
- 			return true;
 
- 		return csi->getExpRank() < minRank || csi->getExpRank() > maxRank;
 
- 	}
 
- 	return true;
 
- }
 
- int StackOwnerLimiter::limit(const BonusLimitationContext &context) const
 
- {
 
- 	const CStack * s = retrieveStackBattle(&context.node);
 
- 	if(s)
 
- 		return s->owner != owner;
 
- 	const CStackInstance * csi = retrieveStackInstance(&context.node);
 
- 	if(csi && csi->armyObj)
 
- 		return csi->armyObj->tempOwner != owner;
 
- 	return true;
 
- }
 
- StackOwnerLimiter::StackOwnerLimiter()
 
- 	: owner(-1)
 
- {
 
- }
 
- StackOwnerLimiter::StackOwnerLimiter(const PlayerColor & Owner):
 
- 	owner(Owner)
 
- {
 
- }
 
- OppositeSideLimiter::OppositeSideLimiter():
 
- 	owner(PlayerColor::CANNOT_DETERMINE)
 
- {
 
- }
 
- OppositeSideLimiter::OppositeSideLimiter(const PlayerColor & Owner):
 
- 	owner(Owner)
 
- {
 
- }
 
- int OppositeSideLimiter::limit(const BonusLimitationContext & context) const
 
- {
 
- 	auto contextOwner = CBonusSystemNode::retrieveNodeOwner(& context.node);
 
- 	auto decision = (owner == contextOwner || owner == PlayerColor::CANNOT_DETERMINE) ? ILimiter::DISCARD : ILimiter::ACCEPT;
 
- 	return decision;
 
- }
 
- // Aggregate/Boolean Limiters
 
- void AggregateLimiter::add(const TLimiterPtr & limiter)
 
- {
 
- 	if(limiter)
 
- 		limiters.push_back(limiter);
 
- }
 
- JsonNode AggregateLimiter::toJsonNode() const
 
- {
 
- 	JsonNode result(JsonNode::JsonType::DATA_VECTOR);
 
- 	result.Vector().push_back(JsonUtils::stringNode(getAggregator()));
 
- 	for(const auto & l : limiters)
 
- 		result.Vector().push_back(l->toJsonNode());
 
- 	return result;
 
- }
 
- const std::string AllOfLimiter::aggregator = "allOf";
 
- const std::string & AllOfLimiter::getAggregator() const
 
- {
 
- 	return aggregator;
 
- }
 
- int AllOfLimiter::limit(const BonusLimitationContext & context) const
 
- {
 
- 	bool wasntSure = false;
 
- 	for(const auto & limiter : limiters)
 
- 	{
 
- 		auto result = limiter->limit(context);
 
- 		if(result == ILimiter::DISCARD)
 
- 			return result;
 
- 		if(result == ILimiter::NOT_SURE)
 
- 			wasntSure = true;
 
- 	}
 
- 	return wasntSure ? ILimiter::NOT_SURE : ILimiter::ACCEPT;
 
- }
 
- const std::string AnyOfLimiter::aggregator = "anyOf";
 
- const std::string & AnyOfLimiter::getAggregator() const
 
- {
 
- 	return aggregator;
 
- }
 
- int AnyOfLimiter::limit(const BonusLimitationContext & context) const
 
- {
 
- 	bool wasntSure = false;
 
- 	for(const auto & limiter : limiters)
 
- 	{
 
- 		auto result = limiter->limit(context);
 
- 		if(result == ILimiter::ACCEPT)
 
- 			return result;
 
- 		if(result == ILimiter::NOT_SURE)
 
- 			wasntSure = true;
 
- 	}
 
- 	return wasntSure ? ILimiter::NOT_SURE : ILimiter::DISCARD;
 
- }
 
- const std::string NoneOfLimiter::aggregator = "noneOf";
 
- const std::string & NoneOfLimiter::getAggregator() const
 
- {
 
- 	return aggregator;
 
- }
 
- int NoneOfLimiter::limit(const BonusLimitationContext & context) const
 
- {
 
- 	bool wasntSure = false;
 
- 	for(const auto & limiter : limiters)
 
- 	{
 
- 		auto result = limiter->limit(context);
 
- 		if(result == ILimiter::ACCEPT)
 
- 			return ILimiter::DISCARD;
 
- 		if(result == ILimiter::NOT_SURE)
 
- 			wasntSure = true;
 
- 	}
 
- 	return wasntSure ? ILimiter::NOT_SURE : ILimiter::ACCEPT;
 
- }
 
- // Updaters
 
- std::shared_ptr<Bonus> Bonus::addUpdater(const TUpdaterPtr & Updater)
 
- {
 
- 	updater = Updater;
 
- 	return this->shared_from_this();
 
- }
 
- std::shared_ptr<Bonus> IUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	return b;
 
- }
 
- std::string IUpdater::toString() const
 
- {
 
- 	return typeid(*this).name();
 
- }
 
- JsonNode IUpdater::toJsonNode() const
 
- {
 
- 	return JsonNode(JsonNode::JsonType::DATA_NULL);
 
- }
 
- GrowsWithLevelUpdater::GrowsWithLevelUpdater(int valPer20, int stepSize) : valPer20(valPer20), stepSize(stepSize)
 
- {
 
- }
 
- std::shared_ptr<Bonus> GrowsWithLevelUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	if(context.getNodeType() == CBonusSystemNode::HERO)
 
- 	{
 
- 		int level = dynamic_cast<const CGHeroInstance &>(context).level;
 
- 		int steps = stepSize ? level / stepSize : level;
 
- 		//rounding follows format for HMM3 creature specialty bonus
 
- 		int newVal = (valPer20 * steps + 19) / 20;
 
- 		//return copy of bonus with updated val
 
- 		std::shared_ptr<Bonus> newBonus = std::make_shared<Bonus>(*b);
 
- 		newBonus->val = newVal;
 
- 		return newBonus;
 
- 	}
 
- 	return b;
 
- }
 
- std::string GrowsWithLevelUpdater::toString() const
 
- {
 
- 	return boost::str(boost::format("GrowsWithLevelUpdater(valPer20=%d, stepSize=%d)") % valPer20 % stepSize);
 
- }
 
- JsonNode GrowsWithLevelUpdater::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "GROWS_WITH_LEVEL";
 
- 	root["parameters"].Vector().push_back(JsonUtils::intNode(valPer20));
 
- 	if(stepSize > 1)
 
- 		root["parameters"].Vector().push_back(JsonUtils::intNode(stepSize));
 
- 	return root;
 
- }
 
- std::shared_ptr<Bonus> TimesHeroLevelUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	if(context.getNodeType() == CBonusSystemNode::HERO)
 
- 	{
 
- 		int level = dynamic_cast<const CGHeroInstance &>(context).level;
 
- 		std::shared_ptr<Bonus> newBonus = std::make_shared<Bonus>(*b);
 
- 		newBonus->val *= level;
 
- 		return newBonus;
 
- 	}
 
- 	return b;
 
- }
 
- std::string TimesHeroLevelUpdater::toString() const
 
- {
 
- 	return "TimesHeroLevelUpdater";
 
- }
 
- JsonNode TimesHeroLevelUpdater::toJsonNode() const
 
- {
 
- 	return JsonUtils::stringNode("TIMES_HERO_LEVEL");
 
- }
 
- ArmyMovementUpdater::ArmyMovementUpdater():
 
- 	base(20),
 
- 	divider(3),
 
- 	multiplier(10),
 
- 	max(700)
 
- {
 
- }
 
- ArmyMovementUpdater::ArmyMovementUpdater(int base, int divider, int multiplier, int max):
 
- 	base(base),
 
- 	divider(divider),
 
- 	multiplier(multiplier),
 
- 	max(max)
 
- {
 
- }
 
- std::shared_ptr<Bonus> ArmyMovementUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	if(b->type == Bonus::MOVEMENT && context.getNodeType() == CBonusSystemNode::HERO)
 
- 	{
 
- 		auto speed = static_cast<const CGHeroInstance &>(context).getLowestCreatureSpeed();
 
- 		si32 armySpeed = speed * base / divider;
 
- 		auto counted = armySpeed * multiplier;
 
- 		auto newBonus = std::make_shared<Bonus>(*b);
 
- 		newBonus->source = Bonus::ARMY;
 
- 		newBonus->val += vstd::amin(counted, max);
 
- 		return newBonus;
 
- 	}
 
- 	if(b->type != Bonus::MOVEMENT)
 
- 		logGlobal->error("ArmyMovementUpdater should only be used for MOVEMENT bonus!");
 
- 	return b;
 
- }
 
- std::string ArmyMovementUpdater::toString() const
 
- {
 
- 	return "ArmyMovementUpdater";
 
- }
 
- JsonNode ArmyMovementUpdater::toJsonNode() const
 
- {
 
- 	JsonNode root(JsonNode::JsonType::DATA_STRUCT);
 
- 	root["type"].String() = "ARMY_MOVEMENT";
 
- 	root["parameters"].Vector().push_back(JsonUtils::intNode(base));
 
- 	root["parameters"].Vector().push_back(JsonUtils::intNode(divider));
 
- 	root["parameters"].Vector().push_back(JsonUtils::intNode(multiplier));
 
- 	root["parameters"].Vector().push_back(JsonUtils::intNode(max));
 
- 	return root;
 
- }
 
- std::shared_ptr<Bonus> TimesStackLevelUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	if(context.getNodeType() == CBonusSystemNode::STACK_INSTANCE)
 
- 	{
 
- 		int level = dynamic_cast<const CStackInstance &>(context).getLevel();
 
- 		std::shared_ptr<Bonus> newBonus = std::make_shared<Bonus>(*b);
 
- 		newBonus->val *= level;
 
- 		return newBonus;
 
- 	}
 
- 	else if(context.getNodeType() == CBonusSystemNode::STACK_BATTLE)
 
- 	{
 
- 		const auto & stack = dynamic_cast<const CStack &>(context);
 
- 		//only update if stack doesn't have an instance (summons, war machines)
 
- 		//otherwise we'd end up multiplying twice
 
- 		if(stack.base == nullptr)
 
- 		{
 
- 			int level = stack.type->level;
 
- 			std::shared_ptr<Bonus> newBonus = std::make_shared<Bonus>(*b);
 
- 			newBonus->val *= level;
 
- 			return newBonus;
 
- 		}
 
- 	}
 
- 	return b;
 
- }
 
- std::string TimesStackLevelUpdater::toString() const
 
- {
 
- 	return "TimesStackLevelUpdater";
 
- }
 
- JsonNode TimesStackLevelUpdater::toJsonNode() const
 
- {
 
- 	return JsonUtils::stringNode("TIMES_STACK_LEVEL");
 
- }
 
- std::string OwnerUpdater::toString() const
 
- {
 
- 	return "OwnerUpdater";
 
- }
 
- JsonNode OwnerUpdater::toJsonNode() const
 
- {
 
- 	return JsonUtils::stringNode("BONUS_OWNER_UPDATER");
 
- }
 
- std::shared_ptr<Bonus> OwnerUpdater::createUpdatedBonus(const std::shared_ptr<Bonus> & b, const CBonusSystemNode & context) const
 
- {
 
- 	auto owner = CBonusSystemNode::retrieveNodeOwner(&context);
 
- 	if(owner == PlayerColor::UNFLAGGABLE)
 
- 		owner = PlayerColor::NEUTRAL;
 
- 	std::shared_ptr<Bonus> updated =
 
- 		std::make_shared<Bonus>(static_cast<Bonus::BonusDuration>(b->duration), b->type, b->source, b->val, b->sid, b->subtype, b->valType);
 
- 	updated->limiter = std::make_shared<OppositeSideLimiter>(owner);
 
- 	return updated;
 
- }
 
- VCMI_LIB_NAMESPACE_END
 
 
  |