派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -1,26 +1,31 @@ package ruleserver import ( "basic.com/pubsub/protomsg.git" "basic.com/valib/logger.git" "github.com/knetic/govaluate" "plugin" "ruleprocess/cache" "ruleprocess/logger" "ruleprocess/structure" "sort" "strconv" "strings" "time" "basic.com/pubsub/protomsg.git" "github.com/knetic/govaluate" "sync" ) var rw sync.RWMutex // 读写锁 // 对单帧图像的判断 thisSdkDatas 当前传入的这帧数据，cacheSdkData 定时器里缓存的一帧数据 没有就返回nil (thisSdkDatas SdkDatas, cacheSdkDatas SdkDatas) func Judge(args *structure.SdkDatas, message *protomsg.SdkMessage) { defer func() { if err := recover(); err != nil { logger.Error("规则模块儿的异常捕获：",err) } }() if len(args.Sdkdata) > 0 { // 拿到本摄像机的区域 cameraPolygons := GetPolygons(args.CameraId) // 把所有的sdk提取的数据都按所属摄像机的区域归置 //logger.Debug("当前摄像机id为：",message.Cid,"当前摄像机执行的任务是：",message.Tasklab.Taskname) logger.Debug("当前摄像机id为：",message.Cid,"当前摄像机执行的任务是：",message.Tasklab.Taskname,"--任务id为：",message.Tasklab.Taskid) for _, arg := range args.Sdkdata { SdkDataFormat(args.CameraId, arg, cameraPolygons) } @@ -64,83 +69,8 @@ } } func RunRule1(args *structure.SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage, label structure.Others) bool { defer func() { if err := recover(); err != nil { logger.Error("比对规则有误", err) } }() logger.Info("+++++++++++规则开始运行+++++++++++++++++当前大规则--：", (*groupRule).GroupText) //logger.Warn("传进去之后是什么德行：",args.RuleResult["yolo"]) Compare(args, groupRule) resultSplice := []*structure.LittleRuleResult{} sdkNames := "" polygonId := "" // 先过完条件规则 for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { // 根据规则的sdkId查出其对应的ipcId，用ipcId去找该比对的数据 sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId) if err != nil { logger.Error("没查到sdk的信息---", err) } ipcId := sdk.IpcId sdkName := sdk.SdkName //logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId) if ipcId == sdkData.IpcId { //logger.Info("当前走的规则是--：", sdkName, "---","") for _, areaMap := range sdkData.AreaMapList { ruleResult := CallSo(sdk.Id,groupRule.Rules[j],areaMap) //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 + "," polygonId = groupRule.Rules[j].PolygonId + "," } if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(polygonId, groupRule.Rules[j].PolygonId) { polygonId = groupRule.Rules[j].PolygonId + "," } resultSplice = append(resultSplice, &ruleResult) } } } } } return false } func CallSo(sdkId string,rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult{ // 根据sdkId查出其对应的sdk的soName，调用相应so的Entrance方法 var soName = "" if sdkId == "812b674b-2375-4589-919a-5c1c3278a97e" { soName = "face.so" } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a975"{ soName = "intrusion.so" } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a976" { soName = "personUnsual.so" } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a972" { soName = "faceCompare.so" } p,err := plugin.Open("./algorithm/"+soName) if err != nil { panic(err) } f,err1 := p.Lookup("Entrance") if err1 != nil { panic("没有找到入口函数") } ruleResult := f.(func(rule *protomsg.Rule, am *structure.AreaMap)structure.LittleRuleResult)(rule,am) return ruleResult } func CallMiddleware(args *structure.SdkDatas,rule protomsg.GroupRule) ([]*structure.LittleRuleResult, string, string){ p,err := plugin.Open("./algorithm/middleware") p,err := plugin.Open("./algorithm/middleware.so") if err != nil { panic(err) } @@ -148,10 +78,10 @@ if err1 != nil { panic("没有找到中间件入口函数") } _,_,_ := f.(func(args *structure.SdkDatas,rule protomsg.GroupRule)([]*structure.LittleRuleResult, string, string))(args,rule) return nil,"","" a,b,c := f.(func(args *structure.SdkDatas,rule protomsg.GroupRule)([]*structure.LittleRuleResult, string, string))(args,rule) return a,b,c } func RunRule(args *structure.SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage, label structure.Others) bool { func RunRule(args *structure.SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage, label structure.Others) (bool,[]int) { defer func() { if err := recover(); err != nil { logger.Error("比对规则有误", err) @@ -231,12 +161,13 @@ } } } logger.Info("face标签的长度：",len(faces)) //for _,face := range faces { // //logger.Debug("————————————————________________看看人脸的坐标:",face.Location) //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []structure.Rect{} locations := []structure.TargetInfo{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { @@ -251,27 +182,30 @@ } 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,}) 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.Rect{}, polygonId, islink, label,}, faces}) 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) } return true return true,labelTypes } else { return false return false,[]int{} } } else { // 结果为假时也要走，有时候为假的状态反转数据也需要记录下来 // 结果为假时也要走，有杀死定时器的操作 TimerAlarm(&label, groupRule.GroupId, result.(bool)) //fmt.Println(timeFlag) return false return false,[]int{} } } else { return false return false,[]int{} } } @@ -294,11 +228,15 @@ return faces } func putYolosToResult(am *structure.AreaMap) []structure.Rect { locations := []structure.Rect{} func putYolosToResult(am *structure.AreaMap) []structure.TargetInfo { locations := []structure.TargetInfo{} if len(am.FilterData) > 0 { for _, data := range am.FilterData { locations = append(locations, data.Location) location := structure.TargetInfo{} location.Rect = data.Location location.TargetId = data.Id location.TargetScore = data.Score locations = append(locations, location) } } //logger.Println("-----------------------------------------------听说你是空的？",faces) @@ -311,6 +249,7 @@ logger.Info("------------------------------------------当前是联动任务，规则是：", groupRule.GroupText) var flag bool = true var timeEle = TimeElement{N: 2, InitN: 2, GroupId: groupRule.GroupId} rw.Lock() for k, timeEle1 := range TimeEleList { if k == groupRule.GroupId { flag = false // 已经有了这个定时器就置为false 不再新增 @@ -335,8 +274,9 @@ } } } rw.Unlock() // 往数组里赋值 isOk := RunRule(args, groupRule, taskId, message, label) isOk,labelTypes := RunRule(args, groupRule, taskId, message, label) if isOk { logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为true") // 根据cameraId去更新或者插入结果,然后判断是否数组是否可以得出报警的结论 @@ -386,6 +326,7 @@ result, _ := expression.Evaluate(nil) // 得到数学公式的结果 if result.(bool) { logger.Info("___________________________________________________________________联动任务报警") rw.RLock() if TimeEleList[groupRule.GroupId] != nil { // 极偶尔有情况会等于nil，不知道为啥，做个判断以防宕机 // 把数组里缓存的数据取出来一起报警 label.LinkCache = []structure.ResultMsg{} @@ -404,13 +345,26 @@ } } } rw.RUnlock() } } } else { logger.Warn("数组不圆满不打标签") // 倒是把打的组规则标签给去掉了啊 lens := len(args.RuleResult["yolo"].([]structure.Result)) - 1 args.RuleResult["yolo"] = args.RuleResult["yolo"].([]structure.Result)[0:lens] for _,val := range labelTypes { if val == 0 { if len(args.RuleResult["yolo"].([]structure.Result)) >= 1 { lens := len(args.RuleResult["yolo"].([]structure.Result))-1 args.RuleResult["yolo"] = args.RuleResult["yolo"].([]structure.Result)[0:lens] } } 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] } } } } } else { // 没有报警， //logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为false") @@ -420,14 +374,15 @@ // va.Result = strconv.FormatBool(isOk) // } //} rw.Lock() for k, _ := range TimeEleList { if k == groupRule.GroupId { delete(TimeEleList, k) logger.Debug("因为定时器的一帧数据结果为false，干掉定时器") } } rw.Unlock() // 因为本帧数据不符合规则，所以也不用统计结果数组里的东西 } } @@ -437,6 +392,7 @@ if rule.SdkArgAlias == "duration" { // logger.Info("当前小规则是：---------", rule) // 先看看定时器元素队列中是否有这条规则的定时器,如果有就不能再次创建了 rw.Lock() var flag bool = true for k, _ := range TimeEleList { if k == groupId+"+"+rule.Id { @@ -452,63 +408,8 @@ TimeEleList[groupId+"+"+rule.Id] = &timeEle // 定时器元素以当前持续时间小规则id为键 logger.Info("创建了计数器") } rw.Unlock() } } } func timeRuleResult(rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult { if rule.PolygonId == am.AreaId { // 首先规则所对应的区域id要跟区域数据的id对的上 if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规 //logger.Info("----------当前时间规则：---------", rule) // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段（比如当前时间是周三，应根据区域id查出其周三的几个布防时间段，数组） //logger.Info("时间规则的测试") now := time.Now() index := getIndexOfWeek(now.Weekday().String()) timeList := GetTimeById(rule.SdkArgValue, index) //logger.Info("当天的时间段集合：----------", timeList) //logger.Info("从数据库中查出的时间规则：", timeList) // 判断图片数据的时间是否符合当前规则 在一个即为true，全不在为false if rule.Operator == "satisfy" || rule.Operator == "==" { // 满足所选的时间规则 flag := "false" for _, timeSlot := range timeList { formula := "'" + timeSlot.Start + "'" + "<" + "'" + am.Time + "'" expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式 result, _ := expression.Evaluate(nil) // 得到数学公式的结果 formula1 := "'" + timeSlot.End + "'" + ">" + "'" + am.Time + "'" expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式 result1, _ := expression1.Evaluate(nil) // 得到数学公式的结果 //logger.Info("看看这两尊大神", result, result1) if result.(bool) && result1.(bool) { flag = "true" break } } return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + flag, rule.Sort} } if rule.Operator == "unsatisfy" || rule.Operator == "!=" { // 不满足所选的时间规则 flag := "true" for _, timeSlot := range timeList { formula := "'" + timeSlot.Start + "'" + " < " + "'" + am.Time + "'" //logger.Info("-----------------时间规则不满足的公式start：", formula) expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式 result, _ := expression.Evaluate(nil) // 得到数学公式的结果 formula1 := "'" + timeSlot.End + "'" + " > " + "'" + am.Time + "'" //logger.Info("-----------------时间规则不满足的公式end：", formula1) expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式 result1, _ := expression1.Evaluate(nil) // 得到数学公式的结果 if result.(bool) && result1.(bool) { flag = "false" break } } return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + flag, rule.Sort} } } } return structure.LittleRuleResult{} }
