Files
handleaiuinew/HanddleAI/HandleAiUi-new/back/threadPoolManager.cpp
T
2024-05-22 16:47:14 +08:00

240 lines
5.8 KiB
C++

#include <threadPoolManager.h>
#include <FrontendBackendBridge.h>
#include <algorithmThread.h>
/// <summary>
/// 检查目标文件夹是否存在,不存在则创建
/// </summary>
/// <param name="path"></param>
using namespace std;
void createDirectoryIfNotExists_2(const std::string& path) {
// 检查路径是否存在
if (!std::filesystem::exists(path)) {
// 创建路径
std::filesystem::create_directories(path);
//std::cout << "已创建文件夹: " << path << std::endl;
}
else {/*
std::cout << "文件夹已存在: " << path << std::endl;*/
}
}
std::string getCurrentDateTime_2() {
// 获取当前时间点
std::chrono::system_clock::time_point now = std::chrono::system_clock::now();
// 转换为字符串
std::time_t currentTime = std::chrono::system_clock::to_time_t(now);
std::tm* localTime = std::localtime(&currentTime);
std::stringstream ss;
ss << std::put_time(localTime, "%H-%M-%S");
return ss.str();
}
int threadAmount = 0;
int maxThread = 16;
////////////THREAD//Manager////////////////////////
threadManager::threadManager(QObject* parent) : QThread(parent)
{
//sql初始化
sql = new sqlManager;
sql->initPara();
//辅助线程初始化
//oriSaveThread->oriThread = true;
//oriSaveThread->Init(this);
//resSaveThread->Init(this);
threadAmount = bridge.back.json.algoThreads.algoDataMap.size();
for (int i = 0; i < threadAmount; i++)
{
algoThread* aT = new algoThread();
aT->Init(this);
//QObject::connect(aT, &algoThread::subThreadStart, [&]() {
// supportThread->start();
// });
algoThreadList.push_back(aT);
}
}
void threadManager::run()
{
}
void threadManager::addTask(std::shared_ptr <detData> data)
{
//cout << __LINE__ << "insert Data " << data ->timeCode<< endl;
{
algoThreadCls::datastruct _struct = bridge.back.json.algoThreads.getData(data->algoThreadName);
data->algoThreadIndex = _struct.threadIndex;
data->resShowIndex = _struct.showIndex;
data->imgAmountMin = _struct.imgAmountMin;
}
{
CameraListCls::datastruct _struct = bridge.back.json.camera.getData(data->cameraKeyName);
data->oriShowIndex = _struct.showIndex;
}
insertQueueMutex.lock();
//insertQueue.push_back(data);
int threadId = data->algoThreadIndex;
if (threadId == -1)
{
cout << "threadId return -1 ,algoName " << data->algoThreadName.toStdString() << endl;
return;
}
algoThreadList[threadId]->addDetTask(data);
insertQueueMutex.unlock();
//this->start();
}
bool threadManager::addTaskFreeThread(std::shared_ptr <detData> data)
{
#if 0
int index = 0;
for (index = 0; index < threadPoolList.size(); index++)
{
if (threadPoolList[index]->isRunning() == false)
{
threadPoolList[index]->addDetTask(data);
threadPoolList[index]->start();
return true;
}
}
for (index = 0; index < threadPoolList.size(); index++)
{
if (threadPoolList[index]->nextObj.isEmpty() == true)
{
threadPoolList[index]->addDetTask(data);
return true;
}
}
#endif
return false;
}
threadManager::~threadManager()
{
for (auto thread : algoThreadList)
{
thread->m_runFlag = false;
thread->terminate();
while (thread->isRunning() == true);
}
for (auto thread : algoThreadList)
{
delete thread;
}
}
////////////////DET//DATA//////////////////
sncodeCls::sncodeCls()
{
tcpStartPos = bridge.back.json.tcp.startPos;
}
bool sncodeCls::clearSameSnRequestData()
{
std::unordered_map< size_t, size_t >tempMap;
for (int i = snWaittingVec.size() - 1; i >= 0; i--)
{
auto it = tempMap.find(snWaittingVec[i].startpos);
if (it == tempMap.end()) {
tempMap.insert({ snWaittingVec[i].startpos,i });
}
else
{
snWaittingVec.removeAt(i);
return false;
}
}
return true;
}
void sncodeCls::checkSnFromWaittingVec(QByteArray byte)
{
while (clearSameSnRequestData() == false);
for (int i = 0; i < snWaittingVec.size(); i++)
{//第一位是hash,第二位是index
string SN = readSnFromQByteArray(byte, snWaittingVec[i].startpos, false, true, 0);
if (SN.compare(snErrorsnError))//不为空
{
snResList.push_back({ snWaittingVec[i].timecode,SN,snWaittingVec[i].startpos });
while (snResList.size() > 100)
snResList.pop_front();
snWaittingVec.removeAt(i);
}
}
}
//FINSEDIT
string sncodeCls::readSnFromQByteArray(QByteArray byte, int dmPos,
bool highLowConversion, bool clearAfterRead, long long timeCode, int Size)
{
int triggerPos = (dmPos - tcpStartPos) * 2;
if (triggerPos < 0)
{
cout << "Read SN error,trigger is DM:" << dmPos << ",but read start at DM:" << tcpStartPos << endl;
return snErrorsnError;
}
QString str;
if (byte.at(triggerPos + 1) == 0 && byte.at(triggerPos) == 0)
return snErrorsnError;
int snCount = 0;
while (byte.at(snCount + triggerPos) != 0 || byte.at(snCount + triggerPos + 1) != 0)
{
if (byte.at(snCount + triggerPos) == 0)
{
snCount++;
continue;
}
str += QChar(byte.at(snCount + triggerPos));
snCount++;
}
if (highLowConversion == true)
{
snHex(str);
}
if (str.size() < Size && Size != -1)
return snErrorsnError;
//if (clearAfterRead == true)
// //for(int i=0;i< snCount;i++)
// bridge.back.sendFinsMsg(
//QString::fromStdString("writeALL_DM" + to_string(dmPos) + "_" + to_string(snCount) + "_0"));
return str.toStdString();
}
QString sncodeCls::snHex(QString sn)
{
QString resultStr = "";
for (int i = 0; i < sn.size() - 1; i = i + 2)
{
resultStr = resultStr + sn.at(i + 1) + sn.at(i);
}
return resultStr;
}
string sncodeCls::getData(unsigned long long timeCode)
{
for (int i = snResList.size() - 1; i >= 0; i--)
{
if (snResList[i].timecode == timeCode)
return snResList[i].sn;
}
return snErrorsnError;
}
string sncodeCls::waitForSn(unsigned long long timeCode, int timeOut)
{
int cnt = 0;
string sn = getData(timeCode);
if (!sn.compare(snErrorsnError))
{
while (cnt < timeOut)
{
sn = getData(timeCode);
if (sn.compare(snErrorsnError))
{
return sn;
}
cnt += 10;
QThread::msleep(10);
}
}
else
return sn;
return snErrorsnError;
}