diff --git a/lib/IGameCallback.cpp b/lib/IGameCallback.cpp index 4472c6945..98fd54c6e 100644 --- a/lib/IGameCallback.cpp +++ b/lib/IGameCallback.cpp @@ -55,6 +55,11 @@ si8 CBattleInfoCallback::battleCanTeleportTo(const CStack * stack, THex destHex, } std::vector CBattleInfoCallback::battleGetDistances(const CStack * stack, THex hex /*= THex::INVALID*/, THex * predecessors /*= NULL*/) +{ + return battleGetDistancesFromHex(stack, stack->position, predecessors); +} + +std::vector CBattleInfoCallback::battleGetDistancesFromHex(const CStack * stack, THex hex /*= THex::INVALID*/, THex * predecessors /*= NULL*/) { if(!hex.isValid()) hex = stack->position; @@ -65,7 +70,7 @@ std::vector CBattleInfoCallback::battleGetDistances(const CStack * stack, T gs->curB->getAccessibilityMap(ac, stack->doubleWide(), stack->attackerOwned, false, occupyable, stack->hasBonusOfType(Bonus::FLYING), stack); THex pr[BFIELD_SIZE]; int dist[BFIELD_SIZE]; - gs->curB->makeBFS(stack->position, ac, pr, dist, stack->doubleWide(), stack->attackerOwned, stack->hasBonusOfType(Bonus::FLYING), false); + gs->curB->makeBFS(hex, ac, pr, dist, stack->doubleWide(), stack->attackerOwned, stack->hasBonusOfType(Bonus::FLYING), false); for(int i=0; i CBattleInfoCallback::battleGetDistances(const CStack * stack, T return ret; } + std::set CBattleInfoCallback::battleGetAttackedHexes(const CStack* attacker, THex destinationTile, THex attackerPos /*= THex::INVALID*/) { if(!gs->curB) @@ -370,8 +376,6 @@ const CGHeroInstance * CBattleInfoCallback::battleGetFightingHero(ui8 side) cons return gs->curB->heroes[side]; } - - CGameState *const CPrivilagedInfoCallback::gameState () { return gs; diff --git a/lib/IGameCallback.h b/lib/IGameCallback.h index 9c0fbc093..fdc7ce5a9 100644 --- a/lib/IGameCallback.h +++ b/lib/IGameCallback.h @@ -120,9 +120,12 @@ public: /// returns numbers of hexes reachable by creature with id ID std::vector battleGetAvailableHexes(const CStack * stack, bool addOccupiable, std::vector * attackable = NULL); -/// returns vector of distances to [dest hex number] +/// returns vector of distances to [dest hex number]; WARNING: second argument is ignored std::vector battleGetDistances(const CStack * stack, THex hex = THex::INVALID, THex * predecessors = NULL); + /// returns vector of distances to [dest hex number] + std::vector battleGetDistancesFromHex(const CStack * stack, THex hex = THex::INVALID, THex * predecessors = NULL); + std::set battleGetAttackedHexes(const CStack* attacker, THex destinationTile, THex attackerPos = THex::INVALID); /// returns true if unit with id ID can shoot to dest bool battleCanShoot(const CStack * stack, THex dest);