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
|
import pygame
from color import Color
from text import Label
class Button(Label):
cooldown = 0.25 # seconds
def __init__(self, x, y, text, font, size, text_color=Color.BLACK, background=Color.WHITE, border_color=Color.BLACK,
border_width=1, border_radius=0, padding=10, darken=True, func=None):
super().__init__(x, y, text, font, size, text_color, background, border_color, border_width, border_radius,
padding)
self.is_hover = False
self.is_pressed = False
self.is_clicked = False
self.func = func
if darken:
self.hover_color = Color.darken(self.background_color, 20)
self.pressed_color = Color.darken(self.background_color, 40)
else:
self.hover_color = Color.lighten(self.background_color, 20)
self.pressed_color = Color.lighten(self.background_color, 40)
self.last_click = 0
self.still_pressed = False
self.click_disabled = False
def update(self):
mouse_pressed = pygame.mouse.get_pressed()[0]
mouse_pos = pygame.mouse.get_pos()
self.is_hover = self.rect.collidepoint(mouse_pos)
if mouse_pressed and not self.is_hover:
self.click_disabled = True
if self.click_disabled:
if not mouse_pressed:
self.click_disabled = False
else:
if self.is_hover:
self.rect.set_background_color(self.hover_color)
else:
self.rect.set_background_color(self.background_color)
return
self.is_clicked = self.is_hover and not mouse_pressed and self.is_pressed
self.is_pressed = self.is_hover and mouse_pressed
if self.still_pressed and not self.is_pressed:
self.still_pressed = False
if self.func is not None and self.is_hover:
self.func()
if self.is_pressed:
self.rect.set_background_color(self.pressed_color)
if pygame.time.get_ticks() - self.last_click > self.cooldown * 1000 and not self.still_pressed:
self.last_click = pygame.time.get_ticks()
self.still_pressed = True
elif self.is_hover:
self.rect.set_background_color(self.hover_color)
else:
self.rect.set_background_color(self.background_color)
def draw(self, screen):
self.rect.draw(screen)
super().draw(screen)
|
