Experimental: allow zombifying hostiles (by player)

MeridianOXC · Feb 2, 2024 · 5f21290 · 5f21290
1 parent c923f30
commit 5f21290
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/Savegame/BattleUnit.cpp b/src/Savegame/BattleUnit.cpp
@@ -1663,7 +1663,22 @@ int BattleUnit::damage(Position relative, int damage, const RuleDamageType *type
 		&& !specialDamageTransform->getZombieUnit(this).empty()
 		&& getArmor()->getZombiImmune() == false)
 	{
-		specialDamageTransformChance = getOriginalFaction() != FACTION_HOSTILE ? specialDamageTransform->getZombieUnitChance() : 0;
+		if (attack.attacker)
+		{
+			if (getOriginalFaction() == FACTION_HOSTILE && attack.attacker->getOriginalFaction() == FACTION_HOSTILE)
+			{
+				// (mind-controlled) chryssalid on snakeman action still not allowed
+				specialDamageTransformChance = 0;
+			}
+			else
+			{
+				specialDamageTransformChance = specialDamageTransform->getZombieUnitChance();
+			}
+		}
+		else
+		{
+			specialDamageTransformChance = getOriginalFaction() != FACTION_HOSTILE ? specialDamageTransform->getZombieUnitChance() : 0;
+		}
 	}
 	else
 	{