panlei
2019-08-08 ee4eeaa38e11d3cef51c9bb093da69547bd18c4d
labelFilter/readyDataForLabel.go
@@ -4,6 +4,7 @@
   "ruleprocess/cache"
   "ruleprocess/logger"
   "ruleprocess/ruleserver"
   "time"
)
type Label struct {
@@ -17,7 +18,10 @@
}
// 把数据装配到label
func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) {
   logger.Debug("face结果:",result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0)
   logger.Debug("yolo结果:",result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0)
   if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) {
      logger.Info("---------------标签过滤器赋值")
      label.CameraId = result.Cid
      camera, err := cache.GetCameraById(result.Cid)
      if err != nil {
@@ -26,13 +30,19 @@
      label.CameraName = camera.Name
      label.CameraAddr = camera.Addr
      label.TaskId = result.Tasklab.Taskid
      for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) {
         label.Persons = append(label.Persons,faceGroup)
         label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel)
      label.Time = time.Now().Format("2006-01-02 15:04:05")
      if result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0 {
         for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) {
            label.Persons = append(label.Persons,faceGroup)
            label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel)
         }
      }
      for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) {
         label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel)
      if result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
         for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) {
            label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel)
         }
      }
      logger.Info("------------标签过滤器赋值完毕")
   }
}