misakiotoha/Bionic_sphere
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1. 完成了PWR按键相关的测试，测试通过

Browse Source 
2. 完成了语言识别测试，测试基本通过，后续需要修改从sd卡导入模型以配合ota
This commit is contained in:
Misaki
2025-09-01 00:11:00 +08:00
parent f13bb5a230
commit ce0998c1c6
17 changed files with 980 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Lib/MIC_Driver/MIC_Speech.c
+240
View File
@@ -0,0 +1,240 @@
#include "MIC_Speech.h"
#include "driver/gpio.h"
#include "driver/i2s_std.h"
#include "driver/i2s_tdm.h"
#include "soc/soc_caps.h"
#include "esp_err.h"
#include "esp_log.h"
#include "esp_wn_iface.h"
#include "esp_wn_models.h"
#include "esp_afe_sr_models.h"
#include "esp_mn_iface.h"
#include "esp_mn_models.h"
#define I2S_CHANNEL_NUM 1
static const char *TAG = "App/Speech";
static i2s_chan_handle_t rx_handle = NULL; // I2S rx channel handler
static AppSpeech MIC_Speech;
bool play_Music_Flag = 0;
uint8_t LCD_Backlight_original = 0;
static esp_err_t i2s_init(i2s_port_t i2s_num, uint32_t sample_rate, int channel_format, int bits_per_chan)
{
esp_err_t ret_val = ESP_OK;
i2s_chan_config_t chan_cfg = I2S_CHANNEL_DEFAULT_CONFIG(i2s_num, I2S_ROLE_MASTER);
ret_val |= i2s_new_channel(&chan_cfg, NULL, &rx_handle);
i2s_std_config_t std_cfg = I2S_CONFIG_DEFAULT(16000, I2S_SLOT_MODE_MONO, I2S_DATA_BIT_WIDTH_32BIT);
// std_cfg.slot_cfg.slot_mask = I2S_STD_SLOT_LEFT;
std_cfg.slot_cfg.slot_mask = I2S_STD_SLOT_RIGHT;
// std_cfg.clk_cfg.mclk_multiple = EXAMPLE_MCLK_MULTIPLE; //The default is I2S_MCLK_MULTIPLE_256. If not using 24-bit data width, 256 should be enough
ret_val |= i2s_channel_init_std_mode(rx_handle, &std_cfg);
ret_val |= i2s_channel_enable(rx_handle);
return ret_val;
}
static void feed_handler(AppSpeech *self)
{
esp_afe_sr_data_t *afe_data = self->afe_data;
int audio_chunksize = self->afe_handle->get_feed_chunksize(afe_data);
int nch = self->afe_handle->get_channel_num(afe_data);
(void)nch;
size_t samp_len = audio_chunksize;
size_t samp_len_bytes = samp_len * I2S_CHANNEL_NUM * sizeof(int32_t);
int32_t *i2s_buff = (int32_t *)malloc(samp_len_bytes);
assert(i2s_buff);
size_t bytes_read;
// FILE *fp = fopen("/sdcard/out", "a+");
// if (fp == NULL) ESP_LOGE(TAG,"can not open file\n");
while (true)
{
i2s_channel_read(rx_handle, i2s_buff, samp_len_bytes, &bytes_read, portMAX_DELAY);
for (int i = 0; i < samp_len; ++i)
{
i2s_buff[i] = i2s_buff[i] >> 14; // 32:8 is the significant bit, 8:0 is the low 8 bits, all 0, the AFE input is 16 bits of voice data, the 29:13 bit is to amplify the voice signal.
}
// FatfsComboWrite(i2s_buff, audio_chunksize * I2S_CHANNEL_NUM * sizeof(int16_t), 1, fp);
self->afe_handle->feed(afe_data, (int16_t *)i2s_buff);
}
self->afe_handle->destroy(afe_data);
if (i2s_buff) {
free(i2s_buff);
i2s_buff = NULL;
}
vTaskDelete(NULL);
}
static void detect_hander(AppSpeech *self)
{
esp_afe_sr_data_t *afe_data = self->afe_data;
int afe_chunksize = self->afe_handle->get_fetch_chunksize(afe_data);
#if defined(CONFIG_SR_MN_CN_MULTINET5_RECOGNITION_QUANT8) || defined(CONFIG_SR_MN_CN_MULTINET6_QUANT) || defined(CONFIG_SR_MN_CN_MULTINET6_AC_QUANT)
char *mn_name = esp_srmodel_filter(self->models, ESP_MN_PREFIX, ESP_MN_CHINESE);
#else
char *mn_name = esp_srmodel_filter(self->models, ESP_MN_PREFIX, ESP_MN_ENGLISH);
#endif // CONFIG_IDF_TARGET_ESP32S3
ESP_LOGI(TAG, "multinet:%s\n", mn_name);
esp_mn_iface_t *multinet = esp_mn_handle_from_name(mn_name);
model_iface_data_t *model_data = multinet->create(mn_name, 6000);
esp_mn_commands_update_from_sdkconfig(multinet, model_data); // Add speech commands from sdkconfig
int mu_chunksize = multinet->get_samp_chunksize(model_data);
assert(mu_chunksize == afe_chunksize);
// FILE *fp = fopen("/sdcard/out", "w");
// if (fp == NULL) ESP_LOGE(TAG,"can not open file\n");
//print active speech commands
multinet->print_active_speech_commands(model_data);
ESP_LOGI(TAG, "Ready");
self->detected = false;
while (true)
{
afe_fetch_result_t* res = self->afe_handle->fetch(afe_data);
if (!res || res->ret_value == ESP_FAIL) {
ESP_LOGE(TAG, "fetch error!\n");
break;
}
if (res->wakeup_state == WAKENET_DETECTED) {
ESP_LOGI(TAG, "WAKEWORD DETECTED\n");
multinet->clean(model_data); // clean all status of multinet
LCD_Backlight_original = LCD_Backlight;
} else if (res->wakeup_state == WAKENET_CHANNEL_VERIFIED) {
ESP_LOGI(TAG, "AFE_FETCH_CHANNEL_VERIFIED, channel index: %d\n", res->trigger_channel_id);
ESP_LOGI(TAG, ">>> Say your command <<<");
self->detected = true;
self->afe_handle->disable_wakenet(afe_data);
LCD_Backlight = 35;
}
if (self->detected) {
esp_mn_state_t mn_state = multinet->detect(model_data, res->data);
if (mn_state == ESP_MN_STATE_DETECTING) {
self->command = COMMAND_NOT_DETECTED;
continue;
} else if (mn_state == ESP_MN_STATE_DETECTED) {
esp_mn_results_t *mn_result = multinet->get_results(model_data);
// for (int i = 0; i < mn_result->num; i++) {
// ESP_LOGI(TAG, "TOP %d, command_id: %d, phrase_id: %d, string:%s prob: %f\n",
// i+1, mn_result->command_id[i], mn_result->phrase_id[i], mn_result->string, mn_result->prob[i]);
// }
ESP_LOGI(TAG, "TOP %d, command_id: %d, phrase_id: %d, string:%s prob: %f\n",
1, mn_result->command_id[0], mn_result->phrase_id[0], mn_result->string, mn_result->prob[0]);
switch (mn_result->command_id[0]) {
case 0:
LCD_Backlight = 100;
break;
case 1:
LCD_Backlight = 30;
break;
case 2:
LCD_Backlight = 0;
break;
case 3:
LCD_Backlight = 100;
break;
case 4:
play_Music_Flag = 1;
break;
default: printf("Unknown Command!\r\n"); break;
}
self->command = (command_word_t)mn_result->command_id[0];
// self->afe_handle->enable_wakenet(afe_data);
// self->detected = false;
self->afe_handle->disable_wakenet(afe_data);
self->detected = true;
ESP_LOGI(TAG, ">>> Say your command <<<");
self->command = COMMAND_TIMEOUT;
} else if (mn_state == ESP_MN_STATE_TIMEOUT) {
esp_mn_results_t *mn_result = multinet->get_results(model_data);
ESP_LOGI(TAG, "timeout, string:%s\n", mn_result->string);
self->command = COMMAND_TIMEOUT;
self->afe_handle->enable_wakenet(afe_data);
self->detected = false;
ESP_LOGI(TAG, ">>> Waiting to be waken up <<<");
LCD_Backlight = LCD_Backlight_original;
if(play_Music_Flag){
play_Music_Flag = 0;
if(ACTIVE_TRACK_CNT)
_lv_demo_music_resume();
else
printf("No MP3 file found in SD card!\r\n");
}
}
}
}
if (model_data) {
multinet->destroy(model_data);
model_data = NULL;
}
self->afe_handle->destroy(afe_data);
vTaskDelete(NULL);
}
void MIC_Speech_init()
{
MIC_Speech.afe_handle = &ESP_AFE_SR_HANDLE;
MIC_Speech.detected = false;
MIC_Speech.command = COMMAND_TIMEOUT;
MIC_Speech.models = esp_srmodel_init("model"); // 这边配置为SD卡当中的文件路径
i2s_init(I2S_NUM_1, 16000, 2, 32);
// sd_card_mount("/sdcard");
afe_config_t afe_config = {
.aec_init = true,
.se_init = true,
.vad_init = true,
.wakenet_init = true,
.voice_communication_init = false,
.voice_communication_agc_init = false,
.voice_communication_agc_gain = 15,
.vad_mode = VAD_MODE_3,
.wakenet_model_name = NULL,
.wakenet_model_name_2 = NULL,
.wakenet_mode = DET_MODE_2CH_90,
.afe_mode = SR_MODE_LOW_COST,
.afe_perferred_core = 0,
.afe_perferred_priority = 5,
.afe_ringbuf_size = 50,
.memory_alloc_mode = AFE_MEMORY_ALLOC_MORE_PSRAM,
.afe_linear_gain = 1.0,
.agc_mode = AFE_MN_PEAK_AGC_MODE_2,
.pcm_config = {
.total_ch_num = 3,
.mic_num = 2,
.ref_num = 1,
.sample_rate = 16000,
},
.debug_init = false,
.debug_hook = {{AFE_DEBUG_HOOK_MASE_TASK_IN, NULL}, {AFE_DEBUG_HOOK_FETCH_TASK_IN, NULL}},
};
afe_config.aec_init = false;
afe_config.se_init = false;
afe_config.vad_init = false;
afe_config.afe_ringbuf_size = 10;
afe_config.pcm_config.total_ch_num = 2;
afe_config.pcm_config.mic_num = 1;
afe_config.pcm_config.ref_num = 1;
afe_config.pcm_config.sample_rate = 16000;
afe_config.wakenet_model_name = esp_srmodel_filter(MIC_Speech.models, ESP_WN_PREFIX, NULL);
MIC_Speech.afe_data = MIC_Speech.afe_handle->create_from_config(&afe_config);
xTaskCreatePinnedToCore((TaskFunction_t)feed_handler, "App/SR/Feed", 4 * 1024, &MIC_Speech, 5, NULL, 1);
xTaskCreatePinnedToCore((TaskFunction_t)detect_hander, "App/SR/Detect", 5 * 1024, &MIC_Speech, 5, NULL, 1);
}
Lib/MIC_Driver/MIC_Speech.h
+47
View File
@@ -0,0 +1,47 @@
#pragma once
#include "esp_afe_sr_iface.h"
#include "esp_process_sdkconfig.h"
#include "model_path.h"
#include "ST77916.h"
#include "LVGL_Music.h"
#define I2S_CONFIG_DEFAULT(sample_rate, channel_fmt, bits_per_chan) { \
.clk_cfg = I2S_STD_CLK_DEFAULT_CONFIG(sample_rate), \
.slot_cfg = I2S_STD_PHILIPS_SLOT_DEFAULT_CONFIG(bits_per_chan, channel_fmt), \
.gpio_cfg = { \
.mclk = GPIO_NUM_NC, \
.bclk = GPIO_NUM_15, \
.ws = GPIO_NUM_2, \
.dout = GPIO_NUM_NC, \
.din = GPIO_NUM_39, \
.invert_flags = { \
.mclk_inv = false, \
.bclk_inv = false, \
.ws_inv = false, \
}, \
}, \
}
typedef enum
{
COMMAND_TIMEOUT = -2,
COMMAND_NOT_DETECTED = -1,
COMMAND_ID1 = 0,
COMMAND_ID2 = 1,
COMMAND_ID3 = 2,
COMMAND_ID4 = 3,
COMMAND_ID5 = 4,
COMMAND_ID6 = 5,
} command_word_t;
typedef struct {
const esp_afe_sr_iface_t *afe_handle;
esp_afe_sr_data_t *afe_data;
srmodel_list_t *models;
bool detected;
command_word_t command;
} AppSpeech;
void MIC_Speech_init();
Lib/PWR_Key/PWR_Key.c
+58
View File
@@ -0,0 +1,58 @@
#include "PWR_Key.h"
static uint8_t BAT_State = 0; // 电池/电源状态标志
static uint8_t Device_State = 0; // 设备状态标志
static uint16_t Long_Press = 0; // 按键按下持续时间计数器
void PWR_Loop(void)
{
if(BAT_State){
if(!gpio_get_level(PWR_KEY_Input_PIN)){
if(BAT_State == 2){
Long_Press ++;
if(Long_Press >= Device_Sleep_Time){
if(Long_Press >= Device_Sleep_Time && Long_Press < Device_Restart_Time)
Device_State = 1;
else if(Long_Press >= Device_Restart_Time && Long_Press < Device_Shutdown_Time)
Device_State = 2;
else if(Long_Press >= Device_Shutdown_Time)
Shutdown();
}
}
}
else{
if(BAT_State == 1)
BAT_State = 2;
Long_Press = 0;
}
}
}
void Fall_Asleep(void)
{
}
void Restart(void)
{
}
void Shutdown(void)
{
gpio_set_level(PWR_Control_PIN, false);
LCD_Backlight = 0;
}
void configure_GPIO(int pin, gpio_mode_t Mode)
{
gpio_reset_pin(pin);
gpio_set_direction(pin, Mode);
}
void PWR_Init(void) {
configure_GPIO(PWR_KEY_Input_PIN, GPIO_MODE_INPUT);
configure_GPIO(PWR_Control_PIN, GPIO_MODE_OUTPUT);
gpio_set_level(PWR_Control_PIN, false);
vTaskDelay(100);
if(!gpio_get_level(PWR_KEY_Input_PIN)) {
BAT_State = 1;
gpio_set_level(PWR_Control_PIN, true);
}
}
Lib/PWR_Key/PWR_Key.h
+16
View File
@@ -0,0 +1,16 @@
#pragma once
#include "ST77916.h"
#define PWR_KEY_Input_PIN 6 // 电源按键输入引脚(GPIO6)
#define PWR_Control_PIN 7 // 电源控制输出引脚(GPIO7)
#define Device_Sleep_Time 10 // 进入睡眠模式的按键时长阈值
#define Device_Restart_Time 15 // 重启设备的按键时长阈值
#define Device_Shutdown_Time 20 // 关机操作的按键时长阈值
void Fall_Asleep(void);
void Shutdown(void);
void Restart(void);
void PWR_Init(void);
void PWR_Loop(void);