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
|
import config
import threading
import numpy
from ctypes import *
class COLOR(Structure):
_fields_ = [('r', c_uint8), ('g', c_uint8), ('b', c_uint8), ('a', c_uint8)]
class FRAMEBUFFER(Structure):
_fields_ = [('data', POINTER(COLOR)), ('w', c_size_t), ('h', c_size_t)]
lib = CDLL('./libbdf.so')
lib.read_bdf_file.restype = c_void_p
lib.framebuffer_render_text.restype = POINTER(FRAMEBUFFER)
lib.framebuffer_render_text.argtypes= [c_char_p, c_void_p, c_void_p, c_size_t, c_size_t, c_size_t]
dbuf = numpy.zeros(config.frame_size*4, dtype=numpy.uint8)
printlock = threading.Lock()
def printframe(fb):
with printlock:
print('\0337\033[H')
rgba = len(fb) == config.frame_size*4
ip = numpy.frombuffer(fb, dtype=numpy.uint8)
numpy.copyto(dbuf[0::4], ip[0::3+rgba])
numpy.copyto(dbuf[1::4], ip[1::3+rgba])
numpy.copyto(dbuf[2::4], ip[2::3+rgba])
lib.console_render_buffer(dbuf.ctypes.data_as(POINTER(c_uint8)), config.display_width, config.display_height)
class Font:
def __init__(self, fontfile='unifont.bdf'):
self.font = lib.read_bdf_file(fontfile)
assert self.font
# hack to prevent unlocalized memory leak arising from ctypes/numpy/cpython interaction
self.cbuf = create_string_buffer(config.frame_size*sizeof(COLOR))
self.cbuflock = threading.Lock()
def compute_text_bounds(self, text):
textbytes = text.encode()
textw, texth = c_size_t(0), c_size_t(0)
res = lib.framebuffer_get_text_bounds(textbytes, self.font, byref(textw), byref(texth))
if res:
raise RuntimeError('Invalid text')
return textw.value, texth.value
def render_text(self, text, offset):
with self.cbuflock:
textbytes = bytes(str(text), 'UTF-8')
res = lib.framebuffer_render_text(textbytes, self.font, self.cbuf,
config.display_width, config.display_height, offset)
if res:
raise RuntimeError('Invalid text')
return self.cbuf
unifont = Font()
|
