CDefaultSpellMechanics.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. /*
  2. * CDefaultSpellMechanics.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 "CDefaultSpellMechanics.h"
  12. #include "../NetPacks.h"
  13. #include "../BattleState.h"
  14. namespace SRSLPraserHelpers
  15. {
  16. static int XYToHex(int x, int y)
  17. {
  18. return x + GameConstants::BFIELD_WIDTH * y;
  19. }
  20. static int XYToHex(std::pair<int, int> xy)
  21. {
  22. return XYToHex(xy.first, xy.second);
  23. }
  24. static int hexToY(int battleFieldPosition)
  25. {
  26. return battleFieldPosition/GameConstants::BFIELD_WIDTH;
  27. }
  28. static int hexToX(int battleFieldPosition)
  29. {
  30. int pos = battleFieldPosition - hexToY(battleFieldPosition) * GameConstants::BFIELD_WIDTH;
  31. return pos;
  32. }
  33. static std::pair<int, int> hexToPair(int battleFieldPosition)
  34. {
  35. return std::make_pair(hexToX(battleFieldPosition), hexToY(battleFieldPosition));
  36. }
  37. //moves hex by one hex in given direction
  38. //0 - left top, 1 - right top, 2 - right, 3 - right bottom, 4 - left bottom, 5 - left
  39. static std::pair<int, int> gotoDir(int x, int y, int direction)
  40. {
  41. switch(direction)
  42. {
  43. case 0: //top left
  44. return std::make_pair((y%2) ? x-1 : x, y-1);
  45. case 1: //top right
  46. return std::make_pair((y%2) ? x : x+1, y-1);
  47. case 2: //right
  48. return std::make_pair(x+1, y);
  49. case 3: //right bottom
  50. return std::make_pair((y%2) ? x : x+1, y+1);
  51. case 4: //left bottom
  52. return std::make_pair((y%2) ? x-1 : x, y+1);
  53. case 5: //left
  54. return std::make_pair(x-1, y);
  55. default:
  56. throw std::runtime_error("Disaster: wrong direction in SRSLPraserHelpers::gotoDir!\n");
  57. }
  58. }
  59. static std::pair<int, int> gotoDir(std::pair<int, int> xy, int direction)
  60. {
  61. return gotoDir(xy.first, xy.second, direction);
  62. }
  63. static bool isGoodHex(std::pair<int, int> xy)
  64. {
  65. return xy.first >=0 && xy.first < GameConstants::BFIELD_WIDTH && xy.second >= 0 && xy.second < GameConstants::BFIELD_HEIGHT;
  66. }
  67. //helper function for rangeInHexes
  68. static std::set<ui16> getInRange(unsigned int center, int low, int high)
  69. {
  70. std::set<ui16> ret;
  71. if(low == 0)
  72. {
  73. ret.insert(center);
  74. }
  75. std::pair<int, int> mainPointForLayer[6]; //A, B, C, D, E, F points
  76. for(auto & elem : mainPointForLayer)
  77. elem = hexToPair(center);
  78. for(int it=1; it<=high; ++it) //it - distance to the center
  79. {
  80. for(int b=0; b<6; ++b)
  81. mainPointForLayer[b] = gotoDir(mainPointForLayer[b], b);
  82. if(it>=low)
  83. {
  84. std::pair<int, int> curHex;
  85. //adding lines (A-b, B-c, C-d, etc)
  86. for(int v=0; v<6; ++v)
  87. {
  88. curHex = mainPointForLayer[v];
  89. for(int h=0; h<it; ++h)
  90. {
  91. if(isGoodHex(curHex))
  92. ret.insert(XYToHex(curHex));
  93. curHex = gotoDir(curHex, (v+2)%6);
  94. }
  95. }
  96. } //if(it>=low)
  97. }
  98. return ret;
  99. }
  100. }
  101. ///DefaultSpellMechanics
  102. void DefaultSpellMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  103. {
  104. if (packet->castedByHero)
  105. {
  106. if (packet->side < 2)
  107. {
  108. battle->sides[packet->side].castSpellsCount++;
  109. }
  110. }
  111. //handle countering spells
  112. for(auto stackID : packet->affectedCres)
  113. {
  114. if(vstd::contains(packet->resisted, stackID))
  115. continue;
  116. CStack * s = battle->getStack(stackID);
  117. s->popBonuses([&](const Bonus * b) -> bool
  118. {
  119. //check for each bonus if it should be removed
  120. const bool isSpellEffect = Selector::sourceType(Bonus::SPELL_EFFECT)(b);
  121. const int spellID = isSpellEffect ? b->sid : -1;
  122. return isSpellEffect && vstd::contains(owner->counteredSpells, spellID);
  123. });
  124. }
  125. }
  126. bool DefaultSpellMechanics::adventureCast(const SpellCastEnvironment * env, AdventureSpellCastParameters & parameters) const
  127. {
  128. if(!owner->isAdventureSpell())
  129. {
  130. env->complain("Attempt to cast non adventure spell in adventure mode");
  131. return false;
  132. }
  133. const CGHeroInstance * caster = parameters.caster;
  134. const int cost = caster->getSpellCost(owner);
  135. if(!caster->canCastThisSpell(owner))
  136. {
  137. env->complain("Hero cannot cast this spell!");
  138. return false;
  139. }
  140. if(caster->mana < cost)
  141. {
  142. env->complain("Hero doesn't have enough spell points to cast this spell!");
  143. return false;
  144. }
  145. {
  146. AdvmapSpellCast asc;
  147. asc.caster = caster;
  148. asc.spellID = owner->id;
  149. env->sendAndApply(&asc);
  150. }
  151. switch(applyAdventureEffects(env, parameters))
  152. {
  153. case ESpellCastResult::OK:
  154. {
  155. SetMana sm;
  156. sm.hid = caster->id;
  157. sm.absolute = false;
  158. sm.val = -cost;
  159. env->sendAndApply(&sm);
  160. return true;
  161. }
  162. break;
  163. case ESpellCastResult::CANCEL:
  164. return true;
  165. }
  166. return false;
  167. }
  168. ESpellCastResult DefaultSpellMechanics::applyAdventureEffects(const SpellCastEnvironment * env, AdventureSpellCastParameters & parameters) const
  169. {
  170. if(owner->hasEffects())
  171. {
  172. const int schoolLevel = parameters.caster->getSpellSchoolLevel(owner);
  173. std::vector<Bonus> bonuses;
  174. owner->getEffects(bonuses, schoolLevel);
  175. for(Bonus b : bonuses)
  176. {
  177. GiveBonus gb;
  178. gb.id = parameters.caster->id.getNum();
  179. gb.bonus = b;
  180. env->sendAndApply(&gb);
  181. }
  182. return ESpellCastResult::OK;
  183. }
  184. else
  185. {
  186. //There is no generic algorithm of adventure cast
  187. env->complain("Unimplemented adventure spell");
  188. return ESpellCastResult::ERROR;
  189. }
  190. }
  191. void DefaultSpellMechanics::battleCast(const SpellCastEnvironment * env, BattleSpellCastParameters & parameters) const
  192. {
  193. BattleSpellCast sc;
  194. sc.side = parameters.casterSide;
  195. sc.id = owner->id;
  196. sc.skill = parameters.spellLvl;
  197. sc.tile = parameters.destination;
  198. sc.dmgToDisplay = 0;
  199. sc.castedByHero = nullptr != parameters.caster;
  200. sc.casterStack = (parameters.casterStack ? parameters.casterStack->ID : -1);
  201. sc.manaGained = 0;
  202. int spellCost = 0;
  203. //calculate spell cost
  204. if(parameters.caster)
  205. {
  206. spellCost = parameters.cb->battleGetSpellCost(owner, parameters.caster);
  207. if(parameters.secHero && parameters.mode == ECastingMode::HERO_CASTING) //handle mana channel
  208. {
  209. int manaChannel = 0;
  210. for(const CStack * stack : parameters.cb->battleGetAllStacks(true)) //TODO: shouldn't bonus system handle it somehow?
  211. {
  212. if(stack->owner == parameters.secHero->tempOwner)
  213. {
  214. vstd::amax(manaChannel, stack->valOfBonuses(Bonus::MANA_CHANNELING));
  215. }
  216. }
  217. sc.manaGained = (manaChannel * spellCost) / 100;
  218. }
  219. }
  220. //calculating affected creatures for all spells
  221. //must be vector, as in Chain Lightning order matters
  222. std::vector<const CStack*> attackedCres; //CStack vector is somewhat more suitable than ID vector
  223. auto creatures = owner->getAffectedStacks(parameters.cb, parameters.mode, parameters.casterColor, parameters.spellLvl, parameters.destination, parameters.caster);
  224. std::copy(creatures.begin(), creatures.end(), std::back_inserter(attackedCres));
  225. for (auto cre : attackedCres)
  226. {
  227. sc.affectedCres.insert(cre->ID);
  228. }
  229. //checking if creatures resist
  230. //resistance is applied only to negative spells
  231. if(owner->isNegative())
  232. {
  233. for(auto s : attackedCres)
  234. {
  235. const int prob = std::min((s)->magicResistance(), 100); //probability of resistance in %
  236. if(env->getRandomGenerator().nextInt(99) < prob)
  237. {
  238. sc.resisted.push_back(s->ID);
  239. }
  240. }
  241. }
  242. StacksInjured si;
  243. SpellCastContext ctx(attackedCres, sc, si);
  244. applyBattleEffects(env, parameters, ctx);
  245. env->sendAndApply(&sc);
  246. //spend mana
  247. if(parameters.caster)
  248. {
  249. SetMana sm;
  250. sm.absolute = false;
  251. sm.hid = parameters.caster->id;
  252. sm.val = -spellCost;
  253. env->sendAndApply(&sm);
  254. if(sc.manaGained > 0)
  255. {
  256. assert(parameters.secHero);
  257. sm.hid = parameters.secHero->id;
  258. sm.val = sc.manaGained;
  259. env->sendAndApply(&sm);
  260. }
  261. }
  262. if(!si.stacks.empty()) //after spellcast info shows
  263. env->sendAndApply(&si);
  264. //reduce number of casts remaining
  265. //TODO: this should be part of BattleSpellCast apply
  266. if (parameters.mode == ECastingMode::CREATURE_ACTIVE_CASTING || parameters.mode == ECastingMode::ENCHANTER_CASTING)
  267. {
  268. assert(parameters.casterStack);
  269. BattleSetStackProperty ssp;
  270. ssp.stackID = parameters.casterStack->ID;
  271. ssp.which = BattleSetStackProperty::CASTS;
  272. ssp.val = -1;
  273. ssp.absolute = false;
  274. env->sendAndApply(&ssp);
  275. }
  276. //Magic Mirror effect
  277. if(owner->isNegative() && parameters.mode != ECastingMode::MAGIC_MIRROR && owner->level && owner->getLevelInfo(0).range == "0") //it is actual spell and can be reflected to single target, no recurrence
  278. {
  279. for(auto & attackedCre : attackedCres)
  280. {
  281. int mirrorChance = (attackedCre)->valOfBonuses(Bonus::MAGIC_MIRROR);
  282. if(mirrorChance > env->getRandomGenerator().nextInt(99))
  283. {
  284. std::vector<const CStack *> mirrorTargets;
  285. auto battleStacks = parameters.cb->battleGetAllStacks(true);
  286. for(auto & battleStack : battleStacks)
  287. {
  288. if(battleStack->owner == parameters.casterColor) //get enemy stacks which can be affected by this spell
  289. {
  290. if (ESpellCastProblem::OK == owner->isImmuneByStack(nullptr, battleStack))
  291. mirrorTargets.push_back(battleStack);
  292. }
  293. }
  294. if(!mirrorTargets.empty())
  295. {
  296. int targetHex = (*RandomGeneratorUtil::nextItem(mirrorTargets, env->getRandomGenerator()))->position;
  297. BattleSpellCastParameters mirrorParameters = parameters;
  298. mirrorParameters.spellLvl = 0;
  299. mirrorParameters.casterSide = 1-parameters.casterSide;
  300. mirrorParameters.casterColor = (attackedCre)->owner;
  301. mirrorParameters.caster = nullptr;
  302. mirrorParameters.destination = targetHex;
  303. mirrorParameters.secHero = parameters.caster;
  304. mirrorParameters.mode = ECastingMode::MAGIC_MIRROR;
  305. mirrorParameters.casterStack = (attackedCre);
  306. mirrorParameters.selectedStack = nullptr;
  307. battleCast(env, mirrorParameters);
  308. }
  309. }
  310. }
  311. }
  312. }
  313. int DefaultSpellMechanics::calculateDuration(const CGHeroInstance * caster, int usedSpellPower) const
  314. {
  315. if(!caster)
  316. {
  317. if (!usedSpellPower)
  318. return 3; //default duration of all creature spells
  319. else
  320. return usedSpellPower; //use creature spell power
  321. }
  322. switch(owner->id)
  323. {
  324. case SpellID::FRENZY:
  325. return 1;
  326. default: //other spells
  327. return caster->getPrimSkillLevel(PrimarySkill::SPELL_POWER) + caster->valOfBonuses(Bonus::SPELL_DURATION);
  328. }
  329. }
  330. ui32 DefaultSpellMechanics::calculateHealedHP(const CGHeroInstance* caster, const CStack* stack, const CStack* sacrificedStack) const
  331. {
  332. int healedHealth;
  333. if(!owner->isHealingSpell())
  334. {
  335. logGlobal->errorStream() << "calculateHealedHP called for nonhealing spell "<< owner->name;
  336. return 0;
  337. }
  338. const int spellPowerSkill = caster->getPrimSkillLevel(PrimarySkill::SPELL_POWER);
  339. const int levelPower = owner->getPower(caster->getSpellSchoolLevel(owner));
  340. if (owner->id == SpellID::SACRIFICE && sacrificedStack)
  341. healedHealth = (spellPowerSkill + sacrificedStack->MaxHealth() + levelPower) * sacrificedStack->count;
  342. else
  343. healedHealth = spellPowerSkill * owner->power + levelPower; //???
  344. healedHealth = owner->calculateBonus(healedHealth, caster, stack);
  345. return std::min<ui32>(healedHealth, stack->MaxHealth() - stack->firstHPleft + (owner->isRisingSpell() ? stack->baseAmount * stack->MaxHealth() : 0));
  346. }
  347. void DefaultSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  348. {
  349. //applying effects
  350. if(owner->isOffensiveSpell())
  351. {
  352. int spellDamage = 0;
  353. if(parameters.casterStack && parameters.mode != ECastingMode::MAGIC_MIRROR)
  354. {
  355. int unitSpellPower = parameters.casterStack->valOfBonuses(Bonus::SPECIFIC_SPELL_POWER, owner->id.toEnum());
  356. if(unitSpellPower)
  357. ctx.sc.dmgToDisplay = spellDamage = parameters.casterStack->count * unitSpellPower; //TODO: handle immunities
  358. else //Faerie Dragon
  359. {
  360. parameters.usedSpellPower = parameters.casterStack->valOfBonuses(Bonus::CREATURE_SPELL_POWER) * parameters.casterStack->count / 100;
  361. ctx.sc.dmgToDisplay = 0;
  362. }
  363. }
  364. int chainLightningModifier = 0;
  365. for(auto & attackedCre : ctx.attackedCres)
  366. {
  367. if(vstd::contains(ctx.sc.resisted, (attackedCre)->ID)) //this creature resisted the spell
  368. continue;
  369. BattleStackAttacked bsa;
  370. if(spellDamage)
  371. bsa.damageAmount = spellDamage >> chainLightningModifier;
  372. else
  373. bsa.damageAmount = owner->calculateDamage(parameters.caster, attackedCre, parameters.spellLvl, parameters.usedSpellPower) >> chainLightningModifier;
  374. ctx.sc.dmgToDisplay += bsa.damageAmount;
  375. bsa.stackAttacked = (attackedCre)->ID;
  376. if(parameters.mode == ECastingMode::ENCHANTER_CASTING) //multiple damage spells cast
  377. bsa.attackerID = parameters.casterStack->ID;
  378. else
  379. bsa.attackerID = -1;
  380. (attackedCre)->prepareAttacked(bsa, env->getRandomGenerator());
  381. ctx.si.stacks.push_back(bsa);
  382. if(owner->id == SpellID::CHAIN_LIGHTNING)
  383. ++chainLightningModifier;
  384. }
  385. }
  386. if(owner->hasEffects())
  387. {
  388. int stackSpellPower = 0;
  389. if(parameters.casterStack && parameters.mode != ECastingMode::MAGIC_MIRROR)
  390. {
  391. stackSpellPower = parameters.casterStack->valOfBonuses(Bonus::CREATURE_ENCHANT_POWER);
  392. }
  393. SetStackEffect sse;
  394. Bonus pseudoBonus;
  395. pseudoBonus.sid = owner->id;
  396. pseudoBonus.val = parameters.spellLvl;
  397. pseudoBonus.turnsRemain = calculateDuration(parameters.caster, stackSpellPower ? stackSpellPower : parameters.usedSpellPower);
  398. CStack::stackEffectToFeature(sse.effect, pseudoBonus);
  399. if(owner->id == SpellID::SHIELD || owner->id == SpellID::AIR_SHIELD)
  400. {
  401. sse.effect.back().val = (100 - sse.effect.back().val); //fix to original config: shield should display damage reduction
  402. }
  403. if(owner->id == SpellID::BIND && parameters.casterStack)//bind
  404. {
  405. sse.effect.back().additionalInfo = parameters.casterStack->ID; //we need to know who casted Bind
  406. }
  407. const Bonus * bonus = nullptr;
  408. if(parameters.caster)
  409. bonus = parameters.caster->getBonusLocalFirst(Selector::typeSubtype(Bonus::SPECIAL_PECULIAR_ENCHANT, owner->id));
  410. //TODO does hero specialty should affects his stack casting spells?
  411. si32 power = 0;
  412. for(const CStack * affected : ctx.attackedCres)
  413. {
  414. if(vstd::contains(ctx.sc.resisted, affected->ID)) //this creature resisted the spell
  415. continue;
  416. sse.stacks.push_back(affected->ID);
  417. //Apply hero specials - peculiar enchants
  418. const ui8 tier = std::max((ui8)1, affected->getCreature()->level); //don't divide by 0 for certain creatures (commanders, war machines)
  419. if(bonus)
  420. {
  421. switch(bonus->additionalInfo)
  422. {
  423. case 0: //normal
  424. {
  425. switch(tier)
  426. {
  427. case 1: case 2:
  428. power = 3;
  429. break;
  430. case 3: case 4:
  431. power = 2;
  432. break;
  433. case 5: case 6:
  434. power = 1;
  435. break;
  436. }
  437. Bonus specialBonus(sse.effect.back());
  438. specialBonus.val = power; //it doesn't necessarily make sense for some spells, use it wisely
  439. sse.uniqueBonuses.push_back (std::pair<ui32,Bonus> (affected->ID, specialBonus)); //additional premy to given effect
  440. }
  441. break;
  442. case 1: //only Coronius as yet
  443. {
  444. power = std::max(5 - tier, 0);
  445. Bonus specialBonus = CStack::featureGenerator(Bonus::PRIMARY_SKILL, PrimarySkill::ATTACK, power, pseudoBonus.turnsRemain);
  446. specialBonus.sid = owner->id;
  447. sse.uniqueBonuses.push_back(std::pair<ui32,Bonus> (affected->ID, specialBonus)); //additional attack to Slayer effect
  448. }
  449. break;
  450. }
  451. }
  452. if (parameters.caster && parameters.caster->hasBonusOfType(Bonus::SPECIAL_BLESS_DAMAGE, owner->id)) //TODO: better handling of bonus percentages
  453. {
  454. int damagePercent = parameters.caster->level * parameters.caster->valOfBonuses(Bonus::SPECIAL_BLESS_DAMAGE, owner->id.toEnum()) / tier;
  455. Bonus specialBonus = CStack::featureGenerator(Bonus::CREATURE_DAMAGE, 0, damagePercent, pseudoBonus.turnsRemain);
  456. specialBonus.valType = Bonus::PERCENT_TO_ALL;
  457. specialBonus.sid = owner->id;
  458. sse.uniqueBonuses.push_back (std::pair<ui32,Bonus> (affected->ID, specialBonus));
  459. }
  460. }
  461. if(!sse.stacks.empty())
  462. env->sendAndApply(&sse);
  463. }
  464. if(owner->isHealingSpell())
  465. {
  466. int hpGained = 0;
  467. if(parameters.casterStack)
  468. {
  469. int unitSpellPower = parameters.casterStack->valOfBonuses(Bonus::SPECIFIC_SPELL_POWER, owner->id.toEnum());
  470. if(unitSpellPower)
  471. hpGained = parameters.casterStack->count * unitSpellPower; //Archangel
  472. else //Faerie Dragon-like effect - unused so far
  473. parameters.usedSpellPower = parameters.casterStack->valOfBonuses(Bonus::CREATURE_SPELL_POWER) * parameters.casterStack->count / 100;
  474. }
  475. StacksHealedOrResurrected shr;
  476. shr.lifeDrain = false;
  477. shr.tentHealing = false;
  478. for(auto & attackedCre : ctx.attackedCres)
  479. {
  480. StacksHealedOrResurrected::HealInfo hi;
  481. hi.stackID = (attackedCre)->ID;
  482. if (parameters.casterStack) //casted by creature
  483. {
  484. const bool resurrect = owner->isRisingSpell();
  485. if (hpGained)
  486. {
  487. //archangel
  488. hi.healedHP = std::min<ui32>(hpGained, attackedCre->MaxHealth() - attackedCre->firstHPleft + (resurrect ? attackedCre->baseAmount * attackedCre->MaxHealth() : 0));
  489. }
  490. else
  491. {
  492. //any typical spell (commander's cure or animate dead)
  493. int healedHealth = parameters.usedSpellPower * owner->power + owner->getPower(parameters.spellLvl);
  494. hi.healedHP = std::min<ui32>(healedHealth, attackedCre->MaxHealth() - attackedCre->firstHPleft + (resurrect ? attackedCre->baseAmount * attackedCre->MaxHealth() : 0));
  495. }
  496. }
  497. else
  498. hi.healedHP = calculateHealedHP(parameters.caster, attackedCre, parameters.selectedStack); //Casted by hero
  499. hi.lowLevelResurrection = (parameters.spellLvl <= 1) && (owner->id != SpellID::ANIMATE_DEAD);
  500. shr.healedStacks.push_back(hi);
  501. }
  502. if(!shr.healedStacks.empty())
  503. env->sendAndApply(&shr);
  504. }
  505. }
  506. std::vector<BattleHex> DefaultSpellMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool *outDroppedHexes) const
  507. {
  508. using namespace SRSLPraserHelpers;
  509. std::vector<BattleHex> ret;
  510. std::string rng = owner->getLevelInfo(schoolLvl).range + ','; //copy + artificial comma for easier handling
  511. if(rng.size() >= 2 && rng[0] != 'X') //there is at lest one hex in range (+artificial comma)
  512. {
  513. std::string number1, number2;
  514. int beg, end;
  515. bool readingFirst = true;
  516. for(auto & elem : rng)
  517. {
  518. if(std::isdigit(elem) ) //reading number
  519. {
  520. if(readingFirst)
  521. number1 += elem;
  522. else
  523. number2 += elem;
  524. }
  525. else if(elem == ',') //comma
  526. {
  527. //calculating variables
  528. if(readingFirst)
  529. {
  530. beg = atoi(number1.c_str());
  531. number1 = "";
  532. }
  533. else
  534. {
  535. end = atoi(number2.c_str());
  536. number2 = "";
  537. }
  538. //obtaining new hexes
  539. std::set<ui16> curLayer;
  540. if(readingFirst)
  541. {
  542. curLayer = getInRange(centralHex, beg, beg);
  543. }
  544. else
  545. {
  546. curLayer = getInRange(centralHex, beg, end);
  547. readingFirst = true;
  548. }
  549. //adding abtained hexes
  550. for(auto & curLayer_it : curLayer)
  551. {
  552. ret.push_back(curLayer_it);
  553. }
  554. }
  555. else if(elem == '-') //dash
  556. {
  557. beg = atoi(number1.c_str());
  558. number1 = "";
  559. readingFirst = false;
  560. }
  561. }
  562. }
  563. //remove duplicates (TODO check if actually needed)
  564. range::unique(ret);
  565. return ret;
  566. }
  567. std::set<const CStack *> DefaultSpellMechanics::getAffectedStacks(SpellTargetingContext & ctx) const
  568. {
  569. std::set<const CStack* > attackedCres;//std::set to exclude multiple occurrences of two hex creatures
  570. const ui8 attackerSide = ctx.cb->playerToSide(ctx.casterColor) == 1;
  571. const auto attackedHexes = rangeInHexes(ctx.destination, ctx.schoolLvl, attackerSide);
  572. const CSpell::TargetInfo ti(owner, ctx.schoolLvl, ctx.mode);
  573. //TODO: more generic solution for mass spells
  574. if(owner->getLevelInfo(ctx.schoolLvl).range.size() > 1) //custom many-hex range
  575. {
  576. for(BattleHex hex : attackedHexes)
  577. {
  578. if(const CStack * st = ctx.cb->battleGetStackByPos(hex, ti.onlyAlive))
  579. {
  580. attackedCres.insert(st);
  581. }
  582. }
  583. }
  584. else if(ti.type == CSpell::CREATURE)
  585. {
  586. auto predicate = [=](const CStack * s){
  587. const bool positiveToAlly = owner->isPositive() && s->owner == ctx.casterColor;
  588. const bool negativeToEnemy = owner->isNegative() && s->owner != ctx.casterColor;
  589. const bool validTarget = s->isValidTarget(!ti.onlyAlive); //todo: this should be handled by spell class
  590. //for single target spells select stacks covering destination tile
  591. const bool rangeCovers = ti.massive || s->coversPos(ctx.destination);
  592. //handle smart targeting
  593. const bool positivenessFlag = !ti.smart || owner->isNeutral() || positiveToAlly || negativeToEnemy;
  594. return rangeCovers && positivenessFlag && validTarget;
  595. };
  596. TStacks stacks = ctx.cb->battleGetStacksIf(predicate);
  597. if(ti.massive)
  598. {
  599. //for massive spells add all targets
  600. for (auto stack : stacks)
  601. attackedCres.insert(stack);
  602. }
  603. else
  604. {
  605. //for single target spells we must select one target. Alive stack is preferred (issue #1763)
  606. for(auto stack : stacks)
  607. {
  608. if(stack->alive())
  609. {
  610. attackedCres.insert(stack);
  611. break;
  612. }
  613. }
  614. if(attackedCres.empty() && !stacks.empty())
  615. {
  616. attackedCres.insert(stacks.front());
  617. }
  618. }
  619. }
  620. else //custom range from attackedHexes
  621. {
  622. for(BattleHex hex : attackedHexes)
  623. {
  624. if(const CStack * st = ctx.cb->battleGetStackByPos(hex, ti.onlyAlive))
  625. attackedCres.insert(st);
  626. }
  627. }
  628. return attackedCres;
  629. }
  630. ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::canBeCasted(const CBattleInfoCallback * cb, PlayerColor player) const
  631. {
  632. //no problems by default, this method is for spell-specific problems
  633. return ESpellCastProblem::OK;
  634. }
  635. ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::isImmuneByStack(const CGHeroInstance * caster, const CStack * obj) const
  636. {
  637. //by default use general algorithm
  638. return owner->isImmuneBy(obj);
  639. }
  640. void DefaultSpellMechanics::doDispell(BattleInfo * battle, const BattleSpellCast * packet, const CSelector & selector) const
  641. {
  642. for(auto stackID : packet->affectedCres)
  643. {
  644. if(vstd::contains(packet->resisted, stackID))
  645. continue;
  646. CStack *s = battle->getStack(stackID);
  647. s->popBonuses(selector);
  648. }
  649. }