/* * CSpellWindow.cpp, part of VCMI engine * * Authors: listed in file AUTHORS in main folder * * License: GNU General Public License v2.0 or later * Full text of license available in license.txt file, in main folder * */ #include "StdInc.h" #include "CSpellWindow.h" #include "../../lib/ScopeGuard.h" #include "GUIClasses.h" #include "InfoWindows.h" #include "CCastleInterface.h" #include "../CGameInfo.h" #include "../CPlayerInterface.h" #include "../PlayerLocalState.h" #include "../CVideoHandler.h" #include "../battle/BattleInterface.h" #include "../gui/CGuiHandler.h" #include "../gui/Shortcut.h" #include "../gui/WindowHandler.h" #include "../widgets/MiscWidgets.h" #include "../widgets/CComponent.h" #include "../widgets/TextControls.h" #include "../adventureMap/AdventureMapInterface.h" #include "../render/CAnimation.h" #include "../render/IRenderHandler.h" #include "../render/IImage.h" #include "../render/IImageLoader.h" #include "../render/Canvas.h" #include "../../CCallback.h" #include "../../lib/CConfigHandler.h" #include "../../lib/CGeneralTextHandler.h" #include "../../lib/spells/CSpellHandler.h" #include "../../lib/spells/Problem.h" #include "../../lib/GameConstants.h" #include "../../lib/mapObjects/CGHeroInstance.h" CSpellWindow::InteractiveArea::InteractiveArea(const Rect & myRect, std::function funcL, int helpTextId, CSpellWindow * _owner) { addUsedEvents(LCLICK | SHOW_POPUP | HOVER); pos = myRect; onLeft = funcL; hoverText = CGI->generaltexth->zelp[helpTextId].first; helpText = CGI->generaltexth->zelp[helpTextId].second; owner = _owner; } void CSpellWindow::InteractiveArea::clickPressed(const Point & cursorPosition) { onLeft(); } void CSpellWindow::InteractiveArea::showPopupWindow(const Point & cursorPosition) { CRClickPopup::createAndPush(helpText); } void CSpellWindow::InteractiveArea::hover(bool on) { if(on) owner->statusBar->write(hoverText); else owner->statusBar->clear(); } class SpellbookSpellSorter { public: bool operator()(const CSpell * A, const CSpell * B) { if(A->getLevel() < B->getLevel()) return true; if(A->getLevel() > B->getLevel()) return false; for(auto schoolId = 0; schoolId < GameConstants::DEFAULT_SCHOOLS; schoolId++) { if(A->school.at(SpellSchool(schoolId)) && !B->school.at(SpellSchool(schoolId))) return true; if(!A->school.at(SpellSchool(schoolId)) && B->school.at(SpellSchool(schoolId))) return false; } return A->getNameTranslated() < B->getNameTranslated(); } } spellsorter; CSpellWindow::CSpellWindow(const CGHeroInstance * _myHero, CPlayerInterface * _myInt, bool openOnBattleSpells): CWindowObject(PLAYER_COLORED | (settings["general"]["enableUiEnhancements"].Bool() ? BORDERED : 0)), battleSpellsOnly(openOnBattleSpells), selectedTab(4), currentPage(0), myHero(_myHero), myInt(_myInt), isBigSpellbook(settings["general"]["enableUiEnhancements"].Bool()), spellsPerPage(24), offL(-11), offR(195), offRM(110), offT(-37), offB(56) { OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE); if(isBigSpellbook) { background = std::make_shared(createBigSpellBook(), Point(0, 0)); updateShadow(); } else { background = std::make_shared(ImagePath::builtin("SpelBack"), 0, 0); offL = offR = offT = offB = offRM = 0; spellsPerPage = 12; } pos = background->center(Point(pos.w/2 + pos.x, pos.h/2 + pos.y)); //initializing castable spells mySpells.reserve(CGI->spellh->objects.size()); for(const CSpell * spell : CGI->spellh->objects) { if(!spell->isCreatureAbility() && myHero->canCastThisSpell(spell)) mySpells.push_back(spell); } std::sort(mySpells.begin(), mySpells.end(), spellsorter); //initializing sizes of spellbook's parts for(auto & elem : sitesPerTabAdv) elem = 0; for(auto & elem : sitesPerTabBattle) elem = 0; for(const auto spell : mySpells) { int * sitesPerOurTab = spell->isCombat() ? sitesPerTabBattle : sitesPerTabAdv; ++sitesPerOurTab[4]; spell->forEachSchool([&sitesPerOurTab](const SpellSchool & school, bool & stop) { ++sitesPerOurTab[school]; }); } if(sitesPerTabAdv[4] % spellsPerPage == 0) sitesPerTabAdv[4]/=spellsPerPage; else sitesPerTabAdv[4] = sitesPerTabAdv[4]/spellsPerPage + 1; for(int v=0; v<4; ++v) { if(sitesPerTabAdv[v] <= spellsPerPage - 2) sitesPerTabAdv[v] = 1; else { if((sitesPerTabAdv[v] - spellsPerPage - 2) % spellsPerPage == 0) sitesPerTabAdv[v] = (sitesPerTabAdv[v] - spellsPerPage - 2) / spellsPerPage + 1; else sitesPerTabAdv[v] = (sitesPerTabAdv[v] - spellsPerPage - 2) / spellsPerPage + 2; } } if(sitesPerTabBattle[4] % spellsPerPage == 0) sitesPerTabBattle[4]/=spellsPerPage; else sitesPerTabBattle[4] = sitesPerTabBattle[4]/spellsPerPage + 1; for(int v=0; v<4; ++v) { if(sitesPerTabBattle[v] <= spellsPerPage - 2) sitesPerTabBattle[v] = 1; else { if((sitesPerTabBattle[v] - spellsPerPage - 2) % spellsPerPage == 0) sitesPerTabBattle[v] = (sitesPerTabBattle[v] - spellsPerPage - 2) / spellsPerPage + 1; else sitesPerTabBattle[v] = (sitesPerTabBattle[v] - spellsPerPage - 2) / spellsPerPage + 2; } } //numbers of spell pages computed leftCorner = std::make_shared(ImagePath::builtin("SpelTrnL.bmp"), 97 + offL, 77 + offT); rightCorner = std::make_shared(ImagePath::builtin("SpelTrnR.bmp"), 487 + offR, 72 + offT); spellIcons = GH.renderHandler().loadAnimation(AnimationPath::builtin("Spells")); schoolTab = std::make_shared(AnimationPath::builtin("SpelTab"), selectedTab, 0, 524 + offR, 88); schoolPicture = std::make_shared(AnimationPath::builtin("Schools"), 0, 0, 117 + offL, 74 + offT); schoolBorders[0] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevA.def")); schoolBorders[1] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevF.def")); schoolBorders[2] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevW.def")); schoolBorders[3] = GH.renderHandler().loadAnimation(AnimationPath::builtin("SplevE.def")); for(auto item : schoolBorders) item->preload(); mana = std::make_shared(435 + (isBigSpellbook ? 159 : 0), 426 + offB, FONT_SMALL, ETextAlignment::CENTER, Colors::YELLOW, std::to_string(myHero->mana)); if(isBigSpellbook) statusBar = CGStatusBar::create(400, 587); else statusBar = CGStatusBar::create(7, 569, ImagePath::builtin("Spelroll.bmp")); Rect schoolRect( 549 + pos.x + offR, 94 + pos.y, 45, 35); interactiveAreas.push_back(std::make_shared( Rect( 479 + pos.x + (isBigSpellbook ? 175 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fexitb, this), 460, this)); interactiveAreas.push_back(std::make_shared( Rect( 221 + pos.x + (isBigSpellbook ? 43 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fbattleSpellsb, this), 453, this)); interactiveAreas.push_back(std::make_shared( Rect( 355 + pos.x + (isBigSpellbook ? 110 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fadvSpellsb, this), 452, this)); interactiveAreas.push_back(std::make_shared( Rect( 418 + pos.x + (isBigSpellbook ? 142 : 0), 405 + pos.y + offB, isBigSpellbook ? 60 : 36, 56), std::bind(&CSpellWindow::fmanaPtsb, this), 459, this)); interactiveAreas.push_back(std::make_shared( schoolRect + Point(0, 0), std::bind(&CSpellWindow::selectSchool, this, 0), 454, this)); interactiveAreas.push_back(std::make_shared( schoolRect + Point(0, 57), std::bind(&CSpellWindow::selectSchool, this, 3), 457, this)); interactiveAreas.push_back(std::make_shared( schoolRect + Point(0, 116), std::bind(&CSpellWindow::selectSchool, this, 1), 455, this)); interactiveAreas.push_back(std::make_shared( schoolRect + Point(0, 176), std::bind(&CSpellWindow::selectSchool, this, 2), 456, this)); interactiveAreas.push_back(std::make_shared( schoolRect + Point(0, 236), std::bind(&CSpellWindow::selectSchool, this, 4), 458, this)); interactiveAreas.push_back(std::make_shared( Rect( 97 + offL + pos.x, 77 + offT + pos.y, leftCorner->pos.h, leftCorner->pos.w ), std::bind(&CSpellWindow::fLcornerb, this), 450, this)); interactiveAreas.push_back(std::make_shared( Rect( 487 + offR + pos.x, 72 + offT + pos.y, rightCorner->pos.h, rightCorner->pos.w ), std::bind(&CSpellWindow::fRcornerb, this), 451, this)); //areas for spells int xpos = 117 + offL + pos.x, ypos = 90 + offT + pos.y; for(int v=0; v( Rect(xpos, ypos, 65, 78), this); if(v == (spellsPerPage / 2) - 1) //to right page { xpos = offRM + 336 + pos.x; ypos = 90 + offT + pos.y; } else { if(v%(isBigSpellbook ? 3 : 2) == 0 || (v%3 == 1 && isBigSpellbook)) { xpos+=85; } else { xpos -= (isBigSpellbook ? 2 : 1)*85; ypos+=97; } } } selectedTab = battleSpellsOnly ? myInt->localState->spellbookSettings.spellbookLastTabBattle : myInt->localState->spellbookSettings.spellbookLastTabAdvmap; schoolTab->setFrame(selectedTab, 0); int cp = battleSpellsOnly ? myInt->localState->spellbookSettings.spellbookLastPageBattle : myInt->localState->spellbookSettings.spellbokLastPageAdvmap; // spellbook last page battle index is not reset after battle, so this needs to stay here vstd::abetween(cp, 0, std::max(0, pagesWithinCurrentTab() - 1)); setCurrentPage(cp); computeSpellsPerArea(); addUsedEvents(KEYBOARD); } CSpellWindow::~CSpellWindow() { } std::shared_ptr CSpellWindow::createBigSpellBook() { std::shared_ptr img = GH.renderHandler().loadImage(ImagePath::builtin("SpelBack")); Canvas canvas = Canvas(Point(800, 600)); // edges canvas.draw(img, Point(0, 0), Rect(15, 38, 90, 45)); canvas.draw(img, Point(0, 460), Rect(15, 400, 90, 141)); canvas.draw(img, Point(705, 0), Rect(509, 38, 95, 45)); canvas.draw(img, Point(705, 460), Rect(509, 400, 95, 141)); // left / right Canvas tmp1 = Canvas(Point(90, 355 - 45)); tmp1.draw(img, Point(0, 0), Rect(15, 38 + 45, 90, 355 - 45)); canvas.drawScaled(tmp1, Point(0, 45), Point(90, 415)); Canvas tmp2 = Canvas(Point(95, 355 - 45)); tmp2.draw(img, Point(0, 0), Rect(509, 38 + 45, 95, 355 - 45)); canvas.drawScaled(tmp2, Point(705, 45), Point(95, 415)); // top / bottom Canvas tmp3 = Canvas(Point(409, 45)); tmp3.draw(img, Point(0, 0), Rect(100, 38, 409, 45)); canvas.drawScaled(tmp3, Point(90, 0), Point(615, 45)); Canvas tmp4 = Canvas(Point(409, 141)); tmp4.draw(img, Point(0, 0), Rect(100, 400, 409, 141)); canvas.drawScaled(tmp4, Point(90, 460), Point(615, 141)); // middle Canvas tmp5 = Canvas(Point(409, 141)); tmp5.draw(img, Point(0, 0), Rect(100, 38 + 45, 509 - 15, 400 - 38)); canvas.drawScaled(tmp5, Point(90, 45), Point(615, 415)); return GH.renderHandler().createImage(canvas.getInternalSurface()); } void CSpellWindow::fexitb() { (myInt->battleInt ? myInt->localState->spellbookSettings.spellbookLastTabBattle : myInt->localState->spellbookSettings.spellbookLastTabAdvmap) = selectedTab; (myInt->battleInt ? myInt->localState->spellbookSettings.spellbookLastPageBattle : myInt->localState->spellbookSettings.spellbokLastPageAdvmap) = currentPage; close(); } void CSpellWindow::fadvSpellsb() { if(battleSpellsOnly == true) { turnPageRight(); battleSpellsOnly = false; setCurrentPage(0); } computeSpellsPerArea(); } void CSpellWindow::fbattleSpellsb() { if(battleSpellsOnly == false) { turnPageLeft(); battleSpellsOnly = true; setCurrentPage(0); } computeSpellsPerArea(); } void CSpellWindow::fmanaPtsb() { } void CSpellWindow::selectSchool(int school) { if(selectedTab != school) { if(selectedTab < school) turnPageLeft(); else turnPageRight(); selectedTab = school; schoolTab->setFrame(selectedTab, 0); setCurrentPage(0); } computeSpellsPerArea(); } void CSpellWindow::fLcornerb() { if(currentPage>0) { turnPageLeft(); setCurrentPage(currentPage - 1); } computeSpellsPerArea(); } void CSpellWindow::fRcornerb() { if((currentPage + 1) < (pagesWithinCurrentTab())) { turnPageRight(); setCurrentPage(currentPage + 1); } computeSpellsPerArea(); } void CSpellWindow::show(Canvas & to) { statusBar->show(to); } void CSpellWindow::computeSpellsPerArea() { std::vector spellsCurSite; spellsCurSite.reserve(mySpells.size()); for(const CSpell * spell : mySpells) { if(spell->isCombat() ^ !battleSpellsOnly && ((selectedTab == 4) || spell->school.at(SpellSchool(selectedTab))) ) { spellsCurSite.push_back(spell); } } if(selectedTab == 4) { if(spellsCurSite.size() > spellsPerPage) { spellsCurSite = std::vector(spellsCurSite.begin() + currentPage*spellsPerPage, spellsCurSite.end()); if(spellsCurSite.size() > spellsPerPage) { spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage, spellsCurSite.end()); } } } else //selectedTab == 0, 1, 2 or 3 { if(spellsCurSite.size() > spellsPerPage - 2) { if(currentPage == 0) { spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage-2, spellsCurSite.end()); } else { spellsCurSite = std::vector(spellsCurSite.begin() + (currentPage-1)*spellsPerPage + spellsPerPage-2, spellsCurSite.end()); if(spellsCurSite.size() > spellsPerPage) { spellsCurSite.erase(spellsCurSite.begin()+spellsPerPage, spellsCurSite.end()); } } } } //applying if(selectedTab == 4 || currentPage != 0) { for(size_t c=0; csetSpell(spellsCurSite[c]); } else { spellAreas[c]->setSpell(nullptr); } } } else { spellAreas[0]->setSpell(nullptr); spellAreas[1]->setSpell(nullptr); for(size_t c=0; csetSpell(spellsCurSite[c]); else spellAreas[c+2]->setSpell(nullptr); } } redraw(); } void CSpellWindow::setCurrentPage(int value) { currentPage = value; schoolPicture->visible = selectedTab!=4 && currentPage == 0; if(selectedTab != 4) schoolPicture->setFrame(selectedTab, 0); leftCorner->visible = currentPage != 0; rightCorner->visible = (currentPage+1) < pagesWithinCurrentTab(); mana->setText(std::to_string(myHero->mana));//just in case, it will be possible to cast spell without closing book } void CSpellWindow::turnPageLeft() { if(settings["video"]["spellbookAnimation"].Bool() && !isBigSpellbook) CCS->videoh->openAndPlayVideo(VideoPath::builtin("PGTRNLFT.SMK"), pos.x+13, pos.y+15); } void CSpellWindow::turnPageRight() { if(settings["video"]["spellbookAnimation"].Bool() && !isBigSpellbook) CCS->videoh->openAndPlayVideo(VideoPath::builtin("PGTRNRGH.SMK"), pos.x+13, pos.y+15); } void CSpellWindow::keyPressed(EShortcut key) { switch(key) { case EShortcut::GLOBAL_RETURN: fexitb(); break; case EShortcut::MOVE_LEFT: fLcornerb(); break; case EShortcut::MOVE_RIGHT: fRcornerb(); break; case EShortcut::MOVE_UP: case EShortcut::MOVE_DOWN: { bool down = key == EShortcut::MOVE_DOWN; static const int schoolsOrder[] = { 0, 3, 1, 2, 4 }; int index = -1; while(schoolsOrder[++index] != selectedTab); index += (down ? 1 : -1); vstd::abetween(index, 0, std::size(schoolsOrder) - 1); if(selectedTab != schoolsOrder[index]) selectSchool(schoolsOrder[index]); break; } case EShortcut::SPELLBOOK_TAB_COMBAT: fbattleSpellsb(); break; case EShortcut::SPELLBOOK_TAB_ADVENTURE: fadvSpellsb(); break; } } int CSpellWindow::pagesWithinCurrentTab() { return battleSpellsOnly ? sitesPerTabBattle[selectedTab] : sitesPerTabAdv[selectedTab]; } CSpellWindow::SpellArea::SpellArea(Rect pos, CSpellWindow * owner) { this->pos = pos; this->owner = owner; addUsedEvents(LCLICK | SHOW_POPUP | HOVER); schoolLevel = -1; mySpell = nullptr; OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE); image = std::make_shared(owner->spellIcons, 0, 0); image->visible = false; name = std::make_shared(39, 70, FONT_TINY, ETextAlignment::CENTER); level = std::make_shared(39, 82, FONT_TINY, ETextAlignment::CENTER); cost = std::make_shared(39, 94, FONT_TINY, ETextAlignment::CENTER); for(auto l : {name, level, cost}) l->setAutoRedraw(false); } CSpellWindow::SpellArea::~SpellArea() = default; void CSpellWindow::SpellArea::clickPressed(const Point & cursorPosition) { if(mySpell) { auto spellCost = owner->myInt->cb->getSpellCost(mySpell, owner->myHero); if(spellCost > owner->myHero->mana) //insufficient mana { LOCPLINT->showInfoDialog(boost::str(boost::format(CGI->generaltexth->allTexts[206]) % spellCost % owner->myHero->mana)); return; } //anything that is not combat spell is adventure spell //this not an error in general to cast even creature ability with hero const bool combatSpell = mySpell->isCombat(); if(combatSpell == mySpell->isAdventure()) { logGlobal->error("Spell have invalid flags"); return; } const bool inCombat = owner->myInt->battleInt != nullptr; const bool inCastle = owner->myInt->castleInt != nullptr; //battle spell on adv map or adventure map spell during combat => display infowindow, not cast if((combatSpell ^ inCombat) || inCastle) { std::vector> hlp(1, std::make_shared(ComponentType::SPELL, mySpell->id)); LOCPLINT->showInfoDialog(mySpell->getDescriptionTranslated(schoolLevel), hlp); } else if(combatSpell) { spells::detail::ProblemImpl problem; if(mySpell->canBeCast(problem, owner->myInt->battleInt->getBattle().get(), spells::Mode::HERO, owner->myHero)) { owner->myInt->battleInt->castThisSpell(mySpell->id); owner->fexitb(); } else { std::vector texts; problem.getAll(texts); if(!texts.empty()) LOCPLINT->showInfoDialog(texts.front()); else LOCPLINT->showInfoDialog(CGI->generaltexth->translate("vcmi.adventureMap.spellUnknownProblem")); } } else //adventure spell { const CGHeroInstance * h = owner->myHero; GH.windows().popWindows(1); auto guard = vstd::makeScopeGuard([this]() { owner->myInt->localState->spellbookSettings.spellbookLastTabAdvmap = owner->selectedTab; owner->myInt->localState->spellbookSettings.spellbokLastPageAdvmap = owner->currentPage; }); if(mySpell->getTargetType() == spells::AimType::LOCATION) adventureInt->enterCastingMode(mySpell); else if(mySpell->getTargetType() == spells::AimType::NO_TARGET) owner->myInt->cb->castSpell(h, mySpell->id); else logGlobal->error("Invalid spell target type"); } } } void CSpellWindow::SpellArea::showPopupWindow(const Point & cursorPosition) { if(mySpell) { std::string dmgInfo; auto causedDmg = owner->myInt->cb->estimateSpellDamage(mySpell, owner->myHero); if(causedDmg == 0 || mySpell->id == SpellID::TITANS_LIGHTNING_BOLT) //Titan's Lightning Bolt already has damage info included dmgInfo.clear(); else { dmgInfo = CGI->generaltexth->allTexts[343]; boost::algorithm::replace_first(dmgInfo, "%d", std::to_string(causedDmg)); } CRClickPopup::createAndPush(mySpell->getDescriptionTranslated(schoolLevel) + dmgInfo, std::make_shared(ComponentType::SPELL, mySpell->id)); } } void CSpellWindow::SpellArea::hover(bool on) { if(mySpell) { if(on) owner->statusBar->write(boost::str(boost::format("%s (%s)") % mySpell->getNameTranslated() % CGI->generaltexth->allTexts[171+mySpell->getLevel()])); else owner->statusBar->clear(); } } void CSpellWindow::SpellArea::setSpell(const CSpell * spell) { schoolBorder.reset(); image->visible = false; name->setText(""); level->setText(""); cost->setText(""); mySpell = spell; if(mySpell) { SpellSchool whichSchool; //0 - air magic, 1 - fire magic, 2 - water magic, 3 - earth magic, schoolLevel = owner->myHero->getSpellSchoolLevel(mySpell, &whichSchool); auto spellCost = owner->myInt->cb->getSpellCost(mySpell, owner->myHero); image->setFrame(mySpell->id); image->visible = true; { OBJECT_CONSTRUCTION_CAPTURING(255-DISPOSE); schoolBorder = std::make_shared(owner->schoolBorders[owner->selectedTab >= 4 ? whichSchool.getNum() : owner->selectedTab], schoolLevel); } ColorRGBA firstLineColor, secondLineColor; if(spellCost > owner->myHero->mana) //hero cannot cast this spell { firstLineColor = Colors::WHITE; secondLineColor = Colors::ORANGE; } else { firstLineColor = Colors::YELLOW; secondLineColor = Colors::WHITE; } name->color = firstLineColor; name->setText(mySpell->getNameTranslated()); level->color = secondLineColor; if(schoolLevel > 0) { boost::format fmt("%s/%s"); fmt % CGI->generaltexth->allTexts[171 + mySpell->getLevel()]; fmt % CGI->generaltexth->levels[3+(schoolLevel-1)];//lines 4-6 level->setText(fmt.str()); } else level->setText(CGI->generaltexth->allTexts[171 + mySpell->getLevel()]); cost->color = secondLineColor; boost::format costfmt("%s: %d"); costfmt % CGI->generaltexth->allTexts[387] % spellCost; cost->setText(costfmt.str()); } }