audio_reader.h

Go to the documentation of this file.

#pragma once
 
#ifdef USE_ESP32
 
#include "audio.h"
#include "audio_transfer_buffer.h"
 
#include "esphome/components/ring_buffer/ring_buffer.h"
 
#include "esp_err.h"
 
#include <esp_http_client.h>
 
namespace esphome::audio {
 
enum class AudioReaderState : uint8_t {
  READING = 0,  // More data is available to read
  FINISHED,     // All data has been read and transferred
  FAILED,       // Encountered an error
};
 
class AudioReader {
  /*
   * @brief Class that facilitates reading a raw audio file.
   * Files can be read from flash (stored in a AudioFile struct) or from an http source.
   * The file data is sent to a ring buffer sink.
   */
 public:
  AudioReader(size_t buffer_size) : buffer_size_(buffer_size) {}
  ~AudioReader();
 
  esp_err_t add_sink(const std::weak_ptr<ring_buffer::RingBuffer> &output_ring_buffer);
 
  esp_err_t start(const std::string &uri, AudioFileType &file_type);
 
  esp_err_t start(AudioFile *audio_file, AudioFileType &file_type);
 
  AudioReaderState read();
 
 protected:
  static esp_err_t http_event_handler(esp_http_client_event_t *evt);
 
  AudioReaderState file_read_();
  AudioReaderState http_read_();
 
  std::shared_ptr<ring_buffer::RingBuffer> file_ring_buffer_;
  std::unique_ptr<AudioSinkTransferBuffer> output_transfer_buffer_;
  void cleanup_connection_();
 
  size_t buffer_size_;
  uint32_t last_data_read_ms_;
 
  esp_http_client_handle_t client_{nullptr};
 
  AudioFile *current_audio_file_{nullptr};
  AudioFileType audio_file_type_{AudioFileType::NONE};
  const uint8_t *file_current_{nullptr};
};
}  // namespace esphome::audio
 
#endif