CArtifactsOfHeroBase.cpp 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311
  1. /*
  2. * CArtifactsOfHeroBase.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 "CArtifactsOfHeroBase.h"
  12. #include "../GameEngine.h"
  13. #include "../gui/Shortcut.h"
  14. #include "Buttons.h"
  15. #include "../CPlayerInterface.h"
  16. #include "../../CCallback.h"
  17. #include "../../lib/ArtifactUtils.h"
  18. #include "../../lib/mapObjects/CGHeroInstance.h"
  19. #include "../../lib/networkPacks/ArtifactLocation.h"
  20. CArtifactsOfHeroBase::CArtifactsOfHeroBase()
  21. : curHero(nullptr)
  22. {
  23. }
  24. void CArtifactsOfHeroBase::putBackPickedArtifact()
  25. {
  26. // Artifact located in artifactsTransitionPos should be returned
  27. if(const auto art = getPickedArtifact())
  28. {
  29. auto slot = ArtifactUtils::getArtAnyPosition(curHero, art->getTypeId());
  30. if(slot == ArtifactPosition::PRE_FIRST)
  31. {
  32. LOCPLINT->cb->eraseArtifactByClient(ArtifactLocation(curHero->id, ArtifactPosition::TRANSITION_POS));
  33. }
  34. else
  35. {
  36. LOCPLINT->cb->swapArtifacts(ArtifactLocation(curHero->id, ArtifactPosition::TRANSITION_POS), ArtifactLocation(curHero->id, slot));
  37. }
  38. }
  39. }
  40. void CArtifactsOfHeroBase::init(
  41. const Point & position,
  42. const BpackScrollFunctor & scrollCallback)
  43. {
  44. // CArtifactsOfHeroBase::init may be transform to CArtifactsOfHeroBase::CArtifactsOfHeroBase if OBJECT_CONSTRUCTION is removed
  45. OBJECT_CONSTRUCTION;
  46. pos += position;
  47. for(int g = 0; g < ArtifactPosition::BACKPACK_START; g++)
  48. {
  49. artWorn[ArtifactPosition(g)] = std::make_shared<CArtPlace>(slotPos[g]);
  50. }
  51. backpack.clear();
  52. for(int s = 0; s < 5; s++)
  53. {
  54. auto artPlace = std::make_shared<CArtPlace>(Point(403 + 46 * s, 365));
  55. backpack.push_back(artPlace);
  56. }
  57. for(auto & artPlace : artWorn)
  58. {
  59. artPlace.second->slot = artPlace.first;
  60. artPlace.second->setArtifact(ArtifactID(ArtifactID::NONE));
  61. }
  62. for(const auto & artPlace : backpack)
  63. {
  64. artPlace->setArtifact(ArtifactID(ArtifactID::NONE));
  65. }
  66. leftBackpackRoll = std::make_shared<CButton>(Point(379, 364), AnimationPath::builtin("hsbtns3.def"), CButton::tooltip(),
  67. [scrollCallback](){scrollCallback(true);}, EShortcut::MOVE_LEFT);
  68. rightBackpackRoll = std::make_shared<CButton>(Point(632, 364), AnimationPath::builtin("hsbtns5.def"), CButton::tooltip(),
  69. [scrollCallback](){scrollCallback(false);}, EShortcut::MOVE_RIGHT);
  70. leftBackpackRoll->block(true);
  71. rightBackpackRoll->block(true);
  72. backpackScroller = std::make_shared<BackpackScroller>(this, Rect(380, 30, 278, 382));
  73. backpackScroller->setScrollingEnabled(false);
  74. setRedrawParent(true);
  75. }
  76. void CArtifactsOfHeroBase::setClickPressedArtPlacesCallback(const CArtPlace::ClickFunctor & callback) const
  77. {
  78. for(const auto & [slot, artPlace] : artWorn)
  79. artPlace->setClickPressedCallback(callback);
  80. for(const auto & artPlace : backpack)
  81. artPlace->setClickPressedCallback(callback);
  82. }
  83. void CArtifactsOfHeroBase::setShowPopupArtPlacesCallback(const CArtPlace::ClickFunctor & callback) const
  84. {
  85. for(const auto & [slot, artPlace] : artWorn)
  86. artPlace->setShowPopupCallback(callback);
  87. for(const auto & artPlace : backpack)
  88. artPlace->setShowPopupCallback(callback);
  89. }
  90. void CArtifactsOfHeroBase::clickPressedArtPlace(CComponentHolder & artPlace, const Point & cursorPosition)
  91. {
  92. if(auto ownedPlace = getArtPlace(cursorPosition))
  93. {
  94. if(ownedPlace->isLocked())
  95. return;
  96. if(clickPressedCallback)
  97. clickPressedCallback(*ownedPlace, cursorPosition);
  98. }
  99. }
  100. void CArtifactsOfHeroBase::showPopupArtPlace(CComponentHolder & artPlace, const Point & cursorPosition)
  101. {
  102. if(auto ownedPlace = getArtPlace(cursorPosition))
  103. {
  104. if(ownedPlace->isLocked())
  105. return;
  106. if(showPopupCallback)
  107. showPopupCallback(*ownedPlace, cursorPosition);
  108. }
  109. }
  110. void CArtifactsOfHeroBase::gestureArtPlace(CComponentHolder & artPlace, const Point & cursorPosition)
  111. {
  112. if(auto ownedPlace = getArtPlace(cursorPosition))
  113. {
  114. if(ownedPlace->isLocked())
  115. return;
  116. if(gestureCallback)
  117. gestureCallback(*ownedPlace, cursorPosition);
  118. }
  119. }
  120. void CArtifactsOfHeroBase::setHero(const CGHeroInstance * hero)
  121. {
  122. curHero = hero;
  123. if (!hero)
  124. return;
  125. for(auto slot : artWorn)
  126. {
  127. setSlotData(slot.second, slot.first);
  128. }
  129. updateBackpackSlots();
  130. }
  131. const CGHeroInstance * CArtifactsOfHeroBase::getHero() const
  132. {
  133. return curHero;
  134. }
  135. void CArtifactsOfHeroBase::scrollBackpack(bool left)
  136. {
  137. LOCPLINT->cb->scrollBackpackArtifacts(curHero->id, left);
  138. }
  139. void CArtifactsOfHeroBase::markPossibleSlots(const CArtifact * art, bool assumeDestRemoved)
  140. {
  141. for(const auto & artPlace : artWorn)
  142. artPlace.second->selectSlot(art->canBePutAt(curHero, artPlace.second->slot, assumeDestRemoved));
  143. }
  144. void CArtifactsOfHeroBase::unmarkSlots()
  145. {
  146. for(auto & artPlace : artWorn)
  147. artPlace.second->selectSlot(false);
  148. for(auto & artPlace : backpack)
  149. artPlace->selectSlot(false);
  150. }
  151. CArtifactsOfHeroBase::ArtPlacePtr CArtifactsOfHeroBase::getArtPlace(const ArtifactPosition & slot)
  152. {
  153. if(ArtifactUtils::isSlotEquipment(slot) && artWorn.find(slot) != artWorn.end())
  154. return artWorn[slot];
  155. if(ArtifactUtils::isSlotBackpack(slot) && slot - ArtifactPosition::BACKPACK_START < backpack.size())
  156. return(backpack[slot - ArtifactPosition::BACKPACK_START]);
  157. logGlobal->error("CArtifactsOfHero::getArtPlace: invalid slot %d", slot);
  158. return nullptr;
  159. }
  160. CArtifactsOfHeroBase::ArtPlacePtr CArtifactsOfHeroBase::getArtPlace(const Point & cursorPosition)
  161. {
  162. for(const auto & [slot, artPlace] : artWorn)
  163. {
  164. if(artPlace->pos.isInside(cursorPosition))
  165. return artPlace;
  166. }
  167. for(const auto & artPlace : backpack)
  168. {
  169. if(artPlace->pos.isInside(cursorPosition))
  170. return artPlace;
  171. }
  172. return nullptr;
  173. }
  174. void CArtifactsOfHeroBase::updateWornSlots()
  175. {
  176. for(auto place : artWorn)
  177. updateSlot(place.first);
  178. }
  179. void CArtifactsOfHeroBase::updateBackpackSlots()
  180. {
  181. ArtifactPosition slot = ArtifactPosition::BACKPACK_START;
  182. for(const auto & artPlace : backpack)
  183. {
  184. setSlotData(artPlace, slot);
  185. slot = slot + 1;
  186. }
  187. auto scrollingPossible = static_cast<int>(curHero->artifactsInBackpack.size()) > backpack.size();
  188. // Blocking scrolling if there is not enough artifacts to scroll
  189. if(leftBackpackRoll)
  190. leftBackpackRoll->block(!scrollingPossible);
  191. if(rightBackpackRoll)
  192. rightBackpackRoll->block(!scrollingPossible);
  193. if (backpackScroller)
  194. backpackScroller->setScrollingEnabled(scrollingPossible);
  195. }
  196. void CArtifactsOfHeroBase::updateSlot(const ArtifactPosition & slot)
  197. {
  198. setSlotData(getArtPlace(slot), slot);
  199. }
  200. const CArtifactInstance * CArtifactsOfHeroBase::getPickedArtifact()
  201. {
  202. // Returns only the picked up artifact. Not just highlighted like in the trading window.
  203. if(curHero)
  204. return curHero->getArt(ArtifactPosition::TRANSITION_POS);
  205. else
  206. return nullptr;
  207. }
  208. void CArtifactsOfHeroBase::enableGesture()
  209. {
  210. for(auto & artPlace : artWorn)
  211. {
  212. artPlace.second->setGestureCallback(std::bind(&CArtifactsOfHeroBase::gestureArtPlace, this, _1, _2));
  213. artPlace.second->addUsedEvents(GESTURE);
  214. }
  215. }
  216. const CArtifactInstance * CArtifactsOfHeroBase::getArt(const ArtifactPosition & slot) const
  217. {
  218. return curHero ? curHero->getArt(slot) : nullptr;
  219. }
  220. void CArtifactsOfHeroBase::enableKeyboardShortcuts()
  221. {
  222. addUsedEvents(AEventsReceiver::KEYBOARD);
  223. }
  224. void CArtifactsOfHeroBase::setSlotData(ArtPlacePtr artPlace, const ArtifactPosition & slot)
  225. {
  226. // Spurious call from artifactMoved in attempt to update hidden backpack slot
  227. if(!artPlace && ArtifactUtils::isSlotBackpack(slot))
  228. {
  229. return;
  230. }
  231. artPlace->slot = slot;
  232. if(auto slotInfo = curHero->getSlot(slot))
  233. {
  234. artPlace->lockSlot(slotInfo->locked);
  235. artPlace->setArtifact(slotInfo->artifact->getTypeId(), slotInfo->artifact->getScrollSpellID());
  236. if(slotInfo->locked || slotInfo->artifact->isCombined())
  237. return;
  238. // If the artifact is part of at least one combined artifact, add additional information
  239. std::map<const ArtifactID, std::vector<ArtifactID>> arts;
  240. for(const auto combinedArt : slotInfo->artifact->getType()->getPartOf())
  241. {
  242. assert(combinedArt->isCombined());
  243. arts.try_emplace(combinedArt->getId());
  244. CArtifactFittingSet fittingSet(*curHero);
  245. for(const auto part : combinedArt->getConstituents())
  246. {
  247. const auto partSlot = fittingSet.getArtPos(part->getId(), false, false);
  248. if(partSlot != ArtifactPosition::PRE_FIRST)
  249. {
  250. arts.at(combinedArt->getId()).emplace_back(part->getId());
  251. fittingSet.lockSlot(partSlot);
  252. }
  253. }
  254. }
  255. artPlace->addCombinedArtInfo(arts);
  256. }
  257. else
  258. {
  259. artPlace->setArtifact(ArtifactID(ArtifactID::NONE));
  260. }
  261. }
  262. BackpackScroller::BackpackScroller(CArtifactsOfHeroBase * owner, const Rect & dimensions)
  263. : Scrollable(0, Point(), Orientation::HORIZONTAL)
  264. , owner(owner)
  265. {
  266. pos = dimensions + pos.topLeft();
  267. setPanningStep(46);
  268. }
  269. void BackpackScroller::scrollBy(int distance)
  270. {
  271. if (distance != 0)
  272. owner->scrollBackpack(distance < 0);
  273. }