添加A3平层传感器数据获取方式

zhangzengfei

2024-10-11 539a78196da60eb97cf7057c5c85dfaa9b240741

 repository/captureRepo.go

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

   "gat1400Exchange/client"
   "gat1400Exchange/config"
   "gat1400Exchange/models"
   "gat1400Exchange/pkg"
   "gat1400Exchange/pkg/logger"
   "gat1400Exchange/service"
   "gat1400Exchange/util"
   "gat1400Exchange/vo"
)
@@ -36,8 +37,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 +46,6 @@
            continue
         }

         bgImageWidth = image.Width
         bgImageHeight = image.Height
         if len(image.Data) > 0 {
            if len(image.Data) > len(bgImageStr) {
               bgImageStr = image.Data
@@ -72,17 +70,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,16 +92,10 @@
            return
         }

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

         // 尝试从faceId提取楼层
         if pd.CameraFloor == "" && config.ClientConf.AddFloorToFaceId {
            pd.CameraFloor, _ = pkg.ParseFloorFromId(face.FaceID)
         }
         //logger.Debug("device %s, CameraFloor:%s", deviceId, pd.CameraFloor)

         payload, err := json.Marshal(pd)
         if err != nil {
@@ -118,7 +108,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)
         }
      }
   }
@@ -186,10 +176,10 @@
            pd.CameraFloor = v.BehaviorDescription
         }

         // 尝试从faceId提取楼层
         if pd.CameraFloor == "" && config.ClientConf.AddFloorToFaceId {
            pd.CameraFloor, _ = pkg.ParseFloorFromId(v.PersonID)
         }
         //// 尝试从faceId提取楼层
         //if pd.CameraFloor == "" && config.ClientConf.AddFloorToFaceId {
         //   pd.CameraFloor, _ = pkg.ParseFloorFromId(v.PersonID)
         //}
         //logger.Debug("device %s, CameraFloor:%s", deviceId, pd.CameraFloor)

         payload, err := json.Marshal(pd)
@@ -214,6 +204,7 @@
func (c CaptureRepository) PackPushDataV2(deviceId, faceId, appearTime string, bgImgBytes, faceImgBytes []byte) *vo.PushDataInfoV2 {
   var pd = new(vo.PushDataInfoV2)
   var floor string
   var runDir string

   faceAppearTime, err := time.ParseInLocation("20060102150405", appearTime, time.Local)
   if err != nil {
@@ -221,12 +212,31 @@
      faceAppearTime = time.Now()
   }

   if config.ServeConf.Role == "server" {
   // 判断是否需要匹配楼层
   if config.NVCSConf.Model != "" {
      // 匹配楼层
      var devPos models.Positions
      _ = devPos.FindDevicePosition(deviceId, faceAppearTime.Unix()+5) // 加5秒电梯关门的时间

      floor = devPos.Pos

      for i := 0; i < config.NVCSConf.WaitRunTime; i++ {
         var dbPos models.Positions
         if err := dbPos.FindMovePosition(faceAppearTime.Unix()+5, floor); err == nil {
            switch dbPos.RunDir {
            case service.RunUp:
               runDir = "in"
            case service.RunDown:
               runDir = "out"
            case service.RunStop:
               runDir = ""
            }

            break
         }

         time.Sleep(1 * time.Second)
      }
   }

   pd.PicMaxImages = append(pd.PicMaxImages, bgImgBytes)
@@ -240,6 +250,7 @@
   pd.DataSource = "camera"
   pd.CameraFloor = floor
   pd.CameraId = deviceId
   pd.Direction = runDir

   return pd
}
@@ -257,24 +268,50 @@

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 floor, runDir string
      var devPos models.Positions
      _ = devPos.FindPositionByTime(faceAppearTime.Unix() + 5) // 加5秒电梯关门的时间
      if devPos.Pos == "" {
         devPos.Pos = "1F"
      }

      floor = devPos.Pos

      for i := 0; i < config.NVCSConf.WaitRunTime; i++ {
         var dbPos models.Positions
         if err := dbPos.FindMovePosition(faceAppearTime.Unix()+5, floor); err == nil {
            switch dbPos.RunDir {
            case service.RunUp:
               runDir = "1"
            case service.RunDown:
               runDir = "2"
            case service.RunStop:
               runDir = "0"
            }

            break
         }

         time.Sleep(1 * time.Second)
      }

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

   }

   b, _ := json.Marshal(msg)
@@ -302,11 +339,11 @@
      devPos.Pos = "1F"
   }

   for idx, v := range msg.PersonListObject.PersonObject {
   for idx, _ := range msg.PersonListObject.PersonObject {
      msg.PersonListObject.PersonObject[idx].BehaviorDescription = devPos.Pos
      if config.ClientConf.AddFloorToFaceId {
         msg.PersonListObject.PersonObject[idx].PersonID = pkg.GenerateFaceIdContainFloor(v.PersonID, devPos.Pos)
      }
      //if config.ClientConf.AddFloorToFaceId {
      //   msg.PersonListObject.PersonObject[idx].PersonID = pkg.GenerateFaceIdContainFloor(v.PersonID, devPos.Pos)
      //}
   }

   b, _ := json.Marshal(msg)
@@ -318,3 +355,18 @@

   return
}

func (c CaptureRepository) PubRecordMessage(deviceId, msgId string) {
   var msg = vo.RecordMsg{
      CamId: deviceId,
      MsgId: msgId,
   }

   logger.Debug("Pub record message %s, %s", deviceId, msgId)
   body, _ := json.Marshal(&msg)

   _, err := util.HttpPost(config.ForwardConf.RecordServer, nil, body)
   if err != nil {
      logger.Warn(err.Error())
   }
}