Make EMP deactivate shield completely

Warzone2100 · Oct 28, 2024 · f1d501c · f1d501c
1 parent 935263b
commit f1d501c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/src/combat.cpp b/src/combat.cpp
@@ -429,6 +429,13 @@ int32_t objDamage(BASE_OBJECT *psObj, unsigned damage, unsigned originalhp, WEAP
 	// EMP weapon radius should not do actual damage
 	if (weaponSubClass == WSC_EMP && empRadiusHit)
 	{
+		if (psObj->type == OBJ_DROID)
+		{
+			DROID *psDroid = castDroid(psObj);
+
+			// EMP weapons kills droid shields completely
+			psDroid->shieldPoints = 0;
+		}
 		return 0;
 	}
 
@@ -514,7 +521,6 @@ int32_t objDamage(BASE_OBJECT *psObj, unsigned damage, unsigned originalhp, WEAP
 				psDroid->shieldPoints = 0;
 				// shields are interrupted, wait for a while until regeneration starts again
 				psDroid->shieldInterruptRegenTime = psDroid->time;
-				psDroid->shieldRegenTime = psDroid->time;
 			}
 
 			Vector3i dv;

diff --git a/src/droid.cpp b/src/droid.cpp
@@ -969,19 +969,18 @@ void droidUpdateShields(DROID *psDroid)
 		}
 		else
 		{
-			if (gameTime - psDroid->shieldInterruptRegenTime >= droidCalculateShieldInterruptRegenTime(psDroid))
+			if (!((psDroid->lastHitWeapon == WSC_EMP) && ((gameTime - psDroid->timeLastHit) < EMP_DISABLE_TIME)) &&
+				gameTime - psDroid->shieldInterruptRegenTime > droidCalculateShieldInterruptRegenTime(psDroid) &&
+				gameTime - psDroid->shieldRegenTime > droidCalculateShieldRegenTime(psDroid))
 			{
-				if (gameTime - psDroid->shieldRegenTime >= droidCalculateShieldRegenTime(psDroid))
+				for (uint32_t i = 0; i < DROID_SHIELD_POINTS_STEP; i++)
 				{
-					for (uint32_t i = 0; i < DROID_SHIELD_POINTS_STEP; i++)
+					if (psDroid->shieldPoints < droidGetMaxShieldPoints(psDroid))
 					{
-						if (psDroid->shieldPoints < droidGetMaxShieldPoints(psDroid))
-						{
-							psDroid->shieldPoints += 1;
-						}
+						psDroid->shieldPoints += 1;
 					}
-					psDroid->shieldRegenTime = gameTime;
 				}
+				psDroid->shieldRegenTime = gameTime;
 			}
 		}
 	}