You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
195 lines
5.7 KiB
195 lines
5.7 KiB
#pragma once
|
|
|
|
#include "../common.h"
|
|
|
|
namespace esphome {
|
|
namespace xiaomi {
|
|
namespace bslamp2 {
|
|
|
|
class Preset {
|
|
public:
|
|
std::string name;
|
|
Preset *next_preset = nullptr;
|
|
std::string group_name;
|
|
light::LightCall *light_call;
|
|
|
|
explicit Preset(std::string name, std::string group_name) :
|
|
name(name), group_name(group_name) {}
|
|
|
|
void set_light_call(light::LightCall *light_call) {
|
|
this->light_call = light_call;
|
|
}
|
|
|
|
void apply() {
|
|
ESP_LOGI(TAG, "Activating light preset: %s/%s",
|
|
group_name.c_str(), name.c_str());
|
|
light_call->perform();
|
|
}
|
|
};
|
|
|
|
class PresetGroup {
|
|
public:
|
|
std::string name;
|
|
PresetGroup *next_group = nullptr;
|
|
Preset *first_preset = nullptr;
|
|
Preset *last_preset = nullptr;
|
|
Preset *active_preset = nullptr;
|
|
|
|
explicit PresetGroup(std::string g_name) : name(g_name) {}
|
|
|
|
Preset *make_preset(std::string p_name) {
|
|
auto p = get_preset(p_name);
|
|
if (p == nullptr) {
|
|
p = new Preset(p_name, this->name);
|
|
if (first_preset == nullptr) {
|
|
first_preset = last_preset = active_preset = p;
|
|
} else {
|
|
last_preset->next_preset = p;
|
|
last_preset = p;
|
|
}
|
|
}
|
|
return p;
|
|
}
|
|
|
|
Preset *get_preset(std::string p_name) {
|
|
for (auto p = first_preset; p != nullptr; p = p->next_preset) {
|
|
if (p->name == p_name) {
|
|
return p;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
};
|
|
|
|
class PresetsContainer : public Component {
|
|
public:
|
|
PresetGroup *first_group = nullptr;
|
|
PresetGroup *last_group = nullptr;
|
|
PresetGroup *active_group = nullptr;
|
|
|
|
explicit PresetsContainer(light::LightState *light) : _light(light) { }
|
|
|
|
void dump_config() {
|
|
if (first_group == nullptr) {
|
|
return;
|
|
}
|
|
ESP_LOGCONFIG(TAG, "Light Presets:");
|
|
for (auto g = first_group; g != nullptr; g = g->next_group) {
|
|
ESP_LOGCONFIG(TAG, " Preset group: %s", g->name.c_str());
|
|
for (auto p = g->first_preset; p != nullptr; p = p->next_preset) {
|
|
ESP_LOGCONFIG(TAG, " Preset: %s", p->name.c_str());
|
|
}
|
|
}
|
|
}
|
|
|
|
void add(std::string g_name, std::string p_name, float r, float g, float b) {
|
|
auto call = make_preset_call_(g_name, p_name);
|
|
call->set_rgb(r, g, b);
|
|
}
|
|
|
|
void add(std::string g_name, std::string p_name, float t) {
|
|
auto call = make_preset_call_(g_name, p_name);
|
|
call->set_color_temperature(t);
|
|
}
|
|
|
|
PresetGroup *get_group(std::string g_name) {
|
|
for (auto g = first_group; g != nullptr; g = g->next_group) {
|
|
if (g->name == g_name) {
|
|
return g;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void activate_next_group() {
|
|
if (active_group == nullptr) {
|
|
ESP_LOGW(TAG, "activate_next_group(): no preset groups defined");
|
|
return;
|
|
}
|
|
active_group = active_group->next_group == nullptr
|
|
? first_group : active_group->next_group;
|
|
if (active_group->active_preset == nullptr) {
|
|
ESP_LOGW(TAG, "activate_next_group(): no presets defined for group %s",
|
|
active_group->name.c_str());
|
|
return;
|
|
}
|
|
active_group->active_preset->apply();
|
|
}
|
|
|
|
void activate_next_preset() {
|
|
if (active_group == nullptr) {
|
|
ESP_LOGW(TAG, "activate_next_preset(): no preset groups defined");
|
|
return;
|
|
}
|
|
auto p = active_group->active_preset;
|
|
if (p == nullptr) {
|
|
ESP_LOGW(TAG, "activate_next_preset(): no presets defined for group %s",
|
|
active_group->name.c_str());
|
|
return;
|
|
}
|
|
active_group->active_preset = p->next_preset == nullptr
|
|
? active_group->first_preset : p->next_preset;
|
|
active_group->active_preset->apply();
|
|
}
|
|
|
|
void activate_group(std::string g_name) {
|
|
auto g = get_group(g_name);
|
|
if (g == nullptr) {
|
|
ESP_LOGE(TAG, "activate_group(%s): preset group does not exist",
|
|
g_name.c_str());
|
|
return;
|
|
}
|
|
auto p = g->active_preset;
|
|
if (p == nullptr) {
|
|
ESP_LOGW(TAG, "activate_group(%s): no presets defined for group",
|
|
g_name.c_str());
|
|
return;
|
|
}
|
|
p->apply();
|
|
}
|
|
|
|
void activate_preset(std::string g_name, std::string p_name) {
|
|
auto g = get_group(g_name);
|
|
if (g == nullptr) {
|
|
ESP_LOGE(TAG, "activate_preset(%s, %s): preset group %s does not exist",
|
|
g_name.c_str(), p_name.c_str(), g_name.c_str());
|
|
return;
|
|
}
|
|
auto p = g->get_preset(p_name);
|
|
if (p == nullptr) {
|
|
ESP_LOGE(TAG, "activate_preset(%s, %s): preset %s does not exist in group %s",
|
|
g_name.c_str(), p_name.c_str(), p_name.c_str(), g_name.c_str());
|
|
return;
|
|
}
|
|
p->apply();
|
|
}
|
|
|
|
protected:
|
|
light::LightState *_light;
|
|
|
|
PresetGroup *make_preset_group_(std::string g_name) {
|
|
auto g = get_group(g_name);
|
|
if (g == nullptr) {
|
|
g = new PresetGroup(g_name);
|
|
if (first_group == nullptr) {
|
|
first_group = last_group = active_group = g;
|
|
} else {
|
|
last_group->next_group = g;
|
|
last_group = g;
|
|
}
|
|
}
|
|
return g;
|
|
}
|
|
|
|
light::LightCall *make_preset_call_(std::string g_name, std::string p_name) {
|
|
auto g = make_preset_group_(g_name);
|
|
auto p = g->make_preset(p_name);
|
|
auto c = new light::LightCall(_light);
|
|
p->set_light_call(c);
|
|
return c;
|
|
}
|
|
};
|
|
|
|
} // namespace bslamp2
|
|
} // namespace xiaomi
|
|
} // namespace esphome
|