113 lines
3.6 KiB
C
113 lines
3.6 KiB
C
/*
|
||
******************************************************************
|
||
* 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*/
|
||
|