forked from xiaozhi/xiaozhi-esp32
83f6f8c703
* Adapt boards to v2 partition tables * fix esp log error * fix display style * reset emotion after download assets * fix compiling * update assets default url * Add user only tools * Add image cache * smaller cache and buffer, more heap * use MAIN_EVENT_CLOCK_TICK to avoid audio glitches * bump to 2.0.0 * fix compiling errors --------- Co-authored-by: Xiaoxia <terrence.huang@tenclass.com>
45 lines
1.3 KiB
C++
45 lines
1.3 KiB
C++
#ifndef AFE_AUDIO_PROCESSOR_H
|
|
#define AFE_AUDIO_PROCESSOR_H
|
|
|
|
#include <esp_afe_sr_models.h>
|
|
#include <freertos/FreeRTOS.h>
|
|
#include <freertos/task.h>
|
|
#include <freertos/event_groups.h>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <functional>
|
|
|
|
#include "audio_processor.h"
|
|
#include "audio_codec.h"
|
|
|
|
class AfeAudioProcessor : public AudioProcessor {
|
|
public:
|
|
AfeAudioProcessor();
|
|
~AfeAudioProcessor();
|
|
|
|
void Initialize(AudioCodec* codec, int frame_duration_ms, srmodel_list_t* models_list) override;
|
|
void Feed(std::vector<int16_t>&& data) override;
|
|
void Start() override;
|
|
void Stop() override;
|
|
bool IsRunning() override;
|
|
void OnOutput(std::function<void(std::vector<int16_t>&& data)> callback) override;
|
|
void OnVadStateChange(std::function<void(bool speaking)> callback) override;
|
|
size_t GetFeedSize() override;
|
|
void EnableDeviceAec(bool enable) override;
|
|
|
|
private:
|
|
EventGroupHandle_t event_group_ = nullptr;
|
|
esp_afe_sr_iface_t* afe_iface_ = nullptr;
|
|
esp_afe_sr_data_t* afe_data_ = nullptr;
|
|
std::function<void(std::vector<int16_t>&& data)> output_callback_;
|
|
std::function<void(bool speaking)> vad_state_change_callback_;
|
|
AudioCodec* codec_ = nullptr;
|
|
int frame_samples_ = 0;
|
|
bool is_speaking_ = false;
|
|
std::vector<int16_t> output_buffer_;
|
|
|
|
void AudioProcessorTask();
|
|
};
|
|
|
|
#endif |