i2s_audio_speaker.h

Go to the documentation of this file.

#pragma once
 
#ifdef USE_ESP32
 
#include "../i2s_audio.h"
 
#include <freertos/event_groups.h>
#include <freertos/queue.h>
#include <freertos/FreeRTOS.h>
 
#include "esphome/components/audio/audio.h"
#include "esphome/components/speaker/speaker.h"
 
#include "esphome/core/component.h"
#include "esphome/core/gpio.h"
#include "esphome/core/helpers.h"
#include "esphome/core/ring_buffer.h"
 
namespace esphome {
namespace i2s_audio {
 
class I2SAudioSpeaker : public I2SAudioOut, public speaker::Speaker, public Component {
 public:
  float get_setup_priority() const override { return esphome::setup_priority::PROCESSOR; }
 
  void setup() override;
  void loop() override;
 
  void set_buffer_duration(uint32_t buffer_duration_ms) { this->buffer_duration_ms_ = buffer_duration_ms; }
  void set_timeout(uint32_t ms) { this->timeout_ = ms; }
#ifdef USE_I2S_LEGACY
#if SOC_I2S_SUPPORTS_DAC
  void set_internal_dac_mode(i2s_dac_mode_t mode) { this->internal_dac_mode_ = mode; }
#endif
  void set_dout_pin(uint8_t pin) { this->dout_pin_ = pin; }
  void set_i2s_comm_fmt(i2s_comm_format_t mode) { this->i2s_comm_fmt_ = mode; }
#else
  void set_dout_pin(uint8_t pin) { this->dout_pin_ = (gpio_num_t) pin; }
  void set_i2s_comm_fmt(std::string mode) { this->i2s_comm_fmt_ = std::move(mode); }
#endif
 
  void start() override;
  void stop() override;
  void finish() override;
 
  void set_pause_state(bool pause_state) override { this->pause_state_ = pause_state; }
  bool get_pause_state() const override { return this->pause_state_; }
 
  size_t play(const uint8_t *data, size_t length, TickType_t ticks_to_wait) override;
  size_t play(const uint8_t *data, size_t length) override { return play(data, length, 0); }
 
  bool has_buffered_data() const override;
 
  void set_volume(float volume) override;
 
  void set_mute_state(bool mute_state) override;
 
 protected:
  static void speaker_task(void *params);
 
  void stop_(bool wait_on_empty);
 
  bool send_esp_err_to_event_group_(esp_err_t err);
 
#ifndef USE_I2S_LEGACY
  static bool i2s_overflow_cb(i2s_chan_handle_t handle, i2s_event_data_t *event, void *user_ctx);
#endif
 
  esp_err_t allocate_buffers_(size_t data_buffer_size, size_t ring_buffer_size);
 
  esp_err_t start_i2s_driver_(audio::AudioStreamInfo &audio_stream_info);
 
  void delete_task_(size_t buffer_size);
 
  TaskHandle_t speaker_task_handle_{nullptr};
  EventGroupHandle_t event_group_{nullptr};
 
  QueueHandle_t i2s_event_queue_;
 
  uint8_t *data_buffer_;
  std::shared_ptr<RingBuffer> audio_ring_buffer_;
 
  uint32_t buffer_duration_ms_;
 
  optional<uint32_t> timeout_;
 
  bool task_created_{false};
  bool pause_state_{false};
 
  int16_t q15_volume_factor_{INT16_MAX};
 
  size_t bytes_written_{0};
 
#ifdef USE_I2S_LEGACY
#if SOC_I2S_SUPPORTS_DAC
  i2s_dac_mode_t internal_dac_mode_{I2S_DAC_CHANNEL_DISABLE};
#endif
  uint8_t dout_pin_;
  i2s_comm_format_t i2s_comm_fmt_;
#else
  gpio_num_t dout_pin_;
  std::string i2s_comm_fmt_;
  i2s_chan_handle_t tx_handle_;
#endif
 
  uint32_t accumulated_frames_written_{0};
};
 
}  // namespace i2s_audio
}  // namespace esphome
 
#endif  // USE_ESP32