p2_gui.py

from Tkinter import *
from p2_game import Game, State

# Default bots, overridden by redteam and blueteam
import first_bot as red_bot
import first_bot as blue_bot
BOTS = {'red': red_bot, 'blue': blue_bot}

master = Tk()

UNDO_STACK = []
RED_AI = IntVar(master)
BLUE_AI = IntVar(master)
AI_THOUGHTS = StringVar(master)
BOARD_SIZE = 4

master.title("Dots and Boxes")

w = 600
h = 600

canvas = Canvas(master, width=w, height=h)

def display(state):

	canvas.delete(ALL)

	square_width = min(int(canvas['width']),int(canvas['height']))

	step = square_width/state.game.width
	r = int(step/10.0)
	w = int(step/15.0)

	def make_callback(move):
		def callback(event):
			if state.whos_turn == 'red' and RED_AI.get():
				print "Give the red guy a chance to think!"
				return
			if state.whos_turn == 'blue' and BLUE_AI.get():
				print "The blue lady needs more time to think!"
				return
			make_move(state, move)
		return callback


	for i,j in state.game.h_lines:
		x = (i+0.5)*step
		y = (j+0.5)*step
		if (i,j) in state.h_line_owners:
			owner = state.h_line_owners[(i,j)]
			canvas.create_line(x,y,x+step,y,width=w)
		else:
			line = canvas.create_line(x,y,x+step,y,width=w,dash=(w,w),fill=state.whos_turn)
			canvas.tag_bind(line,"<Button-1>",make_callback(('h',(i,j))))
			
	for i,j in state.game.v_lines:
		x = (i+0.5)*step
		y = (j+0.5)*step
		if (i,j) in state.v_line_owners:
			owner = state.v_line_owners[(i,j)]
			canvas.create_line(x,y,x,y+step,width=w)
		else:
			line = canvas.create_line(x,y,x,y+step,width=w,dash=(w,w),fill=state.whos_turn)
			canvas.tag_bind(line,"<Button-1>",make_callback(('v',(i,j))))
			
	for i,j in state.game.boxes:
		x = (i+0.5)*step
		y = (j+0.5)*step
		if (i,j) in state.box_owners:
			owner = state.box_owners[(i,j)]
			canvas.create_rectangle(x+r,y+r,x+step-r,y+step-r,fill=owner)

	for i,j in state.game.dots:
		x = (i+0.5)*step
		y = (j+0.5)*step
		canvas.create_oval(x-r,y-r,x+r,y+r,fill='black')

	if not state.is_terminal():
		if state.whos_turn == 'red' and RED_AI.get():
			think(state)
		elif state.whos_turn == 'blue' and BLUE_AI.get():
			think(state)

def make_move(state, move):
	moves = state.get_moves()
	if move in moves:
		UNDO_STACK.append(state)
		next_state = state.copy()
		next_state.apply_move(move)
		display(next_state)
		if next_state.is_terminal():
			try:
				from pygame import mixer
				mixer.init()
				bustin = mixer.Sound('bustin.wav')
				bustin.play()
			except ImportError:
				print "For the full dots and boxes experience install pygame"
			AI_THOUGHTS.set("BOXIN' MAKES ME FEEL GOOD!")
	else:
		print move, "not in legal moves!"

def think(state):
	import threading
	class ThinkingThread(threading.Thread):
		def run(self):
			def quip(line):
				AI_THOUGHTS.set(line)
			move = BOTS[state.whos_turn].think(state.copy(), quip)
			make_move(state, move)
			#AI_THOUGHTS.set("")

	ThinkingThread().start()


def restart():
	game = Game(BOARD_SIZE)
	initial_state = State(game)
	UNDO_STACK[:] = [initial_state]
	display(initial_state)

def undo():
	if len(UNDO_STACK) > 1:
		UNDO_STACK.pop()
		display(UNDO_STACK[-1])

def rungui(redteam=None, blueteam=None, boardSize=4):
	# Override the bots if provided
	global BOARD_SIZE, BOTS
	if redteam:
		BOTS['red'] = redteam
	if blueteam:
		BOTS['blue'] = blueteam
	BOARD_SIZE = boardSize

	toolbar = Frame(master, width=w, height=h+20)
	toolbar.pack(side=BOTTOM)

	undo_btn = Button(toolbar, text="Undo", command=undo)
	undo_btn.pack(side=LEFT)

	restart_btn = Button(toolbar, text="Restart", command=restart)
	restart_btn.pack(side=LEFT)

	red_ai_btn = Checkbutton(toolbar, text="Red AI", variable=RED_AI)
	red_ai_btn.pack(side=LEFT)
	blue_ai_btn = Checkbutton(toolbar, text="Blue AI", variable=BLUE_AI)
	blue_ai_btn.pack(side=LEFT)
	BLUE_AI.set(1)

	ai_thoughts_ent = Entry(toolbar, textvariable=AI_THOUGHTS, state=DISABLED, width=50)
	ai_thoughts_ent.pack(side=LEFT)

	canvas.pack(side=RIGHT)

	restart()

	mainloop()

if __name__ == '__main__':
	rungui()