1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
import random
from common import *
import pygame
pygame.init()
class SnakeEngine(object):
def __init__(self, rows, columns, n_apples, width=800, height=600, fullscreen=False):
super(SnakeEngine, self).__init__()
flags = 0
if fullscreen:
flags |= pygame.FULLSCREEN
pygame.display.set_mode((width, height), flags)
self.bots = {}
self.new_game(rows, columns, n_apples)
def new_game(self, rows, columns, n_apples):
self.board = [[Squares.EMPTY for x in xrange(columns)] for y in xrange(rows)]
for i in xrange(n_apples):
x = random.randint(0, columns - 1)
y = random.randint(0, rows - 1)
self.board[y][x] = Squares.APPLE
def add_bot(self, name, bot):
"""
A bot is a callable object, with this method signature:
def bot_callable(
board=[[cell for cell in row] for row in board],
position=(snake_x, snake_y)
):
return random.choice('RULD')
"""
self.bots[name] = bot
def remove_bot(self, name):
del self.bots[name]
def run(self):
pass
if __name__ == '__main__':
game = SnakeEngine(8, 16, 10)
|
