FWMM/text_based_widget.py at main · DedFishy/FWMM · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
from widget import Widget as WidgetBase
import numpy as np
from font_loader import load_preloaded_font, get_fonts


class TextBasedWidget(WidgetBase):
    """
    A helper class for creating widgets that are composed entirely of text.
    To use this widget, just override get_text, get_font, get_brightness, get_spacing, and optionally get_desired_spf.
    This class will deal with font loading and rendering of text.
    """
    name = "Text Based Widget"
    allow_rotation = True
    current_render: np.matrix = np.matrix([[]])
    rotation = 0
    font = None
    loaded_font_name = None

    def __init__(self):
        super().__init__()
        self.configuration = {}

    def get_text(self): return ""
    def get_font(self): return get_fonts()[0]
    def get_brightness(self): return 255
    def get_spacing(self): return 1

    def get_current_size(self) -> list[int]:
        return [self.current_render.shape[1], self.current_render.shape[0]]

    def get_desired_spf(self) -> int:
        return -1

    def append_character(self, matrix: np.matrix, character: np.matrix, spacing=1) -> np.matrix:
        """Add a character to the given matrix and return the edited matrix"""
        spacer = [
            [0 for _ in range(0, spacing)] for _ in range(0, matrix.shape[0])
        ]
        matrix = np.concatenate((matrix, np.matrix(spacer)), axis=1) # type: ignore
        matrix = np.concatenate((matrix, character), axis=1) # type: ignore
        return matrix

    def render(self):
        """Render the text and return it as an array"""
        if self.font == None or self.loaded_font_name != self.get_font():
            self.loaded_font_name = self.get_font()
            self.font = load_preloaded_font(self.loaded_font_name)
        rows = np.matrix([
            [] for _ in range(0, self.font.get_char_height())
        ])
        i = 0
        for char in self.get_text():
            try:
                if char == " ":
                    rows = self.append_character(rows, [[0 for _ in range(0, self.font.get_char_width())] for _ in range(0, self.font.get_char_height())], spacing= 0 if i == 0 else self.get_spacing())
                else:
                    rows = self.append_character(rows, self.font[char.lower()], spacing= 0 if i == 0 else self.get_spacing())
            except KeyError: # If the character isn't present...
                try: # ...try to render a ? in it's place...
                    rows = self.append_character(rows, self.font["?"], spacing= 0 if i == 0 else self.get_spacing())
                except KeyError: # ...unless that doesn't exist either, then just give up
                    pass
            i += 1

        for y in range(0, len(rows)):
            for x in range(0, len(rows[y])):
                rows[y][x] *= self.get_brightness()
        self.current_render = rows
        return rows

    def get_frame(self):
        return self.render()