Player: add unitEntry(Pawn) and revertUnitEntry(Pawn)

1 files changed, 12 insertions, 0 deletions

diff --git a/core/src/ch/asynk/tankontank/engine/Player.java b/core/src/ch/asynk/tankontank/engine/Player.java
index 385a3ae..fa6a923 100644
--- a/core/src/ch/asynk/tankontank/engine/Player.java
+++ b/core/src/ch/asynk/tankontank/engine/Player.java
@@ -65,6 +65,18 @@ public abstract class Player
         reinforcement.add(pawn);
     }
 
+    public void unitEntry(Pawn pawn)
+    {
+        reinforcement.remove(pawn);
+        units.add(pawn);
+    }
+
+    public void revertUnitEntry(Pawn pawn)
+    {
+        units.remove(pawn);
+        reinforcement.add(pawn);
+    }
+
     public void casualty(Pawn pawn)
     {
         units.remove(pawn);