Added some statistics functionality, TDD

1 files changed, 80 insertions, 0 deletions

diff --git a/src/eu/equalparts/cardbase/decks/Statistics.java b/src/eu/equalparts/cardbase/decks/Statistics.java
new file mode 100644
index 0000000..9efc82a
--- /dev/null
+++ b/src/eu/equalparts/cardbase/decks/Statistics.java
@@ -0,0 +1,80 @@
+package eu.equalparts.cardbase.decks;
+
+import eu.equalparts.cardbase.cards.Card;
+
+public final class Statistics {
+
+	private Statistics() {}
+	
+	
+	
+	public static double calculatePercentage(StandaloneDeck deck, String type) {
+		double allCardsByType = count(deck, type);
+		double allCards = count(deck);
+		return allCardsByType / allCards;
+	}
+	
+	public static int count(StandaloneDeck deck, String type) {
+		int count = type.contains("Land") ? countBasicLands(deck) : 0;
+		for (Card card : deck.cards) {
+			if (card.type != null &&
+					card.type.contains(type)) {
+				count += card.count;
+			}
+		}
+		return count;
+	}
+
+	public static int count(StandaloneDeck deck) {
+		int totalCards = countBasicLands(deck);
+		
+		for (Card card : deck.cards) {
+			totalCards += card.count;
+		}
+		
+		return totalCards;
+	}
+	
+	private static int countBasicLands(StandaloneDeck deck) {
+		return deck.plains + 
+				deck.islands + 
+				deck.swamps + 
+				deck.mountains + 
+				deck.forests;
+	}
+	
+	public static int[] computeDistribution(StandaloneDeck deck, String type) {
+		int arraySize = 0;
+		for (Card card : deck.cards) {
+			if (card.type != null && card.type.contains(type))
+				if (card.cmc != null && card.cmc >= arraySize)
+					arraySize = card.cmc + 1;
+		}
+
+		int[] costs = new int[arraySize];
+		for (Card card : deck.cards) {
+			if (card.type != null && card.type.contains(type))
+				if (card.cmc != null)
+					costs[card.cmc] += card.count; 
+		}
+		
+		return costs;
+	}
+
+	public static int[] computeDistribution(StandaloneDeck deck) {
+		int arraySize = 0;
+		for (Card card : deck.cards) {
+			if (card.cmc != null && card.cmc >= arraySize)
+				arraySize = card.cmc + 1;
+		}
+		
+		int[] costs = new int[arraySize];
+		for (Card card : deck.cards) {
+			if (card.cmc != null)
+				costs[card.cmc] += card.count; 
+		}
+		
+		return costs;
+	}
+	
+}