SPAServer.cpp 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942
  1. #include "SPAServer.h"
  2. #include "spdlog/spdlog.h"
  3. // #include "CurlHttp.h"
  4. #include <filesystem>
  5. // #include "CurlFtp.h"
  6. #include "ThreadPool/ThreadPool.h"
  7. #include <QVector>
  8. SPAServer::SPAServer()
  9. {
  10. m_logger = spdlog::get("SPAServer");
  11. if(m_logger == nullptr)
  12. {
  13. SPDLOG_ERROR("APAServer logger is nullptr");
  14. return;
  15. }
  16. m_threadRunning = true;
  17. /* 初始化WebAPI */
  18. m_toEQMDataBase.initWebApi("http://192.1.3.133:31000/v6/", "", "4c2f9fc91c22dd98331e47af2e2964f4");
  19. /* 模拟违禁品算法ID,后续需要动态调整 */
  20. m_keyContraband = "OD210_026_005246_001-IZRTKyEx";
  21. }
  22. SPAServer::~SPAServer()
  23. {
  24. }
  25. /* 启动服务 */
  26. void SPAServer::startServer()
  27. {
  28. /* 添加获取基础信息的线程 */
  29. // CPPTP.add_task(&SPAServer::threadFromSuperBrain, this);
  30. /* 测试Redis读取并解析数据线程 */
  31. CameraThreadInfo info;
  32. info.RedisIP = "172.16.36.80";
  33. info.RedisPort = 32222;
  34. info.RedisPWD = "Ff1z@TOFr^iwd%Ra";
  35. info.DeviceID = 117;
  36. info.vecAction.push_back("OD210_026_005246_001-IZRTKyEx");
  37. m_keyContraband = "OD210_026_005246_001-IZRTKyEx";
  38. // CPPTP.add_task(&SPAServer::threadFromRedis, this, info);
  39. threadFromRedis(info);
  40. }
  41. /**
  42. * @brief 从基础平台获取算法信息和设备信息的线程函数
  43. *
  44. */
  45. void SPAServer::threadFromSuperBrain()
  46. {
  47. SPDLOG_LOGGER_INFO(m_logger, "开启 fromSuperBrainThread 线程");
  48. /* 创建变量 */
  49. std::vector<AlgorithmInfo> vecAlgNewInfo;
  50. std::vector<DeviceInfo> vecDevNewInfo;
  51. /* 获取一次token,后续失效了再获取 */
  52. m_fromSuperBrain.getToken();
  53. while (m_threadRunning)
  54. {
  55. SPDLOG_LOGGER_INFO(m_logger, "刷新算法和设备信息");
  56. /* 先更新数据库的信息,防止从其他地方更改了数据库,这里没有刷新本地缓存 */
  57. m_toEQMDataBase.getAlgorithmInfo(m_vecEqmAlgInfo);
  58. m_toEQMDataBase.getDeviceInfo(m_vecEqmDevInfo);
  59. m_toEQMDataBase.getDeviceAlgorithmInfo(m_vecEqmDevInfo, m_listDevIDDelete);
  60. /* 从超脑获取基础信息 */
  61. m_fromSuperBrain.getTaskTypeList(vecAlgNewInfo);
  62. m_fromSuperBrain.getDeviceList(vecDevNewInfo);
  63. /* 处理算法信息 */
  64. bool algIsUpdate = processAlgorithmInfo(vecAlgNewInfo);
  65. /* 处理设备信息 */
  66. bool devIsUpdate = processDeviceInfo(vecDevNewInfo);
  67. vecAlgNewInfo.clear();
  68. vecDevNewInfo.clear();
  69. /* 更新算法详细信息 */
  70. m_mutexActionInfo.lock();
  71. m_toEQMDataBase.getActionInfo(m_listActionInfo);
  72. m_mutexActionInfo.unlock();
  73. std::this_thread::sleep_for(std::chrono::seconds(10));
  74. }
  75. }
  76. /* 处理算法信息,返回值为true,说明有改变,需要重新读取 */
  77. bool SPAServer::processAlgorithmInfo(std::vector<AlgorithmInfo> vecNewAlgInfo)
  78. {
  79. std::vector<AlgorithmInfo> vecAlgUpdate;
  80. std::vector<AlgorithmInfo> vecAlgDelete;
  81. /* 对比数据库表格信息,这里只有插入和删除,没有更新 */
  82. compareAlgorithmInfo(vecNewAlgInfo, vecAlgUpdate, vecAlgDelete);
  83. /* 更新数据库,先删除,再写入刷新 */
  84. bool isUpdate = false;
  85. if(vecAlgDelete.size() > 0)
  86. {
  87. SPDLOG_LOGGER_DEBUG(m_logger, "删除算法信息");
  88. m_toEQMDataBase.deleteAlgorithmInfo(vecAlgDelete);
  89. isUpdate = true;
  90. }
  91. if(vecAlgUpdate.size() > 0)
  92. {
  93. SPDLOG_LOGGER_DEBUG(m_logger, "写入算法信息");
  94. m_toEQMDataBase.writeAlgorithmInfo(vecAlgUpdate);
  95. isUpdate = true;
  96. }
  97. return isUpdate;
  98. }
  99. /**
  100. * @brief 处理设备信息
  101. *
  102. * @param vecNewDevInfo 传入新获取到的值
  103. * @return true 需要重新读取数据库,获取新的数据
  104. * @return false 无需读取数据库
  105. */
  106. bool SPAServer::processDeviceInfo(std::vector<DeviceInfo> vecNewDevInfo)
  107. {
  108. std::vector<DeviceInfo> vecDevInsert;
  109. std::vector<DeviceInfo> vecDevUpdate;
  110. std::vector<DeviceInfo> vecDevDelete;
  111. /*-------------------------------------------------------------------------
  112. ****** 这里只对比设备信息,不对比设备的算法信息,算法信息在下面单独对比 *******
  113. *------------------------------------------------------------------------*/
  114. /* 如果本地缓存没有数据,那么就全部插入 */
  115. if(m_vecEqmDevInfo.size() > 0)
  116. {
  117. for(auto& DevInfo : vecNewDevInfo)
  118. {
  119. bool isExist = false;
  120. for(auto& it0 : m_vecEqmDevInfo)
  121. {
  122. if(DevInfo.DeviceID == it0.DeviceID)
  123. {
  124. isExist = true;
  125. /* 对比其他项是否相等,不相等就更新 */
  126. if(DevInfo == it0)
  127. {
  128. continue;
  129. }else {
  130. vecDevUpdate.push_back(DevInfo);
  131. }
  132. break;
  133. }
  134. }
  135. if(!isExist)
  136. {
  137. vecDevInsert.push_back(DevInfo);
  138. }
  139. }
  140. }else {
  141. vecDevInsert = vecNewDevInfo;
  142. }
  143. /* 获取删除列表 */
  144. if(vecNewDevInfo.size() > 0)
  145. {
  146. bool isExist = false;
  147. for(const auto& it : m_vecEqmDevInfo)
  148. {
  149. isExist = false;
  150. for(const auto& it0 : vecNewDevInfo)
  151. {
  152. if(it.DeviceID == it0.DeviceID)
  153. {
  154. isExist = true;
  155. break;
  156. }
  157. }
  158. if(!isExist)
  159. {
  160. vecDevDelete.push_back(it);
  161. }
  162. }
  163. }else {
  164. vecDevDelete = m_vecEqmDevInfo;
  165. }
  166. bool isUpdate = false;
  167. /* 先删除多余的数据 */
  168. if(vecDevDelete.size() > 0)
  169. {
  170. SPDLOG_LOGGER_DEBUG(m_logger, "删除设备信息");
  171. m_toEQMDataBase.deleteDeviceInfo(vecDevDelete);
  172. isUpdate = true;
  173. }
  174. /* 更新数据 */
  175. if(vecDevUpdate.size() > 0)
  176. {
  177. SPDLOG_LOGGER_DEBUG(m_logger, "更新设备信息");
  178. m_toEQMDataBase.updateDeviceInfo(vecDevUpdate);
  179. isUpdate = true;
  180. }
  181. /* 插入数据 */
  182. if(vecDevInsert.size() > 0)
  183. {
  184. SPDLOG_LOGGER_DEBUG(m_logger, "插入设备信息");
  185. m_toEQMDataBase.insertDeviceInfo(vecDevInsert);
  186. isUpdate = true;
  187. }
  188. /*-------------------------------------------------------------------------
  189. ************* 处理设备和算子关联的表格,单独对比设备的算法信息 *************
  190. *------------------------------------------------------------------------*/
  191. /* 插入新的设备信息 */
  192. if(vecDevInsert.size() > 0)
  193. {
  194. SPDLOG_LOGGER_DEBUG(m_logger, "插入设备和算法关联表");
  195. m_toEQMDataBase.insertDeviceAlgorithmInfo(vecDevInsert);
  196. isUpdate = true;
  197. }
  198. vecDevUpdate.clear();
  199. /* 对比现有的设备是否需要更新算法 */
  200. compareDeviceAlgorithmInfo(vecNewDevInfo, vecDevUpdate);
  201. if(vecDevUpdate.size() > 0)
  202. {
  203. SPDLOG_LOGGER_DEBUG(m_logger, "更新设备和算法关联表, 更新设备数目:{}", vecDevUpdate.size());
  204. m_toEQMDataBase.updateDeviceAlgorithmInfo(vecDevUpdate);
  205. }
  206. /* 删除tActionCamer表中消失的设备信息 */
  207. if(m_listDevIDDelete.size() > 0)
  208. {
  209. SPDLOG_LOGGER_DEBUG(m_logger, "删除消失的设备关联的算法");
  210. m_toEQMDataBase.deleteDeviceAlgorithmInfo(m_listDevIDDelete);
  211. isUpdate = true;
  212. }
  213. return isUpdate;
  214. }
  215. /* 对比现有的数据和新获取到的数据,取出要删除和添加的数据 */
  216. void SPAServer::compareAlgorithmInfo(const std::vector<AlgorithmInfo>& vecNewInfo, std::vector<AlgorithmInfo>& vecAlgUpdate, std::vector<AlgorithmInfo>& vecAlgDelete)
  217. {
  218. /* 取出要添加的,如果本地缓存是0,那么全部都要添加 */
  219. if(m_vecEqmAlgInfo.size() > 0)
  220. {
  221. for(const auto& it : vecNewInfo)
  222. {
  223. bool isExist = false;
  224. for(const auto& it0 : m_vecEqmAlgInfo)
  225. {
  226. /* 如果存在就退出循环 */
  227. if(it.ActionID == it0.ActionID)
  228. {
  229. isExist = true;
  230. break;
  231. }
  232. }
  233. if(!isExist)
  234. {
  235. vecAlgUpdate.push_back(it);
  236. }
  237. }
  238. }else {
  239. vecAlgUpdate = vecNewInfo;
  240. }
  241. /* 取出要删除的,如果新的数据是0,那么全部都要删除 */
  242. if(vecNewInfo.size() > 0)
  243. {
  244. bool isExist = false;
  245. for(const auto& it : m_vecEqmAlgInfo)
  246. {
  247. isExist = false;
  248. for(const auto& it0 : vecNewInfo)
  249. {
  250. if(it.ActionID == it0.ActionID)
  251. {
  252. isExist = true;
  253. break;
  254. }
  255. }
  256. if(!isExist)
  257. {
  258. vecAlgDelete.push_back(it);
  259. }
  260. }
  261. }else {
  262. vecAlgDelete = m_vecEqmAlgInfo;
  263. }
  264. }
  265. /**
  266. * @brief 对比设备和算法关联表是否需要更新
  267. * 对比规则:
  268. * 1、这里只对比已有的设备ID,需要删除的ID在获取到tActionCamer表是就已经取出来了
  269. * 2、如果设备ID相等,那么进一步对比算法信息是否相等
  270. * 3、如果设备ID相等,但是算法信息数目不相等,那么直接加入更新列表
  271. * 4、如果设备ID相等,算法信息数目相等,进一步对比算法信息
  272. *
  273. * @param vecNewInfo
  274. * @param vecDevUpdate
  275. */
  276. void SPAServer::compareDeviceAlgorithmInfo(const std::vector<DeviceInfo>& vecNewInfo, std::vector<DeviceInfo>& vecDevUpdate)
  277. {
  278. vecDevUpdate.clear();
  279. for(const auto& it0 : vecNewInfo)
  280. {
  281. for(const auto& it1 : m_vecEqmDevInfo)
  282. {
  283. if(it0.DeviceID == it1.DeviceID)
  284. {
  285. /* 设备的算法信息数目不相等,直接加入更新列表 */
  286. if(it0.vecAlgorithmInfo.size() != it1.vecAlgorithmInfo.size())
  287. {
  288. vecDevUpdate.push_back(it0);
  289. break;
  290. }
  291. /* 设备的算法信息数目相等,进一步对比算法信息 */
  292. bool isEquality = true;
  293. for(const auto& it2 : it0.vecAlgorithmInfo)
  294. {
  295. bool isEq2 = false;
  296. for(const auto& it3 : it1.vecAlgorithmInfo)
  297. {
  298. /* 这里只对比算法ID */
  299. if(it2.ActionID != it3.ActionID)
  300. {
  301. continue;
  302. }else {
  303. isEq2 = true;
  304. break;
  305. }
  306. }
  307. if(!isEq2)
  308. {
  309. isEquality = false;
  310. break;
  311. }
  312. }
  313. if(!isEquality)
  314. {
  315. vecDevUpdate.push_back(it0);
  316. break;
  317. }
  318. }
  319. }
  320. }
  321. }
  322. /**
  323. * @brief 从Redis获取数据线程函数,这个是摄像机线程
  324. * 一个设备一个线程,这个线程的相关变量只在这个线程中使用
  325. *
  326. * @param info
  327. */
  328. void SPAServer::threadFromRedis(const CameraThreadInfo& info)
  329. {
  330. SPDLOG_LOGGER_INFO(m_logger, "开启 fromRedisThread 线程,设备ID:{}", info.DeviceID);
  331. FromRedis fromRedis;
  332. fromRedis.setRedisIPAndPort(info.RedisIP, info.RedisPort);
  333. fromRedis.setRedisPassword(info.RedisPWD);
  334. if(fromRedis.connectRedis())
  335. {
  336. SPDLOG_LOGGER_INFO(m_logger, "连接Redis成功");
  337. }else {
  338. SPDLOG_LOGGER_ERROR(m_logger, "连接Redis失败");
  339. return;
  340. }
  341. CameraThreadInfo threadInfo = info;
  342. // std::string strKey = "117:OD210_026_005246_001-IZRTKyEx";
  343. /* 取出该设备的Key */
  344. std::vector<std::string> vecKey;
  345. for(const auto& it : info.vecAction)
  346. {
  347. std::string strKey = std::to_string(info.DeviceID) + ":" + it;
  348. vecKey.push_back(strKey);
  349. }
  350. std::string strRetValue;
  351. /* 进入线程循环 */
  352. while (m_threadRunning)
  353. {
  354. /* 循环读取这个设备关联的算法信息 */
  355. for(const auto& it : vecKey)
  356. {
  357. SPDLOG_LOGGER_INFO(m_logger, "读取Redis信息, Key: {}", it);
  358. fromRedis.getRedisString(it, strRetValue);
  359. SPDLOG_LOGGER_TRACE(m_logger, "Redis Value:\n{}", strRetValue);
  360. /* 解析数据 */
  361. AlarmInfo alarmInfo;
  362. alarmInfo.ActionID = it.substr(it.find(":") + 1);
  363. /* 解析数据 */
  364. parseRedisData(strRetValue, alarmInfo);
  365. /* 信息时间有的效性判断,主要是记录Redis数据的有效性,是否长时间没有变化,排查错误时用的
  366. * 如果数据长时间数据不变,那么超脑那里就挂了,写入日志 */
  367. isEventTimeVaild(alarmInfo.EventTime);
  368. /* 是否需要写入EQM数据库判断
  369. * 人员计数和区域人员检测需要多次判断,其他的直接写入即可 */
  370. }
  371. std::this_thread::sleep_for(std::chrono::seconds(2));
  372. }
  373. return;
  374. }
  375. /* 解析Redis基础数据 */
  376. void SPAServer::parseRedisData(const std::string& strData, AlarmInfo& alarmInfo)
  377. {
  378. try
  379. {
  380. nJson json0;
  381. json0 = nJson::parse(strData);
  382. alarmInfo.AlarmID = json0["alarmId"].get<int>();
  383. alarmInfo.ChannelID = json0["channel"].get<int>();
  384. alarmInfo.PicUrl = json0["picUrl"].get<std::string>();
  385. alarmInfo.ImageInfo = json0["imageInfo"].get<std::string>();
  386. /* 解析时间,需要将时间中的“T”换成空格 */
  387. alarmInfo.StartTime = json0["beginTime"].get<std::string>();
  388. std::replace(alarmInfo.StartTime.begin(), alarmInfo.StartTime.end(), 'T', ' ');
  389. alarmInfo.EndTime = json0["endTime"].get<std::string>();
  390. std::replace(alarmInfo.EndTime.begin(), alarmInfo.EndTime.end(), 'T', ' ');
  391. alarmInfo.EventTime = json0["eventTime"].get<std::string>();
  392. std::replace(alarmInfo.EventTime.begin(), alarmInfo.EventTime.end(), 'T', ' ');
  393. /* 判断bBoxes有无数据,有数据就解析,没数据就直接返回了 */
  394. nJson json1 = json0["bBoxes"];
  395. std::string labelList; /* 记录违禁品 */
  396. std::string bboxList; /* 记录bbox */
  397. if(!json1.empty())
  398. {
  399. for(auto& it0 : json1)
  400. {
  401. /* 如果status是true,就不是报警,直接跳过 */
  402. bool status = it0["status"].get<bool>();
  403. if(status)
  404. {
  405. continue;
  406. }
  407. /* 这一条Box数据报警了将其内容存储起来 */
  408. alarmInfo.Is_Alarm = true;
  409. /* 解析label,违禁品关键字,先判断这个是不是违禁品检测的算法ID */
  410. if(alarmInfo.ActionID == m_keyContraband)
  411. {
  412. /* 解析报警,取出报警类型 */
  413. nJson label = it0["label"];
  414. for(auto& it1 : label)
  415. {
  416. std::string strLabel = it1.get<std::string>();
  417. /* 检测是否已经加入到字符串中了 */
  418. strLabel += "|";
  419. if(labelList.find(strLabel) != std::string::npos)
  420. {
  421. continue;
  422. }
  423. labelList += strLabel;
  424. }
  425. }
  426. /* 解析bbox,貌似是在图像中的位置 */
  427. nJson bbox = it0["bbox"];
  428. std::string strBbox;
  429. for(auto& it1 : bbox)
  430. {
  431. strBbox += std::to_string(it1.get<int>()) + ",";
  432. }
  433. /* 去掉最后一个“,” */
  434. if(!strBbox.empty())
  435. {
  436. strBbox.pop_back();
  437. }
  438. bboxList += strBbox + "|";
  439. }
  440. /* 去掉最后一个“|” */
  441. if(!labelList.empty())
  442. {
  443. labelList.pop_back();
  444. }
  445. if(!bboxList.empty())
  446. {
  447. bboxList.pop_back();
  448. }
  449. SPDLOG_LOGGER_DEBUG(m_logger, "违禁品列表:{}", labelList);
  450. SPDLOG_LOGGER_DEBUG(m_logger, "bbox列表:{}", bboxList);
  451. }
  452. /* 如果有报警的Box */
  453. if(alarmInfo.Is_Alarm)
  454. {
  455. /* 添加报警信息的提示信息 */
  456. alarmInfo.BboxList = bboxList;
  457. if( (alarmInfo.ActionID == m_keyContraband) && !labelList.empty() )
  458. {
  459. alarmInfo.ActionDes = fmt::format("出现违禁品[{}]告警", labelList);
  460. SPDLOG_LOGGER_INFO(m_logger, "{}", alarmInfo.ActionDes);
  461. }else {
  462. alarmInfo.ActionDes = json0["actionDes"].get<std::string>();
  463. }
  464. /* 判断有没有报警数据 */
  465. if(alarmInfo.ImageInfo.empty())
  466. {
  467. SPDLOG_LOGGER_ERROR(m_logger, "有报警区域,但是没有图片信息");
  468. return;
  469. }
  470. /* 如果是人员报警,就存储人员报警信息 */
  471. if(alarmInfo.ActionID == m_KeyFace)
  472. {
  473. nJson jsonArray = json0["personList"];
  474. for(auto& it : jsonArray)
  475. {
  476. PersonInfo personInfo;
  477. personInfo.PersonID = it["personId"].get<std::string>();
  478. personInfo.PersonName = it["personName"].get<std::string>();
  479. alarmInfo.vecPersonInfo.push_back(personInfo);
  480. }
  481. }
  482. }
  483. }
  484. catch (const nJson::parse_error& e)
  485. {
  486. SPDLOG_LOGGER_ERROR(m_logger, "解析Redis数据失败:{}, 错误ID:{}", e.what(), e.id);
  487. return;
  488. }
  489. catch (const nJson::type_error& e)
  490. {
  491. SPDLOG_LOGGER_ERROR(m_logger, "解析Redis数据失败:{}, 错误ID:{}", e.what(), e.id);
  492. return;
  493. }
  494. }
  495. /* 判断时间是否长时间没有更新 */
  496. bool SPAServer::isEventTimeVaild(const std::string& strTime)
  497. {
  498. /* 获取当前时间 */
  499. std::chrono::system_clock::time_point now = std::chrono::system_clock::now();
  500. /* 字符串转成时间 */
  501. std::istringstream iss(strTime);
  502. std::tm tmEvent = {};
  503. iss >> std::get_time(&tmEvent, "%Y-%m-%d %H:%M:%S");
  504. /* 时间差 */
  505. std::chrono::system_clock::time_point eventTime = std::chrono::system_clock::from_time_t(std::mktime(&tmEvent));
  506. std::chrono::duration<double> diff = now - eventTime;
  507. // SPDLOG_LOGGER_DEBUG(m_logger, "now:{} eventTime: {} 时间差:{}秒",now, eventTime, diff);
  508. if(diff.count() > 600)
  509. {
  510. // SPDLOG_LOGGER_ERROR(m_logger, "Redis数据长时间没有更新,EventTime:{}", strTime);
  511. return false;
  512. }
  513. return true;
  514. }
  515. /**
  516. * @brief 分派任务的线程
  517. 1、 线程定时刷新房间和摄像机的关联关系,以及和算法Action的关联关系
  518. 2、 将算法信息加入到不同的列表中
  519. 需要多个摄像机配合的加入到m_runListRoomActionInfo列表
  520. 不需要多个摄像机配合的加入到m_runListActionInfo列表
  521. 3、 每次刷新都会清空算法的摄像机列表,但是不会动其他基本信息,如果是列表中没有对应的信息,就会创建新的ActionInfo
  522. 那么RunState的状态是INIT,需要开启新的线程
  523. 如果刷新完成后,算法对应的摄像机列表为空,那么对应的线程就会停止运行,将RunState设置为STOP,本线程的下一轮
  524. 循环就会删除这个ActionInfo
  525. */
  526. void SPAServer::threadRoomCamera()
  527. {
  528. /* 房间相机关联信息 */
  529. std::list<RoomCameraInfo> listRC;
  530. /* 创建连接数据库实例 */
  531. std::shared_ptr<ToEQMDataBase> toEQMDataBase = std::make_shared<ToEQMDataBase>();
  532. while (m_threadRunning)
  533. {
  534. /* 先获取EQM数据库信息,取出房间和摄像机关联信息 */
  535. /* 取出每个房间的所有算法,int是RoomID,string是Action */
  536. std::multimap<int, std::string> mapCameraActionID;
  537. m_mutexRunRAI.lock();
  538. m_mutexActionInfo.lock();
  539. /* 先清理已经退出的线程所用到的Action或者RoomAction */
  540. m_runListRoomActionInfo.clearStopRoomAction();
  541. m_runListActionInfo.clearStopAction();
  542. m_mutexRunActionInfo.lock();
  543. /* 将算法信息加入到不同的列表中
  544. * 需要多个摄像机配合的加入到m_runListRoomActionInfo列表
  545. * 不需要多个摄像机配合的加入到m_runListActionInfo列表
  546. * */
  547. m_runListRoomActionInfo.clearCameraList();
  548. m_runListActionInfo.clearCameraList();
  549. for(const auto& it : m_listActionInfo.getData())
  550. {
  551. if(it->ActionID == ActPersonWork || it->ActionID == ActPersonNumber)
  552. {
  553. m_runListRoomActionInfo.addActionInfo(*it);
  554. }else {
  555. m_runListActionInfo.addActionCamera(it);
  556. }
  557. }
  558. m_mutexActionInfo.unlock();
  559. /* 这个容器只用于多个摄像机融合的算法,一个房间的一个算法是一个线程 */
  560. for(const auto& it0 : m_runListRoomActionInfo.getData())
  561. {
  562. /* 人员在岗识别 */
  563. if(it0->ActionID == ActPersonWork)
  564. {
  565. SPDLOG_LOGGER_INFO(m_logger, "RoomID:{} 人员在岗识别", it0->RoomID);
  566. /* 判断是否需要创建新的线程 */
  567. if(it0->RunState == RunTimeState::RUN_STATE_INIT)
  568. {
  569. /* 创建新的线程 */
  570. CPPTP.add_task(&SPAServer::threadActPersonWork, this, it0);
  571. }
  572. }
  573. /* 区域人员检测 */
  574. else if (it0->ActionID == ActPersonNumber)
  575. {
  576. SPDLOG_LOGGER_INFO(m_logger, "RoomID:{} 区域人员检测", it0->RoomID);
  577. if(it0->RunState == RunTimeState::RUN_STATE_INIT)
  578. {
  579. CPPTP.add_task(&SPAServer::threadActPersonNumber, this, it0);
  580. }
  581. }
  582. }
  583. /* 这个容器用于不关联的算法信息,每个设备上的每个算法是一个线程 */
  584. for(const auto& it0 : m_runListActionInfo.getData())
  585. {
  586. /* 违禁物品识别,不需要多个摄像机融合判断,多少个报警都直接上报 */
  587. if (it0->ActionID == ActContraband)
  588. {
  589. SPDLOG_LOGGER_INFO(m_logger, "RoomID:{} 违禁品检测", it0->RoomID);
  590. if(it0->RunState == RunTimeState::RUN_STATE_INIT)
  591. {
  592. CPPTP.add_task(&SPAServer::threadActContraband, this, it0);
  593. }
  594. }
  595. /* 非法入侵检测,这个也不需要摄像机融合 */
  596. else if (it0->ActionID == ActIllegalInvasion)
  597. {
  598. SPDLOG_LOGGER_INFO(m_logger, "RoomID:{} 非法入侵检测", it0->RoomID);
  599. }
  600. /* 疲劳检测,不需要摄像机融合检测,直接上报 */
  601. else if (it0->ActionID == ActFatigueDetection)
  602. {
  603. SPDLOG_LOGGER_INFO(m_logger, "RoomID:{} 疲劳检测", it0->RoomID);
  604. }
  605. }
  606. m_mutexRunRAI.unlock();
  607. m_mutexRunActionInfo.unlock();
  608. }
  609. }
  610. /* 将该算法对应的摄像机放入摄像机列表 */
  611. bool SPAServer::insertCameraToAction(RoomActionInfo* pRAInfo, std::list<RoomCameraInfo>& listRC, std::multimap<int, std::string>& mapCameraActionID)
  612. {
  613. for(const auto& rci : listRC)
  614. {
  615. if(rci.RoomID == pRAInfo->RoomID)
  616. {
  617. /* 这个摄像机在这个房间内,再判断这个摄像机有没有这个算法 */
  618. for(const auto& camID : rci.listCameraID)
  619. {
  620. for(const auto& cai : mapCameraActionID)
  621. {
  622. if(camID == cai.first)
  623. {
  624. /* 再判断这个摄像机的算法是否是当前需要的 */
  625. if(cai.second == pRAInfo->ActionID)
  626. {
  627. pRAInfo->listCameraID.push_back(camID);
  628. }
  629. }
  630. }
  631. }
  632. }
  633. }
  634. return true;
  635. }
  636. /* 更新房间、算法需要的摄像机个数 */
  637. bool SPAServer::updateRoomActionCameraCount(std::shared_ptr<RoomActionInfo> pRAInfo)
  638. {
  639. std::lock_guard<std::mutex> look(m_mutexRunRAI);
  640. for(auto& it : m_runListRoomActionInfo.getData())
  641. {
  642. if((it->RoomID == pRAInfo->RoomID) && (it->ActionID == pRAInfo->ActionID))
  643. {
  644. it->listCameraID = pRAInfo->listCameraID;
  645. break;
  646. }
  647. }
  648. return true;
  649. }
  650. /* 更新算法的摄像机ID */
  651. bool SPAServer::updateActionCameraID(std::shared_ptr<ActionInfo> pInfo)
  652. {
  653. pInfo->CameraID = -1;
  654. std::lock_guard<std::mutex> look(m_mutexRunActionInfo);
  655. for(auto& it : m_runListActionInfo.getData())
  656. {
  657. if(it->isEqualBaseInfo(*pInfo))
  658. {
  659. pInfo->CameraID = it->CameraID;
  660. break;
  661. }
  662. }
  663. return true;
  664. }
  665. /* 设置线程状态 */
  666. void SPAServer::setThreadState(std::shared_ptr<ActionInfo> pInfo, RunTimeState state)
  667. {
  668. std::lock_guard<std::mutex> look(m_mutexRunActionInfo);
  669. for(auto& it : m_runListActionInfo.getData())
  670. {
  671. if(it->isEqualBaseInfo(*pInfo))
  672. {
  673. it->RunState = state;
  674. break;
  675. }
  676. }
  677. }
  678. /**
  679. * @brief 人员在岗识别线程,应该是人脸识别线程,这个需要房间内多个摄像机共同识别
  680. 1、离岗判断条件:直播间无人算离岗(直播间所有摄像头的区域人员检测算法输出结果都为0时记为离岗开始,
  681. 当结果存在非0时记为离岗结束)
  682. 2、离岗输出结果:当判断出离岗行为时,依据人脸识别结果,找到离岗时间点前有人的最近一帧画面中的人员,
  683. 判断为离岗人员,并计算离岗时长,形成离岗记录(频率、机房、离岗人员、离岗开始时间、离岗结束时间、离
  684. 岗持续时长)
  685. 3、在岗时长计算方式:每天00:00:00-23:59:59人脸识别结果中所有人员名称的出现次数✖抽帧频率(如5秒抽
  686. 取1帧,则出现200次人名表示在岗1000秒时长),形成每日的人员在岗时长记录
  687. 4、报警判断条件:依据离岗次数和单次离岗时长进行报警,展示报警结果
  688. *
  689. * @param pRAInfo 传入房间ID和算法ID
  690. */
  691. void SPAServer::threadActPersonWork(RoomActionInfo* RAInfo)
  692. {
  693. SPDLOG_LOGGER_INFO(m_logger, "开启人员在岗识别线程,RoomID:{} ,Action:{}", RAInfo->RoomID, RAInfo->ActionID);
  694. /* 创建读取Redis的实例 */
  695. std::shared_ptr<FromRedis> fromRedis = std::make_shared<FromRedis>();
  696. /* 局部变量 */
  697. std::shared_ptr<RoomActionInfo> pRAInfo = std::make_shared<RoomActionInfo>();
  698. *pRAInfo = *RAInfo;
  699. int CameraCount = pRAInfo->listCameraID.size();
  700. /* 摄像机数目小于0就退出线程 */
  701. while (m_threadRunning)
  702. {
  703. /* 更新算法关联的摄像机个数 */
  704. pRAInfo->listCameraID.clear();
  705. updateRoomActionCameraCount(pRAInfo);
  706. CameraCount = pRAInfo->listCameraID.size();
  707. if(CameraCount == 0)
  708. {
  709. break;
  710. }
  711. /* 读取Redis数据 */
  712. for(const auto& camID : pRAInfo->listCameraID)
  713. {
  714. std::string strKey = std::to_string(camID) + ":" + pRAInfo->ActionID;
  715. std::string strRetValue;
  716. if(!fromRedis->getRedisString(strKey, strRetValue))
  717. {
  718. SPDLOG_LOGGER_ERROR(m_logger, "读取Redis数据失败, Key:{}", strKey);
  719. continue;
  720. }
  721. /* 解析数据 */
  722. AlarmInfo alarmInfo;
  723. parseRedisData(strRetValue, alarmInfo);
  724. }
  725. }
  726. }
  727. /**
  728. * @brief 区域人员检测,检测这个区域内的人数,不能少于多少人,不能多余多少人
  729. * 1、报警判断条件:直播间报警:当前频率所有直播间摄像头的区域人员检测算法输出结果均大于或小于设定人
  730. 数值时,记为报警行为,直接展示报警结果
  731. 2、录播间报警:当前频率所有录播间摄像头的区域人员检测算法输出结果均大于或小于设定人数值时,记为报
  732. 警行为,直接展示报警结果
  733. 3、直播间+录播间报警:当前频率直播间人数+录播间人数大于或小于设定人数值时,记为报警行为,直接展示
  734. 报警结果
  735. *
  736. * @param RAInfo 传入的房间ID和算法ID
  737. */
  738. void SPAServer::threadActPersonNumber(RoomActionInfo* RAInfo)
  739. {
  740. SPDLOG_LOGGER_INFO(m_logger, "开启区域人员检测线程,RoomID:{} ,Action:{}", RAInfo->RoomID, RAInfo->ActionID);
  741. /* 创建读取Redis的实例 */
  742. std::shared_ptr<FromRedis> fromRedis = std::make_shared<FromRedis>();
  743. /* 局部变量 */
  744. std::shared_ptr<RoomActionInfo> pRAInfo = std::make_shared<RoomActionInfo>();
  745. *pRAInfo = *RAInfo;
  746. int CameraCount = 0;
  747. /* 摄像机数目等于0就退出线程 */
  748. while (m_threadRunning)
  749. {
  750. /* 更新算法关联的摄像机个数 */
  751. pRAInfo->listCameraID.clear();
  752. updateRoomActionCameraCount(pRAInfo);
  753. CameraCount = pRAInfo->listCameraID.size();
  754. if(CameraCount == 0)
  755. {
  756. break;
  757. }
  758. /* 读取Redis数据 */
  759. for(const auto& camID : pRAInfo->listCameraID)
  760. {
  761. std::string strKey = std::to_string(camID) + ":" + pRAInfo->ActionID;
  762. std::string strRetValue;
  763. if(!fromRedis->getRedisString(strKey, strRetValue))
  764. {
  765. SPDLOG_LOGGER_ERROR(m_logger, "读取Redis数据失败, Key:{}", strKey);
  766. continue;
  767. }
  768. /* 解析数据 */
  769. AlarmInfo alarmInfo;
  770. parseRedisData(strRetValue, alarmInfo);
  771. }
  772. }
  773. }
  774. /**
  775. * @brief 报警判断条件:机房内出现摄像头的违禁物品识别算法输出结果包含指定违规内容时,记为报警行为,直接
  776. 展示报警结果
  777. *
  778. * @param info
  779. */
  780. void SPAServer::threadActContraband(ActionInfo* info)
  781. {
  782. SPDLOG_LOGGER_INFO(m_logger, "开启违禁物品识别线程,RoomID:{} ,Action:{}", info->RoomID, info->ActionID);
  783. /* 创建读取Redis的实例 */
  784. std::shared_ptr<FromRedis> fromRedis = std::make_shared<FromRedis>();
  785. /* 局部变量,保存这个线程的信息 */
  786. std::shared_ptr<ActionInfo> pActInfo = std::make_shared<ActionInfo>();
  787. *pActInfo = *info;
  788. /* 摄像机ID小于0就退出线程 */
  789. while (m_threadRunning)
  790. {
  791. /* 更新算法关联的摄像机ID */
  792. updateActionCameraID(pActInfo);
  793. /* 如果摄像机ID是小于0,那么这个算法就没有对应的摄像机了,线程就退出了 */
  794. if(pActInfo->CameraID < 0)
  795. {
  796. break;
  797. }
  798. /* 读取Redis数据 */
  799. std::string strKey = std::to_string(pActInfo->CameraID) + ":" + pActInfo->ActionID;
  800. std::string strRetValue;
  801. if(!fromRedis->getRedisString(strKey, strRetValue))
  802. {
  803. SPDLOG_LOGGER_ERROR(m_logger, "读取Redis数据失败, Key:{}", strKey);
  804. continue;
  805. }
  806. /* 解析数据 */
  807. AlarmInfo alarmInfo;
  808. parseRedisData(strRetValue, alarmInfo);
  809. /* 判断事件的时效性,超过多少秒不更新就可能是超脑挂了 */
  810. if(isEventTimeVaild(alarmInfo.EventTime))
  811. {
  812. SPDLOG_LOGGER_WARN(m_logger, "Redis Key:{} 数据长时间没有更新,EventTime:{}",strKey, alarmInfo.EventTime);
  813. continue;
  814. }
  815. /* 判断有无报警记录,写入到EQM数据库 */
  816. }
  817. /* 设置线程退出的状态 */
  818. setThreadState(pActInfo, RunTimeState::RUN_STATE_STOP);
  819. SPDLOG_LOGGER_INFO(m_logger, "违禁物品识别线程退出,RoomID:{} ,CameraID, Action:{}", pActInfo->RoomID, pActInfo->CameraID, pActInfo->ActionID);
  820. }
  821. /**
  822. * @brief 1、野猫、宠物识别:报警判断条件:机房内出现摄像头的宠物识别算法输出报警结果时,记为报警行为,直接
  823. 展示报警结果
  824. 2、无权限人员识别:报警判断条件:当判断出区域非法入侵行为时,依据人脸识别结果,判断是否为人脸库人
  825. 员,非人脸库的人员时判断为非法入侵行为(背影需要报警需要结合区域人员检测算法判断)
  826. *
  827. * @param info
  828. */
  829. void SPAServer::threadActIllegalInvasion(ActionInfo* info)
  830. {
  831. SPDLOG_LOGGER_INFO(m_logger, "开启非法入侵检测线程,RoomID:{} ,Action:{}", info->RoomID, info->ActionID);
  832. /* 创建读取Redis的实例 */
  833. std::shared_ptr<FromRedis> fromRedis = std::make_shared<FromRedis>();
  834. /* 局部变量,保存这个线程的信息 */
  835. std::shared_ptr<ActionInfo> pActInfo = std::make_shared<ActionInfo>();
  836. *pActInfo = *info;
  837. /* 摄像机ID小于0就退出线程 */
  838. while (m_threadRunning)
  839. {
  840. /* 更新算法关联的摄像机ID */
  841. updateActionCameraID(pActInfo);
  842. /* 如果摄像机ID是小于0,那么这个算法就没有对应的摄像机了,线程就退出了 */
  843. if(pActInfo->CameraID < 0)
  844. {
  845. break;
  846. }
  847. /* 读取Redis数据 */
  848. std::string strKey = std::to_string(pActInfo->CameraID) + ":" + pActInfo->ActionID;
  849. std::string strRetValue;
  850. if(!fromRedis->getRedisString(strKey, strRetValue))
  851. {
  852. SPDLOG_LOGGER_ERROR(m_logger, "读取Redis数据失败, Key:{}", strKey);
  853. continue;
  854. }
  855. /* 解析数据 */
  856. AlarmInfo alarmInfo;
  857. parseRedisData(strRetValue, alarmInfo);
  858. /* 判断事件的时效性,超过多少秒不更新就可能是超脑挂了 */
  859. if(isEventTimeVaild(alarmInfo.EventTime))
  860. {
  861. SPDLOG_LOGGER_WARN(m_logger, "Redis Key:{} 数据长时间没有更新,EventTime:{}",strKey, alarmInfo.EventTime);
  862. continue;
  863. }
  864. /* 判断有无报警记录,写入到EQM数据库 */
  865. }
  866. /* 设置线程退出的状态 */
  867. setThreadState(pActInfo, RunTimeState::RUN_STATE_STOP);
  868. SPDLOG_LOGGER_INFO(m_logger, "非法入侵检测线程退出,RoomID:{} ,CameraID:{}, Action:{}", pActInfo->RoomID, pActInfo->CameraID, pActInfo->ActionID);
  869. }