panlei
2019-08-15 d62130814d09b4c0ad313979902ccefec6260ddf
ruleserver/readyDataForRule.go
@@ -112,7 +112,7 @@
// 取出某个时间规则的第几天的规则段集合
func GetTimeById(id string, index int) []TimeRange {
   _, cameraTimeRule := cache.GetTimeRuleById(id)
   var timeRangeList []day
   var timeRangeList []Day
   err := json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList)
   if err != nil {
      logger.Error("取时间规则时反序列化错误!")
@@ -141,7 +141,7 @@
   Start string `json:"start"`
   End   string `json:"end"`
}
type day struct {
type Day struct {
   Day       int         `json:"day"`        // 标示当前星期几
   TimeRange []TimeRange `json:"time_range"` // 当天的几个时间段
}
@@ -166,7 +166,7 @@
}
// 给目标填充liker
func (arg *Arg) fillLiker(tableId []string, compareThreshold float32) {
func (arg *Arg) FillLiker(tableId []string, compareThreshold float32) {
   //bytes := bigCache.GetComparePersonBaseInfo(tableId, arg.Feature, compareThreshold)
   comArg := &protomsg.CompareArgs{
      TableIds:tableId,
@@ -250,9 +250,9 @@
               logger.Info("============================================进行人脸对比")
               for _, areaMap := range sdkData.AreaMapList {
                  // 拿区域中每个人脸特征值去对比,填充其liker
                  if groupRule.Rules[j].PolygonId == areaMap.areaId {
                  if groupRule.Rules[j].PolygonId == areaMap.AreaId {
                     //logger.Info("--------------看看compareFlag的值和tableId和areaMap.args的长度:",compareFlag,tableIds,len(areaMap.args))
                     for _, arg := range areaMap.args {
                     for _, arg := range areaMap.Args {
                        arg.Liker = arg.Liker[0:0]
                        logger.Info("清空之后看看之前打的人脸标签变了没:")
                        if args.RuleResult["face"] != nil && len(args.RuleResult["face"].([]FaceResult)) > 0 {
@@ -263,14 +263,14 @@
                           }
                        }
                        if compareFlag == 1 {
                           arg.fillLiker(nil, threshold)
                           arg.FillLiker(nil, threshold)
                        }
                        if compareFlag == 2 {
                           arg.fillLiker(tableIds, threshold)
                           arg.FillLiker(tableIds, threshold)
                        }
                        //logger.Info("-------------------成功给liker赋值,长度为:", len(arg.Liker))
                     }
                     areaMap.filterData = areaMap.args
                     areaMap.FilterData = areaMap.Args
                     //logger.Info("=======第一次看args:",(areaMap.filterData))
                  }
                  //logger.Info("-------------------------------人脸对比之后的目标数量",len(areaMap.args))
@@ -285,29 +285,29 @@
// 计算区域内的目标数量以及将相似度、占比、尺寸等打包
func (a *AreaMap) CountAreaObjs(arg *SdkData) {
   a.targetNum = 0
   a.TargetNum = 0
   threshold := 0.0       // 相似度
   intersectionper := 0.2 // 占比
   size := 0.0            // 尺寸
   areaPoints := Json2points(a.areaJson)
   areaPoints := Json2points(a.AreaJson)
   widthScale := float64(arg.ImageWidth / 960)
   heigthScale := float64(arg.ImageHeight / 540)
   for _, obj := range arg.Photo {
      //logger.Info("------------------看看sdkData:", arg.SdkName, "的Photo数据----------------", obj, "----顺便看看占比-----:", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale))
      if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) {
         // 这步要备齐表达式里所需要的所有参数
         a.targetNum++
         a.TargetNum++
         arg1 := Arg{obj.Id,obj.Score, PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale), float64(obj.Rects.Width * obj.Rects.Height), obj.IsYolo, obj.Rects, obj.Feature, obj.ThftRes, []*BaseInfo{}}
         //logger.Println("放进去的arg:-------", arg1)
         a.args = append(a.args, &arg1)
         a.filterData = append(a.filterData, &arg1)
         a.Args = append(a.Args, &arg1)
         a.FilterData = append(a.FilterData, &arg1)
      }
   }
   //logger.Info("区域是:",areaPoints,"区域内目标数量为:",a.targetNum,"---",len(a.filterData))
   a.time = time.Unix(time.Now().Unix(), 0).String()[11:16]
   a.keepRight = arg.KeepRight
   a.isStatic = arg.IsStatic
   a.Time = time.Unix(time.Now().Unix(), 0).String()[11:16]
   a.KeepRight = arg.KeepRight
   a.IsStatic = arg.IsStatic
   //logger.Println("--------------------看看区域数据:",*a)
}
@@ -319,7 +319,7 @@
   }
   for _, polygon := range cameraPolygons {
      //logger.Println("++++++在这儿看一下区域啊:", polygon.Polygon)
      areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine}
      areaMap := AreaMap{CameraId: cameraId, AreaId: polygon.Id, AreaJson: polygon.Polygon, TriggerLine: polygon.TriggerLine, DirectionLine: polygon.DirectionLine}
      // 为每个摄像机区域填充数据
      areaMap.CountAreaObjs(arg)
      arg.AreaMapList = append(arg.AreaMapList, &areaMap)