chenshijun
2019-04-16 f946a62d3921e86b44ff8e2973138304b9cd53cd
QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp
@@ -26,8 +26,28 @@
}
RtspAnalysElement::RtspAnalysElement(LDBTool *_dbTool) : m_lDBTool(_dbTool), maxCount(50), currentCount(0) {
    init();
    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
@@ -67,131 +87,208 @@
    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() {
    auto lst = m_lDBTool->searchCamDevTableAll();
    m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
    Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();
    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) {
        Json::Value json;
        int startCamNO = appPref.getIntData("CamStart") * appPref.getIntData("CamStep");
        int CamCount = appPref.getIntData("CamStep");
        auto itor = lst.begin();
    //#todo GB28181
    if (m_GB28181_Enable) {
        auto lst = m_lDBTool->searchCamDevTableByType(1);
        if(startCamNO >= lst.size())
        {
            ERR("startCamNO > lst.size()");
            return;
        }
        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();
        for(int i=0; i < startCamNO; i++){
            itor++;
        }
        for (int i = 0; i < CamCount; i++) {
            if (itor == lst.end())
            {
                ERR("itor == lst.end()");
            if (startCamNO >= lst.size()) {
                ERR("startCamNO > lst.size()");
                return;
            }
            std::string t_camIdex = itor->str_cam_dev_id.toStdString();
            std::string rtsp_url = rtspAddrBuild(itor->str_ip.toStdString(), 554, itor->str_username.toStdString(),
                                                 itor->str_password.toStdString(), itor->str_brand.toStdString());
            SdkRuleMap ruleMap = m_lDBTool->searchSdkRuleByCamId(itor->str_cam_dev_id);
            for (auto &rule:ruleMap) {
                rule.second.strAddr = itor->str_addr;
                rule.second.weekRuleVec = m_lDBTool->searchCameraWeekRuleByCamId(itor->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;
                }
//                if (rule.second.nThreshold < 5 )
//                {
//                    rule.second.fSensitivity=0.95;
//                }
//                else if(rule.second.nThreshold > 95)
//                {
//                     rule.second.fSensitivity=0.05;
//                }
//                else
//                {
//                    rule.second.fSensitivity = 1-(float) (rule.second.nThreshold) / 100;
//                }
                DBG("fSensitivity" << rule.second.fSensitivity);
            for (int i = 0; i < startCamNO; i++) {
                itor++;
            }
            appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
            // appPref.setStringData(t_camIdex+"addr", item.str_addr.toStdString());
            json["rtsp"] = rtsp_url;
            // json["addr"] = item.str_addr.toStdString();
            INFO("cam add is " << itor->str_addr.toStdString());
            for (int i = 0; i < CamCount; i++) {
                if (itor == lst.end()) {
                    ERR("itor == lst.end()");
                    return;
                }
                auto &item = *itor;
            addCamera(t_camIdex, ruleMap);
            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");
        }
//        for (auto item : lst) {
//            std::string t_camIdex = item.str_cam_dev_id.toStdString();
//            std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554, item.str_username.toStdString(),
//                                                 item.str_password.toStdString(), item.str_brand.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)
//                {
//                    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;
//                }
////                if (rule.second.nThreshold < 5 )
////                {
////                    rule.second.fSensitivity=0.95;
////                }
////                else if(rule.second.nThreshold > 95)
////                {
////                     rule.second.fSensitivity=0.05;
////                }
////                else
////                {
////                    rule.second.fSensitivity = 1-(float) (rule.second.nThreshold) / 100;
////                }
//
//                DBG("fSensitivity" << rule.second.fSensitivity);
//            }
//            appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
//            // appPref.setStringData(t_camIdex+"addr", item.str_addr.toStdString());
//            json["rtsp"] = rtsp_url;
//            // json["addr"] = item.str_addr.toStdString();
//
//            INFO("cam add is " << item.str_addr.toStdString());
//
//            addCamera(t_camIdex, ruleMap);
//        }
    } else {
        ERR("searchCamDevTableAll size is 0");
    }
        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;