138 lines
5.1 KiB
Python
138 lines
5.1 KiB
Python
import pygame
|
|
from math import floor
|
|
|
|
|
|
class TextBox:
|
|
# constructor that can be used to set parameters of TextBox instance
|
|
def __init__(
|
|
self,
|
|
position,
|
|
dimensions,
|
|
text="",
|
|
box_color=(200, 200, 200),
|
|
font=pygame.font.get_default_font(),
|
|
font_color=(0, 0, 0),
|
|
font_size=None,
|
|
is_visible=True,
|
|
fit_text=True,
|
|
outline=True,
|
|
outline_thickness=None,
|
|
outline_color=(50, 50, 50)
|
|
):
|
|
# extracting and setting values
|
|
smaller_dimension = min(dimensions)
|
|
width, height = dimensions
|
|
|
|
# counting default outline thickness
|
|
default_outline_thickness = (smaller_dimension + (width + height) / 4) / 50
|
|
|
|
# setting attributes
|
|
self.position = position
|
|
self.dimensions = dimensions
|
|
self.text = text
|
|
self.box_color = _get_proper_rgb(box_color)
|
|
self.font_color = _get_proper_rgb(font_color)
|
|
self.font_size = _return_value_or_default(font_size, floor(smaller_dimension / 2))
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
|
self.is_visible = is_visible
|
|
self.fit_text = fit_text
|
|
self.outline = outline
|
|
self.outline_thickness = _return_value_or_default(outline_thickness, default_outline_thickness)
|
|
self.outline_color = _get_proper_rgb(outline_color)
|
|
|
|
# rendering text to get it's width and height
|
|
rendered_text = self.font.render(text, 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() / width + 0.1))
|
|
self.font = pygame.font.SysFont(font, self.font_size)
|
|
|
|
# draws the TextBox
|
|
def draw(self, window):
|
|
# if is_visible=True drawing the TextBox
|
|
if self.is_visible:
|
|
# extracting and setting values from attributes
|
|
box_x, box_y = self.position
|
|
width, height = self.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)
|
|
|
|
# drawing the box
|
|
pygame.draw.rect(window, self.box_color, (box_x, box_y, width, height))
|
|
|
|
# rendering text and counting it's coordinates
|
|
rendered_text = self.font.render(self.text, True, self.font_color)
|
|
rendered_text_x = box_x + width/2 - rendered_text.get_width()/2
|
|
rendered_text_y = box_y + height/2 - rendered_text.get_height()/2
|
|
|
|
# drawing the box on the coordinates
|
|
window.blit(rendered_text, (rendered_text_x, rendered_text_y))
|
|
|
|
# sets chosen coordinates
|
|
def set_coordinates(self, position=None, dimensions=None, outline_thickness=None):
|
|
if position is not None:
|
|
self.position = position
|
|
|
|
if dimensions is not None:
|
|
self.dimensions = dimensions
|
|
|
|
if outline_thickness is not None:
|
|
self.outline_thickness = outline_thickness
|
|
|
|
# sets the TextBox's text
|
|
def set_text(self, text):
|
|
self.text = text
|
|
|
|
# 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.text, 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, font_color=None, outline_color=None):
|
|
if box_color is not None:
|
|
self.box_color = _get_proper_rgb(box_color)
|
|
|
|
if font_color is not None:
|
|
self.font_color = _get_proper_rgb(font_color)
|
|
|
|
if outline_color is not None:
|
|
self.outline_color = outline_color
|
|
|
|
# sets chosen flags
|
|
def set_flags(self, is_visible=None, outline=None):
|
|
if is_visible is not None:
|
|
self.is_visible = is_visible
|
|
|
|
if outline is not None:
|
|
self.outline = outline
|
|
|
|
|
|
# returns values that are not out of range
|
|
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))
|
|
|
|
|
|
# returns value or default depending on if passed value is None or not
|
|
def _return_value_or_default(value, default):
|
|
return default if value is None else value
|