tabakrolletjie/utils.py

   1 """ Tabakutilities. """
   2
   3 import functools
   4 import time
   5 import pygame.surface
   6 import pygame.locals as pgl
   7
   8 from .constants import DEBUG
   9 from .loader import loader
  10
  11
  12 def debug_timer(label, debug=False):
  13     """ A decorator for printing how long a function took if debug is true.
  14     """
  15     def debug_inner(f):
  16         @functools.wraps(f)
  17         def wrapper(*args, **kw):
  18             start_time = time.time()
  19             try:
  20                 return f(*args, **kw)
  21             finally:
  22                 duration = time.time() - start_time
  23                 if DEBUG or debug:
  24                     print "%s [%g seconds]" % (label, duration)
  25         return wrapper
  26     return debug_inner
  27
  28
  29 class DetailedTimer(object):
  30     """ A detailed timer for overly complex functions.
  31     """
  32     def __init__(self, title, debug=False):
  33         if DEBUG or debug:
  34             self.title = title
  35             self.times = []
  36         else:
  37             # fast funtions with the correct signature
  38             self.start = int
  39             self.show = str
  40             self.lap = str
  41             self.end = int
  42
  43     def start(self):
  44         print "---- %s ----" % self.title
  45         self.times.append(time.time())
  46
  47     def lap(self, label):
  48         now = time.time()
  49         print "  %s: %s" % (label, now - self.times[-1])
  50         self.times.append(now)
  51
  52     def show(self, text):
  53         print "  %s" % text
  54
  55     def end(self):
  56         print "  %s: %s" % ("total", time.time() - self.times[0])
  57         print "---- ----"
  58
  59
  60 def shadowed_text(text, font_name, size, offset=4):
  61     font_black = loader.load_font(font_name, size=size)
  62     font_white = loader.load_font(font_name, size=size)
  63     background = font_black.render(text, True, (0, 0, 0))
  64     foreground = font_white.render(text, True, (255, 255, 255))
  65     new_size = (background.get_width() + offset,
  66                 background.get_height() + offset)
  67     base = pygame.surface.Surface(new_size, pgl.SWSURFACE).convert_alpha()
  68     base.fill((0, 0, 0, 0))
  69     base.blit(background, (offset, offset), None)
  70     base.blit(foreground, (0, 0), None)
  71     return base