cover.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/component.h"
#include "esphome/core/entity_base.h"
#include "esphome/core/helpers.h"
#include "esphome/core/log.h"
#include "esphome/core/preferences.h"
 
#include "cover_traits.h"
 
namespace esphome::cover {
 
static constexpr float COVER_OPEN = 1.0f;
static constexpr float COVER_CLOSED = 0.0f;
 
#define LOG_COVER(prefix, type, obj) \
  if (log_entity_header(TAG, prefix, LOG_STR_LITERAL(type), (obj))) { \
    auto traits_ = (obj)->get_traits(); \
    if (traits_.get_is_assumed_state()) { \
      ESP_LOGCONFIG(TAG, "%s  Assumed State: YES", prefix); \
    } \
    LOG_ENTITY_DEVICE_CLASS(TAG, prefix, *(obj)); \
  }
 
class Cover;
 
class CoverCall {
 public:
  CoverCall(Cover *parent);
 
  CoverCall &set_command(const char *command);
  CoverCall &set_command_open();
  CoverCall &set_command_close();
  CoverCall &set_command_stop();
  CoverCall &set_command_stop_tilt();
  CoverCall &set_command_toggle();
  CoverCall &set_position(float position);
  CoverCall &set_tilt(float tilt);
  CoverCall &set_stop(bool stop);
  CoverCall &set_stop_tilt(bool stop_tilt);
 
  void perform();
 
  const optional<float> &get_position() const;
  bool get_stop() const { return this->stop_; }
  bool get_stop_tilt() const { return this->stop_tilt_; }
  const optional<float> &get_tilt() const;
  const optional<bool> &get_toggle() const;
 
 protected:
  void validate_();
 
  Cover *parent_;
  bool stop_{false};
  bool stop_tilt_{false};
  optional<float> position_{};
  optional<float> tilt_{};
  optional<bool> toggle_{};
};
 
struct CoverRestoreState {
  float position;
  float tilt;
 
  CoverCall to_call(Cover *cover);
  void apply(Cover *cover);
} __attribute__((packed));
 
enum CoverOperation : uint8_t {
  COVER_OPERATION_IDLE = 0,
  COVER_OPERATION_OPENING,
  COVER_OPERATION_CLOSING,
};
 
const LogString *cover_operation_to_str(CoverOperation op);
 
class Cover : public EntityBase {
 public:
  explicit Cover();
 
  CoverOperation current_operation{COVER_OPERATION_IDLE};
  float position;
  float tilt{COVER_OPEN};
 
  CoverCall make_call() { return {this}; }
 
  template<typename F> void add_on_state_callback(F &&f) { this->state_callback_.add(std::forward<F>(f)); }
 
  void publish_state(bool save = true);
 
  virtual CoverTraits get_traits() = 0;
 
  bool is_fully_open() const { return this->position == COVER_OPEN; }
  bool is_fully_closed() const { return this->position == COVER_CLOSED; }
 
 protected:
  friend CoverCall;
 
  virtual void control(const CoverCall &call) = 0;
 
  optional<CoverRestoreState> restore_state_();
 
  LazyCallbackManager<void()> state_callback_{};
 
  ESPPreferenceObject rtc_;
};
 
}  // namespace esphome::cover