⚡ Added condition to check first UMLjointTime (#73)

* ⚡ Added condition to check first UMLjointTime * 🎨 Auto formatted files --------- Co-authored-by: AhmedSobhy01 <[email protected]>
HelanaNady · May 16, 2024 · 701db5b · 701db5b
1 parent 19a3f75
commit 701db5b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/UnitClasses/HealableUnit.cpp b/UnitClasses/HealableUnit.cpp
@@ -2,7 +2,7 @@
 #include "../Game.h"
 
 HealableUnit::HealableUnit(Game* gamePtr, UnitType unitType, double health, int power, int attackCapacity)
-    : Unit(gamePtr, unitType, health, power, attackCapacity), UMLjoinTime(0)
+    : Unit(gamePtr, unitType, health, power, attackCapacity), UMLjoinTime(-1)
 {}
 
 bool HealableUnit::needsHeal() const
@@ -22,6 +22,11 @@ bool HealableUnit::isHealed() const
     return health > initialHealth * 0.2;
 }
 
+bool HealableUnit::hasBeenInUMLbefore() const
+{
+    return UMLjoinTime != -1;
+}
+
 void HealableUnit::receiveHeal(double UHP)
 {
     // Infected units get twice the time to get healed
@@ -33,5 +38,7 @@ void HealableUnit::receiveHeal(double UHP)
 
 void HealableUnit::setUMLjoinTime(int UMLjoinTime)
 {
-    this->UMLjoinTime = UMLjoinTime;
-}
+    // Check if it's the unit first time to join uml 
+    if (!hasBeenInUMLbefore())
+        this->UMLjoinTime = UMLjoinTime;
+}
diff --git a/UnitClasses/HealableUnit.h b/UnitClasses/HealableUnit.h
@@ -17,6 +17,7 @@ class HealableUnit: public Unit
     bool needsHeal() const; // Check if the unit is eligible for healing
     bool hasWaitedForTooLong() const; // Decides whether to kill or heal the unit based on its wait time
     bool isHealed() const; // Checks if the unit's health has increased enough
+    bool hasBeenInUMLbefore() const; // Check if it has been inside uml before or not
 
     void receiveHeal(double); // Increase the health of the unit by "UHP"