naja/widgets/game_bits.py

   1 """
   2 Widget that holds the games's bits.
   3 """
   4
   5 import pygame
   6
   7 from naja.constants import BIT_SIZE, BITS
   8 from naja.resources import resources
   9 from naja.widgets.base import Widget
  10
  11
  12 class GameBitsWidget(Widget):
  13     """
  14     Widget which holds the game's bits.
  15     """
  16     def __init__(self, pos, state):
  17         super(GameBitsWidget, self).__init__(pos, BIT_SIZE)
  18         self.state = state
  19
  20     def prepare(self):
  21         self.surface = pygame.Surface(BIT_SIZE)
  22         health = self.state.gameboard.health
  23         wins = self.state.gameboard.wins
  24
  25         bits = []
  26         bits.extend(
  27             (i < health, 'health', ()) for i in range(4))
  28         bits.extend(
  29             (i < wins, 'win', ()) for i in range(4))
  30
  31         for pos, (is_set, image, transforms) in enumerate(bits):
  32             img = resources.get_image(
  33                 'bits', '%s_%s.png' % (image, 'on' if is_set else 'off'),
  34                 transforms=transforms)
  35             self.surface.blit(img, (img.get_width() * pos, 0))
  36
  37     def draw(self, surface):
  38         surface.blit(self.surface, self.pos)