From 368a79abc1982615dc521e2128abd15ba1d28966 Mon Sep 17 00:00:00 2001
From: Peter Ward <peteraward@gmail.com>
Date: Fri, 10 Sep 2010 11:49:17 +1000
Subject: Added wrapping support.

---
 snake.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/snake.py b/snake.py
index 0429040..add6e62 100644
--- a/snake.py
+++ b/snake.py
@@ -15,9 +15,10 @@ import traceback
 from common import *
 
 class SnakeEngine(object):
-    def __init__(self, rows, columns, n_apples, results=False):
+    def __init__(self, rows, columns, n_apples, wrap=False, results=False):
         super(SnakeEngine, self).__init__()
 
+        self.wrap = wrap
         self.bots = {}
         self.results = None
         if results:
@@ -123,9 +124,13 @@ class SnakeEngine(object):
                 nx = x + dx
                 ny = y + dy
 
-                if ny < 0 or ny >= self.rows or nx < 0 or nx >= self.columns:
-                    self.remove_bot(letter)
-                    continue
+                if self.wrap:
+                    ny %= self.rows
+                    nx %= self.columns
+                else:
+                    if ny < 0 or ny >= self.rows or nx < 0 or nx >= self.columns:
+                        self.remove_bot(letter)
+                        continue
 
                 oldcell = self.board[ny][nx]
                 if oldcell in (Squares.EMPTY, Squares.APPLE):
-- 
cgit v1.2.3