~jinersong/ruleprocess.git

			@@ -9,17 +9,14 @@
			var TrackPond = make(map[string]*PersonTrack)

			type PersonTrack struct {
			Face int // 人脸的id数组改为数量
			Yolo int // yolo对象的id数组改为数量
			Face int // 人脸的id数组改为数量
			Yolo int // yolo对象的id数组改为数量
			}

			func (pt *PersonTrack) Set(msg protomsg.SdkMessage) {

			}

			// 检查是否前后两次的数据id是否完全相同（人脸和yolo）
			// 检查是否前后两次的数据id是否完全相同（人脸和yolo）改为直接检查数量是否一致
			func IsSame(msg *protomsg.SdkMessage) {
			logger.Debug("++++++++++++++++++++++++++++追踪开始++++++++++++++++++++++++")
			logger.Debug("+++++++++++++++++++++追踪开始+++++++++++++++摄像机id为：",msg.Cid)
			logger.Debug("---------------------缓存池--------------------------：",TrackPond)
			for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 遍历各算法的sdkData
			if sdkinfo.Sdktype == "FaceDetect" { // 人脸检测
			if len(sdkinfo.Sdkdata) > 1 {
			@@ -29,19 +26,31 @@
			logger.Info("解析face sdk数据时出现错误", err)
			continue
			}
			logger.Info("================追踪之前人脸的个数：",len(faceParam.Faces))
			if faceParam.Faces != nil && TrackPond[msg.Cid] != nil && len(faceParam.Faces) == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			if TrackPond[msg.Cid] != nil {
			logger.Info("================追踪之前人脸的个数：",len(faceParam.Faces),"现在缓存池中记录的个数：",TrackPond[msg.Cid].Face)
			} else {
			logger.Info("================追踪之前人脸的个数：",len(faceParam.Faces),"还没有这个摄像机的缓存")
			}
			if faceParam.Faces != nil && TrackPond[msg.Cid] != nil && len(faceParam.Faces) == TrackPond[msg.Cid].Face { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			faceParam.Faces = (faceParam.Faces)[0:0]
			sdkinfo.Sdkdata,err = proto.Marshal(&faceParam)
			if err != nil {
			logger.Error("人脸序列化错误",err)
			}
			logger.Info("===================追踪中途人脸的个数",len(faceParam.Faces))
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Face:len(faceParam.Faces)}
			if TrackPond[msg.Cid] != nil {
			TrackPond[msg.Cid].Face = len(faceParam.Faces)
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Face:len(faceParam.Faces)}
			}
			}
			//for i := 0; i < len(faceParam.Faces); {
			// if flag := IsInPond(msg.Cid,faceParam.Faces[i]);flag { // 已经存在，从msg的人脸数组中删除这个元素，不必传到下一个环节
			// faceParam.Faces = append(faceParam.Faces[:i], faceParam.Faces[i+1:]...)
			// } else {
			// i++
			// }
			// if flag := IsInPond(msg.Cid,faceParam.Faces[i]);flag { // 已经存在，从msg的人脸数组中删除这个元素，不必传到下一个环节
			// faceParam.Faces = append(faceParam.Faces[:i], faceParam.Faces[i+1:]...)
			// } else {
			// i++
			// }
			//}
			} else {
			continue
			@@ -55,12 +64,29 @@
			logger.Info("解析yolo数据时出现错误", err)
			continue
			}
			logger.Info("================追踪之前yolo的个数：",len(yoloParam.Infos))
			if yoloParam.Infos != nil && TrackPond[msg.Cid] != nil && len(yoloParam.Infos) == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			yoloParam.Infos = (yoloParam.Infos)[0:0]
			logger.Info("===================追踪中途yolo的个数",len(yoloParam.Faces))
			var yoloNum int = 0
			for i := 0;i < len(yoloParam.Infos); i++ {
			if yoloParam.Infos[i].Typ == 0 {
			yoloNum++
			}
			}
			if TrackPond[msg.Cid] != nil {
			logger.Info("================追踪之前yolo的个数：",yoloNum,"现在缓存池中记录的个数：",TrackPond[msg.Cid].Yolo)
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Yolo:len(yoloParam.Infos)}
			logger.Info("================追踪之前yolo的个数：",yoloNum,"还没有这个摄像机的缓存")
			}
			if yoloParam.Infos != nil && TrackPond[msg.Cid] != nil && yoloNum == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			yoloParam.Infos = (yoloParam.Infos)[0:0]
			sdkinfo.Sdkdata,err = proto.Marshal(&yoloParam)
			if err != nil {
			logger.Error("yolo序列化错误",err)
			}
			} else {
			if TrackPond[msg.Cid] != nil {
			TrackPond[msg.Cid].Yolo = yoloNum
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Yolo:yoloNum}
			}
			}
			} else {
			continue
			@@ -69,6 +95,7 @@
			}
			logger.Debug("---------------------------------------追踪结束--------------------------------------")
			}

			//// 判断一个人脸id在不在追踪池中
			//func FaceIsInPond(cameraId string,face *protomsg.ResultFaceDetect) bool{
			// if TrackPond[cameraId].Face > 0 {
			@@ -93,4 +120,4 @@
			// return false
			// }
			// return false
			//}
			//}