BattleSpellMechanics.cpp 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956
  1. /*
  2. * BattleSpellMechanics.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include "BattleSpellMechanics.h"
  12. #include "../NetPacks.h"
  13. #include "../BattleState.h"
  14. #include "../mapObjects/CGHeroInstance.h"
  15. #include "../mapObjects/CGTownInstance.h"
  16. ///HealingSpellMechanics
  17. void HealingSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  18. {
  19. EHealLevel healLevel = getHealLevel(parameters.effectLevel);
  20. int hpGained = calculateHealedHP(env, parameters, ctx);
  21. StacksHealedOrResurrected shr;
  22. shr.lifeDrain = false;
  23. shr.tentHealing = false;
  24. //special case for Archangel
  25. shr.cure = parameters.mode == ECastingMode::CREATURE_ACTIVE_CASTING && owner->id == SpellID::RESURRECTION;
  26. const bool resurrect = (healLevel != EHealLevel::HEAL);
  27. for(auto & attackedCre : ctx.attackedCres)
  28. {
  29. StacksHealedOrResurrected::HealInfo hi;
  30. hi.stackID = (attackedCre)->ID;
  31. int stackHPgained = parameters.caster->getSpellBonus(owner, hpGained, attackedCre);
  32. hi.healedHP = attackedCre->calculateHealedHealthPoints(stackHPgained, resurrect);
  33. hi.lowLevelResurrection = (healLevel == EHealLevel::RESURRECT);
  34. shr.healedStacks.push_back(hi);
  35. }
  36. if(!shr.healedStacks.empty())
  37. env->sendAndApply(&shr);
  38. }
  39. int HealingSpellMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
  40. {
  41. return parameters.getEffectValue();
  42. }
  43. ///AntimagicMechanics
  44. void AntimagicMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  45. {
  46. DefaultSpellMechanics::applyBattle(battle, packet);
  47. doDispell(battle, packet, [this](const Bonus *b) -> bool
  48. {
  49. if(b->source == Bonus::SPELL_EFFECT)
  50. {
  51. const CSpell * sourceSpell = SpellID(b->sid).toSpell();
  52. if(!sourceSpell)
  53. return false;//error
  54. //keep positive effects
  55. if(sourceSpell->isPositive())
  56. return false;
  57. //keep own effects
  58. if(sourceSpell == owner)
  59. return false;
  60. //remove all others
  61. return true;
  62. }
  63. return false; //not a spell effect
  64. });
  65. }
  66. ///ChainLightningMechanics
  67. std::vector<const CStack *> ChainLightningMechanics::calculateAffectedStacks(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  68. {
  69. std::vector<const CStack *> res;
  70. std::set<BattleHex> possibleHexes;
  71. for(auto stack : cb->battleGetAllStacks())
  72. {
  73. if(stack->isValidTarget())
  74. {
  75. for(auto hex : stack->getHexes())
  76. {
  77. possibleHexes.insert (hex);
  78. }
  79. }
  80. }
  81. int targetsOnLevel[4] = {4, 4, 5, 5};
  82. BattleHex lightningHex = ctx.destination;
  83. for(int i = 0; i < targetsOnLevel[ctx.schoolLvl]; ++i)
  84. {
  85. auto stack = cb->battleGetStackByPos(lightningHex, true);
  86. if(!stack)
  87. break;
  88. res.push_back(stack);
  89. for(auto hex : stack->getHexes())
  90. {
  91. possibleHexes.erase(hex); //can't hit same stack twice
  92. }
  93. if(possibleHexes.empty()) //not enough targets
  94. break;
  95. lightningHex = BattleHex::getClosestTile(stack->attackerOwned, lightningHex, possibleHexes);
  96. }
  97. return res;
  98. }
  99. ///CloneMechanics
  100. void CloneMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  101. {
  102. const CStack * clonedStack = nullptr;
  103. if(ctx.attackedCres.size())
  104. clonedStack = *ctx.attackedCres.begin();
  105. if(!clonedStack)
  106. {
  107. env->complain ("No target stack to clone!");
  108. return;
  109. }
  110. const int attacker = !(bool)parameters.casterSide;
  111. BattleStackAdded bsa;
  112. bsa.creID = clonedStack->type->idNumber;
  113. bsa.attacker = attacker;
  114. bsa.summoned = true;
  115. bsa.pos = parameters.cb->getAvaliableHex(bsa.creID, attacker); //TODO: unify it
  116. bsa.amount = clonedStack->count;
  117. env->sendAndApply(&bsa);
  118. BattleSetStackProperty ssp;
  119. ssp.stackID = bsa.newStackID;//we know stack ID after apply
  120. ssp.which = BattleSetStackProperty::CLONED;
  121. ssp.val = 0;
  122. ssp.absolute = 1;
  123. env->sendAndApply(&ssp);
  124. ssp.stackID = clonedStack->ID;
  125. ssp.which = BattleSetStackProperty::HAS_CLONE;
  126. ssp.val = bsa.newStackID;
  127. ssp.absolute = 1;
  128. env->sendAndApply(&ssp);
  129. SetStackEffect sse;
  130. sse.stacks.push_back(bsa.newStackID);
  131. Bonus lifeTimeMarker(Bonus::N_TURNS, Bonus::NONE, Bonus::SPELL_EFFECT, 0, owner->id.num);
  132. lifeTimeMarker.turnsRemain = parameters.enchantPower;
  133. sse.effect.push_back(lifeTimeMarker);
  134. env->sendAndApply(&sse);
  135. }
  136. ESpellCastProblem::ESpellCastProblem CloneMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  137. {
  138. //can't clone already cloned creature
  139. if(vstd::contains(obj->state, EBattleStackState::CLONED))
  140. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  141. if(obj->cloneID != -1)
  142. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  143. ui8 schoolLevel;
  144. if(caster)
  145. {
  146. schoolLevel = caster->getEffectLevel(owner);
  147. }
  148. else
  149. {
  150. schoolLevel = 3;//todo: remove
  151. }
  152. if(schoolLevel < 3)
  153. {
  154. int maxLevel = (std::max(schoolLevel, (ui8)1) + 4);
  155. int creLevel = obj->getCreature()->level;
  156. if(maxLevel < creLevel) //tier 1-5 for basic, 1-6 for advanced, any level for expert
  157. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  158. }
  159. //use default algorithm only if there is no mechanics-related problem
  160. return DefaultSpellMechanics::isImmuneByStack(caster, obj);
  161. }
  162. ///CureMechanics
  163. void CureMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  164. {
  165. DefaultSpellMechanics::applyBattle(battle, packet);
  166. doDispell(battle, packet, dispellSelector);
  167. }
  168. HealingSpellMechanics::EHealLevel CureMechanics::getHealLevel(int effectLevel) const
  169. {
  170. return EHealLevel::HEAL;
  171. }
  172. bool CureMechanics::dispellSelector(const Bonus * b)
  173. {
  174. if(b->source == Bonus::SPELL_EFFECT)
  175. {
  176. const CSpell * sp = SpellID(b->sid).toSpell();
  177. return sp && sp->isNegative();
  178. }
  179. return false; //not a spell effect
  180. }
  181. ESpellCastProblem::ESpellCastProblem CureMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  182. {
  183. //Selector method name is ok as cashing string. --AVS
  184. if(!obj->canBeHealed() && !canDispell(obj, dispellSelector, "CureMechanics::dispellSelector"))
  185. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  186. return DefaultSpellMechanics::isImmuneByStack(caster, obj);
  187. }
  188. ///DispellMechanics
  189. void DispellMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  190. {
  191. DefaultSpellMechanics::applyBattle(battle, packet);
  192. doDispell(battle, packet, Selector::all);
  193. }
  194. ESpellCastProblem::ESpellCastProblem DispellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  195. {
  196. //just in case
  197. if(!obj->alive())
  198. return ESpellCastProblem::WRONG_SPELL_TARGET;
  199. //DISPELL ignores all immunities, except specific absolute immunity
  200. {
  201. //SPELL_IMMUNITY absolute case
  202. std::stringstream cachingStr;
  203. cachingStr << "type_" << Bonus::SPELL_IMMUNITY << "subtype_" << owner->id.toEnum() << "addInfo_1";
  204. if(obj->hasBonus(Selector::typeSubtypeInfo(Bonus::SPELL_IMMUNITY, owner->id.toEnum(), 1), cachingStr.str()))
  205. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  206. }
  207. if(canDispell(obj, Selector::all, "DefaultSpellMechanics::dispellSelector"))
  208. return ESpellCastProblem::OK;
  209. else
  210. return ESpellCastProblem::WRONG_SPELL_TARGET;
  211. //any other immunities are ignored - do not execute default algorithm
  212. }
  213. void DispellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  214. {
  215. if(parameters.spellLvl > 2)
  216. {
  217. //expert DISPELL also removes spell-created obstacles
  218. ObstaclesRemoved packet;
  219. for(const auto obstacle : parameters.cb->obstacles)
  220. {
  221. if(obstacle->obstacleType == CObstacleInstance::FIRE_WALL
  222. || obstacle->obstacleType == CObstacleInstance::FORCE_FIELD
  223. || obstacle->obstacleType == CObstacleInstance::LAND_MINE)
  224. packet.obstacles.insert(obstacle->uniqueID);
  225. }
  226. if(!packet.obstacles.empty())
  227. env->sendAndApply(&packet);
  228. }
  229. }
  230. ///EarthquakeMechanics
  231. void EarthquakeMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  232. {
  233. if(nullptr == parameters.cb->battleGetDefendedTown())
  234. {
  235. env->complain("EarthquakeMechanics: not town siege");
  236. return;
  237. }
  238. if(CGTownInstance::NONE == parameters.cb->battleGetDefendedTown()->fortLevel())
  239. {
  240. env->complain("EarthquakeMechanics: town has no fort");
  241. return;
  242. }
  243. //start with all destructible parts
  244. std::set<EWallPart::EWallPart> possibleTargets =
  245. {
  246. EWallPart::KEEP,
  247. EWallPart::BOTTOM_TOWER,
  248. EWallPart::BOTTOM_WALL,
  249. EWallPart::BELOW_GATE,
  250. EWallPart::OVER_GATE,
  251. EWallPart::UPPER_WALL,
  252. EWallPart::UPPER_TOWER,
  253. EWallPart::GATE
  254. };
  255. assert(possibleTargets.size() == EWallPart::PARTS_COUNT);
  256. const int targetsToAttack = 2 + std::max<int>(parameters.spellLvl - 1, 0);
  257. CatapultAttack ca;
  258. ca.attacker = -1;
  259. for(int i = 0; i < targetsToAttack; i++)
  260. {
  261. //Any destructible part can be hit regardless of its HP. Multiple hit on same target is allowed.
  262. EWallPart::EWallPart target = *RandomGeneratorUtil::nextItem(possibleTargets, env->getRandomGenerator());
  263. auto & currentHP = parameters.cb->si.wallState;
  264. if(currentHP.at(target) == EWallState::DESTROYED || currentHP.at(target) == EWallState::NONE)
  265. continue;
  266. CatapultAttack::AttackInfo attackInfo;
  267. attackInfo.damageDealt = 1;
  268. attackInfo.attackedPart = target;
  269. attackInfo.destinationTile = parameters.cb->wallPartToBattleHex(target);
  270. ca.attackedParts.push_back(attackInfo);
  271. //removing creatures in turrets / keep if one is destroyed
  272. BattleHex posRemove;
  273. switch(target)
  274. {
  275. case EWallPart::KEEP:
  276. posRemove = -2;
  277. break;
  278. case EWallPart::BOTTOM_TOWER:
  279. posRemove = -3;
  280. break;
  281. case EWallPart::UPPER_TOWER:
  282. posRemove = -4;
  283. break;
  284. }
  285. if(posRemove != BattleHex::INVALID)
  286. {
  287. BattleStacksRemoved bsr;
  288. for(auto & elem : parameters.cb->stacks)
  289. {
  290. if(elem->position == posRemove)
  291. {
  292. bsr.stackIDs.insert(elem->ID);
  293. break;
  294. }
  295. }
  296. if(bsr.stackIDs.size() > 0)
  297. env->sendAndApply(&bsr);
  298. }
  299. };
  300. env->sendAndApply(&ca);
  301. }
  302. ESpellCastProblem::ESpellCastProblem EarthquakeMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  303. {
  304. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  305. {
  306. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  307. return ESpellCastProblem::INVALID;
  308. }
  309. if(nullptr == cb->battleGetDefendedTown())
  310. {
  311. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  312. }
  313. if(CGTownInstance::NONE == cb->battleGetDefendedTown()->fortLevel())
  314. {
  315. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  316. }
  317. CSpell::TargetInfo ti(owner, caster->getSpellSchoolLevel(owner));
  318. if(ti.smart)
  319. {
  320. //if spell targeting is smart, then only attacker can use it
  321. if(cb->playerToSide(caster->getOwner()) != 0)
  322. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  323. }
  324. const auto attackableBattleHexes = cb->getAttackableBattleHexes();
  325. if(attackableBattleHexes.empty())
  326. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  327. return ESpellCastProblem::OK;
  328. }
  329. bool EarthquakeMechanics::requiresCreatureTarget() const
  330. {
  331. return false;
  332. }
  333. ///HypnotizeMechanics
  334. ESpellCastProblem::ESpellCastProblem HypnotizeMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  335. {
  336. //todo: maybe do not resist on passive cast
  337. if(nullptr != caster)
  338. {
  339. //TODO: what with other creatures casting hypnotize, Faerie Dragons style?
  340. ui64 subjectHealth = (obj->count - 1) * obj->MaxHealth() + obj->firstHPleft;
  341. //apply 'damage' bonus for hypnotize, including hero specialty
  342. ui64 maxHealth = caster->getSpellBonus(owner, owner->calculateRawEffectValue(caster->getEffectLevel(owner), caster->getEffectPower(owner)), obj);
  343. if (subjectHealth > maxHealth)
  344. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  345. }
  346. return DefaultSpellMechanics::isImmuneByStack(caster, obj);
  347. }
  348. ///ObstacleMechanics
  349. ESpellCastProblem::ESpellCastProblem ObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  350. {
  351. ui8 side = cb->playerToSide(ctx.caster->getOwner());
  352. bool hexesOutsideBattlefield = false;
  353. auto tilesThatMustBeClear = owner->rangeInHexes(ctx.destination, ctx.schoolLvl, side, &hexesOutsideBattlefield);
  354. for(const BattleHex & hex : tilesThatMustBeClear)
  355. if(!isHexAviable(cb, hex, ctx.ti.clearAffected))
  356. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  357. if(hexesOutsideBattlefield)
  358. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  359. return ESpellCastProblem::OK;
  360. }
  361. bool ObstacleMechanics::isHexAviable(const CBattleInfoCallback * cb, const BattleHex & hex, const bool mustBeClear)
  362. {
  363. if(!hex.isAvailable())
  364. return false;
  365. if(!mustBeClear)
  366. return true;
  367. if(cb->battleGetStackByPos(hex, true))
  368. return false;
  369. auto obst = cb->battleGetObstacleOnPos(hex, false);
  370. if(obst /*&& obst->type != CObstacleInstance::MOAT*/)//todo: issue 2366, uncomment once obstacle tile sharing implemented
  371. return false;
  372. if(nullptr != cb->battleGetDefendedTown() && CGTownInstance::NONE != cb->battleGetDefendedTown()->fortLevel())
  373. {
  374. EWallPart::EWallPart part = cb->battleHexToWallPart(hex);
  375. if(part == EWallPart::INVALID)
  376. return true;//no fortification here
  377. else if(static_cast<int>(part) < 0)
  378. return false;//indestuctible part (cant be checked by battleGetWallState)
  379. else if(part == EWallPart::BOTTOM_TOWER || part == EWallPart::UPPER_TOWER)
  380. return false;//destructible, but should not be available
  381. else if(cb->battleGetWallState(part) != EWallState::DESTROYED && cb->battleGetWallState(part) != EWallState::NONE)
  382. return false;
  383. }
  384. return true;
  385. }
  386. void ObstacleMechanics::placeObstacle(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, const BattleHex & pos) const
  387. {
  388. const int obstacleIdToGive = parameters.cb->obstacles.size()+1;
  389. auto obstacle = std::make_shared<SpellCreatedObstacle>();
  390. setupObstacle(obstacle.get());
  391. obstacle->pos = pos;
  392. obstacle->casterSide = parameters.casterSide;
  393. obstacle->ID = owner->id;
  394. obstacle->spellLevel = parameters.effectLevel;
  395. obstacle->casterSpellPower = parameters.effectPower;
  396. obstacle->uniqueID = obstacleIdToGive;
  397. BattleObstaclePlaced bop;
  398. bop.obstacle = obstacle;
  399. env->sendAndApply(&bop);
  400. }
  401. ///PatchObstacleMechanics
  402. void PatchObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  403. {
  404. std::vector<BattleHex> availableTiles;
  405. for(int i = 0; i < GameConstants::BFIELD_SIZE; i += 1)
  406. {
  407. BattleHex hex = i;
  408. if(isHexAviable(parameters.cb, hex, true))
  409. availableTiles.push_back(hex);
  410. }
  411. RandomGeneratorUtil::randomShuffle(availableTiles, env->getRandomGenerator());
  412. const int patchesForSkill[] = {4, 4, 6, 8};
  413. const int patchesToPut = std::min<int>(patchesForSkill[parameters.spellLvl], availableTiles.size());
  414. //land mines or quicksand patches are handled as spell created obstacles
  415. for (int i = 0; i < patchesToPut; i++)
  416. placeObstacle(env, parameters, availableTiles.at(i));
  417. }
  418. ///LandMineMechanics
  419. ESpellCastProblem::ESpellCastProblem LandMineMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  420. {
  421. //LandMine are useless if enemy has native stack and can see mines, check for LandMine damage immunity is done in general way by CSpell
  422. const ui8 otherSide = !cb->playerToSide(caster->getOwner());
  423. if(cb->battleHasNativeStack(otherSide))
  424. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  425. return DefaultSpellMechanics::canBeCast(cb, mode, caster);
  426. }
  427. bool LandMineMechanics::requiresCreatureTarget() const
  428. {
  429. return true;
  430. }
  431. void LandMineMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  432. {
  433. obstacle->obstacleType = CObstacleInstance::LAND_MINE;
  434. obstacle->turnsRemaining = -1;
  435. obstacle->visibleForAnotherSide = false;
  436. }
  437. ///QuicksandMechanics
  438. bool QuicksandMechanics::requiresCreatureTarget() const
  439. {
  440. return false;
  441. }
  442. void QuicksandMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  443. {
  444. obstacle->obstacleType = CObstacleInstance::QUICKSAND;
  445. obstacle->turnsRemaining = -1;
  446. obstacle->visibleForAnotherSide = false;
  447. }
  448. ///WallMechanics
  449. std::vector<BattleHex> WallMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool * outDroppedHexes) const
  450. {
  451. std::vector<BattleHex> ret;
  452. //Special case - shape of obstacle depends on caster's side
  453. //TODO make it possible through spell config
  454. BattleHex::EDir firstStep, secondStep;
  455. if(side)
  456. {
  457. firstStep = BattleHex::TOP_LEFT;
  458. secondStep = BattleHex::TOP_RIGHT;
  459. }
  460. else
  461. {
  462. firstStep = BattleHex::TOP_RIGHT;
  463. secondStep = BattleHex::TOP_LEFT;
  464. }
  465. //Adds hex to the ret if it's valid. Otherwise sets output arg flag if given.
  466. auto addIfValid = [&](BattleHex hex)
  467. {
  468. if(hex.isValid())
  469. ret.push_back(hex);
  470. else if(outDroppedHexes)
  471. *outDroppedHexes = true;
  472. };
  473. ret.push_back(centralHex);
  474. addIfValid(centralHex.moveInDir(firstStep, false));
  475. if(schoolLvl >= 2) //advanced versions of fire wall / force field cotnains of 3 hexes
  476. addIfValid(centralHex.moveInDir(secondStep, false)); //moveInDir function modifies subject hex
  477. return ret;
  478. }
  479. ///FireWallMechanics
  480. bool FireWallMechanics::requiresCreatureTarget() const
  481. {
  482. return true;
  483. }
  484. void FireWallMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  485. {
  486. const BattleHex destination = parameters.getFirstDestinationHex();
  487. if(!destination.isValid())
  488. {
  489. env->complain("Invalid destination for FIRE_WALL");
  490. return;
  491. }
  492. //firewall is build from multiple obstacles - one fire piece for each affected hex
  493. auto affectedHexes = owner->rangeInHexes(destination, parameters.spellLvl, parameters.casterSide);
  494. for(BattleHex hex : affectedHexes)
  495. placeObstacle(env, parameters, hex);
  496. }
  497. void FireWallMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  498. {
  499. obstacle->obstacleType = CObstacleInstance::FIRE_WALL;
  500. obstacle->turnsRemaining = 2;
  501. obstacle->visibleForAnotherSide = true;
  502. }
  503. ///ForceFieldMechanics
  504. bool ForceFieldMechanics::requiresCreatureTarget() const
  505. {
  506. return false;
  507. }
  508. void ForceFieldMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  509. {
  510. const BattleHex destination = parameters.getFirstDestinationHex();
  511. if(!destination.isValid())
  512. {
  513. env->complain("Invalid destination for FORCE_FIELD");
  514. return;
  515. }
  516. placeObstacle(env, parameters, destination);
  517. }
  518. void ForceFieldMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  519. {
  520. obstacle->obstacleType = CObstacleInstance::FORCE_FIELD;
  521. obstacle->turnsRemaining = 2;
  522. obstacle->visibleForAnotherSide = true;
  523. }
  524. ///RemoveObstacleMechanics
  525. void RemoveObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  526. {
  527. if(auto obstacleToRemove = parameters.cb->battleGetObstacleOnPos(parameters.getFirstDestinationHex(), false))
  528. {
  529. if(canRemove(obstacleToRemove.get(), parameters.spellLvl))
  530. {
  531. ObstaclesRemoved obr;
  532. obr.obstacles.insert(obstacleToRemove->uniqueID);
  533. env->sendAndApply(&obr);
  534. }
  535. else
  536. {
  537. env->complain("Cant remove this obstacle!");
  538. }
  539. }
  540. else
  541. env->complain("There's no obstacle to remove!");
  542. }
  543. ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  544. {
  545. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  546. {
  547. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  548. return ESpellCastProblem::INVALID;
  549. }
  550. const int spellLevel = caster->getSpellSchoolLevel(owner);
  551. for(auto obstacle : cb->battleGetAllObstacles())
  552. if(canRemove(obstacle.get(), spellLevel))
  553. return ESpellCastProblem::OK;
  554. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  555. }
  556. ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  557. {
  558. if(auto obstacle = cb->battleGetObstacleOnPos(ctx.destination, false))
  559. if(canRemove(obstacle.get(), ctx.schoolLvl))
  560. return ESpellCastProblem::OK;
  561. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  562. }
  563. bool RemoveObstacleMechanics::canRemove(const CObstacleInstance * obstacle, const int spellLevel) const
  564. {
  565. switch (obstacle->obstacleType)
  566. {
  567. case CObstacleInstance::ABSOLUTE_OBSTACLE: //cliff-like obstacles can't be removed
  568. case CObstacleInstance::MOAT:
  569. return false;
  570. case CObstacleInstance::USUAL:
  571. return true;
  572. case CObstacleInstance::FIRE_WALL:
  573. if(spellLevel >= 2)
  574. return true;
  575. break;
  576. case CObstacleInstance::QUICKSAND:
  577. case CObstacleInstance::LAND_MINE:
  578. case CObstacleInstance::FORCE_FIELD:
  579. if(spellLevel >= 3)
  580. return true;
  581. break;
  582. default:
  583. break;
  584. }
  585. return false;
  586. }
  587. bool RemoveObstacleMechanics::requiresCreatureTarget() const
  588. {
  589. return false;
  590. }
  591. ///RisingSpellMechanics
  592. HealingSpellMechanics::EHealLevel RisingSpellMechanics::getHealLevel(int effectLevel) const
  593. {
  594. //this may be even distinct class
  595. if((effectLevel <= 1) && (owner->id == SpellID::RESURRECTION))
  596. return EHealLevel::RESURRECT;
  597. return EHealLevel::TRUE_RESURRECT;
  598. }
  599. ///SacrificeMechanics
  600. ESpellCastProblem::ESpellCastProblem SacrificeMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  601. {
  602. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  603. {
  604. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  605. return ESpellCastProblem::INVALID;
  606. }
  607. // for sacrifice we have to check for 2 targets (one dead to resurrect and one living to destroy)
  608. bool targetExists = false;
  609. bool targetToSacrificeExists = false;
  610. for(const CStack * stack : cb->battleGetAllStacks())
  611. {
  612. //using isImmuneBy directly as this mechanics does not have overridden immunity check
  613. //therefore we do not need to check caster and casting mode
  614. //TODO: check that we really should check immunity for both stacks
  615. ESpellCastProblem::ESpellCastProblem res = owner->internalIsImmune(caster, stack);
  616. const bool immune = ESpellCastProblem::OK != res && ESpellCastProblem::NOT_DECIDED != res;
  617. const bool casterStack = stack->owner == caster->getOwner();
  618. if(!immune && casterStack)
  619. {
  620. if(stack->alive())
  621. targetToSacrificeExists = true;
  622. else
  623. targetExists = true;
  624. if(targetExists && targetToSacrificeExists)
  625. break;
  626. }
  627. }
  628. if(targetExists && targetToSacrificeExists)
  629. return ESpellCastProblem::OK;
  630. else
  631. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  632. }
  633. void SacrificeMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  634. {
  635. const CStack * victim = nullptr;
  636. if(parameters.destinations.size() == 2)
  637. {
  638. victim = parameters.destinations[1].stackValue;
  639. }
  640. if(nullptr == victim)
  641. {
  642. env->complain("SacrificeMechanics: No stack to sacrifice");
  643. return;
  644. }
  645. //resurrect target after basic checks
  646. RisingSpellMechanics::applyBattleEffects(env, parameters, ctx);
  647. //it is safe to remove even active stack
  648. BattleStacksRemoved bsr;
  649. bsr.stackIDs.insert(victim->ID);
  650. env->sendAndApply(&bsr);
  651. }
  652. int SacrificeMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
  653. {
  654. const CStack * victim = nullptr;
  655. if(parameters.destinations.size() == 2)
  656. {
  657. victim = parameters.destinations[1].stackValue;
  658. }
  659. if(nullptr == victim)
  660. {
  661. env->complain("SacrificeMechanics: No stack to sacrifice");
  662. return 0;
  663. }
  664. return (parameters.effectPower + victim->MaxHealth() + owner->getPower(parameters.effectLevel)) * victim->count;
  665. }
  666. bool SacrificeMechanics::requiresCreatureTarget() const
  667. {
  668. return false;//canBeCast do all target existence checks
  669. }
  670. ///SpecialRisingSpellMechanics
  671. ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  672. {
  673. //find alive possible target
  674. const CStack * stackToHeal = cb->getStackIf([ctx, this](const CStack * s)
  675. {
  676. const bool ownerMatches = !ctx.ti.smart || s->owner == ctx.caster->getOwner();
  677. return ownerMatches && s->isValidTarget(false) && s->coversPos(ctx.destination) && ESpellCastProblem::OK == owner->isImmuneByStack(ctx.caster, s);
  678. });
  679. if(nullptr == stackToHeal)
  680. {
  681. //find dead possible target if there is no alive target
  682. stackToHeal = cb->getStackIf([ctx, this](const CStack * s)
  683. {
  684. const bool ownerMatches = !ctx.ti.smart || s->owner == ctx.caster->getOwner();
  685. return ownerMatches && s->isValidTarget(true) && s->coversPos(ctx.destination) && ESpellCastProblem::OK == owner->isImmuneByStack(ctx.caster, s);
  686. });
  687. //we have found dead target
  688. if(nullptr != stackToHeal)
  689. {
  690. for(const BattleHex & hex : stackToHeal->getHexes())
  691. {
  692. const CStack * other = cb->getStackIf([hex, stackToHeal](const CStack * s)
  693. {
  694. return s->isValidTarget(false) && s->coversPos(hex) && s != stackToHeal;
  695. });
  696. if(nullptr != other)
  697. return ESpellCastProblem::NO_APPROPRIATE_TARGET;//alive stack blocks resurrection
  698. }
  699. }
  700. }
  701. if(nullptr == stackToHeal)
  702. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  703. return ESpellCastProblem::OK;
  704. }
  705. ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  706. {
  707. // following does apply to resurrect and animate dead(?) only
  708. // for sacrifice health calculation and health limit check don't matter
  709. if(obj->count >= obj->baseAmount)
  710. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  711. //FIXME: Archangels can cast immune stack and this should be applied for them and not hero
  712. // if(caster)
  713. // {
  714. // auto maxHealth = calculateHealedHP(caster, obj, nullptr);
  715. // if (maxHealth < obj->MaxHealth()) //must be able to rise at least one full creature
  716. // return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  717. // }
  718. return DefaultSpellMechanics::isImmuneByStack(caster,obj);
  719. }
  720. ///SummonMechanics
  721. ESpellCastProblem::ESpellCastProblem SummonMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  722. {
  723. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  724. {
  725. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  726. return ESpellCastProblem::INVALID;
  727. }
  728. //check if there are summoned elementals of other type
  729. auto otherSummoned = cb->battleGetStacksIf([caster, this](const CStack * st)
  730. {
  731. return (st->owner == caster->getOwner())
  732. && (vstd::contains(st->state, EBattleStackState::SUMMONED))
  733. && (!vstd::contains(st->state, EBattleStackState::CLONED))
  734. && (st->getCreature()->idNumber != creatureToSummon);
  735. });
  736. if(!otherSummoned.empty())
  737. return ESpellCastProblem::ANOTHER_ELEMENTAL_SUMMONED;
  738. return ESpellCastProblem::OK;
  739. }
  740. void SummonMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  741. {
  742. BattleStackAdded bsa;
  743. bsa.creID = creatureToSummon;
  744. bsa.attacker = !(bool)parameters.casterSide;
  745. bsa.summoned = true;
  746. bsa.pos = parameters.cb->getAvaliableHex(creatureToSummon, !(bool)parameters.casterSide); //TODO: unify it
  747. //TODO stack casting -> probably power will be zero; set the proper number of creatures manually
  748. int percentBonus = parameters.casterHero ? parameters.casterHero->valOfBonuses(Bonus::SPECIFIC_SPELL_DAMAGE, owner->id.toEnum()) : 0;
  749. bsa.amount = parameters.effectPower
  750. * owner->getPower(parameters.spellLvl)
  751. * (100 + percentBonus) / 100.0; //new feature - percentage bonus
  752. if(bsa.amount)
  753. env->sendAndApply(&bsa);
  754. else
  755. env->complain("Summoning didn't summon any!");
  756. }
  757. bool SummonMechanics::requiresCreatureTarget() const
  758. {
  759. return false;
  760. }
  761. ///TeleportMechanics
  762. void TeleportMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  763. {
  764. if(parameters.destinations.size() == 2)
  765. {
  766. //first destination hex to move to
  767. const BattleHex destination = parameters.destinations[0].hexValue;
  768. if(!destination.isValid())
  769. {
  770. env->complain("TeleportMechanics: invalid teleport destination");
  771. return;
  772. }
  773. //second destination creature to move
  774. const CStack * target = parameters.destinations[1].stackValue;
  775. if(nullptr == target)
  776. {
  777. env->complain("TeleportMechanics: no stack to teleport");
  778. return;
  779. }
  780. if(!parameters.cb->battleCanTeleportTo(target, destination, parameters.effectLevel))
  781. {
  782. env->complain("TeleportMechanics: forbidden teleport");
  783. return;
  784. }
  785. BattleStackMoved bsm;
  786. bsm.distance = -1;
  787. bsm.stack = target->ID;
  788. std::vector<BattleHex> tiles;
  789. tiles.push_back(destination);
  790. bsm.tilesToMove = tiles;
  791. bsm.teleporting = true;
  792. env->sendAndApply(&bsm);
  793. }
  794. else
  795. {
  796. env->complain("TeleportMechanics: 2 destinations required.");
  797. return;
  798. }
  799. }
  800. ESpellCastProblem::ESpellCastProblem TeleportMechanics::canBeCast(const CBattleInfoCallback* cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  801. {
  802. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  803. {
  804. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  805. return ESpellCastProblem::INVALID;
  806. }
  807. return DefaultSpellMechanics::canBeCast(cb, mode, caster);
  808. }