146 lines
4.1 KiB
Rust
146 lines
4.1 KiB
Rust
use embedded_graphics::{
|
|
prelude::*,
|
|
pixelcolor::Rgb888,
|
|
primitives::Rectangle
|
|
};
|
|
use rgb::RGB8;
|
|
use ws2812_esp32_rmt_driver::lib_embedded_graphics::{Ws2812DrawTarget, LedPixelShape};
|
|
|
|
use esp_idf_svc::hal::prelude::Peripherals;
|
|
|
|
use std::io;
|
|
use std::fmt::{Formatter, Debug};
|
|
|
|
use crate::power;
|
|
use crate::power::AsMilliwatts;
|
|
use crate::lib8::*;
|
|
use crate::render::*;
|
|
use crate::geometry::*;
|
|
use crate::platform::DisplayInit;
|
|
|
|
pub struct EmbeddedDisplay<T, S>
|
|
where
|
|
T: DrawTarget,
|
|
S: Surface {
|
|
surfaces : SurfacePool<S>,
|
|
target: T,
|
|
total_mw: u32,
|
|
max_mw: u32,
|
|
}
|
|
|
|
|
|
impl<T: DrawTarget, S: Surface> Debug for EmbeddedDisplay<T, S> {
|
|
fn fmt(&self, f: &mut Formatter) -> Result<(), std::fmt::Error> {
|
|
f.debug_struct("EmbeddedDisplay")
|
|
.field("total_mw", &self.total_mw)
|
|
.field("surfaces", &self.surfaces)
|
|
.finish()
|
|
}
|
|
}
|
|
|
|
trait TargetInit<T: DrawTarget> {
|
|
fn setup_target(mut target: T) -> T { target }
|
|
}
|
|
|
|
impl<T, S> EmbeddedDisplay<T, S>
|
|
where
|
|
Self: TargetInit<T>,
|
|
T: DrawTarget,
|
|
S: Surface {
|
|
pub fn new(target: T, max_mw: u32) -> Self {
|
|
EmbeddedDisplay {
|
|
surfaces: SurfacePool::new(),
|
|
target: Self::setup_target(target),
|
|
max_mw: max_mw,
|
|
total_mw: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<T: LedPixelShape, S: Surface> TargetInit<Ws2812DrawTarget<'_, T>> for EmbeddedDisplay<Ws2812DrawTarget<'_, T>, S> {
|
|
fn setup_target(mut target: Ws2812DrawTarget<'_, T>) -> Ws2812DrawTarget<'_, T> {
|
|
target.set_brightness(0);
|
|
target
|
|
}
|
|
}
|
|
|
|
impl<T, S> Surfaces<S> for EmbeddedDisplay<T, S>
|
|
where
|
|
T: DrawTarget,
|
|
S: Surface {
|
|
fn new_surface(&mut self) -> Result<S, io::Error> {
|
|
self.surfaces.new_surface()
|
|
}
|
|
}
|
|
|
|
impl<T: LedPixelShape, S: Surface> Display<S> for EmbeddedDisplay<Ws2812DrawTarget<'_, T>, S> {
|
|
|
|
fn start_frame(&mut self) {
|
|
self.total_mw = 0;
|
|
}
|
|
|
|
fn end_frame(&mut self) {
|
|
let brightness = power::brightness_for_mw(self.total_mw, 255, self.max_mw);
|
|
self.target.set_brightness(brightness);
|
|
self.target.flush().unwrap();
|
|
}
|
|
|
|
fn render_frame(&mut self) {
|
|
let size = T::size();
|
|
let xStride: u8 = 255 / (size.width as u8);
|
|
let yStride: u8 = 255 / (size.height as u8);
|
|
let area = Rectangle::new(Point::new(0, 0), size);
|
|
|
|
self.target.draw_iter(
|
|
area.points()
|
|
.map(|pos| {
|
|
let virtCoords = VirtualCoordinates::new(pos.x as u8 * xStride, pos.y as u8 * yStride);
|
|
let mut pixel = RGB8::new(0, 0, 0);
|
|
for surface in self.surfaces.iter() {
|
|
surface.with_shader(|shader| {
|
|
pixel = pixel.saturating_add(shader.draw(virtCoords.clone()));
|
|
})
|
|
}
|
|
self.total_mw += pixel.as_milliwatts();
|
|
return Pixel(pos, Rgb888::new(pixel.r, pixel.g, pixel.b));
|
|
})
|
|
).unwrap();
|
|
}
|
|
}
|
|
|
|
impl<Shape: LedPixelShape> DisplayInit for Ws2812DrawTarget<'_, Shape> {
|
|
fn new_display<S: Surface>() -> impl Display<S> {
|
|
let peripherals = Peripherals::take().unwrap();
|
|
let led_pin = peripherals.pins.gpio14;
|
|
let channel = peripherals.rmt.channel0;
|
|
|
|
const POWER_VOLTS : u32 = 5;
|
|
const POWER_MA : u32 = 500;
|
|
const MAX_POWER_MW : u32 = POWER_VOLTS * POWER_MA;
|
|
|
|
let target = Self::new(channel, led_pin).unwrap();
|
|
return EmbeddedDisplay::<Self, S>::new(target, MAX_POWER_MW);
|
|
}
|
|
}
|
|
|
|
pub struct PonderjarMatrix {}
|
|
impl LedPixelShape for PonderjarMatrix {
|
|
fn size() -> Size {
|
|
Size::new(17, 17)
|
|
}
|
|
|
|
fn pixel_index(point: Point) -> Option<usize> {
|
|
if (0..Self::size().width as i32).contains(&point.x) && (0..Self::size().height as i32).contains(&point.y) {
|
|
if point.y % 2 == 0 {
|
|
Some((point.y as u32 * Self::size().width as u32 + point.x as u32).try_into().unwrap())
|
|
} else {
|
|
Some((point.y as u32 * Self::size().width as u32 - point.x as u32).try_into().unwrap())
|
|
}
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
pub type PonderjarTarget<'a> = Ws2812DrawTarget<'a, PonderjarMatrix>;
|