current_based_cover.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/components/cover/cover.h"
#include "esphome/components/sensor/sensor.h"
#include "esphome/core/automation.h"
#include "esphome/core/component.h"
#include <cfloat>
 
namespace esphome {
namespace current_based {
 
class CurrentBasedCover : public cover::Cover, public Component {
 public:
  void setup() override;
  void loop() override;
  void dump_config() override;
 
  Trigger<> *get_stop_trigger() { return &this->stop_trigger_; }
 
  Trigger<> *get_open_trigger() { return &this->open_trigger_; }
  void set_open_sensor(sensor::Sensor *open_sensor) { this->open_sensor_ = open_sensor; }
  void set_open_moving_current_threshold(float open_moving_current_threshold) {
    this->open_moving_current_threshold_ = open_moving_current_threshold;
  }
  void set_open_obstacle_current_threshold(float open_obstacle_current_threshold) {
    this->open_obstacle_current_threshold_ = open_obstacle_current_threshold;
  }
  void set_open_duration(uint32_t open_duration) { this->open_duration_ = open_duration; }
 
  Trigger<> *get_close_trigger() { return &this->close_trigger_; }
  void set_close_sensor(sensor::Sensor *close_sensor) { this->close_sensor_ = close_sensor; }
  void set_close_moving_current_threshold(float close_moving_current_threshold) {
    this->close_moving_current_threshold_ = close_moving_current_threshold;
  }
  void set_close_obstacle_current_threshold(float close_obstacle_current_threshold) {
    this->close_obstacle_current_threshold_ = close_obstacle_current_threshold;
  }
  void set_close_duration(uint32_t close_duration) { this->close_duration_ = close_duration; }
 
  void set_max_duration(uint32_t max_duration) { this->max_duration_ = max_duration; }
  void set_obstacle_rollback(float obstacle_rollback) { this->obstacle_rollback_ = obstacle_rollback; }
 
  void set_malfunction_detection(bool malfunction_detection) { this->malfunction_detection_ = malfunction_detection; }
  void set_start_sensing_delay(uint32_t start_sensing_delay) { this->start_sensing_delay_ = start_sensing_delay; }
 
  Trigger<> *get_malfunction_trigger() { return &this->malfunction_trigger_; }
 
  cover::CoverTraits get_traits() override;
 
 protected:
  void control(const cover::CoverCall &call) override;
  void stop_prev_trigger_();
 
  bool is_at_target_() const;
  bool is_opening_() const;
  bool is_opening_blocked_() const;
  bool is_closing_() const;
  bool is_closing_blocked_() const;
  bool is_initial_delay_finished_() const;
 
  void direction_idle_(float new_position = FLT_MAX);
  void start_direction_(cover::CoverOperation dir);
 
  void recompute_position_();
 
  Trigger<> stop_trigger_;
 
  sensor::Sensor *open_sensor_{nullptr};
  Trigger<> open_trigger_;
  float open_moving_current_threshold_;
  float open_obstacle_current_threshold_{FLT_MAX};
  uint32_t open_duration_;
 
  sensor::Sensor *close_sensor_{nullptr};
  Trigger<> close_trigger_;
  float close_moving_current_threshold_;
  float close_obstacle_current_threshold_{FLT_MAX};
  uint32_t close_duration_;
 
  uint32_t max_duration_{UINT32_MAX};
  bool malfunction_detection_{true};
  Trigger<> malfunction_trigger_;
  uint32_t start_sensing_delay_;
  float obstacle_rollback_;
 
  Trigger<> *prev_command_trigger_{nullptr};
  uint32_t last_recompute_time_{0};
  uint32_t start_dir_time_{0};
  uint32_t last_publish_time_{0};
  float target_position_{0};
 
  cover::CoverOperation last_operation_{cover::COVER_OPERATION_OPENING};
};
 
}  // namespace current_based
}  // namespace esphome