CDefaultSpellMechanics.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787
  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 "../BattleState.h"
  13. #include "../CGeneralTextHandler.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. SpellCastContext::SpellCastContext(const DefaultSpellMechanics * mechanics_, const BattleSpellCastParameters & parameters):
  102. mechanics(mechanics_), attackedCres(), sc(), si()
  103. {
  104. prepareBattleCast(parameters);
  105. }
  106. void SpellCastContext::prepareBattleCast(const BattleSpellCastParameters & parameters)
  107. {
  108. sc.side = parameters.casterSide;
  109. sc.id = mechanics->owner->id;
  110. sc.skill = parameters.spellLvl;
  111. sc.tile = parameters.getFirstDestinationHex();
  112. sc.dmgToDisplay = 0;
  113. sc.castByHero = parameters.mode == ECastingMode::HERO_CASTING;
  114. sc.casterStack = (parameters.casterStack ? parameters.casterStack->ID : -1);
  115. sc.manaGained = 0;
  116. }
  117. void SpellCastContext::addDamageToDisplay(const si32 value)
  118. {
  119. sc.dmgToDisplay += value;
  120. }
  121. void SpellCastContext::setDamageToDisplay(const si32 value)
  122. {
  123. sc.dmgToDisplay = value;
  124. }
  125. void SpellCastContext::sendCastPacket(const SpellCastEnvironment * env)
  126. {
  127. for(auto sta : attackedCres)
  128. {
  129. sc.affectedCres.insert(sta->ID);
  130. }
  131. env->sendAndApply(&sc);
  132. }
  133. ///DefaultSpellMechanics
  134. void DefaultSpellMechanics::applyBattle(BattleInfo * battle, const BattleSpellCast * packet) const
  135. {
  136. if (packet->castByHero)
  137. {
  138. if (packet->side < 2)
  139. {
  140. battle->sides[packet->side].castSpellsCount++;
  141. }
  142. }
  143. //handle countering spells
  144. for(auto stackID : packet->affectedCres)
  145. {
  146. CStack * s = battle->getStack(stackID);
  147. s->popBonuses([&](const Bonus * b) -> bool
  148. {
  149. //check for each bonus if it should be removed
  150. const bool isSpellEffect = Selector::sourceType(Bonus::SPELL_EFFECT)(b);
  151. const int spellID = isSpellEffect ? b->sid : -1;
  152. //No exceptions, ANY spell can be countered, even if it can`t be dispelled.
  153. return isSpellEffect && vstd::contains(owner->counteredSpells, spellID);
  154. });
  155. }
  156. }
  157. void DefaultSpellMechanics::battleCast(const SpellCastEnvironment * env, BattleSpellCastParameters & parameters) const
  158. {
  159. logGlobal->debugStream() << "Started spell cast. Spell: "<<owner->name<<"; mode:"<<parameters.mode;
  160. if(nullptr == parameters.caster)
  161. {
  162. env->complain("No spell-caster provided.");
  163. return;
  164. }
  165. SpellCastContext ctx(this, parameters);
  166. //check it there is opponent hero
  167. const ui8 otherSide = 1-parameters.casterSide;
  168. const CGHeroInstance * otherHero = nullptr;
  169. if(parameters.cb->battleHasHero(otherSide))
  170. otherHero = parameters.cb->battleGetFightingHero(otherSide);
  171. int spellCost = 0;
  172. //calculate spell cost
  173. if(parameters.mode == ECastingMode::HERO_CASTING)
  174. {
  175. spellCost = parameters.cb->battleGetSpellCost(owner, parameters.casterHero);
  176. if(nullptr != otherHero) //handle mana channel
  177. {
  178. int manaChannel = 0;
  179. for(const CStack * stack : parameters.cb->battleGetAllStacks(true)) //TODO: shouldn't bonus system handle it somehow?
  180. {
  181. if(stack->owner == otherHero->tempOwner)
  182. {
  183. vstd::amax(manaChannel, stack->valOfBonuses(Bonus::MANA_CHANNELING));
  184. }
  185. }
  186. ctx.sc.manaGained = (manaChannel * spellCost) / 100;
  187. }
  188. }
  189. logGlobal->debugStream() << "spellCost: " << spellCost;
  190. auto creatures = owner->getAffectedStacks(parameters.cb, parameters.mode, parameters.caster, parameters.spellLvl, parameters.getFirstDestinationHex());
  191. std::copy(creatures.begin(), creatures.end(), std::back_inserter(ctx.attackedCres));
  192. logGlobal->debugStream() << "will affect: " << ctx.attackedCres.size() << " stacks";
  193. std::vector <const CStack*> reflected;//for magic mirror
  194. //checking if creatures resist
  195. handleResistance(env, ctx.attackedCres, ctx.sc);
  196. //it is actual spell and can be reflected to single target, no recurrence
  197. const bool tryMagicMirror = owner->isNegative() && owner->level && owner->getLevelInfo(0).range == "0";
  198. if(tryMagicMirror)
  199. {
  200. for(auto s : ctx.attackedCres)
  201. {
  202. const int mirrorChance = (s)->valOfBonuses(Bonus::MAGIC_MIRROR);
  203. if(env->getRandomGenerator().nextInt(99) < mirrorChance)
  204. reflected.push_back(s);
  205. }
  206. vstd::erase_if(ctx.attackedCres, [&reflected](const CStack * s)
  207. {
  208. return vstd::contains(reflected, s);
  209. });
  210. for(auto s : reflected)
  211. {
  212. BattleSpellCast::CustomEffect effect;
  213. effect.effect = 3;
  214. effect.stack = s->ID;
  215. ctx.sc.customEffects.push_back(effect);
  216. }
  217. }
  218. applyBattleEffects(env, parameters, ctx);
  219. ctx.sendCastPacket(env);
  220. if(parameters.mode == ECastingMode::HERO_CASTING)
  221. {
  222. //spend mana
  223. SetMana sm;
  224. sm.absolute = false;
  225. sm.hid = parameters.casterHero->id;
  226. sm.val = -spellCost;
  227. env->sendAndApply(&sm);
  228. if(ctx.sc.manaGained > 0)
  229. {
  230. assert(otherHero);
  231. sm.hid = otherHero->id;
  232. sm.val = ctx.sc.manaGained;
  233. env->sendAndApply(&sm);
  234. }
  235. }
  236. else if (parameters.mode == ECastingMode::CREATURE_ACTIVE_CASTING || parameters.mode == ECastingMode::ENCHANTER_CASTING)
  237. {
  238. //reduce number of casts remaining
  239. assert(parameters.casterStack);
  240. BattleSetStackProperty ssp;
  241. ssp.stackID = parameters.casterStack->ID;
  242. ssp.which = BattleSetStackProperty::CASTS;
  243. ssp.val = -1;
  244. ssp.absolute = false;
  245. env->sendAndApply(&ssp);
  246. }
  247. if(!ctx.si.stacks.empty()) //after spellcast info shows
  248. env->sendAndApply(&ctx.si);
  249. logGlobal->debugStream() << "Finished spell cast. Spell: "<<owner->name<<"; mode:"<<parameters.mode;
  250. //Magic Mirror effect
  251. for(auto & attackedCre : reflected)
  252. {
  253. TStacks mirrorTargets = parameters.cb->battleGetStacksIf([this, parameters](const CStack * battleStack)
  254. {
  255. //Get all enemy stacks. Magic mirror can reflect to immune creature (with no effect)
  256. return battleStack->owner == parameters.casterColor && battleStack->isValidTarget(false);
  257. });
  258. if(!mirrorTargets.empty())
  259. {
  260. int targetHex = (*RandomGeneratorUtil::nextItem(mirrorTargets, env->getRandomGenerator()))->position;
  261. BattleSpellCastParameters mirrorParameters(parameters.cb, attackedCre, owner);
  262. mirrorParameters.spellLvl = 0;
  263. mirrorParameters.aimToHex(targetHex);
  264. mirrorParameters.mode = ECastingMode::MAGIC_MIRROR;
  265. mirrorParameters.selectedStack = nullptr;
  266. mirrorParameters.spellLvl = parameters.spellLvl;
  267. mirrorParameters.effectLevel = parameters.effectLevel;
  268. mirrorParameters.effectPower = parameters.effectPower;
  269. mirrorParameters.effectValue = parameters.effectValue;
  270. mirrorParameters.enchantPower = parameters.enchantPower;
  271. castMagicMirror(env, mirrorParameters);
  272. }
  273. }
  274. }
  275. void DefaultSpellMechanics::battleLogSingleTarget(std::vector<std::string> & logLines, const BattleSpellCast * packet,
  276. const std::string & casterName, const CStack * attackedStack, bool & displayDamage) const
  277. {
  278. const std::string attackedName = attackedStack->getName();
  279. const std::string attackedNameSing = attackedStack->getCreature()->nameSing;
  280. const std::string attackedNamePl = attackedStack->getCreature()->namePl;
  281. auto getPluralFormat = [attackedStack](const int baseTextID) -> boost::format
  282. {
  283. return boost::format(VLC->generaltexth->allTexts[(attackedStack->count > 1 ? baseTextID + 1 : baseTextID)]);
  284. };
  285. auto logSimple = [&logLines, getPluralFormat, attackedName](const int baseTextID)
  286. {
  287. boost::format fmt = getPluralFormat(baseTextID);
  288. fmt % attackedName;
  289. logLines.push_back(fmt.str());
  290. };
  291. auto logPlural = [&logLines, attackedNamePl](const int baseTextID)
  292. {
  293. boost::format fmt(VLC->generaltexth->allTexts[baseTextID]);
  294. fmt % attackedNamePl;
  295. logLines.push_back(fmt.str());
  296. };
  297. displayDamage = false; //in most following cases damage info text is custom
  298. switch(owner->id)
  299. {
  300. case SpellID::STONE_GAZE:
  301. logSimple(558);
  302. break;
  303. case SpellID::POISON:
  304. logSimple(561);
  305. break;
  306. case SpellID::BIND:
  307. logPlural(560);//Roots and vines bind the %s to the ground!
  308. break;
  309. case SpellID::DISEASE:
  310. logSimple(553);
  311. break;
  312. case SpellID::PARALYZE:
  313. logSimple(563);
  314. break;
  315. case SpellID::AGE:
  316. {
  317. boost::format text = getPluralFormat(551);
  318. text % attackedName;
  319. //The %s shrivel with age, and lose %d hit points."
  320. TBonusListPtr bl = attackedStack->getBonuses(Selector::type(Bonus::STACK_HEALTH));
  321. const int fullHP = bl->totalValue();
  322. bl->remove_if(Selector::source(Bonus::SPELL_EFFECT, SpellID::AGE));
  323. text % (fullHP - bl->totalValue());
  324. logLines.push_back(text.str());
  325. }
  326. break;
  327. case SpellID::THUNDERBOLT:
  328. {
  329. logPlural(367);
  330. std::string text = VLC->generaltexth->allTexts[343].substr(1, VLC->generaltexth->allTexts[343].size() - 1); //Does %d points of damage.
  331. boost::algorithm::replace_first(text, "%d", boost::lexical_cast<std::string>(packet->dmgToDisplay)); //no more text afterwards
  332. logLines.push_back(text);
  333. }
  334. break;
  335. case SpellID::DISPEL_HELPFUL_SPELLS:
  336. logPlural(555);
  337. break;
  338. case SpellID::DEATH_STARE:
  339. if (packet->dmgToDisplay > 0)
  340. {
  341. std::string text;
  342. if (packet->dmgToDisplay > 1)
  343. {
  344. text = VLC->generaltexth->allTexts[119]; //%d %s die under the terrible gaze of the %s.
  345. boost::algorithm::replace_first(text, "%d", boost::lexical_cast<std::string>(packet->dmgToDisplay));
  346. boost::algorithm::replace_first(text, "%s", attackedNamePl);
  347. }
  348. else
  349. {
  350. text = VLC->generaltexth->allTexts[118]; //One %s dies under the terrible gaze of the %s.
  351. boost::algorithm::replace_first(text, "%s", attackedNameSing);
  352. }
  353. boost::algorithm::replace_first(text, "%s", casterName); //casting stack
  354. logLines.push_back(text);
  355. }
  356. break;
  357. default:
  358. {
  359. boost::format text(VLC->generaltexth->allTexts[565]); //The %s casts %s
  360. text % casterName % owner->name;
  361. displayDamage = true;
  362. logLines.push_back(text.str());
  363. }
  364. break;
  365. }
  366. }
  367. void DefaultSpellMechanics::applyBattleEffects(const SpellCastEnvironment * env, const BattleSpellCastParameters & parameters, SpellCastContext & ctx) const
  368. {
  369. //applying effects
  370. if(owner->isOffensiveSpell())
  371. {
  372. int spellDamage = parameters.effectValue;
  373. int chainLightningModifier = 0;
  374. for(auto & attackedCre : ctx.attackedCres)
  375. {
  376. BattleStackAttacked bsa;
  377. if(spellDamage != 0)
  378. bsa.damageAmount = owner->adjustRawDamage(parameters.caster, attackedCre, spellDamage) >> chainLightningModifier;
  379. else
  380. bsa.damageAmount = owner->calculateDamage(parameters.caster, attackedCre, parameters.effectLevel, parameters.effectPower) >> chainLightningModifier;
  381. ctx.addDamageToDisplay(bsa.damageAmount);
  382. bsa.stackAttacked = (attackedCre)->ID;
  383. if(parameters.mode == ECastingMode::ENCHANTER_CASTING) //multiple damage spells cast
  384. bsa.attackerID = parameters.casterStack->ID;
  385. else
  386. bsa.attackerID = -1;
  387. (attackedCre)->prepareAttacked(bsa, env->getRandomGenerator());
  388. ctx.si.stacks.push_back(bsa);
  389. if(owner->id == SpellID::CHAIN_LIGHTNING)
  390. ++chainLightningModifier;
  391. }
  392. }
  393. if(owner->hasEffects())
  394. {
  395. SetStackEffect sse;
  396. //get default spell duration (spell power with bonuses for heroes)
  397. int duration = parameters.enchantPower;
  398. //generate actual stack bonuses
  399. {
  400. int maxDuration = 0;
  401. std::vector<Bonus> tmp;
  402. owner->getEffects(tmp, parameters.effectLevel);
  403. for(Bonus& b : tmp)
  404. {
  405. //use configured duration if present
  406. if(b.turnsRemain == 0)
  407. b.turnsRemain = duration;
  408. vstd::amax(maxDuration, b.turnsRemain);
  409. sse.effect.push_back(b);
  410. }
  411. //if all spell effects have special duration, use it
  412. duration = maxDuration;
  413. }
  414. //fix to original config: shield should display damage reduction
  415. if(owner->id == SpellID::SHIELD || owner->id == SpellID::AIR_SHIELD)
  416. {
  417. sse.effect.back().val = (100 - sse.effect.back().val);
  418. }
  419. //we need to know who cast Bind
  420. if(owner->id == SpellID::BIND && parameters.casterStack)
  421. {
  422. sse.effect.back().additionalInfo = parameters.casterStack->ID;
  423. }
  424. const Bonus * bonus = nullptr;
  425. if(parameters.casterHero)
  426. bonus = parameters.casterHero->getBonusLocalFirst(Selector::typeSubtype(Bonus::SPECIAL_PECULIAR_ENCHANT, owner->id));
  427. //TODO does hero specialty should affects his stack casting spells?
  428. si32 power = 0;
  429. for(const CStack * affected : ctx.attackedCres)
  430. {
  431. sse.stacks.push_back(affected->ID);
  432. //Apply hero specials - peculiar enchants
  433. const ui8 tier = std::max((ui8)1, affected->getCreature()->level); //don't divide by 0 for certain creatures (commanders, war machines)
  434. if(bonus)
  435. {
  436. switch(bonus->additionalInfo)
  437. {
  438. case 0: //normal
  439. {
  440. switch(tier)
  441. {
  442. case 1: case 2:
  443. power = 3;
  444. break;
  445. case 3: case 4:
  446. power = 2;
  447. break;
  448. case 5: case 6:
  449. power = 1;
  450. break;
  451. }
  452. Bonus specialBonus(sse.effect.back());
  453. specialBonus.val = power; //it doesn't necessarily make sense for some spells, use it wisely
  454. sse.uniqueBonuses.push_back (std::pair<ui32,Bonus> (affected->ID, specialBonus)); //additional premy to given effect
  455. }
  456. break;
  457. case 1: //only Coronius as yet
  458. {
  459. power = std::max(5 - tier, 0);
  460. Bonus specialBonus = CStack::featureGenerator(Bonus::PRIMARY_SKILL, PrimarySkill::ATTACK, power, duration);
  461. specialBonus.sid = owner->id;
  462. sse.uniqueBonuses.push_back(std::pair<ui32,Bonus> (affected->ID, specialBonus)); //additional attack to Slayer effect
  463. }
  464. break;
  465. }
  466. }
  467. if (parameters.casterHero && parameters.casterHero->hasBonusOfType(Bonus::SPECIAL_BLESS_DAMAGE, owner->id)) //TODO: better handling of bonus percentages
  468. {
  469. int damagePercent = parameters.casterHero->level * parameters.casterHero->valOfBonuses(Bonus::SPECIAL_BLESS_DAMAGE, owner->id.toEnum()) / tier;
  470. Bonus specialBonus = CStack::featureGenerator(Bonus::CREATURE_DAMAGE, 0, damagePercent, duration);
  471. specialBonus.valType = Bonus::PERCENT_TO_ALL;
  472. specialBonus.sid = owner->id;
  473. sse.uniqueBonuses.push_back (std::pair<ui32,Bonus> (affected->ID, specialBonus));
  474. }
  475. }
  476. if(!sse.stacks.empty())
  477. env->sendAndApply(&sse);
  478. }
  479. }
  480. std::vector<BattleHex> DefaultSpellMechanics::rangeInHexes(BattleHex centralHex, ui8 schoolLvl, ui8 side, bool *outDroppedHexes) const
  481. {
  482. using namespace SRSLPraserHelpers;
  483. std::vector<BattleHex> ret;
  484. std::string rng = owner->getLevelInfo(schoolLvl).range + ','; //copy + artificial comma for easier handling
  485. if(rng.size() >= 2 && rng[0] != 'X') //there is at lest one hex in range (+artificial comma)
  486. {
  487. std::string number1, number2;
  488. int beg, end;
  489. bool readingFirst = true;
  490. for(auto & elem : rng)
  491. {
  492. if(std::isdigit(elem) ) //reading number
  493. {
  494. if(readingFirst)
  495. number1 += elem;
  496. else
  497. number2 += elem;
  498. }
  499. else if(elem == ',') //comma
  500. {
  501. //calculating variables
  502. if(readingFirst)
  503. {
  504. beg = atoi(number1.c_str());
  505. number1 = "";
  506. }
  507. else
  508. {
  509. end = atoi(number2.c_str());
  510. number2 = "";
  511. }
  512. //obtaining new hexes
  513. std::set<ui16> curLayer;
  514. if(readingFirst)
  515. {
  516. curLayer = getInRange(centralHex, beg, beg);
  517. }
  518. else
  519. {
  520. curLayer = getInRange(centralHex, beg, end);
  521. readingFirst = true;
  522. }
  523. //adding abtained hexes
  524. for(auto & curLayer_it : curLayer)
  525. {
  526. ret.push_back(curLayer_it);
  527. }
  528. }
  529. else if(elem == '-') //dash
  530. {
  531. beg = atoi(number1.c_str());
  532. number1 = "";
  533. readingFirst = false;
  534. }
  535. }
  536. }
  537. //remove duplicates (TODO check if actually needed)
  538. range::unique(ret);
  539. return ret;
  540. }
  541. std::set<const CStack *> DefaultSpellMechanics::getAffectedStacks(const CBattleInfoCallback * cb, SpellTargetingContext & ctx) const
  542. {
  543. std::set<const CStack* > attackedCres;//std::set to exclude multiple occurrences of two hex creatures
  544. const ui8 attackerSide = cb->playerToSide(ctx.caster->getOwner()) == 1;
  545. const auto attackedHexes = rangeInHexes(ctx.destination, ctx.schoolLvl, attackerSide);
  546. //TODO: more generic solution for mass spells
  547. if(owner->getLevelInfo(ctx.schoolLvl).range.size() > 1) //custom many-hex range
  548. {
  549. for(BattleHex hex : attackedHexes)
  550. {
  551. if(const CStack * st = cb->battleGetStackByPos(hex, ctx.ti.onlyAlive))
  552. {
  553. attackedCres.insert(st);
  554. }
  555. }
  556. }
  557. else if(ctx.ti.type == CSpell::CREATURE)
  558. {
  559. auto predicate = [=](const CStack * s){
  560. const bool positiveToAlly = owner->isPositive() && s->owner == ctx.caster->getOwner();
  561. const bool negativeToEnemy = owner->isNegative() && s->owner != ctx.caster->getOwner();
  562. const bool validTarget = s->isValidTarget(!ctx.ti.onlyAlive); //todo: this should be handled by spell class
  563. //for single target spells select stacks covering destination tile
  564. const bool rangeCovers = ctx.ti.massive || s->coversPos(ctx.destination);
  565. //handle smart targeting
  566. const bool positivenessFlag = !ctx.ti.smart || owner->isNeutral() || positiveToAlly || negativeToEnemy;
  567. return rangeCovers && positivenessFlag && validTarget;
  568. };
  569. TStacks stacks = cb->battleGetStacksIf(predicate);
  570. if(ctx.ti.massive)
  571. {
  572. //for massive spells add all targets
  573. for (auto stack : stacks)
  574. attackedCres.insert(stack);
  575. }
  576. else
  577. {
  578. //for single target spells we must select one target. Alive stack is preferred (issue #1763)
  579. for(auto stack : stacks)
  580. {
  581. if(stack->alive())
  582. {
  583. attackedCres.insert(stack);
  584. break;
  585. }
  586. }
  587. if(attackedCres.empty() && !stacks.empty())
  588. {
  589. attackedCres.insert(stacks.front());
  590. }
  591. }
  592. }
  593. else //custom range from attackedHexes
  594. {
  595. for(BattleHex hex : attackedHexes)
  596. {
  597. if(const CStack * st = cb->battleGetStackByPos(hex, ctx.ti.onlyAlive))
  598. attackedCres.insert(st);
  599. }
  600. }
  601. return attackedCres;
  602. }
  603. ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const ISpellCaster * caster) const
  604. {
  605. //no problems by default, this method is for spell-specific problems
  606. return ESpellCastProblem::OK;
  607. }
  608. ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::canBeCast(const CBattleInfoCallback * cb, const SpellTargetingContext & ctx) const
  609. {
  610. //no problems by default, this method is for spell-specific problems
  611. //common problems handled by CSpell
  612. return ESpellCastProblem::OK;
  613. }
  614. ESpellCastProblem::ESpellCastProblem DefaultSpellMechanics::isImmuneByStack(const ISpellCaster * caster, const CStack * obj) const
  615. {
  616. //by default use general algorithm
  617. return owner->internalIsImmune(caster, obj);
  618. }
  619. void DefaultSpellMechanics::doDispell(BattleInfo * battle, const BattleSpellCast * packet, const CSelector & selector) const
  620. {
  621. auto localSelector = [](const Bonus * bonus)
  622. {
  623. const CSpell * sourceSpell = bonus->sourceSpell();
  624. if(sourceSpell != nullptr)
  625. {
  626. //Special case: DISRUPTING_RAY is "immune" to dispell
  627. //Other even PERMANENT effects can be removed (f.e. BIND)
  628. if(sourceSpell->id == SpellID::DISRUPTING_RAY)
  629. return false;
  630. }
  631. return true;
  632. };
  633. for(auto stackID : packet->affectedCres)
  634. {
  635. CStack *s = battle->getStack(stackID);
  636. s->popBonuses(CSelector(localSelector).And(selector));
  637. }
  638. }
  639. void DefaultSpellMechanics::castMagicMirror(const SpellCastEnvironment* env, BattleSpellCastParameters& parameters) const
  640. {
  641. logGlobal->debugStream() << "Started spell cast. Spell: "<<owner->name<<"; mode: MAGIC_MIRROR";
  642. BattleHex destination = parameters.getFirstDestinationHex();
  643. SpellCastContext ctx(this, parameters);
  644. //calculating affected creatures for all spells
  645. auto creatures = owner->getAffectedStacks(parameters.cb, parameters.mode, parameters.caster, parameters.spellLvl, destination);
  646. std::copy(creatures.begin(), creatures.end(), std::back_inserter(ctx.attackedCres));
  647. logGlobal->debugStream() << "will affect: " << ctx.attackedCres.size() << " stacks";
  648. handleResistance(env, ctx.attackedCres, ctx.sc);
  649. applyBattleEffects(env, parameters, ctx);
  650. ctx.sendCastPacket(env);
  651. if(!ctx.si.stacks.empty()) //after spellcast info shows
  652. env->sendAndApply(&ctx.si);
  653. logGlobal->debugStream() << "Finished spell cast. Spell: "<<owner->name<<"; mode: MAGIC_MIRROR";
  654. }
  655. void DefaultSpellMechanics::handleResistance(const SpellCastEnvironment * env, std::vector<const CStack* >& attackedCres, BattleSpellCast& sc) const
  656. {
  657. //checking if creatures resist
  658. //resistance/reflection is applied only to negative spells
  659. if(owner->isNegative())
  660. {
  661. std::vector <const CStack*> resisted;
  662. for(auto s : attackedCres)
  663. {
  664. //magic resistance
  665. const int prob = std::min((s)->magicResistance(), 100); //probability of resistance in %
  666. if(env->getRandomGenerator().nextInt(99) < prob)
  667. {
  668. resisted.push_back(s);
  669. }
  670. }
  671. vstd::erase_if(attackedCres, [&resisted](const CStack * s)
  672. {
  673. return vstd::contains(resisted, s);
  674. });
  675. for(auto s : resisted)
  676. {
  677. BattleSpellCast::CustomEffect effect;
  678. effect.effect = 78;
  679. effect.stack = s->ID;
  680. sc.customEffects.push_back(effect);
  681. }
  682. }
  683. }
  684. bool DefaultSpellMechanics::requiresCreatureTarget() const
  685. {
  686. //most spells affects creatures somehow regardless of Target Type
  687. //for few exceptions see overrides
  688. return true;
  689. }