squirrel-battle/squirrelbattle/display/mapdisplay.py

#!/usr/bin/env python
from squirrelbattle.interfaces import Map
from .display import Display


class MapDisplay(Display):

    def __init__(self, *args):
        super().__init__(*args)

    def update_map(self, m: Map) -> None:
        self.map = m
        self.pad = self.newpad(m.height, self.pack.tile_width * m.width + 1)

    def update_pad(self) -> None:
        self.init_pair(1, self.pack.tile_fg_color, self.pack.tile_bg_color)
        self.init_pair(2, self.pack.entity_fg_color, self.pack.entity_bg_color)
        self.addstr(self.pad, 0, 0, self.map.draw_string(self.pack),
                    self.color_pair(1))
        for e in self.map.entities:
            self.addstr(self.pad, e.y, self.pack.tile_width * e.x,
                        self.pack[e.name.upper()], self.color_pair(2))

    def display(self) -> None:
        y, x = self.map.currenty, self.pack.tile_width * self.map.currentx
        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.pack.tile_width * self.map.width - \
            (x + deltax) + self.width - 1

        # Wrap perfectly the map according to the width of the tiles
        pmincol = self.pack.tile_width * (pmincol // self.pack.tile_width)
        smincol = self.pack.tile_width * (smincol // self.pack.tile_width)
        smaxcol = self.pack.tile_width \
            * (smaxcol // self.pack.tile_width + 1) - 1

        smaxcol = min(smaxcol, self.width - 1)
        pminrow = max(0, min(self.map.height, pminrow))
        pmincol = max(0, min(self.pack.tile_width * self.map.width, pmincol))

        self.pad.erase()
        self.update_pad()
        self.refresh_pad(self.pad, pminrow, pmincol, sminrow, smincol, smaxrow,
                         smaxcol)