forked from xiaozhi/xiaozhi-esp32
64 lines
1.8 KiB
C++
64 lines
1.8 KiB
C++
#ifndef PROTOCOL_H
|
|
#define PROTOCOL_H
|
|
|
|
#include <cJSON.h>
|
|
#include <string>
|
|
#include <functional>
|
|
|
|
struct BinaryProtocol3 {
|
|
uint8_t type;
|
|
uint8_t reserved;
|
|
uint16_t payload_size;
|
|
uint8_t payload[];
|
|
} __attribute__((packed));
|
|
|
|
enum AbortReason {
|
|
kAbortReasonNone,
|
|
kAbortReasonWakeWordDetected
|
|
};
|
|
|
|
enum ListeningMode {
|
|
kListeningModeAutoStop,
|
|
kListeningModeManualStop,
|
|
kListeningModeAlwaysOn // 需要 AEC 支持
|
|
};
|
|
|
|
class Protocol {
|
|
public:
|
|
virtual ~Protocol() = default;
|
|
|
|
inline int server_sample_rate() const {
|
|
return server_sample_rate_;
|
|
}
|
|
|
|
void OnIncomingAudio(std::function<void(const std::string& data)> callback);
|
|
void OnIncomingJson(std::function<void(const cJSON* root)> callback);
|
|
void OnAudioChannelOpened(std::function<void()> callback);
|
|
void OnAudioChannelClosed(std::function<void()> callback);
|
|
void OnNetworkError(std::function<void(const std::string& message)> callback);
|
|
|
|
virtual bool OpenAudioChannel() = 0;
|
|
virtual void CloseAudioChannel() = 0;
|
|
virtual bool IsAudioChannelOpened() const = 0;
|
|
virtual void SendAudio(const std::string& data) = 0;
|
|
virtual void SendWakeWordDetected(const std::string& wake_word);
|
|
virtual void SendStartListening(ListeningMode mode);
|
|
virtual void SendStopListening();
|
|
virtual void SendAbortSpeaking(AbortReason reason);
|
|
|
|
protected:
|
|
std::function<void(const cJSON* root)> on_incoming_json_;
|
|
std::function<void(const std::string& data)> on_incoming_audio_;
|
|
std::function<void()> on_audio_channel_opened_;
|
|
std::function<void()> on_audio_channel_closed_;
|
|
std::function<void(const std::string& message)> on_network_error_;
|
|
|
|
int server_sample_rate_ = 16000;
|
|
std::string session_id_;
|
|
|
|
virtual void SendText(const std::string& text) = 0;
|
|
};
|
|
|
|
#endif // PROTOCOL_H
|
|
|