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 ValueError('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 ValueError('Invalid text') return self.cbuf unifont = Font()