71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import curses
|
|
from squirrelbattle.display.mapdisplay import MapDisplay
|
|
from squirrelbattle.display.statsdisplay import StatsDisplay
|
|
from squirrelbattle.display.menudisplay import SettingsMenuDisplay, \
|
|
MainMenuDisplay
|
|
from squirrelbattle.display.texturepack import TexturePack
|
|
from typing import Any
|
|
from squirrelbattle.game import Game, GameMode
|
|
from squirrelbattle.enums import DisplayActions
|
|
|
|
|
|
class DisplayManager:
|
|
|
|
def __init__(self, screen: Any, g: Game):
|
|
self.game = g
|
|
self.screen = screen
|
|
pack = TexturePack.get_pack(self.game.settings.TEXTURE_PACK)
|
|
self.mapdisplay = MapDisplay(screen, pack)
|
|
self.statsdisplay = StatsDisplay(screen, pack)
|
|
self.mainmenudisplay = MainMenuDisplay(self.game.main_menu,
|
|
screen, pack)
|
|
self.settingsmenudisplay = SettingsMenuDisplay(screen, pack)
|
|
self.displays = [self.statsdisplay, self.mapdisplay,
|
|
self.mainmenudisplay, self.settingsmenudisplay]
|
|
self.update_game_components()
|
|
|
|
def handle_display_action(self, action: DisplayActions) -> None:
|
|
if action == DisplayActions.REFRESH:
|
|
self.refresh()
|
|
elif action == DisplayActions.UPDATE:
|
|
self.update_game_components()
|
|
|
|
def update_game_components(self) -> None:
|
|
for d in self.displays:
|
|
d.pack = TexturePack.get_pack(self.game.settings.TEXTURE_PACK)
|
|
self.mapdisplay.update_map(self.game.map)
|
|
self.statsdisplay.update_player(self.game.player)
|
|
self.settingsmenudisplay.update_menu(self.game.settings_menu)
|
|
|
|
def refresh(self) -> None:
|
|
if self.game.state == GameMode.PLAY:
|
|
# The map pad has already the good size
|
|
self.mapdisplay.refresh(0, 0, self.rows * 4 // 5, self.cols,
|
|
resize_pad=False)
|
|
self.statsdisplay.refresh(self.rows * 4 // 5, 0,
|
|
self.rows // 5, self.cols)
|
|
if self.game.state == GameMode.MAINMENU:
|
|
self.mainmenudisplay.refresh(0, 0, self.rows, self.cols)
|
|
if self.game.state == GameMode.SETTINGS:
|
|
self.settingsmenudisplay.refresh(0, 0, self.rows, self.cols - 1)
|
|
self.resize_window()
|
|
|
|
def resize_window(self) -> bool:
|
|
"""
|
|
If the window got resized, ensure that the screen size got updated.
|
|
"""
|
|
y, x = self.screen.getmaxyx() if self.screen else (0, 0)
|
|
if self.screen and curses.is_term_resized(self.rows,
|
|
self.cols): # pragma: nocover
|
|
curses.resizeterm(y, x)
|
|
return True
|
|
return False
|
|
|
|
@property
|
|
def rows(self) -> int:
|
|
return curses.LINES if self.screen else 42
|
|
|
|
@property
|
|
def cols(self) -> int:
|
|
return curses.COLS if self.screen else 42
|