派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -9,7 +9,6 @@ "ruleprocess/structure" "sort" "strconv" "strings" "sync" ) @@ -37,7 +36,8 @@ // 得到属于该摄像机的若干组任务的完整规则（跟每一条完整规则比较之后得出本张图像对于某个规则是否报警的结果。放进map，比如本帧图像的id，所碰撞成功的规则id） args.RuleResult = make(map[string]interface{}) args.RuleResult["yolo"] = []structure.Result{} args.RuleResult["face"] = []structure.FaceResult{} args.RuleResult["face"] = []structure.TargetResult{} args.RuleResult["target"] = []structure.TargetResult{} //logger.Warn("传进去之前是什么德行：",args.RuleResult["yolo"]) if taskGroup != nil && len(taskGroup.GroupRules) > 0 { // 先过独立，再过联动 @@ -77,7 +77,7 @@ } } func CallMiddleware(args *structure.SdkDatas,rule protomsg.GroupRule) ([]*structure.LittleRuleResult, string, string){ func CallMiddleware(args *structure.SdkDatas,rule protomsg.GroupRule,lable *structure.Others,message *protomsg.SdkMessage) (bool, []*structure.Arg, string, string){ p,err := plugin.Open("./algorithm/middleware.so") if err != nil { panic(err) @@ -86,8 +86,8 @@ if err1 != nil { panic("没有找到中间件入口函数") } a,b,c := f.(func(args *structure.SdkDatas,rule protomsg.GroupRule)([]*structure.LittleRuleResult, string, string))(args,rule) return a,b,c a,b,c,d := f.(func(args *structure.SdkDatas,rule protomsg.GroupRule,label *structure.Others,message *protomsg.SdkMessage)(bool,[]*structure.Arg, string, string))(args,rule,lable,message) return a,b,c,d } func RunRule(args *structure.SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage, label structure.Others) (bool,[]int) { @@ -99,149 +99,121 @@ logger.Info("+++++++++++规则开始运行+++++++++++++++++当前大规则--：", (*groupRule).GroupText) //logger.Warn("传进去之后是什么德行：",args.RuleResult["yolo"]) Compare(args, groupRule) resultSplice := []*structure.LittleRuleResult{} result := false sdkNames := "" polygonId := "" resultSplice,sdkNames,polygonId = CallMiddleware(args,*groupRule) // 将数组按sort排序 sort.Sort(resultList(resultSplice)) // 排序后取各自的结果和连接符拼出规则表达式得出结果 completeFormula := "" for _, va := range resultSplice { completeFormula = completeFormula + va.Result } if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") || strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") { // 以这些开头的基本是联动任务 if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") { completeFormula = completeFormula[2:] targets := []*structure.Arg{} // 符合条件的目标 // 把一帧数据和一组规则发给算法部分，得出判断结果 result,targets,sdkNames,polygonId = CallMiddleware(args,*groupRule,&label,message) logger.Info("符合条件的目标数据：",targets) if result { // 最后过持续时间等时间维度的条件 把时间规则位置调整到这个位置是为了缓存数据 ！！！！！ps: 对画面中单个目标做定时器的不用再过画面定时器 cacheId := "" for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId) if err != nil { logger.Error("没查到sdk的信息---", err) } ipcId := sdk.IpcId if ipcId == sdkData.IpcId { for _, areaMap := range sdkData.AreaMapList { // 去开启一个定时器 cacheid := duration(groupRule.Rules[j], groupRule.GroupId, areaMap, args, message) if cacheid != "" { cacheId = cacheid } } } } } if strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") { completeFormula = completeFormula[1:] } logger.Info("-------------------看看拔毛后的表达式：", completeFormula) //expression, _ := govaluate.NewEvaluableExpression(completeFormula) //result, _ := expression.Evaluate(nil) // 得到数学公式的结果 //return result.(bool) } if completeFormula != "" { logger.Info("结果公式-----------：", completeFormula) expression, err2 := govaluate.NewEvaluableExpression(completeFormula) if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || err2 != nil { panic("规则有误，得到的数学公式不可解析") } result, _ := expression.Evaluate(nil) // 得到数学公式的结果 if result.(bool) { // 最后过持续时间等时间维度的条件 把时间规则位置调整到这个位置是为了缓存数据 cacheId := "" // 进行定时器的处理和判断 timeFlag := TimerAlarm(&label, groupRule.GroupId, result) if timeFlag == "01" || timeFlag == "10" || timeFlag == "11" || cacheId != ""{ // 没有定时器或者满足定时器条件 // 打人脸标签和yolo标签 // 最后成功报警才把符合条件的人脸数据塞进结果标签里 // 配了人脸的算法才把人脸的数据甩出来打标签 faces := []structure.Arg{} faceFlag := false for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId) if err != nil { logger.Error("没查到sdk的信息---", err) } ipcId := sdk.IpcId if ipcId == sdkData.IpcId { for _, areaMap := range sdkData.AreaMapList { // 去开启一个定时器 cacheid := duration(groupRule.Rules[j], groupRule.GroupId, areaMap, args, message) if cacheid != "" { cacheId = cacheid } 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 { if areaMap.IsEffective { faces = append(faces, putFaceToResult(areaMap, faces)...) } } } } // 进行定时器的处理和判断 timeFlag := TimerAlarm(&label, groupRule.GroupId, result.(bool)) if timeFlag == "01" || timeFlag == "10" || timeFlag == "11" || cacheId != ""{ // 没有定时器或者满足定时器条件 // 打人脸标签和yolo标签 // 最后成功报警才把符合条件的人脸数据塞进结果标签里 // 配了人脸的算法才把人脸的数据甩出来打标签 faces := []structure.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 { if areaMap.IsEffective { faces = append(faces, putFaceToResult(areaMap, faces)...) } //logger.Info("face标签的长度：",len(faces)) //for _,face := range faces { // //logger.Debug("————————————————________________看看人脸的坐标:",face.Location) //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []structure.TargetInfo{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { if areaMap.IsEffective { locations = append(locations, putYolosToResult(areaMap)...) } } } //logger.Info("face标签的长度：",len(faces)) //for _,face := range faces { // //logger.Debug("————————————————________________看看人脸的坐标:",face.Location) //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []structure.TargetInfo{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { if areaMap.IsEffective { locations = append(locations, putYolosToResult(areaMap)...) } } } } //logger.Debug("------locations的内容：", locations) var islink bool if groupRule.SetType == "linkTask" { islink = true } else { islink = false } var labelTypes []int // 0为yolo标签，1为face标签 2为两者标签 if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink,label,}) labelTypes = append(labelTypes,0) //logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result))) } if faceFlag { args.RuleResult["face"] = append(args.RuleResult["face"].([]structure.FaceResult), structure.FaceResult{structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []structure.TargetInfo{}, polygonId, islink,label,}, faces}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) labelTypes = append(labelTypes,1) } // 给持续时间的第一张赋予缓存数据（遍历复制） if cacheId != "" { // 有这帧数据的缓存 tempMap := make(map[string]interface{}) for k, result := range args.RuleResult { if k == "yolo" { tempMap[k] = []structure.Result{} for _, res := range result.([]structure.Result) { tempMap[k] = append(tempMap[k].([]structure.Result), res) } } if k == "face" { tempMap[k] = []structure.FaceResult{} for _, res := range result.([]structure.FaceResult) { tempMap[k] = append(tempMap[k].([]structure.FaceResult), res) } } } rw.Lock() TimeEleList[cacheId].CacheSdkData.RuleResult = tempMap rw.Unlock() } return true,labelTypes } logger.Debug("------locations的内容：", locations) var islink bool if groupRule.SetType == "linkTask" { islink = true } else { return false,[]int{} islink = false } var labelTypes []int // 0为yolo标签，1为face标签 2为两者标签 if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink,label}) labelTypes = append(labelTypes,0) //logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result))) } if faceFlag { args.RuleResult["face"] = append(args.RuleResult["face"].([]structure.TargetResult), structure.TargetResult{structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []structure.TargetInfo{}, polygonId, islink,label}, faces}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) labelTypes = append(labelTypes,1) } // 给持续时间的第一张赋予缓存数据（遍历复制） if cacheId != "" { // 有这帧数据的缓存 tempMap := make(map[string]interface{}) for k, result := range args.RuleResult { if k == "yolo" { tempMap[k] = []structure.Result{} for _, res := range result.([]structure.Result) { tempMap[k] = append(tempMap[k].([]structure.Result), res) } } if k == "face" { tempMap[k] = []structure.TargetResult{} for _, res := range result.([]structure.TargetResult) { tempMap[k] = append(tempMap[k].([]structure.TargetResult), res) } } } rw.Lock() TimeEleList[cacheId].CacheSdkData.RuleResult = tempMap rw.Unlock() } return true,labelTypes } else { // 结果为假时也要走，有杀死定时器的操作 TimerAlarm(&label, groupRule.GroupId, result.(bool)) //fmt.Println(timeFlag) return false,[]int{} } } else { // 结果为假时也要走，有杀死定时器的操作 TimerAlarm(&label, groupRule.GroupId, result) //fmt.Println(timeFlag) return false,[]int{} } } @@ -335,9 +307,9 @@ } } if k == "face" { tempMap[k] = []structure.FaceResult{} for _, res := range result.([]structure.FaceResult) { tempMap[k] = append(tempMap[k].([]structure.FaceResult), res) tempMap[k] = []structure.TargetResult{} for _, res := range result.([]structure.TargetResult) { tempMap[k] = append(tempMap[k].([]structure.TargetResult), res) } } } @@ -381,9 +353,9 @@ args.RuleResult["yolo"].([]structure.Result)[i].Others.LinkCache = label.LinkCache } } for i := 0; i < len(args.RuleResult["face"].([]structure.FaceResult)); i++ { if args.RuleResult["face"].([]structure.FaceResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["face"].([]structure.FaceResult)[i].Others.LinkCache = label.LinkCache for i := 0; i < len(args.RuleResult["face"].([]structure.TargetResult)); i++ { if args.RuleResult["face"].([]structure.TargetResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上 args.RuleResult["face"].([]structure.TargetResult)[i].Others.LinkCache = label.LinkCache } } } @@ -401,9 +373,9 @@ } } if val == 1 { if len(args.RuleResult["face"].([]structure.FaceResult)) >= 1 { lens := len(args.RuleResult["face"].([]structure.FaceResult))-1 args.RuleResult["face"] = args.RuleResult["face"].([]structure.FaceResult)[0:lens] if len(args.RuleResult["face"].([]structure.TargetResult)) >= 1 { lens := len(args.RuleResult["face"].([]structure.TargetResult))-1 args.RuleResult["face"] = args.RuleResult["face"].([]structure.TargetResult)[0:lens] } } }
