CSpellWindow.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788
  1. /*
  2. * CSpellWindow.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 "CSpellWindow.h"
  12. #include "../../lib/ScopeGuard.h"
  13. #include "GUIClasses.h"
  14. #include "InfoWindows.h"
  15. #include "CCastleInterface.h"
  16. #include "../CGameInfo.h"
  17. #include "../CPlayerInterface.h"
  18. #include "../PlayerLocalState.h"
  19. #include "../battle/BattleInterface.h"
  20. #include "../gui/CGuiHandler.h"
  21. #include "../gui/Shortcut.h"
  22. #include "../gui/WindowHandler.h"
  23. #include "../media/IVideoPlayer.h"
  24. #include "../widgets/GraphicalPrimitiveCanvas.h"
  25. #include "../widgets/CComponent.h"
  26. #include "../widgets/CTextInput.h"
  27. #include "../widgets/TextControls.h"
  28. #include "../adventureMap/AdventureMapInterface.h"
  29. #include "../render/CAnimation.h"
  30. #include "../render/IRenderHandler.h"
  31. #include "../render/IImage.h"
  32. #include "../render/IImageLoader.h"
  33. #include "../render/Canvas.h"
  34. #include "../../CCallback.h"
  35. #include "../../lib/CConfigHandler.h"
  36. #include "../../lib/CGeneralTextHandler.h"
  37. #include "../../lib/spells/CSpellHandler.h"
  38. #include "../../lib/spells/ISpellMechanics.h"
  39. #include "../../lib/spells/Problem.h"
  40. #include "../../lib/GameConstants.h"
  41. #include "../../lib/mapObjects/CGHeroInstance.h"
  42. CSpellWindow::InteractiveArea::InteractiveArea(const Rect & myRect, std::function<void()> funcL, int helpTextId, CSpellWindow * _owner)
  43. {
  44. addUsedEvents(LCLICK | SHOW_POPUP | HOVER);
  45. pos = myRect;
  46. onLeft = funcL;
  47. hoverText = CGI->generaltexth->zelp[helpTextId].first;
  48. helpText = CGI->generaltexth->zelp[helpTextId].second;
  49. owner = _owner;
  50. }
  51. void CSpellWindow::InteractiveArea::clickPressed(const Point & cursorPosition)
  52. {
  53. onLeft();
  54. }
  55. void CSpellWindow::InteractiveArea::showPopupWindow(const Point & cursorPosition)
  56. {
  57. CRClickPopup::createAndPush(helpText);
  58. }
  59. void CSpellWindow::InteractiveArea::hover(bool on)
  60. {
  61. if(on)
  62. owner->statusBar->write(hoverText);
  63. else
  64. owner->statusBar->clear();
  65. }
  66. class SpellbookSpellSorter
  67. {
  68. public:
  69. bool operator()(const CSpell * A, const CSpell * B)
  70. {
  71. if(A->getLevel() < B->getLevel())
  72. return true;
  73. if(A->getLevel() > B->getLevel())
  74. return false;
  75. for(auto schoolId = 0; schoolId < GameConstants::DEFAULT_SCHOOLS; schoolId++)
  76. {
  77. if(A->school.at(SpellSchool(schoolId)) && !B->school.at(SpellSchool(schoolId)))
  78. return true;
  79. if(!A->school.at(SpellSchool(schoolId)) && B->school.at(SpellSchool(schoolId)))
  80. return false;
  81. }
  82. return A->getNameTranslated() < B->getNameTranslated();
  83. }
  84. };
  85. CSpellWindow::CSpellWindow(const CGHeroInstance * _myHero, CPlayerInterface * _myInt, bool openOnBattleSpells, std::function<void(SpellID)> onSpellSelect):
  86. CWindowObject(PLAYER_COLORED | (settings["gameTweaks"]["enableLargeSpellbook"].Bool() ? BORDERED : 0)),
  87. battleSpellsOnly(openOnBattleSpells),
  88. selectedTab(4),
  89. currentPage(0),
  90. myHero(_myHero),
  91. myInt(_myInt),
  92. openOnBattleSpells(openOnBattleSpells),
  93. onSpellSelect(onSpellSelect),
  94. isBigSpellbook(settings["gameTweaks"]["enableLargeSpellbook"].Bool()),
  95. spellsPerPage(24),
  96. offL(-11),
  97. offR(195),
  98. offRM(110),
  99. offT(-37),
  100. offB(56)
  101. {
  102. OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE);
  103. if(isBigSpellbook)
  104. {
  105. background = std::make_shared<CPicture>(createBigSpellBook(), Point(0, 0));
  106. updateShadow();
  107. }
  108. else
  109. {
  110. background = std::make_shared<CPicture>(ImagePath::builtin("SpelBack"), 0, 0);
  111. offL = offR = offT = offB = offRM = 0;
  112. spellsPerPage = 12;
  113. }
  114. pos = background->center(Point(pos.w/2 + pos.x, pos.h/2 + pos.y));
  115. if(settings["general"]["enableUiEnhancements"].Bool())
  116. {
  117. Rect r(90, isBigSpellbook ? 480 : 420, isBigSpellbook ? 160 : 110, 16);
  118. const ColorRGBA rectangleColor = ColorRGBA(0, 0, 0, 75);
  119. const ColorRGBA borderColor = ColorRGBA(128, 100, 75);
  120. const ColorRGBA grayedColor = ColorRGBA(158, 130, 105);
  121. searchBoxRectangle = std::make_shared<TransparentFilledRectangle>(r.resize(1), rectangleColor, borderColor);
  122. searchBoxDescription = std::make_shared<CLabel>(r.center().x, r.center().y, FONT_SMALL, ETextAlignment::CENTER, grayedColor, CGI->generaltexth->translate("vcmi.spellBook.search"));
  123. searchBox = std::make_shared<CTextInput>(r, FONT_SMALL, ETextAlignment::CENTER, true);
  124. searchBox->setCallback(std::bind(&CSpellWindow::searchInput, this));
  125. }
  126. processSpells();
  127. //numbers of spell pages computed
  128. leftCorner = std::make_shared<CPicture>(ImagePath::builtin("SpelTrnL.bmp"), 97 + offL, 77 + offT);
  129. rightCorner = std::make_shared<CPicture>(ImagePath::builtin("SpelTrnR.bmp"), 487 + offR, 72 + offT);
  130. spellIcons = GH.renderHandler().loadAnimation(AnimationPath::builtin("Spells"));
  131. schoolTab = std::make_shared<CAnimImage>(AnimationPath::builtin("SpelTab"), selectedTab, 0, 524 + offR, 88);
  132. schoolPicture = std::make_shared<CAnimImage>(AnimationPath::builtin("Schools"), 0, 0, 117 + offL, 74 + offT);
  133. schoolBorders[0] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevA.def"));
  134. schoolBorders[1] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevF.def"));
  135. schoolBorders[2] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevW.def"));
  136. schoolBorders[3] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevE.def"));
  137. for(auto item : schoolBorders)
  138. item->preload();
  139. mana = std::make_shared<CLabel>(435 + (isBigSpellbook ? 159 : 0), 426 + offB, FONT_SMALL, ETextAlignment::CENTER, Colors::YELLOW, std::to_string(myHero->mana));
  140. if(isBigSpellbook)
  141. statusBar = CGStatusBar::create(400, 587);
  142. else
  143. statusBar = CGStatusBar::create(7, 569, ImagePath::builtin("Spelroll.bmp"));
  144. Rect schoolRect( 549 + pos.x + offR, 94 + pos.y, 45, 35);
  145. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 479 + pos.x + (isBigSpellbook ? 175 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fexitb, this), 460, this));
  146. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 221 + pos.x + (isBigSpellbook ? 43 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fbattleSpellsb, this), 453, this));
  147. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 355 + pos.x + (isBigSpellbook ? 110 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fadvSpellsb, this), 452, this));
  148. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 418 + pos.x + (isBigSpellbook ? 142 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fmanaPtsb, this), 459, this));
  149. interactiveAreas.push_back(std::make_shared<InteractiveArea>( schoolRect + Point(0, 0), std::bind(&CSpellWindow::selectSchool, this, 0), 454, this));
  150. interactiveAreas.push_back(std::make_shared<InteractiveArea>( schoolRect + Point(0, 57), std::bind(&CSpellWindow::selectSchool, this, 3), 457, this));
  151. interactiveAreas.push_back(std::make_shared<InteractiveArea>( schoolRect + Point(0, 116), std::bind(&CSpellWindow::selectSchool, this, 1), 455, this));
  152. interactiveAreas.push_back(std::make_shared<InteractiveArea>( schoolRect + Point(0, 176), std::bind(&CSpellWindow::selectSchool, this, 2), 456, this));
  153. interactiveAreas.push_back(std::make_shared<InteractiveArea>( schoolRect + Point(0, 236), std::bind(&CSpellWindow::selectSchool, this, 4), 458, this));
  154. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 97 + offL + pos.x, 77 + offT + pos.y, leftCorner->pos.h, leftCorner->pos.w ), std::bind(&CSpellWindow::fLcornerb, this), 450, this));
  155. interactiveAreas.push_back(std::make_shared<InteractiveArea>( Rect( 487 + offR + pos.x, 72 + offT + pos.y, rightCorner->pos.h, rightCorner->pos.w ), std::bind(&CSpellWindow::fRcornerb, this), 451, this));
  156. //areas for spells
  157. int xpos = 117 + offL + pos.x;
  158. int ypos = 90 + offT + pos.y;
  159. for(int v=0; v<spellsPerPage; ++v)
  160. {
  161. spellAreas[v] = std::make_shared<SpellArea>( Rect(xpos, ypos, 65, 78), this);
  162. if(v == (spellsPerPage / 2) - 1) //to right page
  163. {
  164. xpos = offRM + 336 + pos.x; ypos = 90 + offT + pos.y;
  165. }
  166. else
  167. {
  168. if(v%(isBigSpellbook ? 3 : 2) == 0 || (v%3 == 1 && isBigSpellbook))
  169. {
  170. xpos+=85;
  171. }
  172. else
  173. {
  174. xpos -= (isBigSpellbook ? 2 : 1)*85; ypos+=97;
  175. }
  176. }
  177. }
  178. selectedTab = battleSpellsOnly ? myInt->localState->spellbookSettings.spellbookLastTabBattle : myInt->localState->spellbookSettings.spellbookLastTabAdvmap;
  179. schoolTab->setFrame(selectedTab, 0);
  180. int cp = battleSpellsOnly ? myInt->localState->spellbookSettings.spellbookLastPageBattle : myInt->localState->spellbookSettings.spellbookLastPageAdvmap;
  181. // spellbook last page battle index is not reset after battle, so this needs to stay here
  182. vstd::abetween(cp, 0, std::max(0, pagesWithinCurrentTab() - 1));
  183. setCurrentPage(cp);
  184. computeSpellsPerArea();
  185. addUsedEvents(KEYBOARD);
  186. }
  187. CSpellWindow::~CSpellWindow()
  188. {
  189. }
  190. std::shared_ptr<IImage> CSpellWindow::createBigSpellBook()
  191. {
  192. std::shared_ptr<IImage> img = GH.renderHandler().loadImage(ImagePath::builtin("SpelBack"));
  193. Canvas canvas = Canvas(Point(800, 600));
  194. // edges
  195. canvas.draw(img, Point(0, 0), Rect(15, 38, 90, 45));
  196. canvas.draw(img, Point(0, 460), Rect(15, 400, 90, 141));
  197. canvas.draw(img, Point(705, 0), Rect(509, 38, 95, 45));
  198. canvas.draw(img, Point(705, 460), Rect(509, 400, 95, 141));
  199. // left / right
  200. Canvas tmp1 = Canvas(Point(90, 355 - 45));
  201. tmp1.draw(img, Point(0, 0), Rect(15, 38 + 45, 90, 355 - 45));
  202. canvas.drawScaled(tmp1, Point(0, 45), Point(90, 415));
  203. Canvas tmp2 = Canvas(Point(95, 355 - 45));
  204. tmp2.draw(img, Point(0, 0), Rect(509, 38 + 45, 95, 355 - 45));
  205. canvas.drawScaled(tmp2, Point(705, 45), Point(95, 415));
  206. // top / bottom
  207. Canvas tmp3 = Canvas(Point(409, 45));
  208. tmp3.draw(img, Point(0, 0), Rect(100, 38, 409, 45));
  209. canvas.drawScaled(tmp3, Point(90, 0), Point(615, 45));
  210. Canvas tmp4 = Canvas(Point(409, 141));
  211. tmp4.draw(img, Point(0, 0), Rect(100, 400, 409, 141));
  212. canvas.drawScaled(tmp4, Point(90, 460), Point(615, 141));
  213. // middle
  214. Canvas tmp5 = Canvas(Point(409, 141));
  215. tmp5.draw(img, Point(0, 0), Rect(100, 38 + 45, 509 - 15, 400 - 38));
  216. canvas.drawScaled(tmp5, Point(90, 45), Point(615, 415));
  217. // carpet
  218. Canvas tmp6 = Canvas(Point(590, 59));
  219. tmp6.draw(img, Point(0, 0), Rect(15, 484, 590, 59));
  220. canvas.drawScaled(tmp6, Point(0, 545), Point(800, 59));
  221. // remove bookmarks
  222. for (int i = 0; i < 56; i++)
  223. canvas.draw(Canvas(canvas, Rect(i < 30 ? 268 : 327, 464, 1, 46)), Point(269 + i, 464));
  224. for (int i = 0; i < 56; i++)
  225. canvas.draw(Canvas(canvas, Rect(469, 464, 1, 42)), Point(470 + i, 464));
  226. for (int i = 0; i < 57; i++)
  227. canvas.draw(Canvas(canvas, Rect(i < 30 ? 564 : 630, 464, 1, 44)), Point(565 + i, 464));
  228. for (int i = 0; i < 56; i++)
  229. canvas.draw(Canvas(canvas, Rect(656, 464, 1, 47)), Point(657 + i, 464));
  230. // draw bookmarks
  231. canvas.draw(img, Point(278, 464), Rect(220, 405, 37, 47));
  232. canvas.draw(img, Point(481, 465), Rect(354, 406, 37, 41));
  233. canvas.draw(img, Point(575, 465), Rect(417, 406, 37, 45));
  234. canvas.draw(img, Point(667, 465), Rect(478, 406, 37, 47));
  235. return GH.renderHandler().createImage(canvas.getInternalSurface());
  236. }
  237. void CSpellWindow::searchInput()
  238. {
  239. if(searchBox)
  240. searchBoxDescription->setEnabled(searchBox->getText().empty());
  241. processSpells();
  242. int cp = 0;
  243. // spellbook last page battle index is not reset after battle, so this needs to stay here
  244. vstd::abetween(cp, 0, std::max(0, pagesWithinCurrentTab() - 1));
  245. setCurrentPage(cp);
  246. computeSpellsPerArea();
  247. }
  248. void CSpellWindow::processSpells()
  249. {
  250. mySpells.clear();
  251. //initializing castable spells
  252. mySpells.reserve(CGI->spellh->objects.size());
  253. for(auto const & spell : CGI->spellh->objects)
  254. {
  255. bool searchTextFound = !searchBox || boost::algorithm::contains(boost::algorithm::to_lower_copy(spell->getNameTranslated()), boost::algorithm::to_lower_copy(searchBox->getText()));
  256. if(onSpellSelect)
  257. {
  258. if(spell->isCombat() == openOnBattleSpells && !spell->isSpecial() && !spell->isCreatureAbility() && searchTextFound)
  259. mySpells.push_back(spell.get());
  260. continue;
  261. }
  262. if(!spell->isCreatureAbility() && myHero->canCastThisSpell(spell.get()) && searchTextFound)
  263. mySpells.push_back(spell.get());
  264. }
  265. SpellbookSpellSorter spellsorter;
  266. std::sort(mySpells.begin(), mySpells.end(), spellsorter);
  267. //initializing sizes of spellbook's parts
  268. for(auto & elem : sitesPerTabAdv)
  269. elem = 0;
  270. for(auto & elem : sitesPerTabBattle)
  271. elem = 0;
  272. for(const auto spell : mySpells)
  273. {
  274. int * sitesPerOurTab = spell->isCombat() ? sitesPerTabBattle : sitesPerTabAdv;
  275. ++sitesPerOurTab[4];
  276. spell->forEachSchool([&sitesPerOurTab](const SpellSchool & school, bool & stop)
  277. {
  278. ++sitesPerOurTab[school];
  279. });
  280. }
  281. if(sitesPerTabAdv[4] % spellsPerPage == 0)
  282. sitesPerTabAdv[4]/=spellsPerPage;
  283. else
  284. sitesPerTabAdv[4] = sitesPerTabAdv[4]/spellsPerPage + 1;
  285. for(int v=0; v<4; ++v)
  286. {
  287. if(sitesPerTabAdv[v] <= spellsPerPage - 2)
  288. sitesPerTabAdv[v] = 1;
  289. else
  290. {
  291. if((sitesPerTabAdv[v] - (spellsPerPage - 2)) % spellsPerPage == 0)
  292. sitesPerTabAdv[v] = (sitesPerTabAdv[v] - (spellsPerPage - 2)) / spellsPerPage + 1;
  293. else
  294. sitesPerTabAdv[v] = (sitesPerTabAdv[v] - (spellsPerPage - 2)) / spellsPerPage + 2;
  295. }
  296. }
  297. if(sitesPerTabBattle[4] % spellsPerPage == 0)
  298. sitesPerTabBattle[4]/=spellsPerPage;
  299. else
  300. sitesPerTabBattle[4] = sitesPerTabBattle[4]/spellsPerPage + 1;
  301. for(int v=0; v<4; ++v)
  302. {
  303. if(sitesPerTabBattle[v] <= spellsPerPage - 2)
  304. sitesPerTabBattle[v] = 1;
  305. else
  306. {
  307. if((sitesPerTabBattle[v] - (spellsPerPage - 2)) % spellsPerPage == 0)
  308. sitesPerTabBattle[v] = (sitesPerTabBattle[v] - (spellsPerPage - 2)) / spellsPerPage + 1;
  309. else
  310. sitesPerTabBattle[v] = (sitesPerTabBattle[v] - (spellsPerPage - 2)) / spellsPerPage + 2;
  311. }
  312. }
  313. }
  314. void CSpellWindow::fexitb()
  315. {
  316. (myInt->battleInt ? myInt->localState->spellbookSettings.spellbookLastTabBattle : myInt->localState->spellbookSettings.spellbookLastTabAdvmap) = selectedTab;
  317. (myInt->battleInt ? myInt->localState->spellbookSettings.spellbookLastPageBattle : myInt->localState->spellbookSettings.spellbookLastPageAdvmap) = currentPage;
  318. close();
  319. }
  320. void CSpellWindow::fadvSpellsb()
  321. {
  322. if(battleSpellsOnly == true)
  323. {
  324. turnPageRight();
  325. battleSpellsOnly = false;
  326. setCurrentPage(0);
  327. }
  328. computeSpellsPerArea();
  329. }
  330. void CSpellWindow::fbattleSpellsb()
  331. {
  332. if(battleSpellsOnly == false)
  333. {
  334. turnPageLeft();
  335. battleSpellsOnly = true;
  336. setCurrentPage(0);
  337. }
  338. computeSpellsPerArea();
  339. }
  340. void CSpellWindow::fmanaPtsb()
  341. {
  342. }
  343. void CSpellWindow::selectSchool(int school)
  344. {
  345. if(selectedTab != school)
  346. {
  347. if(selectedTab < school)
  348. turnPageLeft();
  349. else
  350. turnPageRight();
  351. selectedTab = school;
  352. schoolTab->setFrame(selectedTab, 0);
  353. setCurrentPage(0);
  354. }
  355. computeSpellsPerArea();
  356. }
  357. void CSpellWindow::fLcornerb()
  358. {
  359. if(currentPage>0)
  360. {
  361. turnPageLeft();
  362. setCurrentPage(currentPage - 1);
  363. }
  364. computeSpellsPerArea();
  365. }
  366. void CSpellWindow::fRcornerb()
  367. {
  368. if((currentPage + 1) < (pagesWithinCurrentTab()))
  369. {
  370. turnPageRight();
  371. setCurrentPage(currentPage + 1);
  372. }
  373. computeSpellsPerArea();
  374. }
  375. void CSpellWindow::show(Canvas & to)
  376. {
  377. statusBar->show(to);
  378. }
  379. void CSpellWindow::computeSpellsPerArea()
  380. {
  381. std::vector<const CSpell *> spellsCurSite;
  382. spellsCurSite.reserve(mySpells.size());
  383. for(const CSpell * spell : mySpells)
  384. {
  385. if(spell->isCombat() ^ !battleSpellsOnly
  386. && ((selectedTab == 4) || spell->school.at(SpellSchool(selectedTab)))
  387. )
  388. {
  389. spellsCurSite.push_back(spell);
  390. }
  391. }
  392. if(selectedTab == 4)
  393. {
  394. if(spellsCurSite.size() > spellsPerPage)
  395. {
  396. spellsCurSite = std::vector<const CSpell *>(spellsCurSite.begin() + currentPage*spellsPerPage, spellsCurSite.end());
  397. if(spellsCurSite.size() > spellsPerPage)
  398. {
  399. spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage, spellsCurSite.end());
  400. }
  401. }
  402. }
  403. else //selectedTab == 0, 1, 2 or 3
  404. {
  405. if(spellsCurSite.size() > spellsPerPage - 2)
  406. {
  407. if(currentPage == 0)
  408. {
  409. spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage-2, spellsCurSite.end());
  410. }
  411. else
  412. {
  413. spellsCurSite = std::vector<const CSpell *>(spellsCurSite.begin() + (currentPage-1)*spellsPerPage + spellsPerPage-2, spellsCurSite.end());
  414. if(spellsCurSite.size() > spellsPerPage)
  415. {
  416. spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage, spellsCurSite.end());
  417. }
  418. }
  419. }
  420. }
  421. //applying
  422. if(selectedTab == 4 || currentPage != 0)
  423. {
  424. for(size_t c=0; c<spellsPerPage; ++c)
  425. {
  426. if(c < spellsCurSite.size())
  427. {
  428. spellAreas[c]->setSpell(spellsCurSite[c]);
  429. }
  430. else
  431. {
  432. spellAreas[c]->setSpell(nullptr);
  433. }
  434. }
  435. }
  436. else
  437. {
  438. spellAreas[0]->setSpell(nullptr);
  439. spellAreas[1]->setSpell(nullptr);
  440. for(size_t c=0; c<spellsPerPage-2; ++c)
  441. {
  442. if(c < spellsCurSite.size())
  443. spellAreas[c+2]->setSpell(spellsCurSite[c]);
  444. else
  445. spellAreas[c+2]->setSpell(nullptr);
  446. }
  447. }
  448. redraw();
  449. }
  450. void CSpellWindow::setCurrentPage(int value)
  451. {
  452. currentPage = value;
  453. schoolPicture->visible = selectedTab!=4 && currentPage == 0;
  454. if(selectedTab != 4)
  455. schoolPicture->setFrame(selectedTab, 0);
  456. if (currentPage != 0)
  457. leftCorner->enable();
  458. else
  459. leftCorner->disable();
  460. if (currentPage + 1 < pagesWithinCurrentTab())
  461. rightCorner->enable();
  462. else
  463. rightCorner->disable();
  464. mana->setText(std::to_string(myHero->mana));//just in case, it will be possible to cast spell without closing book
  465. }
  466. void CSpellWindow::turnPageLeft()
  467. {
  468. if(settings["video"]["spellbookAnimation"].Bool() && !isBigSpellbook)
  469. CCS->videoh->playSpellbookAnimation(VideoPath::builtin("PGTRNLFT.SMK"), pos.topLeft() + Point(13, 15));
  470. }
  471. void CSpellWindow::turnPageRight()
  472. {
  473. if(settings["video"]["spellbookAnimation"].Bool() && !isBigSpellbook)
  474. CCS->videoh->playSpellbookAnimation(VideoPath::builtin("PGTRNRGH.SMK"), pos.topLeft() + Point(13, 15));
  475. }
  476. void CSpellWindow::keyPressed(EShortcut key)
  477. {
  478. switch(key)
  479. {
  480. case EShortcut::GLOBAL_RETURN:
  481. fexitb();
  482. break;
  483. case EShortcut::MOVE_LEFT:
  484. fLcornerb();
  485. break;
  486. case EShortcut::MOVE_RIGHT:
  487. fRcornerb();
  488. break;
  489. case EShortcut::MOVE_UP:
  490. case EShortcut::MOVE_DOWN:
  491. {
  492. bool down = key == EShortcut::MOVE_DOWN;
  493. static const int schoolsOrder[] = { 0, 3, 1, 2, 4 };
  494. int index = -1;
  495. while(schoolsOrder[++index] != selectedTab);
  496. index += (down ? 1 : -1);
  497. vstd::abetween<int>(index, 0, std::size(schoolsOrder) - 1);
  498. if(selectedTab != schoolsOrder[index])
  499. selectSchool(schoolsOrder[index]);
  500. break;
  501. }
  502. case EShortcut::SPELLBOOK_TAB_COMBAT:
  503. fbattleSpellsb();
  504. break;
  505. case EShortcut::SPELLBOOK_TAB_ADVENTURE:
  506. fadvSpellsb();
  507. break;
  508. }
  509. }
  510. int CSpellWindow::pagesWithinCurrentTab()
  511. {
  512. return battleSpellsOnly ? sitesPerTabBattle[selectedTab] : sitesPerTabAdv[selectedTab];
  513. }
  514. CSpellWindow::SpellArea::SpellArea(Rect pos, CSpellWindow * owner)
  515. {
  516. this->pos = pos;
  517. this->owner = owner;
  518. addUsedEvents(LCLICK | SHOW_POPUP | HOVER);
  519. schoolLevel = -1;
  520. mySpell = nullptr;
  521. OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE);
  522. image = std::make_shared<CAnimImage>(owner->spellIcons, 0, 0);
  523. image->visible = false;
  524. name = std::make_shared<CLabel>(39, 70, FONT_TINY, ETextAlignment::CENTER);
  525. level = std::make_shared<CLabel>(39, 82, FONT_TINY, ETextAlignment::CENTER);
  526. cost = std::make_shared<CLabel>(39, 94, FONT_TINY, ETextAlignment::CENTER);
  527. for(auto l : {name, level, cost})
  528. l->setAutoRedraw(false);
  529. }
  530. CSpellWindow::SpellArea::~SpellArea() = default;
  531. void CSpellWindow::SpellArea::clickPressed(const Point & cursorPosition)
  532. {
  533. if(mySpell)
  534. {
  535. if(owner->onSpellSelect)
  536. {
  537. owner->onSpellSelect(mySpell->id);
  538. owner->fexitb();
  539. return;
  540. }
  541. auto spellCost = owner->myInt->cb->getSpellCost(mySpell, owner->myHero);
  542. if(spellCost > owner->myHero->mana) //insufficient mana
  543. {
  544. LOCPLINT->showInfoDialog(boost::str(boost::format(CGI->generaltexth->allTexts[206]) % spellCost % owner->myHero->mana));
  545. return;
  546. }
  547. //anything that is not combat spell is adventure spell
  548. //this not an error in general to cast even creature ability with hero
  549. const bool combatSpell = mySpell->isCombat();
  550. if(combatSpell == mySpell->isAdventure())
  551. {
  552. logGlobal->error("Spell have invalid flags");
  553. return;
  554. }
  555. const bool inCombat = owner->myInt->battleInt != nullptr;
  556. const bool inCastle = owner->myInt->castleInt != nullptr;
  557. //battle spell on adv map or adventure map spell during combat => display infowindow, not cast
  558. if((combatSpell != inCombat) || inCastle || (!combatSpell && !LOCPLINT->makingTurn))
  559. {
  560. std::vector<std::shared_ptr<CComponent>> hlp(1, std::make_shared<CComponent>(ComponentType::SPELL, mySpell->id));
  561. LOCPLINT->showInfoDialog(mySpell->getDescriptionTranslated(schoolLevel), hlp);
  562. }
  563. else if(combatSpell)
  564. {
  565. spells::detail::ProblemImpl problem;
  566. if(mySpell->canBeCast(problem, owner->myInt->battleInt->getBattle().get(), spells::Mode::HERO, owner->myHero))
  567. {
  568. owner->myInt->battleInt->castThisSpell(mySpell->id);
  569. owner->fexitb();
  570. }
  571. else
  572. {
  573. std::vector<std::string> texts;
  574. problem.getAll(texts);
  575. if(!texts.empty())
  576. LOCPLINT->showInfoDialog(texts.front());
  577. else
  578. LOCPLINT->showInfoDialog(CGI->generaltexth->translate("vcmi.adventureMap.spellUnknownProblem"));
  579. }
  580. }
  581. else //adventure spell
  582. {
  583. const CGHeroInstance * h = owner->myHero;
  584. GH.windows().popWindows(1);
  585. auto guard = vstd::makeScopeGuard([this]()
  586. {
  587. owner->myInt->localState->spellbookSettings.spellbookLastTabAdvmap = owner->selectedTab;
  588. owner->myInt->localState->spellbookSettings.spellbookLastPageAdvmap = owner->currentPage;
  589. });
  590. spells::detail::ProblemImpl problem;
  591. if (mySpell->getAdventureMechanics().canBeCast(problem, LOCPLINT->cb.get(), owner->myHero))
  592. {
  593. if(mySpell->getTargetType() == spells::AimType::LOCATION)
  594. adventureInt->enterCastingMode(mySpell);
  595. else if(mySpell->getTargetType() == spells::AimType::NO_TARGET)
  596. owner->myInt->cb->castSpell(h, mySpell->id);
  597. else
  598. logGlobal->error("Invalid spell target type");
  599. }
  600. else
  601. {
  602. std::vector<std::string> texts;
  603. problem.getAll(texts);
  604. if(!texts.empty())
  605. LOCPLINT->showInfoDialog(texts.front());
  606. else
  607. LOCPLINT->showInfoDialog(CGI->generaltexth->translate("vcmi.adventureMap.spellUnknownProblem"));
  608. }
  609. }
  610. }
  611. }
  612. void CSpellWindow::SpellArea::showPopupWindow(const Point & cursorPosition)
  613. {
  614. if(mySpell)
  615. {
  616. std::string dmgInfo;
  617. auto causedDmg = owner->myInt->cb->estimateSpellDamage(mySpell, owner->myHero);
  618. if(causedDmg == 0 || mySpell->id == SpellID::TITANS_LIGHTNING_BOLT) //Titan's Lightning Bolt already has damage info included
  619. dmgInfo.clear();
  620. else
  621. {
  622. dmgInfo = CGI->generaltexth->allTexts[343];
  623. boost::algorithm::replace_first(dmgInfo, "%d", std::to_string(causedDmg));
  624. }
  625. CRClickPopup::createAndPush(mySpell->getDescriptionTranslated(schoolLevel) + dmgInfo, std::make_shared<CComponent>(ComponentType::SPELL, mySpell->id));
  626. }
  627. }
  628. void CSpellWindow::SpellArea::hover(bool on)
  629. {
  630. if(mySpell)
  631. {
  632. if(on)
  633. owner->statusBar->write(boost::str(boost::format("%s (%s)") % mySpell->getNameTranslated() % CGI->generaltexth->allTexts[171+mySpell->getLevel()]));
  634. else
  635. owner->statusBar->clear();
  636. }
  637. }
  638. void CSpellWindow::SpellArea::setSpell(const CSpell * spell)
  639. {
  640. schoolBorder.reset();
  641. image->visible = false;
  642. name->setText("");
  643. level->setText("");
  644. cost->setText("");
  645. mySpell = spell;
  646. if(mySpell)
  647. {
  648. SpellSchool whichSchool; //0 - air magic, 1 - fire magic, 2 - water magic, 3 - earth magic,
  649. schoolLevel = owner->myHero->getSpellSchoolLevel(mySpell, &whichSchool);
  650. auto spellCost = owner->myInt->cb->getSpellCost(mySpell, owner->myHero);
  651. image->setFrame(mySpell->id);
  652. image->visible = true;
  653. {
  654. OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE);
  655. schoolBorder.reset();
  656. if (owner->selectedTab >= 4)
  657. {
  658. if (whichSchool.getNum() != SpellSchool())
  659. schoolBorder = std::make_shared<CAnimImage>(owner->schoolBorders.at(whichSchool.getNum()), schoolLevel);
  660. }
  661. else
  662. schoolBorder = std::make_shared<CAnimImage>(owner->schoolBorders.at(owner->selectedTab), schoolLevel);
  663. }
  664. ColorRGBA firstLineColor, secondLineColor;
  665. if(spellCost > owner->myHero->mana && !owner->onSpellSelect) //hero cannot cast this spell
  666. {
  667. firstLineColor = Colors::WHITE;
  668. secondLineColor = Colors::ORANGE;
  669. }
  670. else
  671. {
  672. firstLineColor = Colors::YELLOW;
  673. secondLineColor = Colors::WHITE;
  674. }
  675. name->color = firstLineColor;
  676. name->setText(mySpell->getNameTranslated());
  677. level->color = secondLineColor;
  678. if(schoolLevel > 0)
  679. {
  680. boost::format fmt("%s/%s");
  681. fmt % CGI->generaltexth->allTexts[171 + mySpell->getLevel()];
  682. fmt % CGI->generaltexth->levels[3+(schoolLevel-1)];//lines 4-6
  683. level->setText(fmt.str());
  684. }
  685. else
  686. level->setText(CGI->generaltexth->allTexts[171 + mySpell->getLevel()]);
  687. cost->color = secondLineColor;
  688. boost::format costfmt("%s: %d");
  689. costfmt % CGI->generaltexth->allTexts[387] % spellCost;
  690. cost->setText(costfmt.str());
  691. }
  692. }