server1/Server/ThreadCalculate/NoiseDetectThread.h

#ifndef NOISEDETECTTHREAD_H_
#define NOISEDETECTTHREAD_H_


#include "BaseCalculateThread.h"
#include "AudioData.h"
#include "GlobalVariable.h"
#include "SystemConfig.h"
#include "RingQueueManualMutex.hpp"

/* 模拟噪音报警 */
#define CREATE_NOISE_WARN false



class CreateWAVThread;
class CreateRecordFileThread ;


/**
    噪音检测线程类
        1、噪音检测线程是调用动态库检测噪音
        2、噪音检测线程参数和对比项无关，是公共的参数，这里没有对比项信息
        3、因为报警信息需要有对比项信息，所以这里有个map，保存对比项传入的信息，根据对比项信息
           开启报警录制，如果这个通道被多个对比项使用，可能会同时录制多个报警文件。
        
        上面说明作废，现在是由对比项线程来开启噪音检测线程，因为报警、录制报警文件，检测时段和对比项信息绑定太深了，
        所以，这个线程所有权归到对比项线程中，因为噪音检测动态库检测速度很快，即使同一个通道多次开启噪音检测线程，
        也不会有太大影响，噪音检测线程会根据对比项，已有的函数没有删除，依旧保留
    
    噪音判断逻辑
        1、每次检测噪音时，都会检测左右声道的噪音，如果有一个声道有噪音，就认为是噪音
        2、假设设置的噪音检测持续次数是10次，设置的百分比是0.8，10次里有8个是噪音，就判断为噪音
        3、噪音预警，如果连续n次都是噪音，就认为是噪音预警
        4、噪音恢复，连续检测次数中低于设置的百分比就认为噪音恢复了
 */
class NoiseDetectThread : public BaseCalculateThread
{

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

    /* 获取通道信息 */
    const OneSoundCardPCMInfo_t& getRoadInfo() const { return m_roadInfo; }
    /* 获取当前噪音结果 */
    bool isNoise() const { return m_isNoise.load(); }
    /* 获取是否噪音预警 */
    bool isNoiseWarning() const { return m_isNoiseWarning.load(); }

    /* 设置是否在检测时间段内 */
    void setInDetectPeriod(bool isInDetectPeriod) { m_isInDetectPeriod.store(isInDetectPeriod); }

private:
    /* 这两个函数私有化，不再使用 */
    /* 开启对比项通道的噪音报警功能 */
    // void startCompareItemNoiseAlarm(const int itemID, const QString strName, const CompareItemRoadInfo_t& compareItemRoadInfo);
    /* 关闭对比项通道的噪音报警功能 */
    // void stopCompareItemNoiseAlarm(const int itemID, const QString strName, const CompareItemRoadInfo_t& compareItemRoadInfo);

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

private:
    /* 调用动态库检测噪音 */
    bool detectNoise();
    /* 保存结果，不再使用 */
    // void saveResult();
    /* 新的保存结果函数，只有一个对比项信息 */
    void saveResultOnlyOneItem();
    /* 结束报警 */
    void endAlarm();

private:
    OneSoundCardPCMInfo_t m_roadInfo;               /* 录音通道编号 */
    std::string m_roadName;                         /* 录音通道名称 */
    CreateWAVThread* m_pThreadWav = nullptr;        /* WAV小文件生成线程指针 */
    CreateRecordFileThread * m_pThreadCreateAlarm = nullptr; /* 生成报警文件的线程 */
    std::atomic_bool m_isInDetectPeriod = true;     /* 是否在检测时间段内 */

    NoiseDetectBaseConfig_t m_baseNoiseDetectConfig;/* 噪音检测的基础配置 */
    NoiseDetectParam_t m_noiseDetectParam;          /* 噪音检测参数，这个是从数据库中读取过来的 */

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

    /* 计算的结果变量 */
    bool m_currentIsNoise = false;                  /* 当前是否检测到噪音 */
    /* 结果 */
    std::atomic_bool m_isNoiseWarning = false;      /* 噪音预警 */
    std::atomic_bool m_isNoise = false;             /* 是否检测到噪音 */
    bool m_isNoiseLast = false;                     /* 上一次的噪音检测结果 */

    /* ------------------------------------ 噪音检测的一些参数 ------------------------------------ */
    

    /* 噪音检测动态库需要的参数，这些是默认参数，实际启动后会从数据库中获取，就是 NoiseDetectParam_t 中的参数 */
    const std::vector<std::string> m_window_params = {"tukey", "0.25"};
    double m_sample_rate = 48000;                   /* 采样率 */
    double m_silence_threshold = -60.0;          /* 静音检测阈值 */
    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点数 */

    /* ------------------------------------ 报警相关变量 ------------------------------------ */
    std::mutex m_mutexAlarm;                    /* 报警信息互斥锁 */
    // std::map<int, AlarmInfo_t> m_mapAlarmInfo;  /* 报警信息，key是对比项ID */
    RingQueueManualMutex<bool> m_ringQueueIsNoise; /* 噪音检测结果环形队列，保存最近的噪音检测结果 */
    AlarmInfo_t m_currentAlarmInfo;             /* 当前报警信息 */
};



#endif // NOISEDETECTTHREAD_H_