Added args parameter to all button/hover classes to be able to execute a function with arguments.
This commit is contained in:
parent
8d4bb14583
commit
7113e6a8c8
1 changed files with 81 additions and 6 deletions
85
pg.py
85
pg.py
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue