73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import os
|
|
import fcntl
|
|
import struct
|
|
import mmap
|
|
from time import sleep
|
|
|
|
with open('/dev/tty1', 'w') as tty:
|
|
tty.write("\033[?25l")
|
|
|
|
|
|
# Constants for the ioctl call
|
|
FBIOGET_VSCREENINFO = 0x4600
|
|
|
|
# Define a function to perform an ioctl call to get screen info
|
|
def get_screen_info(fb_device):
|
|
# We will get the variable screen info (resolution and bits per pixel)
|
|
# The struct for variable screen info is defined in the Linux kernel headers
|
|
# as fb_var_screeninfo, which is 56 bytes
|
|
screen_info_format = '8I12I16I2I'
|
|
screen_info = struct.pack(screen_info_format, *([0] * 38))
|
|
screen_info = fcntl.ioctl(fb_device, FBIOGET_VSCREENINFO, screen_info)
|
|
return struct.unpack(screen_info_format, screen_info)
|
|
|
|
# Open the framebuffer device in read-write mode
|
|
with open("/dev/fb0", "r+b") as fb:
|
|
# Get screen information using ioctl
|
|
info = get_screen_info(fb)
|
|
xres, yres, bits_per_pixel = info[0], info[1], info[6]
|
|
screensize = xres * yres * bits_per_pixel // 8
|
|
|
|
# Create a memory-mapped area to the framebuffer
|
|
fb_bytes = mmap.mmap(fb.fileno(), screensize, mmap.MAP_SHARED, mmap.PROT_WRITE | mmap.PROT_READ)
|
|
|
|
# Now you can write pixel data to the framebuffer
|
|
# Here is an example that just sets all pixels to black
|
|
fb_bytes.seek(0) # Go to the beginning of the framebuffer
|
|
fb_bytes.write(b'\x00' * screensize)
|
|
|
|
# Your framebuffer application code here
|
|
|
|
# Once your program is finished, you can unmap and close the framebuffer device
|
|
fb_bytes.close()
|
|
|
|
import sys
|
|
import termios
|
|
import atexit
|
|
from select import select
|
|
|
|
# Save the terminal settings
|
|
fd = sys.stdin.fileno()
|
|
old_settings = termios.tcgetattr(fd)
|
|
atexit.register(lambda: termios.tcsetattr(fd, termios.TCSADRAIN, old_settings))
|
|
|
|
def wait_for_key_press():
|
|
# Disable canonical input and echo
|
|
new_settings = termios.tcgetattr(fd)
|
|
new_settings[3] = new_settings[3] & ~termios.ICANON & ~termios.ECHO
|
|
termios.tcsetattr(fd, termios.TCSADRAIN, new_settings)
|
|
|
|
# Wait for input
|
|
rlist, _, _ = select([sys.stdin], [], [], None)
|
|
if rlist:
|
|
sys.stdin.read(1)
|
|
|
|
# Call the function to wait for any key press
|
|
wait_for_key_press()
|
|
|
|
with open('/dev/tty1', 'w') as tty:
|
|
tty.write("\033[0;0H")
|
|
tty.write("\033[?5h\033[?5l")
|
|
tty.write("\033[?25h")
|
|
|