import re
import sys
from mtg import *
import database

def graph(data, title, hideEmpty=True):
    categories = sorted(data.keys())
    cjust = sorted([len(str(k)) for k in data.keys()]).pop()
    vjust = sorted([len(str(v)) for v in data.values()]).pop()
    max = sorted(data.values()).pop()
    print title
    print '-' * 80
    for category in categories:
        if hideEmpty and not data[category]:
            continue
        heading = str(category).ljust(cjust)
        value = str(data[category]).rjust(vjust)
        bar = '*' * int(data[category] * (80 - cjust - vjust - 6) / float(max))
        print '{0} [{1}] | {2}'.format(heading, value, bar)
    if hideEmpty and not max:
        print ' *** Nothing to display ***'

def mana_curve(deck):
    costs = [c.cost.converted() for c in deck if c.cost.converted() > 0]
    data = {}
    for i in range(1, sorted(costs).pop() + 1):
        data[i] = 0
    for c in costs:
        data[c] += 1
    return data
def types(deck):
    return {
        'Lands': len([c for c in deck if 'land' in c.type.split(' ')]),
        'Creatures': len([c for c in deck if 'creature' in c.type.split(' ')]),
        'Artifacts': len([c for c in deck if 'artifact' in c.type.split(' ')]),
        'Enchantments': len([c for c in deck if 'enchantment' in c.type.split(' ')]),
        'Sorceries': len([c for c in deck if 'sorcery' in c.type.split(' ')]),
        'Instants': len([c for c in deck if 'instant' in c.type.split(' ')]),
    }
def colors(deck):
    data = {}
    for color in Mana.types.values():
        data[color] = 0
    for card in deck:
        if 'land' in card.type.split(' '):
            if 'swamp' in card.attributes: data[Mana.types['B']] += 1
            if 'island' in card.attributes: data[Mana.types['U']] += 1
            if 'mountain' in card.attributes: data[Mana.types['R']] += 1
            if 'forest' in card.attributes: data[Mana.types['G']] += 1
            if 'plains' in card.attributes: data[Mana.types['W']] += 1
        else:
            for color, value in card.cost.mana.mana.iteritems():
                data[Mana.types[color]] += 1 if value > 0 else 0
    return data

db = database.TextDB('db.txt')

if len(sys.argv) < 2:
    print 'No decks specified.'
    sys.exit(1)
for filename in sys.argv[1:]:
    print 'Loading deck in "{0}"...'.format(filename)
    deck = Deck()
    with open(filename, 'r') as f:
        for line in f:
            line = line.strip()
            if not re.match('^\d+ .+$', line):
                continue
            line = line.split(' ')
            count = int(line.pop(0))
            name = ' '.join(line)
            card = db.getCard(name)
            if not card:
                print 'Invalid card: ', name
                continue
            deck.extend(card * count)
    
    print 'Loaded', len(deck), 'cards.'
    graph(mana_curve(deck), '\nMana Curve', hideEmpty=False)
    graph(types(deck), '\nCard Types')
    graph(colors(deck), '\nColors')