squirrel-battle/dungeonbattle/display/menudisplay.py

from dungeonbattle.menus import Menu, MainMenu
from .display import Display


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) -> None:
        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, " ")
        # set a marker on the selected line
        self.pad.addstr(self.menu.position, 0, ">")

    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)