X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Ftext.py;h=d753e8a12aed87987e9f6f273e861d678eda5612;hb=0cef06f581d48315a59a19ac06ca1d2be14cbb4c;hp=53030c55b8ad730c4a7725f07dc1bf9c5dcf1f52;hpb=d7291141b604ffc7941074ae1adb9596c5f314ce;p=naja.git diff --git a/naja/widgets/text.py b/naja/widgets/text.py index 53030c5..d753e8a 100644 --- a/naja/widgets/text.py +++ b/naja/widgets/text.py @@ -1,28 +1,98 @@ import pygame -from naja.constants import FONT, FONT_SIZE +from naja.constants import FONT, FONT_SIZE, EIGHT_BIT_SCALE from naja.widgets.base import Widget from naja.resources import resources from naja.utils import convert_colour class TextWidget(Widget): + def __init__(self, pos, text, size=None, fontname=None, fontsize=None, colour=None): super(TextWidget, self).__init__(pos, size) self.text = text self.fontname = fontname or FONT - self.fontsize = (fontsize or FONT_SIZE) / 2 + self.fontsize = (fontsize or FONT_SIZE) // EIGHT_BIT_SCALE self.colour = convert_colour(colour or (0, 0, 0)) + def render_line(self, text): + text_surf = self.font.render(text, True, self.colour) + text_rect = text_surf.get_rect() + return pygame.transform.scale( + text_surf, (text_rect.width * EIGHT_BIT_SCALE, + text_rect.height * EIGHT_BIT_SCALE)) + + def prepare(self): + self.font = resources.get_font(self.fontname, self.fontsize) + self.surface = self.render_line(self.text) + self.size = self.surface.get_rect().size + + def draw(self, surface): + surface.blit(self.surface, self.pos) + + +class TextBoxWidget(TextWidget): + def __init__(self, *args, **kwargs): + self.padding = kwargs.pop('padding', 5) + self.border = kwargs.pop('border', 2) + self.bg_colour = convert_colour(kwargs.pop('bg_colour', + (255, 255, 255, 192))) + self.border_colour = convert_colour(kwargs.pop('border_colour', + (0, 0, 0))) + self.box_width = kwargs.pop('box_width', 0) + + super(TextBoxWidget, self).__init__(*args, **kwargs) + + def lines(self): + if self.box_width != 0: + return self._wrapped_lines() + else: + return self.text.splitlines() + + def _wrapped_lines(self): + def words_fit(words): + words_line = ' '.join(words) + width = self.font.size(words_line)[0] + if width < self.box_width: + return True + elif len(words) == 1: + Exception("Word %r too long for box." % (words[0],)) + return False + + for line in self.text.splitlines(): + current_words = [] + for word in line.split(): + current_words.append(word) + if words_fit(current_words): + continue + else: + yield ' '.join(current_words[:-1]) + current_words = current_words[-1:] + if current_words and words_fit(current_words): + yield ' '.join(current_words) + def prepare(self): self.font = resources.get_font(self.fontname, self.fontsize) - text = self.font.render(self.text, True, self.colour) - text_rect = text.get_rect() - self.surface = pygame.transform.scale(text, (text_rect.width * 2, - text_rect.height * 2)) + rendered_lines = [] + width, height = self.padding, self.padding + for line in self.lines(): + line_surface = self.render_line(line) + line_rect = line_surface.get_rect() + rendered_lines.append(line_surface) + width = max(width, line_rect.width + self.padding) + height += line_rect.height + + self.surface = pygame.surface.Surface((width, height), + pygame.locals.SRCALPHA) + self.surface.fill(self.bg_colour) self.size = self.surface.get_rect().size + x, y = self.padding, self.padding + for line_surface in rendered_lines: + self.surface.blit(line_surface, (x, y)) + y += line_surface.get_rect().height + def draw(self, surface): surface.blit(self.surface, self.rect)