CDefaultSpellMechanics.cpp 25 KB

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