CDefaultSpellMechanics.cpp 25 KB

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