59 lines
1.9 KiB
Python
Executable File
59 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Import pygame without printing anything on the terminal
|
|
import pygame
|
|
|
|
from tron.map import Map
|
|
from tron.game import Game, PositionPlayer
|
|
from tron.window import Window
|
|
from tron.player import Direction, KeyboardPlayer, Mode
|
|
|
|
from ais.wallavoider import Ai
|
|
|
|
# This script shows how to create a game with human players and play it interactively.
|
|
# It shows how to create the game, to setup interactive controls for users, and
|
|
# to run the game while rendering it on a window with a reasonnable framerate.
|
|
|
|
def main():
|
|
# Initialize the game engine
|
|
pygame.init()
|
|
|
|
# Prepare the size for the game.
|
|
# Those values may be good if you want to play, they might not be so good
|
|
# to train your AI. Decreasing them will make the learning faster.
|
|
width = 14
|
|
height = 14
|
|
|
|
# Create a game from its size and its players
|
|
game = Game(width, height, [
|
|
# We create two PositionPlayer for each player of the game.
|
|
# The first one has the id 1, and will use keyboard interaction, with a
|
|
# default direction that will be to the right, and that will use the Z,
|
|
# Q, S and D keys.
|
|
# The last array defines the initial position of the player.
|
|
PositionPlayer(1, Ai(), [0, 0]),
|
|
|
|
# We create a second player that will use the arrow keys.
|
|
PositionPlayer(2, KeyboardPlayer(Direction.LEFT, Mode.ARROWS), [width - 1, height - 1]),
|
|
])
|
|
|
|
# Create a window for the game so the players can see what they're doing.
|
|
window = Window(game, 10)
|
|
|
|
# Hide mouse
|
|
pygame.mouse.set_visible(False)
|
|
|
|
# Run the game.
|
|
game.main_loop(window)
|
|
|
|
# Once the game is finished, if game.winner is None, it means it's a draw
|
|
# Otherwise, game.winner will tell us which player has won the game.
|
|
if game.winner is None:
|
|
print("It's a draw!")
|
|
else:
|
|
print('Player {} wins!'.format(game.winner))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|