tanchiki: tk_ui.py annotate

tanchiki

annotate tk_ui.py @ 53:5450c97fbc38

check_wall() added

author	Olga Zolotareva <olya_zol@inbox.ru>
date	Fri, 24 Dec 2010 23:28:21 +0300
parents	f6d3c0539e8c
children	5cb3fef573f6

rev	line source
me@43	1 #!/usr/bin/python
me@34	2 import math
me@34	3 import Tkinter as tk
me@38	4 import body
me@50	5 from body import Tank as BaseTank, Bullet
me@34	6 from game import Game
me@40	7 from user import User
me@46	8 from vector import Vector
me@38	9
me@51	10 game_size = 500, 500
me@38	11 keys = {
olya_zol@52	12 'ocircumflex': 'base_left',
olya_zol@52	13 'acircumflex': 'base_right',
olya_zol@52	14 'eacute': 'turret_left',
olya_zol@52	15 'oacute': 'turret_right',
olya_zol@52	16 'odiaeresis': 'accelerate',
olya_zol@52	17 'ucircumflex': 'decelerate',
olya_zol@52	18 'division': 'fire',
me@38	19 }
me@38	20
me@44	21 welcome = """Press F5 to start
me@44	22
me@44	23 Keys are:
me@44	24 a, d -- turn tank
me@44	25 q, e -- turn muzzle
me@44	26 w, s -- change speed
me@44	27 x -- fire"""
me@44	28
me@50	29 class Tank(BaseTank):
me@38	30
me@38	31 def on_tick(self, tanks, bullets):
me@38	32 if self.user.base_left:
olya_zol@52	33 self.rotate_base(-1 * body.base_angle)
me@49	34 if self.user.base_right:
me@46	35 self.rotate_base(body.base_angle)
me@38	36 if self.user.turret_left:
olya_zol@52	37 self.rotate_turret(-1 * body.turret_angle)
me@49	38 if self.user.turret_right:
me@46	39 self.rotate_turret(body.turret_angle)
me@38	40 if self.user.accelerate:
me@46	41 self.accelerate(body.speed_delta)
me@38	42 if self.user.decelerate:
olya_zol@52	43 self.accelerate(-1 * body.speed_delta)
me@38	44 if self.user.fire:
me@38	45 self.fire()
me@34	46
me@34	47 class UI(object):
me@38	48
me@34	49 def __init__(self):
me@34	50 self.init_game()
me@34	51 self.init_ui()
me@45	52 self.display_welcome()
me@34	53
me@38	54 def init_game(self):
me@38	55 self.user = User(keys)
olya_zol@52	56 w, h = game_size
me@46	57 game = self.game = Game([], [self.user], w, h)
me@51	58 tank = Tank(Vector(game_size) 0.5, self.user, self.game)
me@46	59 game.bodies.append(tank)
me@38	60
me@34	61 def init_ui(self):
me@34	62 root = self.root = tk.Tk()
me@47	63 root.bind("<KeyPress>", self.on_keypress)
me@47	64 root.bind("<KeyRelease>", self.on_keyrelease)
me@51	65 w, h = game_size
me@51	66 canvas = self.canvas = tk.Canvas(root,
me@51	67 background="black", width=w, height=h)
me@34	68 canvas.pack(fill="both", expand="yes")
me@34	69
me@44	70 def display_welcome(self):
me@45	71 root, canvas = self.root, self.canvas
me@41	72 root.update()
me@41	73 w, h = canvas.winfo_width(), canvas.winfo_height()
me@44	74 canvas.create_rectangle(0, 0, w, h, fill="#211")
me@44	75 canvas.create_text(w/2, h/2, text=welcome,
me@44	76 fill="red", justify="center", font="bold 15")
me@41	77
me@47	78 def on_keypress(self, ev):
me@41	79 if ev.keysym == "F5":
me@41	80 self.step()
me@47	81 self.user.on_key(ev.keysym, True)
me@47	82
me@47	83 def on_keyrelease(self, ev):
me@47	84 self.user.on_key(ev.keysym, False)
me@34	85
me@34	86 def step(self):
me@34	87 self.root.after(100, self.step)
me@40	88 self.game.step()
me@34	89 self.redraw()
me@34	90
me@34	91 def redraw(self):
me@34	92 self.canvas.delete("all")
olya_zol@52	93 for i in self.game.bodies:
olya_zol@52	94 p = i.position
olya_zol@52	95 lt = p + Vector(-1,-1) * i.radius
olya_zol@52	96 rb = p + Vector(1,1) * i.radius
me@47	97
olya_zol@52	98 v1 = Vector(i.radius, 0)
olya_zol@52	99 if isinstance(i, body.Tank):
olya_zol@52	100 v1.phi = i.base_orientation.phi
olya_zol@52	101 else :
olya_zol@52	102 v1.phi = i.velocity.phi
me@48	103 vb = p - v1
me@48	104 ve = p + v1 * 2
olya_zol@52	105 self.canvas.create_line(vb.x, vb.y, ve.x, ve.y, fill="darkgray", width=i.radius*2)
me@48	106
olya_zol@52	107 if isinstance(i, BaseTank):
olya_zol@52	108 t = i.turret * i.radius * 1.5 + p
me@48	109 self.canvas.create_oval(lt.x, lt.y, rb.x, rb.y, fill="darkgreen")
me@48	110 self.canvas.create_line(p.x, p.y, t.x, t.y, fill="orange", width=2)
olya_zol@52	111 elif isinstance(i, Bullet):
me@50	112 self.canvas.create_oval(lt.x, lt.y, rb.x, rb.y, outline="red")
me@50	113 else:
olya_zol@52	114 raise AssertionError("Unknown object type: %s" % i.__class__.__name__)
me@34	115
me@34	116 if __name__ == "__main__":
me@34	117 UI().root.mainloop()