TychesGame/button.py at master · RealTheBeastBoss/TychesGame · 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
from game import *


class Button:
    def __init__(self, text, x_pos, y_pos, height, colour_one = BLUE, colour_two = ORANGE, font = SMALL_FONT, width = 0):
        self.text = text
        self.xPos = x_pos
        self.yPos = y_pos
        self.height = height
        self.font = font
        self.colourOne = colour_one
        self.colourTwo = colour_two
        self.width = width
        self.draw()

    def draw(self):
        if self.width == 0:
            button_text = self.font.render(self.text, True, self.colourOne)
            self.width = button_text.get_width() + 20
        button_rect = pygame.rect.Rect((self.xPos - (self.width/2), self.yPos - (self.height/2)), (self.width, self.height))
        if not self.check_hover():
            button_text = self.font.render(self.text, True, self.colourTwo)
            pygame.draw.rect(WINDOW, self.colourOne, button_rect, 0, 5)
            pygame.draw.rect(WINDOW, self.colourTwo, button_rect, 3, 5)
        else:
            button_text = self.font.render(self.text, True, self.colourOne)
            pygame.draw.rect(WINDOW, self.colourTwo, button_rect, 0, 5)
            pygame.draw.rect(WINDOW, self.colourOne, button_rect, 3, 5)
        text_width_offset = button_text.get_width() / 2
        text_height_offset = button_text.get_height() / 2
        WINDOW.blit(button_text, ((self.xPos - text_width_offset), self.yPos - text_height_offset))

    def check_hover(self):
        mouse_pos = pygame.mouse.get_pos()
        button_rect = pygame.rect.Rect((self.xPos - (self.width/2), self.yPos - (self.height/2)), (self.width, self.height))
        if button_rect.collidepoint(mouse_pos):
            return True
        else:
            return False

    def check_click(self):
        button_rect = pygame.rect.Rect((self.xPos - (self.width/2), self.yPos - (self.height/2)), (self.width, self.height))
        mouse_pos = pygame.mouse.get_pos()
        if Game.LEFT_MOUSE_RELEASED and button_rect.collidepoint(mouse_pos) and Game.BUTTONS_ENABLED:
            pygame.time.set_timer(BUTTON_COOLDOWN_EVENT, 100, 1)
            Game.BUTTONS_ENABLED = False
            return True
        return False