TetrisPyGame/menu.py at main · Listex-Dev/TetrisPyGame · 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
import os
import sounds
import pygame as pg


class Menu:
    margin_x = 0
    margin_y = 50  # space between items(y axis)
    text_color = (255, 255, 255)
    selected_color = (0, 255, 255)

    def __init__(self, title, background_path, screen_res, font, coord):
        self.title = title
        self.background = pg.image.load(background_path).convert()
        self.background = pg.transform.scale(self.background, screen_res)

        self.items = []
        self.selected_item = 0

        self.title_font = pg.font.Font(os.path.join('fonts', font), 58)
        self.items_font = pg.font.Font(os.path.join('fonts', font), 46)

        self.title_coord = tuple(coord)
        self.items_coord = (coord[0] + Menu.margin_x, coord[1] + Menu.margin_y)

    def add_item(self, item):
        self.items.append(item)

    def show(self, screen):
        """
        Blit all menu items.
        :param screen: screen surface
        """
        screen.blit(self.background, (0, 0))
        title_surface = write(self.title_font, self.title, self.text_color)
        screen.blit(title_surface, (self.title_coord[0] - title_surface.get_width()
                                    / 2, self.title_coord[1]))

        # blit menu items
        offset_y = Menu.margin_y
        for i in range(len(self.items)):
            if i == self.selected_item:
                color = self.selected_color
            else:
                color = self.text_color

            item_surface = write(self.items_font, self.items[i], color)
            screen.blit(item_surface, (self.items_coord[0] - item_surface.get_width()
                                       / 2, self.items_coord[1] + offset_y))
            offset_y += Menu.margin_y

    def check_input(self, event):
        """
        Check keyboard input and change selected item
        :param event: pygame.event
        :return: selected item name if an item is selected, else None
        """
        if event.type == pg.KEYDOWN and event.key == pg.K_UP:
            sounds.menu_move_sound.play()
            if self.selected_item == 0:
                self.selected_item = len(self.items) - 1
            else:
                self.selected_item -= 1
        elif event.type == pg.KEYDOWN and event.key == pg.K_DOWN:
            sounds.menu_move_sound.play()
            self.selected_item = (self.selected_item + 1) % len(self.items)
        elif event.type == pg.KEYDOWN and event.key == pg.K_RETURN:
            sounds.select_sound.play()
            return self.items[self.selected_item]

        return None


def write(font, message, color):
    text = font.render(str(message), True, color)
    text = text.convert_alpha()
    return text