mirror of
https://github.com/esphome/esphome.git
synced 2024-12-25 23:14:54 +01:00
c61abf6aca
* null initialize total sensor. * pedantic styling fix Co-authored-by: Guillermo Ruffino <glm.net@gmail.com> Co-authored-by: Guillermo Ruffino <glm.net@gmail.com>
88 lines
2.5 KiB
C++
88 lines
2.5 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/hal.h"
|
|
#include "esphome/components/sensor/sensor.h"
|
|
|
|
#if defined(USE_ESP32) && !defined(USE_ESP32_VARIANT_ESP32C3)
|
|
#include <driver/pcnt.h>
|
|
#define HAS_PCNT
|
|
#endif
|
|
|
|
namespace esphome {
|
|
namespace pulse_counter {
|
|
|
|
enum PulseCounterCountMode {
|
|
PULSE_COUNTER_DISABLE = 0,
|
|
PULSE_COUNTER_INCREMENT,
|
|
PULSE_COUNTER_DECREMENT,
|
|
};
|
|
|
|
#ifdef HAS_PCNT
|
|
using pulse_counter_t = int16_t;
|
|
#else
|
|
using pulse_counter_t = int32_t;
|
|
#endif
|
|
|
|
struct PulseCounterStorageBase {
|
|
virtual bool pulse_counter_setup(InternalGPIOPin *pin) = 0;
|
|
virtual pulse_counter_t read_raw_value() = 0;
|
|
|
|
InternalGPIOPin *pin;
|
|
PulseCounterCountMode rising_edge_mode{PULSE_COUNTER_INCREMENT};
|
|
PulseCounterCountMode falling_edge_mode{PULSE_COUNTER_DISABLE};
|
|
uint32_t filter_us{0};
|
|
pulse_counter_t last_value{0};
|
|
};
|
|
|
|
struct BasicPulseCounterStorage : public PulseCounterStorageBase {
|
|
static void gpio_intr(BasicPulseCounterStorage *arg);
|
|
|
|
bool pulse_counter_setup(InternalGPIOPin *pin) override;
|
|
pulse_counter_t read_raw_value() override;
|
|
|
|
volatile pulse_counter_t counter{0};
|
|
volatile uint32_t last_pulse{0};
|
|
|
|
ISRInternalGPIOPin isr_pin;
|
|
};
|
|
|
|
#ifdef HAS_PCNT
|
|
struct HwPulseCounterStorage : public PulseCounterStorageBase {
|
|
bool pulse_counter_setup(InternalGPIOPin *pin) override;
|
|
pulse_counter_t read_raw_value() override;
|
|
|
|
pcnt_unit_t pcnt_unit;
|
|
};
|
|
#endif
|
|
|
|
PulseCounterStorageBase *get_storage(bool hw_pcnt = false);
|
|
|
|
class PulseCounterSensor : public sensor::Sensor, public PollingComponent {
|
|
public:
|
|
explicit PulseCounterSensor(bool hw_pcnt = false) : storage_(*get_storage(hw_pcnt)) {}
|
|
|
|
void set_pin(InternalGPIOPin *pin) { pin_ = pin; }
|
|
void set_rising_edge_mode(PulseCounterCountMode mode) { storage_.rising_edge_mode = mode; }
|
|
void set_falling_edge_mode(PulseCounterCountMode mode) { storage_.falling_edge_mode = mode; }
|
|
void set_filter_us(uint32_t filter) { storage_.filter_us = filter; }
|
|
void set_total_sensor(sensor::Sensor *total_sensor) { total_sensor_ = total_sensor; }
|
|
|
|
void set_total_pulses(uint32_t pulses);
|
|
|
|
/// Unit of measurement is "pulses/min".
|
|
void setup() override;
|
|
void update() override;
|
|
float get_setup_priority() const override { return setup_priority::DATA; }
|
|
void dump_config() override;
|
|
|
|
protected:
|
|
InternalGPIOPin *pin_;
|
|
PulseCounterStorageBase &storage_;
|
|
uint32_t last_time_{0};
|
|
uint32_t current_total_{0};
|
|
sensor::Sensor *total_sensor_{nullptr};
|
|
};
|
|
|
|
} // namespace pulse_counter
|
|
} // namespace esphome
|