light_state.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/component.h"
#include "esphome/core/entity_base.h"
#include "esphome/core/optional.h"
#include "esphome/core/preferences.h"
#include "esphome/core/string_ref.h"
#include "light_call.h"
#include "light_color_values.h"
#include "light_effect.h"
#include "light_traits.h"
#include "light_transformer.h"
 
#include "esphome/core/hal.h"
#include "esphome/core/helpers.h"
#include "esphome/core/progmem.h"
#include <strings.h>
#include <vector>
 
namespace esphome::light {
 
class LightOutput;
class LightState;
 
class LightRemoteValuesListener {
 public:
  virtual void on_light_remote_values_update() = 0;
};
 
class LightTargetStateReachedListener {
 public:
  virtual void on_light_target_state_reached() = 0;
};
 
enum LightRestoreMode : uint8_t {
  LIGHT_RESTORE_DEFAULT_OFF,
  LIGHT_RESTORE_DEFAULT_ON,
  LIGHT_ALWAYS_OFF,
  LIGHT_ALWAYS_ON,
  LIGHT_RESTORE_INVERTED_DEFAULT_OFF,
  LIGHT_RESTORE_INVERTED_DEFAULT_ON,
  LIGHT_RESTORE_AND_OFF,
  LIGHT_RESTORE_AND_ON,
};
 
struct LightStateRTCState {
  LightStateRTCState(ColorMode color_mode, bool state, float brightness, float color_brightness, float red, float green,
                     float blue, float white, float color_temp, float cold_white, float warm_white)
      : brightness(brightness),
        color_brightness(color_brightness),
        red(red),
        green(green),
        blue(blue),
        white(white),
        color_temp(color_temp),
        cold_white(cold_white),
        warm_white(warm_white),
        effect(0),
        color_mode(color_mode),
        state(state) {}
  LightStateRTCState() = default;
  // Group 4-byte aligned members first
  float brightness{1.0f};
  float color_brightness{1.0f};
  float red{1.0f};
  float green{1.0f};
  float blue{1.0f};
  float white{1.0f};
  float color_temp{1.0f};
  float cold_white{1.0f};
  float warm_white{1.0f};
  uint32_t effect{0};
  // Group smaller members at the end
  ColorMode color_mode{ColorMode::UNKNOWN};
  bool state{false};
};
 
class LightState : public EntityBase, public Component {
 public:
  LightState(LightOutput *output);
 
  LightTraits get_traits();
 
  LightCall turn_on();
  LightCall turn_off();
  LightCall toggle();
  LightCall make_call();
 
  // ========== INTERNAL METHODS ==========
  // (In most use cases you won't need these)
  void setup() override;
  void dump_config() override;
  void loop() override;
  float get_setup_priority() const override { return setup_priority::HARDWARE - 1.0f; }
 
  LightColorValues current_values;
 
  LightColorValues remote_values;
 
  void publish_state();
 
  LightOutput *get_output() const;
 
  StringRef get_effect_name();
 
  void add_remote_values_listener(LightRemoteValuesListener *listener);
 
  void add_target_state_reached_listener(LightTargetStateReachedListener *listener);
 
  void set_default_transition_length(uint32_t default_transition_length) {
    this->default_transition_length_ = default_transition_length;
  }
  uint32_t get_default_transition_length() const { return this->default_transition_length_; }
 
  void set_flash_transition_length(uint32_t flash_transition_length) {
    this->flash_transition_length_ = flash_transition_length;
  }
  uint32_t get_flash_transition_length() const { return this->flash_transition_length_; }
 
  void set_gamma_correct(float gamma_correct) { this->gamma_correct_ = gamma_correct; }
  float get_gamma_correct() const { return this->gamma_correct_; }
 
#ifdef USE_LIGHT_GAMMA_LUT
  void set_gamma_table(const uint16_t *forward) { this->gamma_table_ = forward; }
 
  const uint16_t *get_gamma_table() const { return this->gamma_table_; }
 
  float gamma_correct_lut(float value) const;
  float gamma_uncorrect_lut(float value) const;
#else
  float gamma_correct_lut(float value) const { return value; }
  float gamma_uncorrect_lut(float value) const { return value; }
#endif  // USE_LIGHT_GAMMA_LUT
 
  void set_restore_mode(LightRestoreMode restore_mode) { this->restore_mode_ = restore_mode; }
 
  void set_initial_state(void (*callback)(LightStateRTCState &)) { this->initial_state_callback_ = callback; }
 
  bool supports_effects() const { return !this->effects_.empty(); }
 
  const FixedVector<LightEffect *> &get_effects() const { return this->effects_; }
 
  void add_effects(const std::initializer_list<LightEffect *> &effects);
 
  size_t get_effect_count() const { return this->effects_.size(); }
 
  uint32_t get_current_effect_index() const { return this->active_effect_index_; }
 
  uint32_t get_effect_index(const std::string &effect_name) const {
    if (str_equals_case_insensitive(effect_name, "none")) {
      return 0;
    }
    for (size_t i = 0; i < this->effects_.size(); i++) {
      if (str_equals_case_insensitive(effect_name, this->effects_[i]->get_name())) {
        return i + 1;  // Effects are 1-indexed in active_effect_index_
      }
    }
    return 0;  // Effect not found
  }
 
  uint32_t get_effect_index(const char *name, size_t len) const {
    if (len == 4 && ESPHOME_strncasecmp_P(name, ESPHOME_PSTR("none"), 4) == 0) {
      return 0;
    }
    StringRef ref(name, len);
    for (size_t i = 0; i < this->effects_.size(); i++) {
      if (str_equals_case_insensitive(ref, this->effects_[i]->get_name())) {
        return i + 1;
      }
    }
    return 0;
  }
 
  LightEffect *get_effect_by_index(uint32_t index) const {
    if (index == 0 || index > this->effects_.size()) {
      return nullptr;
    }
    return this->effects_[index - 1];  // Effects are 1-indexed in active_effect_index_
  }
 
  std::string get_effect_name_by_index(uint32_t index) const {
    if (index == 0) {
      return "None";
    }
    if (index > this->effects_.size()) {
      return "";  // Invalid index
    }
    return std::string(this->effects_[index - 1]->get_name());
  }
 
  void current_values_as_binary(bool *binary) { this->current_values.as_binary(binary); }
 
  void current_values_as_brightness(float *brightness);
 
  void current_values_as_rgb(float *red, float *green, float *blue);
 
  void current_values_as_rgbw(float *red, float *green, float *blue, float *white);
 
  void current_values_as_rgbww(float *red, float *green, float *blue, float *cold_white, float *warm_white,
                               bool constant_brightness = false);
 
  void current_values_as_rgbct(float *red, float *green, float *blue, float *color_temperature,
                               float *white_brightness);
 
  void current_values_as_cwww(float *cold_white, float *warm_white, bool constant_brightness = false);
 
  void current_values_as_ct(float *color_temperature, float *white_brightness);
 
  bool is_transformer_active() const { return this->is_transformer_active_; }
 
 protected:
  friend LightOutput;
  friend LightCall;
  friend class AddressableLight;
 
  void start_effect_(uint32_t effect_index);
  LightEffect *get_active_effect_();
  void stop_effect_();
  void start_transition_(const LightColorValues &target, uint32_t length, bool set_remote_values);
 
  void start_flash_(const LightColorValues &target, uint32_t length, bool set_remote_values);
 
  void set_immediately_(const LightColorValues &target, bool set_remote_values);
 
  void save_remote_values_();
 
  void disable_loop_if_idle_();
 
  void schedule_write_() {
    this->next_write_ = true;
    this->enable_loop();
  }
 
  LightOutput *output_;
  std::unique_ptr<LightTransformer> transformer_{nullptr};
  FixedVector<LightEffect *> effects_;
  ESPPreferenceObject rtc_;
 
  std::unique_ptr<std::vector<LightRemoteValuesListener *>> remote_values_listeners_;
 
  std::unique_ptr<std::vector<LightTargetStateReachedListener *>> target_state_reached_listeners_;
 
  void (*initial_state_callback_)(LightStateRTCState &){nullptr};
 
  uint32_t active_effect_index_{};
  uint32_t default_transition_length_{};
  uint32_t flash_transition_length_{};
  float gamma_correct_{};
#ifdef USE_LIGHT_GAMMA_LUT
  const uint16_t *gamma_table_{nullptr};
#endif  // USE_LIGHT_GAMMA_LUT
 
  bool next_write_{true};
  // for effects, true if a transformer (transition) is active.
  bool is_transformer_active_{false};
  LightRestoreMode restore_mode_;
};
 
}  // namespace esphome::light