派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -40,6 +40,7 @@ args []*Arg // 区域内目标集合 filterData []*Arg // 过滤后区域内目标集合 time string // 当前时间（用以匹配时间规则） isEffective bool // 规则中是否用到了此区域 keepRight bool // 是否靠右行 isStatic bool // 是否静止 } @@ -90,6 +91,7 @@ RuleText string // 文字版规则组 Location []Rect // 目标的坐标 AlarmPolygon string // 触发的报警框 IsLink bool // 是否是联动任务 Others } type Others struct { @@ -120,7 +122,7 @@ SdkDataFormat(args.CameraId, arg, cameraPolygons) } // 跑本摄像机的所有规则组 一组一组跑 taskGroup := GetRuleGroup(args.CameraId, args.TaskId) // 本摄像机下所有任务组 taskGroup := GetRuleGroup(args.CameraId, args.TaskId) // 本摄像机本任务下所有规则组 //logger.Println("看下摄像机下的任务组：",taskRuleList) // 得到属于该摄像机的若干组任务的完整规则（跟每一条完整规则比较之后得出本张图像对于某个规则是否报警的结果。放进map，比如本帧图像的id，所碰撞成功的规则id） args.RuleResult = make(map[string]interface{}) @@ -188,6 +190,10 @@ ruleResult := filterRule(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { logger.Info("条件规则结果：", ruleResult.Result) // 如果结果为真，把这条规则中的区域置为有效 if strings.Contains(ruleResult.Result,"true") { areaMap.isEffective = true } // 如果此结果为真且当前过的是yolo算法，应记下此规则所对应的sdkName,另外，还要去重 (后加：把此条触碰的区域id也记录下来) if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) { sdkNames = sdkName + "," @@ -217,6 +223,9 @@ for _, areaMap := range sdkData.AreaMapList { ruleResult := transferParameters(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { if strings.Contains(ruleResult.Result,"true") { areaMap.isEffective = true } logger.Info("数量规则结果：", ruleResult.Result) if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) { sdkNames = sdkName + "," @@ -243,6 +252,9 @@ for _, areaMap := range sdkData.AreaMapList { ruleResult := timeRuleResult(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { if strings.Contains(ruleResult.Result,"true") { areaMap.isEffective = true } logger.Info("时间规则结果：", ruleResult.Result) if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) { sdkNames = sdkName + "," @@ -318,7 +330,9 @@ for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData里有人脸数据且配置了算法才把符合条件的数据塞进标签里去 for _, areaMap := range sdkData.AreaMapList { faces = append(faces, putFaceToResult(areaMap, faces)...) if areaMap.isEffective { faces = append(faces, putFaceToResult(areaMap, faces)...) } } } } @@ -336,21 +350,19 @@ } } //logger.Debug("------locations的内容：", locations) var islink bool if groupRule.SetType == "linkTask" { islink = true } else { islink = false } if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, label}) args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink,label,}) //logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result))) } if faceFlag { args.RuleResult["face"] = append(args.RuleResult["face"].([]FaceResult), FaceResult{Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []Rect{}, polygonId, label}, faces}) args.RuleResult["face"] = append(args.RuleResult["face"].([]FaceResult), FaceResult{Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []Rect{}, polygonId, islink,label,}, faces}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) logger.Info("过完规则时查看人脸标签：") if args.RuleResult["face"] != nil && len(args.RuleResult["face"].([]FaceResult)) > 0 { for _, faceResult := range args.RuleResult["face"].([]FaceResult) { for _,arg := range faceResult.Args { logger.Info("人员分值是：",arg.Score,"liker的数量为",arg.Liker) } } } } return true } else { @@ -479,26 +491,31 @@ result, _ := expression.Evaluate(nil) // 得到数学公式的结果 if result.(bool) { logger.Info("___________________________________________________________________联动任务报警") // 把数组里缓存的数据取出来一起报警 label.LinkCache = []ResultMsg{} for _, ruleRes := range TimeEleList[groupRule.GroupId].RuleResults { label.LinkCache = append(label.LinkCache, ruleRes.CacheData) } logger.Debug("联动任务缓存了几个数据", len(label.LinkCache)) for i := 0; i < len(args.RuleResult["yolo"].([]Result)); i++ { if args.RuleResult["yolo"].([]Result)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["yolo"].([]Result)[i].Others.LinkCache = label.LinkCache if TimeEleList[groupRule.GroupId] != nil { // 极偶尔有情况会等于nil，不知道为啥，做个判断以防宕机 // 把数组里缓存的数据取出来一起报警 label.LinkCache = []ResultMsg{} for _, ruleRes := range TimeEleList[groupRule.GroupId].RuleResults { label.LinkCache = append(label.LinkCache, ruleRes.CacheData) } } for i := 0; i < len(args.RuleResult["face"].([]FaceResult)); i++ { if args.RuleResult["face"].([]FaceResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["face"].([]FaceResult)[i].Others.LinkCache = label.LinkCache logger.Debug("联动任务缓存了几个数据", len(label.LinkCache)) for i := 0; i < len(args.RuleResult["yolo"].([]Result)); i++ { if args.RuleResult["yolo"].([]Result)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["yolo"].([]Result)[i].Others.LinkCache = label.LinkCache } } for i := 0; i < len(args.RuleResult["face"].([]FaceResult)); i++ { if args.RuleResult["face"].([]FaceResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["face"].([]FaceResult)[i].Others.LinkCache = label.LinkCache } } } } } } else { logger.Warn("数组不圆满不打标签") // 倒是把打的组规则标签给去掉了啊 lens := len(args.RuleResult["yolo"].([]Result))-1 args.RuleResult["yolo"] = args.RuleResult["yolo"].([]Result)[0:lens] } } else { // 没有报警， //logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为false")
