Map: activatedPawns is a Board.PawnCollection

2 files changed, 5 insertions, 3 deletions

diff --git a/core/src/ch/asynk/tankontank/game/Hex.java b/core/src/ch/asynk/tankontank/game/Hex.java
index b357e56..45d3d43 100644
--- a/core/src/ch/asynk/tankontank/game/Hex.java
+++ b/core/src/ch/asynk/tankontank/game/Hex.java
@@ -5,6 +5,7 @@ import java.util.Iterator;
 
 import com.badlogic.gdx.graphics.g2d.TextureAtlas;
 
+import ch.asynk.tankontank.engine.Board;
 import ch.asynk.tankontank.engine.Pawn;
 import ch.asynk.tankontank.engine.Tile;
 import ch.asynk.tankontank.engine.Orientation;
@@ -109,7 +110,7 @@ public class Hex extends Tile
     }
 
     @Override
-    public int defenseFor(Pawn target, List<Pawn> foes)
+    public int defenseFor(Pawn target, Board.PawnCollection foes)
     {
         Unit u = (Unit) target;
         boolean terrainBonus = true;
diff --git a/core/src/ch/asynk/tankontank/game/Map.java b/core/src/ch/asynk/tankontank/game/Map.java
index 4f876b0..ce587a1 100644
--- a/core/src/ch/asynk/tankontank/game/Map.java
+++ b/core/src/ch/asynk/tankontank/game/Map.java
@@ -1,6 +1,5 @@
 package ch.asynk.tankontank.game;
 
-import java.util.ArrayList;
 import java.util.Iterator;
 
 import com.badlogic.gdx.assets.AssetManager;
@@ -28,7 +27,7 @@ public abstract class Map extends Board
     public final Board.PawnCollection moveablePawns;
     public final Board.PawnCollection possibleTargets;
     public final Board.PawnCollection attackAssists;
-    public final ArrayList<Pawn> activatedPawns = new ArrayList<Pawn>(7);
+    public final Board.PawnCollection activatedPawns;
 
     private final SpriteAnimation explosion;
     private final SpriteAnimation explosions;
@@ -49,6 +48,8 @@ public abstract class Map extends Board
 
         possibleTargets = new PawnSet(this, 10);
         attackAssists = new PawnSet(this, 6);
+
+        activatedPawns = new PawnSet(this, 7);
     }
 
     @Override