X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Ftile.py;h=acbc28df7cf3b7fbf2d95907d4f61f62a01f9efe;hb=7ce574e6d8521f74fab9ca65847a49492c5783b5;hp=6750584f071f1587052f09396964bdd5c8eae2fd;hpb=9299925383d4746e7f9c4e948e083a1bff1e339a;p=naja.git

diff --git a/naja/widgets/tile.py b/naja/widgets/tile.py
index 6750584..acbc28d 100644
--- a/naja/widgets/tile.py
+++ b/naja/widgets/tile.py
@@ -1,15 +1,91 @@
-#from naja.constants import TILE_SIZE
-from naja.widgets.base import Widget
+# These will probably need to go away when we have images
+import pygame
+
+from naja.constants import TILE_SIZE, BITS, LOCK_HEIGHT, EXAMINE
 from naja.resources import resources
+from naja.resources.mutators import EIGHT_BIT
+from naja.widgets.base import Widget
+
+
+BIT_MAP = {
+    frozenset([BITS.RED]): 'board/tile_red.png',
+    frozenset([BITS.GREEN]): 'board/tile_green.png',
+    frozenset([BITS.BLUE]): 'board/tile_blue.png',
+    frozenset([BITS.RED, BITS.GREEN]): 'board/tile_red_green.png',
+    frozenset([BITS.RED, BITS.BLUE]): 'board/tile_red_blue.png',
+    frozenset([BITS.GREEN, BITS.BLUE]): 'board/tile_green_blue.png',
+    frozenset([BITS.RED, BITS.GREEN, BITS.BLUE]):
+        'board/tile_red_green_blue.png',
+}
 
 
 class TileWidget(Widget):
     """Widget which holds a tile on the game board."""
-    def __init__(self, pos, image=None):
-        super(TileWidget, self).__init__(pos, (96, 96))
+    def __init__(self, pos, state=None, board_pos=None):
+        super(TileWidget, self).__init__(pos, TILE_SIZE)
+        self.state = state
+        self.board_pos = board_pos
+        self.highlighted = False
 
     def prepare(self):
-        pass
+        # Draw background
+        x, y = abs(self.board_pos[0] - 2), abs(self.board_pos[1] - 2)
+
+        if (x + y) % 2 == 0:
+            bg = resources.get_image('board/tile_2.png',
+                                     transforms=(EIGHT_BIT,))
+        else:
+            bg = resources.get_image('board/tile_1.png',
+                                     transforms=(EIGHT_BIT,))
+        overlays = []
+
+        legal_move = (self.board_pos in self.state.player.legal_moves())
+
+        if self.state.gameboard.player_mode == EXAMINE and legal_move:
+            overlays.append(resources.get_image('board/tile_available.png',
+                                     transforms=(EIGHT_BIT,)))
+        if self.highlighted:
+            overlays.append(resources.get_image('board/tile_selected.png',
+                                          transforms=(EIGHT_BIT,)))
+
+        self.surface = pygame.surface.Surface(TILE_SIZE)
+        self.surface.blit(bg, (0, 0))
+        for overlay in overlays:
+            self.surface.blit(overlay, (0, 0))
+        # Look up the required bits on the board location
+        card = self.state.board_locations[self.board_pos]
+        player_pos = self.state.player.position
+        bits = []
+        for action in card.actions:
+            if action.required_bits:
+                bits.append(action.required_bits)
+        # Sort so we have longer lists of bits later
+        self.size = self.surface.get_rect().size
+        if bits:
+            bits.sort(key=lambda x: len(x))
+            y_offset = TILE_SIZE[1] - LOCK_HEIGHT * len(bits)
+            for pattern in bits:
+                if self.board_pos != player_pos:
+                    img_name = BIT_MAP[pattern]
+                    x_offset = 0
+                else:
+                    img_name = BIT_MAP[pattern].replace('.png', '_small.png')
+                    x_offset = 4
+                    if y_offset >= TILE_SIZE[1] // 2:
+                        # FIXME: Hack'ish
+                        # Bump the lock down by some hand-tuned factor
+                        # to not overlap with the robot
+                        y_offset += LOCK_HEIGHT // 2 - 4
+                img = resources.get_image(img_name,
+                                          transforms=(EIGHT_BIT,))
+                self.surface.blit(img, (x_offset, y_offset))
+                y_offset += LOCK_HEIGHT
+
+    def set_highlight(self, pos):
+        self.highlighted = False
+        if (self.state.gameboard.player_mode == EXAMINE and
+                self.board_pos == pos):
+            self.highlighted = True
 
     def draw(self, surface):
-        surface.blit(self.surface, self.rect)
+        surface.blit(self.surface, self.pos)