69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from typing import Optional
|
|
|
|
from pygame import Surface
|
|
|
|
from physics.PhysicsElementsHandler import PhysicsElementsHandler
|
|
from physics.TickData import TickData
|
|
from sprite.PositionScale import PositionScale
|
|
from sprite.Sprite import Sprite
|
|
from ui_elements.UiElement import UiElement
|
|
|
|
|
|
class DrawLayers:
|
|
BACKGROUND = 'background'
|
|
LEVEL = 'level'
|
|
OBJECTS_BACK = 'objects_back'
|
|
OBJECTS = 'objects'
|
|
OBJECTS_FRONT = 'objects_front'
|
|
PLAYER = 'player'
|
|
UI = 'ui'
|
|
|
|
DRAW_ORDER = [BACKGROUND, LEVEL, OBJECTS_BACK, OBJECTS, OBJECTS_FRONT, PLAYER, UI]
|
|
|
|
|
|
class SpriteManager:
|
|
def __init__(self):
|
|
self.physics_handler = PhysicsElementsHandler()
|
|
self.physics_handler.add_collision_callback(self.collision_detected)
|
|
|
|
self.layers: dict[str, list[UiElement]] = {}
|
|
for layer in DrawLayers.DRAW_ORDER:
|
|
self.layers[layer] = []
|
|
|
|
def add_ui_element(self, layer: str, sprite: UiElement):
|
|
self.layers[layer].append(sprite)
|
|
|
|
def remove_ui_element(self, sprite: UiElement):
|
|
for layer in self.layers:
|
|
if sprite in self.layers[layer]:
|
|
self.layers[layer].remove(sprite)
|
|
|
|
def tick(self, tick_data: TickData):
|
|
self.physics_handler.tick(tick_data, self.layers)
|
|
|
|
def draw(self, screen: Surface, screen_transform: PositionScale):
|
|
for layer in DrawLayers.DRAW_ORDER:
|
|
for sprite in self.layers[layer]:
|
|
sprite.draw(screen, screen_transform)
|
|
|
|
def find_elements_at_position(self, position: tuple[float, float]) -> list[UiElement]:
|
|
elements = []
|
|
|
|
for layer in DrawLayers.DRAW_ORDER:
|
|
for sprite in self.layers[layer]:
|
|
bounding_box = sprite.get_bounding_box()
|
|
if bounding_box is not None and bounding_box.contains_point(position):
|
|
elements.append(sprite)
|
|
|
|
return elements
|
|
|
|
def collision_detected(self, sprite_a: Sprite, sprite_b: Sprite):
|
|
pass
|
|
|
|
def find_sprite_by_uuid(self, uuid: str) -> Optional[UiElement]:
|
|
for layer in DrawLayers.DRAW_ORDER:
|
|
for sprite in self.layers[layer]:
|
|
if sprite.uuid == uuid:
|
|
return sprite
|
|
return None
|