| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865 | 
							- /*
 
-  * CDefaultSpellMechanics.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 "CDefaultSpellMechanics.h"
 
- #include "../CStack.h"
 
- #include "../battle/BattleInfo.h"
 
- #include "../CGeneralTextHandler.h"
 
- namespace SRSLPraserHelpers
 
- {
 
- 	static int XYToHex(int x, int y)
 
- 	{
 
- 		return x + GameConstants::BFIELD_WIDTH * y;
 
- 	}
 
- 	static int XYToHex(std::pair<int, int> xy)
 
- 	{
 
- 		return XYToHex(xy.first, xy.second);
 
- 	}
 
- 	static int hexToY(int battleFieldPosition)
 
- 	{
 
- 		return battleFieldPosition/GameConstants::BFIELD_WIDTH;
 
- 	}
 
- 	static int hexToX(int battleFieldPosition)
 
- 	{
 
- 		int pos = battleFieldPosition - hexToY(battleFieldPosition) * GameConstants::BFIELD_WIDTH;
 
- 		return pos;
 
- 	}
 
- 	static std::pair<int, int> hexToPair(int battleFieldPosition)
 
- 	{
 
- 		return std::make_pair(hexToX(battleFieldPosition), hexToY(battleFieldPosition));
 
- 	}
 
- 	//moves hex by one hex in given direction
 
- 	//0 - left top, 1 - right top, 2 - right, 3 - right bottom, 4 - left bottom, 5 - left
 
- 	static std::pair<int, int> gotoDir(int x, int y, int direction)
 
- 	{
 
- 		switch(direction)
 
- 		{
 
- 		case 0: //top left
 
- 			return std::make_pair((y%2) ? x-1 : x, y-1);
 
- 		case 1: //top right
 
- 			return std::make_pair((y%2) ? x : x+1, y-1);
 
- 		case 2:  //right
 
- 			return std::make_pair(x+1, y);
 
- 		case 3: //right bottom
 
- 			return std::make_pair((y%2) ? x : x+1, y+1);
 
- 		case 4: //left bottom
 
- 			return std::make_pair((y%2) ? x-1 : x, y+1);
 
- 		case 5: //left
 
- 			return std::make_pair(x-1, y);
 
- 		default:
 
- 			throw std::runtime_error("Disaster: wrong direction in SRSLPraserHelpers::gotoDir!\n");
 
- 		}
 
- 	}
 
- 	static std::pair<int, int> gotoDir(std::pair<int, int> xy, int direction)
 
- 	{
 
- 		return gotoDir(xy.first, xy.second, direction);
 
- 	}
 
- 	static bool isGoodHex(std::pair<int, int> xy)
 
- 	{
 
- 		return xy.first >=0 && xy.first < GameConstants::BFIELD_WIDTH && xy.second >= 0 && xy.second < GameConstants::BFIELD_HEIGHT;
 
- 	}
 
- 	//helper function for rangeInHexes
 
- 	static std::set<ui16> getInRange(unsigned int center, int low, int high)
 
- 	{
 
- 		std::set<ui16> ret;
 
- 		if(low == 0)
 
- 		{
 
- 			ret.insert(center);
 
- 		}
 
- 		std::pair<int, int> mainPointForLayer[6]; //A, B, C, D, E, F points
 
- 		for(auto & elem : mainPointForLayer)
 
- 			elem = hexToPair(center);
 
- 		for(int it=1; it<=high; ++it) //it - distance to the center
 
- 		{
 
- 			for(int b=0; b<6; ++b)
 
- 				mainPointForLayer[b] = gotoDir(mainPointForLayer[b], b);
 
- 			if(it>=low)
 
- 			{
 
- 				std::pair<int, int> curHex;
 
- 				//adding lines (A-b, B-c, C-d, etc)
 
- 				for(int v=0; v<6; ++v)
 
- 				{
 
- 					curHex = mainPointForLayer[v];
 
- 					for(int h=0; h<it; ++h)
 
- 					{
 
- 						if(isGoodHex(curHex))
 
- 							ret.insert(XYToHex(curHex));
 
- 						curHex = gotoDir(curHex, (v+2)%6);
 
- 					}
 
- 				}
 
- 			} //if(it>=low)
 
- 		}
 
- 		return ret;
 
- 	}
 
- }
 
- SpellCastContext::SpellCastContext(const DefaultSpellMechanics * mechanics_, const SpellCastEnvironment * env_, const BattleSpellCastParameters & parameters_):
 
- 	mechanics(mechanics_), env(env_), attackedCres(), sc(), si(), parameters(parameters_), otherHero(nullptr), spellCost(0), damageToDisplay(0)
 
- {
 
- 	sc.side = parameters.casterSide;
 
- 	sc.id = mechanics->owner->id;
 
- 	sc.skill = parameters.spellLvl;
 
- 	sc.tile = parameters.getFirstDestinationHex();
 
- 	sc.castByHero = parameters.mode == ECastingMode::HERO_CASTING;
 
- 	sc.casterStack = (parameters.casterStack ? parameters.casterStack->ID : -1);
 
- 	sc.manaGained = 0;
 
- 	//check it there is opponent hero
 
- 	const ui8 otherSide = 1-parameters.casterSide;
 
- 	if(parameters.cb->battleHasHero(otherSide))
 
- 		otherHero = parameters.cb->battleGetFightingHero(otherSide);
 
- }
 
- SpellCastContext::~SpellCastContext()
 
- {
 
- }
 
- void SpellCastContext::addDamageToDisplay(const si32 value)
 
- {
 
- 	damageToDisplay += value;
 
- }
 
- void SpellCastContext::setDamageToDisplay(const si32 value)
 
- {
 
- 	damageToDisplay = value;
 
- }
 
- void SpellCastContext::addCustomEffect(const CStack * target, ui32 effect)
 
- {
 
- 	BattleSpellCast::CustomEffect customEffect;
 
- 	customEffect.effect = effect;
 
- 	customEffect.stack = target->ID;
 
- 	sc.customEffects.push_back(customEffect);
 
- }
 
- void SpellCastContext::prepareBattleLog()
 
- {
 
- 	bool displayDamage = true;
 
- 	mechanics->battleLog(sc.battleLog, parameters, attackedCres, damageToDisplay, displayDamage);
 
- 	displayDamage = displayDamage && damageToDisplay > 0;
 
- 	if(displayDamage)
 
- 	{
 
-         MetaString line;
 
-         line.addTxt(MetaString::GENERAL_TXT, 376);
 
-         line.addReplacement(MetaString::SPELL_NAME, mechanics->owner->id.toEnum());
 
-         line.addReplacement(damageToDisplay);
 
-         sc.battleLog.push_back(line);
 
- 	}
 
- }
 
- void SpellCastContext::beforeCast()
 
- {
 
- 	//calculate spell cost
 
- 	if(parameters.mode == ECastingMode::HERO_CASTING)
 
- 	{
 
- 		spellCost = parameters.cb->battleGetSpellCost(mechanics->owner, parameters.casterHero);
 
- 		if(nullptr != otherHero) //handle mana channel
 
- 		{
 
- 			int manaChannel = 0;
 
- 			for(const CStack * stack : parameters.cb->battleGetAllStacks(true)) //TODO: shouldn't bonus system handle it somehow?
 
- 			{
 
- 				if(stack->owner == otherHero->tempOwner)
 
- 				{
 
- 					vstd::amax(manaChannel, stack->valOfBonuses(Bonus::MANA_CHANNELING));
 
- 				}
 
- 			}
 
- 			sc.manaGained = (manaChannel * spellCost) / 100;
 
- 		}
 
- 	}
 
- }
 
- void SpellCastContext::afterCast()
 
- {
 
- 	for(auto sta : attackedCres)
 
- 	{
 
- 		sc.affectedCres.insert(sta->ID);
 
- 	}
 
- 	prepareBattleLog();
 
- 	env->sendAndApply(&sc);
 
- 	if(parameters.mode == ECastingMode::HERO_CASTING)
 
- 	{
 
- 		//spend mana
 
- 		SetMana sm;
 
- 		sm.absolute = false;
 
- 		sm.hid = parameters.casterHero->id;
 
- 		sm.val = -spellCost;
 
- 		env->sendAndApply(&sm);
 
- 		if(sc.manaGained > 0)
 
- 		{
 
- 			assert(otherHero);
 
- 			sm.hid = otherHero->id;
 
- 			sm.val = sc.manaGained;
 
- 			env->sendAndApply(&sm);
 
- 		}
 
- 	}
 
- 	else if (parameters.mode == ECastingMode::CREATURE_ACTIVE_CASTING || parameters.mode == ECastingMode::ENCHANTER_CASTING)
 
- 	{
 
- 		//reduce number of casts remaining
 
- 		assert(parameters.casterStack);
 
- 		BattleSetStackProperty ssp;
 
- 		ssp.stackID = parameters.casterStack->ID;
 
- 		ssp.which = BattleSetStackProperty::CASTS;
 
- 		ssp.val = -1;
 
- 		ssp.absolute = false;
 
- 		env->sendAndApply(&ssp);
 
- 	}
 
- 	if(!si.stacks.empty()) //after spellcast info shows
 
- 		env->sendAndApply(&si);
 
- }
 
- ///DefaultSpellMechanics
 
- DefaultSpellMechanics::DefaultSpellMechanics(const CSpell * s):
 
- 	ISpellMechanics(s)
 
- {
 
- };
 
- void DefaultSpellMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
 
- {
 
- 	if (packet->castByHero)
 
- 	{
 
- 		if (packet->side < 2)
 
- 		{
 
- 			battle->sides[packet->side].castSpellsCount++;
 
- 		}
 
- 	}
 
- 	//handle countering spells
 
- 	for(auto stackID : packet->affectedCres)
 
- 	{
 
- 		CStack * s = battle->getStack(stackID);
 
- 		s->popBonuses([&](const Bonus * b) -> bool
 
- 		{
 
- 			//check for each bonus if it should be removed
 
- 			const bool isSpellEffect = Selector::sourceType(Bonus::SPELL_EFFECT)(b);
 
- 			const int spellID = isSpellEffect ? b->sid : -1;
 
- 			//No exceptions, ANY spell can be countered, even if it can`t be dispelled.
 
- 			return isSpellEffect && vstd::contains(owner->counteredSpells, spellID);
 
- 		});
 
- 	}
 
- }
 
- void DefaultSpellMechanics::battleCast(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters) const
 
- {
 
- 	if(nullptr == parameters.caster)
 
- 	{
 
- 		env->complain("No spell-caster provided.");
 
- 		return;
 
- 	}
 
- 	std::vector <const CStack*> reflected;//for magic mirror
 
- 	cast(env, parameters, reflected);
 
- 	//Magic Mirror effect
 
- 	for(auto & attackedCre : reflected)
 
- 	{
 
- 		if(parameters.mode == ECastingMode::MAGIC_MIRROR)
 
- 		{
 
- 			logGlobal->error("Magic mirror recurrence!");
 
- 			return;
 
- 		}
 
- 		TStacks mirrorTargets = parameters.cb->battleGetStacksIf([this, parameters](const CStack * battleStack)
 
- 		{
 
- 			//Get all enemy stacks. Magic mirror can reflect to immune creature (with no effect)
 
- 			return battleStack->owner == parameters.casterColor && battleStack->isValidTarget(false);
 
- 		});
 
- 		if(!mirrorTargets.empty())
 
- 		{
 
- 			int targetHex = (*RandomGeneratorUtil::nextItem(mirrorTargets, env->getRandomGenerator()))->position;
 
- 			BattleSpellCastParameters mirrorParameters(parameters, attackedCre);
 
- 			mirrorParameters.aimToHex(targetHex);
 
- 			mirrorParameters.cast(env);
 
- 		}
 
- 	}
 
- }
 
- void DefaultSpellMechanics::cast(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, std::vector <const CStack*> & reflected) const
 
- {
 
- 	SpellCastContext ctx(this, env, parameters);
 
- 	ctx.beforeCast();
 
- 	ctx.attackedCres = owner->getAffectedStacks(parameters.cb, parameters.mode, parameters.caster, parameters.spellLvl, parameters.getFirstDestinationHex());
 
- 	handleResistance(env, ctx);
 
- 	if(parameters.mode != ECastingMode::MAGIC_MIRROR)
 
- 		handleMagicMirror(env, ctx, reflected);
 
- 	applyBattleEffects(env, parameters, ctx);
 
- 	ctx.afterCast();
 
- }
 
- void DefaultSpellMechanics::battleLog(std::vector<MetaString> & logLines, const BattleSpellCastParameters & parameters,
 
- 	const std::vector<const CStack *> & attacked, const si32 damageToDisplay, bool & displayDamage) const
 
- {
 
- 	if(attacked.size() != 1)
 
- 	{
 
- 		displayDamage = true;
 
- 		battleLogDefault(logLines, parameters, attacked);
 
- 		return;
 
- 	}
 
- 	auto attackedStack = attacked.at(0);
 
- 	auto addLogLine = [attackedStack, &logLines](const int baseTextID, const boost::logic::tribool & plural)
 
- 	{
 
- 		MetaString line;
 
- 		attackedStack->addText(line, MetaString::GENERAL_TXT, baseTextID, plural);
 
- 		attackedStack->addNameReplacement(line, plural);
 
- 		logLines.push_back(line);
 
- 	};
 
- 	displayDamage = false; //in most following cases damage info text is custom
 
- 	switch(owner->id)
 
- 	{
 
- 	case SpellID::STONE_GAZE:
 
- 		addLogLine(558, boost::logic::indeterminate);
 
- 		break;
 
- 	case SpellID::POISON:
 
- 		addLogLine(561, boost::logic::indeterminate);
 
- 		break;
 
- 	case SpellID::BIND:
 
- 		addLogLine(-560, true);//"Roots and vines bind the %s to the ground!"
 
- 		break;
 
- 	case SpellID::DISEASE:
 
- 		addLogLine(553, boost::logic::indeterminate);
 
- 		break;
 
- 	case SpellID::PARALYZE:
 
- 		addLogLine(563, boost::logic::indeterminate);
 
- 		break;
 
- 	case SpellID::AGE:
 
- 		{
 
- 			//"The %s shrivel with age, and lose %d hit points."
 
- 			MetaString line;
 
- 			attackedStack->addText(line, MetaString::GENERAL_TXT, 551);
 
- 			attackedStack->addNameReplacement(line);
 
- 			//todo: display effect from only this cast
 
- 			TBonusListPtr bl = attackedStack->getBonuses(Selector::type(Bonus::STACK_HEALTH));
 
- 			const int fullHP = bl->totalValue();
 
- 			bl->remove_if(Selector::source(Bonus::SPELL_EFFECT, SpellID::AGE));
 
- 			line.addReplacement(fullHP - bl->totalValue());
 
- 			logLines.push_back(line);
 
- 		}
 
- 		break;
 
- 	case SpellID::THUNDERBOLT:
 
- 		{
 
- 			addLogLine(-367, true);
 
- 			MetaString line;
 
- 			//todo: handle newlines in metastring
 
- 			std::string text = VLC->generaltexth->allTexts[343].substr(1, VLC->generaltexth->allTexts[343].size() - 1); //Does %d points of damage.
 
- 			line << text;
 
- 			line.addReplacement(damageToDisplay); //no more text afterwards
 
- 			logLines.push_back(line);
 
- 		}
 
- 		break;
 
- 	case SpellID::DISPEL_HELPFUL_SPELLS:
 
- 		addLogLine(-555, true);
 
- 		break;
 
- 	case SpellID::DEATH_STARE:
 
- 		if(damageToDisplay > 0)
 
- 		{
 
- 			MetaString line;
 
- 			if(damageToDisplay > 1)
 
- 			{
 
- 				line.addTxt(MetaString::GENERAL_TXT, 119); //%d %s die under the terrible gaze of the %s.
 
- 				line.addReplacement(damageToDisplay);
 
- 				attackedStack->addNameReplacement(line, true);
 
- 			}
 
- 			else
 
- 			{
 
- 				line.addTxt(MetaString::GENERAL_TXT, 118); //One %s dies under the terrible gaze of the %s.
 
- 				attackedStack->addNameReplacement(line, false);
 
- 			}
 
- 			parameters.caster->getCasterName(line);
 
- 			logLines.push_back(line);
 
- 		}
 
- 		break;
 
- 	default:
 
- 		displayDamage = true;
 
- 		battleLogDefault(logLines, parameters, attacked);
 
- 		break;
 
- 	}
 
- }
 
- void DefaultSpellMechanics::battleLogDefault(std::vector<MetaString> & logLines, const BattleSpellCastParameters & parameters, const std::vector<const CStack*> & attacked) const
 
- {
 
- 	MetaString line;
 
- 	parameters.caster->getCastDescription(owner, attacked, line);
 
- 	logLines.push_back(line);
 
- }
 
- void DefaultSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	if(owner->isOffensiveSpell())
 
- 		defaultDamageEffect(env, parameters, ctx);
 
- 	if(owner->hasEffects())
 
- 		defaultTimedEffect(env, parameters, ctx);
 
- }
 
- void DefaultSpellMechanics::defaultDamageEffect(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	const int rawDamage = parameters.getEffectValue();
 
- 	int chainLightningModifier = 0;
 
- 	for(auto & attackedCre : ctx.attackedCres)
 
- 	{
 
- 		BattleStackAttacked bsa;
 
- 		bsa.damageAmount = owner->adjustRawDamage(parameters.caster, attackedCre, rawDamage) >> chainLightningModifier;
 
- 		ctx.addDamageToDisplay(bsa.damageAmount);
 
- 		bsa.stackAttacked = (attackedCre)->ID;
 
- 		if(parameters.mode == ECastingMode::ENCHANTER_CASTING) //multiple damage spells cast
 
- 			bsa.attackerID = parameters.casterStack->ID;
 
- 		else
 
- 			bsa.attackerID = -1;
 
- 		(attackedCre)->prepareAttacked(bsa, env->getRandomGenerator());
 
- 		ctx.si.stacks.push_back(bsa);
 
- 		if(owner->id == SpellID::CHAIN_LIGHTNING)
 
- 			++chainLightningModifier;
 
- 	}
 
- }
 
- void DefaultSpellMechanics::defaultTimedEffect(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	SetStackEffect sse;
 
- 	//get default spell duration (spell power with bonuses for heroes)
 
- 	int duration = parameters.enchantPower;
 
- 	//generate actual stack bonuses
 
- 	{
 
- 		si32 maxDuration = 0;
 
- 		owner->getEffects(sse.effect, parameters.effectLevel, false, duration, &maxDuration);
 
- 		owner->getEffects(sse.cumulativeEffects, parameters.effectLevel, true, duration, &maxDuration);
 
- 		//if all spell effects have special duration, use it later for special bonuses
 
- 		duration = maxDuration;
 
- 	}
 
- 	//fix to original config: shield should display damage reduction
 
- 	if(owner->id == SpellID::SHIELD || owner->id == SpellID::AIR_SHIELD)
 
- 	{
 
- 		sse.effect.at(sse.effect.size() - 1).val = (100 - sse.effect.back().val);
 
- 	}
 
- 	//we need to know who cast Bind
 
- 	else if(owner->id == SpellID::BIND && parameters.casterStack)
 
- 	{
 
- 		sse.effect.at(sse.effect.size() - 1).additionalInfo = parameters.casterStack->ID;
 
- 	}
 
- 	std::shared_ptr<Bonus> bonus = nullptr;
 
- 	if(parameters.casterHero)
 
- 		bonus = parameters.casterHero->getBonusLocalFirst(Selector::typeSubtype(Bonus::SPECIAL_PECULIAR_ENCHANT, owner->id));
 
- 	//TODO does hero specialty should affects his stack casting spells?
 
- 	for(const CStack * affected : ctx.attackedCres)
 
- 	{
 
- 		si32 power = 0;
 
- 		sse.stacks.push_back(affected->ID);
 
- 		//Apply hero specials - peculiar enchants
 
- 		const ui8 tier = std::max((ui8)1, affected->getCreature()->level); //don't divide by 0 for certain creatures (commanders, war machines)
 
- 		if(bonus)
 
- 		{
 
- 			switch(bonus->additionalInfo)
 
- 			{
 
- 			case 0: //normal
 
- 				{
 
- 					switch(tier)
 
- 					{
 
- 					case 1:
 
- 					case 2:
 
- 						power = 3;
 
- 						break;
 
- 					case 3:
 
- 					case 4:
 
- 						power = 2;
 
- 						break;
 
- 					case 5:
 
- 					case 6:
 
- 						power = 1;
 
- 						break;
 
- 					}
 
- 					for(const Bonus & b : sse.effect)
 
- 					{
 
- 						Bonus specialBonus(b);
 
- 						specialBonus.val = power; //it doesn't necessarily make sense for some spells, use it wisely
 
- 						specialBonus.turnsRemain = duration;
 
- 						sse.uniqueBonuses.push_back(std::pair<ui32, Bonus>(affected->ID, specialBonus)); //additional premy to given effect
 
- 					}
 
- 					for(const Bonus & b : sse.cumulativeEffects)
 
- 					{
 
- 						Bonus specialBonus(b);
 
- 						specialBonus.val = power; //it doesn't necessarily make sense for some spells, use it wisely
 
- 						specialBonus.turnsRemain = duration;
 
- 						sse.cumulativeUniqueBonuses.push_back(std::pair<ui32, Bonus>(affected->ID, specialBonus)); //additional premy to given effect
 
- 					}
 
- 				}
 
- 				break;
 
- 			case 1: //only Coronius as yet
 
- 				{
 
- 					power = std::max(5 - tier, 0);
 
- 					Bonus specialBonus(Bonus::N_TURNS, Bonus::PRIMARY_SKILL, Bonus::SPELL_EFFECT, power, owner->id, PrimarySkill::ATTACK);
 
- 					specialBonus.turnsRemain = duration;
 
- 					sse.uniqueBonuses.push_back(std::pair<ui32,Bonus>(affected->ID, specialBonus)); //additional attack to Slayer effect
 
- 				}
 
- 				break;
 
- 			}
 
- 		}
 
- 		if(parameters.casterHero && parameters.casterHero->hasBonusOfType(Bonus::SPECIAL_BLESS_DAMAGE, owner->id)) //TODO: better handling of bonus percentages
 
- 		{
 
- 			int damagePercent = parameters.casterHero->level * parameters.casterHero->valOfBonuses(Bonus::SPECIAL_BLESS_DAMAGE, owner->id.toEnum()) / tier;
 
- 			Bonus specialBonus(Bonus::N_TURNS, Bonus::CREATURE_DAMAGE, Bonus::SPELL_EFFECT, damagePercent, owner->id, 0, Bonus::PERCENT_TO_ALL);
 
- 			specialBonus.turnsRemain = duration;
 
- 			sse.uniqueBonuses.push_back(std::pair<ui32,Bonus>(affected->ID, specialBonus));
 
- 		}
 
- 	}
 
- 	if(!sse.stacks.empty())
 
- 		env->sendAndApply(&sse);
 
- }
 
- std::vector<BattleHex> DefaultSpellMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool *outDroppedHexes) const
 
- {
 
- 	using namespace SRSLPraserHelpers;
 
- 	std::vector<BattleHex> ret;
 
- 	std::string rng = owner->getLevelInfo(schoolLvl).range + ','; //copy + artificial comma for easier handling
 
- 	if(rng.size() >= 2 && rng[0] != 'X') //there is at least one hex in range (+artificial comma)
 
- 	{
 
- 		std::string number1, number2;
 
- 		int beg, end;
 
- 		bool readingFirst = true;
 
- 		for(auto & elem : rng)
 
- 		{
 
- 			if(std::isdigit(elem) ) //reading number
 
- 			{
 
- 				if(readingFirst)
 
- 					number1 += elem;
 
- 				else
 
- 					number2 += elem;
 
- 			}
 
- 			else if(elem == ',') //comma
 
- 			{
 
- 				//calculating variables
 
- 				if(readingFirst)
 
- 				{
 
- 					beg = atoi(number1.c_str());
 
- 					number1 = "";
 
- 				}
 
- 				else
 
- 				{
 
- 					end = atoi(number2.c_str());
 
- 					number2 = "";
 
- 				}
 
- 				//obtaining new hexes
 
- 				std::set<ui16> curLayer;
 
- 				if(readingFirst)
 
- 				{
 
- 					curLayer = getInRange(centralHex, beg, beg);
 
- 				}
 
- 				else
 
- 				{
 
- 					curLayer = getInRange(centralHex, beg, end);
 
- 					readingFirst = true;
 
- 				}
 
- 				//adding abtained hexes
 
- 				for(auto & curLayer_it : curLayer)
 
- 				{
 
- 					ret.push_back(curLayer_it);
 
- 				}
 
- 			}
 
- 			else if(elem == '-') //dash
 
- 			{
 
- 				beg = atoi(number1.c_str());
 
- 				number1 = "";
 
- 				readingFirst = false;
 
- 			}
 
- 		}
 
- 	}
 
- 	//remove duplicates (TODO check if actually needed)
 
- 	range::unique(ret);
 
- 	return ret;
 
- }
 
- std::vector<const CStack *> DefaultSpellMechanics::getAffectedStacks(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster, int spellLvl, BattleHex destination) const
 
- {
 
- 	std::vector<const CStack *> result = calculateAffectedStacks(cb, mode, caster, spellLvl, destination);
 
- 	CSpell::TargetInfo ti(owner, spellLvl, mode);
 
- 	auto predicate = [&, this](const CStack * s)->bool
 
- 	{
 
- 		const bool hitDirectly = ti.alwaysHitDirectly && s->coversPos(destination);
 
- 		const bool immune = (ESpellCastProblem::OK != owner->isImmuneByStack(caster, s));
 
- 		return !hitDirectly && immune;
 
- 	};
 
- 	vstd::erase_if(result, predicate);
 
- 	return result;
 
- }
 
- std::vector<const CStack *> DefaultSpellMechanics::calculateAffectedStacks(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster, int spellLvl, BattleHex destination) const
 
- {
 
- 	std::set<const CStack *> attackedCres;//std::set to exclude multiple occurrences of two hex creatures
 
- 	CSpell::TargetInfo ti(owner, spellLvl, mode);
 
- 	const auto side = cb->playerToSide(caster->getOwner());
 
- 	if(!side)
 
- 		return std::vector<const CStack *>();
 
- 	auto attackedHexes = rangeInHexes(destination, spellLvl, side.get());
 
- 	//hackfix for banned creature massive spells
 
- 	if(!ti.massive && owner->getLevelInfo(spellLvl).range == "X")
 
- 		attackedHexes.push_back(destination);
 
- 	auto mainFilter = [&](const CStack * s)
 
- 	{
 
- 		const bool ownerMatches = cb->battleMatchOwner(caster->getOwner(), s, owner->getPositiveness());
 
- 		const bool validTarget = s->isValidTarget(!ti.onlyAlive); //todo: this should be handled by spell class
 
- 		const bool positivenessFlag = !ti.smart || ownerMatches;
 
- 		return positivenessFlag && validTarget;
 
- 	};
 
- 	if(ti.type == CSpell::CREATURE && attackedHexes.size() == 1)
 
- 	{
 
- 		//for single target spells we must select one target. Alive stack is preferred (issue #1763)
 
- 		auto predicate = [&](const CStack * s)
 
- 		{
 
- 			return s->coversPos(attackedHexes.at(0)) && mainFilter(s);
 
- 		};
 
- 		TStacks stacks = cb->battleGetStacksIf(predicate);
 
- 		for(auto stack : stacks)
 
- 		{
 
- 			if(stack->alive())
 
- 			{
 
- 				attackedCres.insert(stack);
 
- 				break;
 
- 			}
 
- 		}
 
- 		if(attackedCres.empty() && !stacks.empty())
 
- 			attackedCres.insert(stacks.front());
 
- 	}
 
- 	else if(ti.massive)
 
- 	{
 
- 		TStacks stacks = cb->battleGetStacksIf(mainFilter);
 
- 		for(auto stack : stacks)
 
- 			attackedCres.insert(stack);
 
- 	}
 
- 	else //custom range from attackedHexes
 
- 	{
 
- 		for(BattleHex hex : attackedHexes)
 
- 		{
 
- 			if(const CStack * st = cb->battleGetStackByPos(hex, ti.onlyAlive))
 
- 				if(mainFilter(st))
 
- 					attackedCres.insert(st);
 
- 		}
 
- 	}
 
- 	std::vector<const CStack *> res;
 
- 	std::copy(attackedCres.begin(), attackedCres.end(), std::back_inserter(res));
 
- 	return res;
 
- }
 
- ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
 
- {
 
- 	//no problems by default, this method is for spell-specific problems
 
- 	return ESpellCastProblem::OK;
 
- }
 
- ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::canBeCastAt(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster, BattleHex destination) const
 
- {
 
- 	if(mode == ECastingMode::CREATURE_ACTIVE_CASTING || mode == ECastingMode::HERO_CASTING)
 
- 	{
 
- 		const auto level = caster->getSpellSchoolLevel(owner);
 
- 		std::vector<const CStack *> affected = getAffectedStacks(cb, mode, caster, level, destination);
 
- 		//allow to cast spell if it affects at least one smart target
 
- 		bool targetExists = false;
 
- 		for(const CStack * stack : affected)
 
- 		{
 
- 			targetExists = cb->battleMatchOwner(caster->getOwner(), stack, owner->getPositiveness());
 
- 			if(targetExists)
 
- 				break;
 
- 		}
 
- 		if(!targetExists)
 
- 			return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	}
 
- 	return ESpellCastProblem::OK;
 
- }
 
- ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
 
- {
 
- 	//by default use general algorithm
 
- 	return owner->internalIsImmune(caster, obj);
 
- }
 
- bool DefaultSpellMechanics::dispellSelector(const Bonus * bonus)
 
- {
 
- 	if(bonus->source == Bonus::SPELL_EFFECT)
 
- 	{
 
- 		const CSpell * sourceSpell = SpellID(bonus->sid).toSpell();
 
- 		if(!sourceSpell)
 
- 			return false;//error
 
- 		//Special case: DISRUPTING_RAY is "immune" to dispell
 
- 		//Other even PERMANENT effects can be removed (f.e. BIND)
 
- 		if(sourceSpell->id == SpellID::DISRUPTING_RAY)
 
- 			return false;
 
- 		//Special case:do not remove lifetime marker
 
- 		if(sourceSpell->id == SpellID::CLONE)
 
- 			return false;
 
- 		//stack may have inherited effects
 
- 		if(sourceSpell->isAdventureSpell())
 
- 			return false;
 
- 		return true;
 
- 	}
 
- 	//not spell effect
 
- 	return false;
 
- }
 
- void DefaultSpellMechanics::doDispell(BattleInfo * battle, const BattleSpellCast * packet, const CSelector & selector) const
 
- {
 
- 	for(auto stackID : packet->affectedCres)
 
- 	{
 
- 		CStack *s = battle->getStack(stackID);
 
- 		s->popBonuses(selector.And(dispellSelector));
 
- 	}
 
- }
 
- bool DefaultSpellMechanics::canDispell(const IBonusBearer * obj, const CSelector & selector, const std::string & cachingStr) const
 
- {
 
- 	return obj->hasBonus(selector.And(dispellSelector), Selector::all, cachingStr);
 
- }
 
- void DefaultSpellMechanics::handleMagicMirror(const SpellCastEnvironment * env, SpellCastContext & ctx, std::vector <const CStack*> & reflected) const
 
- {
 
- 	//reflection is applied only to negative spells
 
- 	//if it is actual spell and can be reflected to single target, no recurrence
 
- 	const bool tryMagicMirror = owner->isNegative() && owner->level && owner->getLevelInfo(0).range == "0";
 
- 	if(tryMagicMirror)
 
- 	{
 
- 		for(auto s : ctx.attackedCres)
 
- 		{
 
- 			const int mirrorChance = (s)->valOfBonuses(Bonus::MAGIC_MIRROR);
 
- 			if(env->getRandomGenerator().nextInt(99) < mirrorChance)
 
- 				reflected.push_back(s);
 
- 		}
 
- 		vstd::erase_if(ctx.attackedCres, [&reflected](const CStack * s)
 
- 		{
 
- 			return vstd::contains(reflected, s);
 
- 		});
 
- 		for(auto s : reflected)
 
- 			ctx.addCustomEffect(s, 3);
 
- 	}
 
- }
 
- void DefaultSpellMechanics::handleResistance(const SpellCastEnvironment * env, SpellCastContext & ctx) const
 
- {
 
- 	//checking if creatures resist
 
- 	//resistance is applied only to negative spells
 
- 	if(owner->isNegative())
 
- 	{
 
- 		std::vector <const CStack*> resisted;
 
- 		for(auto s : ctx.attackedCres)
 
- 		{
 
- 			//magic resistance
 
- 			const int prob = std::min((s)->magicResistance(), 100); //probability of resistance in %
 
- 			if(env->getRandomGenerator().nextInt(99) < prob)
 
- 			{
 
- 				resisted.push_back(s);
 
- 			}
 
- 		}
 
- 		vstd::erase_if(ctx.attackedCres, [&resisted](const CStack * s)
 
- 		{
 
- 			return vstd::contains(resisted, s);
 
- 		});
 
- 		for(auto s : resisted)
 
- 			ctx.addCustomEffect(s, 78);
 
- 	}
 
- }
 
- bool DefaultSpellMechanics::requiresCreatureTarget() const
 
- {
 
- 	//most spells affects creatures somehow regardless of Target Type
 
- 	//for few exceptions see overrides
 
- 	return true;
 
- }
 
- ///SpecialSpellMechanics
 
- SpecialSpellMechanics::SpecialSpellMechanics(const CSpell * s):
 
- 	DefaultSpellMechanics(s)
 
- {
 
- }
 
- ESpellCastProblem::ESpellCastProblem SpecialSpellMechanics::canBeCastAt(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster, BattleHex destination) const
 
- {
 
- 	//no problems by default
 
- 	//common problems handled by CSpell
 
- 	return ESpellCastProblem::OK;
 
- }
 
- std::vector<const CStack *> SpecialSpellMechanics::calculateAffectedStacks(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster, int spellLvl, BattleHex destination) const
 
- {
 
- 	return std::vector<const CStack *>();
 
- }
 
 
  |