pedantic correntions
[naja.git] / naja / widgets / tile.py
index c9326670251d2988704aa2e35d0a7829048682f9..cf6f04c92d4ef498163dc1f694ff6255fca46af7 100644 (file)
@@ -34,10 +34,19 @@ class TileWidget(Widget):
         # Draw background
         x, y = abs(self.board_pos[0] - 2), abs(self.board_pos[1] - 2)
 
+        if self.state.gameboard.puzzle:
+            tile_1_name = 'board/tile_1.png'
+            tile_2_name = 'board/tile_2_puzzle.png'
+            tile_available_name = 'board/tile_available_puzzle.png'
+        else:
+            tile_1_name = 'board/tile_1.png'
+            tile_2_name = 'board/tile_2.png'
+            tile_available_name = 'board/tile_available.png'
+
         if (x + y) % 2 == 0:
-            bg_name = 'board/tile_2.png'
+            bg_name = tile_2_name
         else:
-            bg_name = 'board/tile_1.png'
+            bg_name = tile_1_name
         bg = resources.get_image(bg_name, transforms=(EIGHT_BIT,))
         overlays = []
 
@@ -45,7 +54,7 @@ class TileWidget(Widget):
 
         if self.state.gameboard.player_mode == EXAMINE and legal_move:
             overlays.append(resources.get_image(
-                'board/tile_available.png', transforms=(EIGHT_BIT,)))
+                tile_available_name, transforms=(EIGHT_BIT,)))
         if self.highlighted:
             overlays.append(resources.get_image(
                 'board/tile_selected.png',
@@ -65,8 +74,27 @@ class TileWidget(Widget):
         for action in card.actions:
             y_offset = self._prepare_action(action, y_offset)
 
+        self._prepare_countdown(card)
+
+    def _prepare_countdown(self, card):
+        if card.replacement_time is None:
+            return
+        elif card.replacement_time <= 1:
+            glyph = 'glyphs/countdown_1.png'
+        elif card.replacement_time == 2:
+            glyph = 'glyphs/countdown_2.png'
+        elif card.replacement_time == 3:
+            glyph = 'glyphs/countdown_3.png'
+        elif card.replacement_time < 8:
+            glyph = 'glyphs/countdown_4.png'
+        else:
+            glyph = 'glyphs/countdown_5.png'
+        img = resources.get_image(
+            glyph, transforms=(EIGHT_BIT, blender(PALETTE.DARK_VIOLET)))
+        self.surface.blit(img, (TILE_SIZE[0] - 20, 0))
+
     def _prepare_lock(self, action, y_offset):
-        if not action.required_bits:
+        if action.required_bits not in BIT_MAP:
             return 4
 
         img_name = BIT_MAP[action.required_bits]