squirrel-battle/dungeonbattle/display/mapdisplay.py

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

from dungeonbattle.display.display import Display
from dungeonbattle.display.texturepack import TexturePack
from dungeonbattle.entities.player import Player
from dungeonbattle.interfaces import Map


class MapDisplay(Display):
    def __init__(self, screen: Any, m: Map, player: Player, pack: TexturePack):
        super().__init__(screen)
        self.height = self.rows
        self.width = self.cols
        self.pack = pack
        self.map = m
        self.player = player
        self.pad = self.newpad(m.height, m.width + 1)

    def update_pad(self) -> None:
        self.pad.addstr(0, 0, self.map.draw_string(self.pack))
        for e in self.map.entities:
            self.pad.addstr(e.y, e.x, self.pack.PLAYER)

    def display(self, y: int, x: int) -> None:
        deltay, deltax = (self.height // 2) + 1, (self.width // 2) + 1
        pminrow, pmincol = y - deltay, x - deltax
        sminrow, smincol = max(-pminrow, 0), max(-pmincol, 0)
        deltay, deltax = self.height - deltay, self.width - deltax
        smaxrow = self.map.height - (y + deltay) + self.height - 1
        smaxrow = min(smaxrow, self.height - 1)
        smaxcol = self.map.width - (x + deltax) + self.width - 1
        smaxcol = min(smaxcol, self.width - 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)

    def refresh(self) -> None:
        return self.display(self.player.y, self.player.x)