Merge pull request #4 from smart-leds-rs/generify

Make SmartLedsWrite more generic

Author: david-sawatzke

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "smart-leds-trait"
-version = "0.1.1"
+version = "0.2.0"
 authors = ["David Sawatzke <[email protected]>"]
 edition = "2018"
 categories = [
@@ -13,6 +13,5 @@ license = "MIT OR Apache-2.0"
 readme = "README.md"
 repository = "https://github.com/smart-leds-rs/smart-leds-trait"
 
-
 [dependencies]
 rgb = "0.8"

src/lib.rs

@@ -1,10 +1,17 @@
 #![no_std]
 
-pub type Color = rgb::RGB8;
+pub use {rgb::RGB, rgb::RGB16, rgb::RGB8, rgb::RGBA};
+
+pub struct White<C>(pub C);
+
+pub type RGBW<ComponentType, WhiteComponentType = ComponentType> =
+    RGBA<ComponentType, White<WhiteComponentType>>;
 
 pub trait SmartLedsWrite {
     type Error;
-    fn write<T>(&mut self, iterator: T) -> Result<(), Self::Error>
+    type Color;
+    fn write<T, I>(&mut self, iterator: T) -> Result<(), Self::Error>
     where
-        T: Iterator<Item = Color>;
+        T: Iterator<Item = I>,
+        I: Into<Self::Color>;
 }