counter_sensor.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
 
#include "esphome/core/component.h"
#include "esphome/core/preferences.h"
#include "esphome/components/sensor/sensor.h"
 
namespace esphome::counter {
 
class CounterSensor final : public sensor::Sensor, public Component {
 public:
  CounterSensor(bool restore, int64_t initial_value) : value_(initial_value), restore_(restore) {}
 
  void setup() override;
  void dump_config() override;
  // restore value before `on_boot` automations run
  float get_setup_priority() const override { return setup_priority::DATA + 50.0f; }
 
  void count_updates_from(sensor::Sensor *source) {
    source->add_on_state_callback([this](float) { this->increment(); });
  }
 
  template<typename T> void count_true_from(T *source) {
    source->add_on_state_callback([this](bool state) {
      if (state)
        this->increment();
    });
  }
 
  void set_value(int64_t value);
  void increment(int64_t amount = 1) {
    this->set_value(static_cast<int64_t>(static_cast<uint64_t>(this->value_) + static_cast<uint64_t>(amount)));
  }
 
 protected:
  ESPPreferenceObject pref_;
  int64_t value_;
  bool restore_;
};
 
}  // namespace esphome::counter