Ctrl: add setAfterAnimationState(StateType)

1 files changed, 5 insertions, 7 deletions

diff --git a/core/src/ch/asynk/tankontank/game/Ctrl.java b/core/src/ch/asynk/tankontank/game/Ctrl.java
index f430846..203bed9 100644
--- a/core/src/ch/asynk/tankontank/game/Ctrl.java
+++ b/core/src/ch/asynk/tankontank/game/Ctrl.java
@@ -185,17 +185,10 @@ public class Ctrl implements Disposable
 
     public void setState(StateType nextState)
     {
-        setState(nextState, battle.getState(player));
-    }
-
-    public void setState(StateType nextState, StateType whenDone)
-    {
         if (nextState == StateType.ABORT)
             nextState = actionAborted();
         else if (nextState == StateType.DONE)
             nextState = actionDone();
-        else if (nextState == StateType.ANIMATION)
-            stateAfterAnimation = whenDone;
 
         this.state.leave(nextState);
 
@@ -255,6 +248,11 @@ public class Ctrl implements Disposable
             state.touchUp();
     }
 
+    public void setAfterAnimationState(StateType after)
+    {
+        stateAfterAnimation = after;
+    }
+
     public void endGame()
     {
         game.setScreen(new OptionsScreen(game));