package util
|
|
import (
|
"basic.com/pubsub/protomsg.git"
|
)
|
|
type BaseCompareInfo struct {
|
TableId string `json:"tableId"`
|
TableName string `json:"tableName"`
|
BwType string `json:"bwType"`
|
CompareScore float64 `json:"compareScore"`
|
TargetId string `json:"targetId"`
|
TargetName string `json:"targetName"`
|
TargetPicUrl string `json:"targetPicUrl"`
|
MonitorLevel string `json:"monitorLevel"`
|
Content string `json:"content"`
|
DbLabel string `json:"labels"`
|
}
|
|
type AlarmRule struct {
|
GroupId string `json:"groupId"`
|
AlarmLevel string `json:"alarmLevel"`
|
RuleText string `json:"ruleText"`
|
DefenceState bool `json:"defenceState"`
|
LinkInfo string `json:"linkInfo"`
|
}
|
|
type Point struct {
|
X float64 `json:"x"`
|
Y float64 `json:"y"`
|
}
|
|
type Points struct {
|
TopLeft Point `json:"topLeft"`
|
BottomRight Point `json:"bottomRight"`
|
}
|
|
type TargetInfo struct {
|
TargetId string `json:"targetId"`
|
AreaId string `json:"areaId"`
|
AreaName string `json:"areaName"`
|
TargetScore float64 `json:"targetScore"`
|
TargetType string `json:"targetType"`
|
Feature string `json:"feature"`
|
PicSmUrl string `json:"picSmUrl"`
|
TargetLocation Points `json:"targetLocation"`
|
BelongsTargetID string `json:"belongsTargetId"`
|
Attribute string `json:"attribute"`
|
}
|
|
//type Target struct {
|
// TargetInfo
|
// AttachTarget TargetInfo `json:"attachTarget"`
|
//}
|
|
//type GroupRuleResult struct {
|
// AlarmRules AlarmRule `json:"alarmRules"`
|
// PicDate string `json:"picDate"`
|
// PicMaxUrl string `json:"picMaxUrl"`
|
//}
|
|
type PicWidHei struct {
|
PicW int `json:"picW"`
|
PicH int `json:"picH"`
|
}
|
|
type TaskResultInfo struct {
|
Id string `json:"id"`
|
CameraId string `json:"cameraId"`
|
CameraAddr string `json:"cameraAddr"`
|
CameraName string `json:"cameraName"`
|
PicDate string `json:"picDate"`
|
PicId int64 `json:"picId"`
|
PicMaxUrl []string `json:"picMaxUrl"`
|
PicWH PicWidHei `json:"picWH"`
|
TaskId string `json:"taskId"`
|
TaskName string `json:"taskName"`
|
SdkName string `json:"sdkName"`
|
Content string `json:"content"`
|
AlarmRules []AlarmRule `json:"alarmRules"`
|
LikeDate string `json:"likeDate"`
|
ShowLabels string `json:"showLabels"`
|
OtherLabels string `json:"otherLabels"`
|
VideoUrl string `json:"videoUrl"`
|
AnalyServerId string `json:"analyServerId"`
|
AnalyServerName string `json:"analyServerName"`
|
AnalyServerIp string `json:"analyServerIp"`
|
ClusterId string `json:"clusterId"`
|
IsAlarm bool `json:"isAlarm"`
|
IsAckAlarm bool `json:"isAckAlarm"`
|
IsCollect bool `json:"isCollect"`
|
IsDelete bool `json:"isDelete"`
|
BaseInfo []*BaseCompareInfo `json:"baseInfo"`
|
TargetInfo []TargetInfo `json:"targetInfo"`
|
FileId string `json:"fileId"` // 数据栈文件id
|
DataSource string `json:"dataSource"` // 数据来源:摄像机, 数据栈
|
}
|
|
// 统一的数据结构
|
type ESInfo struct {
|
TaskResultInfo
|
Version string `json:"version"`
|
UpdateTime string `json:"updateTime"`
|
LinkTag string `json:"linkTag"`
|
LinkTagInfo []*TaskResultInfo `json:"linkTagInfo"`
|
}
|
|
////so处理后的target
|
//type SoTarget struct {
|
// SrcTarget protomsg.Target
|
// BelongsTargetID string
|
//}
|
|
type TaskBasicInfo struct {
|
CamId string
|
CamName string
|
TaskID string
|
TaskName string
|
HandleTime string
|
ImageId int64
|
ImageTime string
|
IsEndRecord bool
|
FileId string // 数据栈中的文件id
|
}
|
|
//缓存数据信息
|
type CacheInfo struct {
|
TBI TaskBasicInfo
|
|
RuleGroup *protomsg.GroupRule
|
IsAlarm bool
|
AlarmResult *SOResult
|
Image *protomsg.Image
|
}
|
|
//SO处理后的结果
|
//做目标检测的（如人脸检测）IsEvent==false，且需要和底库比对，需要分成多个SOResult，每个SOResult包含一个AlarmTargets，可能多个CompDbinfo
|
//做目标检测的（如人脸检测）IsEvent==false，但不需要和底库比对，也需要分成多个SOResult，每个SOResult包含一个AlarmTargets，只是没有CompDbinfo
|
//不是目标检测IsEvent==true，不需要和底库比对，就只返回一个SOResult，可能包含多个AlarmTargets，但是没有CompDbinfo
|
type SOResult struct {
|
IsEvent bool
|
IsEndRecord bool
|
Targets []SoTarget
|
FirstImage *protomsg.Image
|
CacheImage *protomsg.Image
|
SoRule *SoRule
|
GroupRule *protomsg.GroupRule
|
SoType string
|
}
|
|
type SoTarget struct {
|
SrcTarget protomsg.Target
|
AreaId string
|
AreaName string
|
BelongsTargetID string
|
CompDbInfo []*BaseCompareInfo
|
ImageId int64
|
}
|
|
type EventCache struct {
|
Targets map[uint64]*SoTarget
|
RuleGroups map[string]*protomsg.GroupRule
|
Images map[int64]*protomsg.Image
|
}
|
|
type TargetCache struct {
|
Target *SoTarget
|
RuleGroups map[string]*protomsg.GroupRule
|
Image *protomsg.Image
|
}
|
|
type RuleMessage struct {
|
BaseInfo TaskBasicInfo
|
Events map[string]EventCache
|
Targets map[string]TargetCache
|
}
|
|
type RuleParams struct {
|
SdkArgAlias string
|
Operator string
|
OperatorType string
|
SdkArgValue string
|
}
|
|
type SoRule struct {
|
Id string
|
CameraId string
|
PolygonId string
|
SdkId string
|
Sort int32
|
RuleWithPre string
|
IsSaveAnyhow bool
|
GroupId string
|
Params []RuleParams
|
}
|
|
var AlarmLevelMap = map[int32]string{
|
0: "撤防",
|
1: "一级",
|
2: "二级",
|
3: "三级",
|
4: "四级",
|
5: "五级",
|
6: "撤防",
|
}
|
|
func ParamRestrict(src, min, max, def int32) int32 {
|
if min <= src && src <= max {
|
return src
|
}
|
return def
|
}
|
