osd 添加上下行

zhangzengfei

2024-06-13 15dc6df83a839f5913997e4e2a252a0714dc2978

 repository/captureRepo.go

@@ -3,12 +3,12 @@
import (
   "encoding/base64"
   "encoding/json"
   "gat1400Exchange/pkg"
   "time"

   "gat1400Exchange/client"
   "gat1400Exchange/config"
   "gat1400Exchange/models"
   "gat1400Exchange/pkg"
   "gat1400Exchange/pkg/logger"
   "gat1400Exchange/util"
   "gat1400Exchange/vo"
@@ -36,8 +36,7 @@
      }
      var deviceId = face.DeviceID
      var faceId = face.FaceID
      var bgImageStr, imageType string
      var bgImageWidth, bgImageHeight int
      var bgImageStr string
      var bgImageBytes, faceImageBytes []byte = nil, nil

      // 获取大图, 目前海康的小图分辨率太低
@@ -46,8 +45,6 @@
            continue
         }

         bgImageWidth = image.Width
         bgImageHeight = image.Height
         if len(image.Data) > 0 {
            if len(image.Data) > len(bgImageStr) {
               bgImageStr = image.Data
@@ -72,17 +69,15 @@
      }

      // 判断图片类型是否为场景图, 根据人脸坐标切小图.
      if imageType == "14" && config.ImageConf.CutFaceImage {
         scaleX := bgImageWidth / config.ImageConf.OriginWidth
         scaleY := bgImageHeight / config.ImageConf.OriginHeight
         faceRect := &vo.Rect{
            Left:   face.LeftTopX * scaleX,
            Top:    face.LeftTopY * scaleY,
            Right:  face.RightBtmX * scaleX,
            Bottom: face.RightBtmY * scaleY,
      if config.ImageConf.CutFaceImage && face.LeftTopX != 0 {
         faceRect := &pkg.Rect{
            Left:   face.LeftTopX,
            Top:    face.LeftTopY,
            Right:  face.RightBtmX,
            Bottom: face.RightBtmY,
         }

         faceImageBytes, err = util.SubCutImg(bgImageBytes, faceRect, 90)
         faceImageBytes, err = pkg.SubCutImage(bgImageBytes, faceRect, config.ImageConf.Enlarge)
         if err != nil {
            logger.Warn("Cut face image failure, %s", err.Error())
         }
@@ -96,9 +91,9 @@
            return
         }

         // 处理梯控填写的楼层信息 暂时使用oherFeature字段
         // 处理梯控填写的楼层信息 暂时使用otherFeature字段
         if face.OtherFeature != "" {
            pd.CameraFloor = face.OtherFeature
            pd.CameraFloor, _ = pkg.RestoreFloor(face.OtherFeature)
         }

         // 尝试从faceId提取楼层
@@ -118,7 +113,7 @@
            c.CacheData(cacheItem, "basic")
            logger.Warn("The data forwarding failed, adding to local cache.")
         } else {
            logger.Debug("The data forwarding successful. deviceId:%s", deviceId)
            logger.Debug("The data forwarding successful. deviceId:%s, picId:%s", deviceId, face.FaceID)
         }
      }
   }
@@ -257,24 +252,29 @@

func (c CaptureRepository) VIIDFaceMsgForward(msg *vo.RequestFaceList) {
   faceInfo := msg.FaceListObject.FaceObject[0]
   // 匹配楼层
   faceAppearTime, err := time.ParseInLocation("20060102150405", faceInfo.FaceAppearTime, time.Local)
   if err != nil {
      logger.Warn("Parse face appear time error, %s", err.Error())
      faceAppearTime = time.Now()
   }

   var devPos models.Positions
   _ = devPos.FindPositionByTime(faceAppearTime.Unix() + 5) // 加5秒电梯关门的时间
   if devPos.Pos == "" {
      devPos.Pos = "1F"
   }

   for idx, face := range msg.FaceListObject.FaceObject {
      msg.FaceListObject.FaceObject[idx].OtherFeature = devPos.Pos
      if config.ClientConf.AddFloorToFaceId {
         msg.FaceListObject.FaceObject[idx].FaceID = pkg.GenerateFaceIdContainFloor(face.FaceID, devPos.Pos)
   // 判断是否开启了梯控
   if config.NVCSConf.Model != "" {
      // 匹配楼层
      faceAppearTime, err := time.ParseInLocation("20060102150405", faceInfo.FaceAppearTime, time.Local)
      if err != nil {
         logger.Warn("Parse face appear time error, %s", err.Error())
         faceAppearTime = time.Now()
      }

      var devPos models.Positions
      _ = devPos.FindPositionByTime(faceAppearTime.Unix() + 5) // 加5秒电梯关门的时间
      if devPos.Pos == "" {
         devPos.Pos = "1F"
      }

      for idx, face := range msg.FaceListObject.FaceObject {
         msg.FaceListObject.FaceObject[idx].OtherFeature, _ = pkg.ParseFloor(devPos.Pos)
         if config.ClientConf.AddFloorToFaceId {
            msg.FaceListObject.FaceObject[idx].FaceID = pkg.GenerateFaceIdContainFloor(face.FaceID, devPos.Pos)
         }
      }

   }

   b, _ := json.Marshal(msg)