abc_trainer/strings_on_screen.py at main · Wartem/abc_trainer · 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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import pygame

from color_constants import ColorConst
from fonts import get_pygame_font


class StringImage(pygame.sprite.Sprite):
    def __init__(self, font_path: str, letter_str: str, screen: pygame.Surface, font_size = 1800, x: int = 0, y: int = 0, bg_color = ColorConst.BEIGE, text_color = ColorConst.BLACK):
        self.text = letter_str
        self.screen = screen
        self.screen_rect = self.screen.get_rect()
        self.x = x
        self.y = y
        self.font_size = font_size
        self.font = None
        self.font_path = font_path
        self.background_color = bg_color
        self.text_color = text_color
        self.move = lambda x: self.rect.move_ip(*x)
        self.img = None
        self.rect = None

        self.update_text_size()


    def update_text_size(self):
        self.font = get_pygame_font(self.font_path, self.font_size)
        self.img = self._generate_img()
        self.img = self.scale_img(self.img)
        self.rect = self.img.get_rect()
        self.center_text(self.screen)


    def center_text(self, screen):
        self.screen = screen
        self.screen_rect = self.screen.get_rect()
        self.rect.x = self.screen_rect.centerx - self.rect.centerx
        self.rect.y = self.screen_rect.centery - self.rect.centery


    def _draw(self):
        self.screen.fill(self.background_color)

        self.screen.blit(
            self.img,
            (
                self.rect.x,
                self.rect.y,
            ),
        )


    def update(self):
        self._draw()


    def scale_img(self, image_surface: pygame.Surface) -> pygame.Surface:
        """Scales the image to fit the screen.

        Args:
            image_surface (pygame.Surface): The image to be scaled.

        Returns:
            _type_: Scaled version of the image.
        """
        screen_info = pygame.display.Info()
        screen_width = screen_info.current_w - 50
        screen_height = screen_info.current_h - 50

        surface = image_surface
        surface_width, surface_height = surface.get_size()

        scale_x = screen_width / surface_width if surface_width else screen_width
        scale_y = screen_height / surface_height if surface_height else screen_height

        scale = min(scale_x, scale_y)

        return pygame.transform.scale(
            surface, (int(surface_width * scale), int(surface_height * scale))
            #surface, (int(surface_width * 0.5), int(surface_height * 0.5))
        ).convert_alpha()


    def _generate_img(self):
        return self.font.render(
            self.text, True, self.text_color
        )