fix 追踪bug

panlei

2019-12-06 bf32a74cbb28405eb116495eb2046880ecfceb66

 ruleserver/ruleToformula.go

@@ -37,6 +37,7 @@
      args.RuleResult["face"] = []structure.Result{}
      args.RuleResult["target"] = []structure.Result{}
      args.RuleResult["plate"] = []structure.Result{}
      args.RuleResult["track"] = []structure.Result{}
      //logger.Warn("传进去之前是什么德行：",args.RuleResult["yolo"])
      if taskGroup != nil && len(taskGroup.GroupRules) > 0 {
         // 先过独立，再过联动
@@ -137,132 +138,7 @@
      // 进行定时器的处理和判断
      timeFlag := TimerAlarm(&label, groupRule.GroupId, result)
      if timeFlag == "01" || timeFlag == "10" || timeFlag == "11"{ // 没有定时器或者满足定时器条件
         // 打人脸标签和yolo标签
         // 最后成功报警才把符合条件的人脸数据塞进结果标签里
         // 配了人脸的算法才把人脸的数据甩出来打标签
         faces := []*structure.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 {
            if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData里有人脸数据且配置了算法才把符合条件的数据塞进标签里去
               for _, areaMap := range sdkData.AreaMapList {
                  if areaMap.IsEffective {
                     faces = append(faces, putFaceToResult(areaMap, faces)...)
                  }
               }
            }
         }
         //logger.Info("face标签的长度：",len(faces))
         //for _,face := range faces  {
         //   //logger.Debug("————————————————________________看看人脸的坐标:",face.Location)
         //}
         //logger.Warn("___________________________________________________________________________终于走完万里长征")
         // 把他们的位置数据也传下去
         yolos := []*structure.Arg{}
         for _, sdkData := range args.Sdkdata {
            if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框
               for _, areaMap := range sdkData.AreaMapList {
                  if areaMap.IsEffective {
                     yolos = append(yolos, putYolosToResult(areaMap)...)
                  }
               }
            }
         }
         //logger.Debug("------locations的内容：", yolos)
         // 处理目标定时数据
         targets := []*structure.Arg{}
         for _, sdkData := range args.Sdkdata {
            if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 输出目标数据
               for _, areaMap := range sdkData.AreaMapList {
                  if areaMap.IsEffective {
                     targets = append(targets, putTargetsToResult(areaMap)...)
                  }
               }
            }
         }
         // 车辆目标统计
         cars := []*structure.Arg{}
         for _, sdkData := range args.Sdkdata {
            //logger.Info("看看sdkNames",sdkNames)
            if sdkData.IpcId == "91d923ef-6200-4549-ab1b-8e773e85d729" { // 把车牌数据的各个目标的坐标输出方便后面画框
               for _, areaMap := range sdkData.AreaMapList {
                  if areaMap.IsEffective {
                     cars = append(cars, putFaceToResult(areaMap, cars)...)
                  }
               }
            }
         }
         // 跟踪目标统计
         personTrack := []*structure.Arg{}
         for _, sdkData := range args.Sdkdata {
            //logger.Info("看看sdkNames",sdkNames)
            if sdkData.IpcId == "跟踪的ipcId" { // 把跟踪页面的各个目标的坐标输出
               for _, areaMap := range sdkData.AreaMapList {
                  if areaMap.IsEffective {
                     personTrack = append(personTrack, putFaceToResult(areaMap, personTrack)...)
                  }
               }
            }
         }
         var islink bool
         if groupRule.SetType == "linkTask" {
            islink = true
         } else {
            islink = false
         }
         //logger.Info("触发的区域id：",polygonId)
         var labelTypes []int // 0为yolo标签，1为face标签 2为两者标签
         if sdkNames != "" && len(targets) == 0 {
            args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, yolos, 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.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, faces, polygonId, islink,label})
            //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]structure.Result)))
            labelTypes = append(labelTypes,1)
         }
         if len(targets) > 0 {
            args.RuleResult["target"] = append(args.RuleResult["target"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, targets, polygonId, islink,label})
            //logger.Info("-------------------目标持续结果标签", len(args.RuleResult["target"].([]structure.Result)))
            //labelTypes = append(labelTypes,2)
         }
         if len(cars) > 0 {
            args.RuleResult["plate"] = append(args.RuleResult["plate"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, cars, polygonId, islink,label})
            //logger.Info("-------------------车牌结果标签", len(args.RuleResult["plate"].([]structure.Result)))
            //labelTypes = append(labelTypes,2)
         }
         if len(personTrack) > 0 {
            args.RuleResult["track"] = append(args.RuleResult["track"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, personTrack, polygonId, islink,label})
            //logger.Info("-------------------车牌结果标签", len(args.RuleResult["plate"].([]structure.Result)))
            //labelTypes = append(labelTypes,2)
         }
         // 给持续时间的第一张赋予缓存数据（遍历复制）
         //if cacheId != "" { // 有这帧数据的缓存
         //   tempMap := make(map[string]interface{})
         //   for k, result := range args.RuleResult {
         //      if k == "yolo" {
         //         tempMap[k] = []structure.Result{}
         //         for _, res := range result.([]structure.Result) {
         //            tempMap[k] = append(tempMap[k].([]structure.Result), res)
         //         }
         //      }
         //      if k == "face" {
         //         tempMap[k] = []structure.Result{}
         //         for _, res := range result.([]structure.Result) {
         //            tempMap[k] = append(tempMap[k].([]structure.Result), res)
         //         }
         //      }
         //   }
         //   rw.Lock()
         //   TimeEleList[cacheId].CacheSdkData.RuleResult = tempMap
         //   rw.Unlock()
         //}

         labelTypes := AssembResultLabel(args,groupRule,sdkNames,taskId,polygonId,label)
         return true,labelTypes
      } else {
         return false,[]int{}
@@ -493,3 +369,87 @@
   return
}

func AssembResultLabel(args *structure.SdkDatas, groupRule *protomsg.GroupRule,sdkNames string,taskId string,polygonId string,label structure.Others) []int{
   faces := []*structure.Arg{}
   yolos := []*structure.Arg{}
   targets := []*structure.Arg{}
   personTrack := []*structure.Arg{}
   cars := []*structure.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 {
      switch sdkData.IpcId {
      case "A8B73405-373D-4F23-CED2-A617EBD7EC55" :
         if faceFlag { // sdkData里有人脸数据且配置了算法才把符合条件的数据塞进标签里去
            for _, areaMap := range sdkData.AreaMapList {
               if areaMap.IsEffective {
                  faces = append(faces, putFaceToResult(areaMap, faces)...)
               }
            }
         }
      case "02D54B61-0F16-C604-8567-FC4BE493C523":
         if sdkNames != "" { // 把yolo数据的各个目标的坐标输出方便后面画框
            for _, areaMap := range sdkData.AreaMapList {
               if areaMap.IsEffective {
                  yolos = append(yolos, putYolosToResult(areaMap)...)
                  // 处理目标定时数据
                  targets = append(targets, putTargetsToResult(areaMap)...)
               }
            }
         }
      case "91d923ef-6200-4549-ab1b-8e773e85d729":
         // 车辆数据
         for _, areaMap := range sdkData.AreaMapList {
            if areaMap.IsEffective {
               cars = append(cars, putFaceToResult(areaMap, cars)...)
            }
         }
      case "跟踪的ipcId":
         // 跟踪目标
         for _, areaMap := range sdkData.AreaMapList {
            if areaMap.IsEffective {
               personTrack = append(personTrack, putFaceToResult(areaMap, personTrack)...)
            }
         }
      }
   }

   var islink bool
   if groupRule.SetType == "linkTask" {
      islink = true
   } else {
      islink = false
   }
   //logger.Info("触发的区域id：",polygonId)
   var labelTypes []int // 0为yolo标签，1为face标签 2为两者标签
   if sdkNames != "" && len(targets) == 0 {
      args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, yolos, 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.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, faces, polygonId, islink,label})
      //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]structure.Result)))
      labelTypes = append(labelTypes,1)
   }
   if len(targets) > 0 {
      args.RuleResult["target"] = append(args.RuleResult["target"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, targets, polygonId, islink,label})
      //logger.Info("-------------------目标持续结果标签", len(args.RuleResult["target"].([]structure.Result)))
      labelTypes = append(labelTypes,2)
   }
   if len(cars) > 0 {
      args.RuleResult["plate"] = append(args.RuleResult["plate"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, cars, polygonId, islink,label})
      //logger.Info("-------------------车牌结果标签", len(args.RuleResult["plate"].([]structure.Result)))
      labelTypes = append(labelTypes,3)
   }
   if len(personTrack) > 0 {
      args.RuleResult["track"] = append(args.RuleResult["track"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, personTrack, polygonId, islink,label})
      //logger.Info("-------------------车牌结果标签", len(args.RuleResult["plate"].([]structure.Result)))
      labelTypes = append(labelTypes,4)
   }
   return labelTypes
}