GlobalFuncThread.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933
  1. #include "GlobalFuncThread.h"
  2. #include "GlobalConfig.h"
  3. #include "UniversalFunc.h"
  4. #include "spdlog/spdlog.h"
  5. FuncThreadInfo::FuncThreadInfo()
  6. {
  7. ChannelID = -1;
  8. strChannelName = "";
  9. appFunction = AppFunction::APP_NONE;
  10. RunState = RunTimeState::RUN_STATE_NONE;
  11. strFunctionName = "";
  12. // StartTime = QDateTime::currentDateTime();
  13. // EndTime = QDateTime::currentDateTime();
  14. listRoomCamActInfo.clear();
  15. mapCameraName.clear();
  16. }
  17. FuncThreadInfo::~FuncThreadInfo()
  18. {
  19. }
  20. FuncThreadInfo& FuncThreadInfo::operator=(FuncThreadInfo& other)
  21. {
  22. if(this != &other)
  23. {
  24. ChannelID = other.ChannelID;
  25. strChannelName = other.strChannelName;
  26. appFunction = other.appFunction;
  27. RunState = other.RunState;
  28. strFunctionName = other.strFunctionName;
  29. // StartTime = other.StartTime;
  30. // EndTime = other.EndTime;
  31. listRoomCamActInfo = other.listRoomCamActInfo;
  32. mapCameraName = other.mapCameraName;
  33. }
  34. return *this;
  35. }
  36. /**
  37. * @brief 添加算法信息,这里不能直接创建房间,需要先判断有没有已有的房间,因为有的应用功能可能会包含多个房间
  38. * 已经提前被创建过了
  39. *
  40. * @param info
  41. * @return true
  42. * @return false
  43. */
  44. bool FuncThreadInfo::addActionInfo(const ActionInfo& info)
  45. {
  46. /* 根据此类的功能,添加算法信息 */
  47. if(appFunction == AppFunction::APP_NONE)
  48. {
  49. return false;
  50. }
  51. /* 将其添加到对应的房间 */
  52. bool isFind = false;
  53. for(auto& roomInfo : listRoomCamActInfo)
  54. {
  55. if((roomInfo.RoomID == info.RoomID) && (roomInfo.RoomType == info.RoomType))
  56. {
  57. isFind = true;
  58. // roomInfo.mapCameraAction.insert(std::make_pair(info.CameraID, info.ActionID));
  59. roomInfo.addCameraAction(info.CameraID, info.ActionID);
  60. break;
  61. }
  62. }
  63. /* 没找到这个房间,就创建 */
  64. if(!isFind)
  65. {
  66. RoomCamActInfo roomCamActInfo;
  67. roomCamActInfo.RoomID = info.RoomID;
  68. roomCamActInfo.RoomType = info.RoomType;
  69. roomCamActInfo.strRoomName = info.strRoomName;
  70. roomCamActInfo.addCameraAction(info.CameraID, info.ActionID);
  71. listRoomCamActInfo.push_back(roomCamActInfo);
  72. }
  73. return true;
  74. }
  75. /* 清空算法信息 */
  76. void FuncThreadInfo::clearActionList()
  77. {
  78. listRoomCamActInfo.clear();
  79. }
  80. /* 根据房间ID查找房间信息 */
  81. RoomCamActInfo FuncThreadInfo::findRoomInfo(const int RoomID)
  82. {
  83. for(auto& roomInfo : listRoomCamActInfo)
  84. {
  85. if(roomInfo.RoomID == RoomID)
  86. {
  87. return roomInfo;
  88. }
  89. }
  90. RoomCamActInfo roomCamActInfo;
  91. roomCamActInfo.RoomID = -1;
  92. return roomCamActInfo;
  93. }
  94. /* 获取摄像机名称 */
  95. std::string FuncThreadInfo::getCameraName(int CameraID)
  96. {
  97. auto it = mapCameraName.find(CameraID);
  98. if(it != mapCameraName.end())
  99. {
  100. return it->second;
  101. }
  102. return "";
  103. }
  104. /**
  105. * @brief 添加应用信息,一个应用功能在一个频道内只有一个实例
  106. * 这里是添加应用功能和时间段信息
  107. *
  108. * @param func
  109. * @return true
  110. * @return false
  111. */
  112. // bool ListFuncActInfo::addFuncThreadInfo(const AppAndTimeInfo& func)
  113. // {
  114. // if(func.AppType == 0)
  115. // {
  116. // return false;
  117. // }
  118. // /* 解出这条信息里包含几个App,AppType按位计算,总共8个应用信息 */
  119. // for(int i = 0; i < 8; ++i)
  120. // {
  121. // if(func.AppType & 0x01)
  122. // {
  123. // /* 查找有没有这个应用 */
  124. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_OnWork);
  125. // if(pFuncThreadInfo != nullptr)
  126. // {
  127. // /* 更新时间信息 */
  128. // pFuncThreadInfo->StartTime = func.StartTime;
  129. // pFuncThreadInfo->EndTime = func.EndTime;
  130. // continue;
  131. // }
  132. // FuncThreadInfo* fa = new FuncThreadInfo;
  133. // fa->ChannelID = func.ChannelID;
  134. // fa->appFunction = AppFunction::APP_OnWork;
  135. // fa->strFunctionName = "人员在岗识别";
  136. // fa->StartTime = func.StartTime;
  137. // fa->EndTime = func.EndTime;
  138. // listFuncThreadInfo.push_back(fa);
  139. // }
  140. // else if(func.AppType & 0x02)
  141. // {
  142. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Contraband);
  143. // if(pFuncThreadInfo != nullptr)
  144. // {
  145. // /* 更新时间信息 */
  146. // pFuncThreadInfo->StartTime = func.StartTime;
  147. // pFuncThreadInfo->EndTime = func.EndTime;
  148. // continue;
  149. // }
  150. // FuncThreadInfo* fa = new FuncThreadInfo;
  151. // fa->ChannelID = func.ChannelID;
  152. // fa->appFunction = AppFunction::APP_Contraband;
  153. // fa->strFunctionName = "违禁品识别";
  154. // fa->StartTime = func.StartTime;
  155. // fa->EndTime = func.EndTime;
  156. // listFuncThreadInfo.push_back(fa);
  157. // }
  158. // else if (func.AppType & 0x04)
  159. // {
  160. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Illegal);
  161. // if(pFuncThreadInfo != nullptr)
  162. // {
  163. // /* 更新时间信息 */
  164. // pFuncThreadInfo->StartTime = func.StartTime;
  165. // pFuncThreadInfo->EndTime = func.EndTime;
  166. // continue;
  167. // }
  168. // FuncThreadInfo* fa = new FuncThreadInfo;
  169. // fa->ChannelID = func.ChannelID;
  170. // fa->appFunction = AppFunction::APP_Illegal;
  171. // fa->strFunctionName = "非法入侵检测";
  172. // fa->StartTime = func.StartTime;
  173. // fa->EndTime = func.EndTime;
  174. // listFuncThreadInfo.push_back(fa);
  175. // }
  176. // else if (func.AppType & 0x08)
  177. // {
  178. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Fatigue);
  179. // if(pFuncThreadInfo != nullptr)
  180. // {
  181. // /* 更新时间信息 */
  182. // pFuncThreadInfo->StartTime = func.StartTime;
  183. // pFuncThreadInfo->EndTime = func.EndTime;
  184. // continue;
  185. // }
  186. // FuncThreadInfo* fa = new FuncThreadInfo;
  187. // fa->ChannelID = func.ChannelID;
  188. // fa->appFunction = AppFunction::APP_Fatigue;
  189. // fa->strFunctionName = "疲劳检测";
  190. // fa->StartTime = func.StartTime;
  191. // fa->EndTime = func.EndTime;
  192. // listFuncThreadInfo.push_back(fa);
  193. // }
  194. // else if (func.AppType & 0x10)
  195. // {
  196. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Regional);
  197. // if(pFuncThreadInfo != nullptr)
  198. // {
  199. // /* 更新时间信息 */
  200. // pFuncThreadInfo->StartTime = func.StartTime;
  201. // pFuncThreadInfo->EndTime = func.EndTime;
  202. // continue;
  203. // }
  204. // FuncThreadInfo* fa = new FuncThreadInfo;
  205. // fa->ChannelID = func.ChannelID;
  206. // fa->appFunction = AppFunction::APP_Regional;
  207. // fa->strFunctionName = "区域人员检测";
  208. // fa->StartTime = func.StartTime;
  209. // fa->EndTime = func.EndTime;
  210. // listFuncThreadInfo.push_back(fa);
  211. // }
  212. // else if (func.AppType & 0x20)
  213. // {
  214. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Mouse);
  215. // if(pFuncThreadInfo != nullptr)
  216. // {
  217. // /* 更新时间信息 */
  218. // pFuncThreadInfo->StartTime = func.StartTime;
  219. // pFuncThreadInfo->EndTime = func.EndTime;
  220. // continue;
  221. // }
  222. // FuncThreadInfo* fa = new FuncThreadInfo;
  223. // fa->ChannelID = func.ChannelID;
  224. // fa->appFunction = AppFunction::APP_Mouse;
  225. // fa->strFunctionName = "老鼠识别";
  226. // fa->StartTime = func.StartTime;
  227. // fa->EndTime = func.EndTime;
  228. // listFuncThreadInfo.push_back(fa);
  229. // }
  230. // else if (func.AppType & 0x40)
  231. // {
  232. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_PlayPhone);
  233. // if(pFuncThreadInfo != nullptr)
  234. // {
  235. // /* 更新时间信息 */
  236. // pFuncThreadInfo->StartTime = func.StartTime;
  237. // pFuncThreadInfo->EndTime = func.EndTime;
  238. // continue;
  239. // }
  240. // FuncThreadInfo* fa = new FuncThreadInfo;
  241. // fa->ChannelID = func.ChannelID;
  242. // fa->appFunction = AppFunction::APP_PlayPhone;
  243. // fa->strFunctionName = "玩手机识别";
  244. // fa->StartTime = func.StartTime;
  245. // fa->EndTime = func.EndTime;
  246. // listFuncThreadInfo.push_back(fa);
  247. // }
  248. // else if (func.AppType & 0x80)
  249. // {
  250. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_NoMask);
  251. // if(pFuncThreadInfo != nullptr)
  252. // {
  253. // /* 更新时间信息 */
  254. // pFuncThreadInfo->StartTime = func.StartTime;
  255. // pFuncThreadInfo->EndTime = func.EndTime;
  256. // continue;
  257. // }
  258. // FuncThreadInfo* fa = new FuncThreadInfo;
  259. // fa->ChannelID = func.ChannelID;
  260. // fa->appFunction = AppFunction::APP_NoMask;
  261. // fa->strFunctionName = "未戴口罩识别";
  262. // fa->StartTime = func.StartTime;
  263. // fa->EndTime = func.EndTime;
  264. // listFuncThreadInfo.push_back(fa);
  265. // }
  266. // else if (func.AppType & 0x0100)
  267. // {
  268. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_AllDown);
  269. // if(pFuncThreadInfo != nullptr)
  270. // {
  271. // /* 更新时间信息 */
  272. // pFuncThreadInfo->StartTime = func.StartTime;
  273. // pFuncThreadInfo->EndTime = func.EndTime;
  274. // continue;
  275. // }
  276. // FuncThreadInfo* fa = new FuncThreadInfo;
  277. // fa->ChannelID = func.ChannelID;
  278. // fa->appFunction = AppFunction::APP_AllDown;
  279. // fa->strFunctionName = "摔倒识别";
  280. // fa->StartTime = func.StartTime;
  281. // fa->EndTime = func.EndTime;
  282. // listFuncThreadInfo.push_back(fa);
  283. // }
  284. // }
  285. // return true;
  286. // }
  287. /**
  288. * @brief 添加算法信息,根据传进来的算法ID,将其加入到对应的功能中
  289. *
  290. * @param info
  291. * @return true
  292. * @return false
  293. */
  294. bool ListFuncActInfo::addActionInfo(const ActionInfo& info)
  295. {
  296. if(info.ActionID.empty())
  297. {
  298. return false;
  299. }
  300. /* 人脸识别算法(人员在岗识别、非法入侵检测需要) */
  301. if(info.ActionID == GVariable.ActFaceIdentify())
  302. {
  303. /* 人员在岗识别 */
  304. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_OnWork);
  305. if(pFuncThreadInfo != nullptr)
  306. {
  307. pFuncThreadInfo->addActionInfo(info);
  308. }
  309. /* 非法入侵检测 */
  310. pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Illegal);
  311. if(pFuncThreadInfo != nullptr)
  312. {
  313. pFuncThreadInfo->addActionInfo(info);
  314. }
  315. }
  316. /* 人员计数 */
  317. else if (info.ActionID == GVariable.ActPersonCount())
  318. {
  319. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Regional);
  320. if(pFuncThreadInfo != nullptr)
  321. {
  322. pFuncThreadInfo->addActionInfo(info);
  323. }
  324. }
  325. /* 违禁物品 */
  326. else if (info.ActionID == GVariable.ActContraband())
  327. {
  328. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Contraband);
  329. if(pFuncThreadInfo != nullptr)
  330. {
  331. pFuncThreadInfo->addActionInfo(info);
  332. }
  333. }
  334. /* 玩手机 */
  335. else if (info.ActionID == GVariable.ActPlayPhone())
  336. {
  337. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_PlayPhone);
  338. if(pFuncThreadInfo != nullptr)
  339. {
  340. pFuncThreadInfo->addActionInfo(info);
  341. }
  342. }
  343. /* 睡岗识别 */
  344. else if (info.ActionID == GVariable.ActSleep())
  345. {
  346. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Fatigue);
  347. if(pFuncThreadInfo != nullptr)
  348. {
  349. pFuncThreadInfo->addActionInfo(info);
  350. }
  351. }
  352. /* 疲劳检测 */
  353. else if(info.ActionID == GVariable.ActFatigueDetection())
  354. {
  355. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Fatigue);
  356. if(pFuncThreadInfo != nullptr)
  357. {
  358. pFuncThreadInfo->addActionInfo(info);
  359. }
  360. }
  361. /* 动物识别 */
  362. else if (info.ActionID == GVariable.ActAnimalDetect())
  363. {
  364. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Mouse);
  365. if(pFuncThreadInfo != nullptr)
  366. {
  367. pFuncThreadInfo->addActionInfo(info);
  368. }
  369. }
  370. /* 老鼠识别 */
  371. else if (info.ActionID == GVariable.ActMouseDetect())
  372. {
  373. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_Mouse);
  374. if(pFuncThreadInfo != nullptr)
  375. {
  376. pFuncThreadInfo->addActionInfo(info);
  377. }
  378. }
  379. /* 口罩识别 */
  380. else if (info.ActionID == GVariable.ActNoMask())
  381. {
  382. auto pFuncThreadInfo = findAppFunction(info.ChannelID, AppFunction::APP_NoMask);
  383. if(pFuncThreadInfo != nullptr)
  384. {
  385. pFuncThreadInfo->addActionInfo(info);
  386. }
  387. }
  388. else {
  389. SPDLOG_WARN("未知的算法ID: {}", info.ActionID);
  390. return false;
  391. }
  392. return true;
  393. }
  394. /**
  395. * @brief 清空无用的功能信息
  396. * 摄像机和算法信息为空的,或者运行状态为STOP,都会被清理掉
  397. *
  398. */
  399. void ListFuncActInfo::clearNoneFuncThreadInfo()
  400. {
  401. for(auto it0 = listFuncThreadInfo.begin(); it0 != listFuncThreadInfo.end();)
  402. {
  403. if((*it0)->listRoomCamActInfo.empty() || ((*it0)->RunState == RunTimeState::RUN_STATE_STOP))
  404. {
  405. delete *it0;
  406. it0 = listFuncThreadInfo.erase(it0);
  407. }else {
  408. ++it0;
  409. }
  410. }
  411. }
  412. /* 清空算法列表 */
  413. void ListFuncActInfo::clearActionList()
  414. {
  415. for(auto& it0 : listFuncThreadInfo)
  416. {
  417. it0->listRoomCamActInfo.clear();
  418. }
  419. }
  420. /* 查找应用信息 */
  421. bool ListFuncActInfo::findAppFunction(const AppFunction func)
  422. {
  423. for(const auto& it0 : listFuncThreadInfo)
  424. {
  425. if(it0->appFunction == func)
  426. {
  427. return true;
  428. }
  429. }
  430. return false;
  431. }
  432. /* 根据频率和功能查找实例 */
  433. FuncThreadInfo* ListFuncActInfo::findAppFunction(const int ChannelID, const AppFunction func)
  434. {
  435. for(const auto& it0 : listFuncThreadInfo)
  436. {
  437. if( (it0->appFunction == func) && (it0->ChannelID == ChannelID) )
  438. {
  439. return it0;
  440. }
  441. }
  442. return nullptr;
  443. }
  444. /**
  445. * @brief 查找这个应用信息
  446. *
  447. * @param func
  448. * @return FuncThreadInfo*
  449. */
  450. FuncThreadInfo* ListFuncActInfo::findAppFunction(const FuncThreadInfo& func)
  451. {
  452. for(const auto& it0 : listFuncThreadInfo)
  453. {
  454. if(it0->ChannelID == func.ChannelID && it0->appFunction == func.appFunction)
  455. {
  456. return it0;
  457. }
  458. }
  459. return nullptr;
  460. }
  461. /* ====================================================================================
  462. * ************************ GlobalThreadInfo成员函数 *****************************
  463. * ====================================================================================*/
  464. /* 获取线程运行标志位 */
  465. bool GlobalThreadInfo::getRunning() const
  466. {
  467. return m_bRunning;
  468. }
  469. /* 设置线程运行标志位 */
  470. void GlobalThreadInfo::setRunning(bool bRunning)
  471. {
  472. m_bRunning = bRunning;
  473. }
  474. /* 手动给功能块列表加锁 */
  475. void GlobalThreadInfo::lockRunFAI()
  476. {
  477. m_mutexRunFAI.lock();
  478. }
  479. /* 手动解锁 */
  480. void GlobalThreadInfo::unlockRunFAI()
  481. {
  482. m_mutexRunFAI.unlock();
  483. }
  484. /**
  485. * @brief 添加应用信息,一个应用功能在一个频道内只有一个实例
  486. * 这里是添加应用功能,检测时间段在每个功能线程内部自己获取
  487. *
  488. * @param func
  489. * @return true
  490. * @return false
  491. */
  492. bool GlobalThreadInfo::addFuncThreadInfo(const ChannelAppInfo& func)
  493. {
  494. if(func.AppType == 0)
  495. {
  496. return false;
  497. }
  498. /* 转换成应用内使用的枚举值 */
  499. AppFunction appFunc = getAppFunctionID(func.AppType);
  500. /* 解出这条信息里包含几个App,AppType按位计算,总共8个应用信息 */
  501. FuncThreadInfo* fa = new FuncThreadInfo;
  502. fa->ChannelID = func.ChannelID;
  503. fa->appFunction = appFunc;
  504. fa->strChannelName = func.strChannelName;
  505. fa->strFunctionName = func.strAppName;
  506. fa->RunState = RunTimeState::RUN_STATE_NONE;
  507. // bool bFind = false;
  508. auto pFuncThreadInfo = findAppFunction(func.ChannelID, appFunc);
  509. if(pFuncThreadInfo == nullptr)
  510. {
  511. m_listFuncThreadInfo.push_back(fa);
  512. } else
  513. {
  514. /* 删除fa */
  515. delete fa;
  516. fa = nullptr;
  517. }
  518. // if(AppFunction::APP_OnWork == appFunc)
  519. // {
  520. // /* 查找有没有这个应用 */
  521. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_OnWork);
  522. // if(pFuncThreadInfo == nullptr)
  523. // {
  524. // fa->appFunction = appFunc;
  525. // fa->strFunctionName = "人员在岗识别";
  526. // m_listFuncThreadInfo.push_back(fa);
  527. // bFind = true;
  528. // }
  529. // }
  530. // else if(AppFunction::APP_Contraband == appFunc)
  531. // {
  532. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Contraband);
  533. // if(pFuncThreadInfo == nullptr)
  534. // {
  535. // fa->appFunction = appFunc;
  536. // fa->strFunctionName = "违禁品识别";
  537. // m_listFuncThreadInfo.push_back(fa);
  538. // bFind = true;
  539. // }
  540. // }
  541. // else if (AppFunction::APP_Illegal == appFunc)
  542. // {
  543. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Illegal);
  544. // if(pFuncThreadInfo == nullptr)
  545. // {
  546. // fa->appFunction = appFunc;
  547. // fa->strFunctionName = "非法入侵检测";
  548. // m_listFuncThreadInfo.push_back(fa);
  549. // bFind = true;
  550. // }
  551. // }
  552. // else if (AppFunction::APP_Fatigue == appFunc)
  553. // {
  554. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Fatigue);
  555. // if(pFuncThreadInfo != nullptr)
  556. // {
  557. // fa->appFunction = appFunc;
  558. // fa->strFunctionName = "疲劳检测";
  559. // m_listFuncThreadInfo.push_back(fa);
  560. // bFind = true;
  561. // }
  562. // }
  563. // else if (AppFunction::APP_Regional == appFunc)
  564. // {
  565. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Regional);
  566. // if(pFuncThreadInfo != nullptr)
  567. // {
  568. // fa->appFunction = appFunc;
  569. // fa->strFunctionName = "区域人员检测";
  570. // m_listFuncThreadInfo.push_back(fa);
  571. // bFind = true;
  572. // }
  573. // }
  574. // else if (AppFunction::APP_Mouse == appFunc)
  575. // {
  576. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_Mouse);
  577. // if(pFuncThreadInfo != nullptr)
  578. // {
  579. // fa->appFunction = appFunc;
  580. // fa->strFunctionName = "老鼠识别";
  581. // m_listFuncThreadInfo.push_back(fa);
  582. // bFind = true;
  583. // }
  584. // }
  585. // else if (AppFunction::APP_PlayPhone == appFunc)
  586. // {
  587. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_PlayPhone);
  588. // if(pFuncThreadInfo != nullptr)
  589. // {
  590. // fa->appFunction = appFunc;
  591. // fa->strFunctionName = "玩手机识别";
  592. // m_listFuncThreadInfo.push_back(fa);
  593. // bFind = true;
  594. // }
  595. // }
  596. // else if (AppFunction::APP_NoMask == appFunc)
  597. // {
  598. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_NoMask);
  599. // if(pFuncThreadInfo != nullptr)
  600. // {
  601. // fa->appFunction = appFunc;
  602. // fa->strFunctionName = "未戴口罩识别";
  603. // m_listFuncThreadInfo.push_back(fa);
  604. // bFind = true;
  605. // }
  606. // }
  607. // else if (AppFunction::APP_AllDown == appFunc)
  608. // {
  609. // auto pFuncThreadInfo = findAppFunction(func.ChannelID, AppFunction::APP_AllDown);
  610. // if(pFuncThreadInfo != nullptr)
  611. // {
  612. // fa->appFunction = appFunc;
  613. // fa->strFunctionName = "摔倒识别";
  614. // m_listFuncThreadInfo.push_back(fa);
  615. // bFind = true;
  616. // }
  617. // }
  618. // if(!bFind)
  619. // {
  620. // delete fa;
  621. // fa = nullptr;
  622. // }
  623. return true;
  624. }
  625. /**
  626. * @brief 添加算法信息,根据传进来的算法ID,将其加入到对应的功能中
  627. * 根据功能需要,将算法ID加入到对应的功能中
  628. *
  629. * @param info
  630. * @return true
  631. * @return false
  632. */
  633. bool GlobalThreadInfo::addActionInfo(const ActionInfo& info)
  634. {
  635. if(info.ActionID.empty())
  636. {
  637. return false;
  638. }
  639. /* 人脸识别算法(人员在岗识别、非法入侵检测需要) */
  640. if(info.ActionID == GVariable.ActFaceIdentify())
  641. {
  642. /* 人员在岗识别 */
  643. setAppFunction(info, AppFunction::APP_OnWork);
  644. /* 非法入侵检测 */
  645. setAppFunction(info, AppFunction::APP_Illegal);
  646. }
  647. /* 人员计数 */
  648. else if (info.ActionID == GVariable.ActPersonCount())
  649. {
  650. /* 区域人员检测(人员计数?) */
  651. setAppFunction(info, AppFunction::APP_Regional);
  652. /* 非法入侵检测 */
  653. setAppFunction(info, AppFunction::APP_Illegal);
  654. /* 人员在岗识别 */
  655. setAppFunction(info, AppFunction::APP_OnWork);
  656. }
  657. /* 违禁物品 */
  658. else if (info.ActionID == GVariable.ActContraband())
  659. {
  660. setAppFunction(info, AppFunction::APP_Contraband);
  661. }
  662. /* 玩手机 */
  663. else if (info.ActionID == GVariable.ActPlayPhone())
  664. {
  665. setAppFunction(info, AppFunction::APP_PlayPhone);
  666. }
  667. /* 睡岗识别 */
  668. else if (info.ActionID == GVariable.ActSleep())
  669. {
  670. setAppFunction(info, AppFunction::APP_Fatigue);
  671. }
  672. /* 疲劳检测 */
  673. else if(info.ActionID == GVariable.ActFatigueDetection())
  674. {
  675. setAppFunction(info, AppFunction::APP_Fatigue);
  676. }
  677. /* 动物识别 */
  678. else if (info.ActionID == GVariable.ActAnimalDetect())
  679. {
  680. setAppFunction(info, AppFunction::APP_Mouse);
  681. }
  682. /* 老鼠识别 */
  683. else if (info.ActionID == GVariable.ActMouseDetect())
  684. {
  685. setAppFunction(info, AppFunction::APP_Mouse);
  686. }
  687. /* 口罩识别 */
  688. else if (info.ActionID == GVariable.ActNoMask())
  689. {
  690. setAppFunction(info, AppFunction::APP_NoMask);
  691. }
  692. else {
  693. SPDLOG_WARN("未知的算法ID: {}", info.ActionID);
  694. return false;
  695. }
  696. return true;
  697. }
  698. /**
  699. * @brief 清空无用的功能信息
  700. * 摄像机和算法信息为空的,或者运行状态为RUN_STATE_EXITCOMPLET,都会被清理掉
  701. *
  702. */
  703. void GlobalThreadInfo::clearNoneFuncThreadInfo()
  704. {
  705. for(auto it0 = m_listFuncThreadInfo.begin(); it0 != m_listFuncThreadInfo.end();)
  706. {
  707. if((*it0)->listRoomCamActInfo.empty() || ((*it0)->RunState == RunTimeState::RUN_STATE_EXITCOMPLET))
  708. {
  709. delete *it0;
  710. it0 = m_listFuncThreadInfo.erase(it0);
  711. }else {
  712. ++it0;
  713. }
  714. }
  715. }
  716. /* 清空算法列表(直接清空房间信息) */
  717. void GlobalThreadInfo::clearActionList()
  718. {
  719. for(auto& it0 : m_listFuncThreadInfo)
  720. {
  721. // for(auto& it1 : it0->listRoomCamActInfo)
  722. // {
  723. // it1.mapCameraAction.clear();
  724. // }
  725. it0->listRoomCamActInfo.clear();
  726. }
  727. }
  728. /* 设置没有配置摄像机的功能退出 */
  729. void GlobalThreadInfo::setNoneCameraFuncStop()
  730. {
  731. for(auto& it : m_listFuncThreadInfo)
  732. {
  733. if(it->listRoomCamActInfo.empty())
  734. {
  735. if(it->RunState == RunTimeState::RUN_STATE_NONE)
  736. {
  737. SPDLOG_INFO("频道:{:<15} 功能:{:<20}房间信息为空,无需开启线程",it->strChannelName, it->strFunctionName);
  738. }
  739. else if(it->RunState == RunTimeState::RUN_STATE_RUN)
  740. {
  741. SPDLOG_INFO("频道:{:<10}功能:{:<20}房间信息为空,停止线程",it->strChannelName, it->strFunctionName);
  742. }
  743. it->RunState = RunTimeState::RUN_STATE_STOP;
  744. }
  745. }
  746. }
  747. /* 查找应用信息 */
  748. bool GlobalThreadInfo::findAppFunction(const AppFunction func)
  749. {
  750. for(const auto& it0 : m_listFuncThreadInfo)
  751. {
  752. if(it0->appFunction == func)
  753. {
  754. return true;
  755. }
  756. }
  757. return false;
  758. }
  759. /* 根据频率和功能查找实例 */
  760. FuncThreadInfo* GlobalThreadInfo::findAppFunction(const int ChannelID, const AppFunction func)
  761. {
  762. for(const auto& it0 : m_listFuncThreadInfo)
  763. {
  764. if( (it0->appFunction == func) && (it0->ChannelID == ChannelID) )
  765. {
  766. return it0;
  767. }
  768. }
  769. return nullptr;
  770. }
  771. /**
  772. * @brief 查找这个应用信息
  773. *
  774. * @param func
  775. * @return FuncThreadInfo*
  776. */
  777. FuncThreadInfo* GlobalThreadInfo::findAppFunction(const FuncThreadInfo& func)
  778. {
  779. for(const auto& it0 : m_listFuncThreadInfo)
  780. {
  781. if(it0->ChannelID == func.ChannelID && it0->appFunction == func.appFunction)
  782. {
  783. return it0;
  784. }
  785. }
  786. return nullptr;
  787. }
  788. /* 设置应用信息 */
  789. void GlobalThreadInfo::setAppFunction(const ActionInfo& info, const AppFunction func)
  790. {
  791. auto pFuncThreadInfo = findAppFunction(info.ChannelID, func);
  792. if(pFuncThreadInfo != nullptr)
  793. {
  794. pFuncThreadInfo->addActionInfo(info);
  795. if(pFuncThreadInfo->RunState == RunTimeState::RUN_STATE_NONE)
  796. {
  797. /* 线程状态是RUN_STATE_NONE,说明是新创建的,设置成RUN_STATE_INIT,等会会创建线程 */
  798. pFuncThreadInfo->RunState = RunTimeState::RUN_STATE_INIT;
  799. }
  800. }
  801. }
  802. /* 设置线程状态 */
  803. void GlobalThreadInfo::setThreadState(FuncThreadInfo* pInfo, RunTimeState state)
  804. {
  805. std::lock_guard<std::mutex> look(m_mutexRunFAI);
  806. auto p = findAppFunction(*pInfo);
  807. if(p != nullptr)
  808. {
  809. p->RunState = state;
  810. }
  811. }
  812. void GlobalThreadInfo::setThreadState(FuncThreadInfo& pInfo, RunTimeState state)
  813. {
  814. std::lock_guard<std::mutex> look(m_mutexRunFAI);
  815. auto p = findAppFunction(pInfo);
  816. if(p != nullptr)
  817. {
  818. pInfo.RunState = state;
  819. p->RunState = state;
  820. }
  821. }
  822. /**
  823. * @brief 功能线程更新功能信息,这里是从内存中的数组里获取
  824. *
  825. * @param pInfo
  826. * @return true
  827. * @return false
  828. */
  829. bool GlobalThreadInfo::updateFuncInfo(FuncThreadInfo* pInfo)
  830. {
  831. pInfo->clearActionList();
  832. std::lock_guard<std::mutex> look(m_mutexRunFAI);
  833. auto fa = findAppFunction(*pInfo);
  834. if(fa == nullptr)
  835. {
  836. return false;
  837. }
  838. *pInfo = *fa;
  839. return true;
  840. }
  841. bool GlobalThreadInfo::updateFuncInfo(FuncThreadInfo& pInfo)
  842. {
  843. /* 清空算法列表 */
  844. pInfo.clearActionList();
  845. std::lock_guard<std::mutex> look(m_mutexRunFAI);
  846. auto fa = findAppFunction(pInfo);
  847. if(fa == nullptr)
  848. {
  849. return false;
  850. }
  851. pInfo = *fa;
  852. return true;
  853. }
  854. /* 更新频率信息 */
  855. // void GlobalThreadInfo::setChannelInfo(std::map<int, std::string> mapChannelName)
  856. // {
  857. // }
  858. /* 设置摄像机信息 */
  859. void GlobalThreadInfo::setCameraInfo(std::map<int, std::string> mapCameraName)
  860. {
  861. for(auto& it : m_listFuncThreadInfo)
  862. {
  863. it->mapCameraName.clear();
  864. it->mapCameraName = mapCameraName;
  865. }
  866. }