freecell_game_loader.py

#!/usr/bin/env python
from __future__ import print_function

from collections import defaultdict
import click


num_to_class = {'A':1, '2':2, '3':3, '4':4, '5':5, '6':6, '7':7, '8':8, '9':9, '10':10, 'J':11, 'Q':12, 'K':13,
                '0':10}  # 0 for abbrev.
suite_to_class = {'s':1, 'd':2, 'h':3, 'c':4}

class_to_num = [None, 'A', '2', '3', '4', '5', '6', '7', '8', '9', '10', 'J', 'Q', 'K']
class_to_suite = [None, 's', 'd', 'h', 'c']


def parse(game_path):  # returns 7 rows of 8 cols
    c, n, s = defaultdict(int), defaultdict(int), defaultdict(int)
    data = []

    with open(game_path, 'r') as fp:
        for row in fp:
            row = row.rstrip()
            if row == '': continue

            items = row.split(' ')
            if len(items) not in (4, 8): continue
            cards_in_row = [(num_to_class[item[:-1]]-1, suite_to_class[item[-1]]-1) for item in items]

            for num, suite in cards_in_row:
                assert 0 <= num <= 12
                assert 0 <= suite <= 3
                c[num*13 + suite] += 1
                n[num] += 1
                s[suite] += 1

            data.append(cards_in_row)

    for card, occ in c.items():
        if occ >= 2:
            print(class_to_num[1+(card / 13)] + class_to_suite[1+(card % 13)], occ)
    assert len(c) == 52 and len(n) == 13 and len(s) == 4

    return data


def transpose(data): # 7x8 -> 8x7
    lines = []
    for c in range(8):
        line = []
        for r in range(7 if c < 4 else 6):
            line.append(data[r][c])
        lines.append(line)
    return lines


def dump_data(data):
    for row in data:
        print(' '.join(class_to_num[1+num] + class_to_suite[1+suite] for num, suite in row))


@click.command()
@click.argument('game-path', type=click.Path(exists=True, readable=True), default='game_8758887.teacher')
def main(game_path):
    data = parse(game_path)

    dump_data(data)
    print()
    dump_data(transpose(data))


if __name__ == '__main__':
    main()