[draft] AI picks based on card ratings. Added debug logging for a while.

Mage.Server.Plugins/Mage.Player.AI/src/main/java/mage/player/ai/ComputerPlayer.java

@@ -90,6 +90,7 @@ import mage.game.Game;
 import mage.game.Table;
 import mage.game.combat.CombatGroup;
 import mage.game.permanent.Permanent;
+import mage.player.ai.utils.RateCard;
 import mage.players.Player;
 import mage.players.PlayerImpl;
 import mage.target.Target;
@@ -789,8 +790,24 @@ public class ComputerPlayer<T extends ComputerPlayer<T>> extends PlayerImpl<T> i
 
 	@Override
 	public void pickCard(List<Card> cards, Deck deck, Draft draft) {
-		//TODO: improve this
-		draft.addPick(playerId, cards.get(0).getId());
+		if (cards.size() == 0) {
+			throw new IllegalArgumentException("No cards to pick from.");
+		}
+		try {
+			Card bestCard = null;
+			int maxScore = 0;
+			for (Card card : cards) {
+				int score = RateCard.rateCard(card, null);
+				if (bestCard == null || score > maxScore) {
+					maxScore = score;
+					bestCard = card;
+				}
+			}
+			System.out.println("[DEBUG] AI picked: " + bestCard.getName() + ", score=" + maxScore);
+			draft.addPick(playerId, bestCard.getId());
+		} catch (Exception e) {
+			draft.addPick(playerId, cards.get(0).getId());
+		}
 	}
 
 	protected Attackers getPotentialAttackers(Game game) {

Mage.Server.Plugins/Mage.Player.AI/src/main/java/mage/player/ai/utils/RateCard.java

@@ -0,0 +1,169 @@
+package mage.player.ai.utils;
+
+import mage.Constants;
+import mage.cards.Card;
+
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Scanner;
+
+/**
+ * Class responsible for reading ratings from resources and rating gived cards.
+ * Based on card relative ratings from resources and card parameters.
+ *
+ * @author nantuko
+ */
+public class RateCard {
+
+	private static Map<String, Integer> ratings;
+	private static Integer min = Integer.MAX_VALUE, max = 0;
+
+	/**
+	 * Rating that is given for new cards.
+	 * Ratings are in [1,10] range, so setting it high will make new cards appear more often.
+	 */
+	private static final int DEFAULT_NOT_RATED_CARD_RATING = 4;
+
+	/**
+	 * Hide constructor.
+	 */
+	private RateCard() {
+	}
+
+	/**
+	 * Get absolute score of the card.
+	 * Depends on type, manacost, rating.
+	 *
+	 * @param card
+	 * @return
+	 */
+	public static int rateCard(Card card, List<Constants.ColoredManaSymbol> allowedColors) {
+		int type = 0;
+		if (card.getCardType().contains(Constants.CardType.CREATURE)) {
+			type = 10;
+		} else if (card.getSubtype().contains("Equipment")) {
+			type = 8;
+		} else if (card.getSubtype().contains("Aura")) {
+			type = 5;
+		} else if (card.getCardType().contains(Constants.CardType.INSTANT)) {
+			type = 7;
+		} else {
+			type = 6;
+		}
+		int score = 3 * getCardRating(card) + type + getManaCostScore(card, allowedColors);
+		return score;
+	}
+
+	/**
+	 * Return rating of the card.
+	 *
+	 * @param card Card to rate.
+	 * @return Rating number from [1;10].
+	 */
+	private static int getCardRating(Card card) {
+		if (ratings == null) {
+			readRatings();
+		}
+		if (ratings.containsKey(card.getName())) {
+			int r = ratings.get(card.getName());
+			// normalize to [1..10]
+			float f = 10.0f * (r - min) / (max - min);
+			return (int) Math.round(f);
+		}
+		return DEFAULT_NOT_RATED_CARD_RATING;
+	}
+
+	/**
+	 * Reads ratings from resources.
+	 */
+	private synchronized static void readRatings() {
+		if (ratings == null) {
+			ratings = new HashMap<String, Integer>();
+			String filename = "/ratings.txt";
+			try {
+				InputStream is = RateCard.class.getResourceAsStream(filename);
+				Scanner scanner = new Scanner(is);
+				while (scanner.hasNextLine()) {
+					String line = scanner.nextLine();
+					String[] s = line.split(":");
+					if (s.length == 2) {
+						Integer rating = Integer.parseInt(s[0].trim());
+						String name = s[1].trim();
+						if (rating > max) {
+							max = rating;
+						}
+						if (rating < min) {
+							min = rating;
+						}
+						ratings.put(name, rating);
+					}
+				}
+			} catch (Exception e) {
+				e.printStackTrace();
+				ratings.clear(); // no rating available on exception
+			}
+		}
+	}
+
+	private static final int SINGLE_PENALTY[] = {0, 1, 1, 3, 6, 9};
+
+	/**
+	 * Get manacost score.
+	 * Depends on chosen colors. Returns negative score for those cards that doesn't fit allowed colors.
+	 * If allowed colors are not chosen, then score based on converted cost is returned with penalty for heavy colored cards.
+	 *
+	 *
+	 * @param card
+	 * @param allowedColors Can be null.
+	 * @return
+	 */
+	private static int getManaCostScore(Card card, List<Constants.ColoredManaSymbol> allowedColors) {
+		int converted = card.getManaCost().convertedManaCost();
+		if (allowedColors == null) {
+			int colorPenalty = 0;
+			for (String symbol : card.getManaCost().getSymbols()) {
+				if (isColoredMana(symbol)) {
+				 	colorPenalty++;
+				}
+			}
+			return 2 * (converted - colorPenalty + 1);
+		}
+		final Map<String, Integer> singleCount = new HashMap<String, Integer>();
+		int maxSingleCount = 0;
+		for (String symbol : card.getManaCost().getSymbols()) {
+			int count = 0;
+			symbol = symbol.replace("{", "").replace("}", "");
+			if (isColoredMana(symbol)) {
+				for (Constants.ColoredManaSymbol allowed : allowedColors) {
+					if (allowed.toString().equals(symbol)) {
+						count++;
+					}
+				}
+				if (count == 0) {
+					return -30;
+				}
+				Integer typeCount = singleCount.get(symbol);
+				if (typeCount == null) {
+					typeCount = new Integer(0);
+				}
+				typeCount += 1;
+				singleCount.put(symbol, typeCount);
+				maxSingleCount = Math.max(maxSingleCount, typeCount);
+			}
+		}
+		return 2 * converted + 3 * (10 - SINGLE_PENALTY[maxSingleCount]/*-DOUBLE_PENALTY[doubleCount]*/);
+	}
+
+	/**
+	 * Determines whether mana symbol is color.
+	 *
+ 	 * @param symbol
+	 * @return
+	 */
+	private static boolean isColoredMana(String symbol) {
+        return symbol.equals("W") || symbol.equals("G") || symbol.equals("U") || symbol.equals("B") || symbol.equals("R");
+    }
+
+}