squirrel-battle/squirrelbattle/display/display_manager.py

120 lines
5.5 KiB
Python
Raw Normal View History

2020-11-27 15:33:17 +00:00
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later
import curses
2020-12-02 15:01:32 +00:00
from squirrelbattle.display.display import VerticalSplit, HorizontalSplit, Display
2020-11-19 01:18:08 +00:00
from squirrelbattle.display.mapdisplay import MapDisplay
2020-11-27 16:32:26 +00:00
from squirrelbattle.display.messagedisplay import MessageDisplay
2020-11-19 01:18:08 +00:00
from squirrelbattle.display.statsdisplay import StatsDisplay
from squirrelbattle.display.menudisplay import SettingsMenuDisplay, \
2020-11-13 18:08:40 +00:00
MainMenuDisplay
from squirrelbattle.display.logsdisplay import LogsDisplay
2020-11-19 01:18:08 +00:00
from squirrelbattle.display.texturepack import TexturePack
from typing import Any
2020-11-19 01:18:08 +00:00
from squirrelbattle.game import Game, GameMode
from squirrelbattle.enums import DisplayActions
class DisplayManager:
2020-11-10 19:34:22 +00:00
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)
2020-11-10 19:34:22 +00:00
self.mainmenudisplay = MainMenuDisplay(self.game.main_menu,
screen, pack)
self.settingsmenudisplay = SettingsMenuDisplay(screen, pack)
self.logsdisplay = LogsDisplay(screen, pack)
2020-11-27 16:52:26 +00:00
self.messagedisplay = MessageDisplay(screen=screen, pack=None)
self.hbar = HorizontalSplit(screen, pack)
self.vbar = VerticalSplit(screen, pack)
2020-11-10 19:34:22 +00:00
self.displays = [self.statsdisplay, self.mapdisplay,
self.mainmenudisplay, self.settingsmenudisplay,
2020-11-27 16:32:26 +00:00
self.logsdisplay, self.messagedisplay]
self.update_game_components()
2020-11-12 00:58:10 +00:00
def handle_display_action(self, action: DisplayActions) -> None:
2020-11-11 23:12:30 +00:00
if action == DisplayActions.REFRESH:
self.refresh()
2020-11-11 23:12:30 +00:00
elif action == DisplayActions.UPDATE:
self.update_game_components()
2020-11-10 19:34:22 +00:00
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)
2020-11-11 23:12:30 +00:00
self.settingsmenudisplay.update_menu(self.game.settings_menu)
self.logsdisplay.update_logs(self.game.logs)
2020-11-27 16:32:26 +00:00
self.messagedisplay.update_message(self.game.message)
def refresh(self) -> None:
2020-12-02 15:01:32 +00:00
self.resize_window()
Display.erase_screen_lines()
if self.game.state == GameMode.PLAY:
# The map pad has already the good size
self.mapdisplay.refresh(0, 0, self.rows * 4 // 5,
self.mapdisplay.pack.tile_width
* (self.cols * 4 // 5
// self.mapdisplay.pack.tile_width),
resize_pad=False)
self.statsdisplay.refresh(0, self.cols * 4 // 5 + 1,
self.rows, self.cols // 5 - 1)
self.logsdisplay.refresh(self.rows * 4 // 5 + 1, 0,
self.rows // 5 - 1, self.cols * 4 // 5)
self.hbar.refresh(self.rows * 4 // 5, 0, 1, self.cols * 4 // 5)
self.vbar.refresh(0, self.cols * 4 // 5, self.rows, 1)
if self.game.state == GameMode.MAINMENU:
2020-11-10 19:34:22 +00:00
self.mainmenudisplay.refresh(0, 0, self.rows, self.cols)
if self.game.state == GameMode.SETTINGS:
2020-11-11 19:36:43 +00:00
self.settingsmenudisplay.refresh(0, 0, self.rows, self.cols - 1)
2020-11-27 16:32:26 +00:00
if self.game.message:
height, width = 0, 0
for line in self.game.message.split("\n"):
height += 1
width = max(width, len(line))
y, x = (self.rows - height) // 2, (self.cols - width) // 2
self.messagedisplay.refresh(y, x, height, width)
2020-12-02 15:01:32 +00:00
with open("squirrel-battle.html", "w") as f:
f.write("<!doctype html>\n<html>\n<head></head>\n<body>\n"
"<div style=\"font-family: Monospace, Noto Color Emoji;\">\n")
f.write("<table style=\"border-collapse: collapse;\"><tbody>\n")
for line in Display.print_screen().split("\n"):
f.write("<tr style=\"margin: 0; padding: 0;\">")
for c in line:
if c in TexturePack.SQUIRREL_PACK.PLAYER and c != TexturePack.SQUIRREL_PACK.PLAYER[0] and c != " ":
continue
squirrel_pack = TexturePack.SQUIRREL_PACK
width = 2 if c in squirrel_pack and c != " " and c != "" else 1
f.write(f"<td style=\"margin: 0; padding: 0;\" colspan=\"{width}\">{c}</td>")
f.write("</tr>")
f.write("\n</tbody></table>\n</div>\n</body>\n<script>\n"
"function reload() {location.reload()}\n"
"setTimeout(reload, 300)\n"
"</script>\n</html>")
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)
2020-12-02 15:01:32 +00:00
Display.resize_screen_lines(y, x)
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:
2020-11-10 19:34:22 +00:00
return curses.COLS if self.screen else 42