diff options
| -rw-r--r-- | common.py | 13 | ||||
| -rwxr-xr-x | pygame_snake.py | 13 | ||||
| -rwxr-xr-x | pyglet_snake.py | 140 |
3 files changed, 153 insertions, 13 deletions
@@ -11,16 +11,3 @@ class Squares(object): EMPTY = '.' APPLE = '*' -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() - diff --git a/pygame_snake.py b/pygame_snake.py index d1639aa..dece4a4 100755 --- a/pygame_snake.py +++ b/pygame_snake.py @@ -10,6 +10,19 @@ from pygame.locals 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 diff --git a/pyglet_snake.py b/pyglet_snake.py new file mode 100755 index 0000000..6cae702 --- /dev/null +++ b/pyglet_snake.py @@ -0,0 +1,140 @@ +#!/usr/bin/env python + +from __future__ import division + +import time + +import pyglet +pyglet.resource.path = ['images'] +pyglet.resource.reindex() + +from snake import SnakeEngine + +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 PygletSnakeEngine(SnakeEngine, pyglet.window.Window): + EDGE_COLOR = (255, 255, 255, 255) + EDGE_WIDTH = 1 + + def __init__(self, rows, columns, n_apples): + super(PygletSnakeEngine, self).__init__(rows, columns, n_apples) + + pyglet.clock.schedule_interval(lambda t: self.update_snakes(), 0.025) + + def new_game(self, rows, columns, n_apples): + super(PygletSnakeEngine, self).new_game(rows, columns, n_apples) + + # make board surface + self.board_width, self.board_height = scale_aspect( + (columns, rows), (self.width, self.height) + ) + + # load sprites + xscale = self.board_width / self.columns + yscale = self.board_height / self.rows + + self.apple = pyglet.resource.image('apple.png') + self.apple.size = scale_aspect( + (self.apple.width, self.apple.height), + (xscale, yscale) + ) + self.eyes = pyglet.resource.image('eyes.png') + self.eyes.size = scale_aspect( + (self.eyes.width, self.eyes.height), + (xscale, yscale) + ) + + def on_draw(self): + self.clear() + + 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) + right = int((x + 1) * xscale) + bottom = int((y + 1) * yscale) + r = (left, top, right, top, right, bottom, left, bottom) + + # Draw a square. +# width = self.EDGE_WIDTH + pyglet.graphics.draw(4, pyglet.gl.GL_LINE_LOOP, + ('v2f', r), + ('c4B', self.EDGE_COLOR * 4)) + + # Draw the things on the square. + if cell == Squares.APPLE: + w, h = self.apple.size + self.apple.blit(top, left, width=w, height=h) + + elif cell.isalpha(): # Snake... + colour = self.bots[cell.lower()][1] +# self.surface.fill(colour, r) + + if cell.isupper(): # Snake head + w, h = self.eyes.size + self.eyes.blit(top, left, width=w, height=h) + + 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) + + # Early window close, late process cleanup. + pygame.display.quit() + +if __name__ == '__main__': + from bots import * + from oldbot import BotWrapper + + game = PygletSnakeEngine(25, 25, 50) + game.add_bot(right_bot) + game.add_bot(random_bot) + game.add_bot(random_bounds_bot) + game.add_bot(random_square_bot) + game.add_bot(BotWrapper('oldbots/peter.py')) + pyglet.app.run() + |