#ifndef GLOBALINFO_H #define GLOBALINFO_H #include #include #include #include #include "nlohmann/json.hpp" /* ==================================================================================== * ******************************* 全局变量定义 ********************************** * ====================================================================================*/ extern int g_eventTimeVaild; /* 线程运行时的状态 */ enum class RunTimeState { RUN_STATE_INIT = 0, /* 初始化 */ RUN_STATE_RUN, /* 运行 */ RUN_STATE_STOP, /* 停止 */ RUN_STATE_ERROR, /* 错误 */ RUN_STATE_NONE /* 无状态 */ }; /* 应用功能 */ enum class AppFunction { APP_NONE = 0, /* 无功能 */ APP_ONWORK = 1, /* 人员在岗识别 */ APP_CONTRABAND, /* 违禁品识别 */ APP_ILLEGAL, /* 非法入侵检测 */ APP_FATIGUE, /* 疲劳检测 */ APP_REGIONAL, /* 区域人员检测 */ APP_MOUSE, /* 老鼠识别 */ APP_PLAYPHONE, /* 玩手机识别 */ APP_NOMASK, /* 无口罩识别 */ APP_ALLDOWN, /* 摔倒识别 */ }; /** * @brief 全局信息 * */ using nJson = nlohmann::json; /* 算法相关信息 */ struct AlgorithmInfo { int PKID; /* 主键ID */ int ActionTaskID; /* 算法任务ID */ std::string ActionID; /* 算法ID */ std::string ActionName; /* 算法名称 */ AlgorithmInfo() = default; AlgorithmInfo(const AlgorithmInfo& other) : PKID(other.PKID), ActionID(other.ActionID), ActionName(other.ActionName), ActionTaskID(other.ActionTaskID) {} AlgorithmInfo& operator=(const AlgorithmInfo& other) { if (this != &other) { ActionID = other.ActionID; ActionName = other.ActionName; ActionTaskID = other.ActionTaskID; PKID = other.PKID; } return *this; } /* 对比是否相等 */ bool operator==(const AlgorithmInfo& other) const { if(ActionID != other.ActionID) { return false; } if(ActionName != other.ActionName || ActionTaskID != other.ActionTaskID) { return false; } return true; } }; /* 设备列表 */ struct DeviceInfo { int PKID; /* 主键ID */ int DeviceID; /* 设备ID */ int DevicePort; /* 设备端口 */ std::string DeviceName; /* 设备名称 */ std::string DeviceIP; /* 设备IP */ std::string DeviceSerial; /* 设备序列号 */ std::string DeviceType; /* 设备类型 */ std::string UserAccount; /* 用户账号 */ std::string UserPassword; /* 用户密码 */ std::vector vecAlgorithmInfo; /* 算法信息 */ DeviceInfo() = default; DeviceInfo(const DeviceInfo& other) : PKID(other.PKID), DeviceID(other.DeviceID), DevicePort(other.DevicePort), DeviceName(other.DeviceName), DeviceSerial(other.DeviceSerial), DeviceType(other.DeviceType), DeviceIP(other.DeviceIP), UserAccount(other.UserAccount), UserPassword(other.UserPassword), vecAlgorithmInfo(other.vecAlgorithmInfo) {} DeviceInfo& operator=(const DeviceInfo& other) { if (this != &other) { DeviceID = other.DeviceID; DevicePort = other.DevicePort; DeviceName = other.DeviceName; DeviceSerial = other.DeviceSerial; DeviceType = other.DeviceType; DeviceIP = other.DeviceIP; UserAccount = other.UserAccount; UserPassword = other.UserPassword; vecAlgorithmInfo = other.vecAlgorithmInfo; } return *this; } /* 对比是否相等 */ bool operator==(const DeviceInfo& other); /* 对比设备关联的算法信息是否相等 */ bool isEqualAlgorithmInfo(const std::vector& other); }; /* 算法和设备关联的相关信息 */ struct DeviceAlgorithmInfo { int DeviceID; /* 设备ID */ int ActionTaskID; /* 算法任务ID */ std::string ActionID; /* 算法ID */ std::string ActionName; /* 算法名称 */ }; /** * @brief 摄像机线程需要的信息 * */ struct CameraThreadInfo { std::string RedisIP; /* Redis IP */ int RedisPort; /* Redis Port */ std::string RedisPWD; /* Redis Password */ int DeviceID; /* 设备ID */ std::vector vecAction; /* Redis Key,一个设备可能会有多个算法ID */ CameraThreadInfo() = default; CameraThreadInfo& operator=(const CameraThreadInfo& other) { if (this != &other) { RedisIP = other.RedisIP; RedisPort = other.RedisPort; RedisPWD = other.RedisPWD; DeviceID = other.DeviceID; vecAction = other.vecAction; } return *this; } }; /** * @brief 人员信息 * */ struct PersonInfo { std::string PersonID; /* 人员ID */ std::string PersonName; /* 人员名称 */ }; /** * @brief 报警信息 * */ struct AlarmInfo { bool Is_Alarm; /* 是否报警 */ int AlarmID; /* 报警ID */ int DeviceID; /* 设备ID,数据库表格中对应的是CamerID */ int RoomID; /* 房间ID */ int ChannelID; /* 通道ID */ int State; /* 状态 */ int OnWork; /* 是否在工作 */ std::string ActionID; /* 算法ID */ std::string StartTime; /* 报警开始时间 */ std::string EndTime; /* 报警结束时间 */ std::string EventTime; /* 事件时间(报警发生时间,在数据库里对应的是CreateTime) */ std::string PicUrl; /* 报警图片URL */ std::string ImageInfo; /* 图片信息 */ std::string AppID; /* 客户端应用ID */ std::string ActionDes; /* 算法描述信息 */ std::string FaceIDList; /* 人脸ID列表 */ std::string FaceNameList; /* 人脸名称列表 */ std::string BboxList; /* Bbox列表,应该是图片中的报警位置 */ std::vector vecPersonInfo; /* 人员信息 */ AlarmInfo(); AlarmInfo& operator=(AlarmInfo& other); void reInit(); }; /** * @brief 报警规则 * */ struct AlarmRuleInfo { bool LiveMinEnable; /* 启用直播间最小人数 */ bool LiveMaxEnable; /* 启用直播间最大人数 */ bool DicMinEnable; /* 启用导播间最小人数 */ bool DicMaxEnable; /* 启用导播间最大人数 */ bool LiveDicMinEnable; /* 启用直播间和导播间最小人数 */ bool LiveDicMaxEnable; /* 启用直播间和导播间最大人数 */ int LiveMin; /* 直播间最小人数 */ int LiveMax; /* 直播间最大人数 */ int DicMin; /* 导播间最小人数 */ int DicMax; /* 导播间最大人数 */ int LiveDicMin; /* 直播间和导播间最小人数 */ int LiveDicMax; /* 直播间和导播间最大人数 */ std::string RuleName; /* 规则名称 */ AlarmRuleInfo(); AlarmRuleInfo& operator=(AlarmRuleInfo& other); }; /** * @brief 房间和摄像机关联信息,这个是从客户端配置的 * 一个房间对应多个摄像机 * */ struct RoomCameraInfo { int RoomID; /* 房间ID */ std::list listCameraID; /* 摄像机列表 */ RoomCameraInfo() = default; RoomCameraInfo& operator=(const RoomCameraInfo& other) { if (this != &other) { RoomID = other.RoomID; listCameraID = other.listCameraID; } return *this; } /* 这里只对比RoomID */ bool operator==(const RoomCameraInfo& other) const { if(RoomID != other.RoomID) { return false; } return true; } }; /* ==================================================================================== * **************************** 线程功能需要的结构体 ***************************** * ====================================================================================*/ /** * @brief 单个算法信息,包括包含的摄像机,房间、频率信息 * */ struct ActionInfo { RunTimeState RunState; /* 线程运行状态 */ int ChannelID; /* 通道ID */ int RoomID; /* 房间ID */ int CameraID; /* 摄像机ID */ int RoomType; /* 房间类型 */ std::string ActionID; /* 算法ID */ std::string strRoomName; /* 房间名称 */ std::string strActionName; /* 算法名称,这个是自定义的,不需要对比 */ ActionInfo(); ActionInfo& operator=(const ActionInfo& other); /* 对比是否相等,这个需要完全相等,包括算法信息和摄像机信息 */ bool operator==(const ActionInfo& other); /* 对比除摄像机外的基础信息是否相等 */ bool isEqualBaseInfo(const ActionInfo& other); }; /** * @brief 算法信息列表,这个列表既可以存储房间和算法ID的关联信息,也可以用来存储运行线程的信息 * */ class ListActionInfo { public: ListActionInfo() = default; /* 添加关联信息,会自动进行重复判断,如果已有相同的信息,则跳过 */ bool insertActionInfo(ActionInfo* info); /* 删除信息 */ bool deleteActionInfo(ActionInfo* info); /* 给算法添加摄像机,原有的会被替换掉 */ bool addActionCamera(ActionInfo* pInfo); /* 清空算法中的摄像机信息 */ void clearCameraList(); /* 清空设置成STOP或ERROR的Action */ void clearStopAction(); /* 查找算法ID是否已在列表中 */ ActionInfo* findActionInList(ActionInfo* pInfo); /* 查找算法ID是否在列表中,这里查找不会对比摄像机ID */ ActionInfo* findActionIDInListNoCamera(ActionInfo* pInfo); /* 获取容器 */ std::list& getData() { return listActionInfo; } /* 清空容器 */ void clear(); std::list listActionInfo; /* 房间和算法ID关联列表 */ }; /** * @brief 算法信息,以房间分类,算法包含在这干房间内所需要的摄像机ID * */ struct RoomActionInfo { RunTimeState RunState; /* 线程运行状态 */ int ChannelID; /* 通道ID */ int RoomID; /* 房间ID */ int RoomType; /* 房间类型 */ std::string ActionID; /* 算法ID */ std::string strRoomName; /* 房间名称 */ std::list listCameraID; /* 摄像机ID */ RoomActionInfo() { RunState = RunTimeState::RUN_STATE_INIT; ChannelID = -1; RoomID = -1; RoomType = -1; ActionID = ""; strRoomName = ""; listCameraID.clear(); } RoomActionInfo(const RoomActionInfo& other) : ChannelID(other.ChannelID), RoomID(other.RoomID), RoomType(other.RoomType), ActionID(other.ActionID), strRoomName(other.strRoomName), listCameraID(other.listCameraID) {} RoomActionInfo& operator=(const RoomActionInfo& other) { if (this != &other) { ChannelID = other.ChannelID; RoomID = other.RoomID; RoomType = other.RoomType; ActionID = other.ActionID; strRoomName = other.strRoomName; listCameraID = other.listCameraID; } return *this; } /* 对比频道信息、房间信息、算法ID是否相等 */ bool isEqualBaseInfo(const RoomActionInfo& other); }; /* 房间和算法ID关联列表 */ class ListRoomActionInfo { public: ListRoomActionInfo() = default; /* 添加关联信息,会自动进行重复判断,如果已有相同的room和action关联信息,则跳过 */ bool insertRoomActionInfo(const RoomActionInfo& info); /* 添加关联信息,会自动进行重复判断,如果已有相同的room和action关联信息,则跳过 */ bool insertRoomActionInfo(const int ChannelID, const int RoomID, const std::string& strActionID); /* 删除一个容器,注意,这个不能在别的for循环中删除,只能单独删除 */ bool deleteRoomActionInfo(RoomActionInfo* pInfo); /* 清空容器 */ void clear(); /* 添加算法信息,根据传入的算法信息,自动将其加入到对应的容器中 */ void addActionInfo(const ActionInfo& info); /* 清空算法对应的摄像机列表 */ void clearCameraList(); /* 清理设置为STOP或者ERROR的RoomAction */ void clearStopRoomAction(); /* 查找算法ID是否已在列表中 */ RoomActionInfo* findActionIDInList(const int chnID, const int RoomID, const std::string& strActionID); /* 获取容器 */ std::list& getData() { return listRoomActionInfo; } std::list listRoomActionInfo; /* 房间和算法ID关联列表 */ }; /** * @brief 房间内的摄像机和算法关联信息 * */ struct RoomCamActInfo { int RoomID; /* 房间ID */ int RoomType; /* 房间类型 */ std::multimap mapCameraAction; /* 摄像机ID和算法ID关联信息 */ }; /** * @brief 按照功能分类的线程信息,一个实例就是一个功能, 比如区域入侵检测,需要多个算法多个摄像机共同配合 * */ struct FuncActionInfo { int ChannelID; /* 通道ID */ AppFunction appFunction; /* 任务功能 */ RunTimeState RunState; /* 线程运行状态 */ std::string strFunctionName; /* 功能名称 */ std::chrono::system_clock::time_point StartTime; /* 开始时间 */ std::chrono::system_clock::time_point EndTime; /* 结束时间 */ std::list listRoomCamActInfo; /* 房间内的摄像机和算法关联信息 */ FuncActionInfo(); FuncActionInfo& operator=(FuncActionInfo& other); /* 添加算法信息 */ bool addActionInfo(const ActionInfo& info); }; /** * @brief * */ struct ListFuncActInfo { ListFuncActInfo() = default; /* 添加功能信息 */ bool addFuncActionInfo(FuncActionInfo* pInfo); /* 添加算法信息 */ bool addActionInfo(const ActionInfo& info); /* 清空无用的功能信息 */ void clearNoneFuncActionInfo(); /* 查找功能信息 */ FuncActionInfo* findFuncActionInfo(const std::string& strFunctionName); /* 获取容器 */ std::list& getData() { return listFuncActionInfo; } std::list listFuncActionInfo; /* 功能信息列表 */ }; /** * @brief 读取配置文件 * */ class GlobalConfig { public: GlobalConfig(); ~GlobalConfig() = default; /* 读取配置文件 */ bool readConfig(const QString& strConfigFile); /* 打印读取到的值 */ void printValue(); int AppPeopleOnWork; /* 离岗时间 */ int Contraband; /* 违禁物品出现的时间 */ int AppBadMan; /* 非法入侵 */ int AppTired; /* 疲劳检测时间 */ int AppPeopleCont; /* 区域人员统计 */ int AppPlayPhone; /* 玩手机识别 */ int AppMouse; /* 老鼠识别 */ int AppMask; /* 戴口罩识别 */ int CheckSet; /* 服务端多久检测一次配置 */ int EventTimeValid; /* 事件时间有效期 */ int ThreadSleepMS; /* 任务线程休眠时间,单位是ms */ std::string Key; /* Key */ std::string Secret; /* Secret */ }; extern GlobalConfig g_config; #endif /* GLOBALINFO_H */