Paw,Unit: set fire() rotate() move(in), reset()

1 files changed, 42 insertions, 4 deletions

diff --git a/core/src/ch/asynk/tankontank/game/Unit.java b/core/src/ch/asynk/tankontank/game/Unit.java
index 0a066e0..b5924f3 100644
--- a/core/src/ch/asynk/tankontank/game/Unit.java
+++ b/core/src/ch/asynk/tankontank/game/Unit.java
@@ -18,6 +18,15 @@ public class Unit extends HeadedPawn
     public boolean hq;
     public boolean ht;
     public Army army;
+    private boolean hasMoved;
+    private boolean hasFired;
+
+    public Unit(TextureAtlas atlas, String pawn, String head)
+    {
+        super(atlas, pawn, head);
+        hasMoved = false;
+        hasFired = false;
+    }
 
     @Override
     public int getMovementPoints()
@@ -63,17 +72,24 @@ public class Unit extends HeadedPawn
     }
 
     @Override
+    public boolean canRotate()
+    {
+        if (ht) return !hasMoved;
+        return (!hasMoved && !hasFired);
+    }
+
+    @Override
     public boolean canMove()
     {
-        // TODO
-        return true;
+        if (ht) return !hasMoved;
+        return (!hasMoved && !hasFired);
     }
 
     @Override
     public boolean canAttack()
     {
-        // TODO
-        return true;
+        if (ht) return !hasFired;
+        return (!hasMoved && !hasFired);
     }
 
     @Override
@@ -82,6 +98,28 @@ public class Unit extends HeadedPawn
         return isEnemy(other);
     }
 
+    public void fire()
+    {
+        hasFired = true;
+    }
+
+    public void rotate()
+    {
+        hasMoved = true;
+    }
+
+    public void move(int cost)
+    {
+        hasMoved = true;
+        if (cost > mp) System.err.println("Movement point exceeded: " + cost + "/" + mp);
+    }
+
+    public void reset()
+    {
+        hasFired = false;
+        hasMoved = false;
+    }
+
     // hard tager
     public Unit(Army army, boolean hq, int range, int defense, int movementPoints, TextureAtlas atlas, String unit, String head)
     {