panlei
2019-08-05 8be1ea15465a24dc79f113408739c334bf987349
labelFilter/readyDataForLabel.go
@@ -4,6 +4,7 @@
   "ruleprocess/cache"
   "ruleprocess/logger"
   "ruleprocess/ruleserver"
   "time"
)
type Label struct {
@@ -11,20 +12,30 @@
   CameraName    string
   CameraAddr    string
   TaskId        string
   Persons       []ruleserver.Arg
   DefenceLevel  string
   Time        string
   Persons       []ruleserver.FaceResult
   DefenceLevel  []int32
}
// 把数据装配到label
func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) {
   if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) {
      label.CameraId = result.Cid
      camera, err := cache.GetCameraById(result.Cid)
      if err != nil {
         logger.Error("查询摄像机信息出错")
      }
      label.CameraName = camera.Name
      label.CameraAddr = camera.Addr
      label.TaskId = result.Tasklab.Taskid
      label.Time = time.Now().String()
      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)
      }
   }
}
func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) Label {
   label.CameraId = result.Cid
   camera, err := cache.GetCameraById(result.Cid)
   if err != nil {
      logger.Error("查询摄像机信息出错")
   }
   label.CameraName = camera.Name
   label.CameraAddr = camera.Addr
   label.TaskId = result.Tasklab.Taskid
   label.Persons = result.RuleResult["face"].([]ruleserver.Arg)
   //labelData.DefenceLevel = result.
   return Label{}
}
// 从缓存中查出所有的规则数据