BattleSpellMechanics.cpp 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974
  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 "../CStack.h"
  14. #include "../battle/BattleInfo.h"
  15. #include "../mapObjects/CGHeroInstance.h"
  16. #include "../mapObjects/CGTownInstance.h"
  17. ///HealingSpellMechanics
  18. void HealingSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  19. {
  20. EHealLevel healLevel = getHealLevel(parameters.effectLevel);
  21. int hpGained = calculateHealedHP(env, parameters, ctx);
  22. StacksHealedOrResurrected shr;
  23. shr.lifeDrain = false;
  24. shr.tentHealing = false;
  25. //special case for Archangel
  26. shr.cure = parameters.mode == ECastingMode::CREATURE_ACTIVE_CASTING && owner->id == SpellID::RESURRECTION;
  27. const bool resurrect = (healLevel != EHealLevel::HEAL);
  28. for(auto & attackedCre : ctx.attackedCres)
  29. {
  30. StacksHealedOrResurrected::HealInfo hi;
  31. hi.stackID = (attackedCre)->ID;
  32. int stackHPgained = parameters.caster->getSpellBonus(owner, hpGained, attackedCre);
  33. hi.healedHP = attackedCre->calculateHealedHealthPoints(stackHPgained, resurrect);
  34. hi.lowLevelResurrection = (healLevel == EHealLevel::RESURRECT);
  35. shr.healedStacks.push_back(hi);
  36. }
  37. if(!shr.healedStacks.empty())
  38. env->sendAndApply(&shr);
  39. }
  40. int HealingSpellMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
  41. {
  42. return parameters.getEffectValue();
  43. }
  44. ///AntimagicMechanics
  45. void AntimagicMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  46. {
  47. DefaultSpellMechanics::applyBattle(battle, packet);
  48. doDispell(battle, packet, [this](const Bonus *b) -> bool
  49. {
  50. if(b->source == Bonus::SPELL_EFFECT)
  51. {
  52. const CSpell * sourceSpell = SpellID(b->sid).toSpell();
  53. if(!sourceSpell)
  54. return false;//error
  55. //keep positive effects
  56. if(sourceSpell->isPositive())
  57. return false;
  58. //keep own effects
  59. if(sourceSpell == owner)
  60. return false;
  61. //remove all others
  62. return true;
  63. }
  64. return false; //not a spell effect
  65. });
  66. }
  67. ///ChainLightningMechanics
  68. std::vector<const CStack *> ChainLightningMechanics::calculateAffectedStacks(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  69. {
  70. std::vector<const CStack *> res;
  71. std::set<BattleHex> possibleHexes;
  72. for(auto stack : cb->battleGetAllStacks())
  73. {
  74. if(stack->isValidTarget())
  75. {
  76. for(auto hex : stack->getHexes())
  77. {
  78. possibleHexes.insert (hex);
  79. }
  80. }
  81. }
  82. int targetsOnLevel[4] = {4, 4, 5, 5};
  83. BattleHex lightningHex = ctx.destination;
  84. for(int i = 0; i < targetsOnLevel[ctx.schoolLvl]; ++i)
  85. {
  86. auto stack = cb->battleGetStackByPos(lightningHex, true);
  87. if(!stack)
  88. break;
  89. res.push_back(stack);
  90. for(auto hex : stack->getHexes())
  91. {
  92. possibleHexes.erase(hex); //can't hit same stack twice
  93. }
  94. if(possibleHexes.empty()) //not enough targets
  95. break;
  96. lightningHex = BattleHex::getClosestTile(stack->side, lightningHex, possibleHexes);
  97. }
  98. return res;
  99. }
  100. ///CloneMechanics
  101. void CloneMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  102. {
  103. const CStack * clonedStack = nullptr;
  104. if(ctx.attackedCres.size())
  105. clonedStack = *ctx.attackedCres.begin();
  106. if(!clonedStack)
  107. {
  108. env->complain ("No target stack to clone!");
  109. return;
  110. }
  111. BattleStackAdded bsa;
  112. bsa.creID = clonedStack->type->idNumber;
  113. bsa.side = parameters.casterSide;
  114. bsa.summoned = true;
  115. bsa.pos = parameters.cb->getAvaliableHex(bsa.creID, parameters.casterSide);
  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. const auto side = cb->playerToSide(caster->getOwner());
  321. if(!side)
  322. return ESpellCastProblem::INVALID;
  323. //if spell targeting is smart, then only attacker can use it
  324. if(side.get() != BattleSide::ATTACKER)
  325. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  326. }
  327. const auto attackableBattleHexes = cb->getAttackableBattleHexes();
  328. if(attackableBattleHexes.empty())
  329. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  330. return ESpellCastProblem::OK;
  331. }
  332. bool EarthquakeMechanics::requiresCreatureTarget() const
  333. {
  334. return false;
  335. }
  336. ///HypnotizeMechanics
  337. ESpellCastProblem::ESpellCastProblem HypnotizeMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  338. {
  339. //todo: maybe do not resist on passive cast
  340. if(nullptr != caster)
  341. {
  342. //TODO: what with other creatures casting hypnotize, Faerie Dragons style?
  343. ui32 subjectHealth = obj->totalHealth();
  344. //apply 'damage' bonus for hypnotize, including hero specialty
  345. ui32 maxHealth = caster->getSpellBonus(owner, owner->calculateRawEffectValue(caster->getEffectLevel(owner), caster->getEffectPower(owner)), obj);
  346. if (subjectHealth > maxHealth)
  347. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  348. }
  349. return DefaultSpellMechanics::isImmuneByStack(caster, obj);
  350. }
  351. ///ObstacleMechanics
  352. ESpellCastProblem::ESpellCastProblem ObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  353. {
  354. const auto side = cb->playerToSide(ctx.caster->getOwner());
  355. if(!side)
  356. return ESpellCastProblem::INVALID;
  357. bool hexesOutsideBattlefield = false;
  358. auto tilesThatMustBeClear = owner->rangeInHexes(ctx.destination, ctx.schoolLvl, side.get(), &hexesOutsideBattlefield);
  359. for(const BattleHex & hex : tilesThatMustBeClear)
  360. if(!isHexAviable(cb, hex, ctx.ti.clearAffected))
  361. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  362. if(hexesOutsideBattlefield)
  363. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  364. return ESpellCastProblem::OK;
  365. }
  366. bool ObstacleMechanics::isHexAviable(const CBattleInfoCallback * cb, const BattleHex & hex, const bool mustBeClear)
  367. {
  368. if(!hex.isAvailable())
  369. return false;
  370. if(!mustBeClear)
  371. return true;
  372. if(cb->battleGetStackByPos(hex, true))
  373. return false;
  374. auto obst = cb->battleGetAllObstaclesOnPos(hex, false);
  375. for(auto & i : obst)
  376. if(i->obstacleType != CObstacleInstance::MOAT)
  377. return false;
  378. if(cb->battleGetDefendedTown() != nullptr && cb->battleGetDefendedTown()->fortLevel() != CGTownInstance::NONE)
  379. {
  380. EWallPart::EWallPart part = cb->battleHexToWallPart(hex);
  381. if(part == EWallPart::INVALID || part == EWallPart::INDESTRUCTIBLE_PART_OF_GATE)
  382. return true;//no fortification here
  383. else if(static_cast<int>(part) < 0)
  384. return false;//indestuctible part (cant be checked by battleGetWallState)
  385. else if(part == EWallPart::BOTTOM_TOWER || part == EWallPart::UPPER_TOWER)
  386. return false;//destructible, but should not be available
  387. else if(cb->battleGetWallState(part) != EWallState::DESTROYED && cb->battleGetWallState(part) != EWallState::NONE)
  388. return false;
  389. }
  390. return true;
  391. }
  392. void ObstacleMechanics::placeObstacle(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, const BattleHex & pos) const
  393. {
  394. const int obstacleIdToGive = parameters.cb->obstacles.size()+1;
  395. auto obstacle = std::make_shared<SpellCreatedObstacle>();
  396. setupObstacle(obstacle.get());
  397. obstacle->pos = pos;
  398. obstacle->casterSide = parameters.casterSide;
  399. obstacle->ID = owner->id;
  400. obstacle->spellLevel = parameters.effectLevel;
  401. obstacle->casterSpellPower = parameters.effectPower;
  402. obstacle->uniqueID = obstacleIdToGive;
  403. BattleObstaclePlaced bop;
  404. bop.obstacle = obstacle;
  405. env->sendAndApply(&bop);
  406. }
  407. ///PatchObstacleMechanics
  408. void PatchObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  409. {
  410. std::vector<BattleHex> availableTiles;
  411. for(int i = 0; i < GameConstants::BFIELD_SIZE; i += 1)
  412. {
  413. BattleHex hex = i;
  414. if(isHexAviable(parameters.cb, hex, true))
  415. availableTiles.push_back(hex);
  416. }
  417. RandomGeneratorUtil::randomShuffle(availableTiles, env->getRandomGenerator());
  418. const int patchesForSkill[] = {4, 4, 6, 8};
  419. const int patchesToPut = std::min<int>(patchesForSkill[parameters.spellLvl], availableTiles.size());
  420. //land mines or quicksand patches are handled as spell created obstacles
  421. for (int i = 0; i < patchesToPut; i++)
  422. placeObstacle(env, parameters, availableTiles.at(i));
  423. }
  424. ///LandMineMechanics
  425. ESpellCastProblem::ESpellCastProblem LandMineMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  426. {
  427. //LandMine are useless if enemy has native stack and can see mines, check for LandMine damage immunity is done in general way by CSpell
  428. const auto side = cb->playerToSide(caster->getOwner());
  429. if(!side)
  430. return ESpellCastProblem::INVALID;
  431. const ui8 otherSide = cb->otherSide(side.get());
  432. if(cb->battleHasNativeStack(otherSide))
  433. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  434. return DefaultSpellMechanics::canBeCast(cb, mode, caster);
  435. }
  436. bool LandMineMechanics::requiresCreatureTarget() const
  437. {
  438. return true;
  439. }
  440. void LandMineMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  441. {
  442. obstacle->obstacleType = CObstacleInstance::LAND_MINE;
  443. obstacle->turnsRemaining = -1;
  444. obstacle->visibleForAnotherSide = false;
  445. }
  446. ///QuicksandMechanics
  447. bool QuicksandMechanics::requiresCreatureTarget() const
  448. {
  449. return false;
  450. }
  451. void QuicksandMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  452. {
  453. obstacle->obstacleType = CObstacleInstance::QUICKSAND;
  454. obstacle->turnsRemaining = -1;
  455. obstacle->visibleForAnotherSide = false;
  456. }
  457. ///WallMechanics
  458. std::vector<BattleHex> WallMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool * outDroppedHexes) const
  459. {
  460. std::vector<BattleHex> ret;
  461. //Special case - shape of obstacle depends on caster's side
  462. //TODO make it possible through spell config
  463. BattleHex::EDir firstStep, secondStep;
  464. if(side)
  465. {
  466. firstStep = BattleHex::TOP_LEFT;
  467. secondStep = BattleHex::TOP_RIGHT;
  468. }
  469. else
  470. {
  471. firstStep = BattleHex::TOP_RIGHT;
  472. secondStep = BattleHex::TOP_LEFT;
  473. }
  474. //Adds hex to the ret if it's valid. Otherwise sets output arg flag if given.
  475. auto addIfValid = [&](BattleHex hex)
  476. {
  477. if(hex.isValid())
  478. ret.push_back(hex);
  479. else if(outDroppedHexes)
  480. *outDroppedHexes = true;
  481. };
  482. ret.push_back(centralHex);
  483. addIfValid(centralHex.moveInDirection(firstStep, false));
  484. if(schoolLvl >= 2) //advanced versions of fire wall / force field cotnains of 3 hexes
  485. addIfValid(centralHex.moveInDirection(secondStep, false)); //moveInDir function modifies subject hex
  486. return ret;
  487. }
  488. ///FireWallMechanics
  489. bool FireWallMechanics::requiresCreatureTarget() const
  490. {
  491. return true;
  492. }
  493. void FireWallMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  494. {
  495. const BattleHex destination = parameters.getFirstDestinationHex();
  496. if(!destination.isValid())
  497. {
  498. env->complain("Invalid destination for FIRE_WALL");
  499. return;
  500. }
  501. //firewall is build from multiple obstacles - one fire piece for each affected hex
  502. auto affectedHexes = owner->rangeInHexes(destination, parameters.spellLvl, parameters.casterSide);
  503. for(BattleHex hex : affectedHexes)
  504. placeObstacle(env, parameters, hex);
  505. }
  506. void FireWallMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  507. {
  508. obstacle->obstacleType = CObstacleInstance::FIRE_WALL;
  509. obstacle->turnsRemaining = 2;
  510. obstacle->visibleForAnotherSide = true;
  511. }
  512. ///ForceFieldMechanics
  513. bool ForceFieldMechanics::requiresCreatureTarget() const
  514. {
  515. return false;
  516. }
  517. void ForceFieldMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  518. {
  519. const BattleHex destination = parameters.getFirstDestinationHex();
  520. if(!destination.isValid())
  521. {
  522. env->complain("Invalid destination for FORCE_FIELD");
  523. return;
  524. }
  525. placeObstacle(env, parameters, destination);
  526. }
  527. void ForceFieldMechanics::setupObstacle(SpellCreatedObstacle * obstacle) const
  528. {
  529. obstacle->obstacleType = CObstacleInstance::FORCE_FIELD;
  530. obstacle->turnsRemaining = 2;
  531. obstacle->visibleForAnotherSide = true;
  532. }
  533. ///RemoveObstacleMechanics
  534. void RemoveObstacleMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  535. {
  536. auto obstacleToRemove = parameters.cb->battleGetAllObstaclesOnPos(parameters.getFirstDestinationHex(), false);
  537. if(!obstacleToRemove.empty())
  538. {
  539. ObstaclesRemoved obr;
  540. bool complain = true;
  541. for(auto & i : obstacleToRemove)
  542. {
  543. if(canRemove(i.get(), parameters.spellLvl))
  544. {
  545. obr.obstacles.insert(i->uniqueID);
  546. complain = false;
  547. }
  548. }
  549. if(!complain)
  550. env->sendAndApply(&obr);
  551. else if(complain || obr.obstacles.empty())
  552. env->complain("Cant remove this obstacle!");
  553. }
  554. else
  555. env->complain("There's no obstacle to remove!");
  556. }
  557. ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  558. {
  559. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  560. {
  561. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  562. return ESpellCastProblem::INVALID;
  563. }
  564. const int spellLevel = caster->getSpellSchoolLevel(owner);
  565. for(auto obstacle : cb->battleGetAllObstacles())
  566. if(canRemove(obstacle.get(), spellLevel))
  567. return ESpellCastProblem::OK;
  568. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  569. }
  570. ESpellCastProblem::ESpellCastProblem RemoveObstacleMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  571. {
  572. auto obstacles = cb->battleGetAllObstaclesOnPos(ctx.destination, false);
  573. if(!obstacles.empty())
  574. for(auto & i : obstacles)
  575. if(canRemove(i.get(), ctx.schoolLvl))
  576. return ESpellCastProblem::OK;
  577. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  578. }
  579. bool RemoveObstacleMechanics::canRemove(const CObstacleInstance * obstacle, const int spellLevel) const
  580. {
  581. switch (obstacle->obstacleType)
  582. {
  583. case CObstacleInstance::ABSOLUTE_OBSTACLE: //cliff-like obstacles can't be removed
  584. case CObstacleInstance::MOAT:
  585. return false;
  586. case CObstacleInstance::USUAL:
  587. return true;
  588. case CObstacleInstance::FIRE_WALL:
  589. if(spellLevel >= 2)
  590. return true;
  591. break;
  592. case CObstacleInstance::QUICKSAND:
  593. case CObstacleInstance::LAND_MINE:
  594. case CObstacleInstance::FORCE_FIELD:
  595. if(spellLevel >= 3)
  596. return true;
  597. break;
  598. default:
  599. break;
  600. }
  601. return false;
  602. }
  603. bool RemoveObstacleMechanics::requiresCreatureTarget() const
  604. {
  605. return false;
  606. }
  607. ///RisingSpellMechanics
  608. HealingSpellMechanics::EHealLevel RisingSpellMechanics::getHealLevel(int effectLevel) const
  609. {
  610. //this may be even distinct class
  611. if((effectLevel <= 1) && (owner->id == SpellID::RESURRECTION))
  612. return EHealLevel::RESURRECT;
  613. return EHealLevel::TRUE_RESURRECT;
  614. }
  615. ///SacrificeMechanics
  616. ESpellCastProblem::ESpellCastProblem SacrificeMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  617. {
  618. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  619. {
  620. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  621. return ESpellCastProblem::INVALID;
  622. }
  623. // for sacrifice we have to check for 2 targets (one dead to resurrect and one living to destroy)
  624. bool targetExists = false;
  625. bool targetToSacrificeExists = false;
  626. for(const CStack * stack : cb->battleGetAllStacks())
  627. {
  628. //using isImmuneBy directly as this mechanics does not have overridden immunity check
  629. //therefore we do not need to check caster and casting mode
  630. //TODO: check that we really should check immunity for both stacks
  631. ESpellCastProblem::ESpellCastProblem res = owner->internalIsImmune(caster, stack);
  632. const bool immune = ESpellCastProblem::OK != res && ESpellCastProblem::NOT_DECIDED != res;
  633. const bool casterStack = stack->owner == caster->getOwner();
  634. if(!immune && casterStack)
  635. {
  636. if(stack->alive())
  637. targetToSacrificeExists = true;
  638. else
  639. targetExists = true;
  640. if(targetExists && targetToSacrificeExists)
  641. break;
  642. }
  643. }
  644. if(targetExists && targetToSacrificeExists)
  645. return ESpellCastProblem::OK;
  646. else
  647. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  648. }
  649. void SacrificeMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  650. {
  651. const CStack * victim = nullptr;
  652. if(parameters.destinations.size() == 2)
  653. {
  654. victim = parameters.destinations[1].stackValue;
  655. }
  656. if(nullptr == victim)
  657. {
  658. env->complain("SacrificeMechanics: No stack to sacrifice");
  659. return;
  660. }
  661. //resurrect target after basic checks
  662. RisingSpellMechanics::applyBattleEffects(env, parameters, ctx);
  663. //it is safe to remove even active stack
  664. BattleStacksRemoved bsr;
  665. bsr.stackIDs.insert(victim->ID);
  666. env->sendAndApply(&bsr);
  667. }
  668. int SacrificeMechanics::calculateHealedHP(const SpellCastEnvironment* env, const BattleSpellCastParameters& parameters, SpellCastContext& ctx) const
  669. {
  670. const CStack * victim = nullptr;
  671. if(parameters.destinations.size() == 2)
  672. {
  673. victim = parameters.destinations[1].stackValue;
  674. }
  675. if(nullptr == victim)
  676. {
  677. env->complain("SacrificeMechanics: No stack to sacrifice");
  678. return 0;
  679. }
  680. return (parameters.effectPower + victim->MaxHealth() + owner->getPower(parameters.effectLevel)) * victim->count;
  681. }
  682. bool SacrificeMechanics::requiresCreatureTarget() const
  683. {
  684. return false;//canBeCast do all target existence checks
  685. }
  686. ///SpecialRisingSpellMechanics
  687. ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  688. {
  689. //find alive possible target
  690. const CStack * stackToHeal = cb->getStackIf([ctx, this](const CStack * s)
  691. {
  692. const bool ownerMatches = !ctx.ti.smart || s->owner == ctx.caster->getOwner();
  693. return ownerMatches && s->isValidTarget(false) && s->coversPos(ctx.destination) && ESpellCastProblem::OK == owner->isImmuneByStack(ctx.caster, s);
  694. });
  695. if(nullptr == stackToHeal)
  696. {
  697. //find dead possible target if there is no alive target
  698. stackToHeal = cb->getStackIf([ctx, this](const CStack * s)
  699. {
  700. const bool ownerMatches = !ctx.ti.smart || s->owner == ctx.caster->getOwner();
  701. return ownerMatches && s->isValidTarget(true) && s->coversPos(ctx.destination) && ESpellCastProblem::OK == owner->isImmuneByStack(ctx.caster, s);
  702. });
  703. //we have found dead target
  704. if(nullptr != stackToHeal)
  705. {
  706. for(const BattleHex & hex : stackToHeal->getHexes())
  707. {
  708. const CStack * other = cb->getStackIf([hex, stackToHeal](const CStack * s)
  709. {
  710. return s->isValidTarget(false) && s->coversPos(hex) && s != stackToHeal;
  711. });
  712. if(nullptr != other)
  713. return ESpellCastProblem::NO_APPROPRIATE_TARGET;//alive stack blocks resurrection
  714. }
  715. }
  716. }
  717. if(nullptr == stackToHeal)
  718. return ESpellCastProblem::NO_APPROPRIATE_TARGET;
  719. return ESpellCastProblem::OK;
  720. }
  721. ESpellCastProblem::ESpellCastProblem SpecialRisingSpellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  722. {
  723. // following does apply to resurrect and animate dead(?) only
  724. // for sacrifice health calculation and health limit check don't matter
  725. if(obj->count >= obj->baseAmount)
  726. return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  727. //FIXME: Archangels can cast immune stack and this should be applied for them and not hero
  728. // if(caster)
  729. // {
  730. // auto maxHealth = calculateHealedHP(caster, obj, nullptr);
  731. // if (maxHealth < obj->MaxHealth()) //must be able to rise at least one full creature
  732. // return ESpellCastProblem::STACK_IMMUNE_TO_SPELL;
  733. // }
  734. return DefaultSpellMechanics::isImmuneByStack(caster,obj);
  735. }
  736. ///SummonMechanics
  737. ESpellCastProblem::ESpellCastProblem SummonMechanics::canBeCast(const CBattleInfoCallback * cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  738. {
  739. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  740. {
  741. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  742. return ESpellCastProblem::INVALID;
  743. }
  744. //check if there are summoned elementals of other type
  745. auto otherSummoned = cb->battleGetStacksIf([caster, this](const CStack * st)
  746. {
  747. return (st->owner == caster->getOwner())
  748. && (vstd::contains(st->state, EBattleStackState::SUMMONED))
  749. && (!vstd::contains(st->state, EBattleStackState::CLONED))
  750. && (st->getCreature()->idNumber != creatureToSummon);
  751. });
  752. if(!otherSummoned.empty())
  753. return ESpellCastProblem::ANOTHER_ELEMENTAL_SUMMONED;
  754. return ESpellCastProblem::OK;
  755. }
  756. void SummonMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  757. {
  758. BattleStackAdded bsa;
  759. bsa.creID = creatureToSummon;
  760. bsa.side = parameters.casterSide;
  761. bsa.summoned = true;
  762. bsa.pos = parameters.cb->getAvaliableHex(creatureToSummon, parameters.casterSide); //TODO: unify it
  763. //TODO stack casting -> probably power will be zero; set the proper number of creatures manually
  764. int percentBonus = parameters.casterHero ? parameters.casterHero->valOfBonuses(Bonus::SPECIFIC_SPELL_DAMAGE, owner->id.toEnum()) : 0;
  765. bsa.amount = parameters.effectPower
  766. * owner->getPower(parameters.spellLvl)
  767. * (100 + percentBonus) / 100.0; //new feature - percentage bonus
  768. if(bsa.amount)
  769. env->sendAndApply(&bsa);
  770. else
  771. env->complain("Summoning didn't summon any!");
  772. }
  773. bool SummonMechanics::requiresCreatureTarget() const
  774. {
  775. return false;
  776. }
  777. ///TeleportMechanics
  778. void TeleportMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  779. {
  780. if(parameters.destinations.size() == 2)
  781. {
  782. //first destination hex to move to
  783. const BattleHex destination = parameters.destinations[0].hexValue;
  784. if(!destination.isValid())
  785. {
  786. env->complain("TeleportMechanics: invalid teleport destination");
  787. return;
  788. }
  789. //second destination creature to move
  790. const CStack * target = parameters.destinations[1].stackValue;
  791. if(nullptr == target)
  792. {
  793. env->complain("TeleportMechanics: no stack to teleport");
  794. return;
  795. }
  796. if(!parameters.cb->battleCanTeleportTo(target, destination, parameters.effectLevel))
  797. {
  798. env->complain("TeleportMechanics: forbidden teleport");
  799. return;
  800. }
  801. BattleStackMoved bsm;
  802. bsm.distance = -1;
  803. bsm.stack = target->ID;
  804. std::vector<BattleHex> tiles;
  805. tiles.push_back(destination);
  806. bsm.tilesToMove = tiles;
  807. bsm.teleporting = true;
  808. env->sendAndApply(&bsm);
  809. }
  810. else
  811. {
  812. env->complain("TeleportMechanics: 2 destinations required.");
  813. return;
  814. }
  815. }
  816. ESpellCastProblem::ESpellCastProblem TeleportMechanics::canBeCast(const CBattleInfoCallback* cb, const ECastingMode::ECastingMode mode, const ISpellCaster * caster) const
  817. {
  818. if(mode == ECastingMode::AFTER_ATTACK_CASTING || mode == ECastingMode::SPELL_LIKE_ATTACK || mode == ECastingMode::MAGIC_MIRROR)
  819. {
  820. logGlobal->warn("Invalid spell cast attempt: spell %s, mode %d", owner->name, mode); //should not even try to do it
  821. return ESpellCastProblem::INVALID;
  822. }
  823. return DefaultSpellMechanics::canBeCast(cb, mode, caster);
  824. }