Projekt_Sztuczna_Inteligencja/ui/text_box.py

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