Added args parameter to all button/hover classes to be able to execute a function with arguments.

This commit is contained in:
The Wobbler 2024-03-22 19:39:42 +01:00
parent 8d4bb14583
commit 7113e6a8c8

85
pg.py
View file

@ -22,9 +22,10 @@ default_font = pygame.font.Font(pygame.font.get_default_font(), 16)
class Hover: class Hover:
def __init__(self, rect: pygame.Rect, function=None): def __init__(self, rect: pygame.Rect, function=None, args: tuple=None):
self.rect = rect self.rect = rect
self.function = function self.function = function
self.args = args
self.active = True self.active = True
@ -34,22 +35,27 @@ class Hover:
if self.rect.collidepoint((mx, my)): if self.rect.collidepoint((mx, my)):
if not self.function is None: if not self.function is None:
if self.args is None:
self.function() self.function()
else:
self.function(*self.args)
return True return True
return False return False
class Button: class Button:
def __init__(self, rect: pygame.Rect, function=None, key: int=0): # key: 0 = left 1 = mouse wheel pressed 2 = right def __init__(self, rect: pygame.Rect, function=None, args: tuple=None, key: int=0): # key: 0 = left 1 = mouse wheel pressed 2 = right
self.rect = rect self.rect = rect
self.function = function self.function = function
self.args = args
self.key = key self.key = key
self.active = True self.active = True
self.hover = Hover(rect, function) self.hover = Hover(rect, function, args)
if buttonlist: if buttonlist:
buttons.append(self) buttons.append(self)
@ -63,12 +69,13 @@ class Button:
class TextButton: class TextButton:
def __init__(self, text: str, position: tuple, function=None, key: int=0, text_color: tuple=white, bg_color: tuple=gray, font: pygame.font.Font=default_font, padding: tuple=(8, 8), border_radius: int=0, line_thickness: int = 0): def __init__(self, text: str, position: tuple, function=None, args: tuple=None, key: int=0, text_color: tuple=white, bg_color: tuple=gray, font: pygame.font.Font=default_font, padding: tuple=(8, 8), border_radius: int=0, line_thickness: int = 0):
self.text = text self.text = text
self.position = position self.position = position
self.text_color = text_color self.text_color = text_color
self.font = font self.font = font
self.function = function self.function = function
self.args = args
self.key = key self.key = key
self.bg_color = bg_color self.bg_color = bg_color
self.padding = padding self.padding = padding
@ -78,7 +85,7 @@ class TextButton:
self.surface, self.size = self.generate_surface() self.surface, self.size = self.generate_surface()
self.rect = self.make_rect() self.rect = self.make_rect()
self.button = Button(self.rect, self.function, self.key) self.button = Button(self.rect, self.function, args, self.key)
def generate_surface(self): def generate_surface(self):
text_object = self.font.render(self.text, True, self.text_color) text_object = self.font.render(self.text, True, self.text_color)
@ -242,6 +249,74 @@ class TextInput:
self.text_object = crop_surface(self.text_object, (t_width - (t_width - (t_width - self.width)), 0), (t_width - (t_width - self.width), t_height)) self.text_object = crop_surface(self.text_object, (t_width - (t_width - (t_width - self.width)), 0), (t_width - (t_width - self.width), t_height))
class MultilineText:
"""
Creates a surface with text on it.
You can use a "\\n" to create a newline.
When the max_width parameter is set, newlines generate automatically.
When you know the width of a single character in your font, yau can set the char_width parameter. It will make the text creation a bit faster.
Use "surface.blit(multiline_text.surface, pos)" to draw it on a surface.
"""
def __init__(self, text: str, font: pygame.font.Font=default_font, color: tuple=white, max_width: int=None, char_width: int=None):
self.text = text
self.font = font
self.color = color
self.max_width = max_width
self.char_width = char_width
if char_width is None: # get the width of a character by the font
self.char_width = font.render("A", True, white).get_width()
self.surface = self.generate_surface()
def generate_surface(self):
lines = []
line = ""
i = 0
if not self.char_width is None:
for char in self.text:
if char == "\n" or (i + 1) * self.char_width > self.max_width:
lines.append(line)
line = ""
else:
line += char
else:
for char in self.text:
if char == "\n":
lines.append(line)
line = ""
else:
line += char
if not lines:
lines = [line]
texts = []
for line in lines:
texts.append(self.font.render(line, True, self.color))
w = max(texts, key=lambda t: t.get_width()).get_width()
h = texts[0].get_height()
sh = h * len(lines)
surface = pygame.Surface((w, sh), flags=pygame.SRCALPHA)
x, y = 0, 0
for text in texts:
surface.blit(text, (x, y))
y += h
return surface
def set_rot_point(img, pos): def set_rot_point(img, pos):
w, h = img.get_size() w, h = img.get_size()
w, h = w * 2, h * 2 w, h = w * 2, h * 2