派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -1,7 +1,6 @@ package ruleserver import ( "fmt" "ruleprocess/cache" "ruleprocess/logger" "sort" @@ -71,12 +70,12 @@ CameraId string TaskId string Sdkdata []*SdkData RuleResult map[string]interface{} // 过完规则后打的标签 face: []FaceResult, yolo: []Result, cacheData: SdkDatas, timeLabel: 0,1,2,3,4,5 RuleResult map[string]interface{} // 过完规则后打的标签 face: []FaceResult, yolo: []Result } type ResultMsg struct { *protomsg.SdkMessage RuleResult map[string]interface{} // 过完规则后打的标签 RuleResult map[string]interface{} // 过完规则后打的标签 face: []FaceResult, yolo: []Result } // 过规则库打上的标签 @@ -84,11 +83,12 @@ TaskId string // 任务id SdkName string RuleGroupId string // 规则组id DefenceState bool //是否布防 DefenceState bool // 是否布防 AlarmLevel int32 // 报警等级 RuleText string // 文字版规则组 Location []Rect // 目标的坐标 AlarmPolygon string // 触发的报警框 Others map[string]interface{} // cacheData: []ResultMsg,(本组规则中含有持续时间的规则开启的定时器缓存的数据帧)，linkCache: []ResultMsg ，timeLabel: 00,01,10,11,12 } // 过规则库打上的标签 @@ -126,13 +126,14 @@ taskId := taskGroup.TaskId //logger.Println("------------本组任务下的规则组的数量：",len(ruleList)) temp := group.Rules // temp为一组完整规则 在此需要判断规则是否是联动规则 label := make(map[string]interface{}) if len(temp) > 0 { if group.SetType == "linkTask" { // groupId中含有link则为联动任务 LinkTask(args, group, taskId, message) LinkTask(args, group, taskId, message,label) } else { // 独立任务的处理 RunRule(args, group, taskId, message) RunRule(args, group, taskId, message,label) } } } @@ -140,7 +141,7 @@ } } func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage) bool { func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage,label map[string]interface{}) bool { defer func() { if err := recover(); err != nil { logger.Error("比对规则有误", err.(string)) @@ -211,7 +212,7 @@ } } } // 这步过的是时间规则（时间段等） // 这步过的是时间段规则（时间段等） for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId) @@ -267,44 +268,6 @@ result, _ := expression.Evaluate(nil) // 得到数学公式的结果 if result.(bool) { // 最后成功报警才把符合条件的人脸数据塞进结果标签里 // 配了人脸的算法才把人脸的数据甩出来打标签 faces := []Arg{} faceFlag := false for j := 0; j < len(groupRule.Rules); j++ { if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { faceFlag = true } } 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)...) } } } //for _,face := range faces { // //logger.Debug("————————————————________________看看人脸的坐标:",face.Location) //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []Rect{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { locations = append(locations, putYolosToResult(areaMap)...) } } } //logger.Debug("------locations的内容：", locations) if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId,groupRule.DefenceState,groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId}) //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},faces}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) } // 最后过持续时间等时间维度的条件 把时间规则位置调整到这个位置是为了缓存数据 for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { @@ -322,16 +285,56 @@ } } // 进行定时器的处理和判断 timeFlag := TimerAlarm(args, groupRule.GroupId, result.(bool)) timeFlag := TimerAlarm(label, groupRule.GroupId, result.(bool)) if timeFlag == "01" || timeFlag == "10" || timeFlag == "11" { // 满足定时器条件 // 打人脸标签和yolo标签 // 最后成功报警才把符合条件的人脸数据塞进结果标签里 // 配了人脸的算法才把人脸的数据甩出来打标签 faces := []Arg{} faceFlag := false for j := 0; j < len(groupRule.Rules); j++ { if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { faceFlag = true } } 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)...) } } } //for _,face := range faces { // //logger.Debug("————————————————________________看看人脸的坐标:",face.Location) //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []Rect{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { locations = append(locations, putYolosToResult(areaMap)...) } } } //logger.Debug("------locations的内容：", locations) if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId,groupRule.DefenceState,groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId,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}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) } return true } else { return false } } else { // 结果为假时也要走，有时候为假的状态反转数据也需要记录下来 timeFlag := TimerAlarm(args, groupRule.GroupId, result.(bool)) fmt.Println(timeFlag) //timeFlag := TimerAlarm(args, groupRule.GroupId, result.(bool)) //fmt.Println(timeFlag) return false } } else { @@ -371,11 +374,11 @@ } // 联动任务的处理 func LinkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage) { func LinkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage,label map[string]interface{}) { // new一个定时器，如果以此groupId为标志的定时器不存在的话 logger.Info("------------------------------------------当前是联动任务，规则是：", groupRule.GroupText) var flag bool = true var timeEle = TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId} var timeEle = TimeElement{N: 2, InitN: 2, GroupId: groupRule.GroupId} for k, timeEle1 := range TimeEleList { if k == groupRule.GroupId { flag = false // 已经有了这个定时器就置为false 不再新增 @@ -383,7 +386,7 @@ } } if flag { // 如果还没有这个定时器元素就新增一个 timeEle := TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId} // 扔进去一个定时器元素 //timeEle := TimeElement{N: 2, InitN: 2, GroupId: groupRule.GroupId} // 扔进去一个定时器元素 //TimeEleList = make(map[string]timeElement) TimeEleList[groupRule.GroupId] = &timeEle // 定时器元素以规则组id为键 logger.Info("---------------------------------------------联动任务创建了计数器并且计数器集合为：", TimeEleList) @@ -396,12 +399,12 @@ } } if flag1 { TimeEleList[groupRule.GroupId].RuleResults = append(TimeEleList[groupRule.GroupId].RuleResults, &RuleResult{groupRule.Rules[j].CameraId, groupRule.Rules[j].Sort, "", groupRule.Rules[j].RuleWithPre}) TimeEleList[groupRule.GroupId].RuleResults = append(TimeEleList[groupRule.GroupId].RuleResults, &RuleResult{groupRule.Rules[j].CameraId, groupRule.Rules[j].Sort, "", groupRule.Rules[j].RuleWithPre,&ResultMsg{message,args.RuleResult}}) } } } // 往数组里赋值 isOk := RunRule(args, groupRule, taskId, message) isOk := RunRule(args, groupRule, taskId, message,label) if isOk { logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为true") // 根据cameraId去更新或者插入结果,然后判断是否数组是否可以得出报警的结论 @@ -415,12 +418,13 @@ // 判断结果数组是否完满(即被赋值完毕)可得出报警结果 var isPerfect = true for _, va := range timeEle.RuleResults { logger.Info("---------------------瞅瞅当前数组________________：", *va) //logger.Info("---------------------瞅瞅当前数组________________：", *va) if va.Result == "" && va.RuleWithPre != "||" { isPerfect = false } } if isPerfect { logger.Debug("数组完满了，联动任务可以成功报警了！") // 将数组按sort排序 sort.Sort(SubList(timeEle.RuleResults)) // 排序后取各自的结果和连接符拼出规则表达式得出结果 @@ -434,12 +438,13 @@ result, _ := expression.Evaluate(nil) // 得到数学公式的结果 if result.(bool) { logger.Info("___________________________________________________________________联动任务报警") args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, "", groupRule.GroupId, groupRule.DefenceState,groupRule.AlarmLevel, groupRule.GroupText, []Rect{}, ""}) //logger.Info("-------------------yolo结果标签有几个", len(args.RuleResult["yolo"].([]Result))) if args.RuleResult["face"] != nil { //logger.Info("-------------------face结果标签有几个", len(args.RuleResult["face"].([]FaceResult))) // 把数组里缓存的数据取出来一起报警 label["linkCache"] = []*ResultMsg{} for _, ruleRes := range TimeEleList[groupRule.GroupId].RuleResults { label["linkCache"] = append(label["linkCache"].([]*ResultMsg),ruleRes.CacheData) } //os.Exit(0) logger.Debug("联动任务缓存了几个数据",len(label["linkCache"].([]*ResultMsg))) //os.Exit(1) } } } else { @@ -447,13 +452,20 @@ } } else { // 没有报警， //logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为false") // 所以也要去结果数组里放值或更新 for _, va := range timeEle.RuleResults { if args.CameraId != "" && va.CameraId == args.CameraId { va.Result = strconv.FormatBool(isOk) // 所以也要去结果数组里放值或更新 07/30备注： 不应放值，应删除定时器，等为true时再度开启 //for _, va := range timeEle.RuleResults { // if args.CameraId != "" && va.CameraId == args.CameraId { // va.Result = strconv.FormatBool(isOk) // } //} for k,_ := range TimeEleList { if k == groupRule.GroupId { delete(TimeEleList,k) logger.Debug("因为定时器的一帧数据结果为false，干掉定时器") } } // 因为本帧数据不符合规则，所以也不用统计结果数组里的东西 } } @@ -587,10 +599,10 @@ if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 配置的算法要对的上 if rule.SdkArgAlias == "duration" { // logger.Info("当前小规则是：---------", rule) // 先看看定时器元素队列中是否有这个摄像机这个区域的定时器,如果有就不能再次创建了 // 先看看定时器元素队列中是否有这条规则的定时器,如果有就不能再次创建了 var flag bool = true for k, _ := range TimeEleList { if k == groupId+" "+rule.Id { if k == groupId+"+"+rule.Id { flag = false // 有就置为false logger.Info("有这个定时器，不再创建了：") } @@ -600,8 +612,8 @@ timeLength, _ := strconv.Atoi(rule.SdkArgValue) timeEle := TimeElement{N: timeLength, InitN: timeLength, AlarmFlag: false, BufferFlag: 10, CacheSdkData: ResultMsg{message, args.RuleResult}} // 扔进去一个定时器元素（并缓存当前画面帧数据） //TimeEleList = make(map[string]timeElement) TimeEleList[groupId+" "+rule.Id] = &timeEle // 定时器元素以组规则id和当前持续时间小规则id为键 //logger.Info("创建了计数器并且计数器集合为：", TimeEleList) TimeEleList[groupId+"+"+rule.Id] = &timeEle // 定时器元素以当前持续时间小规则id为键 logger.Info("创建了计数器") } } }
