| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857 | 
							- /*
 
-  * BattleSpellMechanics.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 "BattleSpellMechanics.h"
 
- #include "../NetPacks.h"
 
- #include "../BattleState.h"
 
- #include "../mapObjects/CGHeroInstance.h"
 
- #include "../mapObjects/CGTownInstance.h"
 
- ///HealingSpellMechanics
 
- void HealingSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	EHealLevel healLevel = getHealLevel(parameters.effectLevel);
 
- 	int hpGained = calculateHealedHP(env, parameters, ctx);
 
- 	StacksHealedOrResurrected shr;
 
- 	shr.lifeDrain = false;
 
- 	shr.tentHealing = false;
 
- 	const bool resurrect = (healLevel != EHealLevel::HEAL);
 
- 	for(auto & attackedCre : ctx.attackedCres)
 
- 	{
 
- 		StacksHealedOrResurrected::HealInfo hi;
 
- 		hi.stackID = (attackedCre)->ID;
 
- 		int stackHPgained = parameters.caster->getSpellBonus(owner, hpGained, attackedCre);
 
- 		hi.healedHP = attackedCre->calculateHealedHealthPoints(stackHPgained, resurrect);
 
- 		hi.lowLevelResurrection = (healLevel == EHealLevel::RESURRECT);
 
- 		shr.healedStacks.push_back(hi);
 
- 	}
 
- 	if(!shr.healedStacks.empty())
 
- 		env->sendAndApply(&shr);
 
- }
 
- int HealingSpellMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
 
- {
 
- 	if(parameters.effectValue != 0)
 
- 		return parameters.effectValue; //Archangel
 
- 	return owner->calculateRawEffectValue(parameters.effectLevel, parameters.effectPower); //???
 
- }
 
- ///AntimagicMechanics
 
- void AntimagicMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
 
- {
 
- 	DefaultSpellMechanics::applyBattle(battle, packet);
 
- 	doDispell(battle, packet, [this](const Bonus * b) -> bool
 
- 	{
 
- 		if(b->source == Bonus::SPELL_EFFECT)
 
- 		{
 
- 			return b->sid != owner->id; //effect from this spell
 
- 		}
 
- 		return false; //not a spell effect
 
- 	});
 
- }
 
- ///ChainLightningMechanics
 
- std::set<const CStack *> ChainLightningMechanics::getAffectedStacks(const CBattleInfoCallback * cb, SpellTargetingContext & ctx) const
 
- {
 
- 	std::set<const CStack* > attackedCres;
 
- 	std::set<BattleHex> possibleHexes;
 
- 	for(auto stack : cb->battleGetAllStacks())
 
- 	{
 
- 		if(stack->isValidTarget())
 
- 		{
 
- 			for(auto hex : stack->getHexes())
 
- 			{
 
- 				possibleHexes.insert (hex);
 
- 			}
 
- 		}
 
- 	}
 
- 	int targetsOnLevel[4] = {4, 4, 5, 5};
 
- 	BattleHex lightningHex = ctx.destination;
 
- 	for(int i = 0; i < targetsOnLevel[ctx.schoolLvl]; ++i)
 
- 	{
 
- 		auto stack = cb->battleGetStackByPos(lightningHex, true);
 
- 		if(!stack)
 
- 			break;
 
- 		attackedCres.insert (stack);
 
- 		for(auto hex : stack->getHexes())
 
- 		{
 
- 			possibleHexes.erase(hex); //can't hit same place twice
 
- 		}
 
- 		if(possibleHexes.empty()) //not enough targets
 
- 			break;
 
- 		lightningHex = BattleHex::getClosestTile(stack->attackerOwned, ctx.destination, possibleHexes);
 
- 	}
 
- 	return attackedCres;
 
- }
 
- ///CloneMechanics
 
- void CloneMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	const CStack * clonedStack = nullptr;
 
- 	if(ctx.attackedCres.size())
 
- 		clonedStack = *ctx.attackedCres.begin();
 
- 	if(!clonedStack)
 
- 	{
 
- 		env->complain ("No target stack to clone!");
 
- 		return;
 
- 	}
 
- 	const int attacker = !(bool)parameters.casterSide;
 
- 	BattleStackAdded bsa;
 
- 	bsa.creID = clonedStack->type->idNumber;
 
- 	bsa.attacker = attacker;
 
- 	bsa.summoned = true;
 
- 	bsa.pos = parameters.cb->getAvaliableHex(bsa.creID, attacker); //TODO: unify it
 
- 	bsa.amount = clonedStack->count;
 
- 	env->sendAndApply(&bsa);
 
- 	BattleSetStackProperty ssp;
 
- 	ssp.stackID = bsa.newStackID;//we know stack ID after apply
 
- 	ssp.which = BattleSetStackProperty::CLONED;
 
- 	ssp.val = 0;
 
- 	ssp.absolute = 1;
 
- 	env->sendAndApply(&ssp);
 
- 	ssp.stackID = clonedStack->ID;
 
- 	ssp.which = BattleSetStackProperty::HAS_CLONE;
 
- 	ssp.val = bsa.newStackID;
 
- 	ssp.absolute = 1;
 
- 	env->sendAndApply(&ssp);
 
- 	SetStackEffect sse;
 
- 	sse.stacks.push_back(bsa.newStackID);
 
- 	Bonus lifeTimeMarker(Bonus::N_TURNS, Bonus::NONE, Bonus::SPELL_EFFECT, 0, owner->id.num);
 
- 	lifeTimeMarker.turnsRemain = parameters.enchantPower;
 
- 	sse.effect.push_back(lifeTimeMarker);
 
- 	env->sendAndApply(&sse);
 
- }
 
- ESpellCastProblem::ESpellCastProblem CloneMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
 
- {
 
- 	//can't clone already cloned creature
 
- 	if(vstd::contains(obj->state, EBattleStackState::CLONED))
 
- 		return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	if(obj->cloneID != -1)
 
- 		return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	ui8 schoolLevel;
 
- 	if(caster)
 
- 	{
 
- 		schoolLevel = caster->getEffectLevel(owner);
 
- 	}
 
- 	else
 
- 	{
 
- 		schoolLevel = 3;//todo: remove
 
- 	}
 
- 	if(schoolLevel < 3)
 
- 	{
 
- 		int maxLevel = (std::max(schoolLevel, (ui8)1) + 4);
 
- 		int creLevel = obj->getCreature()->level;
 
- 		if(maxLevel < creLevel) //tier 1-5 for basic, 1-6 for advanced, any level for expert
 
- 			return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	}
 
- 	//use default algorithm only if there is no mechanics-related problem
 
- 	return DefaultSpellMechanics::isImmuneByStack(caster, obj);
 
- }
 
- ///CureMechanics
 
- void CureMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
 
- {
 
- 	DefaultSpellMechanics::applyBattle(battle, packet);
 
- 	doDispell(battle, packet, [](const Bonus * b) -> bool
 
- 	{
 
- 		if(b->source == Bonus::SPELL_EFFECT)
 
- 		{
 
- 			CSpell * sp = SpellID(b->sid).toSpell();
 
- 			return sp->isNegative();
 
- 		}
 
- 		return false; //not a spell effect
 
- 	});
 
- }
 
- HealingSpellMechanics::EHealLevel CureMechanics::getHealLevel(int effectLevel) const
 
- {
 
- 	return EHealLevel::HEAL;
 
- }
 
- ///DispellMechanics
 
- void DispellMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
 
- {
 
- 	DefaultSpellMechanics::applyBattle(battle, packet);
 
- 	doDispell(battle, packet, Selector::sourceType(Bonus::SPELL_EFFECT));
 
- }
 
- ESpellCastProblem::ESpellCastProblem DispellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
 
- {
 
- 	{
 
- 		//just in case
 
- 		if(!obj->alive())
 
- 			return ESpellCastProblem::WRONG_SPELL_TARGET;
 
- 	}
 
- 	//DISPELL ignores all immunities, except specific absolute immunity
 
- 	{
 
- 		//SPELL_IMMUNITY absolute case
 
- 		std::stringstream cachingStr;
 
- 		cachingStr << "type_" << Bonus::SPELL_IMMUNITY << "subtype_" << owner->id.toEnum() << "addInfo_1";
 
- 		if(obj->hasBonus(Selector::typeSubtypeInfo(Bonus::SPELL_IMMUNITY, owner->id.toEnum(), 1), cachingStr.str()))
 
- 			return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	}
 
- 	{
 
- 		std::stringstream cachingStr;
 
- 		cachingStr << "source_" << Bonus::SPELL_EFFECT;
 
- 		if(obj->hasBonus(Selector::sourceType(Bonus::SPELL_EFFECT), cachingStr.str()))
 
- 		{
 
- 			return ESpellCastProblem::OK;
 
- 		}
 
- 	}
 
- 	return ESpellCastProblem::WRONG_SPELL_TARGET;
 
- 	//any other immunities are ignored - do not execute default algorithm
 
- }
 
- void DispellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	if(parameters.spellLvl > 2)
 
- 	{
 
- 		//expert DISPELL also removes spell-created obstacles
 
- 		ObstaclesRemoved packet;
 
- 		for(const auto obstacle : parameters.cb->obstacles)
 
- 		{
 
- 			if(obstacle->obstacleType == CObstacleInstance::FIRE_WALL
 
- 				|| obstacle->obstacleType == CObstacleInstance::FORCE_FIELD
 
- 				|| obstacle->obstacleType == CObstacleInstance::LAND_MINE)
 
- 				packet.obstacles.insert(obstacle->uniqueID);
 
- 		}
 
- 		if(!packet.obstacles.empty())
 
- 			env->sendAndApply(&packet);
 
- 	}
 
- }
 
- ///EarthquakeMechanics
 
- void EarthquakeMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	if(nullptr == parameters.cb->battleGetDefendedTown())
 
- 	{
 
- 		env->complain("EarthquakeMechanics: not town siege");
 
- 		return;
 
- 	}
 
- 	if(CGTownInstance::NONE == parameters.cb->battleGetDefendedTown()->fortLevel())
 
- 	{
 
- 		env->complain("EarthquakeMechanics: town has no fort");
 
- 		return;
 
- 	}
 
- 	//start with all destructible parts
 
- 	std::set<EWallPart::EWallPart> possibleTargets =
 
- 	{
 
- 		EWallPart::KEEP,
 
- 		EWallPart::BOTTOM_TOWER,
 
- 		EWallPart::BOTTOM_WALL,
 
- 		EWallPart::BELOW_GATE,
 
- 		EWallPart::OVER_GATE,
 
- 		EWallPart::UPPER_WALL,
 
- 		EWallPart::UPPER_TOWER,
 
- 		EWallPart::GATE
 
- 	};
 
- 	assert(possibleTargets.size() == EWallPart::PARTS_COUNT);
 
- 	const int targetsToAttack = 2 + std::max<int>(parameters.spellLvl - 1, 0);
 
- 	CatapultAttack ca;
 
- 	ca.attacker = -1;
 
- 	for(int i = 0; i < targetsToAttack; i++)
 
- 	{
 
- 		//Any destructible part can be hit regardless of its HP. Multiple hit on same target is allowed.
 
- 		EWallPart::EWallPart target = *RandomGeneratorUtil::nextItem(possibleTargets, env->getRandomGenerator());
 
- 		auto & currentHP = parameters.cb->si.wallState;
 
- 		if(currentHP.at(target) == EWallState::DESTROYED || currentHP.at(target) == EWallState::NONE)
 
- 			continue;
 
- 		CatapultAttack::AttackInfo attackInfo;
 
- 		attackInfo.damageDealt = 1;
 
- 		attackInfo.attackedPart = target;
 
- 		attackInfo.destinationTile = parameters.cb->wallPartToBattleHex(target);
 
- 		ca.attackedParts.push_back(attackInfo);
 
- 		//removing creatures in turrets / keep if one is destroyed
 
- 		BattleHex posRemove;
 
- 		switch(target)
 
- 		{
 
- 		case EWallPart::KEEP:
 
- 			posRemove = -2;
 
- 			break;
 
- 		case EWallPart::BOTTOM_TOWER:
 
- 			posRemove = -3;
 
- 			break;
 
- 		case EWallPart::UPPER_TOWER:
 
- 			posRemove = -4;
 
- 			break;
 
- 		}
 
- 		if(posRemove != BattleHex::INVALID)
 
- 		{
 
- 			BattleStacksRemoved bsr;
 
- 			for(auto & elem : parameters.cb->stacks)
 
- 			{
 
- 				if(elem->position == posRemove)
 
- 				{
 
- 					bsr.stackIDs.insert(elem->ID);
 
- 					break;
 
- 				}
 
- 			}
 
- 			if(bsr.stackIDs.size() > 0)
 
- 				env->sendAndApply(&bsr);
 
- 		}
 
- 	};
 
- 	env->sendAndApply(&ca);
 
- }
 
- ESpellCastProblem::ESpellCastProblem EarthquakeMechanics::canBeCast(const CBattleInfoCallback * cb, const ISpellCaster * caster) const
 
- {
 
- 	if(nullptr == cb->battleGetDefendedTown())
 
- 	{
 
- 		return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	}
 
- 	if(CGTownInstance::NONE == cb->battleGetDefendedTown()->fortLevel())
 
- 	{
 
- 		return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	}
 
- 	CSpell::TargetInfo ti(owner, caster->getSpellSchoolLevel(owner));
 
- 	if(ti.smart)
 
- 	{
 
- 		//if spell targeting is smart, then only attacker can use it
 
- 		if(cb->playerToSide(caster->getOwner()) != 0)
 
- 			return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	}
 
- 	const auto attackableBattleHexes = cb->getAttackableBattleHexes();
 
- 	if(attackableBattleHexes.empty())
 
- 		return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	return ESpellCastProblem::OK;
 
- }
 
- bool EarthquakeMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;
 
- }
 
- ///HypnotizeMechanics
 
- ESpellCastProblem::ESpellCastProblem HypnotizeMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
 
- {
 
- 	//todo: maybe do not resist on passive cast
 
- 	if(nullptr != caster)
 
- 	{
 
- 		//TODO: what with other creatures casting hypnotize, Faerie Dragons style?
 
- 		ui64 subjectHealth = (obj->count - 1) * obj->MaxHealth() + obj->firstHPleft;
 
- 		//apply 'damage' bonus for hypnotize, including hero specialty
 
- 		ui64 maxHealth = caster->getSpellBonus(owner, owner->calculateRawEffectValue(caster->getEffectLevel(owner), caster->getEffectPower(owner)), obj);
 
- 		if (subjectHealth > maxHealth)
 
- 			return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	}
 
- 	return DefaultSpellMechanics::isImmuneByStack(caster, obj);
 
- }
 
- ///ObstacleMechanics
 
- ESpellCastProblem::ESpellCastProblem ObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
 
- {
 
- 	if(ctx.ti.clearAffected)
 
- 	{
 
- 		ui8 side = cb->playerToSide(ctx.caster->getOwner());
 
- 		bool hexesOutsideBattlefield = false;
 
- 		auto tilesThatMustBeClear = owner->rangeInHexes(ctx.destination, ctx.schoolLvl, side, &hexesOutsideBattlefield);
 
- 		for(BattleHex hex : tilesThatMustBeClear)
 
- 		{
 
- 			if(cb->battleGetStackByPos(hex, true) || !!cb->battleGetObstacleOnPos(hex, false) || !hex.isAvailable())
 
- 			{
 
- 				return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 			}
 
- 		}
 
- 		if(hexesOutsideBattlefield)
 
- 		{
 
- 			return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 		}
 
- 	}
 
- 	return ESpellCastProblem::OK;
 
- }
 
- void ObstacleMechanics::placeObstacle(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, const BattleHex & pos) const
 
- {
 
- 	const int obstacleIdToGive = parameters.cb->obstacles.size()+1;
 
- 	auto obstacle = std::make_shared<SpellCreatedObstacle>();
 
- 	setupObstacle(obstacle.get());
 
- 	obstacle->pos = pos;
 
- 	obstacle->casterSide = parameters.casterSide;
 
- 	obstacle->ID = owner->id;
 
- 	obstacle->spellLevel = parameters.effectLevel;
 
- 	obstacle->casterSpellPower = parameters.effectPower;
 
- 	obstacle->uniqueID = obstacleIdToGive;
 
- 	BattleObstaclePlaced bop;
 
- 	bop.obstacle = obstacle;
 
- 	env->sendAndApply(&bop);
 
- }
 
- ///PatchObstacleMechanics
 
- void PatchObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	std::vector<BattleHex> availableTiles;
 
- 	for(int i = 0; i < GameConstants::BFIELD_SIZE; i += 1)
 
- 	{
 
- 		BattleHex hex = i;
 
- 		if(hex.getX() > 0 && hex.getX() < 16 && !(parameters.cb->battleGetStackByPos(hex, false)) && !(parameters.cb->battleGetObstacleOnPos(hex, false)))
 
- 			availableTiles.push_back(hex);
 
- 	}
 
- 	boost::range::random_shuffle(availableTiles);
 
- 	const int patchesForSkill[] = {4, 4, 6, 8};
 
- 	const int patchesToPut = std::min<int>(patchesForSkill[parameters.spellLvl], availableTiles.size());
 
- 	//land mines or quicksand patches are handled as spell created obstacles
 
- 	for (int i = 0; i < patchesToPut; i++)
 
- 		placeObstacle(env, parameters, availableTiles.at(i));
 
- }
 
- ///LandMineMechanics
 
- bool LandMineMechanics::requiresCreatureTarget() const
 
- {
 
- 	return true;
 
- }
 
- void LandMineMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
 
- {
 
- 	obstacle->obstacleType = CObstacleInstance::LAND_MINE;
 
- 	obstacle->turnsRemaining = -1;
 
- 	obstacle->visibleForAnotherSide = false;
 
- }
 
- ///QuicksandMechanics
 
- bool QuicksandMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;
 
- }
 
- void QuicksandMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
 
- {
 
- 	obstacle->obstacleType = CObstacleInstance::QUICKSAND;
 
- 	obstacle->turnsRemaining = -1;
 
- 	obstacle->visibleForAnotherSide = false;
 
- }
 
- ///WallMechanics
 
- std::vector<BattleHex> WallMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool * outDroppedHexes) const
 
- {
 
- 	std::vector<BattleHex> ret;
 
- 	//Special case - shape of obstacle depends on caster's side
 
- 	//TODO make it possible through spell config
 
- 	BattleHex::EDir firstStep, secondStep;
 
- 	if(side)
 
- 	{
 
- 		firstStep = BattleHex::TOP_LEFT;
 
- 		secondStep = BattleHex::TOP_RIGHT;
 
- 	}
 
- 	else
 
- 	{
 
- 		firstStep = BattleHex::TOP_RIGHT;
 
- 		secondStep = BattleHex::TOP_LEFT;
 
- 	}
 
- 	//Adds hex to the ret if it's valid. Otherwise sets output arg flag if given.
 
- 	auto addIfValid = [&](BattleHex hex)
 
- 	{
 
- 		if(hex.isValid())
 
- 			ret.push_back(hex);
 
- 		else if(outDroppedHexes)
 
- 			*outDroppedHexes = true;
 
- 	};
 
- 	ret.push_back(centralHex);
 
- 	addIfValid(centralHex.moveInDir(firstStep, false));
 
- 	if(schoolLvl >= 2) //advanced versions of fire wall / force field cotnains of 3 hexes
 
- 		addIfValid(centralHex.moveInDir(secondStep, false)); //moveInDir function modifies subject hex
 
- 	return ret;
 
- }
 
- ///FireWallMechanics
 
- bool FireWallMechanics::requiresCreatureTarget() const
 
- {
 
- 	return true;
 
- }
 
- void FireWallMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	const BattleHex destination = parameters.getFirstDestinationHex();
 
- 	if(!destination.isValid())
 
- 	{
 
- 		env->complain("Invalid destination for FIRE_WALL");
 
- 		return;
 
- 	}
 
- 	//firewall is build from multiple obstacles - one fire piece for each affected hex
 
- 	auto affectedHexes = owner->rangeInHexes(destination, parameters.spellLvl, parameters.casterSide);
 
- 	for(BattleHex hex : affectedHexes)
 
- 		placeObstacle(env, parameters, hex);
 
- }
 
- void FireWallMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
 
- {
 
- 	obstacle->obstacleType = CObstacleInstance::FIRE_WALL;
 
- 	obstacle->turnsRemaining = 2;
 
- 	obstacle->visibleForAnotherSide = true;
 
- }
 
- ///ForceFieldMechanics
 
- bool ForceFieldMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;
 
- }
 
- void ForceFieldMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	const BattleHex destination = parameters.getFirstDestinationHex();
 
- 	if(!destination.isValid())
 
- 	{
 
- 		env->complain("Invalid destination for FORCE_FIELD");
 
- 		return;
 
- 	}
 
- 	placeObstacle(env, parameters, destination);
 
- }
 
- void ForceFieldMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
 
- {
 
- 	obstacle->obstacleType = CObstacleInstance::FORCE_FIELD;
 
- 	obstacle->turnsRemaining = 2;
 
- 	obstacle->visibleForAnotherSide = true;
 
- }
 
- ///RemoveObstacleMechanics
 
- void RemoveObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	if(auto obstacleToRemove = parameters.cb->battleGetObstacleOnPos(parameters.getFirstDestinationHex(), false))
 
- 	{
 
- 		if(canRemove(obstacleToRemove.get(), parameters.spellLvl))
 
- 		{
 
- 			ObstaclesRemoved obr;
 
- 			obr.obstacles.insert(obstacleToRemove->uniqueID);
 
- 			env->sendAndApply(&obr);
 
- 		}
 
- 		else
 
- 		{
 
- 			env->complain("Cant remove this obstacle!");
 
- 		}
 
- 	}
 
- 	else
 
- 		env->complain("There's no obstacle to remove!");
 
- }
 
- ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const ISpellCaster * caster) const
 
- {
 
- 	const int spellLevel = caster->getSpellSchoolLevel(owner);
 
- 	for(auto obstacle : cb->battleGetAllObstacles())
 
- 		if(canRemove(obstacle.get(), spellLevel))
 
- 			return ESpellCastProblem::OK;
 
- 	return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- }
 
- ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
 
- {
 
- 	if(auto obstacle = cb->battleGetObstacleOnPos(ctx.destination, false))
 
- 		if(canRemove(obstacle.get(), ctx.schoolLvl))
 
- 			return ESpellCastProblem::OK;
 
- 	return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- }
 
- bool RemoveObstacleMechanics::canRemove(const CObstacleInstance * obstacle, const int spellLevel) const
 
- {
 
- 	switch (obstacle->obstacleType)
 
- 	{
 
- 	case CObstacleInstance::ABSOLUTE_OBSTACLE: //cliff-like obstacles can't be removed
 
- 	case CObstacleInstance::MOAT:
 
- 		return false;
 
- 	case CObstacleInstance::USUAL:
 
- 		return true;
 
- 	case CObstacleInstance::FIRE_WALL:
 
- 		if(spellLevel >= 2)
 
- 			return true;
 
- 		break;
 
- 	case CObstacleInstance::QUICKSAND:
 
- 	case CObstacleInstance::LAND_MINE:
 
- 	case CObstacleInstance::FORCE_FIELD:
 
- 		if(spellLevel >= 3)
 
- 			return true;
 
- 		break;
 
- 	default:
 
- 		break;
 
- 	}
 
- 	return false;
 
- }
 
- bool RemoveObstacleMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;
 
- }
 
- ///RisingSpellMechanics
 
- HealingSpellMechanics::EHealLevel RisingSpellMechanics::getHealLevel(int effectLevel) const
 
- {
 
- 	//this may be even distinct class
 
- 	if((effectLevel <= 1) && (owner->id == SpellID::RESURRECTION))
 
- 		return EHealLevel::RESURRECT;
 
- 	return EHealLevel::TRUE_RESURRECT;
 
- }
 
- ///SacrificeMechanics
 
- ESpellCastProblem::ESpellCastProblem SacrificeMechanics::canBeCast(const CBattleInfoCallback * cb, const ISpellCaster * caster) const
 
- {
 
- 	// for sacrifice we have to check for 2 targets (one dead to resurrect and one living to destroy)
 
- 	bool targetExists = false;
 
- 	bool targetToSacrificeExists = false;
 
- 	for(const CStack * stack : cb->battleGetAllStacks())
 
- 	{
 
- 		//using isImmuneBy directly as this mechanics does not have overridden immunity check
 
- 		//therefore we do not need to check caster and casting mode
 
- 		//TODO: check that we really should check immunity for both stacks
 
- 		ESpellCastProblem::ESpellCastProblem res = owner->internalIsImmune(caster, stack);
 
- 		const bool immune =  ESpellCastProblem::OK != res && ESpellCastProblem::NOT_DECIDED != res;
 
- 		const bool casterStack = stack->owner == caster->getOwner();
 
- 		if(!immune && casterStack)
 
- 		{
 
- 			if(stack->alive())
 
- 				targetToSacrificeExists = true;
 
- 			else
 
- 				targetExists = true;
 
- 			if(targetExists && targetToSacrificeExists)
 
- 				break;
 
- 		}
 
- 	}
 
- 	if(targetExists && targetToSacrificeExists)
 
- 		return ESpellCastProblem::OK;
 
- 	else
 
- 		return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- }
 
- void SacrificeMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	const CStack * victim = nullptr;
 
- 	if(parameters.destinations.size() == 2)
 
- 	{
 
- 		victim = parameters.destinations[1].stackValue;
 
- 	}
 
- 	else
 
- 	{
 
- 		//todo: remove and report error
 
- 		victim = parameters.selectedStack;
 
- 	}
 
- 	if(nullptr == victim)
 
- 	{
 
- 		env->complain("SacrificeMechanics: No stack to sacrifice");
 
- 		return;
 
- 	}
 
- 	//resurrect target after basic checks
 
- 	RisingSpellMechanics::applyBattleEffects(env, parameters, ctx);
 
- 	//it is safe to remove even active stack
 
- 	BattleStacksRemoved bsr;
 
- 	bsr.stackIDs.insert(victim->ID);
 
- 	env->sendAndApply(&bsr);
 
- }
 
- int SacrificeMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
 
- {
 
- 	int res = 0;
 
- 	const CStack * victim = nullptr;
 
- 	if(parameters.destinations.size() == 2)
 
- 	{
 
- 		victim = parameters.destinations[1].stackValue;
 
- 	}
 
- 	else
 
- 	{
 
- 		//todo: remove and report error
 
- 		victim = parameters.selectedStack;
 
- 	}
 
- 	if(nullptr == victim)
 
- 	{
 
- 		env->complain("SacrificeMechanics: No stack to sacrifice");
 
- 		return 0;
 
- 	}
 
- 	res = (parameters.effectPower + victim->MaxHealth() + owner->getPower(parameters.effectLevel)) * victim->count;
 
- 	return res;
 
- }
 
- bool SacrificeMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;//canBeCast do all target existence checks
 
- }
 
- ///SpecialRisingSpellMechanics
 
- ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
 
- {
 
- 	const CStack * stack = cb->getStackIf([ctx](const CStack * s)
 
- 	{
 
- 		const bool ownerMatches = !ctx.ti.smart || s->getOwner() == ctx.caster->getOwner();
 
- 		return ownerMatches && s->isValidTarget(true) && s->coversPos(ctx.destination);
 
- 	});
 
- 	if(nullptr == stack)
 
- 		return ESpellCastProblem::NO_APPROPRIATE_TARGET;
 
- 	return ESpellCastProblem::OK;
 
- }
 
- ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
 
- {
 
- 	// following does apply to resurrect and animate dead(?) only
 
- 	// for sacrifice health calculation and health limit check don't matter
 
- 	if(obj->count >= obj->baseAmount)
 
- 		return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- 	//FIXME: Archangels can cast immune stack and this should be applied for them and not hero
 
- //	if(caster)
 
- //	{
 
- //		auto maxHealth = calculateHealedHP(caster, obj, nullptr);
 
- //		if (maxHealth < obj->MaxHealth()) //must be able to rise at least one full creature
 
- //			return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
 
- //	}
 
- 	return DefaultSpellMechanics::isImmuneByStack(caster,obj);
 
- }
 
- ///SummonMechanics
 
- ESpellCastProblem::ESpellCastProblem SummonMechanics::canBeCast(const CBattleInfoCallback * cb, const ISpellCaster * caster) const
 
- {
 
- 	//check if there are summoned elementals of other type
 
- 	auto otherSummoned = cb->battleGetStacksIf([caster, this](const CStack * st)
 
- 	{
 
- 		return (st->owner == caster->getOwner())
 
- 			&& (vstd::contains(st->state, EBattleStackState::SUMMONED))
 
- 			&& (!vstd::contains(st->state, EBattleStackState::CLONED))
 
- 			&& (st->getCreature()->idNumber != creatureToSummon);
 
- 	});
 
- 	if(!otherSummoned.empty())
 
- 		return ESpellCastProblem::ANOTHER_ELEMENTAL_SUMMONED;
 
- 	return ESpellCastProblem::OK;
 
- }
 
- void SummonMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	BattleStackAdded bsa;
 
- 	bsa.creID = creatureToSummon;
 
- 	bsa.attacker = !(bool)parameters.casterSide;
 
- 	bsa.summoned = true;
 
- 	bsa.pos = parameters.cb->getAvaliableHex(creatureToSummon, !(bool)parameters.casterSide); //TODO: unify it
 
- 	//TODO stack casting -> probably power will be zero; set the proper number of creatures manually
 
- 	int percentBonus = parameters.casterHero ? parameters.casterHero->valOfBonuses(Bonus::SPECIFIC_SPELL_DAMAGE, owner->id.toEnum()) : 0;
 
- 	bsa.amount = parameters.effectPower
 
- 		* owner->getPower(parameters.spellLvl)
 
- 		* (100 + percentBonus) / 100.0; //new feature - percentage bonus
 
- 	if(bsa.amount)
 
- 		env->sendAndApply(&bsa);
 
- 	else
 
- 		env->complain("Summoning didn't summon any!");
 
- }
 
- bool SummonMechanics::requiresCreatureTarget() const
 
- {
 
- 	return false;
 
- }
 
- ///TeleportMechanics
 
- void TeleportMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
 
- {
 
- 	//todo: check legal teleport
 
- 	if(parameters.destinations.size() == 2)
 
- 	{
 
- 		//first destination creature to move
 
- 		const CStack * target = parameters.destinations[0].stackValue;
 
- 		if(nullptr == target)
 
- 		{
 
- 			env->complain("TeleportMechanics: no stack to teleport");
 
- 			return;
 
- 		}
 
- 		//second destination hex to move to
 
- 		const BattleHex destination = parameters.destinations[1].hexValue;
 
- 		if(!destination.isValid())
 
- 		{
 
- 			env->complain("TeleportMechanics: invalid teleport destination");
 
- 			return;
 
- 		}
 
- 		BattleStackMoved bsm;
 
- 		bsm.distance = -1;
 
- 		bsm.stack = target->ID;
 
- 		std::vector<BattleHex> tiles;
 
- 		tiles.push_back(destination);
 
- 		bsm.tilesToMove = tiles;
 
- 		bsm.teleporting = true;
 
- 		env->sendAndApply(&bsm);
 
- 	}
 
- 	else
 
- 	{
 
- 		//todo: remove and report error
 
- 		BattleStackMoved bsm;
 
- 		bsm.distance = -1;
 
- 		bsm.stack = parameters.selectedStack->ID;
 
- 		std::vector<BattleHex> tiles;
 
- 		tiles.push_back(parameters.getFirstDestinationHex());
 
- 		bsm.tilesToMove = tiles;
 
- 		bsm.teleporting = true;
 
- 		env->sendAndApply(&bsm);
 
- 	}
 
- }
 
 
  |