派生自 ruleprocess
algorithm/middleware/middleware.go
@@ -18,6 +18,7 @@ resultSplice := []*structure.LittleRuleResult{} sdkNames := "" polygonId := "" // 这步过的是时间段规则（时间段等） for j := 0; j < len(groupRule.Rules); j++ { for _, sdkData := range args.Sdkdata { @@ -32,7 +33,7 @@ ruleResult := timeRuleResult(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { if strings.Contains(ruleResult.Result, "false") { return false,"","" // 不符合时间规则的话直接返回，不再浪费时间 return false,"","" // 不符合时间规则的话直接返回，不再浪费程序执行 } if strings.Contains(ruleResult.Result, "true") { areaMap.IsEffective = true @@ -72,6 +73,12 @@ //logger.Info("当前走的规则是--：", groupRule.Rules[j],sdkName,groupRule.Rules[j].SdkArgAlias, "---","") for _, areaMap := range sdkData.AreaMapList { ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message) logger.Info("前置连接符：",groupRule.Rules[j]) if groupRule.Rules[j].RuleWithPre == "⊇" { assembleDta(args,groupRule.Rules[j-1],groupRule.Rules[j]) // 把当前小规则结果置空 ruleResult = structure.LittleRuleResult{} } if ruleResult.Result != "" { //logger.Info("条件规则结果：", ruleResult.Result) // 如果结果为真，把这条规则中的区域置为有效 @@ -202,7 +209,7 @@ } // 跨摄像机跟踪 for j := 0; j < len(groupRule.Rules); j++ { if groupRule.Rules[j].SdkId == "人体跟踪" && groupRule.Rules[j].SdkArgAlias != "灵魂参数"{ if groupRule.Rules[j].SdkId == "520af340-f257-4fe2-ac3c-695c390e53dc" && groupRule.Rules[j].SdkArgAlias == "cmpThreshold"{ for _, sdkData := range args.Sdkdata { // 根据规则的sdkId查出其对应的ipcId，用ipcId去找该比对的数据 sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId) @@ -210,8 +217,6 @@ logger.Error("没查到sdk的信息---", err) } ipcId := sdk.IpcId sdkName := sdk.SdkName //logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId) if ipcId == sdkData.IpcId { //logger.Info("当前走的规则是--：", groupRule.Rules[j],sdkName,groupRule.Rules[j].SdkArgAlias, "---","") for _, areaMap := range sdkData.AreaMapList { @@ -222,13 +227,7 @@ 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 + " " } 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) } } @@ -378,7 +377,7 @@ soName = "static.so" case "812b674b-2375-4589-919a-5c1c3278a978": soName = "plate.so" case "人体跟踪": case "520af340-f257-4fe2-ac3c-695c390e53dc": soName = "personTrack.so" } //soInfo,err := cache.GetSoInfoById(sdkId) @@ -425,4 +424,87 @@ } } return 0 } } //// 时间段调度 //func timeDispatch(args *structure.SdkDatas,groupRule protomsg.GroupRule,resultSplice []*structure.LittleRuleResult,sdkNames,polygonId string) (bool,string,string){ // // 这步过的是时间段规则（时间段等） // 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 // sdkName := sdk.SdkName // if ipcId == sdkData.IpcId { // for _, areaMap := range sdkData.AreaMapList { // ruleResult := timeRuleResult(groupRule.Rules[j], areaMap) // if ruleResult.Result != "" { // if strings.Contains(ruleResult.Result, "false") { // return false,"","" // 不符合时间规则的话直接返回，不再浪费程序执行 // } // if strings.Contains(ruleResult.Result, "true") { // areaMap.IsEffective = true // } // //logger.Info("时间规则结果：", ruleResult.Result) // if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) { // sdkNames = sdkName + " " // } // 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) // } // } // } // } // } //} //func assembleData(args *structure.SdkDatas,resultSplice []*structure.LittleRuleResult,groupRule protomsg.GroupRule) { // for index,result := range resultSplice { // if strings.Contains(result.Result,"contain") { // assembleDta(args,groupRule.Rules[index-1],groupRule.Rules[index]) // // 删除带有contain的规则结果 // resultSplice = append(resultSplice[:index],resultSplice[index+1:]...) // } // } //} // 如果规则组中有包含符号，把规则组之后的规则对应的算法和区域提取的数据装配到前面去 func assembleDta(args *structure.SdkDatas,rule1,rule2 *protomsg.Rule) { // 拿到后面的那个区域的数据 logger.Info("触发了包含事件，把后者的数据给到前者",*rule1,*rule2) area := structure.AreaMap{} for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { if areaMap.SdkId == rule2.SdkId && areaMap.AreaId == rule2.PolygonId { area = *areaMap goto second } } } second: for _, sdkData := range args.Sdkdata { for _, areaMap := range sdkData.AreaMapList { if areaMap.SdkId == rule1.SdkId && areaMap.AreaId == rule1.PolygonId { // 拿到包含符之前的数据，遍历装配 for _,arg := range areaMap.FilterData { mergeArg(arg,area) logger.Info("合并后的arg是：",*arg) } // 清空人脸，人脸不再单独插入了 area.FilterData = nil return // 装配一次就可以 } } } } func mergeArg(arg *structure.Arg,areaMap structure.AreaMap) { for _,arg1 := range areaMap.FilterData { precent := ruleserver.PgsInterPercent(Rect2Point(arg.Location),arg1.Location,1,1) if precent > 99 { arg.AttachArg = arg1.SourceArg } } }
