fix broken merge
[naja.git] / 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
34         if (x + y) % 2 == 0:
35             bg = resources.get_image('board/tile_2.png',
36                                      transforms=(EIGHT_BIT,))
37         else:
38             bg = resources.get_image('board/tile_1.png',
39                                      transforms=(EIGHT_BIT,))
40         overlays = []
41
42         legal_move = (self.board_pos in self.state.player.legal_moves())
43
44         if self.state.gameboard.player_mode == EXAMINE and legal_move:
45             overlays.append(resources.get_image('board/tile_available.png',
46                                      transforms=(EIGHT_BIT,)))
47         if self.highlighted:
48             overlays.append(resources.get_image('board/tile_selected.png',
49                                           transforms=(EIGHT_BIT,)))
50
51         self.surface = pygame.surface.Surface(TILE_SIZE)
52         self.surface.blit(bg, (0, 0))
53         for overlay in overlays:
54             self.surface.blit(overlay, (0, 0))
55         # Look up the required bits on the board location
56         card = self.state.board_locations[self.board_pos]
57         player_pos = self.state.player.position
58         bits = []
59         for action in card.actions:
60             if action.required_bits:
61                 bits.append(action.required_bits)
62         # Sort so we have longer lists of bits later
63         self.size = self.surface.get_rect().size
64         if bits:
65             bits.sort(key=lambda x: len(x))
66             y_offset = TILE_SIZE[1] - LOCK_HEIGHT * len(bits)
67             for pattern in bits:
68                 if self.board_pos != player_pos:
69                     img_name = BIT_MAP[pattern]
70                     x_offset = 0
71                 else:
72                     img_name = BIT_MAP[pattern].replace('.png', '_small.png')
73                     x_offset = 4
74                     if y_offset >= TILE_SIZE[1] // 2:
75                         # FIXME: Hack'ish
76                         # Bump the lock down by some hand-tuned factor
77                         # to not overlap with the robot
78                         y_offset += LOCK_HEIGHT // 2 - 4
79                 img = resources.get_image(img_name,
80                                           transforms=(EIGHT_BIT,))
81                 self.surface.blit(img, (x_offset, y_offset))
82                 y_offset += LOCK_HEIGHT
83
84     def set_highlight(self, pos):
85         self.highlighted = False
86         if (self.state.gameboard.player_mode == EXAMINE and
87                 self.board_pos == pos):
88             self.highlighted = True
89
90     def draw(self, surface):
91         surface.blit(self.surface, self.pos)