NoiseReduction.c 2.3 KB
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>

#include "JZsdkLib.h"

// T_JZsdkReturnCode PcmNoiseReduction(unsigned char *data, int len, int threshold) 
// {
//     for (int i = 0; i < len; i++)
//     {
//         if (abs(data) >= 0xFF) 
//         {
//             data[i] = 0;
//         }
//     }
    
     
//     return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
// }


#define NOISE_THRESHOLD 0.05 // 降噪阈值

T_JZsdkReturnCode reduce_noise(float* samples, int num_samples) {
    for (int i = 0; i < num_samples; i++) {
        if (fabs(samples[i]) < NOISE_THRESHOLD) {
            samples[i] = 0.0f; // 如果样本小于阈值,则将其置为零
        }
    }
}

T_JZsdkReturnCode PcmNoiseReduction(short *data, int len) 
{
    float float_buffer[len];

    // 转换为浮点数并进行降噪
    for (size_t i = 0; i < len; i++) {
        float_buffer[i] = data[i] / 32768.0f; // 从16-bit转换到 [-1.0, 1.0]
    }

    reduce_noise(float_buffer, len);

    // 转换回16-bit并写入输出文件
    for (size_t i = 0; i < len; i++) 
    {
        data[i] = (short)(float_buffer[i] * 32767.0f); // 溢出处理
    }

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

// #define NOISE_THRESHOLD 0.05 // 降噪阈值

// // 字节交换函数
// short swap_endian(short val) {
//     return (val >> 8) | (val << 8);
// }

// T_JZsdkReturnCode reduce_noise(float* samples, int num_samples) {
//     for (int i = 0; i < num_samples; i++) {
//         if (fabs(samples[i]) < NOISE_THRESHOLD) {
//             samples[i] = 0.0f; // 如果样本小于阈值,则将其置为零
//         }
//     }
// }

// T_JZsdkReturnCode PcmNoiseReduction(U16_t *data, int len) 
// {
//     float float_buffer[len];

//     // 将大端格式转换为浮点数
//     for (size_t i = 0; i < len; i++) {
//         // 字节交换
//         short sample = swap_endian(data[i]);
//         float_buffer[i] = sample / 32768.0f; // 从16-bit转换到 [-1.0, 1.0]
//     }

//     reduce_noise(float_buffer, len);

//     // 转换回16-bit并写入输出文件
//     for (size_t i = 0; i < len; i++) 
//     {
//         data[i] = (U16_t)(float_buffer[i] * 32767.0f); // 溢出处理
        
//         data[i] = swap_endian(data[i]);
//     }

//     return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
// }