From ac7a1ccae0ce038c6d92feaba622b72f1a339226 Mon Sep 17 00:00:00 2001
From: Jérémy Zurcher <jeremy@asynk.ch>
Date: Tue, 14 Oct 2014 12:34:35 +0200
Subject: Pawn,Unit: add boolean isHqOf(Pawn), Player add boolean
 canPromote(Pawn)

---
 core/src/ch/asynk/tankontank/engine/Pawn.java | 1 +
 core/src/ch/asynk/tankontank/game/Player.java | 8 ++++++++
 core/src/ch/asynk/tankontank/game/Unit.java   | 9 +++++++++
 3 files changed, 18 insertions(+)

(limited to 'core/src/ch/asynk/tankontank')

diff --git a/core/src/ch/asynk/tankontank/engine/Pawn.java b/core/src/ch/asynk/tankontank/engine/Pawn.java
index ef91544b..83cd642e 100644
--- a/core/src/ch/asynk/tankontank/engine/Pawn.java
+++ b/core/src/ch/asynk/tankontank/engine/Pawn.java
@@ -46,6 +46,7 @@ public abstract class Pawn implements Drawable, Disposable
     public abstract int getAttackRangeFrom(Tile tile);
 
     public abstract boolean isHq();
+    public abstract boolean isHqOf(Pawn other);
     public abstract boolean isUnit();
     public abstract boolean isEnemy(Pawn other);
 
diff --git a/core/src/ch/asynk/tankontank/game/Player.java b/core/src/ch/asynk/tankontank/game/Player.java
index ca8716cd..56463119 100644
--- a/core/src/ch/asynk/tankontank/game/Player.java
+++ b/core/src/ch/asynk/tankontank/game/Player.java
@@ -110,6 +110,14 @@ public class Player implements Drawable, Disposable
         return ((Unit) pawn).isEnemy(army);
     }
 
+    public boolean canPromote(Pawn pawn)
+    {
+        if (pawn.isHq()) return false;
+        for (Pawn p: casualties)
+            if (p.isHqOf(pawn)) return true;
+        return false;
+    }
+
     public void setPosition(float x, float y)
     {
         flag.setPosition(x, y);
diff --git a/core/src/ch/asynk/tankontank/game/Unit.java b/core/src/ch/asynk/tankontank/game/Unit.java
index a20891b8..bae5fd8d 100644
--- a/core/src/ch/asynk/tankontank/game/Unit.java
+++ b/core/src/ch/asynk/tankontank/game/Unit.java
@@ -128,6 +128,15 @@ public class Unit extends HeadedPawn
         return (type == UnitType.HARD_TARGET_HQ);
     }
 
+    @Override
+    public boolean isHqOf(Pawn other)
+    {
+        if ((id == UnitId.GE_PANZER_IV_HQ) && (((Unit)other).id == UnitId.GE_PANZER_IV)) return true;
+        if ((id == UnitId.US_PERSHING_HQ) && (((Unit)other).id == UnitId.US_PERSHING)) return true;
+        if ((id == UnitId.US_SHERMAN_HQ) && (((Unit)other).id == UnitId.US_SHERMAN)) return true;
+        return false;
+    }
+
     @Override
     public boolean isUnit()
     {
-- 
cgit v1.2.3