X-Git-Url: http://git.localhorst.tv/?a=blobdiff_plain;f=src%2Fbattle%2FBattleState.cpp;h=b5c91f8ca6b03587bf9605111ed0e6814097e1c9;hb=222167ba3722dc7f47ff7510006bd516e0010a50;hp=c294e1a28feda13da5b9f81be3b9e15aa4aecdc3;hpb=049f304d41e9aa8b5821e298fcb68c1b9a7e97d6;p=l2e.git

diff --git a/src/battle/BattleState.cpp b/src/battle/BattleState.cpp
index c294e1a..b5c91f8 100644
--- a/src/battle/BattleState.cpp
+++ b/src/battle/BattleState.cpp
@@ -9,6 +9,7 @@
 
 #include "PartyLayout.h"
 #include "states/SelectMoveAction.h"
+#include "states/PerformAttacks.h"
 #include "../app/Application.h"
 #include "../app/Input.h"
 #include "../common/Ikari.h"
@@ -180,6 +181,7 @@ void BattleState::LoadInventory() {
 			itemMenu.AddEmptyEntry();
 		}
 	}
+	ClearAllAttacks();
 }
 
 void BattleState::ExitState(Application &ctrl, SDL_Surface *screen) {
@@ -192,12 +194,11 @@ void BattleState::ResumeState(Application &ctrl, SDL_Surface *screen) {
 		ctrl.PopState(); // quit the battle scene
 		return;
 	}
-	// reset attack choices
-	activeHero = -1;
-	for (vector<Hero>::size_type i(0), end(heroes.size()); i < end; ++i) {
-		attackChoices[i] = AttackChoice(this);
+	if (AttackSelectionDone()) {
+		ctrl.PushState(new PerformAttacks(this));
+	} else {
+		ctrl.PushState(new SelectMoveAction(this));
 	}
-	ctrl.PushState(new SelectMoveAction(this));
 }
 
 void BattleState::PauseState(Application &ctrl, SDL_Surface *screen) {
@@ -205,6 +206,14 @@ void BattleState::PauseState(Application &ctrl, SDL_Surface *screen) {
 }
 
 
+void BattleState::ClearAllAttacks() {
+	activeHero = -1;
+	for (vector<Hero>::size_type i(0), end(heroes.size()); i < end; ++i) {
+		attackChoices[i] = AttackChoice(this);
+	}
+}
+
+
 void BattleState::HandleEvents(const Input &input) {
 
 }