server1/Server/ThreadCalculate/NoiseDetectThread.h

#ifndef NOISEDETECTTHREAD_H_
#define NOISEDETECTTHREAD_H_


#include "BaseCalculateThread.h"
#include "AudioData.h"

class CreateWAVThread;



/**
 * @brief 噪音检测线程类
 *          1、噪音检测线程是调用动态库检测噪音
 *          2、噪音检测线程参数和对比项无关，是公共的参数，这里没有对比项信息
 * 
 */
class NoiseDetectThread : public BaseCalculateThread
{

public:
    NoiseDetectThread(CalculateThreadInfo_t& threadInfo);
    ~NoiseDetectThread() override;

    /* 获取通道信息 */
    const SoundCardRoadInfo_t& getRoadInfo() const { return m_roadInfo; }
    /* 获取噪音结果 */
    bool isNoise() const { return m_isNoise.load(); }

protected:
    /* 线程功能函数 */
    void task() override;
    /* 初始化数据 */
    bool initData() override;
    /* 清理数据 */
    void clearData() override;

private:
    /* 调用动态库检测噪音 */
    bool detectNoise();
    /* 保存结果 */
    void saveResult();


private:
    SoundCardRoadInfo_t m_roadInfo;                 /* 录音通道编号 */
    std::string m_roadName;                         /* 录音通道名称 */
    CreateWAVThread* m_pWAVThread = nullptr;        /* WAV小文件生成线程指针 */

    AudioLeftRightData m_leftRightData;             /* 左右声道数据 */

    std::atomic_bool m_isNoise = false;             /* 是否检测到噪音 */
    bool m_isNoiseLast = false;                     /* 上一次是否检测到噪音 */


    /* 噪音检测的一些参数 */
    const std::vector<std::string> m_window_params = {"tukey", "0.25"};
    double m_sample_rate = 48000;                   /* 采样率 */
    double m_silence_threshold = 3e-3;        /* 静音检测阈值 */
    double m_db_threshold = -70.0;            /* 分贝阈值 */
    double m_cv_threshold = -70.0;            /* 变异系数阈值 */
    int m_nperseg = 256;                      /* 每段样本数 */
    int m_noverlap = 32;                      /* 重叠样本数 */
    int m_nfft = 256;                         /* FFT点数 */
};



#endif // NOISEDETECTTHREAD_H_