first

apps/common/jl_kws/jl_kws_algo.c

@@ -0,0 +1,172 @@
+#include "jl_kws_common.h"
+
+#if TCFG_KWS_VOICE_RECOGNITION_ENABLE
+
+//===================================================//
+//                    算法接口函数                   //
+//===================================================//
+#define JL_KWS_PARAM_FILE 	SDFILE_RES_ROOT_PATH"jl_kws.cfg"
+#define KWS_FRAME_LEN 		320
+void *JL_kws_init(void *ptr, void *fp1, int online_cmvn, float confidence_yes, float confidence_no, int smooth_yes, int smooth_no);
+extern int jl_kws_get_need_buf_len(void);
+//每帧固定320 bytes, 16k采样率, 10ms
+extern int jl_detect_kws(char *data, int len, void *jl_kws1);
+extern void JL_kws_free(void *_jl_det_kws);
+
+#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;
+    void *kws_algo_buf;
+    u32 *frame_buf;
+    void *param_ptr;
+};
+
+static struct jl_kws_algo __kws_algo = {0};
+
+static void kws_algo_local_init(void *buf, void *param_ptr)
+{
+    //===============================//
+    //       	 算法初始化          //
+    //===============================//
+    void *buf_align = (void *)ALIGN(buf, 4);
+    kws_info("file ptr: 0x%x, buf_align: 0x%x", (u32)param_ptr, (u32)buf_align);
+    memset(buf, 0, jl_kws_get_need_buf_len());
+    int cmn_online = 1;
+    //int online_cmvn = 1;
+    int smooth_yes = 25;
+    int smooth_no = 30;
+    float confidence_yes = KWS_YES_THR;
+    float confidence_no = KWS_NO_THR;
+    __kws_algo.kws_handle = JL_kws_init(buf_align, param_ptr, cmn_online, confidence_yes, confidence_no, smooth_yes, smooth_no);
+}
+
+int jl_kws_algo_init(void)
+{
+    FILE *fp = NULL;
+    struct vfs_attr param_attr = {0};
+    u32 param_len = 0;
+    u8 *param_ptr = NULL;
+    u32 need_buf_len = 0 ;
+    void *buf_ptr = NULL;
+    int ret = 0;
+
+    //===============================//
+    //          打开参数文件         //
+    //===============================//
+    fp = fopen(JL_KWS_PARAM_FILE, "r");
+    if (!fp) {
+        return -ENOENT;
+    }
+
+    fget_attrs(fp, &param_attr);
+    param_len = param_attr.fsize;
+    __kws_algo.param_ptr = (void *)param_attr.sclust; //cpu access addr
+
+    fclose(fp);
+    //===============================//
+    //       申请算法需要的buf       //
+    //===============================//
+    need_buf_len = jl_kws_get_need_buf_len();
+    buf_ptr = zalloc(need_buf_len);
+    kws_debug("need_buf_len = 0x%x", need_buf_len);
+    if (!buf_ptr) {
+        return JL_KWS_ERR_ALGO_NO_BUF;
+    }
+
+    __kws_algo.kws_algo_buf = buf_ptr;
+
+    buf_ptr = zalloc(KWS_FRAME_LEN);
+    kws_debug("KWS_FRAME_LEN = 0x%x", KWS_FRAME_LEN);
+    if (!buf_ptr) {
+        free(__kws_algo.kws_algo_buf);
+        __kws_algo.kws_algo_buf = NULL;
+        return JL_KWS_ERR_ALGO_NO_BUF;
+    }
+    __kws_algo.frame_buf = buf_ptr;
+
+    kws_algo_local_init(__kws_algo.kws_algo_buf, __kws_algo.param_ptr);
+
+    return JL_KWS_ERR_NONE;
+}
+
+
+int jl_kws_algo_detect_run(u8 *buf, u32 len)
+{
+    int ret = 0;
+    /* local_irq_disable(); */
+    /* KWS_IO_DEBUG_1(A,4); */
+    ret = jl_detect_kws((char *)buf, len, __kws_algo.kws_handle);
+    /* 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.kws_algo_buf) {
+        free(__kws_algo.kws_algo_buf);
+        __kws_algo.kws_algo_buf = NULL;
+    }
+
+    if (__kws_algo.frame_buf) {
+        free(__kws_algo.frame_buf);
+        __kws_algo.frame_buf = NULL;
+    }
+}
+
+#endif /* #if TCFG_KWS_VOICE_RECOGNITION_ENABLE */