renderbug/src/power.rs
2024-11-29 10:58:17 +01:00

43 lines
1.2 KiB
Rust

use rgb::RGB8;
pub trait AsMilliwatts {
fn as_milliwatts(&self) -> u32;
}
impl AsMilliwatts for RGB8 {
fn as_milliwatts(&self) -> u32 {
const RED_MW : u32 = 16 * 5; //< 16mA @ 5v = 80mW
const GREEN_MW : u32 = 11 * 5; //< 11mA @ 5v = 55mW
const BLUE_MW : u32 = 15 * 5; //< 15mA @ 5v = 75mW
const DARK_MW : u32 = 5; //< 1mA @ 5v = 5mW
let red = (self.r as u32 * RED_MW).wrapping_shr(8);
let green = (self.g as u32 * GREEN_MW).wrapping_shr(8);
let blue = (self.b as u32 * BLUE_MW).wrapping_shr(8);
return red + green + blue + DARK_MW;
}
}
impl<T> AsMilliwatts for [T] where T: AsMilliwatts {
fn as_milliwatts(&self) -> u32 {
self.iter().map(|p| { p.as_milliwatts() }).sum()
}
}
impl<T, const S: usize> AsMilliwatts for [T; S] where T: AsMilliwatts {
fn as_milliwatts(&self) -> u32 {
self.iter().map(|p| { p.as_milliwatts() }).sum()
}
}
pub fn brightness_for_mw(total_mw : u32, target : u8, max_power: u32) -> u8 {
let target32 = target as u32;
let requested_mw = (total_mw * target32) / 256;
if requested_mw > max_power {
return ((target32 * max_power) / requested_mw).try_into().unwrap();
}
return target;
}