CDefaultSpellMechanics.cpp 26 KB

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