//
|
// Created by ps on 18-9-26.
|
//
|
|
#include "RtspAnalysElement.h"
|
|
#include <basic/http_server/HttpSrvRetRecieve.hpp>
|
|
using std::string;
|
|
//RtspAnalysElement::RtspAnalysElement() {}
|
|
std::string RtspAnalysElement::resetFdfs(std::string ip, unsigned int port, std::string content) {
|
|
std::thread reset([&]() {
|
fdfsClient.rwLock.rdlock();
|
if (fdfsClient.fastFds != nullptr) {
|
delete fdfsClient.fastFds;
|
}
|
fdfsClient.fastFds = new FastFds("fastDfsClient.conf");
|
fdfsClient.rwLock.unlock();
|
});
|
reset.detach();
|
|
return "";
|
}
|
|
RtspAnalysElement::RtspAnalysElement(LDBTool *_dbTool) : m_lDBTool(_dbTool), maxCount(50), currentCount(0) {
|
resetFdfs();
|
init();
|
{
|
IMPORT_APP_ARGS;
|
//nsq set callback func
|
DBG("NsqMsgConsumer Init");
|
DBG("argv[0]:" << argv[0]);
|
if(argv[0][0] == '.' && argv[0][1] == '/'){
|
string pName(argv[0]+2, argv[0]+strlen(argv[0]));
|
DBG("pName:" << pName);
|
nsqMsgConsumer = new BasicMsg::Nsq::NsqMsgConsumer("127.0.0.1", "4150", pName,
|
to_string(appPref.getIntData("RpcVTIMPort")));
|
}else{
|
string pName(argv[0]);
|
DBG("pName:" << pName);
|
nsqMsgConsumer = new BasicMsg::Nsq::NsqMsgConsumer("127.0.0.1", "4150", pName,
|
to_string(appPref.getIntData("RpcVTIMPort")));
|
}
|
nsqMsgConsumer->setMessageCallback(std::bind(&OnMsgFunc, std::placeholders::_1, this));
|
nsqMsgConsumer->init();
|
DBG("NsqMsgConsumer Init END!!");
|
}
|
{
|
// std::thread httpServer([&]() {
|
// //#todo port
|
// HttpSrvRetRecieve httpSrvRetRecieve("0.0.0.0", 12306, 1);
|
// httpSrvRetRecieve.setInfo("^/resetFdfs$", "POST", std::bind(&RtspAnalysElement::resetFdfs, this,
|
// std::placeholders::_1,
|
// std::placeholders::_2,
|
// std::placeholders::_3));
|
// httpSrvRetRecieve.start();
|
// httpSrvRetRecieve.waitForShutDown();
|
// });
|
// httpServer.detach();
|
}
|
}
|
|
#define SETSCORE(VEC, POS, IDENT) appPref.setFloatData(IDENT, POS >= VEC.size() ? 80 : VEC[POS]);
|
//#define GETSCORE(IDENT) appPref.getFloatData(IDENT) == -1 ? 0.1 : appPref.getFloatData(IDENT);
|
|
static std::string rtspAddrBuild(std::string ip, const int port, std::string username,
|
std::string passwd, std::string brand) {
|
std::string rtsp_url;
|
if (brand == "haikang") {
|
rtsp_url = "rtsp://" + username + ":" + passwd + "@" + ip + ":" + std::to_string(port) +
|
"/h264/ch1/main/av_stream";
|
} else if (brand == "dahua") {
|
//rtsp://username:password@ip:port/cam/realmonitor?channel=1&subtype=0
|
rtsp_url = "rtsp://" + username + ":" + passwd + "@" + ip + ":" + std::to_string(port) +
|
"/cam/realmonitor?channel=1&subtype=0";
|
//TODO
|
} else if (brand == "yushi") {
|
rtsp_url = "rtsp://" + username + ":" + passwd + "@" + ip + ":" + std::to_string(port) +
|
"/video1";
|
} else {
|
//TODO
|
}
|
|
return rtsp_url;
|
}
|
|
|
void RtspAnalysElement::OnMsgFunc(void *msgPtr, void *pThisPtr) {
|
|
evnsq::Message *msg = (evnsq::Message *) msgPtr;
|
DBG("Received a message, id=" << msg->id << " message=[" << msg->body.ToString() << "]");
|
RtspAnalysElement *pthis = (RtspAnalysElement *) pThisPtr;
|
Json::Reader reader;
|
Json::Value value;
|
if (reader.parse(msg->body.ToString(), value)) {
|
auto itor_json_end = value.end();
|
Json::Value cfg_val;
|
for (auto itor_json = value.begin(); itor_json != itor_json_end; ++itor_json) {
|
//#todo 使用策略模式?发送端多个指令累积后发送?
|
if (itor_json->isString() && reader.parse(itor_json->asString(), cfg_val)) {
|
if (itor_json.name() == "cam_edit" || itor_json.name() == "editSdkRule") {
|
string cam_idx = "";
|
if(itor_json.name() == "cam_edit"){
|
if (pthis->m_GB28181_Enable){
|
cam_idx = cfg_val["equipCode"].asString();
|
}else{
|
cam_idx = cfg_val["str_cam_dev_id"].asString();
|
}
|
} else {
|
cam_idx = cfg_val["strCamId"].asString();
|
}
|
|
if (pthis->controllers.find(cam_idx) != pthis->controllers.end()) {
|
if (pthis->m_GB28181_Enable) {
|
auto lst = pthis->m_lDBTool->searchCamDevTableByType(1);
|
for (auto &item : lst) {
|
if (item.str_cam_dev_id.toStdString() != cam_idx)continue;
|
string rtsp_url = "GB28181";
|
pthis->addCamWithSearchRuleMap(item, rtsp_url);
|
break;
|
}
|
} else {
|
auto lst = pthis->m_lDBTool->searchCamDevTableByType(0);
|
for (auto &item : lst) {
|
if (item.str_cam_dev_id.toStdString() != cam_idx)continue;
|
std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554,
|
item.str_username.toStdString(),
|
item.str_password.toStdString(),
|
item.str_brand.toStdString());
|
pthis->addCamWithSearchRuleMap(item, rtsp_url);
|
break;
|
}
|
}
|
}
|
} else if (itor_json.name() == "cam_del") {
|
// string cam_idx = cfg_val["str_cam_dev_id"].asString();
|
string cam_idx = "";
|
if(cfg_val["equipCode"].asString().size()){
|
cam_idx = cfg_val["equipCode"].asString();
|
}else if(cfg_val["str_cam_dev_id"].asString().size()){
|
cam_idx = cfg_val["str_cam_dev_id"].asString();
|
}
|
pthis->removeCamera(cam_idx);
|
}
|
} else {
|
DBG("itor value is not string ");
|
continue;
|
}
|
}
|
} else {
|
ERR("Error Message!!");
|
}
|
}
|
|
void RtspAnalysElement::init() {
|
|
m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
|
|
Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();
|
|
//#todo GB28181
|
if (m_GB28181_Enable) {
|
auto lst = m_lDBTool->searchCamDevTableByType(1);
|
|
appPref.setIntData("n_cut_max_duration", lst_dev.n_cut_max_duration);
|
appPref.setIntData("n_cut_min_duration", lst_dev.n_cut_min_duration);
|
if (lst.size() > 0) {
|
int startCamNO = appPref.getIntData("CamStartNO");
|
int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
|
auto itor = lst.begin();
|
|
if (startCamNO >= lst.size()) {
|
ERR("startCamNO > lst.size()");
|
return;
|
}
|
for (int i = 0; i < startCamNO; i++) {
|
itor++;
|
}
|
|
for (int i = 0; i < CamCount; i++) {
|
if (itor == lst.end()) {
|
ERR("itor == lst.end()");
|
return;
|
}
|
auto &item = *itor;
|
|
string rtsp_url = "GB28181";
|
addCamWithSearchRuleMap(item, rtsp_url);
|
|
itor++;
|
}
|
|
// for (auto &item : lst) {
|
// std::string t_camIdex = item.str_cam_dev_id.toStdString();
|
// std::string rtsp_url = "GB28181";
|
//
|
// SdkRuleMap ruleMap = m_lDBTool->searchSdkRuleByCamId(item.str_cam_dev_id);
|
// for (auto &rule:ruleMap) {
|
//
|
// rule.second.strAddr = item.str_addr;
|
// rule.second.weekRuleVec = m_lDBTool->searchCameraWeekRuleByCamId(item.str_cam_dev_id, rule.first);
|
//
|
// if (rule.second.nSdkType == PerStaticSdk) {
|
// float temp = 1 - (float) (rule.second.nThreshold) / 100;
|
// rule.second.fSensitivity = (75 + 25 * temp) / 100;
|
//
|
// } else if (rule.second.nSdkType != FaceSdk) {
|
// float temp = 1 - (float) (rule.second.nThreshold) / 100;
|
// rule.second.fSensitivity = (5 + 90 * temp) / 100;
|
// }
|
// DBG("fSensitivity" << rule.second.fSensitivity);
|
// }
|
// appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
|
// INFO("cam add is " << item.str_addr.toStdString());
|
// addCamera(t_camIdex, ruleMap);
|
// }
|
} else {
|
ERR("searchCamDevTableAll size is 0");
|
}
|
|
} else {
|
|
auto lst = m_lDBTool->searchCamDevTableByType(0);
|
|
appPref.setIntData("n_cut_max_duration", lst_dev.n_cut_max_duration);
|
appPref.setIntData("n_cut_min_duration", lst_dev.n_cut_min_duration);
|
if (lst.size() > 0) {
|
int startCamNO = appPref.getIntData("CamStartNO");
|
int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
|
auto itor = lst.begin();
|
|
if (startCamNO >= lst.size()) {
|
ERR("startCamNO > lst.size()");
|
return;
|
}
|
|
for (int i = 0; i < startCamNO; i++) {
|
itor++;
|
}
|
|
for (int i = 0; i < CamCount; i++) {
|
if (itor == lst.end()) {
|
ERR("itor == lst.end()");
|
return;
|
}
|
auto &item = *itor;
|
std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554, item.str_username.toStdString(),
|
item.str_password.toStdString(), item.str_brand.toStdString());
|
|
addCamWithSearchRuleMap(item, rtsp_url);
|
|
itor++;
|
}
|
} else {
|
ERR("searchCamDevTableAll size is 0");
|
}
|
|
}
|
}
|
|
void RtspAnalysElement::addCamWithSearchRuleMap(const Record_Cam_Dev &item, string rtsp_url) {
|
string t_camIdex = item.str_cam_dev_id.toStdString();
|
|
SdkRuleMap ruleMap = m_lDBTool->searchSdkRuleByCamId(item.str_cam_dev_id);
|
for (auto &rule:ruleMap) {
|
|
rule.second.strAddr = item.str_addr;
|
rule.second.weekRuleVec = m_lDBTool->searchCameraWeekRuleByCamId(item.str_cam_dev_id, rule.first);
|
|
// if (rule.second.nSdkType == PerStaticSdk || rule.second.nSdkType == PerimeterSdk
|
// || rule.second.nSdkType == KeepRightSdk || rule.second.nSdkType == KeepRightSdk) {
|
float temp = (float)1 - (float) (rule.second.nThreshold) / 100;
|
rule.second.fSensitivity = (temp * 50 + 50) / 100;
|
// rule.second.fSensitivity = (temp * 25 + 75) / 100;
|
|
// } else if (rule.second.nSdkType != FaceSdk) {
|
// float temp = (float)1 - (float) (rule.second.nThreshold) / 100;
|
// rule.second.fSensitivity = (temp * 90 + 5) / 100;
|
// }
|
DBG("fSensitivity" << rule.second.fSensitivity);
|
}
|
appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
|
|
INFO("cam add is " << item.str_addr.toStdString());
|
addCamera(t_camIdex, ruleMap);
|
}
|
|
|
RtspAnalysElement::~RtspAnalysElement() {
|
for (auto controller: controllers) {
|
delete controller.second;
|
}
|
controllers.clear();
|
}
|
|
int RtspAnalysElement::addCamera(const std::string &index, const std::map<int, SdkRule> &sdkRuleMap) {
|
if (controllers.find(index) == controllers.end()) {
|
if (currentCount >= maxCount) {
|
ERR("addCamera faild, camera's num is full!")
|
return -1;
|
}
|
|
controllers[index] = new AppPipeController(index, sdkRuleMap, true);
|
controllers[index]->setfdfsClient(&fdfsClient);
|
controllers[index]->start();
|
currentCount++;
|
return 0;
|
|
} else {
|
removeCamera(index);
|
DBG("removeCamera " << index)
|
return addCamera(index, sdkRuleMap);
|
}
|
}
|
|
int RtspAnalysElement::removeCamera(const std::string &index) {
|
if (controllers.find(index) == controllers.end())return -1;
|
auto controller = controllers[index];
|
controller->stop();
|
controller->wait();
|
delete controller;
|
controllers.erase(index);
|
currentCount--;
|
return 0;
|
}
|
|
int RtspAnalysElement::removeAll() {
|
for (auto controller: controllers) {
|
controller.second->stop();
|
}
|
for (auto controller: controllers) {
|
controller.second->wait();
|
delete controller.second;
|
}
|
controllers.clear();
|
currentCount = 0;
|
return 0;
|
}
|
|
int RtspAnalysElement::getMaxCamCount() {
|
return maxCount;
|
}
|
|
int RtspAnalysElement::getCurrentCamCount() {
|
return currentCount;
|
}
|
|
std::string RtspAnalysElement::getRtmp(std::string &index) {
|
if (controllers.find(index) == controllers.end())return "";
|
auto controller = controllers[index];
|
return controller->getRtmp();
|
}
|
|
|
std::vector<int> RtspAnalysElement::chnString2Vec(std::string str_list) {
|
std::vector<int> result;
|
char *property_list = const_cast<char *>(str_list.c_str());
|
const char *c_Analyse = ",";
|
char *t_property;
|
|
char *t_save = NULL;
|
t_property = strtok_r(property_list, c_Analyse, &t_save);
|
|
while (t_property) {
|
std::string str_pro(t_property);
|
|
result.push_back(atoi(str_pro.c_str()));
|
|
t_property = strtok_r(t_save, c_Analyse, &t_save);
|
|
}
|
return result;
|
}
|
|
//void RtspAnalysElement::setDataByType(Json::Value& json,const SdkRule& rule)
|
//{
|
// std::string camId=rule.strCamId.toStdString();
|
// sdkRuleMap[rule.nSdkType].weekRuleVec=m_lDBTool->searchCameraWeekRuleByCamId(rule.strCamId,rule.nSdkType);
|
// switch (rule.nSdkType)
|
// {
|
// case FaceSdk:
|
// {
|
// json["face.enable"] = "1";
|
|
// appPref.setFloatData(camId+"face.det",rule.nQuality);
|
// appPref.setFloatData(camId+"face.cmp",rule.nThreshold);
|
|
|
|
// break;
|
// }
|
// case CarSdk:
|
// {
|
// // DBG("XX.enable " << "1");
|
|
// appPref.setFloatData(camId+"xx.det",rule.nQuality);
|
// appPref.setFloatData(camId+"xx.cmp",rule.nThreshold);
|
|
// break;
|
// }
|
// case YoloSdk:
|
// {
|
// json["yolo.enable"] = "1";
|
// appPref.setFloatData(camId+"yolo.det",rule.nQuality);
|
// appPref.setFloatData(camId+"yolo.cmp",rule.nThreshold);
|
|
// break;
|
// }
|
// case PerimeterSdk:
|
// {
|
// // auto rule=m_lDBTool->searchPerimeterRuleByCamId(camId);
|
// json["perimeter.enable"] = 1;
|
// json["perimeter.area"]=rule.strAreas.toStdString();
|
// json["perimeter.num"]=rule.nAlarmNumLowerLimit;
|
|
// json["perimeter.delay"]=rule.nTriggerDelay;
|
// json["perimeter.tolerance"]=rule.nTriggerTolerance;
|
// appPref.setFloatData(camId+"perimeter.det",rule.nQuality);
|
// appPref.setFloatData(camId+"perimeter.cmp",rule.nThreshold);
|
|
|
// break;
|
// }
|
// case CrowdSdk:
|
// {
|
// json["crowd.enable"] = 1;
|
// json["crowd.area"]=rule.strAreas.toStdString();
|
// json["crowd.num"]=rule.nAlarmNumLowerLimit;
|
// json["crowd.delay"]=rule.nTriggerDelay;
|
// json["crowd.tolerance"]=rule.nTriggerTolerance;
|
// appPref.setFloatData(camId+"crowd.det",rule.nQuality);
|
// appPref.setFloatData(camId+"crowd.cmp",rule.nThreshold);
|
|
// break;
|
// }
|
// case KeepRightSdk:
|
// {
|
// json["keepRight.enable"] = 1;
|
// json["keepRight.leftArea"]=rule.strAreas.toStdString();
|
// json["keepRight.leftLine"]=rule.strLine.toStdString();
|
// json["keepRight.rightArea"]=rule.strExAreas.toStdString();
|
// json["keepRight.rightLine"]=rule.strExLine.toStdString();
|
// json["keepRight.delay"]=rule.nTriggerDelay;
|
// json["keepRight.tolerance"]=rule.nTriggerTolerance;
|
// appPref.setFloatData(camId+"keepRight.det",rule.nQuality);
|
// appPref.setFloatData(camId+"keepRight.cmp",rule.nThreshold);
|
// break;
|
// }
|
// default:
|
// break;
|
// }
|
//}
|
