tangent128/holiday_lights
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Implement white-balancing

Browse source
This commit is contained in:
Tangent 128 2019-11-06 19:33:55 -05:00
parent 7455a35053
commit bc018751c8
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
house/src/lib.rs
View file

@ -9,12 +9,18 @@ use lights::rgb::Rgb;
const PORCH_BACK_LEN: usize = 150; const PORCH_BACK_LEN: usize = 150;
const PORCH_FRONT_LEN: usize = 150; const PORCH_FRONT_LEN: usize = 150;
const PORCH_WHITE: Rgb = Rgb(255, 208, 160);
pub struct Harrogate<PB, PF> { pub struct Harrogate<PB, PF> {
pub porch_back: PB, pub porch_back: PB,
pub porch_front: PF, pub porch_front: PF,
} }
fn white_balance(HardwareRgb(r, g, b): HardwareRgb, white: Rgb) -> HardwareRgb {
let Rgb(r, g, b) = Rgb(r, g, b) * white;
HardwareRgb(r, g, b)
}
impl<PB, PF> Harrogate<PB, PF> impl<PB, PF> Harrogate<PB, PF>
where where
PB: IntoIterator<Item = Rgb>, PB: IntoIterator<Item = Rgb>,
@ -24,10 +30,10 @@ where
let mut buffer = [HardwareRgb(255,0,255); PORCH_BACK_LEN + PORCH_FRONT_LEN]; let mut buffer = [HardwareRgb(255,0,255); PORCH_BACK_LEN + PORCH_FRONT_LEN];
for (i, pixel) in self.porch_back.into_iter().take(PORCH_BACK_LEN).enumerate() { for (i, pixel) in self.porch_back.into_iter().take(PORCH_BACK_LEN).enumerate() {
buffer[PORCH_BACK_LEN - i] = correct(&pixel); buffer[PORCH_BACK_LEN - i] = white_balance(correct(&pixel), PORCH_WHITE);
} }
for (i, pixel) in self.porch_front.into_iter().take(PORCH_FRONT_LEN).enumerate() { for (i, pixel) in self.porch_front.into_iter().take(PORCH_FRONT_LEN).enumerate() {
buffer[PORCH_BACK_LEN + i] = correct(&pixel); buffer[PORCH_BACK_LEN + i] = white_balance(correct(&pixel), PORCH_WHITE);
} }
buffer.iter().render_to(lights); buffer.iter().render_to(lights);