From dca643e758e88d612a9ceb063a4595ac66b9de91 Mon Sep 17 00:00:00 2001 From: Maurice Makaay Date: Fri, 5 Mar 2021 12:05:27 +0100 Subject: [PATCH] Initial import. --- README.md | 1 + yeelight_bs2.h | 196 +++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 197 insertions(+) create mode 100644 README.md create mode 100644 yeelight_bs2.h diff --git a/README.md b/README.md new file mode 100644 index 0000000..a64cd6f --- /dev/null +++ b/README.md @@ -0,0 +1 @@ +# esphome-yeelight_bs2 diff --git a/yeelight_bs2.h b/yeelight_bs2.h new file mode 100644 index 0000000..1931e1e --- /dev/null +++ b/yeelight_bs2.h @@ -0,0 +1,196 @@ +#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