2
0

CDefaultSpellMechanics.cpp 21 KB

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