Map: add boolean toggleAttackAssist(Pawn)

1 files changed, 13 insertions, 0 deletions

diff --git a/core/src/ch/asynk/tankontank/game/Map.java b/core/src/ch/asynk/tankontank/game/Map.java
index 5c50ea5..e5764f0 100644
--- a/core/src/ch/asynk/tankontank/game/Map.java
+++ b/core/src/ch/asynk/tankontank/game/Map.java
@@ -212,6 +212,19 @@ public abstract class Map extends Board
         return attackAssists.size();
     }
 
+    public boolean toggleAttackAssist(Pawn pawn)
+    {
+        if (activablePawns.contains(pawn)) {
+            activablePawns.remove(pawn);
+            activatedPawns.add(pawn);
+            return true;
+        } else {
+            activatedPawns.remove(pawn);
+            activablePawns.add(pawn);
+            return false;
+        }
+    }
+
     public void buildAndShowMovesAndAssits(Pawn pawn, GridPoint2 hex)
     {
         showPossibleMoves(false);