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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#!/usr/bin/env python
from __future__ import division
import os
import time
import pygame
pygame.init()
from pygame.locals import *
from common import *
from snake import SnakeEngine
class Sprites(object):
PREFIX = 'images'
def __getattribute__(self, name):
try:
return object.__getattribute__(self, name.upper())
except AttributeError:
from pygame.image import load
filename = os.path.join(self.PREFIX, name.lower() + ".png")
image = load(filename).convert_alpha()
setattr(self, name, image)
return image
Sprites = Sprites()
def scale_aspect((source_width, source_height), (target_width, target_height)):
source_aspect = source_width / source_height
target_aspect = target_width / target_height
if source_aspect > target_aspect:
# restrict width
width = target_width
height = width / source_aspect
else:
# restrict height
height = target_height
width = height * source_aspect
return (width, height)
class PygameSnakeEngine(SnakeEngine):
EDGE_COLOR = (255, 255, 255)
EDGE_WIDTH = 1
def __init__(self, rows, columns, n_apples,
width=800, height=600, fullscreen=False,
**kwargs):
flags = 0
if fullscreen:
flags |= pygame.FULLSCREEN
self.screen = pygame.display.set_mode((width, height), flags)
self.width = width
self.height = height
super(PygameSnakeEngine, self).__init__(rows, columns, n_apples,
**kwargs)
def new_game(self, rows, columns, n_apples):
super(PygameSnakeEngine, self).new_game(rows, columns, n_apples)
# make board surface
self.board_width, self.board_height = scale_aspect(
(columns, rows), (self.width, self.height)
)
self.surface = pygame.Surface((self.board_width, self.board_height))
# load sprites
xscale = self.board_width / self.columns
yscale = self.board_height / self.rows
def load_image(image):
new_size = scale_aspect(image.get_size(), (xscale, yscale))
return pygame.transform.smoothscale(image, new_size)
self.apple = load_image(Sprites.APPLE)
self.eyes = load_image(Sprites.EYES)
def draw_board(self):
xscale = self.board_width / self.columns
yscale = self.board_height / self.rows
# Draw grid.
for y, row in enumerate(self.board):
for x, cell in enumerate(row):
left = int(x * xscale)
top = int(y * yscale)
w = int((x + 1) * xscale) - left
h = int((y + 1) * yscale) - top
r = Rect(left, top, w, h)
# Draw a square.
pygame.draw.rect(self.surface, self.EDGE_COLOR, r,
self.EDGE_WIDTH)
# Draw the things on the square.
if cell == Squares.APPLE:
self.surface.blit(self.apple, r.topleft)
elif cell.isalpha(): # Snake...
colour = self.bots[cell.lower()][1]
self.surface.fill(colour, r)
if cell.isupper(): # Snake head
self.surface.blit(self.eyes, r.topleft)
def run(self):
clock = pygame.time.Clock()
running = True
while running and self.bots:
for event in pygame.event.get():
if event.type == pygame.QUIT or \
(event.type == pygame.KEYDOWN and event.key == K_ESCAPE):
running = False
break
if not running: break
# Clear the screen.
self.screen.fill((0, 0, 0))
self.surface.fill((0, 0, 0))
# Draw the board.
self.draw_board()
# Center the board.
x = (self.width - self.board_width) / 2
y = (self.height - self.board_height) / 2
self.screen.blit(self.surface, (x, y))
# Update the display.
pygame.display.flip()
clock.tick(20)
# Let the snakes move!
self.update_snakes()
if running:
time.sleep(2)
if __name__ == '__main__':
import sys
from processbot import BotWrapper
rows, columns, apples = map(int, sys.argv[1:4])
game = PygameSnakeEngine(rows, columns, apples)
for filename in sys.argv[4:]:
bot = BotWrapper(filename)
game.add_bot(bot)
game.run()
# Early window close, late process cleanup.
pygame.display.quit()
|
