server1/Server/ThreadCalculate/CalculateDBThread.cpp

#include "CalculateDBThread.h"
#include "GlobalVariable.h"
#include "spdlog.h"

#include "ThreadManager.h"
#include "GlobalInfo.h"
#include "ThreadAlarmManager.h"


CalculateDBThread::CalculateDBThread(CalculateThreadInfo_t& threadInfo)
    : BaseCalculateThread(threadInfo)
{

}

CalculateDBThread::~CalculateDBThread()
{

}

/* 获取结果 */
OneRoadVolume_t CalculateDBThread::getVolumeInfo()
{
    std::lock_guard<std::mutex> lock(m_mutexVolumeInfo);
    return m_roadVolumeInfo;
}

/**
 * @brief   获取最新的结果
 *          结果包括：静音、过载、反相和音量包信息
 * 
 * @param volumeInfo 
 * @return true 结果更新成功
 * @return false 没有更新结果，可能是没有，也可能传入的已经是最新的结果了
 */
bool CalculateDBThread::getlastVolumeInfo(OneRoadVolume_t& volumeInfo)
{
    std::lock_guard<std::mutex> lock(m_mutexVolumeInfo);
    if(m_caculateDBData.ringQueue.isEmpty())
    {
        // SPDLOG_LOGGER_TRACE(m_logger, "{} 获取最新的音量数据，环形队列为空", m_logBase);
        return false; // 环形队列为空，无法获取最新数据
    }
    /* 根据时间判断是否是最新的 */
    if(volumeInfo.dateTime == m_roadVolumeInfo.dateTime)
    {
        return false;
    }
    volumeInfo.isSilence = m_roadVolumeInfo.isSilence;
    volumeInfo.isOverload = m_roadVolumeInfo.isOverload;
    volumeInfo.isReversed = m_roadVolumeInfo.isReversed;
    memcpy(volumeInfo.vecleftDB, m_roadVolumeInfo.vecleftDB, VOLUME_INFO_NUM);
    memcpy(volumeInfo.vecrightDB, m_roadVolumeInfo.vecrightDB, VOLUME_INFO_NUM);

    /* 更新时间 */
    volumeInfo.dateTime = m_roadVolumeInfo.dateTime;

    return true;
}

/* 获取报警数据 */
// AlarmInfo_t CalculateDBThread::getAlarm(EAlarmType alarmType)
// {
//     std::lock_guard<std::mutex> lock(m_mutexVolumeInfo);
//     switch(alarmType)
//     {
//         case EAlarmType::EAT_Silent:
//             {
//                 if(m_alarmSilence.AlarmType != EAlarmType::EAT_None && m_alarmLastSilence.isAlarm == false)
//                 {
//                     /* 报警结束了，返回报警信息，并清空 */
//                     AlarmInfo_t alarmInfo = m_alarmLastSilence;
//                     m_alarmLastSilence = AlarmInfo_t(); // 清空报警信息
//                     return alarmInfo;
//                 }
//                 return m_alarmSilence;
//             }
//         case EAlarmType::EAT_Overload:
//             {
//                 if(m_alarmOverload.AlarmType != EAlarmType::EAT_None && m_alarmLastOverload.isAlarm == false)
//                 {
//                     /* 报警结束了，返回报警信息，并清空 */
//                     AlarmInfo_t alarmInfo = m_alarmLastOverload;
//                     m_alarmLastOverload = AlarmInfo_t(); // 清空报警信息
//                     return alarmInfo;
//                 }
//                 return m_alarmOverload;
//             }
//         case EAlarmType::EAT_Reversed:
//             {
//                 if(m_alarmPhase.AlarmType != EAlarmType::EAT_None && m_alarmLastPhase.isAlarm == false)
//                 {
//                     /* 报警结束了，返回报警信息，并清空 */
//                     AlarmInfo_t alarmInfo = m_alarmLastPhase;
//                     m_alarmLastPhase = AlarmInfo_t(); // 清空报警信息
//                     return alarmInfo;
//                 }
//                 return m_alarmPhase;
//             }
//         default:
//             SPDLOG_LOGGER_ERROR(m_logger, "{} 获取报警数据失败，未知报警类型: {}", m_logBase, static_cast<int>(alarmType));
//             return m_alarmNull; // 返回一个空的报警信息
//     }
// }

/* 线程功能函数 */
void CalculateDBThread::task()
{
    /* 初始化数据 */
    if(!initData())
    {
        return;
    }
    SPDLOG_LOGGER_INFO(m_logger, " ★ {} 开启计算静音、过载和反相等报警的线程 ", m_logBase);
    while(m_isRunning)
    {
        std::this_thread::sleep_for(std::chrono::milliseconds(10));
        /* 判断是否是需要计算的时间段 */

        /* --------------------------------------------------------------------------------
         * 更新最新数据
         * --------------------------------------------------------------------------------*/
        /* 获取最新数据 */
        if(!m_threadCreateDB->getLatestResult(m_caculateDBData.ringQueue))
        {
            // SPDLOG_LOGGER_DEBUG(m_logger, "{} 获取最新的音量数据失败", m_logBase);
            continue; // 没有获取到最新数据，继续等待
        }
        // 获取最新的一秒钟数据
        m_currSecondData = *m_caculateDBData.ringQueue.back();

        /* --------------------------------------------------------------------------------
         * 计算数据
         * --------------------------------------------------------------------------------*/
        calcuDBPhase();
    }

    /* 清理数据 */
    clearData();

    SPDLOG_LOGGER_INFO(m_logger, "★ {} 计算静音、过载和反相等报警的线程结束 ", m_logBase);
}

/* 初始化数据 */
bool CalculateDBThread::initData()
{
    m_logBase = fmt::format("对比项: {}, 通道: {}",
                            m_threadInfo.compareItemInfo.strName.toStdString(),
                            m_threadInfo.compareItemInfo.mapRoad.begin().value().strCompareRoadName.toStdString());
    /* 对于单个录音通道的线程来说，直接取map中的第一个通道 */
    auto begin = m_threadInfo.compareItemInfo.mapRoad.begin();
    m_roadInfo = begin.value(); // 获取第一个通道的信息
    m_roadName = m_roadInfo.strCompareRoadName.toStdString();

    /* 获取线程，设置10秒超时 */
    auto m_startTime = std::chrono::steady_clock::now(); // 记录开始时间
    while(true)
    {
        /* 获取计算音量的线程 */
        if(m_threadCreateDB == nullptr)
        {
            m_threadCreateDB = ThreadMan.getCreateDBThread(m_roadInfo.scRoadInfo.nSoundCardNum, m_roadInfo.scRoadInfo.roadInfo.nRoadNum);
        }
        if(m_threadCreateAlarmFile == nullptr)
        {
            m_threadCreateAlarmFile = ThreadMan.getCreateLongFileThread(m_roadInfo.scRoadInfo.nSoundCardNum, m_roadInfo.scRoadInfo.roadInfo.nRoadNum);
        }
        
        if(m_threadCreateDB != nullptr && m_threadCreateAlarmFile != nullptr)
        {
            break; // 获取到线程，跳出循环
        }
        /* 超过10秒还没有获取到线程，返回失败 */
        if(std::chrono::steady_clock::now() - m_startTime > std::chrono::seconds(10))
        {
            break;
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(10));
        
    }
    if (m_threadCreateDB == nullptr)
    {
        SPDLOG_LOGGER_ERROR(m_logger, "{} 获取创建音量线程失败", m_logBase);
        return false;
    }
    
    /* 初始化参数，这些参数从外面传进来的 */
    m_avgCalculateDBSeconds = GInfo.avgDBCalculateSeconds();
    
    m_numQueueSeconds = GInfo.queueElementCount();
    m_caculateDBData.ringQueue.setQueueCapacity(m_numQueueSeconds);


    /* 设置静音、过载、反相的计算参数 */
    const auto& silentParam = m_threadInfo.compareItemInfo.paramMute;
    m_volumeParam.SetSilentSwitch(silentParam.isEnable);
    m_volumeParam.SetSilentDuration(silentParam.nLen);
    m_volumeParam.SetSilentThreshold(silentParam.threshold.nThreshold);
    m_volumeParam.SetSilentSensitivity(silentParam.nSensitivity);

    const auto& overloadParam = m_threadInfo.compareItemInfo.paramOverload;
    m_volumeParam.SetOverloadSwitch(overloadParam.isEnable);
    m_volumeParam.SetOverloadDuration(overloadParam.nLen);
    m_volumeParam.SetOverloadThreshold(overloadParam.threshold.nThreshold);
    m_volumeParam.SetOverloadSensitivity(overloadParam.nSensitivity);

    const auto& phaseParam = m_threadInfo.compareItemInfo.paramPhase;
    m_volumeParam.SetPhaseSwitch(phaseParam.isEnable);
    m_volumeParam.SetPhaseDuration(phaseParam.nLen);
    m_volumeParam.SetPhaseThreshold(phaseParam.threshold.dThreshold);
    m_volumeParam.SetPhaseSensitivity(phaseParam.nSensitivity);


    return true;
}

/* 清理数据 */
void CalculateDBThread::clearData()
{

}

/* 计算静音过载反相 */
void CalculateDBThread::calcuDBPhase()
{
    /* 有最新数据，进行计算 */
    m_mutexVolumeInfo.lock();
    
    for(int i = 0; i < VOLUME_INFO_NUM; ++i)
    {
        m_roadVolumeInfo.vecleftDB[i] = m_currSecondData.aryLeftDB[i];
        m_roadVolumeInfo.vecrightDB[i] = m_currSecondData.aryRightDB[i];
    }

    /* 计算静音 */
    m_roadVolumeInfo.isSilence = m_caculateDBData.calculateSilent(m_volumeParam, m_avgCalculateDBSeconds, 
                                    m_silentStartPos, m_silentEndPos);
    /* 计算过载 */
    m_roadVolumeInfo.isOverload = m_caculateDBData.calculateOverload(m_volumeParam, m_avgCalculateDBSeconds, 
                                    m_overloadStartPos, m_overloadEndPos, m_bLastOverload);
    /* 计算反相 */
    m_roadVolumeInfo.isReversed = m_caculateDBData.calculatePhase(m_volumeParam, m_avgCalculateDBSeconds, 
                                    m_phaseStartPos, m_phaseEndPos, m_bLastReversed);
    
    /* 处理报警信息 */
    processAlarm();

    /* 更新时间信息 */
    m_roadVolumeInfo.dateTime = m_currSecondData.startTime;

    m_mutexVolumeInfo.unlock();
}

/**
 * @brief 判断是否报警
 *        1、这里报警信息不包含对比项的信息，对比项信息在对比项线程里自己添加
 *        2、报警信息只包含通道编号、通道名称、报警类型等信息
 *        3、静音、过载、反相的判断需要连续多个秒满足才会判断报警，因此报警的开始时间其实
 *           比真正静音的开始时间要晚，差距就是判断报警时长的秒数SilentDuration、OverloadDuration和PhaseDuration
 *           所以需要真正计算出静音开始的时间
 * 
 */
void CalculateDBThread::processAlarm()
{
    /* ----------------------- 处理静音报警 ----------------------- */
    processSilence();
    /* ----------------------- 处理过载报警 ----------------------- */
    processOverload();
    /* ----------------------- 处理反相报警 ----------------------- */
    processPhase();
    
}

/* 处理静音报警 */
void CalculateDBThread::processSilence()
{
    if(m_roadVolumeInfo.isSilence)
    {
        /* 这次是静音，判断静音报警是否已经开启 */
        if(!m_alarmSilence.isAlarm)
        {
            /* 开启静音报警 */
            m_alarmSilence.isAlarm = true;
            m_alarmSilence.strCompareItemName = m_threadInfo.compareItemInfo.strName.toStdString();
            m_alarmSilence.RoadInfo = m_roadInfo; // 设置通道信息
            m_alarmSilence.AlarmType = EAlarmType::EAT_Silent; // 静音报警
            /* 计算开始时间，往前推判断静音需要的时间 */
            QDateTime startTime;
            if(m_silentStartPos > 0)
            {
                /* 上面计算出来的静音开始位置有效，往前推设置的静音时长 */
                startTime = m_caculateDBData.ringQueue.at(m_silentStartPos)->startTime;
                startTime = startTime.addSecs(- m_volumeParam.GetSilentDuration());
            }else 
            {
                startTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetSilentDuration());
            }
            m_alarmSilence.StartTime = startTime; // 静音开始时间
            m_alarmSilence.EndTime = m_currSecondData.endTime;

            /* 开启录音 */
            if(!m_threadCreateAlarmFile->startRecordAlarmFile(m_alarmSilence))
            {
                /* 开启录制报警音频失败 */
                SPDLOG_LOGGER_WARN(m_logger, "{} 开启录制静音报警音频失败", m_logBase);
            }
        }else 
        {
            /* 静音报警已经开启，更新结束时间 */
            m_alarmSilence.EndTime = m_currSecondData.endTime;
        }
    }else 
    {
        /* 这次不是静音，判断静音报警是否已经开启 */
        if(m_alarmSilence.isAlarm)
        {
            /* 静音报警已经开启，结束静音报警 */
            if(m_silentEndPos > 0)
            {
                auto endData = m_caculateDBData.ringQueue.at(m_silentEndPos);
                if(endData != nullptr)
                {
                    m_alarmSilence.EndTime = endData->endTime.addSecs(- m_volumeParam.GetSilentDuration());
                }
            }else 
            {
                m_alarmSilence.EndTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetSilentDuration());
            }
            /* 保存报警信息 */
            // m_alarmLastSilence = m_alarmSilence;
            
            /* 停止录制报警音频 */
            if(!m_threadCreateAlarmFile->stopRecordAlarmFile(m_alarmSilence))
            {
                SPDLOG_LOGGER_WARN(m_logger, "{} 停止录制静音报警音频失败", m_logBase);
            }

            /* 将报警信息发送给写入报警信息的线程 */
            AlarmManager.addAlarmInfo(m_alarmSilence);

            /* 打印日志 */
            SPDLOG_LOGGER_INFO(m_logger, "{} 静音报警结束，开始时间：{}，结束时间：{}",
                m_logBase, 
                m_alarmSilence.StartTime.toString("yyyy-MM-dd hh:mm:ss").toStdString(),
                m_alarmSilence.EndTime.toString("yyyy-MM-dd hh:mm:ss").toStdString());

            /* 清空报警信息 */
            m_alarmSilence = AlarmInfo_t();
        }
    }

    /* 更新结果，实时更新 */
    // if(m_alarmSilence.AlarmType != EAlarmType::EAT_None)
    // {
    //     /* 有报警类型，进一步判断是否有报警 */
    //     if(m_alarmSilence.isAlarm)
    //     {
    //         /* 有报警，更新报警信息，判断一下上次的报警是否被取走 */
    //         if(m_alarmLastSilence.AlarmType == EAlarmType::EAT_None || m_alarmLastSilence.isAlarm == true)
    //         {
    //             /* 报警结束信息被取走了，清空了结果，或者正在报警中，更新信息 */
    //             m_alarmLastSilence = m_alarmSilence;
    //         }
    //     }else 
    //     {
    //         /* 有报警类型无报警，说明是报警结束了 */
    //         m_alarmLastSilence = m_alarmSilence;
    //         m_alarmSilence = AlarmInfo_t(); // 清空报警信息
    //     }
    // }
}

/* 处理过载报警 */
void CalculateDBThread::processOverload()
{
    if(m_roadVolumeInfo.isOverload)
    {
        /* 这次是过载，判断过载报警是否已经开启 */
        if(!m_alarmOverload.isAlarm)
        {
            /* 开启过载报警 */
            m_alarmOverload.isAlarm = true;
            m_alarmOverload.strCompareItemName = m_threadInfo.compareItemInfo.strName.toStdString();
            m_alarmOverload.RoadInfo = m_roadInfo; // 设置通道信息
            m_alarmOverload.AlarmType = EAlarmType::EAT_Overload; // 过载报警
            /* 计算开始时间，往前推设置的过载时长 */
            QDateTime startTime;
            if(m_overloadStartPos > 0)
            {
                /* 上面计算出来的过载开始位置有效，往前推设置的过载时长 */
                startTime = m_caculateDBData.ringQueue.at(m_overloadStartPos)->startTime;
                startTime = startTime.addSecs(- m_volumeParam.GetOverloadDuration());
            }else 
            {
                startTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetOverloadDuration());
            }
            m_alarmOverload.StartTime = startTime; // 过载开始时间
            m_alarmOverload.EndTime = m_currSecondData.endTime;

            /* 开启录音 */
            if(!m_threadCreateAlarmFile->startRecordAlarmFile(m_alarmSilence))
            {
                /* 开启录制报警音频失败 */
                SPDLOG_LOGGER_WARN(m_logger, "{} 开启录制过载报警音频失败", m_logBase);
            }

        }else 
        {
            /* 过载报警已经开启，更新结束时间 */
            m_alarmOverload.EndTime = m_currSecondData.endTime;
        }
    }else 
    {
        /* 这次不是过载，判断过载报警是否已经开启 */
        if(m_alarmOverload.isAlarm)
        {
            /* 过载报警已经开启，结束过载报警 */
            if(m_overloadEndPos > 0)
            {
                auto endData = m_caculateDBData.ringQueue.at(m_overloadEndPos);
                if(endData != nullptr)
                {
                    m_alarmOverload.EndTime = endData->endTime.addSecs(- m_volumeParam.GetOverloadDuration());
                }
            }else 
            {
                m_alarmOverload.EndTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetOverloadDuration());
            }
            /* 保存报警信息 */
            // m_alarmLastOverload = m_alarmOverload;
            
            /* 停止录制报警音频 */
            if(!m_threadCreateAlarmFile->stopRecordAlarmFile(m_alarmOverload))
            {
                SPDLOG_LOGGER_WARN(m_logger, "{} 停止录制过载报警音频失败", m_logBase);
            }

            /* 将报警信息发送给写入报警信息的线程 */
            AlarmManager.addAlarmInfo(m_alarmOverload);


            /* 打印日志 */
            SPDLOG_LOGGER_INFO(m_logger, "{} 过载报警结束，开始时间：{}，结束时间：{}",
                m_logBase, 
                m_alarmOverload.StartTime.toString("yyyy-MM-dd hh:mm:ss").toStdString(),
                m_alarmOverload.EndTime.toString("yyyy-MM-dd hh:mm:ss").toStdString());
            
            /* 清空报警信息 */
            m_alarmOverload = AlarmInfo_t();
        }
    }

    /* 更新结果，实时更新 */
    // if(m_alarmOverload.AlarmType != EAlarmType::EAT_None)
    // {
    //     /* 有报警类型，进一步判断是否有报警 */
    //     if(m_alarmOverload.isAlarm)
    //     {
    //         /* 有报警，更新报警信息，判断一下上次的报警是否被取走 */
    //         if(m_alarmLastOverload.AlarmType == EAlarmType::EAT_None || m_alarmLastOverload.isAlarm == true)
    //         {
    //             /* 报警结束信息被取走了，或者正在报警中，更新信息 */
    //             m_alarmLastOverload = m_alarmOverload;
    //         }
    //     }else 
    //     {
    //         /* 有报警类型无报警，说明是报警结束了 */
    //         m_alarmLastOverload = m_alarmOverload;
    //         m_alarmOverload = AlarmInfo_t(); // 清空报警信息
    //     }
    // }
    
}

/* 处理反相报警 */
void CalculateDBThread::processPhase()
{
    if(m_roadVolumeInfo.isReversed)
    {
        /* 这次是反相，判断反相报警是否已经开启 */
        if(!m_alarmPhase.isAlarm)
        {
            /* 开启反相报警 */
            m_alarmPhase.isAlarm = true;
            m_alarmPhase.strCompareItemName = m_threadInfo.compareItemInfo.strName.toStdString();
            m_alarmPhase.RoadInfo = m_roadInfo; // 设置通道信息
            m_alarmPhase.AlarmType = EAlarmType::EAT_Reversed; // 反相报警
            /* 计算开始时间，往前推设置的反相时长 */
            QDateTime startTime;
            if(m_phaseStartPos > 0)
            {
                /* 上面计算出来的反相开始位置有效，往前推设置的反相时长 */
                startTime = m_caculateDBData.ringQueue.at(m_phaseStartPos)->startTime;
                startTime = startTime.addSecs(- m_volumeParam.GetPhaseDuration());
            }else 
            {
                startTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetPhaseDuration());
            }
            m_alarmPhase.StartTime = startTime; // 反相开始时间
            m_alarmPhase.EndTime = m_currSecondData.endTime;

            /* 开启录音 */
            if(!m_threadCreateAlarmFile->startRecordAlarmFile(m_alarmSilence))
            {
                /* 开启录制报警音频失败 */
                SPDLOG_LOGGER_WARN(m_logger, "{} 开启录制反相报警音频失败", m_logBase);
            }

        }else 
        {
            /* 反相报警已经开启，更新结束时间 */
            m_alarmPhase.EndTime = m_currSecondData.endTime;
        }
    }else 
    {
        /* 这次不是反相，判断反相报警是否已经开启 */
        if(m_alarmPhase.isAlarm)
        {
            /* 反相报警已经开启，结束反相报警 */
            if(m_phaseEndPos > 0)
            {
                auto endData = m_caculateDBData.ringQueue.at(m_phaseEndPos);
                if(endData != nullptr)
                {
                    m_alarmPhase.EndTime = endData->endTime.addSecs(- m_volumeParam.GetPhaseDuration());
                }
            }else 
            {
                m_alarmPhase.EndTime = m_currSecondData.endTime.addSecs(- m_volumeParam.GetPhaseDuration());
            }
            /* 保存报警信息 */
            // m_alarmLastPhase = m_alarmPhase;
            
            /* 停止录制报警音频 */
            if(!m_threadCreateAlarmFile->stopRecordAlarmFile(m_alarmPhase))
            {
                SPDLOG_LOGGER_WARN(m_logger, "{} 停止录制反相报警音频失败", m_logBase);
            }

            /* 将报警信息发送给写入报警信息的线程 */
            AlarmManager.addAlarmInfo(m_alarmPhase);

            /* 打印日志 */
            SPDLOG_LOGGER_INFO(m_logger, "{} 反相报警结束，开始时间：{}，结束时间：{}",
                m_logBase, 
                m_alarmPhase.StartTime.toString("yyyy-MM-dd hh:mm:ss").toStdString(),
                m_alarmPhase.EndTime.toString("yyyy-MM-dd hh:mm:ss").toStdString());
            
            /* 清空报警信息 */
            m_alarmPhase = AlarmInfo_t();
        }
    }

    /* 更新结果，实时更新 */
    // if(m_alarmPhase.AlarmType != EAlarmType::EAT_None)
    // {
    //     /* 有报警类型，进一步判断是否有报警 */
    //     if(m_alarmPhase.isAlarm)
    //     {
    //         /* 有报警，更新报警信息，判断一下上次的报警是否被取走 */
    //         if(m_alarmLastPhase.AlarmType == EAlarmType::EAT_None || m_alarmLastPhase.isAlarm == true)
    //         {
    //             /* 报警结束信息被取走了，或者正在报警中，更新信息 */
    //             m_alarmLastPhase = m_alarmPhase;
    //         }
    //     }else 
    //     {
    //         /* 有报警类型无报警，说明是报警结束了 */
    //         m_alarmLastPhase = m_alarmPhase;
    //         m_alarmPhase = AlarmInfo_t(); // 清空报警信息
    //     }
    // }
}


/* 生成报警文件路径 */
std::string CalculateDBThread::generateAlarmFilePath(const AlarmInfo_t& alarmInfo)
{
    std::string filePath = fmt::format("Alarm_{}_{}_{}.wav", 
        alarmInfo.strCompareItemName, 
        static_cast<int>(alarmInfo.AlarmType), 
        alarmInfo.StartTime.toString("yyyyMMdd_hhmmss").toStdString());
    return filePath;
}