squirrel-battle/dungeonbattle/display/menudisplay.py

80 lines
2.8 KiB
Python

from dungeonbattle.menus import Menu, MainMenu
from typing import Any
from .display import Display
import curses
class MenuDisplay(Display):
position: int
def __init__(self, *args) :
super().__init__(*args)
self.menubox = self.newpad(self.rows,self.cols)
def update_menu(self, menu: Menu):
self.menu = menu
self.values = [ str(a) for a in menu.values ]
self.trueheight = len(self.values)
self.truewidth = max([len(a) for a in self.values])
#Menu values are printed in pad
self.pad = self.newpad(self.trueheight,self.truewidth+2)
for i in range(self.trueheight) :
self.pad.addstr(i,0," " + self.values[i])
def update_pad(self) -> None:
for i in range(self.trueheight):
self.pad.addstr(i,0," ")
self.pad.addstr(self.menu.position,0,">") #set a marker on the selected line
def display(self) -> None:
if self.height-2>=self.menu.position-1 :
cornery = 0
elif self.height-2 >= self.trueheight-self.menu.position :
cornery = self.trueheight-self.height+2
#Menu box
self.menubox.addstr(0,0,""+""*(self.width-2)+"")
for i in range(1,self.height-1) :
self.menubox.addstr(i,0,""+" "*(self.width-2)+"")
self.menubox.addstr(self.height-1, 0, ""+""*(self.width-2)+"")
self.menubox.refresh(0, 0, self.y, self.x,
self.height + self.y,
self.width + self.x)
self.update_pad()
self.pad.refresh(cornery, 0, self.y+1, self.x+2,
self.height-1 + self.y,
self.width-1 + self.x)
@property
def preferred_width(self) -> int:
return self.truewidth + 6
@property
def preferred_height(self) -> int:
return self.trueheight + 2
class MainMenuDisplay(Display):
def __init__(self, menu : MainMenu, *args) :
super().__init__(*args)
self.menu = menu
self.pad = self.newpad(self.rows, self.cols)
with open("resources/ascii_art.txt", "r") as file:
self.title = file.read().split("\n")
self.menudisplay = MenuDisplay(self.screen, self.pack)
self.menudisplay.update_menu(self.menu)
def display(self) -> None:
for i in range(len(self.title)) :
self.pad.addstr(4+i,self.width//2-len(self.title[0])//2-1,self.title[i])
self.pad.refresh(0,0,self.y,self.x,self.height,self.width)
menuwidth = min(self.menudisplay.preferred_width, self.width)
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)