105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
#!/usr/bin/python3
|
|
|
|
from wobbl_tools.pygame_tools.utils import *
|
|
|
|
|
|
class MultilineText:
|
|
"""
|
|
Creates a surface with text on it.
|
|
You can use "\\n" to create a newline.
|
|
When the max_width parameter is set, newlines generate automatically.
|
|
If you know the width of the widest character in your font, you can set the char_width parameter,
|
|
it will make the text creation a bit faster.
|
|
Use "surface.blit(multiline_text.surface, pos)" to draw it on a surface.
|
|
"""
|
|
def __init__(self, text: str, font: pygame.font.Font=default_font, color: tuple=white, max_width: int=None):
|
|
self.text = text
|
|
self.font = font
|
|
self.color = color
|
|
self.max_width = max_width
|
|
|
|
self.surface = self.generate_surface()
|
|
|
|
def generate_surface(self):
|
|
lines = []
|
|
line = ""
|
|
|
|
length = 1
|
|
|
|
if self.max_width is None:
|
|
for char in self.text:
|
|
if char == "\n":
|
|
lines.append(line)
|
|
length = 1
|
|
line = ""
|
|
|
|
else:
|
|
line += char
|
|
|
|
length += 1
|
|
|
|
else:
|
|
for char in self.text:
|
|
if char == "\n":
|
|
lines.append(line)
|
|
|
|
length = 1
|
|
line = ""
|
|
|
|
elif self.font.size(line)[0] >= self.max_width - 8:
|
|
if " " in line:
|
|
# remove last word
|
|
|
|
words = line.split(" ")
|
|
|
|
last_word = words[len(words) - 1]
|
|
words.pop(len(words) - 1)
|
|
|
|
line = ""
|
|
|
|
for word in words: # convert the list back to a string
|
|
line += word + " "
|
|
|
|
line = line[:len(line) - 1] # remove last space character
|
|
|
|
lines.append(line)
|
|
|
|
line = last_word + char
|
|
|
|
else:
|
|
lines.append(line)
|
|
|
|
length = 1
|
|
line = char
|
|
|
|
else:
|
|
line += char
|
|
|
|
length += 1
|
|
|
|
lines.append(line)
|
|
|
|
texts = []
|
|
|
|
for line in lines:
|
|
texts.append(self.font.render(line, True, self.color))
|
|
|
|
if self.max_width is None:
|
|
w = max(texts, key=lambda t: t.get_width()).get_width()
|
|
|
|
else:
|
|
w = self.max_width
|
|
|
|
h = texts[0].get_height()
|
|
sh = h * len(lines)
|
|
|
|
surface = pygame.Surface((w, sh), flags=pygame.SRCALPHA)
|
|
|
|
x, y = 0, 0
|
|
|
|
for text in texts:
|
|
surface.blit(text, (x, y))
|
|
|
|
y += h
|
|
|
|
return surface
|