Render bit pattern markers on tile widgets

[naja.git] / naja / widgets / tile.py

diff --git a/naja/widgets/tile.py b/naja/widgets/tile.py
index 7550ba573ba147c4e39d3e0bccb7325bd318251e..a152f765a102de9511e030b84ec4285337ad04d3 100644 (file)
--- a/naja/widgets/tile.py
+++ b/naja/widgets/tile.py
@@ -2,16 +2,24 @@
 import pygame
 import pygame.locals as pgl
 
-from naja.constants import TILE_SIZE
+from naja.constants import TILE_SIZE, BITS
 from naja.resources import resources
 from naja.widgets.base import Widget
 
 
+BIT_MAP = {
+        BITS['CYAN']: 'board/tile_cyan.png',
+        BITS['MAGENTA']: 'board/tile_magenta.png',
+        BITS['YELLOW']: 'board/tile_yellow.png',
+        }
+
+
 class TileWidget(Widget):
     """Widget which holds a tile on the game board."""
-    def __init__(self, pos, image=None):
+    def __init__(self, pos, state=None, board_pos=None):
         super(TileWidget, self).__init__(pos, TILE_SIZE)
-        self.image = image
+        self.state = state
+        self.board_pos = board_pos
 
     def prepare(self):
         # Placeholder logic - just draw the outline of a square
@@ -19,7 +27,24 @@ class TileWidget(Widget):
         pygame.draw.lines(self.surface, pgl.color.THECOLORS['yellow'],
                           True, [(1, 1), (1, 95), (95, 95), (95, 1)], 2)
         self.surface.convert_alpha(pygame.display.get_surface())
+        # Look up the required bits on the board location
+        card = self.state.gameboard.board_locations[self.board_pos]
+        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] - 32 * len(bits)
+            for pattern in bits:
+                x_offset = 10
+                for bit in pattern:
+                    img = resources.get_image(BIT_MAP[bit])
+                    self.surface.blit(img, (x_offset, y_offset))
+                    x_offset += 32
+                y_offset += 32
 
     def draw(self, surface):
         surface.blit(self.surface, self.rect)