snake: main.py annotate

snake

annotate main.py @ 167:36756bd7e7ed

added oftions for UI.restart()

author	Alex Martynov
date	Thu, 23 Dec 2010 20:24:56 +0300
parents	f99b094a3d5d
children	0f7df983d610

rev	line source
martiran@18	1 import Tkinter as tk
Alex@23	2 import tkFileDialog as tkfd
martiran@20	3 import engine
martiran@32	4 import snake
martiran@18	5
martiran@18	6
martiran@18	7
martiran@18	8 class UI(object):
Alex@160	9 """User Interface:
Alex@160	10
Alex@160	11 Atributes:
Alex@160	12
Alex@160	13 - 'root' - root Window game placed at
Alex@160	14 - 'engine' - engine of the game
Alex@160	15 - 'canvas' - Widget field is pictured at
Alex@160	16 - 'step_id' - current step of the game
Alex@160	17 - 'after_id' - identificator of runing game process
Alex@166	18 - 'step_legth' - length of the step (in ms)
Alex@166	19 - 'game_length' - number of the steps in one round of the game"""
martiran@19	20 def __init__ (self):
martiran@136	21 """Create Python Battle game window.
martiran@136	22 Initialyze engige of the game."""
martiran@29	23 self.root = tk.Tk()
martiran@29	24 self.root.title("Python Battle")
martiran@29	25 self.canvas = tk.Canvas(self.root, background = "black")
martiran@19	26 self.canvas.pack(side ="top", fill="both", expand="yes")
me@96	27 self.buttons_pack(self.root).pack(side ="bottom", fill="both", expand="no")
Alex@165	28 self.step_id = 0
martiran@19	29 self.engine = engine.Engine(self.canvas)
martiran@32	30 self.after_id = None
Alex@160	31 self.step_length = 150
Alex@166	32 self.game_length = 200
martiran@18	33 return
martiran@136	34
me@96	35 def buttons_pack(self, root):
martiran@136	36 """Packing the buttons in root frame.
martiran@136	37 Definition of button functions."""
me@96	38 buttons = tk.Frame(root)
Alex@160	39 load_1 = tk.Button(buttons, text="Load 1", command=lambda: self.load(0))
martiran@132	40 load_1.grid(row=1, column=2, stick="news")
Alex@160	41 load_2 = tk.Button(buttons, text="Load 2", command=lambda: self.load(1))
martiran@132	42 load_2.grid(row=2, column=3, stick="news")
Alex@165	43 run_b = tk.Button(buttons, text="Run", command=lambda: self.start())
Alex@160	44 run_b.grid(row=2, column=2, stick="news")
Alex@167	45 restart_b = tk.Button(buttons, text="Restart", command=lambda: self.restart(survived="no"))
Alex@160	46 restart_b.grid(row=1, column=5, stick="news")
Alex@160	47 load_3 = tk.Button(buttons, text="Load 3", command=lambda: self.load(2))
martiran@132	48 load_3.grid(row=3, column=2, stick="news")
Alex@160	49 load_4 = tk.Button(buttons, text="Load 4", command=lambda: self.load(3))
martiran@132	50 load_4.grid(row=2, column=1, stick="news")
me@96	51 step_b = tk.Button(buttons, text="Step", command=lambda: self.step())
martiran@151	52 step_b.grid(row=2, column=5, stick="news")
martiran@151	53 end_b = tk.Button(buttons, text="End", command=lambda: self.end())
martiran@151	54 end_b.grid(row=3, column=5, stick="news")
me@133	55 for column in range(1, 6):
me@133	56 buttons.grid_columnconfigure(column, weight=1)
me@96	57 return buttons
martiran@29	58
Alex@23	59 def load (self, snake_number):
martiran@136	60 """Ask for snake file loading.
martiran@136	61 Initialyzing snake and draw it on the field.
Alex@164	62 Return field back to default (without snakes) after end of the game."""
Alex@155	63 if self.step_id == 666:
martiran@29	64 self.step_id = 0
martiran@130	65 self.engine.snakes = [None, None, None, None]
martiran@29	66 pass
martiran@107	67 if self.step_id == 0:
me@150	68 file = tkfd.askopenfile(title="Open file")
me@150	69 if file == None:
me@150	70 return
martiran@29	71 snake = self.engine.create_snake(snake_number)
me@150	72 snake.load(file)
martiran@29	73 pass
Alex@103	74 self.engine.refill()
Alex@86	75 self.engine.redraw()
Alex@27	76 return
Alex@23	77
Alex@165	78 def start (self):
Alex@165	79 """Init running of the game."""
Alex@166	80 if self.step_id == self.game_length + 666:
Alex@167	81 self.restart(survived="yes")
Alex@97	82 if self.dead_snake_check() == False:
Alex@97	83 return
Alex@166	84 if self.step_id > self.game_length:
martiran@122	85 self.end()
martiran@122	86 return
Alex@165	87 self.engine.psnakes = self.engine.snakes[:]
Alex@165	88 self.run()
Alex@165	89
Alex@165	90 def run (self):
Alex@165	91 """Run the game with 'step_length' ms step
Alex@165	92 After the end of the game - restarts it with snakes survived in
Alex@165	93 previous game"""
martiran@32	94 self.step_id = self.step_id+1
martiran@28	95 self.engine.step()
Alex@160	96 self.after_id = self.canvas.after(self.step_length, self.run)
martiran@28	97 return
Alex@160	98
martiran@28	99 def step (self):
martiran@136	100 """Do the next game step"""
Alex@97	101 if self.dead_snake_check() == False:
Alex@97	102 return
Alex@166	103 if self.step_id <= self.game_length:
martiran@32	104 if self.after_id != None:
martiran@32	105 self.canvas.after_cancel(self.after_id)
martiran@32	106 pass
martiran@32	107 self.step_id = self.step_id+1
martiran@29	108 self.engine.step()
martiran@29	109 pass
martiran@29	110 else:
martiran@29	111 self.end()
martiran@29	112 pass
martiran@29	113 return
Alex@97	114
Alex@97	115 def dead_snake_check(self):
martiran@136	116 """Check the number of snakes alive.
martiran@136	117 End the game if alive snake number is less than two."""
Alex@97	118 dead_snakes = 0
Alex@97	119 for snake in self.engine.snakes:
Alex@97	120 if snake == None:
Alex@97	121 dead_snakes=dead_snakes+1
Alex@97	122 pass
Alex@97	123 if dead_snakes >= 3:
Alex@97	124 self.end()
Alex@97	125 return False
Alex@160	126
Alex@167	127 def restart(self, survived):
Alex@167	128 """"Restarts snakes positions after the end of the game
Alex@167	129
Alex@167	130 Options:
Alex@167	131 survived = "yes" - restarts next round only with snakes survived in previous round
Alex@167	132 survived = "no" - restart next roun with all snakes played in previous round"""
Alex@167	133 if survived == "yes":
Alex@167	134 snake_set = self.engine.snakes
Alex@167	135 else:
Alex@167	136 snake_set = self.engine.psnakes
Alex@163	137 self.step_id = 0
Alex@167	138 for i, snake in enumerate(snake_set):
Alex@167	139 if snake_set[i] != None:
Alex@163	140 self.engine.snakes[i] = snake
Alex@163	141 self.engine.create_snake(i, snake)
Alex@163	142 self.engine.refill()
Alex@163	143 self.engine.redraw()
Alex@23	144
martiran@28	145 def end (self):
martiran@136	146 """End the game and raise the window that tels about it."""
Alex@60	147 if self.after_id != None:
Alex@60	148 self.canvas.after_cancel(self.after_id)
Alex@60	149 pass
Alex@166	150 self.step_id = self.game_length + 666
martiran@29	151 root = tk.Tk()
martiran@29	152 end_label = tk.Label(root, text="End")
martiran@29	153 end_label.pack()
martiran@29	154 root.mainloop()
martiran@18	155 pass
martiran@32	156
me@42	157 if __name__ == "__main__":
me@42	158 snake_batle = UI()
me@42	159 snake_batle.root.mainloop()