- Expanded the predictThreatenedObjects AI function.

forge-ai/src/main/java/forge/ai/ComputerUtil.java

@@ -1444,6 +1444,21 @@ public class ComputerUtil {
                 }
             }
         }
+        //GainControl
+        else if (threatApi == ApiType.GainControl
+                && (saviourApi == ApiType.ChangeZone || saviourApi == ApiType.Pump || saviourApi == null)) {
+            for (final Object o : objects) {
+                if (o instanceof Card) {
+                    final Card c = (Card) o;
+                    // give Shroud to targeted creatures
+                    if (saviourApi == ApiType.Pump && tgt == null && saviour.hasParam("KW")
+                            && (saviour.getParam("KW").endsWith("Shroud") || saviour.getParam("KW").endsWith("Hexproof"))) {
+                        continue;
+                    }
+                    threatened.add(c);
+                }
+            }
+        }
     
         Iterables.addAll(threatened, ComputerUtil.predictThreatenedObjects(aiPlayer, saviour, topStack.getSubAbility()));
         return threatened;

forge-ai/src/main/java/forge/ai/ability/DamageDealAi.java

@@ -20,10 +20,8 @@ import forge.game.spellability.TargetChoices;
 import forge.game.spellability.TargetRestrictions;
 import forge.game.zone.ZoneType;
 import forge.util.Aggregates;
-import forge.util.MyRandom;
 
 import java.util.List;
-import java.util.Random;
 
 public class DamageDealAi extends DamageAiBase {
     @Override
@@ -89,13 +87,9 @@ public class DamageDealAi extends DamageAiBase {
             return false;
         }
 
-        if (sa.isAbility()) {
+        if (ComputerUtil.preventRunAwayActivations(sa)) {
-            final Random r = MyRandom.getRandom(); // prevent run-away
-                                                   // activations
-            if (r.nextFloat() > Math.pow(.9, sa.getActivationsThisTurn())) {
         	return false;
         }
-        }
 
         if (!this.damageTargetAI(ai, sa, dmg)) {
             return false;