102 lines
3.1 KiB
Rust
102 lines
3.1 KiB
Rust
|
use core::convert::Infallible;
|
||
|
|
|
||
|
|
use embassy_rp::{peripherals::PIO0, pio_programs::ws2812::PioWs2812};
|
||
|
|
use embedded_graphics::{
|
||
|
|
Pixel,
|
||
|
|
pixelcolor::Rgb888,
|
||
|
|
prelude::{DrawTarget, OriginDimensions, Point, RgbColor, Size},
|
||
|
|
};
|
||
|
|
use rgb::RGB8;
|
||
|
|
|
||
|
|
// Sourced from the `smart-leds` crate
|
||
|
|
const GAMMA8: [u8; 256] = [
|
||
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1,
|
||
|
|
1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5, 5,
|
||
|
|
5, 6, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 9, 9, 9, 10, 10, 10, 11, 11, 11, 12, 12, 13, 13, 13, 14,
|
||
|
|
14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 20, 20, 21, 21, 22, 22, 23, 24, 24, 25, 25, 26, 27,
|
||
|
|
27, 28, 29, 29, 30, 31, 32, 32, 33, 34, 35, 35, 36, 37, 38, 39, 39, 40, 41, 42, 43, 44, 45, 46,
|
||
|
|
47, 48, 49, 50, 50, 51, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 66, 67, 68, 69, 70, 72,
|
||
|
|
73, 74, 75, 77, 78, 79, 81, 82, 83, 85, 86, 87, 89, 90, 92, 93, 95, 96, 98, 99, 101, 102, 104,
|
||
|
|
105, 107, 109, 110, 112, 114, 115, 117, 119, 120, 122, 124, 126, 127, 129, 131, 133, 135, 137,
|
||
|
|
138, 140, 142, 144, 146, 148, 150, 152, 154, 156, 158, 160, 162, 164, 167, 169, 171, 173, 175,
|
||
|
|
177, 180, 182, 184, 186, 189, 191, 193, 196, 198, 200, 203, 205, 208, 210, 213, 215, 218, 220,
|
||
|
|
223, 225, 228, 231, 233, 236, 239, 241, 244, 247, 249, 252, 255,
|
||
|
|
];
|
||
|
|
|
||
|
|
pub fn gamma(color: Rgb888) -> Rgb888 {
|
||
|
|
Rgb888::new(
|
||
|
|
GAMMA8[color.r() as usize],
|
||
|
|
GAMMA8[color.g() as usize],
|
||
|
|
GAMMA8[color.b() as usize],
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct Serpentine {
|
||
|
|
pub size: Size,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Serpentine {
|
||
|
|
fn map(&self, mut p: Point) -> Option<usize> {
|
||
|
|
if p.x < 0 || p.x as u32 >= self.size.width || p.y < 0 || p.y as u32 >= self.size.height {
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
|
||
|
|
if p.y % 2 == 0 {
|
||
|
|
p.x = self.size.width as i32 - 1 - p.x;
|
||
|
|
}
|
||
|
|
|
||
|
|
Some(p.y as usize * self.size.width as usize + p.x as usize)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct LedMatrix<'d> {
|
||
|
|
buffer: [RGB8; 320],
|
||
|
|
layout: Serpentine,
|
||
|
|
pio: PioWs2812<'d, PIO0, 0, 320>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'d> LedMatrix<'d> {
|
||
|
|
pub fn new(layout: Serpentine, pio: PioWs2812<'d, PIO0, 0, 320>) -> Self {
|
||
|
|
Self {
|
||
|
|
buffer: [RGB8::default(); 320],
|
||
|
|
layout,
|
||
|
|
pio,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn sync(&mut self) {
|
||
|
|
self.pio.write(&self.buffer).await;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'d> DrawTarget for LedMatrix<'d> {
|
||
|
|
type Color = Rgb888;
|
||
|
|
|
||
|
|
type Error = Infallible;
|
||
|
|
|
||
|
|
fn draw_iter<I>(&mut self, pixels: I) -> Result<(), Self::Error>
|
||
|
|
where
|
||
|
|
I: IntoIterator<Item = Pixel<Self::Color>>,
|
||
|
|
{
|
||
|
|
for Pixel(point, color) in pixels {
|
||
|
|
if let Some(x) = self.layout.map(point).and_then(|i| self.buffer.get_mut(i)) {
|
||
|
|
let color = gamma(color);
|
||
|
|
// Map to 0-64 for brightness limiting
|
||
|
|
*x = RGB8::new(
|
||
|
|
color.r().div_ceil(2),
|
||
|
|
color.g().div_ceil(2),
|
||
|
|
color.b().div_ceil(2),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'d> OriginDimensions for LedMatrix<'d> {
|
||
|
|
fn size(&self) -> Size {
|
||
|
|
self.layout.size
|
||
|
|
}
|
||
|
|
}
|