派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -101,18 +101,6 @@ Liker []LikePerson // 如果是人脸的话尤其是比对，应存下他跟底库的人员的相似情况 yolo的话给nil就行 } // 从通道中获取的sdk输出的图像数据(目前主要是yolo算法的数据) //type ArgsFromSdk struct { // CameraId string // TaskId string // Photo []PhotoMap // yolo算法结构，也可以存人脸的数据，毕竟人脸中能用规则来测的还是那些参数 // KeepRight bool // 是否靠右行 算法判断的与上一帧图像的比较结果 // IsStatic bool // 是否静止 // ImageWidth int // 摄像机拍摄的图像宽 像素 // ImageHeight int // 摄像机拍摄的图像高 像素 // RuleResult []Result // 过完规则后打的标签 //} // 每个算法对于当前帧画面自己提取的数据 type SdkData struct { TaskId string @@ -124,7 +112,7 @@ IsStatic bool // 是否静止 ImageWidth int // 摄像机拍摄的图像宽 像素 ImageHeight int // 摄像机拍摄的图像高 像素 AreaMapList []AreaMap // 本sdk提取的数据按照区域划分后的数据集合 AreaMapList []*AreaMap // 本sdk提取的数据按照区域划分后的数据集合 } // 从算法模块儿拿来的对一帧图像各个算法提取的数据集合 @@ -194,8 +182,8 @@ for _, polygon := range cameraPolygons { areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine} // 为每个摄像机区域填充数据 areaMap.CountAreaObjs(arg) arg.AreaMapList = append(arg.AreaMapList, areaMap) (&areaMap).CountAreaObjs(arg) arg.AreaMapList = append(arg.AreaMapList, &areaMap) } } @@ -210,6 +198,7 @@ } // 跑本摄像机的所有规则组 一组一组跑 taskRuleList := GetRuleGroup(args.CameraId) // 本摄像机下所有任务组 //log.Println("看下摄像机下的任务组：",taskRuleList) // 得到属于该摄像机的若干组任务的完整规则（跟每一条完整规则比较之后得出本张图像对于某个规则是否报警的结果。放进map，比如本帧图像的id，所碰撞成功的规则id） args.RuleResult = make(map[string]interface{}) args.RuleResult["yolo"] = []Result{} @@ -226,6 +215,7 @@ linkTask(args, ruleList[i], taskId) } else { // 独立任务的处理 log.Println("----------------------------------------64摄像机入关") RunRule(args, ruleList[i], taskId) } } @@ -237,10 +227,14 @@ func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) bool { resultSplice := []*LittleRuleResult{} // 先过完条件规则 log.Println("---------------------看下这个组规则：",groupRule.Rules) for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { if sdkData.SdkId == "812b674b-2375-4589-919a-5c1c3278a975" { log.Println("入侵的算法数据",sdkData) } for _, areaMap := range sdkData.AreaMapList { ruleResult := filterRule(groupRule.Rules[j], &areaMap) ruleResult := filterRule(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { log.Println("条件规则结果：", ruleResult.Result) resultSplice = append(resultSplice, &ruleResult) @@ -253,7 +247,7 @@ for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { for j := 0; j < len(groupRule.Rules); j++ { putFaceToResult(groupRule.Rules[j], &areaMap, faces) putFaceToResult(groupRule.Rules[j], areaMap, faces) } } } @@ -262,7 +256,7 @@ for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { ruleResult := transferParameters(groupRule.Rules[j], &areaMap) ruleResult := transferParameters(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { log.Println("数量规则结果：", ruleResult.Result) resultSplice = append(resultSplice, &ruleResult) @@ -274,7 +268,7 @@ for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { ruleResult := timeRuleResult(groupRule.Rules[j], &areaMap) ruleResult := timeRuleResult(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { log.Println("时间规则结果：", ruleResult.Result) resultSplice = append(resultSplice, &ruleResult) @@ -286,7 +280,7 @@ for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { duration(groupRule.Rules[j], &areaMap) duration(groupRule.Rules[j], areaMap) } } } @@ -510,6 +504,7 @@ func filterRule(rule *protomsg.Rule, am *AreaMap) LittleRuleResult { if rule.SdkId == "FaceDetect" || rule.SdkId == "FaceCompare" { // 处理的是人脸算法 如果这条规则配置的是人脸算法，过滤完条件之后直接得出结果，因为肯定没有数量条件，自己拼接 log.Println("------------------------------------------还能进这儿了不成？？？") if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 算法和区域都得对的上 if rule.SdkId == "FaceCompare" {
