naja/utils.py

   1 import pygame
   2
   3 from naja.constants import BITS
   4
   5
   6 def convert_colour(colour):
   7     if isinstance(colour, pygame.Color):
   8         return colour
   9     if isinstance(colour, tuple):
  10         return pygame.Color(*colour)
  11     if isinstance(colour, basestring):
  12         return pygame.Color(colour)
  13     raise ValueError()
  14
  15
  16 def bit_glyphs(bits):
  17     bit_names = dict((v, k) for k, v in BITS.items())
  18     return '{%s}' % ','.join(bit_names[bit] for bit in reversed(range(8))
  19                              if bit in bits)
  20
  21
  22 def move_glyph(move_name):
  23     return {
  24         'CASTLE': u'\u265c',
  25         'BISHOP': u'\u265d',
  26         'KNIGHT': u'\u265e',
  27     }.get(move_name, move_name.lower())
  28
  29
  30 def parse_bits(bit_list):
  31     # Convert names to numbers if applicable.
  32     return frozenset(BITS.get(bit, bit) for bit in bit_list)
  33
  34
  35 def warp_to_game_state(game_state):
  36     from naja.events import LoadGameEvent, SceneChangeEvent
  37     from naja.scenes.game import GameScene
  38     LoadGameEvent.post(game_state)
  39     SceneChangeEvent.post(GameScene)
  40
  41
  42 class Flashlight(object):
  43     def __init__(self, rate):
  44         self.rate = rate
  45         self.reset()
  46
  47     def tick(self):
  48         self.ticks += 1
  49         if self.ticks >= self.rate:
  50             self.on = not self.on
  51             self.ticks = 0
  52             return True
  53         return False
  54
  55     def reset(self):
  56         self.ticks = 0
  57         self.on = True