派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -2,7 +2,7 @@ import ( "ruleprocess/cache" "ruleprocess/logger" "basic.com/valib/logger.git" "sort" "strconv" "strings" @@ -82,6 +82,11 @@ RuleResult map[string]interface{} // 过完规则后打的标签 face: []FaceResult, yolo: []Result } type TargetInfo struct { Rect TargetId uint64 TargetScore float64 } // 过规则库打上的标签 type Result struct { TaskId string // 任务id @@ -90,7 +95,7 @@ DefenceState bool // 是否布防 AlarmLevel int32 // 报警等级 RuleText string // 文字版规则组 Location []Rect // 目标的坐标 Location []TargetInfo // 目标的坐标 AlarmPolygon string // 触发的报警框 IsLink bool // 是否是联动任务 Others @@ -205,9 +210,8 @@ // 如果此结果为真且当前过的是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) { if strings.Contains(ruleResult.Result, "true") && !strings.Contains(polygonId, groupRule.Rules[j].PolygonId) { polygonId = groupRule.Rules[j].PolygonId + "," } resultSplice = append(resultSplice, &ruleResult) @@ -216,7 +220,6 @@ } } } // 再过其他数据 这步直接得到结果（真或假） 过目标数量 for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { @@ -350,7 +353,7 @@ //} logger.Warn("___________________________________________________________________________终于走完万里长征") // 把他们的位置数据也传下去 locations := []Rect{} locations := []TargetInfo{} for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框 for _, areaMap := range sdkData.AreaMapList { @@ -372,7 +375,7 @@ //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, islink,label,}, faces}) args.RuleResult["face"] = append(args.RuleResult["face"].([]FaceResult), FaceResult{Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []TargetInfo{}, polygonId, islink,label,}, faces}) //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult))) labelTypes = append(labelTypes,1) } @@ -411,11 +414,15 @@ return faces } func putYolosToResult(am *AreaMap) []Rect { locations := []Rect{} func putYolosToResult(am *AreaMap) []TargetInfo { locations := []TargetInfo{} if len(am.filterData) > 0 { for _, data := range am.filterData { locations = append(locations, data.Location) location := TargetInfo{} location.Rect = data.Location location.TargetId = data.Id location.TargetScore = data.Score locations = append(locations, location) } } //logger.Println("-----------------------------------------------听说你是空的？",faces) @@ -580,7 +587,7 @@ //logger.Info("规则的算法id和区域的算法id：", rule.SdkId, "===", am.sdkId) if rule.PolygonId == am.areaId { // 算法和区域都得对的上 if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule"{ if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" { if rule.RuleWithPre == "||" { return LittleRuleResult{} } else { @@ -648,17 +655,15 @@ } else { return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort} } } else { return LittleRuleResult{} } } else { return LittleRuleResult{} } else if rule.SdkArgAlias == "" { // 什么参数都不配的情况 if am.targetNum > 0 { return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort} } else { return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort} } } } else { return LittleRuleResult{} } } else { return LittleRuleResult{} } } else { // 处理的都是yolo数据 @@ -698,14 +703,16 @@ return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort} } } else { return LittleRuleResult{} } else if rule.SdkArgAlias == "" { if am.targetNum > 0 { return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort} } else { return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort} } } } else { return LittleRuleResult{} } } return LittleRuleResult{} } // 如果有持续时间条件维护开启一个定时器 @@ -756,18 +763,6 @@ result, _ := expression.Evaluate(nil) // 得到数学公式的结果 return LittleRuleResult{am.sdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort} // 加上关于算法的判断条件，不能只有关于规则的，有的算法本身就是一个规则，如个体静止，靠右行,所以，拿到当前子规则的sdkid来判断是否是那些特殊的规则 } else if rule.SdkId == "IsStatic" { // 静止算法 if am.isStatic { return LittleRuleResult{} } else { return LittleRuleResult{} } } else if rule.SdkId == "KeepRight" { // 靠右行算法 if am.keepRight { return LittleRuleResult{} } else { return LittleRuleResult{} } } } return LittleRuleResult{}
