BattleSpellMechanics.cpp 25 KB

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