Adolfo Reyna
258 lines
11 KiB
Python
258 lines
11 KiB
Python
import time
|
|
from machine import Pin, SPI
|
|
import framebuf
|
|
import micropython
|
|
import struct
|
|
|
|
class RLCD:
|
|
def __init__(self, spi, cs, dc, rst, width=400, height=300):
|
|
self.spi = spi
|
|
self.cs = cs
|
|
self.dc = dc
|
|
self.rst = rst
|
|
self.width = width
|
|
self.height = height
|
|
|
|
# 1. Hardware Buffer (The weird format the screen needs)
|
|
self.hw_len = (self.width * self.height) // 8
|
|
self.hw_buffer = bytearray(self.hw_len)
|
|
|
|
# 2. Virtual Canvas (Standard format for drawing lines, text, bitmaps)
|
|
# We use MONO_HLSB (Standard Horizontal Byte layout)
|
|
self.canvas_buffer = bytearray(self.hw_len)
|
|
self.canvas = framebuf.FrameBuffer(self.canvas_buffer, self.width, self.height, framebuf.MONO_HLSB)
|
|
|
|
# Init Pins
|
|
self.cs.init(self.cs.OUT, value=1)
|
|
self.dc.init(self.dc.OUT, value=0)
|
|
self.rst.init(self.rst.OUT, value=1)
|
|
|
|
self.reset()
|
|
self.init_display()
|
|
|
|
# --- DRAWING WRAPPERS ---
|
|
def pixel(self, x, y, c): self.canvas.pixel(x, y, c)
|
|
def line(self, x1, y1, x2, y2, c): self.canvas.line(x1, y1, x2, y2, c)
|
|
def rect(self, x, y, w, h, c): self.canvas.rect(x, y, w, h, c)
|
|
def fill_rect(self, x, y, w, h, c): self.canvas.fill_rect(x, y, w, h, c)
|
|
def text(self, msg, x, y, c=1): self.canvas.text(msg, x, y, c)
|
|
def clear(self, c=0): self.canvas.fill(c)
|
|
|
|
# --- SCALABLE TEXT ---
|
|
def text_large(self, msg, x, y, scale=2, c=1):
|
|
char_w = 8; char_h = 8
|
|
tmp_buf = bytearray(char_w * char_h // 8)
|
|
tmp_fb = framebuf.FrameBuffer(tmp_buf, char_w, char_h, framebuf.MONO_HLSB)
|
|
for char in msg:
|
|
tmp_fb.fill(0); tmp_fb.text(char, 0, 0, 1)
|
|
for py in range(8):
|
|
for px in range(8):
|
|
if tmp_fb.pixel(px, py):
|
|
self.canvas.fill_rect(x + (px * scale), y + (py * scale), scale, scale, c)
|
|
x += (8 * scale)
|
|
|
|
# --- RAW BITMAPS (1:1 scale) ---
|
|
def bitmap(self, x, y, w, h, pixel_data):
|
|
img = framebuf.FrameBuffer(pixel_data, w, h, framebuf.MONO_HLSB)
|
|
self.canvas.blit(img, x, y)
|
|
|
|
# --- PBM FILE LOADER WITH SCALING (NEW!) ---
|
|
def draw_pbm(self, filename, x, y, scale=1):
|
|
try:
|
|
with open(filename, 'rb') as f:
|
|
line1 = f.readline()
|
|
if not line1.startswith(b'P4'): print("Err: Not P4 PBM"); return
|
|
while True:
|
|
line = f.readline()
|
|
if not line.startswith(b'#'): break
|
|
dims = line.split(); w = int(dims[0]); h = int(dims[1])
|
|
data = bytearray(f.read())
|
|
|
|
# Create temp buffer for source image
|
|
src_fb = framebuf.FrameBuffer(data, w, h, framebuf.MONO_HLSB)
|
|
|
|
if scale == 1:
|
|
self.canvas.blit(src_fb, x, y) # Fast path
|
|
else:
|
|
# Slow path: iterate pixels and draw scaled rects
|
|
for sy in range(h):
|
|
for sx in range(w):
|
|
if src_fb.pixel(sx, sy):
|
|
self.canvas.fill_rect(x + (sx * scale), y + (sy * scale), scale, scale, 1)
|
|
print(f"Loaded {filename} (scale {scale})")
|
|
except OSError:
|
|
print(f"Error: Could not open {filename}")
|
|
|
|
def draw_bmp(self, filename, x=0, y=0):
|
|
"""Draw a 24-bit or 32-bit uncompressed color BMP image converted to 1-bit monochrome at (x, y) coordinates."""
|
|
try:
|
|
with open(filename, 'rb') as f:
|
|
header = f.read(54)
|
|
if len(header) < 54 or header[0:2] != b'BM':
|
|
print("Err: Not a valid BMP file")
|
|
return False
|
|
|
|
pixel_offset = struct.unpack('<I', header[10:14])[0]
|
|
width, height = struct.unpack('<ii', header[18:26])
|
|
planes, bpp = struct.unpack('<HH', header[26:30])
|
|
compression = struct.unpack('<I', header[30:34])[0]
|
|
|
|
if bpp not in (24, 32):
|
|
print("Err: Only 24-bit and 32-bit BMP formats supported")
|
|
return False
|
|
|
|
if compression != 0:
|
|
print("Err: Only uncompressed BMP supported")
|
|
return False
|
|
|
|
f.seek(pixel_offset)
|
|
bottom_up = True
|
|
if height < 0:
|
|
height = -height
|
|
bottom_up = False
|
|
|
|
row_bytes = (width * bpp) // 8
|
|
row_padded = ((width * bpp + 31) // 32) * 4
|
|
|
|
read_buf = bytearray(row_padded)
|
|
|
|
for row_idx in range(height):
|
|
n = f.readinto(read_buf)
|
|
if n < row_padded:
|
|
break
|
|
|
|
screen_y = y + (height - 1 - row_idx) if bottom_up else y + row_idx
|
|
if screen_y < 0 or screen_y >= self.height:
|
|
continue
|
|
|
|
for px in range(width):
|
|
screen_x = x + px
|
|
if screen_x < 0 or screen_x >= self.width:
|
|
continue
|
|
|
|
if bpp == 24:
|
|
b = read_buf[px * 3]
|
|
g = read_buf[px * 3 + 1]
|
|
r = read_buf[px * 3 + 2]
|
|
else: # 32-bit
|
|
b = read_buf[px * 4]
|
|
g = read_buf[px * 4 + 1]
|
|
r = read_buf[px * 4 + 2]
|
|
|
|
# Convert to monochrome (0 = White, 1 = Black)
|
|
lum = (r * 299 + g * 587 + b * 114) // 1000
|
|
c = 1 if lum < 128 else 0
|
|
self.canvas.pixel(screen_x, screen_y, c)
|
|
|
|
self.show()
|
|
return True
|
|
except Exception as e:
|
|
print("Error drawing BMP on RLCD:", e)
|
|
return False
|
|
|
|
def draw_rgb565(self, x, y, w, h, data):
|
|
"""Draw raw RGB565 pixel data converted to 1-bit monochrome on the RLCD."""
|
|
for cy in range(h):
|
|
screen_y = y + cy
|
|
if screen_y < 0 or screen_y >= self.height:
|
|
continue
|
|
for cx in range(w):
|
|
screen_x = x + cx
|
|
if screen_x < 0 or screen_x >= self.width:
|
|
continue
|
|
idx = (cy * w + cx) * 2
|
|
h_byte = data[idx]
|
|
l_byte = data[idx + 1]
|
|
# Extract RGB from RGB565
|
|
r = (h_byte & 0xF8)
|
|
g = ((h_byte & 0x07) << 5) | ((l_byte & 0xE0) >> 3)
|
|
b = (l_byte & 0x1F) << 3
|
|
# Convert to luminance (0 = White, 1 = Black in RLCD)
|
|
lum = (r * 299 + g * 587 + b * 114) // 1000
|
|
c = 1 if lum < 128 else 0
|
|
self.canvas.pixel(screen_x, screen_y, c)
|
|
self.show()
|
|
return True
|
|
|
|
# --- SCREENSHOT ---
|
|
def save_screenshot(self, filename):
|
|
print(f"Saving screenshot to {filename}...")
|
|
try:
|
|
with open(filename, 'wb') as f:
|
|
f.write(b'P4\n')
|
|
f.write(f"{self.width} {self.height}\n".encode())
|
|
f.write(self.canvas_buffer)
|
|
print("Saved!")
|
|
except Exception as e:
|
|
print(f"Error saving screenshot: {e}")
|
|
|
|
# --- HARDWARE LOGIC ---
|
|
def reset(self):
|
|
self.rst(1); time.sleep_ms(50); self.rst(0); time.sleep_ms(20); self.rst(1); time.sleep_ms(50)
|
|
|
|
def write_cmd(self, cmd):
|
|
self.cs(0); self.dc(0); self.spi.write(bytearray([cmd])); self.cs(1)
|
|
|
|
def write_data(self, data):
|
|
self.cs(0); self.dc(1)
|
|
if isinstance(data, int): self.spi.write(bytearray([data]))
|
|
elif isinstance(data, list): self.spi.write(bytearray(data))
|
|
else: self.spi.write(data)
|
|
self.cs(1)
|
|
|
|
def init_display(self):
|
|
self.write_cmd(0xD6); self.write_data(0x17); self.write_data(0x02)
|
|
self.write_cmd(0xD1); self.write_data(0x01)
|
|
self.write_cmd(0xC0); self.write_data(0x11); self.write_data(0x04)
|
|
self.write_cmd(0xC1); self.write_data([0x41, 0x41, 0x41, 0x41])
|
|
self.write_cmd(0xC2); self.write_data([0x19, 0x19, 0x19, 0x19])
|
|
self.write_cmd(0xC4); self.write_data([0x41, 0x41, 0x41, 0x41])
|
|
self.write_cmd(0xC5); self.write_data([0x19, 19, 0x19, 0x19])
|
|
self.write_cmd(0xD8); self.write_data(0xA6); self.write_data(0xE9)
|
|
self.write_cmd(0xB2); self.write_data(0x05)
|
|
self.write_cmd(0xB3); self.write_data([0xE5, 0xF6, 0x05, 0x46, 0x77, 0x77, 0x77, 0x77, 0x76, 0x45])
|
|
self.write_cmd(0xB4); self.write_data([0x05, 0x46, 0x77, 0x77, 0x77, 0x77, 0x76, 0x45])
|
|
self.write_cmd(0x62); self.write_data([0x32, 0x03, 0x1F])
|
|
self.write_cmd(0xB7); self.write_data(0x13)
|
|
self.write_cmd(0xB0); self.write_data(0x64)
|
|
self.write_cmd(0x11); time.sleep_ms(200)
|
|
self.write_cmd(0xC9); self.write_data(0x00)
|
|
self.write_cmd(0x36); self.write_data(0x48)
|
|
self.write_cmd(0x3A); self.write_data(0x11)
|
|
self.write_cmd(0xB9); self.write_data(0x20)
|
|
self.write_cmd(0xB8); self.write_data(0x29)
|
|
self.write_cmd(0x21) # Inversion ON (Black Background by default)
|
|
self.write_cmd(0x2A); self.write_data([0x12, 0x2A])
|
|
self.write_cmd(0x2B); self.write_data([0x00, 0xC7])
|
|
self.write_cmd(0x35); self.write_data(0x00)
|
|
self.write_cmd(0xD0); self.write_data(0xFF)
|
|
self.write_cmd(0x38); self.write_cmd(0x29)
|
|
|
|
def invert(self, enable):
|
|
if enable:
|
|
self.write_cmd(0x21) # Inversion ON (Black Background)
|
|
else:
|
|
self.write_cmd(0x20) # Inversion OFF (White Background)
|
|
|
|
# --- THE HEAVY LIFTER (Optimized) ---
|
|
@micropython.native
|
|
def show(self):
|
|
HEIGHT = 300; WIDTH = 400
|
|
for i in range(len(self.hw_buffer)): self.hw_buffer[i] = 0
|
|
for y in range(HEIGHT):
|
|
for x in range(WIDTH):
|
|
if self.canvas.pixel(x, y):
|
|
inv_y = HEIGHT - 1 - y
|
|
byte_x = x // 2
|
|
block_y = inv_y // 4
|
|
index = byte_x * 75 + block_y
|
|
local_x = x % 2
|
|
local_y = inv_y % 4
|
|
bit = 7 - (local_y * 2 + local_x)
|
|
self.hw_buffer[index] |= (1 << bit)
|
|
self.write_cmd(0x2A); self.write_data(0x12); self.write_data(0x2A)
|
|
self.write_cmd(0x2B); self.write_data(0x00); self.write_data(0xC7)
|
|
self.write_cmd(0x2C)
|
|
self.cs(0); self.dc(1)
|
|
self.spi.write(self.hw_buffer)
|
|
self.cs(1) |