mirror of
https://github.com/esphome/esphome.git
synced 2025-02-20 20:08:20 +00:00
104 lines
3.9 KiB
C++
104 lines
3.9 KiB
C++
#include "addressable_light.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
namespace esphome {
|
|
namespace light {
|
|
|
|
static const char *const TAG = "light.addressable";
|
|
|
|
void AddressableLight::call_setup() {
|
|
this->setup();
|
|
|
|
#ifdef ESPHOME_LOG_HAS_VERY_VERBOSE
|
|
this->set_interval(5000, [this]() {
|
|
const char *name = this->state_parent_ == nullptr ? "" : this->state_parent_->get_name().c_str();
|
|
ESP_LOGVV(TAG, "Addressable Light '%s' (effect_active=%s next_show=%s)", name, YESNO(this->effect_active_),
|
|
YESNO(this->next_show_));
|
|
for (int i = 0; i < this->size(); i++) {
|
|
auto color = this->get(i);
|
|
ESP_LOGVV(TAG, " [%2d] Color: R=%3u G=%3u B=%3u W=%3u", i, color.get_red_raw(), color.get_green_raw(),
|
|
color.get_blue_raw(), color.get_white_raw());
|
|
}
|
|
ESP_LOGVV(TAG, "");
|
|
});
|
|
#endif
|
|
}
|
|
|
|
Color esp_color_from_light_color_values(LightColorValues val) {
|
|
auto r = to_uint8_scale(val.get_color_brightness() * val.get_red());
|
|
auto g = to_uint8_scale(val.get_color_brightness() * val.get_green());
|
|
auto b = to_uint8_scale(val.get_color_brightness() * val.get_blue());
|
|
auto w = to_uint8_scale(val.get_white());
|
|
return Color(r, g, b, w);
|
|
}
|
|
|
|
void AddressableLight::write_state(LightState *state) {
|
|
auto val = state->current_values;
|
|
auto max_brightness = to_uint8_scale(val.get_brightness() * val.get_state());
|
|
this->correction_.set_local_brightness(max_brightness);
|
|
|
|
this->last_transition_progress_ = 0.0f;
|
|
this->accumulated_alpha_ = 0.0f;
|
|
|
|
if (this->is_effect_active())
|
|
return;
|
|
|
|
// don't use LightState helper, gamma correction+brightness is handled by ESPColorView
|
|
|
|
if (state->transformer_ == nullptr || !state->transformer_->is_transition()) {
|
|
// no transformer active or non-transition one
|
|
this->all() = esp_color_from_light_color_values(val);
|
|
} else {
|
|
// transition transformer active, activate specialized transition for addressable effects
|
|
// instead of using a unified transition for all LEDs, we use the current state each LED as the
|
|
// start. Warning: ugly
|
|
|
|
// We can't use a direct lerp smoothing here though - that would require creating a copy of the original
|
|
// state of each LED at the start of the transition
|
|
// Instead, we "fake" the look of the LERP by using an exponential average over time and using
|
|
// dynamically-calculated alpha values to match the look of the
|
|
|
|
float new_progress = state->transformer_->get_progress();
|
|
float prev_smoothed = LightTransitionTransformer::smoothed_progress(last_transition_progress_);
|
|
float new_smoothed = LightTransitionTransformer::smoothed_progress(new_progress);
|
|
this->last_transition_progress_ = new_progress;
|
|
|
|
auto end_values = state->transformer_->get_end_values();
|
|
Color target_color = esp_color_from_light_color_values(end_values);
|
|
|
|
// our transition will handle brightness, disable brightness in correction.
|
|
this->correction_.set_local_brightness(255);
|
|
target_color *= to_uint8_scale(end_values.get_brightness() * end_values.get_state());
|
|
|
|
float denom = (1.0f - new_smoothed);
|
|
float alpha = denom == 0.0f ? 0.0f : (new_smoothed - prev_smoothed) / denom;
|
|
|
|
// We need to use a low-resolution alpha here which makes the transition set in only after ~half of the length
|
|
// We solve this by accumulating the fractional part of the alpha over time.
|
|
float alpha255 = alpha * 255.0f;
|
|
float alpha255int = floorf(alpha255);
|
|
float alpha255remainder = alpha255 - alpha255int;
|
|
|
|
this->accumulated_alpha_ += alpha255remainder;
|
|
float alpha_add = floorf(this->accumulated_alpha_);
|
|
this->accumulated_alpha_ -= alpha_add;
|
|
|
|
alpha255 += alpha_add;
|
|
alpha255 = clamp(alpha255, 0.0f, 255.0f);
|
|
auto alpha8 = static_cast<uint8_t>(alpha255);
|
|
|
|
if (alpha8 != 0) {
|
|
uint8_t inv_alpha8 = 255 - alpha8;
|
|
Color add = target_color * alpha8;
|
|
|
|
for (auto led : *this)
|
|
led = add + led.get() * inv_alpha8;
|
|
}
|
|
}
|
|
|
|
this->schedule_show();
|
|
}
|
|
|
|
} // namespace light
|
|
} // namespace esphome
|