X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Ftext.py;h=899dacec7c095cea5a79540d2f71b75cd2f41232;hb=343dbbaa8126ed72ffec8cb4a3ff45050fd5d06a;hp=fb0c31fd507e3ba3843c7b464c7f052c4b4e566d;hpb=abead158fe29ece7a0bc08b09b7427c0caf47599;p=naja.git

diff --git a/naja/widgets/text.py b/naja/widgets/text.py
index fb0c31f..899dace 100644
--- a/naja/widgets/text.py
+++ b/naja/widgets/text.py
@@ -1,6 +1,6 @@
 import pygame
 
-from naja.constants import FONT, FONT_SIZE, EIGHT_BIT_SCALE
+from naja.constants import FONT, FONT_SIZE, EIGHT_BIT_SCALE, PALETTE
 from naja.resources import resources
 from naja.resources.mutators import EIGHT_BIT, blender
 from naja.utils import convert_colour
@@ -8,13 +8,27 @@ from naja.widgets.base import Widget
 
 
 MARKUP_MAP = {
-    'NORTH': (1, 'glyphs/arrow_up.png'),
-    'SOUTH': (1, 'glyphs/arrow_down.png'),
-    'EAST': (1, 'glyphs/arrow_right.png'),
-    'WEST': (1, 'glyphs/arrow_left.png'),
+    'NORTH': ('glyphs/arrow_up.png', None),
+    'SOUTH': ('glyphs/arrow_down.png', None),
+    'EAST': ('glyphs/arrow_right.png', None),
+    'WEST': ('glyphs/arrow_left.png', None),
+    'HEALTH': ('glyphs/health.png', PALETTE.DARK_RED),
+    'WINTOKEN': ('glyphs/win.png', PALETTE.DARK_OLIVE),
+    'KEY': ('glyphs/key.png', None),
+    'MSB': ('glyphs/msb.png', None),
+    'REDKEY': ('glyphs/key.png', PALETTE.ORANGE),
+    'GREENKEY': ('glyphs/key.png', PALETTE.GREEN),
+    'BLUEKEY': ('glyphs/key.png', PALETTE.BLUE),
 }
 
 
+class Glyph(object):
+    def __init__(self, markup_text, glyph_keys, suffix=''):
+        self.markup_text = markup_text
+        self.glyph_keys = glyph_keys
+        self.text = ' ' * len(self.glyph_keys) + suffix
+
+
 class TextWidget(Widget):
 
     def __init__(self, pos, text, size=None, fontname=None, fontsize=None,
@@ -24,7 +38,7 @@ class TextWidget(Widget):
         self.text = text
         self.fontname = fontname or FONT
         self.fontsize = (fontsize or FONT_SIZE) // EIGHT_BIT_SCALE
-        self.colour = convert_colour(colour or (0, 0, 0))
+        self.colour = convert_colour(colour or PALETTE.BLACK)
 
     def render_line(self, text):
         text_surf = self.font.render(text, True, self.colour)
@@ -44,12 +58,12 @@ class TextWidget(Widget):
 
 class TextBoxWidget(TextWidget):
     def __init__(self, *args, **kwargs):
-        self.padding = kwargs.pop('padding', 5)
+        self.padding = kwargs.pop('padding', 4)
         self.border = kwargs.pop('border', 2)
         self.bg_colour = convert_colour(kwargs.pop('bg_colour',
-                                                   (255, 255, 255, 192)))
+                                                   PALETTE.LIGHT_VIOLET))
         self.border_colour = convert_colour(kwargs.pop('border_colour',
-                                                       (0, 0, 0)))
+                                                       PALETTE.BLACK))
         self.box_width = kwargs.pop('box_width', 0)
 
         super(TextBoxWidget, self).__init__(*args, **kwargs)
@@ -60,6 +74,34 @@ class TextBoxWidget(TextWidget):
         else:
             return self.text.splitlines()
 
+    def _prepare_glyph(self, image_map, glyph, current_words, lines):
+        size = self.font.size(' '.join(current_words[:-1] + ['']))
+        x = size[0] * EIGHT_BIT_SCALE + self.padding
+        y = size[1] * lines * EIGHT_BIT_SCALE + self.padding
+        for glyph_key in glyph.glyph_keys:
+            image_name, colour = MARKUP_MAP[glyph_key]
+            if colour is None:
+                colour = self.colour
+            image = resources.get_image(
+                image_name, transforms=(EIGHT_BIT, blender(colour)))
+            image_map[(x, y)] = image
+            x += image.get_width()
+
+    def _check_markup(self, word):
+        suffix = ''
+        if word[-1] in '.,':
+            suffix = word[-1]
+            word = word[:-1]
+
+        if word[0] == '{' and word[-1] == '}':
+            subwords = word[1:-1].split(',')
+            if all(subword in MARKUP_MAP for subword in subwords):
+                return Glyph(word + suffix, subwords, suffix)
+        elif word in MARKUP_MAP:
+            return Glyph(word + suffix, [word], suffix)
+
+        return None
+
     def _wrapped_lines(self, image_map):
         def words_fit(words):
             words_line = ' '.join(words)
@@ -73,31 +115,24 @@ class TextBoxWidget(TextWidget):
         line_count = 0
         for line in self.text.splitlines():
             current_words = []
-            for word in line.split():
-                suffix = ''
-                if word[-1] in '.,':
-                    suffix = word[-1]
-                    word = word[:-1]
-                markup_data = MARKUP_MAP.get(word, None)
-                if markup_data is not None:
-                    word = ' ' * markup_data[0]
-                word += suffix
+            remaining_words = line.split()
+            while remaining_words:
+                word = remaining_words.pop(0)
+                glyph = self._check_markup(word)
+                if glyph is not None:
+                    word = glyph.text
                 current_words.append(word)
                 if words_fit(current_words):
-                    if markup_data is not None:
-                        size = self.font.size(
-                            ' '.join(current_words[:-1]) + ' ')
-                        pos = (size[0] * EIGHT_BIT_SCALE,
-                               size[1] * line_count * EIGHT_BIT_SCALE)
-                        pos = (pos[0] + self.padding, pos[1] + self.padding)
-                        image_map[pos] = resources.get_image(
-                            markup_data[1],
-                            transforms=(EIGHT_BIT, blender(self.colour)))
-                        continue
+                    if glyph is not None:
+                        self._prepare_glyph(
+                            image_map, glyph, current_words, line_count)
                 else:
                     line_count += 1
                     yield ' '.join(current_words[:-1])
-                    current_words = current_words[-1:]
+                    current_words = []
+                    if glyph is not None:
+                        word = glyph.markup_text
+                    remaining_words.insert(0, word)
             if current_words and words_fit(current_words):
                 yield ' '.join(current_words)