renderbug: first implementation of virtual coordinate based rendering

src/embedded_graphics_lib.rs

@@ -0,0 +1,96 @@
+use embedded_graphics::{
+    prelude::*,
+    pixelcolor::Rgb888,
+    primitives::Rectangle
+};
+use ws2812_esp32_rmt_driver::lib_embedded_graphics::{Ws2812DrawTarget, LedPixelShape};
+
+use std::rc::Rc;
+use std::cell::RefCell;
+use std::sync::{Arc, Mutex};
+use running_average::RealTimeRunningAverage;
+
+use crate::power;
+use crate::lib8::*;
+use crate::render::*;
+use crate::time::Periodically;
+use crate::geometry::*;
+
+pub struct EmbeddedDisplay<T>
+where
+T: DrawTarget {
+    surfaces : RefCell<Vec<Surface>>,
+    target: T,
+    total_mw: u32,
+    max_mw: u32,
+    fps: RealTimeRunningAverage<u32>,
+    frame: u32,
+    fps_display: Periodically
+}
+
+impl<T> EmbeddedDisplay<T>
+where
+T: DrawTarget {
+    pub fn new(target: T, max_mw: u32) -> Self {
+        EmbeddedDisplay {
+            surfaces: RefCell::new(Vec::new()),
+            target: target,
+            max_mw: max_mw,
+            total_mw: 0,
+            fps: RealTimeRunningAverage::default(),
+            frame: 0,
+            fps_display: Periodically::new_every_n_seconds(5)
+        }
+    }
+}
+
+impl<T> Surfaces for EmbeddedDisplay<T>
+where
+T: DrawTarget {
+    fn new_surface(&mut self) -> Surface {
+        let surface = Surface::new();
+        self.surfaces.borrow_mut().push(surface.clone());
+        return surface;
+    }
+}
+
+impl<T: LedPixelShape> Display for EmbeddedDisplay<Ws2812DrawTarget<'_, T>> {
+    fn start_frame(&mut self) {
+        self.total_mw = 0;
+        self.frame = self.frame.wrapping_add(1);
+    }
+
+    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();
+        self.fps.insert(1);
+        self.fps_display.run(|| {
+            log::info!("FPS: {} frame={} brightness={} mw={}", self.fps.measurement(), self.frame, brightness, self.total_mw);
+        });
+    }
+
+    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);
+
+        let surfaces = self.surfaces.borrow();
+
+        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 surfaces.iter() {
+                    surface.with_shader(|shader| {
+                        pixel = shader.draw(virtCoords.clone());
+                    })
+                }
+                self.total_mw += power::color_to_mw(&pixel);
+                return Pixel(pos, Rgb888::new(pixel.red, pixel.green, pixel.blue));
+            })
+        ).unwrap();
+    }
+}

src/geometry.rs

@@ -0,0 +1,53 @@
+use std::marker::PhantomData;
+
+pub trait CoordinateSpace {}
+
+pub trait Coordinates<T, S: CoordinateSpace> {
+    fn x(&self) -> T;
+    fn y(&self) -> T;
+    fn new(x: T, y: T) -> Self;
+
+    const MAX: T;
+    const MIN: T;
+}
+
+#[derive(PartialEq, Debug, Copy, Clone)]
+pub struct Virtual {}
+impl CoordinateSpace for Virtual {}
+
+#[derive(PartialEq, Debug, Copy, Clone)]
+pub struct Physical {}
+impl CoordinateSpace for Physical {}
+
+#[derive(PartialEq, Debug, Copy, Clone)]
+pub struct Coord8<S: CoordinateSpace> {
+    x: u8,
+    y: u8,
+    space: PhantomData<S>
+}
+
+pub type VirtualCoordinates = Coord8<Virtual>;
+pub type PhysicalCoordinates = Coord8<Physical>;
+
+impl<S> Coordinates<u8, S> for Coord8<S>
+where
+S: CoordinateSpace {
+    fn new(x: u8, y: u8) -> Self {
+        Self {
+            x: x,
+            y: y,
+            space: PhantomData
+        }
+    }
+
+    fn x(&self) -> u8 {
+        self.x
+    }
+
+    fn y(&self) -> u8 {
+        self.y
+    }
+
+    const MAX: u8 = 255;
+    const MIN: u8 = 255;
+}

src/lib8.rs

@@ -1,4 +1,5 @@
 use palette::convert::FromColorUnclamped;
+use palette::blend::{PreAlpha, Premultiply};
 use palette::encoding::srgb::Srgb;
 use palette::Hsv;
 use embedded_graphics::pixelcolor::RgbColor;
@@ -13,7 +14,7 @@ pub struct RGB8 {
 }
 
 impl RGB8 {
-    const fn new(red : u8, green : u8, blue : u8) -> Self {
+    pub const fn new(red : u8, green : u8, blue : u8) -> Self {
         Self {
             red: red,
             green: green,

src/main.rs

@@ -1,7 +1,7 @@
 #![feature(trait_upcasting)]
 #![allow(arithmetic_overflow)]
 use esp_idf_svc::hal::prelude::Peripherals;
-use ws2812_esp32_rmt_driver::lib_embedded_graphics::{LedPixelStrip, Ws2812DrawTarget};
+use ws2812_esp32_rmt_driver::lib_embedded_graphics::{LedPixelStrip, LedPixelShape, LedPixelMatrix, Ws2812DrawTarget};
 use embedded_graphics::{
     prelude::*,
 };
@@ -14,8 +14,12 @@ mod lib8;
 mod render;
 mod task;
 mod time;
+mod geometry;
+mod embedded_graphics_lib;
 
 use crate::time::Periodically;
+use crate::geometry::{Coordinates, VirtualCoordinates};
+use crate::embedded_graphics_lib::EmbeddedDisplay;
 
 struct IdleTask {
     frame: u8,
@@ -28,8 +32,8 @@ struct IdleShader {
 }
 
 impl render::Shader for IdleShader {
-    fn draw(&self, coords: Point) -> lib8::RGB8 {
-        Hsv::new_srgb(self.frame.wrapping_add(coords.x as u8), 255, 255).into_color_unclamped()
+    fn draw(&self, coords: VirtualCoordinates) -> lib8::RGB8 {
+        Hsv::new_srgb(self.frame.wrapping_add(coords.x()).wrapping_add(coords.y()), 255, 255).into_color_unclamped()
     }
 }
 
@@ -52,6 +56,30 @@ impl task::Task for IdleTask {
             self.surface.set_shader(Box::new(IdleShader { frame: self.frame }));
         })
     }
+
+    fn stop(&mut self) {
+        self.surface.clear_shader();
+    }
+}
+
+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
+        }
+    }
 }
 
 fn main() {
@@ -68,14 +96,15 @@ fn main() {
     let led_pin = peripherals.pins.gpio14;
     let channel = peripherals.rmt.channel0;
 
-    const NUM_PIXELS : usize = 300;
+    const NUM_PIXELS : usize = 255;
     const POWER_VOLTS : u32 = 5;
     const POWER_MA : u32 = 500;
     const MAX_POWER_MW : u32 = POWER_VOLTS * POWER_MA;
 
     log::info!("Setting up display");
-    let target = Ws2812DrawTarget::<LedPixelStrip<NUM_PIXELS>>::new(channel, led_pin).unwrap();
-    let mut display = render::EmbeddedDisplay::new(target, MAX_POWER_MW);
+    let mut target = Ws2812DrawTarget::<PonderjarMatrix>::new(channel, led_pin).unwrap();
+    target.set_brightness(0);
+    let mut display = EmbeddedDisplay::new(target, MAX_POWER_MW);
 
     log::info!("Creating runner");
     let mut runner = task::Scheduler::new(vec![

src/render.rs

@@ -1,21 +1,18 @@
-use embedded_graphics::{
-    prelude::*,
-    pixelcolor::Rgb888
-};
-use ws2812_esp32_rmt_driver::lib_embedded_graphics::{Ws2812DrawTarget, LedPixelShape};
-
 use std::rc::Rc;
 use std::cell::RefCell;
-
-use running_average::RealTimeRunningAverage;
+use std::sync::{Arc, Mutex};
+use palette::blend::{BlendWith, Equations, Parameter, PreAlpha};
 
 use crate::task;
 use crate::lib8::RGB8;
 use crate::power;
 use crate::time::Periodically;
+use crate::geometry::*;
+
+use std::time::Instant;
 
 pub trait Shader: Send {
-    fn draw(&self, coords: Point) -> RGB8;
+    fn draw(&self, surface_coords: VirtualCoordinates) -> RGB8;
 }
 
 pub trait Surfaces {
@@ -41,94 +38,41 @@ T: Display {
     }
 }
 
-struct ShaderSlot {
-    shader: Option<Box<dyn Shader>>
+struct ShaderBinding {
+    shader: Option<Box<dyn Shader>>,
+    opacity: u8,
 }
 
+#[derive(Clone)]
 pub struct Surface {
-    slot: Rc<RefCell<ShaderSlot>>
+    pub binding: Arc<Mutex<ShaderBinding>>
 }
 
 impl Surface {
-    fn new(slot: Rc<RefCell<ShaderSlot>>) -> Self {
+    pub fn new() -> Self {
         Self {
-            slot: slot
+            binding: Arc::new(Mutex::new(ShaderBinding {
+                shader: None,
+                opacity: 255,
+            })),
+        }
+    }
+
+    pub fn with_shader<F: FnOnce(&dyn Shader)>(&self, f: F) {
+        if let Some(ref shader) = self.binding.lock().unwrap().shader {
+            f(shader.as_ref());
         }
     }
 
     pub fn set_shader(&mut self, shader: Box<dyn Shader>) {
-        self.slot.borrow_mut().shader = Some(shader);
-    }
-}
-
-pub struct EmbeddedDisplay<T>
-where
-T: DrawTarget {
-    shaders : RefCell<Vec<Rc<RefCell<ShaderSlot>>>>,
-    target: T,
-    total_mw: u32,
-    max_mw: u32,
-    fps: RealTimeRunningAverage<u32>,
-    frame: u32,
-    fps_display: Periodically
-}
-
-impl<T> EmbeddedDisplay<T>
-where
-T: DrawTarget {
-    pub fn new(target: T, max_mw: u32) -> Self {
-        EmbeddedDisplay {
-            shaders: RefCell::new(Vec::new()),
-            target: target,
-            max_mw: max_mw,
-            total_mw: 0,
-            fps: RealTimeRunningAverage::default(),
-            frame: 0,
-            fps_display: Periodically::new_every_n_seconds(5)
-        }
-    }
-}
-
-impl<T> Surfaces for EmbeddedDisplay<T>
-where
-T: DrawTarget {
-    fn new_surface(&mut self) -> Surface {
-        let slot = Rc::new(RefCell::new(ShaderSlot {
-            shader: None
-        }));
-        let surface = Surface::new(slot.clone());
-        self.shaders.borrow_mut().push(slot);
-        return surface;
-    }
-}
-
-impl<T: LedPixelShape> Display for EmbeddedDisplay<Ws2812DrawTarget<'_, T>> {
-    fn start_frame(&mut self) {
-        self.target.clear(Rgb888::BLACK).unwrap();
-        self.total_mw = 0;
-        self.frame = self.frame.wrapping_add(1);
-    }
-
-    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();
-        self.fps.insert(1);
-        self.fps_display.run(|| {
-            log::info!("FPS: {} frame={} brightness={} mw={}", self.fps.measurement(), self.frame, brightness, self.total_mw);
-        });
-    }
-
-    fn render_frame(&mut self) {
-        for slot in self.shaders.borrow().iter() {
-            if let Some(ref shader) = slot.borrow().shader {
-                for i in 0..T::size().width {
-                    let coords = Point::new(i as i32, 0);
-                    let color = shader.draw(coords);
-                    self.total_mw += power::color_to_mw(&color);
-                    Pixel(coords, Rgb888::new(color.red, color.green, color.blue)).draw(&mut self.target).unwrap();
-                }
-            }
-        }
+        self.binding.lock().unwrap().shader = Some(shader);
+    }
+
+    pub fn clear_shader(&mut self) {
+        self.binding.lock().unwrap().shader = None;
+    }
+
+    pub fn set_opacity(&mut self, opacity: u8) {
+        self.binding.lock().unwrap().opacity = opacity;
     }
 }

src/task.rs

@@ -1,4 +1,5 @@
 use std::fmt;
+use std::time::{Duration, Instant};
 
 pub trait Task {
     fn tick(&mut self) {}
@@ -82,7 +83,7 @@ impl ScheduledState for Stopped  {
 
 struct ScheduledTask {
     state: Option<Box<dyn ScheduledState>>,
-    task: Box<dyn Task>
+    task: Box<dyn Task>,
 }
 
 impl std::fmt::Debug for ScheduledTask {
@@ -98,7 +99,7 @@ impl ScheduledTask {
     fn new(task: Box<dyn Task>) -> Self {
         ScheduledTask {
             state: Some(Box::new(Starting{})),
-            task: task
+            task: task,
         }
     }
 
@@ -122,7 +123,7 @@ impl ScheduledTask {
 }
 
 pub struct Scheduler {
-    tasks: Vec<ScheduledTask>
+    tasks: Vec<ScheduledTask>,
 }
 
 impl Scheduler {