2.23inch/type.py at main · frogCaller/2.23inch · GitHub

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
104
105
106
107
108
import os
import time
from PIL import Image, ImageDraw, ImageFont
import curses
from drive import SSD1305

# Initialize display
disp = SSD1305.SSD1305()
disp.Init()
disp.clear()
width = disp.width
height = disp.height
image = Image.new('1', (width, height))
draw = ImageDraw.Draw(image)
draw.rectangle((0, 0, width, height), outline=0, fill=0)
padding = 0
top = padding
bottom = height - padding
x = 0
timeframe = 0.01
font = ImageFont.truetype('04B_08__.TTF', 8)
line_height = font.getsize("A")[1]

char_width = font.getsize("A")[0]
max_chars_per_line = width // char_width

MAX_LINES = height // line_height

input_string = ""

def buffer(sec):
    disp.getbuffer(image)
    disp.ShowImage()
    time.sleep(sec)
    draw.rectangle((0, 0, width, height), outline=0, fill=0)

def linetext(content, line_number):
    draw.rectangle((0, line_number * line_height, width, (line_number + 1) * line_height), outline=0, fill=0)
    draw.text((x, line_number * line_height), content, font=font, fill=255)

def wrap_text(text, max_line_length):
    wrapped_lines = []
    lines = text.split('\n')

    for line in lines:
        current_line = ""
        for char in line:
            if len(current_line) < max_line_length:
                current_line += char
            else:
                wrapped_lines.append(current_line)
                current_line = char

        if current_line:
            wrapped_lines.append(current_line)

    return wrapped_lines

def main(stdscr):
    curses.curs_set(0)
    stdscr.nodelay(1)
    stdscr.timeout(0)

    global input_string
    current_line = 0

    with open("type.txt", "a") as f:
        while True:
            key = stdscr.getch()

            if key != -1:
                if key == 27:  # ESC key
                    input_string = ""  # Reset input string
                    current_line = 0
                    buffer(timeframe)
                elif key == curses.KEY_BACKSPACE or key == 127:  # Backspace key
                    if input_string:
                        input_string = input_string[:-1]  # Remove last character
                        wrapped_lines = wrap_text(input_string, max_chars_per_line)
                        start_line = max(0, len(wrapped_lines) - MAX_LINES)
                        for i, line in enumerate(wrapped_lines[start_line:]):
                            linetext(line, i)
                        current_line = len(wrapped_lines) - 1
                        buffer(timeframe)
                        f.seek(0)
                        f.truncate()  # Clear the file content before writing new content
                        f.write(input_string)
                elif key == curses.KEY_ENTER or key == 10:  # Enter key
                    input_string += '\n'
                    current_line += 1
                    f.write('\n')  # Write the newline character to the file
                else:
                    key_char = chr(key) if key < 256 else f"Special key {key}"
                    input_string += key_char
                    wrapped_lines = wrap_text(input_string, max_chars_per_line)
                    start_line = max(0, len(wrapped_lines) - MAX_LINES)
                    for i, line in enumerate(wrapped_lines[start_line:]):
                        linetext(line, i)
                    current_line = len(wrapped_lines) - 1
                    buffer(timeframe)
                    f.write(key_char)

            stdscr.refresh()

            if input_string and input_string[0] == curses.KEY_F1:
                break

curses.wrapper(main)