summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPeter Ward <peteraward@gmail.com>2009-09-28 12:00:43 +1000
committerPeter Ward <peteraward@gmail.com>2009-09-28 12:00:43 +1000
commit84c620ce93dbb9da9fc2328c13b8a48316641d4a (patch)
treec0f8e69bb7fadad122381e6862d8b046f11e549b
parent9cc9e121b18556f6c83cad335e32621c27f656bc (diff)
Added peter bot.
-rw-r--r--peter.py71
-rwxr-xr-xsnake.py8
2 files changed, 78 insertions, 1 deletions
diff --git a/peter.py b/peter.py
new file mode 100644
index 0000000..b182575
--- /dev/null
+++ b/peter.py
@@ -0,0 +1,71 @@
+#!/usr/bin/env python
+"""New and improved bot, OPTIMISED!!"""
+
+import random
+import sys
+
+from common import *
+
+def peter_bot(board, (HEADX, HEADY)):
+ HEIGHT = len(board)
+ WIDTH = len(board[0])
+ SNAKE_BODY = board[HEADY][HEADX]
+
+ SNAKE_BODY = SNAKE_BODY.lower()
+ SNAKE_HEAD = SNAKE_BODY.upper()
+
+ def get_cell(board, x, y):
+ if x < 0 or x >= WIDTH or y < 0 or y >= HEIGHT:
+ raise KeyError, 'out of range.'
+ return board[y][x]
+
+ md_two = {
+ (-1, 0, 'l'): ((-2, 0), (-1, 1), (-1, -1)),
+ (0, -1, 'u'): ((-1, -1), (1, -1), (0, -2)),
+ (1, 0, 'r'): ((2, 0), (1, 1), (1, -1)),
+ (0, 1, 'd'): (((0, 2), (-1, 1), (1, 1))),
+ }
+
+ max_score = 0
+ max_moves = []
+
+ for (dx, dy, move), adj in md_two.items():
+ score = 0
+
+ try:
+ square = get_cell(board, HEADX + dx, HEADY + dy)
+ except KeyError:
+ continue
+
+ if square == Squares.APPLE:
+ score += 2
+ elif square != Squares.EMPTY:
+ continue # Definitely cannot move here.
+
+ for ddx, ddy in adj:
+ try:
+ square = get_cell(board, HEADX + ddx, HEADY + ddy)
+ except KeyError:
+ score -= 1
+ continue
+
+ if square == Squares.APPLE:
+ score += 2
+ elif square == Squares.EMPTY:
+ score += 1
+ elif square == SNAKE_BODY:
+ score -= 1
+ elif square.isupper():
+ score += 3
+
+ if score == max_score:
+ max_moves.append(move)
+ elif score > max_score:
+ max_score = score
+ max_moves = [move]
+
+ if max_moves:
+ return random.choice(max_moves)
+ else:
+ return 'U' # Suicide!
+
diff --git a/snake.py b/snake.py
index f374134..f0bf46b 100755
--- a/snake.py
+++ b/snake.py
@@ -2,6 +2,7 @@
from __future__ import division
+import time
import string
from random import randint
from collections import deque
@@ -199,7 +200,7 @@ class SnakeEngine(object):
clock = pygame.time.Clock()
running = True
- while running:
+ 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):
@@ -226,16 +227,21 @@ class SnakeEngine(object):
# 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 peter import *
game = SnakeEngine(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(peter_bot)
game.run()