From f7ea3420f0e0995aaed18f89b56f3158ad3137dc Mon Sep 17 00:00:00 2001
From: slapshot5 <slapshot5@cardforge.org>
Date: Fri, 30 Sep 2011 06:42:28 +0000
Subject: [PATCH] a couple tweaks to AF_PutCounter: 1. add UpTo param for "add
 up to X counters to" a card 2. add MaxFromEffect param to specify the maximum
 amount of these counters the card can have from this effect

---
 .../AbilityFactory_Counters.java              | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/main/java/forge/card/abilityFactory/AbilityFactory_Counters.java b/src/main/java/forge/card/abilityFactory/AbilityFactory_Counters.java
index 92f6800a4f5..6e53fda4f6a 100644
--- a/src/main/java/forge/card/abilityFactory/AbilityFactory_Counters.java
+++ b/src/main/java/forge/card/abilityFactory/AbilityFactory_Counters.java
@@ -150,8 +150,11 @@ public class AbilityFactory_Counters {
         Card card = af.getHostCard();
         int amount = AbilityFactory.calculateAmount(af.getHostCard(), params.get("CounterNum"), sa);
 
-        sb.append("Put ").append(amount).append(" ").append(cType.getName())
-                .append(" counter");
+        sb.append("Put ");
+        if (params.containsKey("UpTo")) {
+            sb.append("up to ");
+        }
+        sb.append(amount).append(" ").append(cType.getName()).append(" counter");
         if (amount != 1) { sb.append("s"); }
         sb.append(" on ");
 
@@ -577,6 +580,20 @@ public class AbilityFactory_Counters {
         Card card = af.getHostCard();
         String type = params.get("CounterType");
         int counterAmount = AbilityFactory.calculateAmount(af.getHostCard(), params.get("CounterNum"), sa);
+        int max = params.containsKey("MaxFromEffect") ? Integer.parseInt(params.get("MaxFromEffect")) : -1;
+        
+        if (params.containsKey("UpTo")) {
+            Integer[] integers = new Integer[counterAmount+1];
+            for (int j = 0; j <= counterAmount; j++) {
+                integers[j] = Integer.valueOf(j);
+            }
+            Integer i = GuiUtils.getChoiceOptional("How many counters?", integers);
+            if (null == i) {
+                return;
+            } else {
+                counterAmount = i.intValue();
+            }
+        }
 
         ArrayList<Card> tgtCards;
 
@@ -590,6 +607,9 @@ public class AbilityFactory_Counters {
 
         for (Card tgtCard : tgtCards) {
             if (tgt == null || CardFactoryUtil.canTarget(card, tgtCard)) {
+                if (max != -1) {
+                    counterAmount = max - tgtCard.getCounters(Counters.valueOf(type));
+                }
                 PlayerZone zone = AllZone.getZoneOf(tgtCard);
                 if (zone == null){
                     // Do nothing, token disappeared