runner.py

#!/usr/bin/env python

"""
Common run script to apply algorithm to solve 2048 puzzle.
Algorithms are read from subdirectory algorithms and have to supply a function getNextMoves(matrix), which should return a list of next moves.
"""

import os, sys
import logging
import argparse
import importlib
import pprint
import time
import tabulate

import puzzle

def initializeGame(seed = None, showGUI = True):
    return puzzle.GameGrid(is_ai_game=True, useSeed = seed, showGUI = showGUI)

def printSummary(results):
    order = list(results.keys())
    order.sort(key = lambda x: results[x]["Nmoves"], reverse=True)
    order.sort(key = lambda x: results[x]["score"],  reverse=True)
    header = ["Algorithm name", "Score", "moves"]
    lines = []
    for alg in order:
        res = results[alg]
        lines.append([alg, res["score"], res["Nmoves"]])
    print(tabulate.tabulate(lines, header, tablefmt="grid"))

def main(argv):
    parser = argparse.ArgumentParser( description = 'Script that applies a provided algorithm to solve 2048 puzzle.' ,
                                      formatter_class = argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument( "-d", '--debug',       default = False, action='store_true', help = 'Print in debug output' )
    parser.add_argument(       "--gui",         default = False, action="store_true", help = "Shows graphical interface with current status")
    parser.add_argument(       "--ascii",       default = False, action="store_true", help = "Prints current status to terminal")
    parser.add_argument(       "--sleep",       default = 0,  type=float,             help = "Time to wait between moves [s].")
    parser.add_argument( "-s", "--seed",        default = None,                       help = "Set seed to fixed value.")
    parser.add_argument( "-a", "--algorithm",   default = "example",                  help = "Which algorithms to run. multiple algorithms can be split by ','.")
    args = parser.parse_args(argv)

    if args.debug:
        logging.basicConfig(level=logging.DEBUG)
        logging.info("Set log level to DEBUG")
    else:
        logging.basicConfig(level=logging.INFO)
        logging.info("Set log level to INFO")

    results = {}

    for algorithm in args.algorithm.split(","):

        logging.debug("Initializing game")
        gamegrid = initializeGame(seed = args.seed, showGUI = args.gui)

        logging.debug("loading algorithm " + algorithm)
        alg = importlib.import_module("algorithms."+algorithm)

        logging.debug("starting loop")
        done = False
        Nmoves = 0
        NnoMoves = 0
        while (not done):
            logging.debug("Getting next move from algorithm")
            moves = alg.getNextMoves(gamegrid.matrix)
            if not type(moves) == list:
                moves = [moves]
            for move in moves:
                didMove = gamegrid.ai_move(move)
                if not didMove:
                    NnoMoves += 1
                    if NnoMoves > 4:
                        done = True
                        break
                    continue
                
                Nmoves += 1
                NnoMoves = 0

                time.sleep(args.sleep)

                # update game grid
                if args.gui: 
                    gamegrid.update()
                if args.ascii:
                    print( "status: (Score = {})".format(gamegrid.calc_score()))
                    print( tabulate.tabulate(gamegrid.matrix, tablefmt="grid"))

                if gamegrid.game_over():
                    done = True
                    break

        score = gamegrid.calc_score()
        results[algorithm] = {"score": score, "Nmoves": Nmoves}
        print("GAME OVER. Final score: {:8.0f} after {:5.0f} moves (algorithm: {}).".format(score, Nmoves, algorithm))
        if args.gui:
            raw_input("Press Enter to terminate.")

    printSummary(results)

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