esphome-xiaomi_bslamp2/color_white_light.h

#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