初版

SDK/audio/jl_kws/jl_kws_algo.c

@@ -0,0 +1,172 @@
+#ifdef SUPPORT_MS_EXTENSIONS
+#pragma bss_seg(".jl_kws_algo.data.bss")
+#pragma data_seg(".jl_kws_algo.data")
+#pragma const_seg(".jl_kws_algo.text.const")
+#pragma code_seg(".jl_kws_algo.text")
+#endif
+#include "jl_kws_common.h"
+#include"jlsp_simple_dvad.h"
+
+#if TCFG_KWS_VOICE_RECOGNITION_ENABLE
+
+int CONFIG_YESNO_VAD_ENABLE = 0;
+
+//===================================================//
+//                    算法接口函数                   //
+//===================================================//
+#define JL_KWS_PARAM_FILE 	SDFILE_RES_ROOT_PATH"jl_kws.cfg"
+#define KWS_FRAME_LEN 		320
+//每帧固定320 bytes, 16k采样率, 10ms
+int jlsp_wake_word_yesno_heap_size(int *private_heap_size, int *share_heap_size);
+void *JL_kws_init(char *private_heap_buffer, int private_heap_size, char *share_heap_buffer, int share_heap_size,
+                  float confidence_yes, float confidence_no, int smooth_yes, int smooth_no, int online, int init_noise);
+int jl_detect_kws(void *m, dvad_config_t *cfg, char *pcm, int size);
+void JL_kws_free(void *_jl_det_kws);
+int jl_kws_get_need_buf_len(void);
+
+#define ALIGN(a, b) \
+	({ \
+	 	int m = (u32)(a) & ((b)-1); \
+		int rets = (u32)(a) + (m?((b)-m):0);	 \
+		rets;\
+	})
+
+
+//===================================================//
+//             IO 口DEBUG(测量算法时间)              //
+//===================================================//
+#define     KWS_IO_DEBUG_0(i,x)       //{JL_PORT##i->DIR &= ~BIT(x), JL_PORT##i->OUT &= ~BIT(x);}
+#define     KWS_IO_DEBUG_1(i,x)       //{JL_PORT##i->DIR &= ~BIT(x), JL_PORT##i->OUT |= BIT(x);}
+
+/*
+ *Yes/No词条识别阈值:
+ *(1)越大越难识别，相应的，词条识别匹配度就越高，即越不容易误触发
+ *(2)越小越容易识别，相应的，词条识别匹配度就越低，即越容易误触发
+ */
+#define KWS_YES_THR		0.6f
+#define KWS_NO_THR		0.6f
+
+struct jl_kws_algo {
+    void *kws_handle;
+    u32 *frame_buf;
+    char *private_heap;
+    char *share_heap;
+};
+
+static struct jl_kws_algo __kws_algo = {0};
+
+static void kws_algo_local_init(void)
+{
+    //===============================//
+    //       申请算法需要的buf       //
+    //===============================//
+    int private_heap_size, share_heap_size;
+    jlsp_wake_word_yesno_heap_size(&private_heap_size, &share_heap_size);
+    kws_debug("private_heap_size:%d, share_heap_size:%d\n", private_heap_size, share_heap_size);
+    __kws_algo.private_heap = (char *)zalloc(private_heap_size);
+    __kws_algo.share_heap = (char *)zalloc(share_heap_size);
+
+    //===============================//
+    //       	 算法初始化          //
+    //===============================//
+    int cmn_online = 1;
+    int smooth_yes = 25; //值越大对应的准确率越低，同时误触发也越低
+    int smooth_no = 30; //值越大对应的准确率越低，同时误触发也越低
+    float confidence_yes = KWS_YES_THR;
+    float confidence_no = KWS_NO_THR;
+    int init_noise = (45 << 8) | 50;
+    __kws_algo.kws_handle = JL_kws_init(__kws_algo.private_heap, private_heap_size, __kws_algo.share_heap, share_heap_size,
+                                        confidence_yes, confidence_no, smooth_yes, smooth_no, cmn_online, init_noise);
+}
+
+int jl_kws_algo_init(void)
+{
+    FILE *fp = NULL;
+    void *buf_ptr = NULL;
+
+    buf_ptr = zalloc(KWS_FRAME_LEN);
+    kws_debug("KWS_FRAME_LEN = 0x%x", KWS_FRAME_LEN);
+    if (!buf_ptr) {
+        return JL_KWS_ERR_ALGO_NO_BUF;
+    }
+    __kws_algo.frame_buf = buf_ptr;
+
+    kws_algo_local_init();
+
+    return JL_KWS_ERR_NONE;
+}
+
+
+int jl_kws_algo_detect_run(u8 *buf, u32 len)
+{
+    int ret = 0;
+    dvad_config_t dvad_cfg = {0};
+    dvad_cfg.d2a_frame_con = 100;
+    dvad_cfg.d2a_th_db = 20;
+    dvad_cfg.dvad_gain_id = 10;
+    dvad_cfg.d_frame_con = 100;
+    dvad_cfg.d_high_con_th = 3;
+    dvad_cfg.d_low_con_th = 6;
+    dvad_cfg.d_stride1 = 3;
+    dvad_cfg.d_stride2 = 7;
+    /* local_irq_disable(); */
+    /* KWS_IO_DEBUG_1(A,4); */
+    ret = jl_detect_kws(__kws_algo.kws_handle, &dvad_cfg, (char *)buf, len);
+    /* KWS_IO_DEBUG_0(A,4); */
+    /* local_irq_enable(); */
+
+    if ((ret != KWS_VOICE_EVENT_YES) &&
+        (ret != KWS_VOICE_EVENT_NO)) {
+        ret = KWS_VOICE_EVENT_NONE;
+    }
+
+    return ret;
+}
+
+void *jl_kws_algo_get_frame_buf(u32 *buf_len)
+{
+    if (__kws_algo.frame_buf) {
+        *buf_len = KWS_FRAME_LEN;
+        return __kws_algo.frame_buf;
+    }
+
+    return NULL;
+}
+
+int jl_kws_algo_start(void)
+{
+    return JL_KWS_ERR_NONE;
+}
+
+void jl_kws_algo_stop(void)
+{
+    return;
+}
+
+
+void jl_kws_algo_close(void)
+{
+    kws_info("%s", __func__);
+
+    if (__kws_algo.kws_handle) {
+        JL_kws_free(__kws_algo.kws_handle);
+        __kws_algo.kws_handle = NULL;
+    }
+
+    if (__kws_algo.frame_buf) {
+        free(__kws_algo.frame_buf);
+        __kws_algo.frame_buf = NULL;
+    }
+
+    if (__kws_algo.private_heap) {
+        free(__kws_algo.private_heap);
+        __kws_algo.private_heap = NULL;
+    }
+
+    if (__kws_algo.share_heap) {
+        free(__kws_algo.share_heap);
+        __kws_algo.share_heap = NULL;
+    }
+}
+
+#endif /* #if TCFG_KWS_VOICE_RECOGNITION_ENABLE */