mirror of
https://github.com/esphome/esphome.git
synced 2025-02-20 20:08:20 +00:00
113 lines
4.4 KiB
C++
113 lines
4.4 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/helpers.h"
|
|
#include "light_color_values.h"
|
|
|
|
namespace esphome {
|
|
namespace light {
|
|
|
|
/// Base-class for all light color transformers, such as transitions or flashes.
|
|
class LightTransformer {
|
|
public:
|
|
LightTransformer(uint32_t start_time, uint32_t length, const LightColorValues &start_values,
|
|
const LightColorValues &target_values)
|
|
: start_time_(start_time), length_(length), start_values_(start_values), target_values_(target_values) {}
|
|
|
|
LightTransformer() = delete;
|
|
|
|
/// Whether this transformation is finished
|
|
virtual bool is_finished() { return this->get_progress() >= 1.0f; }
|
|
|
|
/// This will be called to get the current values for output.
|
|
virtual LightColorValues get_values() = 0;
|
|
|
|
/// The values that should be reported to the front-end.
|
|
virtual LightColorValues get_remote_values() { return this->get_target_values_(); }
|
|
|
|
/// The values that should be set after this transformation is complete.
|
|
virtual LightColorValues get_end_values() { return this->get_target_values_(); }
|
|
|
|
virtual bool publish_at_end() = 0;
|
|
virtual bool is_transition() = 0;
|
|
|
|
float get_progress() { return clamp((millis() - this->start_time_) / float(this->length_), 0.0f, 1.0f); }
|
|
|
|
protected:
|
|
const LightColorValues &get_start_values_() const { return this->start_values_; }
|
|
|
|
const LightColorValues &get_target_values_() const { return this->target_values_; }
|
|
|
|
uint32_t start_time_;
|
|
uint32_t length_;
|
|
LightColorValues start_values_;
|
|
LightColorValues target_values_;
|
|
};
|
|
|
|
class LightTransitionTransformer : public LightTransformer {
|
|
public:
|
|
LightTransitionTransformer(uint32_t start_time, uint32_t length, const LightColorValues &start_values,
|
|
const LightColorValues &target_values)
|
|
: LightTransformer(start_time, length, start_values, target_values) {
|
|
// When turning light on from off state, use colors from new.
|
|
if (!this->start_values_.is_on() && this->target_values_.is_on()) {
|
|
this->start_values_ = LightColorValues(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_ = LightColorValues(this->get_start_values_());
|
|
this->intermediate_values_.set_state(false);
|
|
}
|
|
}
|
|
|
|
LightColorValues get_values() 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_ = LightColorValues(this->get_end_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);
|
|
}
|
|
|
|
bool publish_at_end() override { return false; }
|
|
bool is_transition() override { return true; }
|
|
|
|
// 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); }
|
|
|
|
protected:
|
|
bool changing_color_mode_{false};
|
|
LightColorValues intermediate_values_{};
|
|
};
|
|
|
|
class LightFlashTransformer : public LightTransformer {
|
|
public:
|
|
LightFlashTransformer(uint32_t start_time, uint32_t length, const LightColorValues &start_values,
|
|
const LightColorValues &target_values)
|
|
: LightTransformer(start_time, length, start_values, target_values) {}
|
|
|
|
LightColorValues get_values() override { return this->get_target_values_(); }
|
|
|
|
LightColorValues get_end_values() override { return this->get_start_values_(); }
|
|
|
|
bool publish_at_end() override { return true; }
|
|
bool is_transition() override { return false; }
|
|
};
|
|
|
|
} // namespace light
|
|
} // namespace esphome
|