CDefaultSpellMechanics.cpp 24 KB

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