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

diff --git a/naja/widgets/text.py b/naja/widgets/text.py
index 4ddc0d6..899dace 100644
--- a/naja/widgets/text.py
+++ b/naja/widgets/text.py
@@ -8,17 +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'),
-    'HEALTH': (1, 'glyphs/health.png'),
-    'WIN': (1, 'glyphs/win.png'),
-    'KEY': (1, 'glyphs/key.png'),
-    'MSB': (1, 'glyphs/msb.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,
@@ -48,7 +58,7 @@ 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',
                                                    PALETTE.LIGHT_VIOLET))
@@ -64,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)
@@ -79,31 +117,22 @@ class TextBoxWidget(TextWidget):
             current_words = []
             remaining_words = line.split()
             while remaining_words:
-                word = remaining_words[0]
-                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
+                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)))
-                    remaining_words.pop(0)
+                    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 = []
+                    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)