378 lines
15 KiB
Python
378 lines
15 KiB
Python
|
import pygame
|
||
|
|
from math import floor
|
||
|
|
|
||
|
|
|
||
|
|
class InputBox:
|
||
|
|
backspace_tick = 0
|
||
|
|
backspace_deleted_streak = 0
|
||
|
|
backspace_breakpoint = 6
|
||
|
|
deleting_speeds = [12, 4]
|
||
|
|
|
||
|
|
# constructor that can be used to set parameters of InputBox instance
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
position,
|
||
|
|
dimensions,
|
||
|
|
input_box_position=None,
|
||
|
|
input_box_dimensions=None,
|
||
|
|
label="",
|
||
|
|
user_input="",
|
||
|
|
box_color=(195, 195, 195),
|
||
|
|
input_box_color=(225, 245, 245),
|
||
|
|
inner_box_color=(205, 205, 205),
|
||
|
|
bottom_strip_color=(180, 180, 180),
|
||
|
|
font=pygame.font.get_default_font(),
|
||
|
|
font_color=(50, 50, 50),
|
||
|
|
font_size=None,
|
||
|
|
is_visible=True,
|
||
|
|
is_active=True,
|
||
|
|
input_centered=False,
|
||
|
|
fit_text=True,
|
||
|
|
outline=True,
|
||
|
|
outline_thickness=None,
|
||
|
|
outline_color=(50, 50, 50),
|
||
|
|
input_box_transform_hover=1.07,
|
||
|
|
input_box_transform_selected=2,
|
||
|
|
input_box_transform_inactive=0.9
|
||
|
|
):
|
||
|
|
# extracting and setting values
|
||
|
|
smaller_dimension = min(dimensions)
|
||
|
|
box_x, box_y = position
|
||
|
|
width, height = dimensions
|
||
|
|
|
||
|
|
# picking default input_box position and dimensions depending on the boxes width
|
||
|
|
if width < 70:
|
||
|
|
default_input_box_position = (box_x + width / 20, box_y + height / 2)
|
||
|
|
default_input_box_dimensions = (width - width / 10, 3 * height / 10)
|
||
|
|
|
||
|
|
else:
|
||
|
|
default_input_box_position = (box_x + width / 12, box_y + height / 2)
|
||
|
|
default_input_box_dimensions = (width - width / 6, 3 * height / 10)
|
||
|
|
|
||
|
|
# counting default outline thickness
|
||
|
|
default_outline_thickness = (smaller_dimension + (width + height) / 4) / 50
|
||
|
|
|
||
|
|
# setting attributes
|
||
|
|
self.position = position
|
||
|
|
self.dimensions = dimensions
|
||
|
|
self.input_box_position = _return_value_or_default(input_box_position, default_input_box_position)
|
||
|
|
self.input_box_dimensions = _return_value_or_default(input_box_dimensions, default_input_box_dimensions)
|
||
|
|
self.label = label
|
||
|
|
self.user_input = user_input
|
||
|
|
self.box_color = _get_proper_rgb(box_color)
|
||
|
|
self.ib_color = _get_proper_rgb(input_box_color)
|
||
|
|
self.inner_box_color = _get_proper_rgb(inner_box_color)
|
||
|
|
self.bottom_strip_color = _get_proper_rgb(bottom_strip_color)
|
||
|
|
self.font_color = _get_proper_rgb(font_color)
|
||
|
|
self.font_size = _return_value_or_default(font_size, floor(smaller_dimension / 2.8))
|
||
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
||
|
|
self.input_font_size = max(15, floor(self.input_box_dimensions[1] - 2))
|
||
|
|
self.input_font = pygame.font.SysFont(font, self.input_font_size)
|
||
|
|
self.is_visible = is_visible
|
||
|
|
self.is_active = is_active
|
||
|
|
self.is_selected = False
|
||
|
|
self.input_centered = input_centered
|
||
|
|
self.fit_text = fit_text
|
||
|
|
self.outline = outline
|
||
|
|
self.outline_thickness = _return_value_or_default(outline_thickness, default_outline_thickness)
|
||
|
|
self.outline_color = outline_color
|
||
|
|
self.ib_transform_hover = input_box_transform_hover
|
||
|
|
self.ib_transform_selected = input_box_transform_selected
|
||
|
|
self.ib_transform_inactive = input_box_transform_inactive
|
||
|
|
self.backspace_pressed_down = False
|
||
|
|
|
||
|
|
# rendering label to get it's width and height
|
||
|
|
rendered_label = self.font.render(label, True, (0, 0, 0))
|
||
|
|
|
||
|
|
# if text is out of bounds and fit_text=True - resizing text to fit the box
|
||
|
|
if self.fit_text and rendered_label.get_width() > self.input_box_dimensions[0]:
|
||
|
|
self.font_size = floor(self.font_size / (rendered_label.get_width() / width + 0.1))
|
||
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
||
|
|
|
||
|
|
# counting colors on: hover, inactive, selected
|
||
|
|
self.ib_hover_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_hover)))
|
||
|
|
self.ib_inactive_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_inactive)))
|
||
|
|
self.ib_selected_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_selected)))
|
||
|
|
|
||
|
|
# draws, updates and selects on mouse click the InputBox instance
|
||
|
|
def run(self, window, mouse_position, events):
|
||
|
|
self.select_on_click(mouse_position, events)
|
||
|
|
self.draw(window, mouse_position)
|
||
|
|
self.update(events)
|
||
|
|
|
||
|
|
# draws an InputBox instance (doesn't run interactions [for everything use run(self, ...) method])
|
||
|
|
def draw(self, window, mouse_position):
|
||
|
|
# if is_visible=True drawing the InputBox
|
||
|
|
if self.is_visible:
|
||
|
|
# extracting and setting values from attributes
|
||
|
|
box_x, box_y = self.position
|
||
|
|
width, height = self.dimensions
|
||
|
|
input_box_x, input_box_y = self.input_box_position
|
||
|
|
input_width, input_height = self.input_box_dimensions
|
||
|
|
|
||
|
|
# if outline=True - drawing an outline
|
||
|
|
if self.outline:
|
||
|
|
padding = self.outline_thickness
|
||
|
|
outline_coordinates = (box_x - padding, box_y - padding, width + 2 * padding, height + 2 * padding)
|
||
|
|
pygame.draw.rect(window, self.outline_color, outline_coordinates)
|
||
|
|
|
||
|
|
# setting "transparent" background color
|
||
|
|
text_background_color = self.ib_color
|
||
|
|
|
||
|
|
# setting values accordingly to InputBox'es state
|
||
|
|
# is selected
|
||
|
|
if self.is_selected:
|
||
|
|
input_box_color = self.ib_selected_color
|
||
|
|
text_background_color = (150, 255, 255)
|
||
|
|
|
||
|
|
# is active and mouse is not over
|
||
|
|
elif not self.is_over(mouse_position) and self.is_active:
|
||
|
|
input_box_color = self.ib_color
|
||
|
|
|
||
|
|
# is active and mouse is over
|
||
|
|
elif self.is_active:
|
||
|
|
input_box_color = self.ib_hover_color
|
||
|
|
|
||
|
|
# is not active
|
||
|
|
else:
|
||
|
|
input_box_color = self.ib_inactive_color
|
||
|
|
|
||
|
|
# drawing outer box
|
||
|
|
pygame.draw.rect(window, self.box_color, (box_x, box_y, width, height))
|
||
|
|
|
||
|
|
# drawing inner (upper) box
|
||
|
|
pygame.draw.rect(window,
|
||
|
|
self.inner_box_color,
|
||
|
|
(input_box_x, box_y + height / 10, input_width, height - input_height - height / 2.5)
|
||
|
|
)
|
||
|
|
|
||
|
|
# drawing input (lower) box
|
||
|
|
pygame.draw.rect(window, input_box_color, (input_box_x, input_box_y, input_width, input_height))
|
||
|
|
|
||
|
|
# drawing bottom strip
|
||
|
|
pygame.draw.rect(window, self.bottom_strip_color, (box_x, box_y + 9 * height / 10, width, 1 * height / 10))
|
||
|
|
|
||
|
|
# rendering label and counting its coordinates
|
||
|
|
rendered_label = self.font.render(self.label, True, self.font_color)
|
||
|
|
rendered_label_x = box_x + width / 2 - rendered_label.get_width() / 2
|
||
|
|
rendered_label_y = box_y + height / 4 - rendered_label.get_height() / 2
|
||
|
|
|
||
|
|
# drawing the label on the coordinates
|
||
|
|
window.blit(rendered_label, (rendered_label_x, rendered_label_y))
|
||
|
|
|
||
|
|
# rendering input
|
||
|
|
rendered_input_text = self.input_font.render(self.user_input, True, self.font_color, text_background_color)
|
||
|
|
|
||
|
|
# picking input coordinates on center or not depending on input_centered flag value
|
||
|
|
if self.input_centered:
|
||
|
|
rendered_input_text_x = input_box_x + input_width / 2 - rendered_input_text.get_width() / 2
|
||
|
|
rendered_input_text_y = input_box_y + input_height / 2 - rendered_input_text.get_height() / 2
|
||
|
|
|
||
|
|
else:
|
||
|
|
rendered_input_text_x = input_box_x + 2
|
||
|
|
rendered_input_text_y = input_box_y + input_height / 2 - rendered_input_text.get_height() / 2
|
||
|
|
|
||
|
|
# drawing the input on the coordinates
|
||
|
|
window.blit(rendered_input_text, (rendered_input_text_x, rendered_input_text_y))
|
||
|
|
|
||
|
|
# updates InputBox'es attributes if user types something in it
|
||
|
|
def update(self, events):
|
||
|
|
# if the InputBox is selected - updating the input text
|
||
|
|
if self.is_selected:
|
||
|
|
self._delete_characters_if_backspace_pressed()
|
||
|
|
|
||
|
|
# rendering input to get it's width and height
|
||
|
|
rendered_input = self.input_font.render(self.user_input, True, self.font_color, self.ib_inactive_color)
|
||
|
|
|
||
|
|
# deleting or adding characters depending on user's input
|
||
|
|
for event in events:
|
||
|
|
if event.type == pygame.KEYDOWN:
|
||
|
|
if event.key == pygame.K_BACKSPACE:
|
||
|
|
if not self.backspace_pressed_down:
|
||
|
|
self.user_input = self.user_input[:-1]
|
||
|
|
self.backspace_pressed_down = True
|
||
|
|
|
||
|
|
elif event.key == pygame.K_RETURN:
|
||
|
|
self.is_selected = False
|
||
|
|
|
||
|
|
# if text isn't too long - adding a character
|
||
|
|
elif rendered_input.get_width() + 10 < self.input_box_dimensions[0]:
|
||
|
|
self.user_input += event.unicode
|
||
|
|
|
||
|
|
elif event.type == pygame.KEYUP:
|
||
|
|
if event.key == pygame.K_BACKSPACE:
|
||
|
|
self.backspace_pressed_down = False
|
||
|
|
self.backspace_tick = 0
|
||
|
|
self.backspace_deleted_streak = 0
|
||
|
|
|
||
|
|
# checks if the InputBox is clicked and changes is_selected to True or False
|
||
|
|
def select_on_click(self, mouse_position, events):
|
||
|
|
# if is active set is_selected attribute accordingly
|
||
|
|
if self.is_active:
|
||
|
|
for event in events:
|
||
|
|
# if mouse is clicked set as selected if mouse is over, or set as not selected otherwise
|
||
|
|
if event.type == pygame.MOUSEBUTTONUP:
|
||
|
|
self.set_is_selected(self.is_over(mouse_position))
|
||
|
|
|
||
|
|
# checks if a position is over the InputBox and returns True or False
|
||
|
|
def is_over(self, position):
|
||
|
|
mouse_x, mouse_y = position
|
||
|
|
button_x, button_y = self.position
|
||
|
|
width, height = self.dimensions
|
||
|
|
|
||
|
|
return button_x <= mouse_x <= (button_x + width) and button_y <= mouse_y <= (button_y + height)
|
||
|
|
|
||
|
|
# returns InputBox'es label
|
||
|
|
def get_label(self):
|
||
|
|
return self.label
|
||
|
|
|
||
|
|
# returns InputBox'es user's input
|
||
|
|
def get_input(self):
|
||
|
|
return self.user_input
|
||
|
|
|
||
|
|
# sets chosen coordinates
|
||
|
|
def set_space(
|
||
|
|
self,
|
||
|
|
position=None,
|
||
|
|
dimensions=None,
|
||
|
|
input_box_position=None,
|
||
|
|
input_box_dimensions=None,
|
||
|
|
outline_thickness=None
|
||
|
|
):
|
||
|
|
if position is not None:
|
||
|
|
self.position = position
|
||
|
|
|
||
|
|
if dimensions is not None:
|
||
|
|
self.dimensions = dimensions
|
||
|
|
|
||
|
|
if input_box_position is not None:
|
||
|
|
self.input_box_position = input_box_position
|
||
|
|
|
||
|
|
if input_box_dimensions is not None:
|
||
|
|
self.input_box_dimensions = input_box_dimensions
|
||
|
|
|
||
|
|
if outline_thickness is not None:
|
||
|
|
self.outline_thickness = outline_thickness
|
||
|
|
|
||
|
|
# sets chosen text attributes
|
||
|
|
def set_texts(self, label=None, user_input=None):
|
||
|
|
if label is not None:
|
||
|
|
self.label = label
|
||
|
|
|
||
|
|
if user_input is not None:
|
||
|
|
self.user_input = user_input
|
||
|
|
|
||
|
|
# sets chosen font attributes
|
||
|
|
def set_font(self, font=None, font_size=None):
|
||
|
|
if font_size is not None:
|
||
|
|
self.font_size = font_size
|
||
|
|
|
||
|
|
# rendering text to get it's width and height
|
||
|
|
rendered_text = self.font.render(self.user_input, True, (0, 0, 0))
|
||
|
|
|
||
|
|
# if text is out of bounds and fit_text=True - resizing text to fit the box
|
||
|
|
if self.fit_text and rendered_text.get_width() > self.dimensions[0]:
|
||
|
|
self.font_size = floor(self.font_size / (rendered_text.get_width() / self.dimensions[0] + 0.1))
|
||
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
||
|
|
|
||
|
|
if font is not None:
|
||
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
||
|
|
|
||
|
|
# sets chosen color attributes
|
||
|
|
def set_colors(
|
||
|
|
self,
|
||
|
|
box_color=None,
|
||
|
|
input_box_color=None,
|
||
|
|
inner_box_color=None,
|
||
|
|
bottom_strip_color=None,
|
||
|
|
font_color=None,
|
||
|
|
outline_color=None,
|
||
|
|
input_box_transform_hover=None,
|
||
|
|
input_box_transform_selected=None,
|
||
|
|
input_box_transform_inactive=None
|
||
|
|
):
|
||
|
|
if box_color is not None:
|
||
|
|
self.box_color = _get_proper_rgb(box_color)
|
||
|
|
|
||
|
|
if input_box_color is not None:
|
||
|
|
self.ib_color = _get_proper_rgb(input_box_color)
|
||
|
|
|
||
|
|
if inner_box_color is not None:
|
||
|
|
self.inner_box_color = _get_proper_rgb(inner_box_color)
|
||
|
|
|
||
|
|
if bottom_strip_color is not None:
|
||
|
|
self.bottom_strip_color = _get_proper_rgb(bottom_strip_color)
|
||
|
|
|
||
|
|
if font_color is not None:
|
||
|
|
self.font_color = _get_proper_rgb(font_color)
|
||
|
|
|
||
|
|
if outline_color is not None:
|
||
|
|
self.outline_color = _get_proper_rgb(outline_color)
|
||
|
|
|
||
|
|
if input_box_transform_hover is not None:
|
||
|
|
self.ib_transform_hover = input_box_transform_hover
|
||
|
|
|
||
|
|
if input_box_transform_selected is not None:
|
||
|
|
self.ib_transform_selected = input_box_transform_selected
|
||
|
|
|
||
|
|
if input_box_transform_inactive is not None:
|
||
|
|
self.ib_transform_inactive = input_box_transform_inactive
|
||
|
|
|
||
|
|
self.ib_hover_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_hover)))
|
||
|
|
self.ib_inactive_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_inactive)))
|
||
|
|
self.ib_selected_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_selected)))
|
||
|
|
|
||
|
|
# sets chosen flag attributes
|
||
|
|
def set_flags(self, is_visible=None, is_active=None, input_centered=None, fit_text=None, outline=None):
|
||
|
|
if is_visible is not None:
|
||
|
|
self.is_visible = is_visible
|
||
|
|
|
||
|
|
if is_active is not None:
|
||
|
|
self.is_active = is_active
|
||
|
|
|
||
|
|
if input_centered is not None:
|
||
|
|
self.input_centered = input_centered
|
||
|
|
|
||
|
|
if fit_text is not None:
|
||
|
|
self.fit_text = fit_text
|
||
|
|
|
||
|
|
if outline is not None:
|
||
|
|
self.outline = outline
|
||
|
|
|
||
|
|
# sets is_selected with a given value
|
||
|
|
def set_is_selected(self, is_selected):
|
||
|
|
self.is_selected = is_selected
|
||
|
|
|
||
|
|
def _delete_characters_if_backspace_pressed(self):
|
||
|
|
if self.backspace_pressed_down and \
|
||
|
|
(self.backspace_tick + 1) % self.deleting_speeds[0] == 0 and \
|
||
|
|
self.backspace_deleted_streak <= 6:
|
||
|
|
self.user_input = self.user_input[:-1]
|
||
|
|
self.backspace_tick = 0
|
||
|
|
self.backspace_deleted_streak += 1
|
||
|
|
|
||
|
|
elif self.backspace_pressed_down and \
|
||
|
|
(self.backspace_tick + 1) % self.deleting_speeds[1] == 0 and \
|
||
|
|
self.backspace_deleted_streak > 6:
|
||
|
|
self.user_input = self.user_input[:-1]
|
||
|
|
self.backspace_tick = 0
|
||
|
|
self.backspace_deleted_streak += 1
|
||
|
|
|
||
|
|
elif self.backspace_pressed_down:
|
||
|
|
self.backspace_tick += 1
|
||
|
|
|
||
|
|
|
||
|
|
def _get_proper_rgb(x):
|
||
|
|
r, g, b = x
|
||
|
|
r, g, b = max(0, min(255, floor(r))), max(0, min(255, floor(g))), max(0, min(255, floor(b)))
|
||
|
|
|
||
|
|
return tuple((r, g, b))
|
||
|
|
|
||
|
|
|
||
|
|
def _transform(iterable, transform_value):
|
||
|
|
return [x * transform_value for x in iterable]
|
||
|
|
|
||
|
|
|
||
|
|
def _return_value_or_default(value, default):
|
||
|
|
return default if value is None else value
|