108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
import curses
|
|
from typing import Any, Optional, Union
|
|
|
|
from squirrelbattle.display.texturepack import TexturePack
|
|
from squirrelbattle.tests.screen import FakePad
|
|
|
|
|
|
class Display:
|
|
x: int
|
|
y: int
|
|
width: int
|
|
height: int
|
|
pad: Any
|
|
|
|
def __init__(self, screen: Any, pack: Optional[TexturePack] = None):
|
|
self.screen = screen
|
|
self.pack = pack or TexturePack.get_pack("ascii")
|
|
|
|
def newpad(self, height: int, width: int) -> Union[FakePad, Any]:
|
|
return curses.newpad(height, width) if self.screen else FakePad()
|
|
|
|
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:
|
|
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()
|
|
|
|
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.pad.addstr(i, 0, "┃")
|
|
self.pad.refresh(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.pad.addstr(0, i, "━")
|
|
self.pad.refresh(0, 0, self.y, self.x, self.y, self.x + self.width - 1)
|
|
|
|
class Box(Display):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.pad = self.newpad(self.rows, self.cols)
|
|
|
|
def display(self) -> None:
|
|
self.pad.addstr(0, 0, "┏" + "━" * (self.width - 2) + "┓")
|
|
for i in range(1, self.height - 1):
|
|
self.pad.addstr(i, 0, "┃")
|
|
self.pad.addstr(i, self.width - 1, "┃")
|
|
self.pad.addstr(self.height - 1, 0,
|
|
"┗" + "━" * (self.width - 2) + "┛")
|
|
self.pad.refresh(0, 0, self.y, self.x, self.y + self.height - 1, self.x + self.width - 1)
|