加入中文UI

main/audio_codecs/audio_codec.cc

@@ -0,0 +1,150 @@
+#include "audio_codec.h"
+#include "board.h"
+#include "settings.h"
+
+#include <esp_log.h>
+#include <cstring>
+#include <driver/i2s_common.h>
+
+#define TAG "AudioCodec"
+
+AudioCodec::AudioCodec() {
+    audio_event_group_ = xEventGroupCreate();
+}
+
+AudioCodec::~AudioCodec() {
+    if (audio_input_task_ != nullptr) {
+        vTaskDelete(audio_input_task_);
+    }
+    if (audio_output_task_ != nullptr) {
+        vTaskDelete(audio_output_task_);
+    }
+    if (audio_event_group_ != nullptr) {
+        vEventGroupDelete(audio_event_group_);
+    }
+}
+
+void AudioCodec::OnInputData(std::function<void(std::vector<int16_t>&& data)> callback) {
+    on_input_data_ = callback;
+}
+
+void AudioCodec::OutputData(std::vector<int16_t>& data) {
+    std::lock_guard<std::mutex> lock(audio_output_queue_mutex_);
+    audio_output_queue_.emplace_back(std::move(data));
+    audio_output_queue_cv_.notify_one();
+}
+
+IRAM_ATTR bool AudioCodec::on_sent(i2s_chan_handle_t handle, i2s_event_data_t *event, void *user_ctx) {
+    auto audio_codec = (AudioCodec*)user_ctx;
+    xEventGroupSetBits(audio_codec->audio_event_group_, AUDIO_EVENT_OUTPUT_DONE);
+    return false;
+}
+
+void AudioCodec::Start() {
+    Settings settings("audio", false);
+    output_volume_ = settings.GetInt("output_volume", output_volume_);
+
+    // 注册音频输出回调
+    i2s_event_callbacks_t callbacks = {};
+    callbacks.on_sent = on_sent;
+    i2s_channel_register_event_callback(tx_handle_, &callbacks, this);
+
+    ESP_ERROR_CHECK(i2s_channel_enable(tx_handle_));
+    ESP_ERROR_CHECK(i2s_channel_enable(rx_handle_));
+
+    EnableInput(true);
+    EnableOutput(true);
+
+    // 创建音频输入任务
+    if (audio_input_task_ == nullptr) {
+        xTaskCreate([](void* arg) {
+            auto audio_device = (AudioCodec*)arg;
+            audio_device->InputTask();
+        }, "audio_input", 4096 * 2, this, 3, &audio_input_task_);
+    }
+    // 创建音频输出任务
+    if (audio_output_task_ == nullptr) {
+        xTaskCreate([](void* arg) {
+            auto audio_device = (AudioCodec*)arg;
+            audio_device->OutputTask();
+        }, "audio_output", 4096 * 2, this, 3, &audio_output_task_);
+    }
+}
+
+void AudioCodec::InputTask() {
+    int duration = 30;
+    int input_frame_size = input_sample_rate_ / 1000 * duration * input_channels_;
+
+    while (true) {
+        std::vector<int16_t> input_data(input_frame_size);
+        int samples = Read(input_data.data(), input_data.size());
+        if (samples > 0) {
+            if (on_input_data_) {
+                on_input_data_(std::move(input_data));
+            }
+        }
+    }
+}
+
+void AudioCodec::OutputTask() {
+    while (true) {
+        std::unique_lock<std::mutex> lock(audio_output_queue_mutex_);
+        if (!audio_output_queue_cv_.wait_for(lock, std::chrono::seconds(30), [this]() {
+                return !audio_output_queue_.empty();
+            })) {
+            // If timeout, disable output
+            EnableOutput(false);
+            continue;
+        }
+        auto data = std::move(audio_output_queue_.front());
+        audio_output_queue_.pop_front();
+        lock.unlock();
+
+        if (!output_enabled_) {
+            EnableOutput(true);
+        }
+
+        xEventGroupClearBits(audio_event_group_, AUDIO_EVENT_OUTPUT_DONE);
+        Write(data.data(), data.size());
+        audio_output_queue_cv_.notify_all();
+    }
+}
+
+void AudioCodec::WaitForOutputDone() {
+    // Wait for the output queue to be empty and the output is done
+    std::unique_lock<std::mutex> lock(audio_output_queue_mutex_);
+    audio_output_queue_cv_.wait(lock, [this]() {
+        return audio_output_queue_.empty();
+    });
+    lock.unlock();
+    xEventGroupWaitBits(audio_event_group_, AUDIO_EVENT_OUTPUT_DONE, pdFALSE, pdFALSE, portMAX_DELAY);
+}
+
+void AudioCodec::ClearOutputQueue() {
+    std::lock_guard<std::mutex> lock(audio_output_queue_mutex_);
+    audio_output_queue_.clear();
+}
+
+void AudioCodec::SetOutputVolume(int volume) {
+    output_volume_ = volume;
+    ESP_LOGI(TAG, "Set output volume to %d", output_volume_);
+    
+    Settings settings("audio", true);
+    settings.SetInt("output_volume", output_volume_);
+}
+
+void AudioCodec::EnableInput(bool enable) {
+    if (enable == input_enabled_) {
+        return;
+    }
+    input_enabled_ = enable;
+    ESP_LOGI(TAG, "Set input enable to %s", enable ? "true" : "false");
+}
+
+void AudioCodec::EnableOutput(bool enable) {
+    if (enable == output_enabled_) {
+        return;
+    }
+    output_enabled_ = enable;
+    ESP_LOGI(TAG, "Set output enable to %s", enable ? "true" : "false");
+}