squirrel-battle/squirrelbattle/display/display.py

161 lines
5.4 KiB
Python
Raw Normal View History

2020-11-27 15:33:17 +00:00
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later
2020-11-06 14:34:24 +00:00
import curses
2020-11-10 19:43:30 +00:00
from typing import Any, Optional, Union
2020-11-08 22:40:03 +00:00
2020-11-19 01:18:08 +00:00
from squirrelbattle.display.texturepack import TexturePack
from squirrelbattle.tests.screen import FakePad
2020-11-06 14:34:24 +00:00
class Display:
2020-11-10 19:34:22 +00:00
x: int
y: int
width: int
height: int
pad: Any
2020-11-10 19:34:22 +00:00
2020-11-10 19:43:30 +00:00
def __init__(self, screen: Any, pack: Optional[TexturePack] = None):
2020-11-06 14:34:24 +00:00
self.screen = screen
2020-11-10 19:43:30 +00:00
self.pack = pack or TexturePack.get_pack("ascii")
2020-11-08 22:40:03 +00:00
def newpad(self, height: int, width: int) -> Union[FakePad, Any]:
2020-11-08 22:48:26 +00:00
return curses.newpad(height, width) if self.screen else FakePad()
def truncate(self, msg: str, height: int, width: int) -> str:
height = max(0, height)
width = max(0, width)
lines = msg.split("\n")
lines = lines[:height]
lines = [line[:width] for line in lines]
return "\n".join(lines)
def addstr(self, pad: Any, y: int, x: int, msg: str, *options) -> None:
"""
Display a message onto the pad.
If the message is too large, it is truncated vertically and horizontally
"""
height, width = pad.getmaxyx()
msg = self.truncate(msg, height - y, width - x - 1)
if msg.replace("\n", "") and x >= 0 and y >= 0:
return pad.addstr(y, x, msg, *options)
def init_pair(self, number: int, foreground: int, background: int) -> None:
return curses.init_pair(number, foreground, background) \
if self.screen else None
def color_pair(self, number: int) -> int:
return curses.color_pair(number) if self.screen else 0
def resize(self, y: int, x: int, height: int, width: int,
resize_pad: bool = True) -> None:
self.x = x
self.y = y
self.width = width
self.height = height
if hasattr(self, "pad") and resize_pad and \
self.height >= 0 and self.width >= 0:
2020-11-20 17:06:43 +00:00
self.pad.resize(self.height + 1, self.width + 1)
def refresh(self, *args, resize_pad: bool = True) -> None:
if len(args) == 4:
self.resize(*args, resize_pad)
self.display()
2020-11-10 19:34:22 +00:00
def refresh_pad(self, pad: Any, top_y: int, top_x: int,
window_y: int, window_x: int,
last_y: int, last_x: int) -> None:
"""
Refresh a pad on a part of the window.
The refresh starts at coordinates (top_y, top_x) from the pad,
and is drawn from (window_y, window_x) to (last_y, last_x).
If coordinates are invalid (negative indexes/length..., then nothing
is drawn and no error is raised.
"""
top_y, top_x = max(0, top_y), max(0, top_x)
window_y, window_x = max(0, window_y), max(0, window_x)
2020-11-26 21:32:25 +00:00
screen_max_y, screen_max_x = self.screen.getmaxyx() if self.screen \
2020-11-27 13:04:51 +00:00
else (42, 42)
last_y, last_x = min(screen_max_y - 1, last_y), \
min(screen_max_x - 1, last_x)
if last_y >= window_y and last_x >= window_x:
# Refresh the pad only if coordinates are valid
pad.refresh(top_y, top_x, window_y, window_x, last_y, last_x)
2020-11-10 19:34:22 +00:00
def display(self) -> None:
raise NotImplementedError
@property
def rows(self) -> int:
return curses.LINES if self.screen else 42
@property
def cols(self) -> int:
return curses.COLS if self.screen else 42
class VerticalSplit(Display):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.pad = self.newpad(self.rows, 1)
@property
def width(self) -> int:
return 1
@width.setter
def width(self, val: Any) -> None:
pass
def display(self) -> None:
for i in range(self.height):
self.addstr(self.pad, i, 0, "")
self.refresh_pad(self.pad, 0, 0, self.y, self.x,
self.y + self.height - 1, self.x)
class HorizontalSplit(Display):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.pad = self.newpad(1, self.cols)
@property
def height(self) -> int:
return 1
@height.setter
def height(self, val: Any) -> None:
pass
def display(self) -> None:
for i in range(self.width):
self.addstr(self.pad, 0, i, "")
self.refresh_pad(self.pad, 0, 0, self.y, self.x, self.y,
self.x + self.width - 1)
2020-11-20 17:06:43 +00:00
2020-11-20 17:09:39 +00:00
2020-11-20 17:06:43 +00:00
class Box(Display):
2020-11-27 16:52:26 +00:00
def __init__(self, *args, fg_border_color: Optional[int] = None, **kwargs):
2020-11-20 17:06:43 +00:00
super().__init__(*args, **kwargs)
self.pad = self.newpad(self.rows, self.cols)
2020-11-27 16:52:26 +00:00
self.fg_border_color = fg_border_color or curses.COLOR_WHITE
pair_number = 4 + self.fg_border_color
self.init_pair(pair_number, self.fg_border_color, curses.COLOR_BLACK)
self.pair = self.color_pair(pair_number)
2020-11-20 17:09:39 +00:00
2020-11-20 17:06:43 +00:00
def display(self) -> None:
2020-11-27 16:52:26 +00:00
self.addstr(self.pad, 0, 0, "" + "" * (self.width - 2) + "",
self.pair)
2020-11-20 17:06:43 +00:00
for i in range(1, self.height - 1):
2020-11-27 16:52:26 +00:00
self.addstr(self.pad, i, 0, "", self.pair)
self.addstr(self.pad, i, self.width - 1, "", self.pair)
self.addstr(self.pad, self.height - 1, 0,
2020-11-27 16:52:26 +00:00
"" + "" * (self.width - 2) + "", self.pair)
self.refresh_pad(self.pad, 0, 0, self.y, self.x,
self.y + self.height - 1, self.x + self.width - 1)