Changeset 167:36756bd7e7ed

Timestamp:

12/23/10 20:24:56 (5 years ago)

Author:

Alex Martynov

Branch:

default

Message:

added oftions for UI.restart()

File:

• main.py (modified) (3 diffs)

main.py

@@ -43 +43 @@
         run_b = tk.Button(buttons, text="Run", command=lambda: self.start())
         run_b.grid(row=2, column=2, stick="news")
-        restart_b = tk.Button(buttons, text="Restart", command=lambda: self.restart())
+        restart_b = tk.Button(buttons, text="Restart", command=lambda: self.restart(survived="no"))
         restart_b.grid(row=1, column=5, stick="news")
         load_3 = tk.Button(buttons, text="Load 3", command=lambda: self.load(2))
@@ -79 +79 @@
         """Init running of the game."""
         if self.step_id == self.game_length + 666:
-            self.next_round()
+            self.restart(survived="yes")
         if self.dead_snake_check() == False:
             return
@@ -125 +125 @@
             return False
         
-    def restart(self):
-        """"Restarts the game after the end of the game with snakes survived"""
+    def restart(self, survived):
+        """"Restarts snakes positions after the end of the game
+
+        Options:
+        survived = "yes" - restarts next round only with snakes survived in previous round
+        survived = "no" - restart next roun with all snakes played in previous round"""
+        if survived == "yes":
+            snake_set = self.engine.snakes
+        else:
+            snake_set = self.engine.psnakes          
         self.step_id = 0
-        for i, snake in enumerate(self.engine.psnakes):
-            if self.engine.psnakes[i] != None:
+        for i, snake in enumerate(snake_set):
+            if snake_set[i] != None:
                 self.engine.snakes[i] = snake
                 self.engine.create_snake(i, snake)
         self.engine.refill()
         self.engine.redraw()
-
-    def next_round(self):
-        pass
 
     def end (self):