mirror of
https://github.com/esphome/esphome.git
synced 2024-12-02 11:44:13 +01:00
116 lines
4.2 KiB
C++
116 lines
4.2 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/helpers.h"
|
|
#include "light_color_values.h"
|
|
#include "light_state.h"
|
|
#include "light_transformer.h"
|
|
|
|
namespace esphome {
|
|
namespace light {
|
|
|
|
class LightTransitionTransformer : public LightTransformer {
|
|
public:
|
|
void start() override {
|
|
// When turning light on from off state, use colors from target state.
|
|
if (!this->start_values_.is_on() && this->target_values_.is_on()) {
|
|
this->start_values_ = LightColorValues(this->target_values_);
|
|
this->start_values_.set_brightness(0.0f);
|
|
}
|
|
|
|
// When changing color mode, go through off state, as color modes are orthogonal and there can't be two active.
|
|
if (this->start_values_.get_color_mode() != this->target_values_.get_color_mode()) {
|
|
this->changing_color_mode_ = true;
|
|
this->intermediate_values_ = this->start_values_;
|
|
this->intermediate_values_.set_state(false);
|
|
}
|
|
}
|
|
|
|
optional<LightColorValues> apply() override {
|
|
float p = this->get_progress_();
|
|
|
|
// Halfway through, when intermediate state (off) is reached, flip it to the target, but remain off.
|
|
if (this->changing_color_mode_ && p > 0.5f &&
|
|
this->intermediate_values_.get_color_mode() != this->target_values_.get_color_mode()) {
|
|
this->intermediate_values_ = this->target_values_;
|
|
this->intermediate_values_.set_state(false);
|
|
}
|
|
|
|
LightColorValues &start = this->changing_color_mode_ && p > 0.5f ? this->intermediate_values_ : this->start_values_;
|
|
LightColorValues &end = this->changing_color_mode_ && p < 0.5f ? this->intermediate_values_ : this->target_values_;
|
|
if (this->changing_color_mode_)
|
|
p = p < 0.5f ? p * 2 : (p - 0.5) * 2;
|
|
|
|
float v = LightTransitionTransformer::smoothed_progress(p);
|
|
return LightColorValues::lerp(start, end, v);
|
|
}
|
|
|
|
protected:
|
|
// This looks crazy, but it reduces to 6x^5 - 15x^4 + 10x^3 which is just a smooth sigmoid-like
|
|
// transition from 0 to 1 on x = [0, 1]
|
|
static float smoothed_progress(float x) { return x * x * x * (x * (x * 6.0f - 15.0f) + 10.0f); }
|
|
|
|
bool changing_color_mode_{false};
|
|
LightColorValues intermediate_values_{};
|
|
};
|
|
|
|
class LightFlashTransformer : public LightTransformer {
|
|
public:
|
|
LightFlashTransformer(LightState &state) : state_(state) {}
|
|
|
|
void start() override {
|
|
this->transition_length_ = this->state_.get_flash_transition_length();
|
|
if (this->transition_length_ * 2 > this->length_)
|
|
this->transition_length_ = this->length_ / 2;
|
|
|
|
// do not create transition if length is 0
|
|
if (this->transition_length_ == 0)
|
|
return;
|
|
|
|
// first transition to original target
|
|
this->transformer_ = this->state_.get_output()->create_default_transition();
|
|
this->transformer_->setup(this->state_.current_values, this->target_values_, this->transition_length_);
|
|
}
|
|
|
|
optional<LightColorValues> apply() override {
|
|
// transition transformer does not handle 0 length as progress returns nan
|
|
if (this->transition_length_ == 0)
|
|
return this->target_values_;
|
|
|
|
if (this->transformer_ != nullptr) {
|
|
if (!this->transformer_->is_finished()) {
|
|
return this->transformer_->apply();
|
|
} else {
|
|
this->transformer_->stop();
|
|
this->transformer_ = nullptr;
|
|
}
|
|
}
|
|
|
|
if (millis() > this->start_time_ + this->length_ - this->transition_length_ &&
|
|
!this->secondary_transition_occurred_) {
|
|
// second transition back to start value
|
|
this->transformer_ = this->state_.get_output()->create_default_transition();
|
|
this->transformer_->setup(this->state_.current_values, this->get_start_values(), this->transition_length_);
|
|
this->secondary_transition_occurred_ = true;
|
|
}
|
|
|
|
// once transition is complete, don't change states until next transition
|
|
return optional<LightColorValues>();
|
|
}
|
|
|
|
// Restore the original values after the flash.
|
|
void stop() override {
|
|
this->state_.current_values = this->get_start_values();
|
|
this->state_.remote_values = this->get_start_values();
|
|
this->state_.publish_state();
|
|
}
|
|
|
|
protected:
|
|
LightState &state_;
|
|
uint32_t transition_length_;
|
|
bool secondary_transition_occurred_{false};
|
|
std::unique_ptr<LightTransformer> transformer_{nullptr};
|
|
};
|
|
|
|
} // namespace light
|
|
} // namespace esphome
|