huxi
158 lines
3.6 KiB
C
158 lines
3.6 KiB
C
#include "ai_interaction/ai_text.h"
|
|
#include "mass_data/mass_data.h"
|
|
#include "system/includes.h"
|
|
#include "JL_rcsp_protocol.h"
|
|
|
|
#if TCFG_AI_INTERACTION_ENABLE
|
|
|
|
#pragma pack(1)
|
|
struct _AI_TEXT_PAYLOAD {
|
|
u8 version; // [0,3]:version // [4,7]:0-localText,1-aiText
|
|
u8 vendor; // 0-杰理,1-科大讯飞
|
|
u16 len;
|
|
u8 dat[0];
|
|
};
|
|
struct _AI_TEXT_PAYLOAD_TTS {
|
|
u8 version;
|
|
u16 len;
|
|
u8 dat[0];
|
|
};
|
|
#pragma pack()
|
|
|
|
struct _AI_TEXT {
|
|
struct _AI_TEXT_PAYLOAD_TTS *ai;
|
|
struct _AI_TEXT_PAYLOAD_TTS *local;
|
|
int (* rx_text)(u8 ai_text, u8 *data, int len);
|
|
OS_MUTEX mutex;
|
|
};
|
|
|
|
|
|
#define AI_TEXT_RULE(x) ((x)->version >> 4)
|
|
#define AI_TEXT_RULED(x) ((x)->version >>3)
|
|
static struct _AI_TEXT ai_text;
|
|
|
|
void ai_text_enter_critical(void)
|
|
{
|
|
os_mutex_pend(&ai_text.mutex, 0);
|
|
}
|
|
|
|
void ai_text_exit_critical(void)
|
|
{
|
|
os_mutex_post(&ai_text.mutex);
|
|
}
|
|
|
|
static void ai_text_rx_cb(void *priv, u8 *data, int len)
|
|
{
|
|
u8 is_ai_text = 0;
|
|
struct _AI_TEXT_PAYLOAD ai;
|
|
struct _AI_TEXT_PAYLOAD_TTS *tts;
|
|
memcpy(&ai, data, sizeof(struct _AI_TEXT_PAYLOAD));
|
|
ai.len = READ_BIG_U16(&ai.len);
|
|
if (AI_TEXT_RULE(&ai)) {
|
|
is_ai_text = 1;
|
|
}
|
|
if (AI_TEXT_RULED(&ai)) {
|
|
is_ai_text = 2;
|
|
}
|
|
if (ai_text.rx_text) {
|
|
if (true == ai_text.rx_text(is_ai_text, &data[sizeof(struct _AI_TEXT_PAYLOAD)], ai.len)) {
|
|
// 外面已经处理，内部不再保存数据
|
|
return ;
|
|
}
|
|
}
|
|
|
|
tts = malloc(ai.len + sizeof(struct _AI_TEXT_PAYLOAD_TTS));
|
|
if (!tts) {
|
|
log_e("malloc err \n");
|
|
return ;
|
|
}
|
|
tts->version = 0;
|
|
WRITE_BIG_U16(&tts->len, ai.len);
|
|
memcpy(tts->dat, &data[sizeof(struct _AI_TEXT_PAYLOAD)], ai.len);
|
|
|
|
ai_text_enter_critical();
|
|
if (is_ai_text) {
|
|
if (ai_text.ai) {
|
|
free(ai_text.ai);
|
|
ai_text.ai = NULL;
|
|
}
|
|
ai_text.ai = tts;
|
|
} else {
|
|
if (ai_text.local) {
|
|
free(ai_text.local);
|
|
ai_text.local = NULL;
|
|
}
|
|
ai_text.local = tts;
|
|
}
|
|
ai_text_exit_critical();
|
|
}
|
|
|
|
int ai_text_get_local_text(u8 **buf)
|
|
{
|
|
if (ai_text.local) {
|
|
*buf = ai_text.local->dat;
|
|
return READ_BIG_U16(&ai_text.local->len);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ai_text_get_ai_text(u8 **buf)
|
|
{
|
|
if (ai_text.ai) {
|
|
*buf = ai_text.ai->dat;
|
|
return READ_BIG_U16(&ai_text.ai->len);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ai_text_local_to_tts(void)
|
|
{
|
|
ai_text_enter_critical();
|
|
if (ai_text.local) {
|
|
mass_data_asyn_send(MASS_DATA_TYPE_TTS, (u8 *)ai_text.local, READ_BIG_U16(&ai_text.local->len) + sizeof(struct _AI_TEXT_PAYLOAD_TTS), NULL, NULL);
|
|
ai_text_exit_critical();
|
|
return true;
|
|
}
|
|
ai_text_exit_critical();
|
|
return false;
|
|
}
|
|
|
|
int ai_text_ai_to_tts(void)
|
|
{
|
|
ai_text_enter_critical();
|
|
if (ai_text.ai) {
|
|
mass_data_asyn_send(MASS_DATA_TYPE_TTS, (u8 *)ai_text.ai, READ_BIG_U16(&ai_text.ai->len) + sizeof(struct _AI_TEXT_PAYLOAD_TTS), NULL, NULL);
|
|
ai_text_exit_critical();
|
|
return true;
|
|
}
|
|
ai_text_exit_critical();
|
|
return false;
|
|
}
|
|
|
|
void ai_text_release(void)
|
|
{
|
|
ai_text_enter_critical();
|
|
if (ai_text.ai) {
|
|
free(ai_text.ai);
|
|
ai_text.ai = NULL;
|
|
}
|
|
if (ai_text.local) {
|
|
free(ai_text.local);
|
|
ai_text.local = NULL;
|
|
}
|
|
ai_text_exit_critical();
|
|
}
|
|
|
|
int ai_text_init(int (* rx_text)(u8 ai_text, u8 *data, int len))
|
|
{
|
|
memset(&ai_text, 0, sizeof(struct _AI_TEXT));
|
|
ai_text.rx_text = rx_text;
|
|
os_mutex_create(&ai_text.mutex);
|
|
mass_data_recv_register(MASS_DATA_TYPE_AI_TEXT, NULL, ai_text_rx_cb);
|
|
return true;
|
|
}
|
|
|
|
|
|
#endif
|
|
|