mirror of
https://github.com/esphome/esphome.git
synced 2025-04-19 01:00:28 +01:00
142 lines
3.6 KiB
C++
142 lines
3.6 KiB
C++
#include "i2s_audio_microphone.h"
|
|
|
|
#ifdef USE_ESP32
|
|
|
|
#include <driver/i2s.h>
|
|
|
|
#include "esphome/core/hal.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
namespace esphome {
|
|
namespace i2s_audio {
|
|
|
|
static const size_t BUFFER_SIZE = 512;
|
|
|
|
static const char *const TAG = "i2s_audio.microphone";
|
|
|
|
void I2SAudioMicrophone::setup() {
|
|
ESP_LOGCONFIG(TAG, "Setting up I2S Audio Microphone...");
|
|
this->buffer_.resize(BUFFER_SIZE);
|
|
|
|
#if SOC_I2S_SUPPORTS_ADC
|
|
if (this->adc_) {
|
|
if (this->parent_->get_port() != I2S_NUM_0) {
|
|
ESP_LOGE(TAG, "Internal ADC only works on I2S0!");
|
|
this->mark_failed();
|
|
return;
|
|
}
|
|
} else
|
|
#endif
|
|
if (this->pdm_) {
|
|
if (this->parent_->get_port() != I2S_NUM_0) {
|
|
ESP_LOGE(TAG, "PDM only works on I2S0!");
|
|
this->mark_failed();
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
void I2SAudioMicrophone::start() {
|
|
if (this->is_failed())
|
|
return;
|
|
this->state_ = microphone::STATE_STARTING;
|
|
}
|
|
void I2SAudioMicrophone::start_() {
|
|
if (!this->parent_->try_lock()) {
|
|
return; // Waiting for another i2s to return lock
|
|
}
|
|
i2s_driver_config_t config = {
|
|
.mode = (i2s_mode_t) (I2S_MODE_MASTER | I2S_MODE_RX),
|
|
.sample_rate = 16000,
|
|
.bits_per_sample = I2S_BITS_PER_SAMPLE_16BIT,
|
|
.channel_format = I2S_CHANNEL_FMT_ONLY_RIGHT,
|
|
.communication_format = I2S_COMM_FORMAT_STAND_I2S,
|
|
.intr_alloc_flags = ESP_INTR_FLAG_LEVEL1,
|
|
.dma_buf_count = 4,
|
|
.dma_buf_len = 256,
|
|
.use_apll = false,
|
|
.tx_desc_auto_clear = false,
|
|
.fixed_mclk = 0,
|
|
.mclk_multiple = I2S_MCLK_MULTIPLE_DEFAULT,
|
|
.bits_per_chan = I2S_BITS_PER_CHAN_DEFAULT,
|
|
};
|
|
|
|
#if SOC_I2S_SUPPORTS_ADC
|
|
if (this->adc_) {
|
|
config.mode = (i2s_mode_t) (config.mode | I2S_MODE_ADC_BUILT_IN);
|
|
i2s_driver_install(this->parent_->get_port(), &config, 0, nullptr);
|
|
|
|
i2s_set_adc_mode(ADC_UNIT_1, this->adc_channel_);
|
|
i2s_adc_enable(this->parent_->get_port());
|
|
} else {
|
|
#endif
|
|
if (this->pdm_)
|
|
config.mode = (i2s_mode_t) (config.mode | I2S_MODE_PDM);
|
|
|
|
i2s_driver_install(this->parent_->get_port(), &config, 0, nullptr);
|
|
|
|
i2s_pin_config_t pin_config = this->parent_->get_pin_config();
|
|
pin_config.data_in_num = this->din_pin_;
|
|
|
|
i2s_set_pin(this->parent_->get_port(), &pin_config);
|
|
#if SOC_I2S_SUPPORTS_ADC
|
|
}
|
|
#endif
|
|
this->state_ = microphone::STATE_RUNNING;
|
|
this->high_freq_.start();
|
|
}
|
|
|
|
void I2SAudioMicrophone::stop() {
|
|
if (this->state_ == microphone::STATE_STOPPED || this->is_failed())
|
|
return;
|
|
if (this->state_ == microphone::STATE_STARTING) {
|
|
this->state_ = microphone::STATE_STOPPED;
|
|
return;
|
|
}
|
|
this->state_ = microphone::STATE_STOPPING;
|
|
}
|
|
|
|
void I2SAudioMicrophone::stop_() {
|
|
i2s_stop(this->parent_->get_port());
|
|
i2s_driver_uninstall(this->parent_->get_port());
|
|
this->parent_->unlock();
|
|
this->state_ = microphone::STATE_STOPPED;
|
|
this->high_freq_.stop();
|
|
}
|
|
|
|
void I2SAudioMicrophone::read_() {
|
|
size_t bytes_read = 0;
|
|
esp_err_t err =
|
|
i2s_read(this->parent_->get_port(), this->buffer_.data(), BUFFER_SIZE, &bytes_read, (100 / portTICK_PERIOD_MS));
|
|
if (err != ESP_OK) {
|
|
ESP_LOGW(TAG, "Error reading from I2S microphone: %s", esp_err_to_name(err));
|
|
this->status_set_warning();
|
|
return;
|
|
}
|
|
|
|
this->status_clear_warning();
|
|
|
|
this->data_callbacks_.call(this->buffer_);
|
|
}
|
|
|
|
void I2SAudioMicrophone::loop() {
|
|
switch (this->state_) {
|
|
case microphone::STATE_STOPPED:
|
|
break;
|
|
case microphone::STATE_STARTING:
|
|
this->start_();
|
|
break;
|
|
case microphone::STATE_RUNNING:
|
|
this->read_();
|
|
break;
|
|
case microphone::STATE_STOPPING:
|
|
this->stop_();
|
|
break;
|
|
}
|
|
}
|
|
|
|
} // namespace i2s_audio
|
|
} // namespace esphome
|
|
|
|
#endif // USE_ESP32
|