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,7 +36,7 @@
      }
      var deviceId = face.DeviceID
      var faceId = face.FaceID
      var bgImageStr, imageType string
      var bgImageStr string
      var bgImageBytes, faceImageBytes []byte = nil, nil

      // 获取大图, 目前海康的小图分辨率太低
@@ -69,31 +69,31 @@
      }

      // 判断图片类型是否为场景图, 根据人脸坐标切小图.
      if imageType == "14" && config.ForwardConf.CutFaceImage {
         faceRect := &vo.Rect{
      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())
         }
      }

      // 转发图像
      logger.Debug("Prepare forward image, deviceId:%s, image len:%d, server:%s", deviceId, len(bgImageBytes), config.ForwardConf.SyncServer)
      logger.Debug("Prepare forward image, deviceId:%s, bgImage len:%d, smImage len:%d server:%s", deviceId, len(bgImageBytes), len(faceImageBytes), config.ForwardConf.SyncServer)
      if deviceId != "" && bgImageStr != "" && config.ForwardConf.SyncServer != "" {
         pd := c.PackPushDataV2(deviceId, faceId, face.FaceAppearTime, bgImageBytes, faceImageBytes)
         if pd == nil {
            return
         }

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

         // 尝试从faceId提取楼层
@@ -113,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)
         }
      }
   }
@@ -252,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)