From 439a23896bda9c4e954092816354914e645579b6 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期一, 10 六月 2019 09:37:13 +0800 Subject: [PATCH] 替换这个文件 --- QiaoJiaSystem/StructureApp/AppPipeController.cpp | 95 ++++++++++++++++++++++------------------------- 1 files changed, 44 insertions(+), 51 deletions(-) diff --git a/QiaoJiaSystem/StructureApp/AppPipeController.cpp b/QiaoJiaSystem/StructureApp/AppPipeController.cpp index 7fd95c9..aa0cbf5 100644 --- a/QiaoJiaSystem/StructureApp/AppPipeController.cpp +++ b/QiaoJiaSystem/StructureApp/AppPipeController.cpp @@ -9,7 +9,7 @@ #include <basic/pipe_element/ffmpeg/basic_struct_for_video_image.h> //AppPipeController::AppPipeController(std::string folderPath, const Json::Value &json) : -// videoCaptureElement("filePath", 25, -1, appPref.getLongData("gpu.index")), m_index(0), +// videoCaptureElement("filePath", 25, -1, appPref.getIntData("gpu.index")), m_index(0), // m_folderPath(folderPath), // m_json(json), fdfsClient(nullptr), yoloRpcElement(folderPath + "yoloRpc"), faceRpcElement(folderPath + "faceRpc"), // faceExtractElement(folderPath + "faceExtract"), triggerElement(25, 10), @@ -19,7 +19,8 @@ //} AppPipeController::AppPipeController(std::string folderPath, const SdkRuleMap &ruleMap) : - videoCaptureElement("filePath", 25, -1, appPref.getLongData("gpu.index")), + m_rpcClient("RtspAnalysServer", "127.0.0.1",appPref.getIntData("RpcVTIMPort"),"tcp"), + videoCaptureElement("filePath", 25, -1, appPref.getIntData("gpu.index")), // videoCaptureElement(40), m_index(0), m_folderPath(folderPath), @@ -29,17 +30,16 @@ faceRpcElement(folderPath + "faceRpc", ruleMap[FaceSdk]), faceExtractElement(folderPath + "faceExtract", ruleMap[FaceSdk]), triggerElement(25, 10), - recordVideoElement(-1, ""), +// recordVideoElement(-1, ""), bRecordVideoEnable(false), - rpcClient("RtspAnalysServer", "127.0.0.1",10009,"tcp"), - m_bSetWH(false) -{ + + m_bSetWH(false) { init(); } //AppPipeController::AppPipeController(int index, const Json::Value &json, bool RecordVideoEnable) : -// videoCaptureElement(json["rtsp"].asString(), 25, 3000, appPref.getLongData("gpu.index"), RecordVideoEnable), +// videoCaptureElement(json["rtsp"].asString(), 25, 3000, appPref.getIntData("gpu.index"), RecordVideoEnable), // m_index(index), fdfsClient(nullptr), yoloRpcElement(to_string(index) + "yoloRpc"), // faceRpcElement(to_string(index) + "faceRpc"), m_json(json), faceExtractElement(to_string(index) + "faceExtract"), // triggerElement(25, 4), recordVideoElement(index, json["rtsp"].asString()), bRecordVideoEnable(RecordVideoEnable) { @@ -48,7 +48,7 @@ //AppPipeController::AppPipeController(std::string camId, const Json::Value &json, bool RecordVideoEnable) : -// videoCaptureElement(json["rtsp"].asString(), 25, 3000, appPref.getLongData("gpu.index")), +// videoCaptureElement(json["rtsp"].asString(), 25, 3000, appPref.getIntData("gpu.index")), // m_index(0), // m_camId(camId), // fdfsClient(nullptr), @@ -69,7 +69,8 @@ // intKeepRight(); //} AppPipeController::AppPipeController(std::string camId, const SdkRuleMap &ruleMap, bool RecordVideoEnable) : - videoCaptureElement(appPref.getStringData(camId + "rtsp"), 25, 3000, appPref.getLongData("gpu.index")), + m_rpcClient("RtspAnalysServer", "127.0.0.1",appPref.getIntData("RpcVTIMPort"),"tcp"), + videoCaptureElement(appPref.getStringData(camId + "rtsp"), 25, 3000, appPref.getIntData("gpu.index")), // videoCaptureElement(40), m_index(0), m_camId(camId), @@ -78,20 +79,17 @@ faceRpcElement(camId + "faceRpc", ruleMap[FaceSdk]), faceExtractElement(camId + "faceExtract", ruleMap[FaceSdk]), triggerElement(25, 4), - recordVideoElement(camId, appPref.getStringData(camId + "rtsp")), - newRecordVideoElement(camId), +// recordVideoElement(camId, appPref.getStringData(camId + "rtsp")), +// newRecordVideoElement(camId), perimeterElement(ruleMap[PerimeterSdk]), crowdElement(ruleMap[CrowdSdk]), perStaticElement(ruleMap[PerStaticSdk]), perHubElement(ruleMap[PerHubSdk]), leftJudgment(ruleMap[KeepRightSdk]), rightJudgment(ruleMap[KeepRightSdk]), - bRecordVideoEnable(RecordVideoEnable), +// bRecordVideoEnable(RecordVideoEnable), m_sdkRuleMap(ruleMap), - rpcClient("RtspAnalysServer", "127.0.0.1",10009,"tcp"), - m_bSetWH(false) - - { + m_bSetWH(false) { DBG("camId" << camId); init(); initPerimeter(); @@ -179,7 +177,7 @@ void AppPipeController::RecordVideo() { - std::string imgKeyYolo = yoloRpcElement.getProperty("imgKey"); + /*std::string imgKeyYolo = yoloRpcElement.getProperty("imgKey"); std::string imgKeyFace = faceRpcElement.getProperty("imgKey"); if(faceRpcElement.getTriggerState() || @@ -226,8 +224,9 @@ "crowdElement: "<<crowdElement.getTriggerState()<<" "<< "perHubElement: "<<perHubElement.getTriggerState()<<" "<< "perStaticElement: "<<perStaticElement.getTriggerState()); - } + }*/ } + void AppPipeController::init() { unsigned char ip_old[15] = {0}; @@ -257,30 +256,27 @@ videoCaptureElement.registerConnector([&] { std::string imgKey; - m_hiredisTool.listRpop(m_camId,imgKey); - if(imgKey.empty()) - { - ERR("CamId "<<m_camId<<" No ImgKey"); + m_hiredisTool.listRpop(m_camId, imgKey); + if (imgKey.empty()) { +// ERR("CamId "<<m_camId<<" No ImgKey"); return; } cv::Mat imageTemp; - m_hiredisTool.getKeyImage(imgKey,imageTemp); - if(!m_bSetWH) - { - INFO("SetWidth:"<<imageTemp.cols<<" Height:"<<imageTemp.rows); - appPref.setIntData(m_camId+"width",imageTemp.cols); - appPref.setIntData(m_camId+"height",imageTemp.rows); - m_bSetWH=true; + m_hiredisTool.getKeyImage(imgKey, imageTemp); + if (!m_bSetWH) { + INFO("SetWidth:" << imageTemp.cols << " Height:" << imageTemp.rows); + appPref.setIntData(m_camId + "width", imageTemp.cols); + appPref.setIntData(m_camId + "height", imageTemp.rows); + m_bSetWH = true; } m_hiredisTool.delKey(imgKey); - if(imageTemp.empty()) - { - ERR("No Image Data In: "<<m_camId<<" ImgKey:"<<imgKey); + if (imageTemp.empty()) { + ERR("No Image Data In: " << m_camId << " ImgKey:" << imgKey); return; } - std::string strNewTime = AppUtil::getTimeUSecString(); + std::string strNewTime = AppUtil::getTimeUSecString(); ImageName_s_t imgSt = ImageName_s_t::fromString(imgKey); @@ -288,8 +284,8 @@ if (!faceRpcElement.isBusy()) { //#todo faceRpcElement.setProperty("time", strNewTime); - faceRpcElement.setProperty("imgKey",imgKey); - INFO("Write To FaceRPC ES time:"<<strNewTime<< " ImgKey: "<<imgKey); + faceRpcElement.setProperty("imgKey", imgKey); +// INFO("Write To FaceRPC ES time:" << strNewTime << " ImgKey: " << imgKey); faceRpcElement.setImage(imageTemp); faceRpcElement.submit(); } @@ -297,8 +293,8 @@ if (!yoloRpcElement.isBusy()) { //#todo yoloRpcElement.setProperty("time", strNewTime); - yoloRpcElement.setProperty("imgKey",imgKey); - INFO("Write To YoloES time:"<<strNewTime<< " ImgKey: "<<imgKey); + yoloRpcElement.setProperty("imgKey", imgKey); +// INFO("Write To YoloES time:" << strNewTime << " ImgKey: " << imgKey); yoloRpcElement.setImage(imageTemp); yoloRpcElement.submit(); } @@ -309,7 +305,7 @@ imageDrawElement.submit(); } //this->RecordVideo(); - if(faceRpcElement.getTriggerState() || + /*if(faceRpcElement.getTriggerState() || personElement.getTriggerState() || leftJudgment.getTriggerState() || rightJudgment.getTriggerState() || @@ -343,7 +339,7 @@ "crowdElement: "<<crowdElement.getTriggerState()<<" "<< "perHubElement: "<<perHubElement.getTriggerState()<<" "<< "perStaticElement: "<<perStaticElement.getTriggerState()); - } + }*/ } }); @@ -384,7 +380,7 @@ imageDrawElement.registerConnector([&] { if (appPref.getIntData("show.image") == 1) { - ImageShowElement::showImage(to_string(this->m_index), *imageDrawElement.getImage()); + ImageShowElement::showImage(to_string(this->m_index), *imageDrawElement.getImage()); } }); @@ -397,8 +393,7 @@ registerElement(faceRpcElement); registerElement(faceExtractElement); } - if(appPref.getIntData("show.image") == 1) - { + if (appPref.getIntData("show.image") == 1) { imageDrawElement.registerConnector([&] { ImageShowElement::showImage(to_string(this->m_index), *imageDrawElement.getImage()); @@ -408,8 +403,8 @@ registerElement(imageDrawElement); } - // registerElement(newRecordVideoElement); - videoCaptureElement.setOutPutInterval(3); + // registerElement(newRecordVideoElement); + //videoCaptureElement.setOutPutInterval(3); faceExtractElement.setProperty("index", to_string(m_index)); registerElement(yoloRpcElement); //#todo setProperty @@ -542,14 +537,12 @@ leftJudgment.init(rule.strAreas, rule.strLine); rightJudgment.init(rule.strExAreas, rule.strExLine); yoloRpcElement.registerConnector([&] { - if(!leftJudgment.isBusy()) - { - leftJudgment.setYoloObjects(yoloRpcElement.getLastScoreRects()); - leftJudgment.setImage(yoloRpcElement.getImage()); - leftJudgment.submit(); + if (!leftJudgment.isBusy()) { + leftJudgment.setYoloObjects(yoloRpcElement.getLastScoreRects()); + leftJudgment.setImage(yoloRpcElement.getImage()); + leftJudgment.submit(); } - if(!rightJudgment.isBusy()) - { + if (!rightJudgment.isBusy()) { rightJudgment.setYoloObjects(yoloRpcElement.getLastScoreRects()); rightJudgment.setImage(yoloRpcElement.getImage()); -- Gitblit v1.8.0