BattleCommon: add addReinforcement(Player, EntryPoint, UnitId, boolean) -> setAce()

1 files changed, 6 insertions, 0 deletions

diff --git a/core/src/ch/asynk/tankontank/game/battles/BattleCommon.java b/core/src/ch/asynk/tankontank/game/battles/BattleCommon.java
index eecad83..d7e6b23 100644
--- a/core/src/ch/asynk/tankontank/game/battles/BattleCommon.java
+++ b/core/src/ch/asynk/tankontank/game/battles/BattleCommon.java
@@ -55,7 +55,13 @@ public abstract class BattleCommon implements Battle
 
     public void addReinforcement(Player player, EntryPoint entryPoint, UnitId unitId)
     {
+        addReinforcement(player, entryPoint, unitId, false);
+    }
+
+    public void addReinforcement(Player player, EntryPoint entryPoint, UnitId unitId, boolean ace)
+    {
         Unit unit = factory.getUnit(unitId);
+        unit.setAce(ace);
         player.addReinforcement(unit);
         pawnEntry.put(unit, entryPoint);
     }