62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import abc
|
|
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 = []
|
|
|
|
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)
|