resampler_microphone.h

Go to the documentation of this file.

#pragma once
 
#ifdef USE_ESP32
 
#include "esphome/components/audio/audio.h"
#include "esphome/components/microphone/microphone.h"
#include "esphome/components/microphone/microphone_source.h"
 
#include "esphome/core/component.h"
 
#include <resampler.h>  // esp-audio-libs
 
#include <memory>
#include <vector>
 
namespace esphome::resampler {
 
class ResamplerMicrophone final : public Component, public microphone::Microphone {
 public:
  explicit ResamplerMicrophone(microphone::MicrophoneSource *source) : source_(source) {}
 
  void setup() override;
  void loop() override;
  void dump_config() override;
 
  void start() override;
  void stop() override;
 
  void set_target_sample_rate(uint32_t target_sample_rate) { this->target_sample_rate_ = target_sample_rate; }
  void set_filters(uint16_t filters) { this->filters_ = filters; }
  void set_taps(uint16_t taps) { this->taps_ = taps; }
 
 protected:
  bool init_resampler_(const audio::AudioStreamInfo &input_stream_info);
 
  void process_audio_(const std::vector<uint8_t> &data);
 
  microphone::MicrophoneSource *source_;
  std::unique_ptr<esp_audio_libs::resampler::Resampler> resampler_;
  // Reused for every chunk so resampling does not allocate
  std::vector<uint8_t> output_buffer_;
 
  // Format the resampler is set up for
  audio::AudioStreamInfo input_stream_info_;
 
  uint32_t target_sample_rate_;
  uint16_t taps_;
  uint16_t filters_;
 
  uint8_t active_listeners_{0};
  bool resampler_ready_{false};
};
 
}  // namespace esphome::resampler
 
#endif  // USE_ESP32