From 9492e4212c26dba42688409b6fc76aaba6d032b3 Mon Sep 17 00:00:00 2001 From: Northmoc Date: Wed, 17 Aug 2022 15:24:10 -0400 Subject: [PATCH] AiCostDecision.visit for CostChooseColor --- forge-ai/src/main/java/forge/ai/AiCostDecision.java | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/forge-ai/src/main/java/forge/ai/AiCostDecision.java b/forge-ai/src/main/java/forge/ai/AiCostDecision.java index 16b01575594..210209a702e 100644 --- a/forge-ai/src/main/java/forge/ai/AiCostDecision.java +++ b/forge-ai/src/main/java/forge/ai/AiCostDecision.java @@ -2,10 +2,12 @@ package forge.ai; import static forge.ai.ComputerUtilCard.getBestCreatureAI; +import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.Map; +import forge.card.MagicColor; import forge.game.cost.*; import org.apache.commons.lang3.ObjectUtils; @@ -55,6 +57,14 @@ public class AiCostDecision extends CostDecisionMakerBase { return PaymentDecision.number(c); } + @Override + public PaymentDecision visit(CostChooseColor cost) { + int c = cost.getAbilityAmount(ability); + List choices = player.getController().chooseColors("Color", ability, c, c, + new ArrayList<>(MagicColor.Constant.ONLY_COLORS)); + return PaymentDecision.colors(choices); + } + @Override public PaymentDecision visit(CostChooseCreatureType cost) { String choice = player.getController().chooseSomeType("Creature", ability, CardType.getAllCreatureTypes(),