GameCtrl: add animationCount and setAnnimationCount(int)

1 files changed, 13 insertions, 3 deletions

diff --git a/core/src/ch/asynk/tankontank/game/GameCtrl.java b/core/src/ch/asynk/tankontank/game/GameCtrl.java
index 6202ef4..149596c 100644
--- a/core/src/ch/asynk/tankontank/game/GameCtrl.java
+++ b/core/src/ch/asynk/tankontank/game/GameCtrl.java
@@ -44,6 +44,8 @@ public class GameCtrl implements Disposable
     private GameState rotateState;
     private GameState animationState;
 
+    private int animationCount = 0;
+
     private GameState state;
 
     public GameCtrl(final TankOnTank game)
@@ -85,11 +87,19 @@ public class GameCtrl implements Disposable
         return (state != selectState);
     }
 
+    public void setAnimationCount(int count)
+    {
+        animationCount = count;
+    }
+
     public void animationDone()
     {
-        GameState.State next = state.getNextState();
-        state.setNextState(GameState.State.SELECT);
-        setState(next, (next == GameState.State.SELECT));
+        animationCount -= 1;
+        if (animationCount == 0) {
+            GameState.State next = state.getNextState();
+            state.setNextState(GameState.State.SELECT);
+            setState(next, (next == GameState.State.SELECT));
+        }
     }
 
     public void setState(GameState.State state)