526 lines
17 KiB
Python
526 lines
17 KiB
Python
import re
|
|
import codecs
|
|
import sys
|
|
import random
|
|
|
|
import utils
|
|
|
|
# format a list of rows of data into nice columns
|
|
def padrows(l):
|
|
# get length for each field
|
|
lens = []
|
|
for ll in l:
|
|
for i, field in enumerate(ll):
|
|
if i < len(lens):
|
|
lens[i] = max(len(str(field)), lens[i])
|
|
else:
|
|
lens += [len(str(field))]
|
|
# now pad out to that length
|
|
padded = []
|
|
for ll in l:
|
|
padded += ['']
|
|
for i, field in enumerate(ll):
|
|
s = str(field)
|
|
pad = ' ' * (lens[i] - len(s))
|
|
padded[-1] += (s + pad + ' ')
|
|
return padded
|
|
def printrows(l):
|
|
for row in l:
|
|
print row
|
|
|
|
# so this stuff still needs to be cleaned up
|
|
punctuation_chars = r'[+\-*",.:;WUBRGPV/XTQ|\\&^\{\}@ \n=~%\[\]]'
|
|
creature_keywords = [
|
|
# evergreen
|
|
'deathtouch',
|
|
'defender',
|
|
'double strike',
|
|
'first strike',
|
|
'flash',
|
|
'flying',
|
|
'haste',
|
|
'hexproof',
|
|
'indestructible',
|
|
'lifelink',
|
|
'menace',
|
|
'prowess',
|
|
'reach',
|
|
'trample',
|
|
'vigilance',
|
|
# no longer evergreen
|
|
'banding',
|
|
'fear',
|
|
'shroud',
|
|
'intimidate',
|
|
# expert level keywords
|
|
'absorb',
|
|
'amplify',
|
|
'annihilator',
|
|
'battle cry',
|
|
'bolster',
|
|
'bloodthirst',
|
|
'bushido',
|
|
'changeling',
|
|
'convoke',
|
|
'devour',
|
|
'evolve',
|
|
'exalted',
|
|
'extort',
|
|
'fading',
|
|
'flanking',
|
|
'frenzy',
|
|
'graft',
|
|
'haunt',
|
|
'horsemanship',
|
|
'infect',
|
|
'modular',
|
|
#'morph',
|
|
#'ninjutsu',
|
|
'persist',
|
|
'poisonous',
|
|
'provoke',
|
|
#'prowl',
|
|
'rampage',
|
|
'ripple',
|
|
#'scavenge',
|
|
'shadow',
|
|
'soulbond',
|
|
'soulshift',
|
|
'split second',
|
|
'sunburst',
|
|
'undying',
|
|
#'unearth',
|
|
'unleash',
|
|
'vanishing',
|
|
'wither',
|
|
] # there are other keywords out there, these are just easy to detect
|
|
|
|
# data aggregating classes
|
|
class Manacost:
|
|
'''mana cost representation with data'''
|
|
|
|
# hardcoded to be dependent on the symbol structure... ah well
|
|
def get_colors(self):
|
|
colors = ''
|
|
for sym in self.symbols:
|
|
if self.symbols[sym] > 0:
|
|
symcolors = re.sub(r'2|P|S|X', '', sym)
|
|
for symcolor in symcolors:
|
|
if symcolor not in colors:
|
|
colors += symcolor
|
|
# sort so the order is always consistent
|
|
return ''.join(sorted(colors))
|
|
|
|
def check_colors(self, symbolstring):
|
|
for sym in symbolstring:
|
|
if not sym in self.colors:
|
|
return False
|
|
return True
|
|
|
|
def __init__(self, text):
|
|
self.raw = text
|
|
self.cmc = 0
|
|
self.colorless = 0
|
|
self.sequence = []
|
|
self.symbols = {sym : 0 for sym in utils.mana_syms}
|
|
self.allsymbols = {sym : 0 for sym in utils.mana_symall}
|
|
|
|
if text == '':
|
|
self._parsed = True
|
|
self._valid = True
|
|
self.none = True
|
|
self.inner = ''
|
|
|
|
elif not (len(self.raw) >= 2 and self.raw[0] == '{' and self.raw[-1] == '}'):
|
|
self._parsed = False
|
|
self._valid = False
|
|
self.none = False
|
|
|
|
else:
|
|
self._parsed = True
|
|
self._valid = True
|
|
self.none = False
|
|
self.inner = self.raw[1:-1]
|
|
|
|
# structure mirrors the decoding in utils, but we pull out different data here
|
|
idx = 0
|
|
while idx < len(self.inner):
|
|
# taking this branch is an infinite loop if unary_marker is empty
|
|
if (len(utils.mana_unary_marker) > 0 and
|
|
self.inner[idx:idx+len(utils.mana_unary_marker)] == utils.mana_unary_marker):
|
|
idx += len(utils.mana_unary_marker)
|
|
self.sequence += [utils.mana_unary_marker]
|
|
elif self.inner[idx:idx+len(utils.mana_unary_counter)] == utils.mana_unary_counter:
|
|
idx += len(utils.mana_unary_counter)
|
|
self.sequence += [utils.mana_unary_counter]
|
|
self.colorless += 1
|
|
self.cmc += 1
|
|
else:
|
|
old_idx = idx
|
|
for symlen in range(utils.mana_symlen_min, utils.mana_symlen_max + 1):
|
|
encoded_sym = self.inner[idx:idx+symlen]
|
|
if encoded_sym in utils.mana_symall_decode:
|
|
idx += symlen
|
|
# leave the sequence encoded for convenience
|
|
self.sequence += [encoded_sym]
|
|
sym = utils.mana_symall_decode[encoded_sym]
|
|
self.allsymbols[sym] += 1
|
|
if sym in utils.mana_symalt:
|
|
self.symbols[utils.mana_alt(sym)] += 1
|
|
else:
|
|
self.symbols[sym] += 1
|
|
if sym == utils.mana_X:
|
|
self.cmc += 0
|
|
elif utils.mana_2 in sym:
|
|
self.cmc += 2
|
|
else:
|
|
self.cmc += 1
|
|
break
|
|
# otherwise we'll go into an infinite loop if we see a symbol we don't know
|
|
if idx == old_idx:
|
|
idx += 1
|
|
self._valid = False
|
|
|
|
self.colors = self.get_colors()
|
|
|
|
def __str__(self):
|
|
return utils.mana_untranslate(''.join(self.sequence))
|
|
|
|
def format(self, for_forum):
|
|
return utils.mana_untranslate(''.join(self.sequence, for_forum))
|
|
|
|
def reencode(self, randomize = False):
|
|
if randomize:
|
|
# so this won't work very well if mana_unary_marker isn't empty
|
|
return (utils.mana_open_delimiter
|
|
+ ''.join(random.sample(self.sequence, len(self.sequence)))
|
|
+ utils.mana_close_delimiter)
|
|
else:
|
|
return utils.mana_open_delimiter + ''.join(self.sequence) + utils.mana_close_delimiter
|
|
|
|
class Card:
|
|
'''card representation with data'''
|
|
|
|
def __init__(self, text):
|
|
self.raw = text
|
|
self._parsed = True
|
|
self._valid = True
|
|
|
|
if '\n' in self.raw:
|
|
halves = self.raw.split('\n')
|
|
if not len(halves) == 2:
|
|
self._parsed = False
|
|
self._valid = False
|
|
self.fields = halves
|
|
return
|
|
else:
|
|
self.raw = halves[0]
|
|
self.bside = Card(halves[1])
|
|
if not self.bside._valid:
|
|
self._valid = False
|
|
else:
|
|
self.bside = None
|
|
|
|
fields = self.raw.split(utils.fieldsep)
|
|
if not len(fields) >= 10:
|
|
self._parsed = False
|
|
self._valid = False
|
|
self.fields = fields
|
|
else:
|
|
if not fields[1] == '':
|
|
self.name = fields[1]
|
|
else:
|
|
self.name = ''
|
|
self._valid = False
|
|
|
|
if not fields[2] == '':
|
|
self.supertypes = fields[2].split(' ')
|
|
else:
|
|
self.supertypes = []
|
|
|
|
if not fields[3] == '':
|
|
self.types = fields[3].split(' ')
|
|
else:
|
|
self.types = []
|
|
self._valid = False
|
|
|
|
if not fields[4] == '':
|
|
self.loyalty = fields[4]
|
|
try:
|
|
self.loyalty_value = int(self.loyalty)
|
|
except ValueError:
|
|
self.loyalty_value = None
|
|
# strictly speaking, '* where * is something' is valid...
|
|
# self._valid = False
|
|
else:
|
|
self.loyalty = None
|
|
self.loyalty_value = None
|
|
|
|
if not fields[5] == '':
|
|
self.subtypes = fields[5].split(' ')
|
|
if 'creature' in self.types:
|
|
self.creaturetypes = self.subtypes
|
|
else:
|
|
self.creaturetypes = []
|
|
else:
|
|
self.subtypes = []
|
|
self.creaturetypes = []
|
|
|
|
if not fields[6] == '':
|
|
self.pt = fields[6]
|
|
self.power = None
|
|
self.power_value = None
|
|
self.toughness = None
|
|
self.toughness_value = None
|
|
p_t = self.pt.split('/')
|
|
if len(p_t) == 2:
|
|
self.power = p_t[0]
|
|
try:
|
|
self.power_value = int(self.power)
|
|
except ValueError:
|
|
self.power_value = None
|
|
self.toughness = p_t[1]
|
|
try:
|
|
self.toughness_value = int(self.toughness)
|
|
except ValueError:
|
|
self.toughness_value = None
|
|
else:
|
|
self._valid = False
|
|
else:
|
|
self.pt = None
|
|
self.power = None
|
|
self.power_value = None
|
|
self.toughness = None
|
|
self.toughness_value = None
|
|
|
|
# if there's no cost (lands) then cost.none will be True
|
|
self.cost = Manacost(fields[7])
|
|
|
|
if not fields[8] == '':
|
|
self.text = fields[8]
|
|
self.text_lines = self.text.split(utils.newline)
|
|
self.text_words = re.sub(punctuation_chars, ' ', self.text).split()
|
|
self.creature_words = []
|
|
# SUPER HACK
|
|
if 'creature' in self.types:
|
|
for line in self.text_lines:
|
|
orig_line = line
|
|
guess = []
|
|
for keyword in creature_keywords:
|
|
if keyword in line:
|
|
guess += [keyword]
|
|
line = line.replace(keyword, '')
|
|
# yeah, I said it was a hack
|
|
if re.sub(punctuation_chars, ' ', line).split() == [] or 'protect' in line or 'walk' in line or 'sliver creatures' in line or 'you control have' in line:
|
|
for word in guess:
|
|
if word not in self.creature_words:
|
|
self.creature_words += [word]
|
|
# elif len(guess) > 0 and len(line) < 30:
|
|
# print orig_line
|
|
else:
|
|
self.text = ''
|
|
self.text_lines = []
|
|
self.text_words = []
|
|
self.creature_words = []
|
|
|
|
def __str__(self):
|
|
return ''.join([
|
|
utils.fieldsep,
|
|
self.name,
|
|
utils.fieldsep,
|
|
(' ' + utils.dash_marker + ' ').join([' '.join(self.supertypes + self.types),
|
|
' '.join(self.subtypes)]),
|
|
utils.fieldsep,
|
|
str(self.cost.cmc) if self.cost.colors == ''
|
|
else str(self.cost.cmc) + ', ' + self.cost.colors,
|
|
utils.fieldsep,
|
|
])
|
|
|
|
def reencode(self, randomize = False):
|
|
return ''.join([
|
|
utils.fieldsep,
|
|
self.name,
|
|
utils.fieldsep,
|
|
' '.join(self.supertypes),
|
|
utils.fieldsep,
|
|
' '.join(self.types),
|
|
utils.fieldsep,
|
|
self.loyalty if self.loyalty else '',
|
|
utils.fieldsep,
|
|
' '.join(self.subtypes),
|
|
utils.fieldsep,
|
|
self.pt if self.pt else '',
|
|
utils.fieldsep,
|
|
self.cost.reencode(randomize) if not self.cost.none else '',
|
|
utils.fieldsep,
|
|
self.text,
|
|
utils.fieldsep,
|
|
utils.bsidesep + self.bside.reencode(randomize) if self.bside else '',
|
|
])
|
|
|
|
# global card pools
|
|
unparsed_cards = []
|
|
invalid_cards = []
|
|
cards = []
|
|
allcards = []
|
|
|
|
# global indices
|
|
by_name = {}
|
|
by_type = {}
|
|
by_type_inclusive = {}
|
|
by_supertype = {}
|
|
by_supertype_inclusive = {}
|
|
by_subtype = {}
|
|
by_subtype_inclusive = {}
|
|
by_color = {}
|
|
by_color_inclusive = {}
|
|
by_cmc = {}
|
|
by_cost = {}
|
|
by_power = {}
|
|
by_toughness = {}
|
|
by_pt = {}
|
|
by_loyalty = {}
|
|
by_textlines = {}
|
|
by_textlen = {}
|
|
|
|
def inc(d, k, obj):
|
|
if k:
|
|
if k in d:
|
|
d[k] += obj
|
|
else:
|
|
d[k] = obj
|
|
|
|
# build the global indices
|
|
def analyze(cardtexts):
|
|
global unparsed_cards, invalid_cards, cards, allcards
|
|
for cardtext in cardtexts:
|
|
# the empty card is not interesting
|
|
if not cardtext:
|
|
continue
|
|
card = Card(cardtext)
|
|
if card._valid:
|
|
cards += [card]
|
|
allcards += [card]
|
|
elif card._parsed:
|
|
invalid_cards += [card]
|
|
allcards += [card]
|
|
else:
|
|
unparsed_cards += [card]
|
|
|
|
if card._parsed:
|
|
inc(by_name, card.name, [card])
|
|
|
|
inc(by_type, ' '.join(card.types), [card])
|
|
for t in card.types:
|
|
inc(by_type_inclusive, t, [card])
|
|
inc(by_supertype, ' '.join(card.supertypes), [card])
|
|
for t in card.supertypes:
|
|
inc(by_supertype_inclusive, t, [card])
|
|
inc(by_subtype, ' '.join(card.subtypes), [card])
|
|
for t in card.subtypes:
|
|
inc(by_subtype_inclusive, t, [card])
|
|
|
|
if card.cost.colors:
|
|
inc(by_color, card.cost.colors, [card])
|
|
for c in card.cost.colors:
|
|
inc(by_color_inclusive, c, [card])
|
|
else:
|
|
# colorless, still want to include in these tables
|
|
inc(by_color, 'A', [card])
|
|
inc(by_color_inclusive, 'A', [card])
|
|
|
|
inc(by_cmc, card.cost.cmc, [card])
|
|
inc(by_cost, card.cost.reencode(), [card])
|
|
|
|
|
|
inc(by_power, card.power, [card])
|
|
inc(by_toughness, card.toughness, [card])
|
|
inc(by_pt, card.pt, [card])
|
|
|
|
|
|
inc(by_loyalty, card.loyalty, [card])
|
|
|
|
inc(by_textlines, len(card.text_lines), [card])
|
|
inc(by_textlen, len(card.text), [card])
|
|
|
|
# summarize the indices
|
|
def summarize():
|
|
print '===================='
|
|
print str(len(cards)) + ' valid cards, ' + str(len(invalid_cards)) + ' invalid cards.'
|
|
print str(len(allcards)) + ' cards parsed, ' + str(len(unparsed_cards)) + ' failed to parse'
|
|
print '--------------------'
|
|
print str(len(by_name)) + ' unique card names'
|
|
print '--------------------'
|
|
print (str(len(by_color)) + ' represented colors (including colorless as \'A\'), '
|
|
+ str(len(by_color_inclusive)) + ' combinations')
|
|
print 'Breakdown by color:'
|
|
rows = [by_color_inclusive.keys()]
|
|
rows += [[len(by_color_inclusive[k]) for k in rows[0]]]
|
|
printrows(padrows(rows))
|
|
print '--------------------'
|
|
print str(len(by_type_inclusive)) + ' unique card types, ' + str(len(by_type)) + ' combinations'
|
|
print 'Breakdown by type:'
|
|
d = sorted(by_type_inclusive,
|
|
lambda x,y: cmp(len(by_type_inclusive[x]), len(by_type_inclusive[y])),
|
|
reverse = True)
|
|
rows = [[k for k in d[:10]]]
|
|
rows += [[len(by_type_inclusive[k]) for k in rows[0]]]
|
|
printrows(padrows(rows))
|
|
print '--------------------'
|
|
print (str(len(by_subtype_inclusive)) + ' unique subtypes, '
|
|
+ str(len(by_subtype)) + ' combinations')
|
|
print '-- Popular subtypes: --'
|
|
d = sorted(by_subtype_inclusive,
|
|
lambda x,y: cmp(len(by_subtype_inclusive[x]), len(by_subtype_inclusive[y])),
|
|
reverse = True)
|
|
rows = []
|
|
for k in d[0:10]:
|
|
rows += [[k, len(by_subtype_inclusive[k])]]
|
|
printrows(padrows(rows))
|
|
print '-- Top combinations: --'
|
|
d = sorted(by_subtype,
|
|
lambda x,y: cmp(len(by_subtype[x]), len(by_subtype[y])),
|
|
reverse = True)
|
|
rows = []
|
|
for k in d[0:10]:
|
|
rows += [[k, len(by_subtype[k])]]
|
|
printrows(padrows(rows))
|
|
print '--------------------'
|
|
print (str(len(by_supertype_inclusive)) + ' unique supertypes, '
|
|
+ str(len(by_supertype)) + ' combinations')
|
|
print 'Breakdown by supertype:'
|
|
d = sorted(by_supertype_inclusive,
|
|
lambda x,y: cmp(len(by_supertype_inclusive[x]),len(by_supertype_inclusive[y])),
|
|
reverse = True)
|
|
rows = [[k for k in d]]
|
|
rows += [[len(by_supertype_inclusive[k]) for k in rows[0]]]
|
|
printrows(padrows(rows))
|
|
print '===================='
|
|
# TODO: more to come
|
|
|
|
# describe outliers in the indices
|
|
def outliers():
|
|
pass
|
|
|
|
def main(fname, oname = None, verbose = False):
|
|
if verbose:
|
|
print 'Opening encoded card file: ' + fname
|
|
|
|
with open(fname, 'rt') as f:
|
|
text = f.read()
|
|
|
|
cardtexts = text.split(utils.cardsep)
|
|
analyze(cardtexts)
|
|
summarize()
|
|
outliers()
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
if len(sys.argv) == 2:
|
|
main(sys.argv[1])
|
|
elif len(sys.argv) == 3:
|
|
main(sys.argv[1], oname = sys.argv[2])
|
|
else:
|
|
print 'Usage: ' + sys.argv[0] + ' ' + '<encoded file> [output filename]'
|
|
exit(1)
|
|
|