1 files changed, 118 insertions, 0 deletions
diff --git a/snakegame/engines/pyglet.py b/snakegame/engines/pyglet.py
new file mode 100644
index 0000000..5a17ac0
--- /dev/null
+++ b/snakegame/engines/pyglet.py
@@ -0,0 +1,118 @@
+#!/usr/bin/env python
+
+from __future__ import division
+
+import pyglet
+pyglet.resource.path = ['images']
+pyglet.resource.reindex()
+
+from pyglet import gl
+
+import common
+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 = 2
+
+    def __init__(self, rows, columns, n_apples, *args, **kwargs):
+        super(PygletSnakeEngine, self).__init__(rows, columns, n_apples, *args, **kwargs)
+
+        gl.glEnable(gl.GL_BLEND)
+        gl.glBlendFunc(gl.GL_SRC_ALPHA, gl.GL_ONE_MINUS_SRC_ALPHA)
+
+        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 = self.height - int(y * yscale)
+                right = int((x + 1) * xscale)
+                bottom = self.height - int((y + 1) * yscale)
+                r = (left, top, right, top, right, bottom, left, bottom)
+
+                # Draw a square.
+                gl.glLineWidth(self.EDGE_WIDTH)
+                pyglet.graphics.draw(4, gl.GL_LINE_LOOP,
+                                     ('v2f', r),
+                                     ('c4B', self.EDGE_COLOR * 4))
+
+                # Draw the things on the square.
+                if cell == common.Squares.APPLE:
+                    w, h = self.apple.size
+                    self.apple.blit(left + (xscale - w) / 2.0, top - h, width=w, height=h)
+
+                elif cell.isalpha(): # Snake...
+                    colour = self.bots[cell.lower()][1] + (255,)
+                    gl.glPolygonMode(gl.GL_FRONT, gl.GL_FILL)
+                    pyglet.graphics.draw(4, gl.GL_POLYGON,
+                                         ('v2f', r),
+                                         ('c4B', colour * 4),
+                    )
+
+                    if cell.isupper(): # Snake head
+                        w, h = self.eyes.size
+                        self.eyes.blit(left, top - h, width=w, height=h)
+
+    def update_snakes(self, *args):
+        if not self.bots:
+            pyglet.app.exit()
+        super(PygletSnakeEngine, self).update_snakes(*args)
+
+    def run(self):
+        pyglet.app.run()
+
+if __name__ == '__main__':
+    import sys
+    from processbot import BotWrapper
+
+    rows, columns, apples = map(int, sys.argv[1:4])
+    game = PygletSnakeEngine(rows, columns, apples)
+    for filename in sys.argv[4:]:
+        bot = BotWrapper(filename)
+        game.add_bot(bot)
+    game.run()
+