From ba7371fa91a3af11d36f7f332e0823691a311414 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9my=20Zurcher?= <jeremy@asynk.ch>
Date: Tue, 18 Nov 2014 16:24:53 +0100
Subject: Unit: add promote(), degrade()

---
 core/src/ch/asynk/tankontank/game/Unit.java | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/core/src/ch/asynk/tankontank/game/Unit.java b/core/src/ch/asynk/tankontank/game/Unit.java
index e9c2af4..d5576be 100644
--- a/core/src/ch/asynk/tankontank/game/Unit.java
+++ b/core/src/ch/asynk/tankontank/game/Unit.java
@@ -192,6 +192,34 @@ public class Unit extends HeadedPawn
         return false;
     }
 
+    public void promote()
+    {
+        if (isA(UnitId.GE_PANZER_IV))
+            id = UnitId.GE_PANZER_IV_HQ;
+        else if (isA(UnitId.US_PERSHING))
+            id = UnitId.US_PERSHING_HQ;
+        else if (isA(UnitId.US_SHERMAN))
+            id = UnitId.US_SHERMAN_HQ;
+        else
+            return;
+
+        type = UnitType.HARD_TARGET_HQ;
+    }
+
+    public void degrade()
+    {
+        if (isA(UnitId.GE_PANZER_IV_HQ))
+            id = UnitId.GE_PANZER_IV;
+        else if (isA(UnitId.US_PERSHING_HQ))
+            id = UnitId.US_PERSHING;
+        else if (isA(UnitId.US_SHERMAN_HQ))
+            id = UnitId.US_SHERMAN;
+        else
+            return;
+
+        type = UnitType.HARD_TARGET;
+    }
+
     @Override
     public boolean isHardTarget()
     {
-- 
cgit v1.1-2-g2b99