summaryrefslogtreecommitdiff
path: root/tui/tui.py
blob: d8c6a5dfc96a42025e749beb42a67ad37cc68c58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import termios, fcntl, sys, os
from tui.static import Action, ActionKeys, ActionEscKeys, Colors, TileTypes

class Tui():
    def __init__(self):
        # Vaatii hieman terminaaliasetusten muokkaamista jotta yksittäiset
        # napin painallukset voidaan lukea
        # https://stackoverflow.com/questions/983354/how-do-i-wait-for-a-pressed-key
        fd = sys.stdin.fileno()
        self.oldterm = termios.tcgetattr(fd)
        
        newattr = termios.tcgetattr(fd)
        newattr[3] = newattr[3] & ~termios.ICANON & ~termios.ECHO
        termios.tcsetattr(fd, termios.TCSANOW, newattr)
        
        self.oldflags = fcntl.fcntl(fd, fcntl.F_GETFL)
        fcntl.fcntl(fd, fcntl.F_SETFL, self.oldflags | os.O_NONBLOCK)
        

    def __del__(self):
        # palautetaan terminaali takaisin alkupetäiseen uskoon
        fd = sys.stdin.fileno()
        termios.tcsetattr(fd, termios.TCSAFLUSH, self.oldterm)
        fcntl.fcntl(fd, fcntl.F_SETFL, self.oldflags)
        

    def set_color(self, color):
        if color>=0 and color<16:
            print(end=f"\033[{'1;' if color//8 else ''}3{color%8}m")


    def set_bg(self, color):
        if color>=0 and color<8:
            print(end=f"\033[4{color}m")


    def cursor_up(self, lines):
        print(end=f"\033[{lines}F")


    def reset_color(self):
        print(end="\033[0m")
        

    def draw_tile(self, tile, hilighted):
        for i in range(len(TileTypes[tile].text)):
            color, bg = TileTypes[tile].colors[i]
            self.set_color(Colors.BLACK if hilighted else color)
            self.set_bg(Colors.CYAN if hilighted else bg)
            print(end=TileTypes[tile].text[i])
            self.reset_color()
        

    def draw_matrix(self, matrix, hx, hy ):
        self.cursor_up(len(matrix[0]))
        for y in range(len(matrix[0])):
            for x in range(len(matrix)):
                self.draw_tile( matrix[x][y], 
                        x == hx and y == hy )
            print()
            
    
    def read_action(self):
        escape = 0
        while True:
            try:
                c = sys.stdin.read(1)
            except:
                continue
            if escape:
                if c in "[0123456789":
                    continue
                if c in ActionEscKeys:
                    return ActionEscKeys[c]
                escape = 0
                continue
            else:
                if c == '\033':
                    escape = 1
                    continue
                if c in ActionKeys:
                    return ActionKeys[c]


    def matrix_selector(self, matrix, x, y ):
        self.draw_matrix(matrix, x, y)
        while True:
            action = self.read_action()
            match action:
                case Action.QUIT:
                    return (action,x,y)
                case Action.OPEN | Action.FLAG:
                    if matrix[x][y]>=10:
                        return (action,x,y)
                case Action.UP:
                    y = y-1 if y>0 else 0
                case Action.LEFT:
                    x = x-1 if x>0 else 0
                case Action.DOWN:
                    y = y+1 if y<len(matrix[0])-1 else y
                case Action.RIGHT:
                    x = x+1 if x<len(matrix)-1 else x
            self.draw_matrix(matrix, x, y)