sep-pm-platformer/project/main.py

165 lines
6.3 KiB
Python

from typing import Optional
import pygame
from level.Level import Level
from level.LevelManager import LevelManager
from level.selection.LevelScreenManager import LevelScreenManager
from level.selection.LevelSelectionScreenManager import LevelSelectionScreenManager
from level.selection.MainMenuScreenManager import MainMenuScreenManager
from level.selection.ScreenManager import ScreenManager
from physics import ConstantsParser
from physics.SpriteManager import SpriteManager
from physics.TickData import TickData
from sprite.PositionScale import PositionScale
from sprite.SpritesheetManager import SpritesheetManager
from ui_elements.ClickEvent import ClickEvent
from ui_elements.KeyManager import KeyManager
BACKGROUND_IMAGES_UNSCALED = {
'tutorial': pygame.image.load('data/sprites/tutorial_bg.png'),
'castle': pygame.image.load('data/sprites/castle_bg.png'),
'cave': pygame.image.load('data/sprites/cave_bg.png'),
}
BACKGROUND_IMAGES_SCALED = {
'tutorial': pygame.transform.scale(BACKGROUND_IMAGES_UNSCALED['tutorial'], (1, 1)),
'castle': pygame.transform.scale(BACKGROUND_IMAGES_UNSCALED['castle'], (1, 1)),
'cave': pygame.transform.scale(BACKGROUND_IMAGES_UNSCALED['cave'], (1, 1)),
}
def apply_frame_rate(number: float, frame_rate: float = 30):
"""
this function calculates a factor that will be multiplied with the
physics of the game to provide a constant speed
:param frame_rate: the frame rate of the game
:param number: The number to scale by the factor
:return: The scaled number
"""
return number / (frame_rate / 30)
class MainLoop:
def __init__(self):
self.GAME_STATE_MENU = 'main_menu'
self.GAME_STATE_LEVEL_SELECTION = 'level_selection'
self.GAME_STATE_LEVEL = 'level'
self.screen_transform: PositionScale = PositionScale((0, 0), (1.5, 1.5))
self.window_size: tuple[float, float] = (1, 1)
pygame.init()
pygame.display.set_caption("PM GAME")
self.update_position_scale(self.screen_transform)
self.screen = self.screen
self.clock = pygame.time.Clock()
self.frame_rate = 30
self.spritesheet_manager = SpritesheetManager("data/sprites", "data/sprites/sprites.json")
self.sprite_manager = SpriteManager()
self.key_manager = KeyManager()
self.parsed_levels_manager = LevelManager('data/levels')
self.parsed_levels_manager.load_from_config('data/levels/levels.json')
self.screen_manager: Optional[ScreenManager] = None
self.game_state = self.GAME_STATE_MENU
self.set_game_state(self.GAME_STATE_MENU)
self.level: Optional[Level] = None
def update_position_scale(self, position_scale: PositionScale):
self.screen_transform = position_scale
self.window_size = (
self.screen_transform.scale[0] * ConstantsParser.CONFIG.block_size[0] *
ConstantsParser.CONFIG.level_size[0],
self.screen_transform.scale[1] * ConstantsParser.CONFIG.block_size[1] *
ConstantsParser.CONFIG.level_size[1]
)
self.screen = pygame.display.set_mode((self.window_size[0], self.window_size[1]))
for key in BACKGROUND_IMAGES_UNSCALED:
BACKGROUND_IMAGES_SCALED[key] = pygame.transform.scale(
BACKGROUND_IMAGES_UNSCALED[key], self.window_size
)
def select_level(self, level: Level):
print('Loading level', level.name)
self.level = level
self.set_game_state(self.GAME_STATE_LEVEL)
def select_level_selection(self, theme: str = 'tutorial'):
self.set_game_state(self.GAME_STATE_LEVEL_SELECTION)
if self.screen_manager is not None and isinstance(self.screen_manager, LevelSelectionScreenManager):
self.screen_manager.select_theme(theme)
def select_main_menu(self):
self.set_game_state(self.GAME_STATE_MENU)
def set_game_state(self, game_state: str):
self.game_state = game_state
if self.screen_manager is not None:
self.screen_manager.destroy()
if self.game_state == self.GAME_STATE_MENU:
self.screen_manager = MainMenuScreenManager(
self.sprite_manager, self.spritesheet_manager, self
)
elif self.game_state == self.GAME_STATE_LEVEL:
self.screen_manager = LevelScreenManager(
self.sprite_manager, self.spritesheet_manager, self, self.level
)
elif self.game_state == self.GAME_STATE_LEVEL_SELECTION:
self.screen_manager = LevelSelectionScreenManager(
self.sprite_manager, self.spritesheet_manager, self, self.parsed_levels_manager
)
else:
print('Invalid game state', self.game_state)
if self.screen_manager is not None:
self.screen_manager.initialize()
def main_loop(self):
while True:
self.clock.tick(self.frame_rate)
pygame_events: list[pygame.event.Event] = pygame.event.get()
self.key_manager.update_key_events(pygame_events)
click_events: list[ClickEvent] = ClickEvent.create_events(pygame_events, self.screen_transform)
for event in click_events:
for layer in self.sprite_manager.layers:
for sprite in self.sprite_manager.layers[layer]:
if sprite.get_bounding_box().contains_point(event.world_position):
sprite.click(event)
for event in pygame_events:
if event.type == pygame.QUIT:
pygame.quit()
exit()
self.screen.fill((0, 0, 0))
tick_data = TickData(apply_frame_rate(1, self.frame_rate),
pygame_events,
self.key_manager,
click_events,
self.screen_transform)
self.screen_manager.tick(tick_data)
self.sprite_manager.tick(tick_data)
if self.level is not None and self.game_state == self.GAME_STATE_LEVEL:
self.screen.blit(BACKGROUND_IMAGES_SCALED[self.level.theme], (0, 0))
self.sprite_manager.draw(self.screen, self.screen_transform)
pygame.display.update()
main_loop: MainLoop = MainLoop()
main_loop.main_loop()