sep-pm-platformer/project/ui_elements/UiElement.py

import abc
import uuid
from typing import Optional

import pygame
from pygame import Surface

from sprite.BoundingBox import BoundingBox
from sprite.PositionScale import PositionScale
from physics.TickData import TickData
from ui_elements import CoordinateTransform


class UiElement:
    def __init__(self):
        self.position_scale = PositionScale()
        self.visible = True
        self.click_listeners = []

        self.bounding_box = BoundingBox(0, 0, 0, 0)

        self.uuid = uuid.uuid4()

    def add_click_listener(self, listener):
        self.click_listeners.append(listener)

    def remove_click_listener(self, listener):
        self.click_listeners.remove(listener)

    def click(self, click_event):
        for listener in self.click_listeners:
            listener(self, click_event)

    @abc.abstractmethod
    def tick(self, tick_data: TickData):
        pass

    @abc.abstractmethod
    def render_sprite_image(self) -> Optional[Surface]:
        pass

    @abc.abstractmethod
    def get_bounding_box(self) -> BoundingBox:
        pass

    def draw(self, screen: pygame.Surface, screen_transform: PositionScale):
        if not self.visible:
            return

        image = self.render_sprite_image()

        if image is not None:
            target_position = CoordinateTransform.transform_world_to_screen(self.position_scale.position,
                                                                            screen_transform)

            screen_scale = screen_transform.scale
            object_scale = self.position_scale.scale

            target_size = (int(screen_scale[0] * object_scale[0] * image.get_width()),
                           int(screen_scale[1] * object_scale[1] * image.get_height()))

            target_image = UiElement.get_scaled_image(image, target_size)
            screen.blit(target_image, target_position)

    @staticmethod
    def get_scaled_image(image, resize):
        return pygame.transform.scale(image, resize)