MusicScalesApp/RootNoteMenu.py at main · MrToastMachine/MusicScalesApp · 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
import pygame
from button import Button
from AppManager import ALL_NOTES
from AppManager import FONT
import AppManager
from colours import Colours

PADDING = 50

SECT_PADDING = AppManager.SECTION_PADDING

class RootNoteMenu():
    def __init__(self, pos, dimensions, button_size, bg_col, butt_colour, butt_colour_highlighted, is_vertical):

        self.xPos, self.yPos = pos
        self.width, self.height = dimensions
        self.button_size = button_size
        self.is_vertical = is_vertical

        self.button_area_width = self.width - 2*PADDING
        self.button_area_height = self.height - 2*PADDING

        self.all_buttons = []

        self.padding = PADDING
        self.y_padding = (self.height - self.button_size) / 2

        self.bg_colour = bg_col
        self.butt_colour = butt_colour
        self.butt_colour_highlighted = butt_colour_highlighted

        if self.is_vertical:
            self.createVerticalMenu()
        else:
            self.createHorizontalMenu()

    # Designate positions of all buttons
    # Called once at instantiation
    def createHorizontalMenu(self):
        self.button_gap = (self.button_area_width - 12*self.button_size)/11

        #button_area
        self.rect = (self.xPos + SECT_PADDING/2, self.yPos + SECT_PADDING/2, self.width - SECT_PADDING, self.height - SECT_PADDING)
        for i, note in enumerate(ALL_NOTES):
            xStart = self.padding + i*(self.button_size + self.button_gap)
            yStart = self.y_padding + self.yPos

            rect = pygame.Rect(xStart, yStart, self.button_size, self.button_size)

            button = Button(rect,note, FONT, self.butt_colour, Colours.BLACK, None)

            self.all_buttons.append(button)

    def createVerticalMenu(self):
        self.button_gap = (self.button_area_height - 12*self.button_size)/11

        self.rect = (self.xPos + SECT_PADDING/2, self.yPos + SECT_PADDING/2, self.width - SECT_PADDING, self.height - SECT_PADDING)
        for i, note in enumerate(ALL_NOTES):
            xStart = self.padding + i*(self.button_size + self.button_gap)
            yStart = self.y_padding + self.yPos

            rect = pygame.Rect(xStart, yStart, self.button_size, self.button_size)

            button = Button(rect,note, FONT, self.butt_colour, Colours.BLACK, None)

            self.all_buttons.append(button)

    def drawMenu(self, screen):
        pygame.draw.rect(screen, self.bg_colour, self.rect, border_radius=AppManager.SECTION_CORNER_RADIUS)
        for butt in self.all_buttons:
            if butt.text_input == AppManager.ACTIVE_ROOT:
                butt.base_colour = self.butt_colour_highlighted
            else:
                butt.base_colour = self.butt_colour

            butt.update(screen)

    def checkAllButtonsForInput(self, mousePos):
        for butt in self.all_buttons:
            if butt.checkForInput(mousePos):
                AppManager.ACTIVE_ROOT = butt.text_input
                print(butt.text_input)
                return True