add models, daos and controllers for tables.

xuxiuxi

2019-03-04 17675f1c6447b6e014b520608ce6d5f1f2e9707a

 QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp

@@ -9,6 +9,7 @@
using std::string;

RtspAnalysManager::RtspAnalysManager(LDBTool *_dbTool) : m_lDBTool(nullptr), m_maxCount(50), m_currentCount(0) {
    INFO("MYH DEBUG HERE");
    m_lDBTool=new LDBTool;
    init();

@@ -37,8 +38,9 @@
    return rtsp_url;
}

//初始化函数
void RtspAnalysManager::init() {

    INFO("MYH DEBUG HERE")
    auto lst = m_lDBTool->searchCamDevTableAll();
    Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();

@@ -46,17 +48,51 @@
    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("CamStart") * appPref.getIntData("CamStep");
        int CamCount = appPref.getIntData("CamStep");
        INFO("StartCamNO: "<<startCamNO<<" CamStep: "<<CamCount);
        auto itor = lst.begin();

        for (auto item : lst)
        if(startCamNO >= lst.size())
        {
            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());
            INFO("cam add is " << item.str_addr.toStdString());
            ERR("startCamNO > lst.size()");
            return;
        }

        for(int i=0; i < startCamNO; i++){

            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());
            INFO("JumpCam: "<<t_camIdex<<"  URL: "<<rtsp_url);
            itor++;
        }

        for (int i = 0; i < CamCount; i++) {
            if (itor == lst.end())
            {
                ERR("itor == lst.end()");
                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());
            INFO("cam add is " << itor->str_addr.toStdString());

            addCamera(t_camIdex, rtsp_url);

            itor++;
        }

//        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());
//            INFO("cam add is " << item.str_addr.toStdString());
//
//            addCamera(t_camIdex, rtsp_url);
//
//        }
    } else {
        ERR("searchCamDevTableAll size is 0");
    }
@@ -64,22 +100,33 @@
}

RtspAnalysManager::~RtspAnalysManager() {
    INFO("Program Exit");
    for (auto controller: m_controllers) {
        INFO("Delete Controller: "<<controller.first);
        delete controller.second;
    }
    m_controllers.clear();
}

/**
 * 根据摄像机ID和RTSP路径,增加摄像机
 * 对于每个摄像机有一个 RtspImageRedisElement 对象用于保存图片到Redis
 * 有一个 RtspCaptureElement 用户录取和保存视频
 * @param index 摄像机id
 * @param rtsp 上面的摄像机对应的视频路径
 * @return -1 添加失败,0 添加成功
 */
int RtspAnalysManager::addCamera(const std::string &index, const std::string& rtsp) {
    auto lst = m_lDBTool->searchCamDevTableAll();
    Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();

    if (m_controllers.find(index) == m_controllers.end()) {
        INFO("MYH DEBUG HERE");
        if (m_currentCount >= m_maxCount) {
            ERR("addCamera faild, camera's num is full!")
            return -1;
        }
        DBG("RTSP: "<<rtsp<<"  INDEX:"<<index);
        INFO("RTSP: "<<rtsp<<"  INDEX:"<<index);
        m_imgRedisControllers[index] = new RtspImageRedisElement(index);
        m_imgRedisControllers[index]->start();

@@ -91,12 +138,19 @@

    } else {
        removeCamera(index);
        DBG("removeCamera " << index)
        return addCamera(rtsp, index);
        INFO("removeCamera " << index);
        //DBG("removeCamera " << index);
        return addCamera(index, rtsp);
    }
}

/**
 * 根据CamID移除摄像机,同时停止视频的抓取和保存图片到Redis
 * @param index 摄像机ID
 * @return 总是0
 */
int RtspAnalysManager::removeCamera(const std::string &index) {
    INFO("MYH DEBUG HERE");
    if (m_controllers.find(index) == m_controllers.end())return -1;
    auto controller = m_controllers[index];
    controller->stop();
@@ -110,10 +164,16 @@
    imgRedis->wait();
    delete imgRedis;
    m_imgRedisControllers.erase(index);
    INFO("MYH DEBUG HERE");
    return 0;
}

/**
 * 移除所有的摄像机
 * @return
 */
int RtspAnalysManager::removeAll() {
    INFO("MYH DEBUG HERE");
    for (auto controller: m_controllers) {
        controller.second->stop();
    }
@@ -122,6 +182,7 @@
        delete controller.second;
    }
    m_controllers.clear();
    INFO("MYH DEBUG HERE");


    for (auto controller: m_imgRedisControllers) {
@@ -135,16 +196,22 @@


    m_currentCount = 0;
    INFO("MYH DEBUG HERE");
    return 0;
}

//获取最大的摄像机数量
int RtspAnalysManager::getMaxCamCount() {
    return m_maxCount;
}

//获取当前的摄像机数量
int RtspAnalysManager::getCurrentCamCount() {
    return m_currentCount;
}


//录取视频的RPC的接口函数
::std::string RtspAnalysManager::recordVideo(const ::std::string& name, const ::Ice::Current&)
{
    INFO("Record Video For: "<<name);
@@ -165,14 +232,19 @@
    return name;
}


//保存视频到RtspImageRedis的队列,由RtspCaptureElement调用
bool RtspAnalysManager::SaveImageToRedis(const std::string& camId,const std::string& imageName,const cv::Mat& img)
{
    INFO("MYH DEBUG HERE");
    auto item = m_imgRedisControllers.find(camId);
    if(item!=m_imgRedisControllers.end())
    {
        INFO("Save Succeed Cam: "<<camId<<" ImageKey: "<<imageName);
        item->second->SaveImage(imageName,img);
    } else {
        ERR("Save Failed Cam: "<<camId<<" ImageKey: "<<imageName);
    }
    INFO("MYH DEBUG HERE");
    return true;
}