You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.3 KiB
67 lines
2.3 KiB
2 years ago
|
import os
|
||
|
|
from typing import List
|
||
|
|
from graphics.colors import Colors
|
||
|
|
|
||
|
|
class Layer:
|
||
|
|
def __init__(self, screen, z_index):
|
||
|
|
self.frame = {}
|
||
|
|
self.z_index = z_index
|
||
|
|
self.x = screen.x
|
||
|
|
self.y = screen.y
|
||
|
|
|
||
|
|
def clear(self):
|
||
|
|
self.frame = {}
|
||
|
|
|
||
|
|
def put_char(self, char, x, y, colors: List[Colors]):
|
||
|
|
if x > self.x or x < 0 or y > self.y or y < 0:
|
||
|
|
raise ValueError("out of range pixel")
|
||
|
|
self.frame[(x, y)] = "".join(colors) + char + Colors.RESET
|
||
|
|
|
||
|
|
def put_string(self, string, x, y, colors: List[Colors]):
|
||
|
|
if x > self.x or x < 0 or y > self.y or y < 0:
|
||
|
|
raise ValueError("out of range pixel")
|
||
|
|
string = string.split("\n")
|
||
|
|
for string_y in range(len(string)):
|
||
|
|
for string_x in range(len(string[string_y])):
|
||
|
|
self.frame[(x+string_x, y+string_y)] = "".join(colors) + string[string_y][string_x] + Colors.RESET
|
||
|
|
|
||
|
|
def rect(self, x1, y1, x2, y2, colors: List[Colors], template=' '):
|
||
|
|
if (x1 > self.x or x1 < 0 or y1 > self.y or y1 < 0) or (x2 > self.x or x2 < 0 or y2 > self.y or y2 < 0):
|
||
|
|
raise ValueError("out of range pixel")
|
||
|
|
if len(template) > 1:
|
||
|
|
raise ValueError("template should be 1 char long")
|
||
|
|
for x in range(x1, x2+1):
|
||
|
|
for y in range(y1, y2+1):
|
||
|
|
self.frame[(x, y)] = "".join(colors) + template + Colors.RESET
|
||
|
|
|
||
|
|
|
||
|
|
class Screen:
|
||
|
|
def __init__(self):
|
||
|
|
size = os.get_terminal_size()
|
||
|
|
self.x = size.columns
|
||
|
|
self.y = size.lines - 2
|
||
|
|
self.frame = {}
|
||
|
|
self.__layers = {}
|
||
|
|
|
||
|
|
def set_layer(self, name, layer: Layer):
|
||
|
|
self.__layers[name] = layer
|
||
|
|
|
||
|
|
def __combine_layers(self):
|
||
|
|
self.frame = {}
|
||
|
|
layers = sorted(self.__layers.values(), key=lambda layer: layer.z_index)
|
||
|
|
for layer in layers:
|
||
|
|
self.frame.update(layer.frame)
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
self.__combine_layers()
|
||
|
|
content = ""
|
||
|
|
for y in range(self.y):
|
||
|
|
#content += "\n"
|
||
|
|
for x in range(self.x):
|
||
|
|
if (x, y) in self.frame:
|
||
|
|
content += self.frame[(x, y)]
|
||
|
|
else:
|
||
|
|
content += " "
|
||
|
|
content += "\n"
|
||
|
|
print("\033[H\033[J", end="")
|
||
|
|
print(content)
|