X-Git-Url: http://git.localhorst.tv/?a=blobdiff_plain;f=src%2Fbattle%2FBattle.h;h=d794ebb3cefee126175a8faee2bd5398e3466214;hb=63a2a96b70a18d08a2714571f79fe7d99612cb9e;hp=cd2894fa753dbba0cbc0f39bc4aea765c524df0b;hpb=e1dab8a680a76f8621e967a693dbf2b481ba8f75;p=l2e.git

diff --git a/src/battle/Battle.h b/src/battle/Battle.h
index cd2894f..d794ebb 100644
--- a/src/battle/Battle.h
+++ b/src/battle/Battle.h
@@ -8,7 +8,9 @@ namespace battle {
 	class TargetSelection;
 }
 namespace common {
+	struct GameState;
 	class Stats;
+	class Upgrade;
 }
 
 #include "Capsule.h"
@@ -47,7 +49,9 @@ public:
 	bool MonsterAlive(int index) const;
 	bool CapsuleAlive() const;
 
+	std::vector<Hero>::iterator HeroesBegin() { return heroes.begin(); }
 	std::vector<Hero>::const_iterator HeroesBegin() const { return heroes.begin(); }
+	std::vector<Hero>::iterator HeroesEnd() { return heroes.end(); }
 	std::vector<Hero>::const_iterator HeroesEnd() const { return heroes.end(); }
 	Hero &HeroAt(int index);
 	const Hero &HeroAt(int index) const;
@@ -109,6 +113,9 @@ public:
 
 	int ExpReward() const { return expReward; }
 	int GoldReward() const { return goldReward; }
+	void ApplyRewards(
+			common::GameState &,
+			std::vector<common::Upgrade> &);
 
 private:
 	const PartyLayout *heroesLayout;