103 lines
3.3 KiB
C++
103 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include "../common.h"
|
|
#include "../light_hal.h"
|
|
#include "color_handler_chain.h"
|
|
#include "light_transformer.h"
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/components/ledc/ledc_output.h"
|
|
|
|
namespace esphome {
|
|
namespace xiaomi {
|
|
namespace bslamp2 {
|
|
|
|
/**
|
|
* A LightOutput class for the Xiaomi Mijia Bedside Lamp 2.
|
|
*
|
|
* The function of this class is to translate a required light state
|
|
* into actual physicial GPIO output signals to drive the device's LED
|
|
* circuitry. It forms the glue between the physical device and the
|
|
* logical light color input.
|
|
*/
|
|
class XiaomiBslamp2LightOutput : public Component, public light::LightOutput {
|
|
public:
|
|
void set_parent(LightHAL *light) { light_ = light; }
|
|
|
|
void set_night_light_color_temperature_calibration(float red, float green, float blue) {
|
|
night_light_calibration_ = {red, green, blue};
|
|
color_handler_chain.set_night_light_color_temperature_calibration(night_light_calibration_);
|
|
apply_current_state();
|
|
}
|
|
|
|
/**
|
|
* Returns a LightTraits object, which is used to explain to the outside
|
|
* world (e.g. Home Assistant) what features are supported by this device.
|
|
*/
|
|
light::LightTraits get_traits() override {
|
|
auto traits = light::LightTraits();
|
|
traits.set_supported_color_modes({light::ColorMode::RGB, light::ColorMode::COLOR_TEMPERATURE});
|
|
traits.set_min_mireds(MIRED_MIN);
|
|
traits.set_max_mireds(MIRED_MAX);
|
|
return traits;
|
|
}
|
|
|
|
std::unique_ptr<light::LightTransformer> create_default_transition() override {
|
|
return make_unique<XiaomiBslamp2LightTransitionTransformer>(
|
|
light_,
|
|
&light_mode_callback_,
|
|
&state_callback_,
|
|
night_light_calibration_);
|
|
}
|
|
|
|
void add_on_light_mode_callback(std::function<void(std::string)> &&callback) {
|
|
light_mode_callback_.add(std::move(callback));
|
|
}
|
|
|
|
void add_on_state_callback(std::function<void(light::LightColorValues)> &&callback) {
|
|
state_callback_.add(std::move(callback));
|
|
}
|
|
|
|
/**
|
|
* Applies a requested light state to the physicial GPIO outputs.
|
|
*/
|
|
void write_state(light::LightState *state) {
|
|
state_ = state;
|
|
auto values = state->current_values;
|
|
|
|
color_handler_chain.set_light_color_values(values);
|
|
light_mode_callback_.call(color_handler_chain.light_mode);
|
|
state_callback_.call(values);
|
|
|
|
// Note: one might think that it is more logical to turn on the LED
|
|
// circuitry master switch after setting the individual channels,
|
|
// but this is the order that was used by the original firmware. I
|
|
// tried to stay as close as possible to the original behavior, so
|
|
// that's why these GPIOs are turned on at this point.
|
|
if (values.get_state() != 0)
|
|
light_->turn_on();
|
|
|
|
// Apply the GPIO output levels as defined by the color handler.
|
|
light_->set_state(&color_handler_chain);
|
|
|
|
if (values.get_state() == 0)
|
|
light_->turn_off();
|
|
}
|
|
|
|
void apply_current_state() {
|
|
if (state_ != nullptr)
|
|
write_state(state_);
|
|
}
|
|
|
|
protected:
|
|
LightHAL *light_;
|
|
light::LightState *state_{nullptr};
|
|
ColorHandlerChain color_handler_chain;
|
|
CallbackManager<void(std::string)> light_mode_callback_{};
|
|
CallbackManager<void(light::LightColorValues)> state_callback_{};
|
|
NightLightCalibration night_light_calibration_{DEFAULT_NIGHT_LIGHT_CALIBRATION};
|
|
};
|
|
|
|
} // namespace bslamp2
|
|
} // namespace xiaomi
|
|
} // namespace esphome
|