#!/usr/bin/env python3
import sys
import pickle

from pprint import pprint

from copycat import Problem
from copycat.statistics import cross_table

def compare_sets():
    pass

def main(args):
    branchProblemSets = dict()
    problemSets = []
    for filename in args:
        with open(filename, 'rb') as infile:
            pSet = pickle.load(infile)
            branchProblemSets[filename] = pSet
            problemSets.append((filename, pSet))
    crossTable = cross_table(problemSets)
    key_sorted_items = lambda d : sorted(d.items(), key=lambda t:t[0])

    tableItems = key_sorted_items(crossTable)
    assert len(tableItems) > 0, 'Empty table'

    headKey, headSubDict = tableItems[0]
    # Create table and add headers
    table = [['problems x variants']]
    for subkey, subsubdict in key_sorted_items(headSubDict):
        for subsubkey, result in key_sorted_items(subsubdict):
            table[-1].append('{} x {} for {} x {}'.format(*subkey, *subsubkey))

    # Add test results to table
    for key, subdict in tableItems:
        table.append([])
        problem = '{}:{}::{}:_'.format(*key)
        table[-1].append(problem)
        for subkey, subsubdict in key_sorted_items(subdict):
            for subsubkey, result in key_sorted_items(subsubdict):
                table[-1].append(str(result))

    # A simple transpose
    table = list(map(list, zip(*table)))

    with open('output/cross_compare.csv', 'w') as outfile:
        outfile.write('\n'.join(','.join(row) for row in table) + '\n')
    return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))