moved upgrade process to battle class

[l2e.git] / src / battle / Battle.cpp

diff --git a/src/battle/Battle.cpp b/src/battle/Battle.cpp
index 7bfa940461e12cff44f90400d94327b58fc00e00..cc481e3c993eb19033e3f3e6e5ccb3fbd8e3e16a 100644 (file)
--- a/src/battle/Battle.cpp
+++ b/src/battle/Battle.cpp
@@ -3,12 +3,17 @@
 #include "AttackChoice.h"
 #include "PartyLayout.h"
 #include "TargetSelection.h"
+#include "../common/GameState.h"
 #include "../common/Stats.h"
+#include "../common/Upgrade.h"
 
 #include <cassert>
 #include <stdexcept>
 
+using common::GameState;
 using common::Stats;
+using common::Upgrade;
+using std::vector;
 
 
 namespace battle {
@@ -382,4 +387,19 @@ bool Battle::Defeat() const {
        return true;
 }
 
+
+void Battle::ApplyRewards(
+               GameState &state,
+               vector<Upgrade> &info) {
+       for (vector<Hero>::iterator i(HeroesBegin()), end(HeroesEnd());
+                       i != end; ++i) {
+               if (i->Health() <= 0) continue;
+               i->Master().AddExperience(expReward, info);
+       }
+       if (capsule.Health() > 0) {
+               capsule.Master().AddExperience(expReward, info);
+       }
+       state.money += goldReward;
+}
+
 }