package labelFilter
|
|
import (
|
"basic.com/valib/logger.git"
|
"ruleprocess/cache"
|
"ruleprocess/structure"
|
"time"
|
)
|
|
type Label struct {
|
CameraId string
|
CameraName string
|
CameraAddr string
|
TaskId string
|
Time string
|
Persons []structure.FaceResult
|
DefenceLevel []int32
|
}
|
// 把数据装配到label
|
func (label *Label)DataFormatToLabel(result structure.ResultMsg) {
|
logger.Debug("face结果：",result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0)
|
logger.Debug("yolo结果：",result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0)
|
if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0) {
|
logger.Info("---------------标签过滤器赋值")
|
label.CameraId = result.Cid
|
camera, err := cache.GetCameraById(result.Cid)
|
if err != nil {
|
logger.Error("查询摄像机信息出错")
|
}
|
label.CameraName = camera.Name
|
label.CameraAddr = camera.Addr
|
label.TaskId = result.Tasklab.Taskid
|
label.Time = time.Now().Format("2006-01-02 15:04:05")
|
if result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0 {
|
for _,faceGroup := range result.RuleResult["face"].([]structure.FaceResult) {
|
label.Persons = append(label.Persons,faceGroup)
|
label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel)
|
}
|
}
|
if result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0 {
|
|
for _,yoloGroup := range result.RuleResult["yolo"].([]structure.Result) {
|
label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel)
|
}
|
}
|
logger.Info("------------标签过滤器赋值完毕")
|
}
|
}
|
|
// 从缓存中查出所有的规则数据
|
