派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -163,22 +163,9 @@ // 根据摄像机id拿到摄像机所有区域 func GetPolygons(cameraId string) []protomsg.CameraPolygon { //var api dbapi.CameraApi //data := api.FindAllPolygons() ////logger.Info("查到的所有区域：", data) //// 根据id从map中拿到区域 var cameraPolygons []protomsg.CameraPolygon //for _, item := range data { // if item.CameraId == cameraId && item.Type != "line" { // // 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小 x坐标分别*image.width/页面区域宽 y坐标分别*image.height/页面区域高 // // 前台宽高固定 // //logger.Println("------------符合条件的区域：",item) // cameraPolygons = append(cameraPolygons, item) // } //} cameraPolygons = cache.GetPolygonsByCameraId(cameraId) //logger.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为：", len(cameraPolygons)) return cameraPolygons } @@ -197,7 +184,7 @@ } } // 对单帧图像的判断 是舍弃（或者说对于某些需求可以放ES数据库一份）还是返回 // 对单帧图像的判断 thisSdkDatas 当前传入的这帧数据，cacheSdkData 定时器里缓存的一帧数据 没有就返回nil (thisSdkDatas SdkDatas, cacheSdkDatas SdkDatas) func Judge(args *SdkDatas) { if len(args.Sdkdata) > 0 { // 拿到本摄像机的区域 @@ -256,11 +243,14 @@ logger.Info("没有配置对比底库参数") return } logger.Info("============================================进入compare") for j := 0; j < len(groupRule.Rules); j++ { // 只有是人脸对比并且配置的参数是对比底库才走（若配的对比阈值则不走） if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { logger.Info("============================================遍历到人脸比对规则") for _, sdkData := range args.Sdkdata { logger.Info("==========================================IpcId",sdkData.IpcId) if sdkData.IpcId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 搜索到人脸检测的sdkdata 现在关于人脸的只有他一个sdk，全找他 logger.Info("============================================第一关") for _, areaMap := range sdkData.AreaMapList { // 拿区域中每个人脸特征值去对比，填充其liker if groupRule.Rules[j].PolygonId == areaMap.areaId { @@ -273,6 +263,7 @@ logger.Error("getBaseInfo解压错误", err1) } for key, val := range m { logger.Info("_______________________________________对比库的返回值",key,val) baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { logger.Error("查询底库人员信息出错", err1) @@ -404,7 +395,7 @@ ipcId := sdk.IpcId if ipcId == sdkData.IpcId { for _, areaMap := range sdkData.AreaMapList { duration(groupRule.Rules[j], &areaMap) duration(groupRule.Rules[j], &areaMap, args) } } } @@ -538,12 +529,6 @@ areaPoints := Json2points(a.areaJson) widthScale := float64(arg.ImageWidth / 960) heigthScale := float64(arg.ImageHeight / 540) // for _, sdkInfo := range arg.SdkMessage.Tasklab.Sdkinfos { // if sdkInfo.Sdktype == "yolo" { // //sdkInfo.Sdkdata. // } // } for _, obj := range arg.Photo { //logger.Info("------------------看看sdkData:", arg.SdkName, "的Photo数据----------------", obj, "----顺便看看占比-----：", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale)) if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) { @@ -581,17 +566,6 @@ // 以摄像机id查出跟其相关的所有任务下的所有规则组 func GetRuleGroup(cameraId string) []*protomsg.TaskGroupArgs { // 查询数据库 // 第一步查出跟这个摄像机相关的group_id(大规则) //var api dbapi.CameraTaskArgsApi // ////logger.Info("所有规则:", all) //var taskArgs []*protomsg.TaskGroupArgs //for _, camArg := range all { // if camArg.CameraId == cameraId { // taskArgs = camArg.TaskArgs // } //} all := cache.GetCameraTaskRulesByCameraId(cameraId) return all } @@ -661,9 +635,9 @@ if result.(bool) { logger.Info("___________________________________________________________________联动任务报警") args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, "", groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText, []Rect{}}) logger.Info("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result)) logger.Info("-------------------yolo结果标签有几个", len(args.RuleResult["yolo"].([]Result))) if args.RuleResult["face"] != nil { logger.Info("-------------------face结果标签", args.RuleResult["face"].([]Arg)) logger.Info("-------------------face结果标签有几个", len(args.RuleResult["face"].([]Arg))) } //os.Exit(0) } @@ -675,7 +649,7 @@ logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为false") // 所以也要去结果数组里放值或更新 for _, va := range timeEle.RuleResults { if args.CameraId != "" && va.CameraId == args.CameraId { // arg.CameraId 随便找一个数据 if args.CameraId != "" && va.CameraId == args.CameraId { va.Result = strconv.FormatBool(isOk) } } @@ -739,7 +713,7 @@ } // 先清空过滤后的数据，再往里塞本次过滤后的数据 am.filterData = am.filterData[0:0] logger.Info("-----------------------人脸过滤的args：", args) //logger.Info("-----------------------人脸过滤的args：", args) for _, arg := range args { var formula string if rule.SdkArgAlias == "score" { @@ -824,7 +798,7 @@ } // 都过滤完条件之后看看是否要创建一个定时器元素 创建定时器的条件：是否有靠右行，个体静止等自带定时器含义的算法以及是否有持续时间 func duration(rule *protomsg.Rule, am *AreaMap) { func duration(rule *protomsg.Rule, am *AreaMap, args *SdkDatas) { if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 配置的算法要对的上 if rule.SdkArgAlias == "duration" { // // 先看看定时器元素队列中是否有这个摄像机这个区域的定时器,如果有就不能再次创建了 @@ -838,7 +812,7 @@ if flag { timeLength, _ := strconv.Atoi(rule.SdkArgValue) timeEle := TimeElement{N: timeLength, InitN: timeLength} // 扔进去一个定时器元素 timeEle := TimeElement{N: timeLength, InitN: timeLength, AlarmFlag: false, CacheSdkData: *args} // 扔进去一个定时器元素 //TimeEleList = make(map[string]timeElement) TimeEleList[am.taskId+" "+am.sdkId+" "+am.areaId] = &timeEle // 定时器元素以摄像机id拼接区域id为键 logger.Info("创建了计数器并且计数器集合为：", TimeEleList) @@ -846,21 +820,6 @@ } } } //// 冗余拼接一个目标数量大于0 //func splice1(am *AreaMap) string { // var num int = 0 // for _, data := range am.filterData { // if data.isYolo { // num++ // } // } // logger.Println("看看区域内符合条件的目标数量：----------", num) // formula := strconv.Itoa(num) + " " + ">" + "0" // expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式 // result, _ := expression.Evaluate(nil) // 得到数学公式的结果 // return strconv.FormatBool(result.(bool)) //} // 给数据库的规则表达式代参 args: 一条子规则，区域数据 func transferParameters(rule *protomsg.Rule, am *AreaMap) LittleRuleResult { @@ -977,14 +936,6 @@ // 取出某个时间规则的第几天的规则段集合 func GetTimeById(id string, index int) []TimeRange { //var cameraTimeRule protomsg.CameraTimerule //var api dbapi.CameraApi //_, rules := api.FindAllTimeRules() //for _, rule := range rules { // if rule.Id == id { // cameraTimeRule = rule // } //} _, cameraTimeRule := cache.GetTimeRuleById(id) var timeRangeList []day err := json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList)
