naja/widgets/board.py

   1 """
   2 Widget that holds the game tiles.
   3 """
   4 import pygame.locals as pgl
   5
   6 from naja.constants import BOARD_SIZE, TILE_SIZE, KEYS, EXAMINE, ACT
   7 from naja.events import finish_event
   8
   9 from naja.widgets.base import Widget
  10 from naja.widgets.tile import TileWidget
  11
  12
  13 class BoardWidget(Widget):
  14     """
  15     Widget which holds all the tiles that make up the gameboard.
  16     """
  17     def __init__(self, pos, state, info):
  18         super(BoardWidget, self).__init__(pos, BOARD_SIZE)
  19         self.info = info
  20         self.state = state
  21         self.card_pos = (2, 2)
  22         self._tiles = []
  23         self.legal = False
  24         for y in range(0, 5):
  25             for x in range(0, 5):
  26                 tile_pos = (pos[0] + x * TILE_SIZE[0],
  27                             pos[1] + y * TILE_SIZE[1])
  28                 self._tiles.append(TileWidget(tile_pos, state, (x, y)))
  29
  30     def prepare(self):
  31         for tile in self._tiles:
  32             tile.set_highlight(self.card_pos)
  33             tile.prepare()
  34         self.size = BOARD_SIZE
  35         if self.state.gameboard.player_mode == ACT:
  36             self.card_pos = self.state.player.position
  37
  38     def draw(self, surface):
  39         for tile in self._tiles:
  40             tile.draw(surface)
  41
  42     def change_pos(self, offset):
  43         tpos = (offset[0] + self.card_pos[0], offset[1] + self.card_pos[1])
  44         if tpos[0] < 0 or tpos[1] < 0 or tpos[0] > 4 or tpos[1] > 4:
  45             return False
  46         self.card_pos = tpos
  47         self.info.set_position(self.card_pos)
  48         self.legal = self.card_pos in self.state.player.legal_moves()
  49         return True
  50
  51     def handle_event(self, ev):
  52         if self.state.gameboard.player_mode == ACT:
  53             return super(BoardWidget, self).handle_event(ev)
  54         if ev.type == pgl.KEYDOWN:
  55             if ev.key in KEYS.UP:
  56                 if self.change_pos((0, -1)):
  57                     return finish_event()
  58             if ev.key in KEYS.DOWN:
  59                 if self.change_pos((0, +1)):
  60                     return finish_event()
  61             if ev.key in KEYS.LEFT:
  62                 if self.change_pos((-1, 0)):
  63                     return finish_event()
  64             if ev.key in KEYS.RIGHT:
  65                 if self.change_pos((+1, 0)):
  66                     return finish_event()
  67             if (ev.key in KEYS.SELECT and
  68                     self.state.player.set_position(self.card_pos)):
  69                 self.state.gameboard.change_mode(ACT)
  70                 self.info.set_position(self.card_pos)
  71                 return finish_event()
  72         return super(BoardWidget, self).handle_event(ev)