#pragma once
|
|
|
|
#include "esphome.h"
|
|
|
|
#define CONSTANT_BRIGHTNESS true
|
|
|
|
// The lamp circuitry does not support having RGB and white
|
|
// channels active at the same time. Therefore, color interlock
|
|
// must be enabled.
|
|
#define COLOR_INTERLOCK true
|
|
|
|
// Same range as supported by the original Yeelight firmware.
|
|
#define HOME_ASSISTANT_MIRED_MIN 153
|
|
#define HOME_ASSISTANT_MIRED_MAX 588
|
|
|
|
namespace esphome {
|
|
namespace rgbww {
|
|
|
|
class YeelightBedsideLampV2LightOutput : public Component, public LightOutput
|
|
{
|
|
public:
|
|
YeelightBedsideLampV2LightOutput(
|
|
FloatOutput *r, FloatOutput *g, FloatOutput *b, FloatOutput *w,
|
|
esphome::gpio::GPIOBinaryOutput *m1, esphome::gpio::GPIOBinaryOutput *m2) :
|
|
red_(r), green_(g), blue_(b), white_(w), master1_(m1), master2_(m2) {}
|
|
|
|
LightTraits get_traits() override
|
|
{
|
|
auto traits = LightTraits();
|
|
traits.set_supports_rgb(true);
|
|
traits.set_supports_color_temperature(true);
|
|
traits.set_supports_brightness(true);
|
|
traits.set_supports_rgb_white_value(false);
|
|
traits.set_supports_color_interlock(COLOR_INTERLOCK);
|
|
traits.set_min_mireds(HOME_ASSISTANT_MIRED_MIN);
|
|
traits.set_max_mireds(HOME_ASSISTANT_MIRED_MAX);
|
|
return traits;
|
|
}
|
|
|
|
void write_state(LightState *state) override
|
|
{
|
|
auto values = state->current_values;
|
|
|
|
ESP_LOGD("custom", "B = State %f, RGB %f %f %f, BRI %f, TEMP %f",
|
|
values.get_state(),
|
|
values.get_red(), values.get_green(), values.get_blue(),
|
|
values.get_brightness(), values.get_color_temperature());
|
|
|
|
// Power down the light when its state is 'off'.
|
|
if (values.get_state() == 0) {
|
|
this->turn_off_();
|
|
return;
|
|
}
|
|
|
|
// Leave it to the default tooling to figure out the basics.
|
|
// Because of the color interlocking, there are two possible outcomes:
|
|
// - red, green, blue zero -> the light is in color temperature mode
|
|
// - cwhite, wwhite zero -> the light is in RGB mode
|
|
float red, green, blue, cwhite, wwhite;
|
|
state->current_values_as_rgbww(
|
|
&red, &green, &blue, &cwhite, &wwhite,
|
|
CONSTANT_BRIGHTNESS, COLOR_INTERLOCK);
|
|
|
|
if (cwhite > 0 || wwhite > 0) {
|
|
this->turn_on_in_color_temperature_mode_(
|
|
values.get_color_temperature(), values.get_brightness());
|
|
} else {
|
|
this->turn_on_in_rgb_mode_(
|
|
values.get_red(), values.get_green(), values.get_blue(), values.get_brightness());
|
|
}
|
|
}
|
|
|
|
private:
|
|
FloatOutput *red_;
|
|
FloatOutput *green_;
|
|
FloatOutput *blue_;
|
|
FloatOutput *white_;
|
|
esphome::gpio::GPIOBinaryOutput *master1_;
|
|
esphome::gpio::GPIOBinaryOutput *master2_;
|
|
|
|
void turn_off_()
|
|
{
|
|
master1_->turn_off();
|
|
red_->set_level(0);
|
|
green_->set_level(0);
|
|
blue_->set_level(0);
|
|
white_->set_level(0);
|
|
}
|
|
|
|
void turn_on_in_rgb_mode_(float red, float green, float blue, float brightness)
|
|
{
|
|
ESP_LOGD("custom", "Activate RGB %f, %f, %f, BRIGHTNESS %f", red, green, blue, brightness);
|
|
|
|
// This tunes the power for the red channel a bit, to bring
|
|
// the red=1, green=1, blue=1 color more towards white.
|
|
// (on my lamps, there is a faint hint of red in the color)
|
|
red = red * 0.95f;
|
|
|
|
// Compensate for brightness.
|
|
red = red * brightness;
|
|
green = green * brightness;
|
|
blue = blue * brightness;
|
|
|
|
// Inverse the signal. The LEDs in the lamp's circuit are brighter
|
|
// when the voltages on the GPIO pins are lower.
|
|
red = 1.0f - red;
|
|
green = 1.0f - green;
|
|
blue = 1.0f - blue;
|
|
|
|
float white = 0.0;
|
|
|
|
ESP_LOGD("rgb_mode", "LED state : RGBW %f, %f, %f, %f", red, green, blue, white);
|
|
|
|
// Drive the LEDs.
|
|
red_->set_level(red);
|
|
green_->set_level(green);
|
|
blue_->set_level(blue);
|
|
white_->set_level(white);
|
|
master1_->turn_on();
|
|
}
|
|
|
|
void turn_on_in_color_temperature_mode_(float temperature, float brightness)
|
|
{
|
|
ESP_LOGD("temperature_mode", "Activate TEMPERATURE %f, BRIGHTNESS %f", temperature, brightness);
|
|
|
|
float volt_scaler = 3.23f;
|
|
|
|
if (temperature <= HOME_ASSISTANT_MIRED_MAX && temperature >= 371)
|
|
{
|
|
float start = 371;
|
|
float end = 588;
|
|
float band = end - start;
|
|
|
|
float red_volt = 2.86f * (1.0f - brightness);
|
|
float red = red_volt / volt_scaler;
|
|
|
|
float green_1 = 2.90f + (temperature - start) * (2.97f - 2.90f) / band;
|
|
float green_100 = 0.45f + (temperature - start) * (1.13f - 0.45f) / band;
|
|
float green_volt = green_1 + brightness * (green_100 - green_1);
|
|
float green = green_volt / volt_scaler;
|
|
|
|
float blue = 1.0f;
|
|
|
|
float white_1 = 0.28f - (temperature - start) * (0.28f - 0.19f) / band;
|
|
float white_100 = 1.07f - (temperature - start) * (1.07f - 0.22f) / band;
|
|
float white_volt = white_1 + brightness * (white_100 - white_1);
|
|
float white = white_volt / volt_scaler;
|
|
|
|
ESP_LOGD("temperature_mode", "LED state : RGBW %f, %f, %f, %f", red, green, blue, white);
|
|
|
|
red_->set_level(red);
|
|
green_->set_level(green);
|
|
blue_->set_level(blue);
|
|
white_->set_level(white);
|
|
master2_->turn_on();
|
|
master1_->turn_on();
|
|
return;
|
|
}
|
|
else if (temperature >= 334)
|
|
{
|
|
float red_volt = (1.0f - brightness) * 2.86f;
|
|
float red = red_volt / volt_scaler;
|
|
|
|
float green_volt = 2.9f - brightness * (2.9f - 0.45f);
|
|
float green = green_volt / volt_scaler;
|
|
|
|
float blue = 1.0f;
|
|
|
|
float white_volt = 0.28f + brightness * (1.07f - 0.28f);
|
|
float white = white_volt / volt_scaler;
|
|
|
|
ESP_LOGD("temperature_mode", "LED state : RGBW %f, %f, %f, %f", red, green, blue, white);
|
|
|
|
red_->set_level(red);
|
|
green_->set_level(green);
|
|
blue_->set_level(blue);
|
|
white_->set_level(white);
|
|
master2_->turn_on();
|
|
master1_->turn_on();
|
|
return;
|
|
} else if (temperature >= 313) {
|
|
} else if (temperature >= 251) {
|
|
} else if (temperature >= 223) {
|
|
} else if (temperature >= HOME_ASSISTANT_MIRED_MIN) {
|
|
}
|
|
|
|
red_->set_level(0.5);
|
|
green_->set_level(0.5);
|
|
blue_->set_level(0.5);
|
|
master2_->turn_off();
|
|
master1_->turn_off();
|
|
}
|
|
};
|
|
|
|
} // namespace rgbww
|
|
} // namespace esphome
|