X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Ftext.py;h=de77a60d7320c993644ae6a958220a9d6097c9f3;hb=a1c561e26fae79ce8a9570c9404975a8a0367d65;hp=3df09488e3a170502426b79d7724b196d2d22c32;hpb=c7c0aeb2347992dfb69844700711bd0ded2caf0c;p=naja.git

diff --git a/naja/widgets/text.py b/naja/widgets/text.py
index 3df0948..de77a60 100644
--- a/naja/widgets/text.py
+++ b/naja/widgets/text.py
@@ -13,6 +13,7 @@ MARKUP_MAP = {
     'SOUTH': ('glyphs/arrow_down.png', None),
     'EAST': ('glyphs/arrow_right.png', None),
     'WEST': ('glyphs/arrow_left.png', None),
+    'RETURN': ('glyphs/arrow_left.png', None),
     'HEALTH': ('glyphs/health.png', PALETTE.DARK_RED),
     'WINTOKEN': ('glyphs/win.png', PALETTE.DARK_OLIVE),
     'KEY': ('glyphs/key.png', None),
@@ -22,6 +23,9 @@ MARKUP_MAP = {
     'BLUE': ('glyphs/key.png', PALETTE.BLUE),
     'CLOCKWISE': ('glyphs/clockwise.png', None),
     'ANTICLOCKWISE': ('glyphs/anticlockwise.png', None),
+    'SHIFT_LEFT': ('glyphs/shift_left.png', None),
+    'SHIFT_RIGHT': ('glyphs/shift_right.png', None),
+    'COUNTDOWN': ('glyphs/countdown_4.png', PALETTE.DARK_VIOLET),
 
     'HEALTH_NOCOLOUR': ('glyphs/health.png', None),
     'WINTOKEN_NOCOLOUR': ('glyphs/win.png', None),
@@ -41,7 +45,8 @@ class TextWidget(Widget):
 
     def __init__(self, pos, text, fontname=None, fontsize=None,
                  colour=None, unselectable_colour=None, view_port=None,
-                 centre=False):
+                 centre=False, border=0, border_colour=PALETTE.GREY,
+                 padding=2):
         super(TextWidget, self).__init__(pos)
 
         self.text = text
@@ -55,6 +60,9 @@ class TextWidget(Widget):
             pygame.Rect((0, 0), view_port) if view_port is not None else None)
         self.centre = centre
         self.centre_pos = pos
+        self.border = border
+        self.border_colour = convert_colour(border_colour)
+        self.padding = padding
 
     def render_line(self, text):
         colour = self.colour
@@ -66,6 +74,13 @@ class TextWidget(Widget):
             text_surf, (text_rect.width * EIGHT_BIT_SCALE,
                         text_rect.height * EIGHT_BIT_SCALE))
 
+    def render_border(self, surface):
+        if not self.border or not self.border_colour:
+            return
+        rect = surface.get_rect()
+        rect = rect.inflate(- self.border / 2, - self.border / 2)
+        pygame.draw.rect(surface, self.border_colour, rect, self.border)
+
     def update_size(self):
         if self.view_port is not None:
             self.size = self.view_port.size
@@ -78,6 +93,7 @@ class TextWidget(Widget):
     def prepare(self):
         self.font = resources.get_font(self.fontname, self.fontsize)
         self.surface = self.render_line(self.text)
+        self.render_border(self.surface)
         self.update_size()
 
     def handle_event(self, ev):
@@ -125,15 +141,11 @@ class TextWidget(Widget):
 
 class TextBoxWidget(TextWidget):
     def __init__(self, *args, **kwargs):
-        self.padding = kwargs.pop('padding', 4)
-        self.border = kwargs.pop('border', 2)
         self.bg_colour = convert_colour(kwargs.pop('bg_colour',
                                                    PALETTE.LIGHT_VIOLET))
-        self.border_colour = convert_colour(kwargs.pop('border_colour',
-                                                       PALETTE.BLACK))
         self.box_width = kwargs.pop('box_width', 0)
-
         self.full_width = kwargs.pop('full_width', True)
+        kwargs.setdefault('padding', 4)
 
         super(TextBoxWidget, self).__init__(*args, **kwargs)
 
@@ -162,6 +174,8 @@ class TextBoxWidget(TextWidget):
             suffix = word[-1]
             word = word[:-1]
 
+        if not word:
+            return None
         if word[0] == '{' and word[-1] == '}':
             subwords = word[1:-1].split(',')
             if all(subword in MARKUP_MAP for subword in subwords):
@@ -238,3 +252,5 @@ class TextBoxWidget(TextWidget):
             y += line_surface.get_rect().height
         for pos, img in image_map.items():
             self.surface.blit(img, pos)
+
+        self.render_border(self.surface)