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 "light_call.h"
#include "light_color_values.h"
#include "light_effect.h"
#include "light_traits.h"
#include "light_transformer.h"

#include <vector>

namespace esphome {
namespace light {

class LightOutput;

enum LightRestoreMode {
  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)
      : color_mode(color_mode),
        state(state),
        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) {}
  LightStateRTCState() = default;
  ColorMode color_mode{ColorMode::UNKNOWN};
  bool state{false};
  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};
};

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;

  LightColorValues current_values;

  LightColorValues remote_values;

  void publish_state();

  LightOutput *get_output() const;

  std::string get_effect_name();

  void add_new_remote_values_callback(std::function<void()> &&send_callback);

  void add_new_target_state_reached_callback(std::function<void()> &&send_callback);

  void set_default_transition_length(uint32_t default_transition_length);
  uint32_t get_default_transition_length() const;

  void set_flash_transition_length(uint32_t flash_transition_length);
  uint32_t get_flash_transition_length() const;

  void set_gamma_correct(float gamma_correct);
  float get_gamma_correct() const { return this->gamma_correct_; }

  void set_restore_mode(LightRestoreMode restore_mode);

  void set_initial_state(const LightStateRTCState &initial_state);

  bool supports_effects();

  const std::vector<LightEffect *> &get_effects() const;

  void add_effects(const std::vector<LightEffect *> &effects);

  void current_values_as_binary(bool *binary);

  void current_values_as_brightness(float *brightness);

  void current_values_as_rgb(float *red, float *green, float *blue, bool color_interlock = false);

  void current_values_as_rgbw(float *red, float *green, float *blue, float *white, bool color_interlock = false);

  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();

 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_();

  LightOutput *output_;
  uint32_t active_effect_index_{};
  std::unique_ptr<LightTransformer> transformer_{nullptr};
  bool next_write_{true};

  ESPPreferenceObject rtc_;

  CallbackManager<void()> remote_values_callback_{};

  CallbackManager<void()> target_state_reached_callback_{};

  uint32_t default_transition_length_{};
  uint32_t flash_transition_length_{};
  float gamma_correct_{};
  LightRestoreMode restore_mode_;
  optional<LightStateRTCState> initial_state_{};
  std::vector<LightEffect *> effects_;

  // for effects, true if a transformer (transition) is active.
  bool is_transformer_active_ = false;
};

}  // namespace light
}  // namespace esphome