first

2025-10-29 13:10:02 +08:00
commit 49a07fa419
2284 changed files with 642060 additions and 0 deletions
--- a/apps/common/audio/audio_ns.c
+++ b/apps/common/audio/audio_ns.c
@ -0,0 +1,112 @@
+/*
+ ******************************************************************
+ *				Noise Suppress Module(降噪模块)
+ *Notes:
+ *(1)支持降噪的数据采样率：8k/16k
+ *(2)增加多5k左右代码
+ *(3)内存消耗跟数据采样率有关，具体可通过以下接口查询：
+ *   int ns_mem_size = noise_suppress_mem_query(&ans->ns_para);
+ ******************************************************************
+ */
+
+#include "audio_ns.h"
+
+#ifdef CONFIG_MEDIA_NEW_ENABLE
+
+/*
+*********************************************************************
+*                  NoiseSuppress Process
+* Description: 降噪处理主函数
+* Arguments  : ns	降噪句柄
+*			   in	输入数据
+*			   out	输出数据
+*			   len  输入数据长度
+* Return	 : 降噪输出长度
+* Note(s)    : 由于降噪是固定处理帧长的，所以如果输入数据长度不是降噪
+*			   帧长整数倍，则某一次会输出0长度，即没有输出
+*********************************************************************
+*/
+int audio_ns_run(audio_ns_t *ns, short *in, short *out, u16 len)
+{
+    if (ns == NULL) {
+        return len;
+    }
+#if 0
+    int wlen = cbuf_write(&ns->cbuf, in, len);
+    if (wlen != len) {
+        printf("ns_cbuf full\n");
+    }
+    if (ns->cbuf.data_len >= NS_FRAME_SIZE) {
+        cbuf_read(&ns->cbuf, out, NS_FRAME_SIZE);
+        putchar('.');
+        noise_suppress_run(out, out, NS_FRAME_POINTS);
+        return NS_FRAME_SIZE;
+    } else {
+        return 0;
+    }
+#else
+    //putchar('.');
+    int nOut = noise_suppress_run(in, out, (len / 2));
+    return (nOut << 1);
+#endif
+}
+
+/*
+*********************************************************************
+*                  	Noise Suppress Open
+* Description: 初始化降噪模块
+* Arguments  : sr				数据采样率
+*			   mode				降噪模式(0,1,2:越大越耗资源，效果越好)
+*			   NoiseLevel	 	初始噪声水平(评估初始噪声，加快收敛)
+*			   AggressFactor	降噪强度(越大越强:1~2)
+*			   MinSuppress		降噪最小压制(越小越强:0~1)
+* Return	 : 降噪模块句柄
+* Note(s)    : (1)采样率只支持8k、16k
+*			   (2)如果内存不足，可以考虑wideband强制为0，即只做窄带降噪
+*				  ans-ns_para.wideband = 0;
+*********************************************************************
+*/
+audio_ns_t *audio_ns_open(u16 sr, u8 mode, float NoiseLevel, float AggressFactor, float MinSuppress)
+{
+    audio_ns_t *ans = zalloc(sizeof(audio_ns_t));
+    //cbuf_init(&ans->cbuf, ans->in, sizeof(ans->in));
+
+    ans->ns_para.wideband = (sr == 16000) ? 1 : 0;
+    ans->ns_para.mode = mode;
+    ans->ns_para.NoiseLevel = NoiseLevel;
+    ans->ns_para.AggressFactor = AggressFactor;
+    ans->ns_para.MinSuppress = MinSuppress;
+    printf("ns wideband:%d\n", ans->ns_para.wideband);
+    //int ns_mem_size = noise_suppress_mem_query(&ans->ns_para);
+    //printf("ns mem_size:%d\n", ns_mem_size);
+    noise_suppress_open(&ans->ns_para);
+    float lowcut = -60.f;
+    noise_suppress_config(NS_CMD_LOWCUTTHR, 0, &lowcut);
+    float noise_floor = -60.f;
+    noise_suppress_config(NS_CMD_NOISE_FLOOR, 0, &noise_floor);
+    printf("audio_ns_open ok\n");
+    return ans;
+}
+
+/*
+*********************************************************************
+*                  	Noise Suppress Close
+* Description: 关闭降噪模块
+* Arguments  : ns 降噪模块句柄
+* Return	 : 0 成功 其他 失败
+* Note(s)    : None.
+*********************************************************************
+*/
+int audio_ns_close(audio_ns_t *ns)
+{
+    noise_suppress_close();
+    if (ns) {
+        free(ns);
+        ns = NULL;
+    }
+    printf("esco_ans_close ok\n");
+    return 0;
+}
+
+#endif /*CONFIG_MEDIA_NEW_ENABLE*/
+