#include "PerimeterElement.h" //#include #include #include #include #include #include #include #include #include #include #include #include #define ALARM_PERCENT (25) //25% PerimeterElement::PerimeterElement(const SdkRule &rule) : m_sdkRule(rule), PipeElement(true), trackingTrigger(nullptr), mRealNum(0), //每秒几张,就乘以几 m_triggerElement(rule.nTriggerDelay * 3, 0), pManagerEsDB(nullptr), m_bIsMask(true), m_bSetWH(false), pointArray(nullptr), m_rpcClient("RtspAnalysServer", "127.0.0.1", appPref.getIntData("RpcVTIMPort"), "tcp") { pManagerEsDB = new EsDBTool(appPref.getStringData("ipAdd"), appPref.getIntData("ipPort")); } PerimeterElement::~PerimeterElement() { if (pManagerEsDB) { delete pManagerEsDB; pManagerEsDB = nullptr; } if (pointArray) { // delete[] pointArray[0]; delete[] pointArray; } } void PerimeterElement::threadInitial() { // setMask(m_sdkRule.strAreas.toStdString()); } void PerimeterElement::threadFunc() { //ClockTimer ct("PerimeterElement::threadFunc"); if (!isInWeek(m_sdkRule.weekRuleVec)) return; int num = 0; std::string picDate; std::string imgKey; bool state = false; m_recVec.clear(); //todo debug m_scoreVec.clear(); m_objTypeVec.clear(); // DBG("Objs="< poly2; std::vector interPoly; poly2.push_back(Point(rect.x(), rect.y())); poly2.push_back(Point(rect.x() + rect.width(), rect.y())); poly2.push_back(Point(rect.x() + rect.width(), rect.y() + rect.height())); poly2.push_back(Point(rect.x(), rect.y() + rect.height())); IntAreaCalcUtil::PolygonClip(poly1, poly2, interPoly); float inter = IntAreaCalcUtil::intAreaCalc(interPoly); float total1 = IntAreaCalcUtil::intAreaCalc(poly1); float total2 = IntAreaCalcUtil::intAreaCalc(poly2); int per1 = (int) (inter / total1 * 100); int per2 = (int) (inter / total2 * 100); int per = std::max(per1, per2); if (ALARM_PERCENT <= per) { //DBG("per 1="< 0 ; state = (num != mRealNum); // DBG("num="< m_sdkRule.nAlarmNumLowerLimit && num != mRealNum; // INFO("SDK Type: CrowdSdk"); break; case PerHubSdk: DBG("nAlarmNumLowerLimit:" << m_sdkRule.nAlarmNumLowerLimit << " nAlarmNumUpperLimit:" << m_sdkRule.nAlarmNumUpperLimit << " mRealNum:" << mRealNum); if(m_sdkRule.nAlarmNumUpperLimit == -1){ state = num >= m_sdkRule.nAlarmNumLowerLimit && num !=mRealNum; }else if(m_sdkRule.nAlarmNumLowerLimit == -1){ state = num <= m_sdkRule.nAlarmNumUpperLimit && num !=mRealNum; } else { state = num >= m_sdkRule.nAlarmNumLowerLimit && num <= m_sdkRule.nAlarmNumUpperLimit && num != mRealNum; } // INFO("SDK Type: PerHubSdk"); // if(m_sdkRule.nAlarmNumLowerLimit>=0) // { // if( m_sdkRule.nAlarmNumUp2019-01-16 09:49:57perLimit>m_sdkRule.nAlarmNumLowerLimit+1) // { // m_triggerElement.setState(num>m_sdkRule.nAlarmNumLowerLimit && numm_sdkRule.nAlarmNumLowerLimit); // } // } // else // { // if(m_sdkRule.nAlarmNumUpperLimit>0) // { // m_triggerElement.setState(num>m_sdkRule.nAlarmNumUpperLimit); // } // else // { // m_triggerElement.setState(false); // } // } break; default: break; } INFO("State:" << state << " Num:" << num << " mRealNum:" << mRealNum << " LowerLimit :" << m_sdkRule.nAlarmNumLowerLimit << " HightLimit: " << m_sdkRule.nAlarmNumUpperLimit << " m_triggerElement.getTriggerState() :" << m_triggerElement.getTriggerState()); m_triggerElement.setState(state); m_triggerElement.triggerOnce(); DBG("m_triggerElement.getTriggerState():" << m_triggerElement.getTriggerState()); if (m_triggerElement.getTriggerState()) { if (num > 0) { if (pointArray != nullptr) { const cv::Point2i *ppt[1] = {pointArray}; int npt[] = {npts}; cv::polylines(image, ppt, npt, 1, true, cv::Scalar(255, 255, 0), 2); } // for(auto rect:m_recVec) // { // cv::rectangle(image, rect, cv::Scalar(0, 0, 255), 2); // } for (int i = 0; i < m_recVec.size(); i++) { cv::rectangle(image, m_recVec[i], cv::Scalar(0, 0, 255), 2); cv::putText(image, to_string(m_scoreVec[i]), cv::Point(m_recVec[i].x, m_recVec[i].y + 12), cv::HersheyFonts::FONT_HERSHEY_PLAIN, 1.5, cv::Scalar(255, 255, 0)); cv::putText(image, m_objTypeVec[i], cv::Point(m_recVec[i].x, m_recVec[i].y + 24), cv::HersheyFonts::FONT_HERSHEY_PLAIN, 1.5, cv::Scalar(255, 255, 0)); } // cv::Mat t_image = image(CutMask()).clone(); // if(t_image.empty()) // { // DBG("t_image empty"); // return; // } std::string imgUrl = uploadImgToFdfs(image); saveInfoToEs(imgUrl, picDate, imgKey); } DBG("num=" << num << " lastnum=" << mRealNum); mRealNum = num; //todo 重置报警trigger m_triggerElement.setState(false); m_triggerElement.triggerOnce(); } // if (m_triggerElement.getTriggerState() && num!=mRealNum) // { // cv::Mat t_image = image(CutMask()).clone(); // if(t_image.empty()) // { // DBG("t_image empty"); // return; // } // std::string imgUrl = uploadImgToFdfs(t_image); // saveInfoToEs(imgUrl, picDate); // DBG("num=" << num << " lastnum=" << mRealNum); // mRealNum = num; // } // if(!state) // { // mRealNum = num; // } // DBG("m_sdkRule.nSdkType="<data()); // } value.copyTo(image); if (!m_bSetWH) { setMask(m_sdkRule.strAreas.toStdString()); m_bSetWH = true; } } bool PerimeterElement::getTriggerState() const { return m_triggerElement.getTriggerState(); } std::string PerimeterElement::uploadImgToFdfs(cv::Mat &image) { std::string strImgUrl = "http://"; if (fdfsClient != nullptr && fdfsClient->fastFds != nullptr) { fdfsClient->rwLock.rdlock(); std::vector buffer; CvUtil::cvMat2Buffer(image, buffer); std::string strImgUrlTmp = ""; fdfsClient->fastFds->uploadFile(buffer, strImgUrlTmp, "jpg"); // strImgUrl.append(fdfsClient->fastFds->getIp() + "/" + strImgUrlTmp); strImgUrl.clear(); strImgUrl = strImgUrlTmp; // strImgUrl.append("/").append(strImgUrlTmp); fdfsClient->rwLock.unlock(); } return strImgUrl; } bool PerimeterElement::saveInfoToEs(const std::string &imgUrl, const std::string &time, const std::string &imgKey) { string str_uuid; uuid_t t_uuid; char str[36]; uuid_generate(t_uuid); uuid_unparse(t_uuid, str); str_uuid = str; Json::Value t_json; t_json["Id"] = str_uuid; //主键 //#todo t_json["videoReqNum"] = m_sdkRule.strCamId.toStdString(); t_json["sdkType"] = to_string(m_sdkRule.nSdkType); // DBG("to_string(m_sdkRule.nSdkType)="<recordVideo(imgKey); } catch (std::exception &e) { ERR("Record Video Err: " << imgKey << " Message: " << e.what()); //return false; } INFO("SaveImgKeyToES: " << imgKey); bool retface = false; if (pManagerEsDB) retface = pManagerEsDB->insertData("personaction", "perVideoAction", t_json.toStyledString(), str_uuid); if (retface) { INFO("perVideoAction db success"); } else { ERR("personaction db fail"); } } bool PerimeterElement::isInWeek(const std::vector &ruleWeek) { int nWeek = QDate::currentDate().dayOfWeek(); for (int i = 0; i < ruleWeek.size(); ++i) { if (ruleWeek[i].m_nType == nWeek) { QString strCurrent = QDateTime::currentDateTime().toString("hh:mm"); if (strCurrent >= ruleWeek[i].m_strBegin && strCurrent <= ruleWeek[i].m_strEnd) { return true; } } } return false; } cv::Rect PerimeterElement::CutMask() { int size = m_recVec.size(); //QPoint max,secMax; cv::Rect rect, max, min; if (size > 0) { max = m_recVec[0]; min = m_recVec[0]; } for (int i = 1; i < size; ++i) { if (m_recVec[i].x > max.x) { max.x = m_recVec[i].x; } if (m_recVec[i].y > max.y) { max.y = m_recVec[i].y; } if (m_recVec[i].x < min.x) { min.x = m_recVec[i].x; } if (m_recVec[i].y < min.y) { min.y = m_recVec[i].y; } } if (size == 1) { rect = max; } else { rect.x = min.x; rect.y = min.y; rect.width = max.x + max.width - min.x; rect.height = max.y + max.height - min.y; } rect = CvUtil::zoomRectEqual(rect, 1.2, 1.2) & cv::Rect(0, 0, image.cols, image.rows); // DBG("min x="<