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.
125 lines
3.9 KiB
125 lines
3.9 KiB
#pragma once
|
|
|
|
#include <array>
|
|
#include <stdexcept>
|
|
|
|
#include "common.h"
|
|
#include "gpio_outputs.h"
|
|
|
|
namespace esphome {
|
|
namespace yeelight {
|
|
namespace bs2 {
|
|
|
|
/**
|
|
* The minimum color temperature in mired. Same as supported by
|
|
* the original Yeelight firmware.
|
|
*/
|
|
static const int MIRED_MIN = 153;
|
|
|
|
/**
|
|
* The maximum color temperature in mired. Same as supported by
|
|
* the original Yeelight firmware.
|
|
*/
|
|
static const int MIRED_MAX = 588;
|
|
|
|
struct RGBWLevelsByTemperature {
|
|
float from_temperature;
|
|
float red;
|
|
float green;
|
|
float blue;
|
|
float white;
|
|
};
|
|
|
|
using RGBWLevelsTable = std::array<RGBWLevelsByTemperature, 15>;
|
|
|
|
static const RGBWLevelsTable rgbw_levels_1_ {{
|
|
{ 501.0f, 0.873f, 0.907f, 1.000f, 0.063f },
|
|
{ 455.0f, 0.873f, 0.896f, 1.000f, 0.063f },
|
|
{ 417.0f, 0.873f, 0.891f, 1.000f, 0.068f },
|
|
{ 371.0f, 0.873f, 0.880f, 1.000f, 0.070f },
|
|
{ 334.0f, 0.873f, 0.887f, 1.000f, 0.088f },
|
|
{ 313.0f, 0.882f, 0.904f, 1.000f, 0.128f },
|
|
{ 295.0f, 0.947f, 1.000f, 0.968f, 0.145f },
|
|
{ 251.0f, 0.999f, 1.000f, 1.000f, 0.155f },
|
|
{ 223.0f, 1.000f, 0.899f, 0.921f, 0.130f },
|
|
{ 201.0f, 1.000f, 0.873f, 0.908f, 0.115f },
|
|
{ 182.0f, 1.000f, 0.873f, 0.901f, 0.103f },
|
|
{ 173.0f, 1.000f, 0.873f, 0.904f, 0.094f },
|
|
{ 167.0f, 1.000f, 0.873f, 0.891f, 0.098f },
|
|
{ 154.0f, 1.000f, 0.873f, 0.894f, 0.090f },
|
|
{ 153.0f, 1.000f, 0.873f, 0.892f, 0.088f }
|
|
}};
|
|
|
|
static const RGBWLevelsTable rgbw_levels_100_ {{
|
|
{ 501.0f, 0.000f, 0.344f, 1.000f, 0.068f },
|
|
{ 455.0f, 0.000f, 0.237f, 1.000f, 0.093f },
|
|
{ 417.0f, 0.000f, 0.186f, 1.000f, 0.120f },
|
|
{ 371.0f, 0.000f, 0.149f, 1.000f, 0.167f },
|
|
{ 334.0f, 0.000f, 0.135f, 1.000f, 0.325f },
|
|
{ 313.0f, 0.097f, 0.314f, 1.000f, 0.740f },
|
|
{ 295.0f, 0.745f, 1.000f, 0.953f, 0.905f },
|
|
{ 251.0f, 1.000f, 1.000f, 1.000f, 1.000f },
|
|
{ 223.0f, 1.000f, 0.267f, 0.485f, 0.765f },
|
|
{ 201.0f, 1.000f, 0.000f, 0.355f, 0.609f },
|
|
{ 182.0f, 1.000f, 0.000f, 0.282f, 0.489f },
|
|
{ 173.0f, 1.000f, 0.000f, 0.313f, 0.392f },
|
|
{ 167.0f, 1.000f, 0.000f, 0.180f, 0.422f },
|
|
{ 154.0f, 1.000f, 0.000f, 0.218f, 0.368f },
|
|
{ 153.0f, 1.000f, 0.000f, 0.187f, 0.335f }
|
|
}};
|
|
|
|
/**
|
|
* This class can handle the GPIO outputs for the white light mode,
|
|
* based on color temperature + brightness.
|
|
*/
|
|
class ColorWhiteLight : public GPIOOutputs {
|
|
protected:
|
|
bool set_light_color_values(light::LightColorValues v) {
|
|
if (v.get_white() == 0.0f) {
|
|
return false;
|
|
}
|
|
|
|
auto temperature = clamp_temperature_(v.get_color_temperature());
|
|
auto brightness = clamp_brightness_(v.get_brightness());
|
|
|
|
auto levels_1 = lookup_in_table_(rgbw_levels_1_, temperature);
|
|
auto levels_100 = lookup_in_table_(rgbw_levels_100_, temperature);
|
|
|
|
red = esphome::lerp(brightness, levels_1.red, levels_100.red);
|
|
green = esphome::lerp(brightness, levels_1.green, levels_100.green);
|
|
blue = esphome::lerp(brightness, levels_1.blue, levels_100.blue);
|
|
white = esphome::lerp(brightness, levels_1.white, levels_100.white);
|
|
|
|
return true;
|
|
}
|
|
|
|
float clamp_temperature_(float temperature)
|
|
{
|
|
if (temperature > MIRED_MAX)
|
|
temperature = MIRED_MAX;
|
|
else if (temperature < MIRED_MIN)
|
|
temperature = MIRED_MIN;
|
|
return temperature;
|
|
}
|
|
|
|
float clamp_brightness_(float brightness)
|
|
{
|
|
if (brightness < 0.01f)
|
|
brightness = 0.01f;
|
|
else if (brightness > 1.00f)
|
|
brightness = 1.00f;
|
|
return brightness;
|
|
}
|
|
|
|
RGBWLevelsByTemperature lookup_in_table_(RGBWLevelsTable table, float temperature)
|
|
{
|
|
for (RGBWLevelsByTemperature& item : table)
|
|
if (temperature >= item.from_temperature)
|
|
return item;
|
|
throw std::invalid_argument("received too low temperature");
|
|
}
|
|
};
|
|
|
|
} // namespace bs2
|
|
} // namespace yeelight
|
|
} // namespace esphome
|