naja/widgets/tile.py

   1 # These will probably need to go away when we have images
   2 import pygame
   3 import pygame.locals as pgl
   4
   5 from naja.constants import TILE_SIZE, BITS, LOCK_HEIGHT, EXAMINE
   6 from naja.resources import resources
   7 from naja.resources.mutators import EIGHT_BIT
   8 from naja.widgets.base import Widget
   9
  10
  11 BIT_MAP = {
  12     frozenset([BITS.CYAN]): 'board/tile_cyan.png',
  13     frozenset([BITS.MAGENTA]): 'board/tile_magenta.png',
  14     frozenset([BITS.YELLOW]): 'board/tile_yellow.png',
  15     frozenset([BITS.CYAN, BITS.MAGENTA]): 'board/tile_cyan_magenta.png',
  16     frozenset([BITS.CYAN, BITS.YELLOW]): 'board/tile_cyan_yellow.png',
  17     frozenset([BITS.MAGENTA, BITS.YELLOW]): 'board/tile_magenta_yellow.png',
  18     frozenset([BITS.CYAN, BITS.MAGENTA, BITS.YELLOW]): 'board/tile_cyan_magenta_yellow.png',
  19     }
  20
  21
  22 class TileWidget(Widget):
  23     """Widget which holds a tile on the game board."""
  24     def __init__(self, pos, state=None, board_pos=None):
  25         super(TileWidget, self).__init__(pos, TILE_SIZE)
  26         self.state = state
  27         self.board_pos = board_pos
  28         self.highlighted = False
  29
  30     def prepare(self):
  31         # Draw background
  32         x, y = abs(self.board_pos[0] - 2), abs(self.board_pos[1] - 2)
  33         legal_move = (self.board_pos in self.state.player.legal_moves())
  34         if self.state.gameboard.player_mode == EXAMINE and legal_move:
  35             bg = resources.get_image('board/tile_available.png',
  36                                      transforms=(EIGHT_BIT,))
  37         elif (x + y) % 2 == 0:
  38             bg = resources.get_image('board/tile_2.png',
  39                                      transforms=(EIGHT_BIT,))
  40         else:
  41             bg = resources.get_image('board/tile_1.png',
  42                                      transforms=(EIGHT_BIT,))
  43         overlay = None
  44         if self.highlighted:
  45             overlay = resources.get_image('board/tile_selected.png',
  46                                           transforms=(EIGHT_BIT,))
  47         self.surface = pygame.surface.Surface(TILE_SIZE)
  48         self.surface.blit(bg, (0, 0))
  49         if overlay is not None:
  50             self.surface.blit(overlay, (0, 0))
  51         # Look up the required bits on the board location
  52         card = self.state.board_locations[self.board_pos]
  53         player_pos = self.state.player.position
  54         bits = []
  55         for action in card.actions:
  56             if action.required_bits:
  57                 bits.append(action.required_bits)
  58         # Sort so we have longer lists of bits later
  59         self.size = self.surface.get_rect().size
  60         if bits:
  61             bits.sort(key=lambda x: len(x))
  62             y_offset = TILE_SIZE[1] - LOCK_HEIGHT * len(bits)
  63             for pattern in bits:
  64                 if self.board_pos != player_pos:
  65                     img_name = BIT_MAP[pattern]
  66                     x_offset = 0
  67                 else:
  68                     img_name = BIT_MAP[pattern].replace('.png', '_small.png')
  69                     x_offset = 4
  70                     if y_offset >= TILE_SIZE[1] // 2:
  71                         # FIXME: Hack'ish
  72                         # Bump the lock down by some hand-tuned factor
  73                         # to not overlap with the robot
  74                         y_offset += LOCK_HEIGHT // 2 - 4
  75                 img = resources.get_image(img_name,
  76                                           transforms=(EIGHT_BIT,))
  77                 self.surface.blit(img, (x_offset, y_offset))
  78                 y_offset += LOCK_HEIGHT
  79
  80     def set_highlight(self, pos):
  81         self.highlighted = False
  82         if (self.state.gameboard.player_mode == EXAMINE and
  83                 self.board_pos == pos):
  84             self.highlighted = True
  85
  86     def draw(self, surface):
  87         surface.blit(self.surface, self.pos)