FaceAccess/Record/main.h
2024-09-06 23:06:54 +08:00

107 lines
2.3 KiB
C++

#ifndef __MAIN_H__
#define __MAIN_H__
#include "RkAudio.h"
#include <fstream>
#include <memory>
class SpeexDsp;
class WebRtcAecm;
enum Dsp {
Vqe,
Speex,
AecMobile,
Aec3,
};
constexpr auto DumpPath = "/sdcard/data";
class Task {
public:
virtual void run() = 0;
virtual ~Task() {
}
};
class RecorderTask : public Task {
public:
void run() final;
private:
std::shared_ptr<RkAudio::Input> m_input;
std::shared_ptr<std::ofstream> m_ofs;
};
class PlayerTask : public Task {
public:
void setChannels(int channels);
void setPath(const std::string &path);
void run() final;
protected:
void play();
private:
int m_channels = 2;
std::string m_path;
std::vector<char> m_buffer;
std::shared_ptr<std::ifstream> m_ifs;
std::shared_ptr<RkAudio::Output> m_output;
};
class EchoRecordPrivate;
class EchoRecordTask : public Task {
public:
EchoRecordTask();
~EchoRecordTask();
void setDsp(Dsp dsp);
void setDumpEnabled(bool enabled);
void setChannels(int channels);
void run() final;
private:
int m_channels = 2;
Dsp m_dsp = Vqe;
std::shared_ptr<RkAudio::Output> m_output;
std::shared_ptr<RkAudio::Input> m_input;
std::shared_ptr<SpeexDsp> m_speex;
std::shared_ptr<WebRtcAecm> m_webRtcAecm;
std::vector<uint8_t> m_outBuffer;
std::vector<uint8_t> m_nearendBuffer;
std::vector<uint8_t> m_farendBuffer;
EchoRecordPrivate *m_d = nullptr;
std::shared_ptr<std::ofstream> m_outOfs;
std::shared_ptr<std::ofstream> m_micOfs;
};
class ProcessFileTaskPrivate;
class ProcessFileTask : public Task {
public:
ProcessFileTask();
~ProcessFileTask();
void setDsp(Dsp dsp);
void run() final;
protected:
void process();
private:
ProcessFileTaskPrivate *m_d = nullptr;
Dsp m_dsp = Speex;
std::shared_ptr<SpeexDsp> m_speex;
std::shared_ptr<WebRtcAecm> m_webRtcAecm;
std::shared_ptr<std::ifstream> m_speakerIfs;
std::shared_ptr<std::ifstream> m_micIfs;
std::shared_ptr<std::ofstream> m_ofs;
std::shared_ptr<std::ofstream> m_nsOfs; // 降噪后的文件
std::string m_outFilename;
std::chrono::system_clock::time_point m_begin;
};
Dsp dspFromString(const std::string &dsp);
std::string dspToString(Dsp dsp);
#endif // __MAIN_H__