squirrel-battle/squirrelbattle/display/menudisplay.py

231 lines
7.9 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
2020-12-11 16:43:46 +00:00
2020-12-04 15:02:03 +00:00
import curses
from random import randint
2020-11-11 19:36:43 +00:00
from typing import List
from squirrelbattle.menus import Menu, MainMenu, SettingsMenu, StoreMenu
from .display import Box, Display
from ..entities.player import Player
from ..enums import KeyValues, GameMode
2020-12-11 16:43:46 +00:00
from ..game import Game
2020-11-19 01:49:59 +00:00
from ..resources import ResourceManager
2020-11-27 21:19:41 +00:00
from ..translations import gettext as _
2020-11-10 19:34:22 +00:00
2020-11-08 15:36:21 +00:00
class MenuDisplay(Display):
"""
A class to display the menu objects
"""
2020-12-18 14:07:09 +00:00
menu: Menu
2020-11-09 00:33:23 +00:00
position: int
2020-11-10 19:34:22 +00:00
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.menubox = Box(*args, **kwargs)
2020-11-10 19:34:22 +00:00
def update_menu(self, menu: Menu) -> None:
self.menu = menu
2020-11-08 15:36:21 +00:00
2020-11-10 19:34:22 +00:00
# Menu values are printed in pad
self.pad = self.newpad(self.trueheight, self.truewidth + 2)
2020-11-09 00:33:23 +00:00
def update_pad(self) -> None:
for i in range(self.trueheight):
self.addstr(self.pad, i, 0, " " + self.values[i])
2020-11-10 19:34:22 +00:00
# set a marker on the selected line
self.addstr(self.pad, self.menu.position, 0, " >")
2020-11-09 00:33:23 +00:00
def display(self) -> None:
cornery = 0 if self.height - 2 >= self.menu.position - 1 \
else self.trueheight - self.height + 2 \
if self.height - 2 >= self.trueheight - self.menu.position else 0
2020-11-10 19:34:22 +00:00
# Menu box
self.menubox.refresh(self.y, self.x, self.height, self.width)
self.pad.erase()
self.update_pad()
self.refresh_pad(self.pad, cornery, 0, self.y + 1, self.x + 1,
self.height - 2 + self.y,
self.width - 2 + self.x)
2020-11-10 19:34:22 +00:00
2020-12-11 16:43:46 +00:00
def handle_click(self, y: int, x: int, game: Game) -> None:
2020-12-11 16:40:56 +00:00
"""
We can select a menu item with the mouse.
"""
2020-12-11 17:07:24 +00:00
self.menu.position = max(0, min(len(self.menu.values) - 1, y - 1))
2020-12-11 16:43:46 +00:00
game.handle_key_pressed(KeyValues.ENTER)
2020-12-11 16:40:56 +00:00
@property
def truewidth(self) -> int:
return max([len(str(a)) for a in self.values])
@property
def trueheight(self) -> int:
return len(self.values)
@property
def preferred_width(self) -> int:
return self.truewidth + 6
2020-11-10 19:34:22 +00:00
@property
def preferred_height(self) -> int:
return self.trueheight + 2
@property
2020-11-11 19:36:43 +00:00
def values(self) -> List[str]:
return [str(a) for a in self.menu.values]
2020-11-13 18:08:40 +00:00
class SettingsMenuDisplay(MenuDisplay):
"""
A class to display specifically a settingsmenu object
"""
2020-12-18 14:07:09 +00:00
menu: SettingsMenu
def update(self, game: Game) -> None:
self.update_menu(game.settings_menu)
@property
def values(self) -> List[str]:
2020-11-27 21:19:41 +00:00
return [_(a[1][1]) + (" : "
+ ("?" if self.menu.waiting_for_key
2020-11-27 21:19:41 +00:00
and a == self.menu.validate() else a[1][0]
.replace("\n", "\\n"))
2020-11-13 18:08:40 +00:00
if a[1][0] else "") for a in self.menu.values]
2020-11-10 19:34:22 +00:00
class MainMenuDisplay(Display):
"""
A class to display specifically a mainmenu object
"""
2020-11-10 19:34:22 +00:00
def __init__(self, menu: MainMenu, *args):
super().__init__(*args)
self.menu = menu
2020-11-19 01:49:59 +00:00
with open(ResourceManager.get_asset_path("ascii_art.txt"), "r") as file:
self.title = file.read().split("\n")
2020-11-13 18:08:40 +00:00
self.pad = self.newpad(max(self.rows, len(self.title) + 30),
2020-11-19 00:17:46 +00:00
max(len(self.title[0]) + 5, self.cols))
2020-11-13 18:08:40 +00:00
self.fg_color = curses.COLOR_WHITE
self.menudisplay = MenuDisplay(self.screen, self.pack)
self.menudisplay.update_menu(self.menu)
2020-11-10 19:34:22 +00:00
def display(self) -> None:
2020-11-10 19:34:22 +00:00
for i in range(len(self.title)):
self.addstr(self.pad, 4 + i, max(self.width // 2
- len(self.title[0]) // 2 - 1, 0), self.title[i],
self.fg_color)
self.refresh_pad(self.pad, 0, 0, self.y, self.x,
self.height + self.y - 1,
2020-11-20 17:09:39 +00:00
self.width + self.x - 1)
menuwidth = min(self.menudisplay.preferred_width, self.width)
2020-11-10 19:34:22 +00:00
menuy, menux = len(self.title) + 8, self.width // 2 - menuwidth // 2 - 1
self.menudisplay.refresh(
menuy, menux, min(self.menudisplay.preferred_height,
self.height - menuy), menuwidth)
2020-12-04 13:41:59 +00:00
2020-12-18 14:07:09 +00:00
def update(self, game: Game) -> None:
self.menudisplay.update_menu(game.main_menu)
2020-12-11 16:43:46 +00:00
def handle_click(self, y: int, x: int, game: Game) -> None:
2020-12-11 16:40:56 +00:00
menuwidth = min(self.menudisplay.preferred_width, self.width)
menuy, menux = len(self.title) + 8, self.width // 2 - menuwidth // 2 - 1
menuheight = min(self.menudisplay.preferred_height, self.height - menuy)
if menuy <= y < menuy + menuheight and menux <= x < menux + menuwidth:
2020-12-11 16:43:46 +00:00
self.menudisplay.handle_click(y - menuy, x - menux, game)
2020-12-11 16:40:56 +00:00
if y <= len(self.title):
self.fg_color = randint(0, 1000), randint(0, 1000), randint(0, 1000)
2020-12-04 13:41:59 +00:00
class PlayerInventoryDisplay(MenuDisplay):
player: Player = None
selected: bool = True
store_mode: bool = False
2020-12-18 14:07:09 +00:00
def update(self, game: Game) -> None:
self.player = game.player
2020-12-18 14:07:09 +00:00
self.update_menu(game.inventory_menu)
self.store_mode = game.state == GameMode.STORE
self.selected = game.state == GameMode.INVENTORY \
or (self.store_mode and not game.is_in_store_menu)
2020-12-18 14:07:09 +00:00
2020-12-04 13:41:59 +00:00
def update_pad(self) -> None:
self.menubox.update_title(_("INVENTORY"))
2020-12-04 15:28:37 +00:00
for i, item in enumerate(self.menu.values):
rep = self.pack[item.name.upper()]
selection = f"[{rep}]" if i == self.menu.position \
and self.selected else f" {rep} "
self.addstr(self.pad, i + 1, 0, selection
+ " " + item.translated_name.capitalize()
+ (": " + str(item.price) + " Hazels"
if self.store_mode else ""))
2020-12-04 15:28:37 +00:00
if self.store_mode:
price = f"{self.pack.HAZELNUT} {self.player.hazel} Hazels"
width = len(price) + (self.pack.tile_width - 1)
self.addstr(self.pad, self.height - 3, self.width - width - 2,
price, italic=True)
2020-12-04 15:28:37 +00:00
@property
def truewidth(self) -> int:
return max(1, self.height if hasattr(self, "height") else 10)
@property
def trueheight(self) -> int:
return 2 + super().trueheight
2020-12-11 17:07:24 +00:00
def handle_click(self, y: int, x: int, game: Game) -> None:
"""
We can select a menu item with the mouse.
"""
self.menu.position = max(0, min(len(self.menu.values) - 1, y - 2))
game.is_in_store_menu = False
2020-12-11 17:07:24 +00:00
game.handle_key_pressed(KeyValues.ENTER)
2020-12-07 20:22:06 +00:00
class StoreInventoryDisplay(MenuDisplay):
menu: StoreMenu
selected: bool = False
2020-12-18 14:07:09 +00:00
def update(self, game: Game) -> None:
self.update_menu(game.store_menu)
self.selected = game.is_in_store_menu
2020-12-18 14:07:09 +00:00
def update_pad(self) -> None:
self.menubox.update_title(_("STALL"))
for i, item in enumerate(self.menu.values):
rep = self.pack[item.name.upper()]
selection = f"[{rep}]" if i == self.menu.position \
and self.selected else f" {rep} "
self.addstr(self.pad, i + 1, 0, selection
2020-12-11 17:17:08 +00:00
+ " " + item.translated_name.capitalize()
+ ": " + str(item.price) + " Hazels")
price = f"{self.pack.HAZELNUT} {self.menu.merchant.hazel} Hazels"
width = len(price) + (self.pack.tile_width - 1)
self.addstr(self.pad, self.height - 3, self.width - width - 2, price,
italic=True)
@property
def truewidth(self) -> int:
return max(1, self.height if hasattr(self, "height") else 10)
2020-12-07 20:22:06 +00:00
@property
def trueheight(self) -> int:
return 2 + super().trueheight
def handle_click(self, y: int, x: int, game: Game) -> None:
"""
We can select a menu item with the mouse.
"""
self.menu.position = max(0, min(len(self.menu.values) - 1, y - 2))
game.is_in_store_menu = True
game.handle_key_pressed(KeyValues.ENTER)