mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-12 10:03:53 +02:00
278 lines
7.5 KiB
C++
278 lines
7.5 KiB
C++
/*
|
|
* CArtifactsOfHeroBase.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 "CArtifactsOfHeroBase.h"
|
|
|
|
#include "../gui/CGuiHandler.h"
|
|
#include "../gui/Shortcut.h"
|
|
|
|
#include "Buttons.h"
|
|
|
|
#include "../CPlayerInterface.h"
|
|
#include "../CGameInfo.h"
|
|
|
|
#include "../../CCallback.h"
|
|
|
|
#include "../../lib/ArtifactUtils.h"
|
|
#include "../../lib/mapObjects/CGHeroInstance.h"
|
|
#include "../../lib/networkPacks/ArtifactLocation.h"
|
|
|
|
CArtifactsOfHeroBase::CArtifactsOfHeroBase()
|
|
: backpackPos(0),
|
|
curHero(nullptr),
|
|
putBackPickedArtCallback(nullptr)
|
|
{
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::putBackPickedArtifact()
|
|
{
|
|
// Artifact located in artifactsTransitionPos should be returned
|
|
if(getPickedArtifact())
|
|
{
|
|
auto slot = ArtifactUtils::getArtAnyPosition(curHero, curHero->artifactsTransitionPos.begin()->artifact->getTypeId());
|
|
if(slot == ArtifactPosition::PRE_FIRST)
|
|
{
|
|
LOCPLINT->cb->eraseArtifactByClient(ArtifactLocation(curHero->id, ArtifactPosition::TRANSITION_POS));
|
|
}
|
|
else
|
|
{
|
|
LOCPLINT->cb->swapArtifacts(ArtifactLocation(curHero->id, ArtifactPosition::TRANSITION_POS), ArtifactLocation(curHero->id, slot));
|
|
}
|
|
}
|
|
if(putBackPickedArtCallback)
|
|
putBackPickedArtCallback();
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::setPutBackPickedArtifactCallback(PutBackPickedArtCallback callback)
|
|
{
|
|
putBackPickedArtCallback = callback;
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::init(
|
|
CHeroArtPlace::ClickFunctor lClickCallback,
|
|
CHeroArtPlace::ClickFunctor showPopupCallback,
|
|
const Point & position,
|
|
BpackScrollFunctor scrollCallback)
|
|
{
|
|
// CArtifactsOfHeroBase::init may be transform to CArtifactsOfHeroBase::CArtifactsOfHeroBase if OBJECT_CONSTRUCTION_CAPTURING is removed
|
|
OBJECT_CONSTRUCTION_CAPTURING(255 - DISPOSE);
|
|
pos += position;
|
|
for(int g = 0; g < ArtifactPosition::BACKPACK_START; g++)
|
|
{
|
|
artWorn[ArtifactPosition(g)] = std::make_shared<CHeroArtPlace>(slotPos[g]);
|
|
}
|
|
backpack.clear();
|
|
for(int s = 0; s < 5; s++)
|
|
{
|
|
auto artPlace = std::make_shared<CHeroArtPlace>(Point(403 + 46 * s, 365));
|
|
backpack.push_back(artPlace);
|
|
}
|
|
for(auto artPlace : artWorn)
|
|
{
|
|
artPlace.second->slot = artPlace.first;
|
|
artPlace.second->setArtifact(nullptr);
|
|
artPlace.second->leftClickCallback = lClickCallback;
|
|
artPlace.second->showPopupCallback = showPopupCallback;
|
|
}
|
|
for(auto artPlace : backpack)
|
|
{
|
|
artPlace->setArtifact(nullptr);
|
|
artPlace->leftClickCallback = lClickCallback;
|
|
artPlace->showPopupCallback = showPopupCallback;
|
|
}
|
|
leftBackpackRoll = std::make_shared<CButton>(Point(379, 364), AnimationPath::builtin("hsbtns3.def"), CButton::tooltip(), [scrollCallback]() {scrollCallback(-1);}, EShortcut::MOVE_LEFT);
|
|
rightBackpackRoll = std::make_shared<CButton>(Point(632, 364), AnimationPath::builtin("hsbtns5.def"), CButton::tooltip(), [scrollCallback]() {scrollCallback(+1);}, EShortcut::MOVE_RIGHT);
|
|
leftBackpackRoll->block(true);
|
|
rightBackpackRoll->block(true);
|
|
|
|
setRedrawParent(true);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::leftClickArtPlace(CHeroArtPlace & artPlace)
|
|
{
|
|
if(leftClickCallback)
|
|
leftClickCallback(*this, artPlace);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::rightClickArtPlace(CHeroArtPlace & artPlace)
|
|
{
|
|
if(showPopupCallback)
|
|
showPopupCallback(*this, artPlace);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::setHero(const CGHeroInstance * hero)
|
|
{
|
|
curHero = hero;
|
|
if(curHero->artifactsInBackpack.size() > 0)
|
|
backpackPos %= curHero->artifactsInBackpack.size();
|
|
else
|
|
backpackPos = 0;
|
|
|
|
for(auto slot : artWorn)
|
|
{
|
|
setSlotData(slot.second, slot.first, *curHero);
|
|
}
|
|
scrollBackpack(0);
|
|
}
|
|
|
|
const CGHeroInstance * CArtifactsOfHeroBase::getHero() const
|
|
{
|
|
return curHero;
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::scrollBackpack(int offset)
|
|
{
|
|
scrollBackpackForArtSet(offset, *curHero);
|
|
redraw();
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::scrollBackpackForArtSet(int offset, const CArtifactSet & artSet)
|
|
{
|
|
// offset==-1 => to left; offset==1 => to right
|
|
using slotInc = std::function<ArtifactPosition(ArtifactPosition&)>;
|
|
auto artsInBackpack = static_cast<int>(artSet.artifactsInBackpack.size());
|
|
auto scrollingPossible = artsInBackpack > backpack.size();
|
|
|
|
slotInc inc_straight = [](ArtifactPosition & slot) -> ArtifactPosition
|
|
{
|
|
return slot + 1;
|
|
};
|
|
slotInc inc_ring = [artsInBackpack](ArtifactPosition & slot) -> ArtifactPosition
|
|
{
|
|
return ArtifactPosition::BACKPACK_START + (slot - ArtifactPosition::BACKPACK_START + 1) % artsInBackpack;
|
|
};
|
|
slotInc inc;
|
|
if(scrollingPossible)
|
|
inc = inc_ring;
|
|
else
|
|
inc = inc_straight;
|
|
|
|
backpackPos += offset;
|
|
if(backpackPos < 0)
|
|
backpackPos += artsInBackpack;
|
|
|
|
if(artsInBackpack)
|
|
backpackPos %= artsInBackpack;
|
|
|
|
auto slot = ArtifactPosition(ArtifactPosition::BACKPACK_START + backpackPos);
|
|
for(auto artPlace : backpack)
|
|
{
|
|
setSlotData(artPlace, slot, artSet);
|
|
slot = inc(slot);
|
|
}
|
|
|
|
// Blocking scrolling if there is not enough artifacts to scroll
|
|
if(leftBackpackRoll)
|
|
leftBackpackRoll->block(!scrollingPossible);
|
|
if(rightBackpackRoll)
|
|
rightBackpackRoll->block(!scrollingPossible);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::markPossibleSlots(const CArtifactInstance * art, bool assumeDestRemoved)
|
|
{
|
|
for(auto artPlace : artWorn)
|
|
artPlace.second->selectSlot(art->canBePutAt(curHero, artPlace.second->slot, assumeDestRemoved));
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::unmarkSlots()
|
|
{
|
|
for(auto & artPlace : artWorn)
|
|
artPlace.second->selectSlot(false);
|
|
|
|
for(auto & artPlace : backpack)
|
|
artPlace->selectSlot(false);
|
|
}
|
|
|
|
CArtifactsOfHeroBase::ArtPlacePtr CArtifactsOfHeroBase::getArtPlace(const ArtifactPosition & slot)
|
|
{
|
|
if(ArtifactUtils::isSlotEquipment(slot))
|
|
{
|
|
if(artWorn.find(slot) == artWorn.end())
|
|
{
|
|
logGlobal->error("CArtifactsOfHero::getArtPlace: invalid slot %d", slot);
|
|
return nullptr;
|
|
}
|
|
return artWorn[slot];
|
|
}
|
|
if(ArtifactUtils::isSlotBackpack(slot))
|
|
{
|
|
for(ArtPlacePtr artPlace : backpack)
|
|
if(artPlace->slot == slot)
|
|
return artPlace;
|
|
return nullptr;
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::updateWornSlots()
|
|
{
|
|
for(auto place : artWorn)
|
|
updateSlot(place.first);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::updateBackpackSlots()
|
|
{
|
|
if(curHero->artifactsInBackpack.size() <= backpack.size() && backpackPos != 0)
|
|
backpackPos = 0;
|
|
scrollBackpack(0);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::updateSlot(const ArtifactPosition & slot)
|
|
{
|
|
setSlotData(getArtPlace(slot), slot, *curHero);
|
|
}
|
|
|
|
const CArtifactInstance * CArtifactsOfHeroBase::getPickedArtifact()
|
|
{
|
|
// Returns only the picked up artifact. Not just highlighted like in the trading window.
|
|
if(!curHero || curHero->artifactsTransitionPos.empty())
|
|
return nullptr;
|
|
else
|
|
return curHero->getArt(ArtifactPosition::TRANSITION_POS);
|
|
}
|
|
|
|
void CArtifactsOfHeroBase::setSlotData(ArtPlacePtr artPlace, const ArtifactPosition & slot, const CArtifactSet & artSet)
|
|
{
|
|
// Spurious call from artifactMoved in attempt to update hidden backpack slot
|
|
if(!artPlace && ArtifactUtils::isSlotBackpack(slot))
|
|
{
|
|
return;
|
|
}
|
|
|
|
artPlace->slot = slot;
|
|
if(auto slotInfo = artSet.getSlot(slot))
|
|
{
|
|
artPlace->lockSlot(slotInfo->locked);
|
|
artPlace->setArtifact(slotInfo->artifact);
|
|
if(!slotInfo->artifact->isCombined())
|
|
{
|
|
// If the artifact is part of at least one combined artifact, add additional information
|
|
std::map<const CArtifact*, int> arts;
|
|
for(const auto combinedArt : slotInfo->artifact->artType->getPartOf())
|
|
{
|
|
arts.insert(std::pair(combinedArt, 0));
|
|
for(const auto part : combinedArt->getConstituents())
|
|
{
|
|
if(artSet.hasArt(part->getId(), false))
|
|
arts.at(combinedArt)++;
|
|
}
|
|
}
|
|
artPlace->addCombinedArtInfo(arts);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
artPlace->setArtifact(nullptr);
|
|
}
|
|
}
|