squirrel-battle/dungeonbattle/mapdisplay.py

#!/usr/bin/env python
import curses
from typing import Any

from dungeonbattle.interfaces import Map


class MapDisplay:
    def __init__(self, m: Map, player: Any):
        # TODO Type the player field with the good type
        self.map = m
        self.pad = curses.newpad(m.height, m.width + 1)
        self.player = player

    def update_pad(self) -> None:
        self.pad.addstr(0, 0, self.map.draw_string())
        for e in self.map.entities:
            self.pad.addch(e.y, e.x, e.img)
        self.pad.addstr(self.player.y, self.player.x, '🐿️')

    def display(self, y: int, x: int) -> None:
        deltay, deltax = (curses.LINES // 2) + 1, (curses.COLS // 2) + 1
        pminrow, pmincol = y - deltay, x - deltax
        sminrow, smincol = max(-pminrow, 0), max(-pmincol, 0)
        deltay, deltax = curses.LINES - deltay, curses.COLS - deltax
        smaxrow = self.map.height - (y + deltay) + curses.LINES - 1
        smaxrow = min(smaxrow, curses.LINES - 1)
        smaxcol = self.map.width - (x + deltax) + curses.COLS - 1
        smaxcol = min(smaxcol, curses.COLS - 1)
        pminrow = max(0, min(self.map.height, pminrow))
        pmincol = max(0, min(self.map.width, pmincol))
        self.pad.clear()
        self.update_pad()
        self.pad.refresh(pminrow, pmincol, sminrow, smincol, smaxrow, smaxcol)