Changeset 90:5d7d9c19b6c2

Timestamp:

12/20/10 01:49:55 (5 years ago)

Author:

Alex Martynov

Branch:

default

Message:

engine.legal_moves added random shuffle + engine.move_do() added

File:

• engine.py (modified) (2 diffs)

engine.py

@@ -97 +97 @@
             else:
                 dir_cell = self.field[snake.cells[0].y + applied_dir[0], snake.cells[0].x + applied_dir[1]]
-                if dir_cell.type == 'empty':
-                    snake.cells.insert(0,dir_cell)
-                    del snake.cells[-1]
-                    pass
-                elif (dir_cell.type == 'tail' and dir_cell.snake != snake):
-                    snake.cells.insert(0,dir_cell)
-                    del dir_cell.snake.cells[-1]
-                    pass
+                self.move_do()
                 break
         if applied_dir == None:
             applied_dir = snake.legal_dir[int(rnd.random()*len(snake.legal_dir))]
             dir_cell = self.field[snake.cells[0].y + applied_dir[0], snake.cells[0].x + applied_dir[1]]
-            if dir_cell.type == 'empty':
-                snake.cells.insert(0,dir_cell)
-                del snake.cells[-1]
-                pass
-            elif (dir_cell.type == 'tail' and dir_cell.snake != snake):
-                snake.cells.insert(0,dir_cell)
-                del dir_cell.snake.cells[-1]
-                pass
+            self.move_do()
             pass
         return
+    def move_do(self, snake, dir_cell):
+        if dir_cell.type == 'empty':
+            snake.cells.insert(0,dir_cell)
+            del snake.cells[-1]
+            pass
+        elif (dir_cell.type == 'tail' and dir_cell.snake != snake):
+            snake.cells.insert(0,dir_cell)
+            del dir_cell.snake.cells[-1]
+            pass
+
     def create_snake(self, snake_number):
         cells_id = []
@@ -160 +156 @@
             if (dir_cell.type == 'empty' or (dir_cell.type == 'tail' and dir_cell.snake != snake)):
                 snake.legal_dir.append(direction)
-                return
+        rnd.shuffle(snake.legal_dir)
         return