esp_adc.hpp

#pragma once
 
#include <atomic>
#include <cstdint>
 
#include "adc.hpp"
#include "esp_adc/adc_continuous.h"
#include "hal/adc_types.h"
 
#if (SOC_ADC_DIGI_RESULT_BYTES == 2)
#define XR_ADC_DMA_OUTPUT_TYPE ADC_DIGI_OUTPUT_FORMAT_TYPE1
#define XR_ADC_DMA_GET_CHANNEL(p_data) ((p_data)->type1.channel)
#define XR_ADC_DMA_GET_DATA(p_data) ((p_data)->type1.data)
#else
#define XR_ADC_DMA_OUTPUT_TYPE ADC_DIGI_OUTPUT_FORMAT_TYPE2
#define XR_ADC_DMA_GET_CHANNEL(p_data) ((p_data)->type2.channel)
#define XR_ADC_DMA_GET_DATA(p_data) ((p_data)->type2.data)
#endif
 
namespace LibXR
{
 
class ESP32ADC
{
 public:
  class Channel : public ADC
  {
   public:
    Channel() : parent_(nullptr), idx_(0), channel_num_(0) {}
 
    Channel(ESP32ADC *parent, uint8_t idx, uint8_t channel_num)
        : parent_(parent), idx_(idx), channel_num_(channel_num)
    {
    }
 
    float Read() override { return parent_ ? parent_->ReadChannel(idx_) : 0.f; }
 
    uint8_t ChannelNumber() const { return channel_num_; }
 
   private:
    ESP32ADC *parent_;     
    uint8_t idx_;          
    uint8_t channel_num_;  
  };
 
  ESP32ADC(
      adc_unit_t unit, const adc_channel_t *channels, uint8_t num_channels,
      uint32_t freq = SOC_ADC_SAMPLE_FREQ_THRES_LOW,
      adc_atten_t attenuation = ADC_ATTEN_DB_12,
      adc_bitwidth_t bitwidth = static_cast<adc_bitwidth_t>(SOC_ADC_DIGI_MAX_BITWIDTH),
      float reference_voltage = 3.3f, size_t dma_buf_size = 256)
      : m_unit_(unit),
        m_num_channels_(num_channels),
        m_attenuation_(attenuation),
        m_bitwidth_(bitwidth),
        m_reference_voltage_(reference_voltage),
        m_max_raw_((1 << bitwidth) - 1)
  {
    m_patterns_ = new adc_digi_pattern_config_t[m_num_channels_];
    m_channels_ = new Channel[m_num_channels_];
    m_latest_values_ = new float[m_num_channels_];
    m_sum_buf_ = new int[m_num_channels_];
    m_cnt_buf_ = new int[m_num_channels_];
 
    for (uint8_t i = 0; i < m_num_channels_; ++i)
    {
      m_patterns_[i].atten = static_cast<uint8_t>(m_attenuation_);
      m_patterns_[i].channel = channels[i];
      m_patterns_[i].unit = static_cast<uint8_t>(m_unit_);
      m_patterns_[i].bit_width = static_cast<uint8_t>(m_bitwidth_);
      m_channels_[i] = Channel(this, i, channels[i]);
      m_latest_values_[i] = 0.f;
    }
 
    adc_continuous_handle_cfg_t adc_config = {
        .max_store_buf_size = dma_buf_size,
        .conv_frame_size = dma_buf_size / 2,
    };
    ESP_ERROR_CHECK(adc_continuous_new_handle(&adc_config, &m_handle_));
 
    adc_continuous_config_t dig_cfg = {};
    dig_cfg.sample_freq_hz = freq;
    if (m_unit_ == ADC_UNIT_1)
    {
      dig_cfg.conv_mode = ADC_CONV_SINGLE_UNIT_1;
    }
    else
    {
      dig_cfg.conv_mode = ADC_CONV_SINGLE_UNIT_2;
    }
    dig_cfg.format = XR_ADC_DMA_OUTPUT_TYPE;
    dig_cfg.adc_pattern = m_patterns_;
    dig_cfg.pattern_num = m_num_channels_;
    ESP_ERROR_CHECK(adc_continuous_config(m_handle_, &dig_cfg));
 
    adc_continuous_evt_cbs_t cbs = {
        .on_conv_done = &ESP32ADC::OnConvDone,
    };
    ESP_ERROR_CHECK(adc_continuous_register_event_callbacks(m_handle_, &cbs, this));
    ESP_ERROR_CHECK(adc_continuous_start(m_handle_));
  }
 
  ~ESP32ADC()
  {
    if (m_handle_)
    {
      adc_continuous_stop(m_handle_);
      adc_continuous_deinit(m_handle_);
    }
    delete[] m_patterns_;
    delete[] m_channels_;
    delete[] m_latest_values_;
    delete[] m_sum_buf_;
    delete[] m_cnt_buf_;
  }
 
  Channel &GetChannel(uint8_t idx) { return m_channels_[idx]; }
 
  float ReadChannel(uint8_t idx) { return m_latest_values_[idx]; }
 
 private:
  static bool IRAM_ATTR OnConvDone(adc_continuous_handle_t handle,
                                   const adc_continuous_evt_data_t *edata,
                                   void *user_data)
  {
    auto *self = reinterpret_cast<ESP32ADC *>(user_data);
    self->HandleSamples(edata->conv_frame_buffer, edata->size);
    return false;
  }
 
  void HandleSamples(const void *buf, size_t size_bytes)
  {
    for (uint8_t idx = 0; idx < m_num_channels_; ++idx)
    {
      m_sum_buf_[idx] = 0;
      m_cnt_buf_[idx] = 0;
    }
 
    size_t n = size_bytes / sizeof(adc_digi_output_data_t);
    const adc_digi_output_data_t *samples =
        static_cast<const adc_digi_output_data_t *>(buf);
 
    for (size_t i = 0; i < n; ++i)
    {
      uint8_t ch = XR_ADC_DMA_GET_CHANNEL(&samples[i]);
      for (uint8_t idx = 0; idx < m_num_channels_; ++idx)
      {
        if (ch == m_channels_[idx].ChannelNumber())
        {
          int raw = XR_ADC_DMA_GET_DATA(&samples[i]);
          m_sum_buf_[idx] += raw;
          m_cnt_buf_[idx] += 1;
          break;
        }
      }
    }
 
    for (uint8_t idx = 0; idx < m_num_channels_; ++idx)
    {
      if (m_cnt_buf_[idx] > 0)
      {
        float avg = Normalize(static_cast<float>(m_sum_buf_[idx]) / m_cnt_buf_[idx]);
        m_latest_values_[idx] = avg;
      }
    }
  }
 
  float Normalize(float raw) const
  {
    return (raw / static_cast<float>(m_max_raw_)) * m_reference_voltage_;
  }
 
  adc_digi_pattern_config_t *m_patterns_;  
  Channel *m_channels_;                    
  float *m_latest_values_;  
  int *m_sum_buf_;          
  int *m_cnt_buf_;          
 
  adc_unit_t m_unit_;                           
  uint8_t m_num_channels_;                      
  adc_atten_t m_attenuation_;                   
  adc_bitwidth_t m_bitwidth_;                   
  float m_reference_voltage_;                   
  uint16_t m_max_raw_;                          
  adc_continuous_handle_t m_handle_ = nullptr;  
};
 
}  // namespace LibXR