108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#include "esphome/components/sensor/sensor.h"
|
|
#include "esphome/components/voltage_sampler/voltage_sampler.h"
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/hal.h"
|
|
|
|
#ifdef USE_ESP32
|
|
#include <esp_adc_cal.h>
|
|
#include "driver/adc.h"
|
|
#endif // USE_ESP32
|
|
|
|
namespace esphome {
|
|
namespace adc {
|
|
|
|
#ifdef USE_ESP32
|
|
// clang-format off
|
|
#if (ESP_IDF_VERSION_MAJOR == 5 && \
|
|
((ESP_IDF_VERSION_MINOR == 0 && ESP_IDF_VERSION_PATCH >= 5) || \
|
|
(ESP_IDF_VERSION_MINOR == 1 && ESP_IDF_VERSION_PATCH >= 3) || \
|
|
(ESP_IDF_VERSION_MINOR >= 2)) \
|
|
)
|
|
// clang-format on
|
|
static const adc_atten_t ADC_ATTEN_DB_12_COMPAT = ADC_ATTEN_DB_12;
|
|
#else
|
|
static const adc_atten_t ADC_ATTEN_DB_12_COMPAT = ADC_ATTEN_DB_11;
|
|
#endif
|
|
#endif // USE_ESP32
|
|
|
|
enum class SamplingMode : uint8_t {
|
|
AVG = 0,
|
|
MIN = 1,
|
|
MAX = 2,
|
|
};
|
|
|
|
const LogString *sampling_mode_to_str(SamplingMode mode);
|
|
|
|
class Aggregator {
|
|
public:
|
|
Aggregator(SamplingMode mode);
|
|
void add_sample(uint32_t value);
|
|
uint32_t aggregate();
|
|
|
|
protected:
|
|
uint32_t aggr_{0};
|
|
uint32_t samples_{0};
|
|
SamplingMode mode_{SamplingMode::AVG};
|
|
};
|
|
|
|
class ADCSensor : public sensor::Sensor, public PollingComponent, public voltage_sampler::VoltageSampler {
|
|
public:
|
|
#ifdef USE_ESP32
|
|
/// Set the attenuation for this pin. Only available on the ESP32.
|
|
void set_attenuation(adc_atten_t attenuation) { this->attenuation_ = attenuation; }
|
|
void set_channel1(adc1_channel_t channel) {
|
|
this->channel1_ = channel;
|
|
this->channel2_ = ADC2_CHANNEL_MAX;
|
|
}
|
|
void set_channel2(adc2_channel_t channel) {
|
|
this->channel2_ = channel;
|
|
this->channel1_ = ADC1_CHANNEL_MAX;
|
|
}
|
|
void set_autorange(bool autorange) { this->autorange_ = autorange; }
|
|
#endif // USE_ESP32
|
|
|
|
/// Update ADC values
|
|
void update() override;
|
|
/// Setup ADC
|
|
void setup() override;
|
|
void dump_config() override;
|
|
/// `HARDWARE_LATE` setup priority
|
|
float get_setup_priority() const override;
|
|
void set_pin(InternalGPIOPin *pin) { this->pin_ = pin; }
|
|
void set_output_raw(bool output_raw) { this->output_raw_ = output_raw; }
|
|
void set_sample_count(uint8_t sample_count);
|
|
void set_sampling_mode(SamplingMode sampling_mode);
|
|
float sample() override;
|
|
|
|
#ifdef USE_ESP8266
|
|
std::string unique_id() override;
|
|
#endif // USE_ESP8266
|
|
|
|
#ifdef USE_RP2040
|
|
void set_is_temperature() { this->is_temperature_ = true; }
|
|
#endif // USE_RP2040
|
|
|
|
protected:
|
|
uint8_t sample_count_{1};
|
|
bool output_raw_{false};
|
|
InternalGPIOPin *pin_;
|
|
SamplingMode sampling_mode_{SamplingMode::AVG};
|
|
|
|
#ifdef USE_RP2040
|
|
bool is_temperature_{false};
|
|
#endif // USE_RP2040
|
|
|
|
#ifdef USE_ESP32
|
|
adc_atten_t attenuation_{ADC_ATTEN_DB_0};
|
|
adc1_channel_t channel1_{ADC1_CHANNEL_MAX};
|
|
adc2_channel_t channel2_{ADC2_CHANNEL_MAX};
|
|
bool autorange_{false};
|
|
esp_adc_cal_characteristics_t cal_characteristics_[SOC_ADC_ATTEN_NUM] = {};
|
|
#endif // USE_ESP32
|
|
};
|
|
|
|
} // namespace adc
|
|
} // namespace esphome
|