---看看比分

panlei

2019-08-01 408290cf3fe23f639c8718d984d66809fad50ed8

---看看比分

 ruleserver/readyDataForRule.go

@@ -13,7 +13,9 @@
    "strconv"
    "time"
)
var serverIp,serverPort,dbTablePersons string

var serverIp, serverPort, dbTablePersons string

type conf struct {
    PhotoUrl       string `yaml:"photoUrl"`
    VideoPersons   string `yaml:"videoPersons"`
@@ -22,6 +24,7 @@
    ServerPort     string `yaml:"serverPort"`
    DbTablePersons string `yaml:"dbTablePersons"`
}

func Init() {
    data, err := ioutil.ReadFile("./config/conf.yml")
    if err != nil {
@@ -31,7 +34,7 @@
    c := conf{}
    //把yaml形式的字符串解析成struct类型
    yaml.Unmarshal(data, &c)
    logger.Debug("配置文件的值为：",c)
    logger.Debug("配置文件的值为：", c)
    serverIp = c.ServerIp
    serverPort = c.ServerPort
    dbTablePersons = c.DbTablePersons
@@ -39,36 +42,38 @@
    if err1 != nil {
        logger.Error("查询本机信息失败！")
    }
    logger.Debug("本机信息和server信息：",localConfig,serverIp,serverPort)
    bigCache.Init(dbTablePersons,serverIp,serverPort,localConfig.ServerId)
    logger.Debug("本机信息和server信息：", localConfig, serverIp, serverPort)
    bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId)
}

type BaseInfo struct {
    TableId string `json:"tableId"`
    TableName string `json:"tableName"`
    BwType string `json:"bwType"`
    TableId      string  `json:"tableId"`
    TableName    string  `json:"tableName"`
    BwType       string  `json:"bwType"`
    CompareScore float64 `json:"compareScore"`
    PersonId string     `json:"personId"`
    PersonName string    `json:"personName"`
    PersonPicUrl string `json:"personPicUrl"`
    PhoneNum string `json:"phoneNum"`
    Sex string `json:"sex"`
    IdCard string `json:"idCard"`
    MonitorLevel string `json:"monitorLevel"`
    Content string `json:"content"`
    PersonId     string  `json:"personId"`
    PersonName   string  `json:"personName"`
    PersonPicUrl string  `json:"personPicUrl"`
    PhoneNum     string  `json:"phoneNum"`
    Sex          string  `json:"sex"`
    IdCard       string  `json:"idCard"`
    MonitorLevel string  `json:"monitorLevel"`
    Content      string  `json:"content"`
}

// 以摄像机id查出跟其相关的所有任务下的所有规则组
func GetRuleGroup(cameraId string,taskId string) *protomsg.TaskGroupArgs {
func GetRuleGroup(cameraId string, taskId string) *protomsg.TaskGroupArgs {
    all := cache.GetCameraTaskRulesByCameraId(cameraId)
    //logger.Info("==================所有的任务的规则组：",all)
    var taskGroup  *protomsg.TaskGroupArgs
    for _,task := range all  {
    var taskGroup *protomsg.TaskGroupArgs
    for _, task := range all {
        if task.TaskId == taskId {
            taskGroup = task
        }
    }
    return taskGroup
}

// 根据摄像机id拿到摄像机所有区域
func GetPolygons(cameraId string) []protomsg.CameraPolygon {
    var cameraPolygons []protomsg.CameraPolygon
@@ -78,8 +83,8 @@
}

// 保留四位小数
func Retain(f float32)(float64) {
    s:= strconv.FormatFloat(float64(f), 'f', 4, 64)
func Retain(f float32) (float64) {
    s := strconv.FormatFloat(float64(f), 'f', 4, 64)
    v, err := strconv.ParseFloat(s, 64)
    if err != nil {
        logger.Error("保留四位小数转换错误")
@@ -105,7 +110,6 @@
    return nil
}


// 根据传入的字符串得到其在一周内的索引 周一到周日分别对应1到7
func getIndexOfWeek(weekday string) int {
    var weekdays = [7]string{"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"}
@@ -129,7 +133,7 @@
// 将字符串格式的坐标序列化为Point格式
func Json2points(areaPoints string) []Point {
    var pts []Point
    if areaPoints == "[]" || areaPoints == ""{
    if areaPoints == "[]" || areaPoints == "" {
        logger.Error("=====================此区域为全部区域")
        pts = append(pts, Point{0, 0})
        pts = append(pts, Point{0, 540})
@@ -146,8 +150,8 @@
}

// 给目标填充liker
func (arg *Arg) fillLiker(tableId []string,compareThreshold float32) {
    bytes := bigCache.GetComparePersonBaseInfo(tableId,arg.Feature,compareThreshold)
func (arg *Arg) fillLiker(tableId []string, compareThreshold float32) {
    bytes := bigCache.GetComparePersonBaseInfo(tableId, arg.Feature, compareThreshold)
    var m map[string]float32
    err1 := json.Unmarshal(bytes, &m)
    if err1 != nil {
@@ -155,33 +159,34 @@
    }
    //logger.Info("----------------------------------------map是", m)
    ids := []string{}
    for key,_ := range m {
        ids = append(ids,key)
    for key, _ := range m {
        ids = append(ids, key)
    }
    logger.Info("------------------------------------------------------------------------------------------------------------------------")
    //logger.Info("=====================人员id的集合为：",ids)
    if len(ids) > 0 {
        baseinfos, err1 := esutil.Dbpersoninfosbyid(ids,dbTablePersons,serverIp,serverPort)
        baseinfos, err1 := esutil.Dbpersoninfosbyid(ids, dbTablePersons, serverIp, serverPort)
        if err1 != nil {
            logger.Error("查询底库人员信息出错", err1)
        }
        for _,baseinfo := range baseinfos  {
        for _, baseinfo := range baseinfos {
            // 根据tableId查询底库信息给liker赋值
            //logger.Info("---------看看每个底库人员的信息：",baseinfo.Id,baseinfo.PersonName)
            var tableIds []string
            tableIds = append(tableIds,baseinfo.TableId) // 虽然是传入数组返回数组的接口，但我按单个的使用了
            table,err := esutil.Dbtablefosbyid(tableIds,"dbtables",serverIp,serverPort)
            if err != nil || len(table) == 0{
                logger.Error("根据id查询底库信息出错！",err,"--返回值长度为：",len(table))
            tableIds = append(tableIds, baseinfo.TableId) // 虽然是传入数组返回数组的接口，但我按单个的使用了
            table, err := esutil.Dbtablefosbyid(tableIds, "dbtables", serverIp, serverPort)
            if err != nil || len(table) == 0 {
                logger.Error("根据id查询底库信息出错！", err, "--返回值长度为：", len(table))
            }
            base := BaseInfo{TableId:baseinfo.TableId,TableName:table[0].TableName,BwType:table[0].BwType,CompareScore:Retain(m[baseinfo.Id])*100,PersonId:baseinfo.Id,PersonName:baseinfo.PersonName,PersonPicUrl:baseinfo.PersonPicUrl,PhoneNum:baseinfo.PhoneNum,Sex:baseinfo.Sex,IdCard:baseinfo.IdCard,MonitorLevel:baseinfo.MonitorLevel,Content:baseinfo.Reserved}
            //logger.Debug("看看这个base的对比值是多少：",base)
            logger.Debug("看看这个base的对比值是多少：",Retain(m[baseinfo.Id]) * 100)
            base := BaseInfo{TableId: baseinfo.TableId, TableName: table[0].TableName, BwType: table[0].BwType, CompareScore: Retain(m[baseinfo.Id]) * 100, PersonId: baseinfo.Id, PersonName: baseinfo.PersonName, PersonPicUrl: baseinfo.PersonPicUrl, PhoneNum: baseinfo.PhoneNum, Sex: baseinfo.Sex, IdCard: baseinfo.IdCard, MonitorLevel: baseinfo.MonitorLevel, Content: baseinfo.Reserved}
            //os.Exit(1)
            arg.Liker = append(arg.Liker, &base)
        }
    }

}

// 人脸比对
func Compare(args *SdkDatas, groupRule *protomsg.GroupRule) {
    compareFlag := 0
@@ -192,7 +197,7 @@
        if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { // 配了人脸比对
            if groupRule.Rules[j].SdkArgAlias == "compareBase" && groupRule.Rules[j].SdkArgValue != "" { // 配了比对底库的参数但不是配的全部底库
                compareFlag = 2
                tableIds = append(tableIds,groupRule.Rules[j].SdkArgValue)// 最后会多一个tableId，切出来的数组取len-1就可以
                tableIds = append(tableIds, groupRule.Rules[j].SdkArgValue) // 最后会多一个tableId，切出来的数组取len-1就可以
            }
            if groupRule.Rules[j].SdkArgAlias == "compareBase" && groupRule.Rules[j].SdkArgValue == "" { // 配的参数是比对全部底库
                compareFlag = 1
@@ -212,7 +217,7 @@
        logger.Info("没有配置对比底库参数")
        return
    }
    logger.Debug("=====配置的阈值为：",threshold)
    logger.Debug("=====配置的阈值为：", threshold)
    for j := 0; j < len(groupRule.Rules); j++ {
        if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
            for _, sdkData := range args.Sdkdata {
@@ -224,10 +229,10 @@
                            //logger.Info("--------------看看compareFlag的值和tableId和areaMap.args的长度：",compareFlag,tableIds,len(areaMap.args))
                            for _, arg := range areaMap.args {
                                if compareFlag == 1 {
                                    arg.fillLiker(nil,threshold)
                                    arg.fillLiker(nil, threshold)
                                }
                                if compareFlag == 2 {
                                    arg.fillLiker(tableIds,threshold)
                                    arg.fillLiker(tableIds, threshold)
                                }
                                //logger.Info("-------------------成功给liker赋值,长度为：", len(arg.Liker))
                            }
@@ -284,4 +289,4 @@
        areaMap.CountAreaObjs(arg)
        arg.AreaMapList = append(arg.AreaMapList, &areaMap)
    }
}
}