From ea2a1177163647f1188edfa653f046348532ca60 Mon Sep 17 00:00:00 2001 From: Simon Cross Date: Sat, 10 Sep 2016 21:06:30 +0200 Subject: [PATCH] Add count down bar. --- tabakrolletjie/infobar.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/tabakrolletjie/infobar.py b/tabakrolletjie/infobar.py index 58aa3b6..ac9f1a9 100644 --- a/tabakrolletjie/infobar.py +++ b/tabakrolletjie/infobar.py @@ -54,3 +54,41 @@ class InfoBar(object): if text != self._text: self._text = text self._update_infobar() + + +class CountDownBar(object): + """ Count Down Bar. """ + + def __init__(self, units="h", color=(255, 255, 255), pos=(1024, 10)): + self.units = units + self.color = color + self.pos = pos + self.template = "{remaining:d}{units}" + self._font = loader.load_font(FONTS['sans'], size=48) + self._text = None + self._countbar = None + self._dest = None + + def render(self, surface, remaining): + self._update(remaining) + surface.blit(self._countbar, self._dest, None) + + def _update_countbar(self): + text_img = self._font.render(self._text, True, self.color) + width = text_img.get_width() + 10 + height = text_img.get_height() + 10 + self._countbar = pygame.surface.Surface( + (width, height), pgl.SWSURFACE).convert_alpha() + self._countbar.fill((0, 0, 0, 64)) + self._countbar.blit(text_img, (5, 3), None) + self._dest = (self.pos[0] - width, self.pos[1]) + + def _update(self, remaining): + options = { + "remaining": remaining, + "units": self.units, + } + text = self.template.format(**options) + if text != self._text: + self._text = text + self._update_countbar() -- 2.34.1