You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.3 KiB
61 lines
2.3 KiB
#pragma once
|
|
|
|
#include <array>
|
|
#include <stdexcept>
|
|
|
|
namespace esphome {
|
|
namespace rgbww {
|
|
namespace yeelight_bs2 {
|
|
|
|
class RGBLight
|
|
{
|
|
public:
|
|
float red = 0;
|
|
float green = 0;
|
|
float blue = 0;
|
|
float white = 0;
|
|
|
|
void set_color(float red, float green, float blue, float brightness, float state)
|
|
{
|
|
// Overall, the RGB colors are very usable when simply scaling the
|
|
// RGB channels with the brightness, but around the white point,
|
|
// the color is a bit on the red side of the spectrum. The following
|
|
// scaling was created to fix that.
|
|
// These functions were created, based on actual measurements while
|
|
// using the original firmware.
|
|
auto b = brightness * 100.0f;
|
|
auto red_w = 1.00f - (-0.0000121426 * b * b - 0.147576 * b + 93.2335) / 100.0f;
|
|
auto green_w = 1.00f - (-0.0000242425 * b * b - 0.340449 * b + 88.4423) / 100.0f;
|
|
auto blue_w = 1.00f - (-0.0000085869 * b * b - 0.109649 * b + 94.2026) / 100.0f;
|
|
|
|
// For colors that are not around the white point, we can scale the
|
|
// RGB channels with the requested brightness, resulting in a very
|
|
// usable color. Not 100% the same as the original firmware, but
|
|
// sometimes even better IMO.
|
|
auto red_c = red * brightness;
|
|
auto green_c = green * brightness;
|
|
auto blue_c = blue * brightness;
|
|
|
|
// The actual RGB values are a weighed mix of the above two.
|
|
// The closer to the white point, the more the white point
|
|
// value applies.
|
|
auto level_red = (red_w * ((green+blue)/2)) + (red_c * (1-(green+blue)/2));
|
|
auto level_green = (green_w * ((red+blue)/2)) + (green_c * (1-(red+blue)/2));
|
|
auto level_blue = (blue_w * ((red+green)/2)) + (blue_c * (1-(red+green)/2));
|
|
if (red == 1 && green == 1 && blue == 1) {
|
|
level_red = red_w;
|
|
level_green = green_w;
|
|
level_blue = blue_w;
|
|
}
|
|
|
|
// Invert the signal. The LEDs in the lamp's circuit are brighter
|
|
// when the pwm levels on the GPIO pins are lower.
|
|
this->red = 1.0f - level_red;
|
|
this->green = 1.0f - level_green;
|
|
this->blue = 1.0f - level_blue;
|
|
}
|
|
};
|
|
|
|
} // namespace yeelight_bs2
|
|
} // namespace rgbww
|
|
} // namespace esphome
|