GlobalFuncThread.cpp 30 KB

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