play.py

import os
import sys
from argparse import ArgumentParser

from agent import RandomAgent
from agent import HumanAgent
from driver import GameDriver
from utils import InvalidMapError

MAP_TYPES = ['ascii', 'emoji']


def main(args):
    parser = ArgumentParser(description='')

    parser.add_argument('--height', type=int, required=True,
                        help='Heigth of the map')
    parser.add_argument('--width', type=int, required=True,
                        help='Width of the map')
    parser.add_argument('--num-powerups', type=int, required=True,
                        help='Number of powerups to put in the game map')
    parser.add_argument('--num-monsters', type=int, required=True,
                        help='Number of monsters to put in the game map')
    parser.add_argument('--num-dynamic-monsters', type=int, required=True,
                        help='Number of dynamic monsters to put in the game')
    parser.add_argument('--initial-strength', default=100, type=int,
                        help='Initial strength of each agent')
    parser.add_argument('--save-dir', type=str,
                        help='Save directory for saving the map')
    parser.add_argument('--map-file', type=str,
                        help='Path to the map JSON file')
    parser.add_argument('--play-against-human', action='store_true',
                        help='Whether to have a Human player as one of the '
                        'agents in the game')
    parser.add_argument('--show-map', action='store_true',
                        help='Whether to display the map in the terminal')
    parser.add_argument('--map-type', choices=MAP_TYPES, default='ascii',
                        help='Select map type. Choices are {' +
                        ', '.join(MAP_TYPES) + '}')
    parser.add_argument('--verbose', action='store_true',
                        help='Whether to be verbose when playing game')

    args = parser.parse_args(args)

    # TODO: Change how agents are populated
    agent = RandomAgent(args.height, args.width, args.initial_strength)

    agents = [agent]
    if args.play_against_human:
        human = HumanAgent(args.height, args.width, args.initial_strength)
        agents.append(human)

    try:
        game_driver = GameDriver(
            height=args.height, width=args.width,
            num_powerups=args.num_powerups,
            num_monsters=args.num_monsters,
            num_dynamic_monsters=args.num_dynamic_monsters,
            agents=agents,
            initial_strength=args.initial_strength,
            show_map=args.show_map, map_type=args.map_type,
            save_dir=args.save_dir, map_file=args.map_file)
    except InvalidMapError as e:
        print('The game map could not be created!')
        print(e)
        print('Restart the game')

    print('Starting game')
    try:
        game_driver.play(verbose=args.verbose)
    except StopIteration as e:
        print(e)
        return


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