def _add_grids(self):
'''Create UI containers for the player grids.'''
self.my_grid_frame = PlayerGridFrame(self)
self.my_grid_frame.place(x=self.X_PADDING + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
l1 = Label(self.my_grid_frame, text="Your Grid")
l1.pack()
self.my_grid = ShipGrid(self.my_grid_frame, True)
self.my_grid.pack(side=LEFT, pady=20)
self.their_grid_frame = PlayerGridFrame(self)
self.their_grid_frame.place(x=self.my_grid.size + self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
self.their_grid_label = Label(self.their_grid_frame, text="Opponent's Grid")
self.their_grid_label.pack()
self.their_grid = ShipGrid(self.their_grid_frame, False)
self.their_grid.pack(side=LEFT, pady=20)
self._add_grid_events()
def _add_grids(self):
'''Create UI containers for the player grids.'''
self.my_grid_frame = PlayerGridFrame(self)
self.my_grid_frame.place(x=self.X_PADDING + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
l1 = Label(self.my_grid_frame, text="Your Grid")
l1.pack()
self.my_grid = ShipGrid(self.my_grid_frame, True)
self.my_grid.pack(side=LEFT, pady=20)
self.their_grid_frame = PlayerGridFrame(self)
self.their_grid_frame.place(x=self.my_grid.size + self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
self.their_grid_label = Label(self.their_grid_frame, text="Opponent's Grid")
self.their_grid_label.pack()
self.their_grid = ShipGrid(self.their_grid_frame, False)
self.their_grid.pack(side=LEFT, pady=20)
self._add_grid_events()
class Game(Frame):
'''Top-level Frame managing top-level events. Interact directly with user.'''
############ geometry ###############
X_PADDING = 25
Y_PADDING = 25
SHIP_PANEL_WIDTH = 150
BUTTON_PANEL_HEIGHT = 50
BUTTON_PADDING = 5
WARNING_BAR_HEIGHT = 40
#####################################
########### states ##################
PLACING = 0
PLAYING = 1
GAME_OVER = 2
#####################################
############ colors #################
BACKGROUND_COLOR = "white"
WARNING_BACKGROUND = "khaki1"
#####################################
###### window titles and messages ################
GAME_OVER_POPUP_TITLE = "Game Over"
GAME_OVER_WIN_MSG = "You win!"
GAME_OVER_LOSE_MSG = "Game over. You lose."
WINDOW_TITLE_GAME_OVER = "Battleship (Game Over)"
WINDOW_TITLE_NORMAL = "Battleship"
##################################################
def __init__(self, master):
'''Create the UI for a game of battleship.'''
Frame.__init__(self, master)
self._create_ui()
# these are 'controller' elements that should really be in another class
self.ai = ShipAI(self.their_grid._model, self.my_grid._model)
self.reset()
def show_warning(self, msg, title=None):
'''Show a warning msg that a certain action is illegal.
Allows user to understand what's going on (i.e. why their action failed.
'''
tkMessageBox.showwarning(title, msg)
def _create_ui(self):
'''Create all UI elements for the game.'''
self._create_menu()
self._add_grids()
self._add_staging_panel()
self._addship_panels()
self._make_buttons()
self.config(height=self.Y_PADDING * 3 + self.my_grid.size + self.BUTTON_PANEL_HEIGHT + self.WARNING_BAR_HEIGHT)
self.set_all_bgs(self.BACKGROUND_COLOR, self)
def _destroy_popup(self, event=None):
'''Process removal of the popup.'''
self.master.focus_set()
self._popup.grab_release()
self._popup.destroy()
def _show_rules(self):
'''Show the help dialog in a new window.'''
# load the help page
help_page_location = "help/rules.txt"
f = open(help_page_location, "r")
lines = f.read()
f.close()
tkMessageBox.showinfo("Rules", lines)
def show_keyboard_shortcuts(self):
'''Show a dialog box with the keyboard shortcuts.'''
# load the keyboard shortcuts page
ks_page_location = "help/keyboard_shortcuts.txt"
f = open(ks_page_location, "r")
lines = f.read()
f.close()
tkMessageBox.showinfo("Keyboard Shortcuts", lines)
def _create_menu(self):
'''Create the menu in the GUI.'''
menubar = Menu(self)
self.menus = {}
count = 0
self.file_menu = Menu(menubar, tearoff=0)
self.file_menu.add_command(label="New Game")#, command=self.reset)
self.menus["file_new_game"] = count
count += 1
self.file_menu.add_command(label="Save")
self.menus["file_save"] = count
count += 1
self.file_menu.add_command(label="Open")
self.menus["file_open"] = count
count += 1
self.file_menu.add_command(label="Exit")#, command=self.master.destroy)
self.menus["file_exit"] = count
count += 1
menubar.add_cascade(label="File", menu=self.file_menu)
if battleship.GameController.DEV_FLAG:
count = 0
self.dev_menu = Menu(menubar, tearoff=0)
self.dev_menu.add_command(label="Auto Place")
self.menus["dev_auto_place"] = count
count += 1
self.dev_menu.add_command(label="Random Shot")
self.menus["dev_random_shot"] = count
count += 1
self.dev_menu.add_command(label="Auto Load")
self.menus["dev_auto_load"] = count
count += 1
menubar.add_cascade(label="Dev", menu=self.dev_menu)
help_menu = Menu(menubar, tearoff=0)
help_menu.add_command(label="Rules", command=self._show_rules)
help_menu.add_command(label="Keyboard Shortcuts", command=self.show_keyboard_shortcuts)
menubar.add_cascade(label="Help", menu=help_menu)
self.master.config(menu=menubar)
def _add_staging_panel(self):
'''Create the placement/ship staging panel.'''
self.my_grid_frame.staging_panel = ShipPlacementPanel(self)
self.my_grid_frame.staging_panel.place(
x=self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH + self.my_grid.size,
y=self.Y_PADDING
)
def set_all_bgs(self, color, parent):
'''Set all the backgrounds of the child widgets to a certain color.'''
parent.config(background=color)
for child in parent.winfo_children():
self.set_all_bgs(color, child)
def _addship_panels(self):
'''Add a list of ships to select from, for adding.
Note that staging area must be added FIRST'''
############################## ShipPanel ########################
self.my_grid_frame.ship_panel = ShipPanel(self)
self.my_grid_frame.ship_panel.place(x=self.X_PADDING, y=self.Y_PADDING * 4)
self.unselect_ship()
##################################################################
###################### ShipWarPanel ##############################
self.my_grid_frame._ship_war_panel = ShipWarPanel(self)
self.my_grid_frame._ship_war_panel.config(height=self.my_grid_frame.winfo_height())
self.my_grid_frame._ship_war_panel.place(x=self.X_PADDING, y=self.Y_PADDING * 2)
##################################################################
###################### EnemyShipPanel ############################
self.their_grid_frame.ship_panel = EnemyShipPanel(self)
self.their_grid_frame.ship_panel.place(x=self.my_grid.size * 2 + self.X_PADDING * 3 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING * 4)
##################################################################
def unselect_ship(self):
'''Deselect all ships in the placement and staging GUIs.'''
self.my_grid_frame.ship_panel._ship_var.set(10)
self.my_grid_frame.staging_panel.reset()
def _hide_frame(self, frame):
'''Since you can't hide a frame per se, 'unpack' the frame's child widgets.
WARNING: this removes all packing directions for children'''
frame.lower()
for child in frame.winfo_children():
child.pack_forget()
def show_game_over_popup(self, winner):
'''Show a popup with a dialog saying the game is over, and showing the winning player.'''
msg = {
battleship.GameController. HUMAN_PLAYER : self.GAME_OVER_WIN_MSG,
battleship.GameController.AI_PLAYER : self.GAME_OVER_LOSE_MSG
} [winner]
tkMessageBox.showinfo(self.GAME_OVER_POPUP_TITLE, msg)
def process_placing_state(self):
'''Basic stuff to do during placing state.'''
self.config(width=self.X_PADDING * 3 + self.my_grid.size + self.SHIP_PANEL_WIDTH + self.my_grid_frame.staging_panel.CANVAS_WIDTH)
# show staging panel
self.my_grid_frame.staging_panel.pack_ui()
self.my_grid_frame.staging_panel.lift(aboveThis=self.their_grid_frame)
self.my_grid_frame.staging_panel.reset()
#re-pack
self.play_game_button.pack(side=LEFT, padx=self.BUTTON_PADDING, pady=self.BUTTON_PADDING)
self.play_game_button.config(state=DISABLED)
self._hide_frame(self.their_grid_frame)
self._hide_frame(self.my_grid_frame._ship_war_panel)
self.my_grid_frame.ship_panel.lift(aboveThis=self.my_grid_frame._ship_war_panel)
# allow the AI to place ships
self.ai.place_ships()
def process_playing_state(self):
'''Basic stuff to do during playing state.'''
self.config(width=self.X_PADDING * 4 + self.my_grid.size * 2 + self.SHIP_PANEL_WIDTH * 2)
self.my_grid._model.finalize()
self.their_grid._model.finalize()
self._hide_frame(self.my_grid_frame.staging_panel)
self.their_grid.config(state=NORMAL)
self.their_grid.enable()
# hide while playing
self.play_game_button.pack_forget()
self.unselect_ship()
self.my_grid_frame._ship_war_panel.pack_ui()
self.my_grid_frame._ship_war_panel.lift(aboveThis=self.my_grid_frame.ship_panel)
# show opponent's grid
self.their_grid_frame.lift(aboveThis=self.my_grid_frame.staging_panel)
self.their_grid_label.pack()
self.their_grid.pack(side=LEFT, pady=20)
def process_game_over_state(self):
'''Change the view to reflect the game is over.'''
self.their_grid.disable()
self.master.title(self.WINDOW_TITLE_GAME_OVER)
def process_state(self):
'''Simple state controller to enable and disable certain widgets depending on the state.'''
if self._state == self.PLACING:
self.process_placing_state()
elif self._state == self.PLAYING:
self.process_playing_state()
elif self._state == self.GAME_OVER:
self.process_game_over_state()
def _add_grid_events(self):
'''Add events to the grids.'''
#self.their_grid.tag_bind("tile", "<Button-1>", self._shot)
pass
def _add_grids(self):
'''Create UI containers for the player grids.'''
self.my_grid_frame = PlayerGridFrame(self)
self.my_grid_frame.place(x=self.X_PADDING + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
l1 = Label(self.my_grid_frame, text="Your Grid")
l1.pack()
self.my_grid = ShipGrid(self.my_grid_frame, True)
self.my_grid.pack(side=LEFT, pady=20)
self.their_grid_frame = PlayerGridFrame(self)
self.their_grid_frame.place(x=self.my_grid.size + self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
self.their_grid_label = Label(self.their_grid_frame, text="Opponent's Grid")
self.their_grid_label.pack()
self.their_grid = ShipGrid(self.their_grid_frame, False)
self.their_grid.pack(side=LEFT, pady=20)
self._add_grid_events()
def reset(self):
'''New game!'''
self.master.title(self.WINDOW_TITLE_NORMAL)
# reset selected ship
self.unselect_ship()
# reset staging area
self.my_grid_frame.staging_panel.reset()
# reset indicators on ships in panels
self.my_grid_frame._ship_war_panel.reset()
for ship, button in self.my_grid_frame.ship_panel.ship_buttons.items():
button.config(foreground="black")
for x, y in self.my_grid.get_tiles():
self.reset_closure(x, y)
self._state = self.PLACING
self.process_state()
def reset_closure(self, x, y):
'''Add a placement event to the given tile.
TODO this is badly named'''
tag_id = self.my_grid._get_tile_name(x, y)
c = self.get_add_ship_callback()
f = lambda event: self.add_staged_ship(x, y, c)
self.my_grid.tag_bind(tag_id, "<Button-1>", f)
def add_staged_ship(self, x, y, callback):
'''Take the stage from the staging area, and place it on the board at position (x, y).
After ship has been placed, execute the function <callback>.'''
s = self.my_grid_frame.staging_panel.get_staged_ship()
if s is not None:
self.my_grid.add_ship(x, y, s.get_short_name(), s.is_vertical(), callback)
def get_add_ship_callback(self):
'''Return the callback function for adding a ship.'''
return lambda: self.ship_set(self.my_grid_frame.ship_panel.get_current_ship())
def _set_ship_sunk(self, ship):
'''This is a callback, to be called when a ship has been sunk.
TODO for now only called when one of MY ships is sunk.
UI shows that the given ship has been sunk.'''
self.my_grid_frame.ship_panel.set_sunk(ship)
def _set_ship_hit(self, ship):
'''This is a callback, to be called when a ship has been hit.
TODO for now only called when one of MY ships is hit.
UI shows that the given ship has been hit.'''
assert ship is not None
self.my_grid_frame._ship_war_panel.update(ship)
def ship_set(self, ship):
'''This is a callback, to be called when a ship has been placed.
UI shows that the given ship has been placed.'''
self._set_ships[ship] = True
self.my_grid_frame.ship_panel.set_placed(ship)
if all(self._set_ships.values()):
self.play_game_button.config(state=NORMAL)
def _make_buttons(self):
'''Create action buttons at the bottom.'''
button_row = self.my_grid.size + self.Y_PADDING + (self.BUTTON_PANEL_HEIGHT - 2 * self.BUTTON_PADDING)
button_frame = Frame(self)
button_frame.place(x=self.my_grid.size - self.X_PADDING, y=button_row)
self.play_game_button = Button(button_frame, text="Play")
self.play_game_button.pack(side=LEFT, padx=self.BUTTON_PADDING, pady=self.BUTTON_PADDING)
def redraw(self):
'''Redraw the GUI, reloading all info from the model.
TODO this is a work-in-progress'''
# first, figure out the state
# are the ships placed? are they sunk?
grids = [self.my_grid._model, self.their_grid._model]
# set panels to the correct state
self.my_grid_frame.ship_panel.redraw(grids[0])
self.my_grid_frame._ship_war_panel.redraw(grids[0])
self.their_grid_frame.ship_panel.redraw(grids[1])
# set the grid to the correct state
self.my_grid.redraw(grids[0])
self.their_grid.redraw(grids[1])
if all([g.has_all_ships() for g in grids]) and all([g.all_sunk() for g in grids]):
#TODO
# state is game over.
pass
elif all([g.has_all_ships() for g in grids]):
self.process_playing_state()
else:
#TODO
# state is placing
pass
class Game(Frame):
'''Top-level Frame managing top-level events. Interact directly with user.'''
############ geometry ###############
X_PADDING = 25
Y_PADDING = 25
SHIP_PANEL_WIDTH = 150
BUTTON_PANEL_HEIGHT = 50
BUTTON_PADDING = 5
WARNING_BAR_HEIGHT = 40
#####################################
########### states ##################
PLACING = 0
PLAYING = 1
GAME_OVER = 2
#####################################
############ colors #################
BACKGROUND_COLOR = "white"
WARNING_BACKGROUND = "khaki1"
#####################################
###### window titles and messages ################
GAME_OVER_POPUP_TITLE = "Game Over"
GAME_OVER_WIN_MSG = "You win!"
GAME_OVER_LOSE_MSG = "Game over. You lose."
WINDOW_TITLE_GAME_OVER = "Battleship (Game Over)"
WINDOW_TITLE_NORMAL = "Battleship"
##################################################
def __init__(self, master):
'''Create the UI for a game of battleship.'''
Frame.__init__(self, master)
self._create_ui()
# these are 'controller' elements that should really be in another class
self.ai = ShipAI(self.their_grid._model, self.my_grid._model)
self.reset()
def show_warning(self, msg, title=None):
'''Show a warning msg that a certain action is illegal.
Allows user to understand what's going on (i.e. why their action failed.
'''
tkMessageBox.showwarning(title, msg)
def _create_ui(self):
'''Create all UI elements for the game.'''
self._create_menu()
self._add_grids()
self._add_staging_panel()
self._addship_panels()
self._make_buttons()
self.config(height=self.Y_PADDING * 3 + self.my_grid.size + self.BUTTON_PANEL_HEIGHT + self.WARNING_BAR_HEIGHT)
self.set_all_bgs(self.BACKGROUND_COLOR, self)
def _destroy_popup(self, event=None):
'''Process removal of the popup.'''
self.master.focus_set()
self._popup.grab_release()
self._popup.destroy()
def _show_rules(self):
'''Show the help dialog in a new window.'''
# load the help page
help_page_location = "help/rules.txt"
f = open(help_page_location, "r")
lines = f.read()
f.close()
tkMessageBox.showinfo("Rules", lines)
def show_keyboard_shortcuts(self):
'''Show a dialog box with the keyboard shortcuts.'''
# load the keyboard shortcuts page
ks_page_location = "help/keyboard_shortcuts.txt"
f = open(ks_page_location, "r")
lines = f.read()
f.close()
tkMessageBox.showinfo("Keyboard Shortcuts", lines)
def _create_menu(self):
'''Create the menu in the GUI.'''
menubar = Menu(self)
self.menus = {}
count = 0
self.file_menu = Menu(menubar, tearoff=0)
self.file_menu.add_command(label="New Game")#, command=self.reset)
self.menus["file_new_game"] = count
count += 1
self.file_menu.add_command(label="Save")
self.menus["file_save"] = count
count += 1
self.file_menu.add_command(label="Open")
self.menus["file_open"] = count
count += 1
self.file_menu.add_command(label="Exit")#, command=self.master.destroy)
self.menus["file_exit"] = count
count += 1
menubar.add_cascade(label="File", menu=self.file_menu)
if battleship.GameController.DEV_FLAG:
count = 0
self.dev_menu = Menu(menubar, tearoff=0)
self.dev_menu.add_command(label="Auto Place")
self.menus["dev_auto_place"] = count
count += 1
self.dev_menu.add_command(label="Random Shot")
self.menus["dev_random_shot"] = count
count += 1
self.dev_menu.add_command(label="Auto Load")
self.menus["dev_auto_load"] = count
count += 1
menubar.add_cascade(label="Dev", menu=self.dev_menu)
help_menu = Menu(menubar, tearoff=0)
help_menu.add_command(label="Rules", command=self._show_rules)
help_menu.add_command(label="Keyboard Shortcuts", command=self.show_keyboard_shortcuts)
menubar.add_cascade(label="Help", menu=help_menu)
self.master.config(menu=menubar)
def _add_staging_panel(self):
'''Create the placement/ship staging panel.'''
self.my_grid_frame.staging_panel = ShipPlacementPanel(self)
self.my_grid_frame.staging_panel.place(
x=self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH + self.my_grid.size,
y=self.Y_PADDING
)
def set_all_bgs(self, color, parent):
'''Set all the backgrounds of the child widgets to a certain color.'''
parent.config(background=color)
for child in parent.winfo_children():
self.set_all_bgs(color, child)
def _addship_panels(self):
'''Add a list of ships to select from, for adding.
Note that staging area must be added FIRST'''
############################## ShipPanel ########################
self.my_grid_frame.ship_panel = ShipPanel(self)
self.my_grid_frame.ship_panel.place(x=self.X_PADDING, y=self.Y_PADDING * 4)
self.unselect_ship()
##################################################################
###################### ShipWarPanel ##############################
self.my_grid_frame._ship_war_panel = ShipWarPanel(self)
self.my_grid_frame._ship_war_panel.config(height=self.my_grid_frame.winfo_height())
self.my_grid_frame._ship_war_panel.place(x=self.X_PADDING, y=self.Y_PADDING * 2)
##################################################################
###################### EnemyShipPanel ############################
self.their_grid_frame.ship_panel = EnemyShipPanel(self)
self.their_grid_frame.ship_panel.place(x=self.my_grid.size * 2 + self.X_PADDING * 3 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING * 4)
##################################################################
def unselect_ship(self):
'''Deselect all ships in the placement and staging GUIs.'''
self.my_grid_frame.ship_panel._ship_var.set(10)
self.my_grid_frame.staging_panel.reset()
def _hide_frame(self, frame):
'''Since you can't hide a frame per se, 'unpack' the frame's child widgets.
WARNING: this removes all packing directions for children'''
frame.lower()
for child in frame.winfo_children():
child.pack_forget()
def show_game_over_popup(self, winner):
'''Show a popup with a dialog saying the game is over, and showing the winning player.'''
msg = {
battleship.GameController. HUMAN_PLAYER : self.GAME_OVER_WIN_MSG,
battleship.GameController.AI_PLAYER : self.GAME_OVER_LOSE_MSG
} [winner]
tkMessageBox.showinfo(self.GAME_OVER_POPUP_TITLE, msg)
def process_placing_state(self):
'''Basic stuff to do during placing state.'''
self.config(width=self.X_PADDING * 3 + self.my_grid.size + self.SHIP_PANEL_WIDTH + self.my_grid_frame.staging_panel.CANVAS_WIDTH)
# show staging panel
self.my_grid_frame.staging_panel.pack_ui()
self.my_grid_frame.staging_panel.lift(aboveThis=self.their_grid_frame)
self.my_grid_frame.staging_panel.reset()
#re-pack
self.play_game_button.pack(side=LEFT, padx=self.BUTTON_PADDING, pady=self.BUTTON_PADDING)
self.play_game_button.config(state=DISABLED)
self._hide_frame(self.their_grid_frame)
self._hide_frame(self.my_grid_frame._ship_war_panel)
self.my_grid_frame.ship_panel.lift(aboveThis=self.my_grid_frame._ship_war_panel)
# allow the AI to place ships
self.ai.place_ships()
def process_playing_state(self):
'''Basic stuff to do during playing state.'''
self.config(width=self.X_PADDING * 4 + self.my_grid.size * 2 + self.SHIP_PANEL_WIDTH * 2)
self.my_grid._model.finalize()
self.their_grid._model.finalize()
self._hide_frame(self.my_grid_frame.staging_panel)
self.their_grid.config(state=NORMAL)
self.their_grid.enable()
# hide while playing
self.play_game_button.pack_forget()
self.unselect_ship()
self.my_grid_frame._ship_war_panel.pack_ui()
self.my_grid_frame._ship_war_panel.lift(aboveThis=self.my_grid_frame.ship_panel)
# show opponent's grid
self.their_grid_frame.lift(aboveThis=self.my_grid_frame.staging_panel)
self.their_grid_label.pack()
self.their_grid.pack(side=LEFT, pady=20)
def process_game_over_state(self):
'''Change the view to reflect the game is over.'''
self.their_grid.disable()
self.master.title(self.WINDOW_TITLE_GAME_OVER)
def process_state(self):
'''Simple state controller to enable and disable certain widgets depending on the state.'''
if self._state == self.PLACING:
self.process_placing_state()
elif self._state == self.PLAYING:
self.process_playing_state()
elif self._state == self.GAME_OVER:
self.process_game_over_state()
def _add_grid_events(self):
'''Add events to the grids.'''
#self.their_grid.tag_bind("tile", "<Button-1>", self._shot)
pass
def _add_grids(self):
'''Create UI containers for the player grids.'''
self.my_grid_frame = PlayerGridFrame(self)
self.my_grid_frame.place(x=self.X_PADDING + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
l1 = Label(self.my_grid_frame, text="Your Grid")
l1.pack()
self.my_grid = ShipGrid(self.my_grid_frame, True)
self.my_grid.pack(side=LEFT, pady=20)
self.their_grid_frame = PlayerGridFrame(self)
self.their_grid_frame.place(x=self.my_grid.size + self.X_PADDING * 2 + self.SHIP_PANEL_WIDTH, y=self.Y_PADDING)
self.their_grid_label = Label(self.their_grid_frame, text="Opponent's Grid")
self.their_grid_label.pack()
self.their_grid = ShipGrid(self.their_grid_frame, False)
self.their_grid.pack(side=LEFT, pady=20)
self._add_grid_events()
def reset(self):
'''New game!'''
self.master.title(self.WINDOW_TITLE_NORMAL)
# reset selected ship
self.unselect_ship()
# reset staging area
self.my_grid_frame.staging_panel.reset()
# reset indicators on ships in panels
self.my_grid_frame._ship_war_panel.reset()
for ship, button in self.my_grid_frame.ship_panel.ship_buttons.iteritems():
button.config(foreground="black")
for x, y in self.my_grid.get_tiles():
self.reset_closure(x, y)
self._state = self.PLACING
self.process_state()
def reset_closure(self, x, y):
'''Add a placement event to the given tile.
TODO this is badly named'''
tag_id = self.my_grid._get_tile_name(x, y)
c = self.get_add_ship_callback()
f = lambda event: self.add_staged_ship(x, y, c)
self.my_grid.tag_bind(tag_id, "<Button-1>", f)
def add_staged_ship(self, x, y, callback):
'''Take the stage from the staging area, and place it on the board at position (x, y).
After ship has been placed, execute the function <callback>.'''
s = self.my_grid_frame.staging_panel.get_staged_ship()
if s is not None:
self.my_grid.add_ship(x, y, s.get_short_name(), s.is_vertical(), callback)
def get_add_ship_callback(self):
'''Return the callback function for adding a ship.'''
return lambda: self.ship_set(self.my_grid_frame.ship_panel.get_current_ship())
def _set_ship_sunk(self, ship):
'''This is a callback, to be called when a ship has been sunk.
TODO for now only called when one of MY ships is sunk.
UI shows that the given ship has been sunk.'''
self.my_grid_frame.ship_panel.set_sunk(ship)
def _set_ship_hit(self, ship):
'''This is a callback, to be called when a ship has been hit.
TODO for now only called when one of MY ships is hit.
UI shows that the given ship has been hit.'''
self.my_grid_frame._ship_war_panel.update(ship)
def ship_set(self, ship):
'''This is a callback, to be called when a ship has been placed.
UI shows that the given ship has been placed.'''
self._set_ships[ship] = True
self.my_grid_frame.ship_panel.set_placed(ship)
if all(self._set_ships.values()):
self.play_game_button.config(state=NORMAL)
def _make_buttons(self):
'''Create action buttons at the bottom.'''
button_row = self.my_grid.size + self.Y_PADDING + (self.BUTTON_PANEL_HEIGHT - 2 * self.BUTTON_PADDING)
button_frame = Frame(self)
button_frame.place(x=self.my_grid.size - self.X_PADDING, y=button_row)
self.play_game_button = Button(button_frame, text="Play")
self.play_game_button.pack(side=LEFT, padx=self.BUTTON_PADDING, pady=self.BUTTON_PADDING)
def redraw(self):
'''Redraw the GUI, reloading all info from the model.
TODO this is a work-in-progress'''
# first, figure out the state
# are the ships placed? are they sunk?
grids = [self.my_grid._model, self.their_grid._model]
# set panels to the correct state
self.my_grid_frame.ship_panel.redraw(grids[0])
self.my_grid_frame._ship_war_panel.redraw(grids[0])
self.their_grid_frame.ship_panel.redraw(grids[1])
# set the grid to the correct state
self.my_grid.redraw(grids[0])
self.their_grid.redraw(grids[1])
if all([g.has_all_ships() for g in grids]) and all([g.all_sunk() for g in grids]):
#TODO
# state is game over.
pass
elif all([g.has_all_ships() for g in grids]):
self.process_playing_state()
else:
#TODO
# state is placing
pass