派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -184,7 +184,10 @@ // 把sdk从数据帧上提取的按照区域分类归置 func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) { logger.Info("--------看看解析出来的数据---", *arg) logger.Info("==================================本sdkData中解出来的目标数据=======================================") for _, photo := range arg.Photo { logger.Info("--------看看解析出来的数据---", cameraId, arg.IpcId, photo.Rects, photo.Score) } for _, polygon := range cameraPolygons { //logger.Println("++++++在这儿看一下区域啊:", polygon.Polygon) areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine} @@ -194,7 +197,7 @@ } } // 对单帧图像的判断 是舍弃（或者说对于某些需求可以放ES数据库一份）还是返回 // 对单帧图像的判断 thisSdkDatas 当前传入的这帧数据，cacheSdkData 定时器里缓存的一帧数据 没有就返回nil (thisSdkDatas SdkDatas, cacheSdkDatas SdkDatas) func Judge(args *SdkDatas) { if len(args.Sdkdata) > 0 { // 拿到本摄像机的区域 @@ -255,7 +258,7 @@ } for j := 0; j < len(groupRule.Rules); j++ { // 只有是人脸对比并且配置的参数是对比底库才走（若配的对比阈值则不走） if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972"{ if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 搜索到人脸检测的sdkdata 现在关于人脸的只有他一个sdk，全找他 for _, areaMap := range sdkData.AreaMapList { @@ -270,9 +273,9 @@ logger.Error("getBaseInfo解压错误", err1) } for key, val := range m { baseinfo,err1 := esutil.Dbpersoninfosbyid(key) baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { logger.Error("查询底库人员信息出错",err1) logger.Error("查询底库人员信息出错", err1) } baseinfo.CompareScore = val arg.Liker = append(arg.Liker, &baseinfo) @@ -290,7 +293,7 @@ for key, val := range m { baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { logger.Error("查询底库人员信息出错",err1) logger.Error("查询底库人员信息出错", err1) } baseinfo.CompareScore = val arg.Liker = append(arg.Liker, &baseinfo) @@ -401,7 +404,7 @@ ipcId := sdk.IpcId if ipcId == sdkData.IpcId { for _, areaMap := range sdkData.AreaMapList { duration(groupRule.Rules[j], &areaMap) duration(groupRule.Rules[j], &areaMap, args) } } } @@ -441,26 +444,26 @@ timeEle.N = timeEle.InitN // 重置定时器 } } logger.Info("-------------------重置定时器并返回false---------------") logger.Info("-------------------重置可能存在的定时器并返回false---------------") return false } else { // 最后成功报警才把符合条件的人脸数据塞进结果标签里 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 { for _, areaMap := range sdkData.AreaMapList { for j := 0; j < len(groupRule.Rules); j++ { faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...) if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData里有人脸数据且配置了算法才把符合条件的数据塞进标签里去 for _, areaMap := range sdkData.AreaMapList { faces = append(faces, putFaceToResult(&areaMap)...) } } } // 配了人脸的算法才把人脸的数据甩出来打标签 flag11 := 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" { flag11 = true } } if flag11 { if faceFlag { args.RuleResult["face"] = faces } // 去看池子里是否有与本帧数据有关的定时器，如果有，看此时是否走到0，没有此定时器或有定时器走到0的话返回成功报警 @@ -478,22 +481,12 @@ // 把他们的位置数据也传下去 locations := []Rect{} for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { for j := 0; j < len(groupRule.Rules); j++ { locations = append(locations, putYolosToResult(groupRule.Rules[j], &areaMap)...) if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { locations = append(locations, putYolosToResult(&areaMap)...) } } } //// 配了人脸的算法才把人脸的数据甩出来打标签 //flag11 := 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-5c1c3278a971" { // flag11 = true // } //} //if flag11 { // args.RuleResult["face"] = faces //} if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText, locations}) logger.Info("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result)) @@ -511,26 +504,23 @@ } } func putFaceToResult(rule *protomsg.Rule, am *AreaMap) []Arg { func putFaceToResult(am *AreaMap) []Arg { faces := []Arg{} if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { if len(am.filterData) > 0 { for _, data := range am.filterData { faces = append(faces, data) } if len(am.filterData) > 0 { for _, data := range am.filterData { faces = append(faces, data) } } //logger.Println("-----------------------------------------------听说你是空的？",faces) return faces } func putYolosToResult(rule *protomsg.Rule, am *AreaMap) []Rect { func putYolosToResult(am *AreaMap) []Rect { locations := []Rect{} if rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a97e" && rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a972" { if len(am.filterData) > 0 { for _, data := range am.filterData { locations = append(locations, data.Location) } if len(am.filterData) > 0 { for _, data := range am.filterData { locations = append(locations, data.Location) } } //logger.Println("-----------------------------------------------听说你是空的？",faces) @@ -671,9 +661,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) } @@ -685,7 +675,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) } } @@ -749,6 +739,7 @@ } // 先清空过滤后的数据，再往里塞本次过滤后的数据 am.filterData = am.filterData[0:0] //logger.Info("-----------------------人脸过滤的args：", args) for _, arg := range args { var formula string if rule.SdkArgAlias == "score" { @@ -833,7 +824,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" { // // 先看看定时器元素队列中是否有这个摄像机这个区域的定时器,如果有就不能再次创建了 @@ -847,7 +838,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)
