panlei
2019-07-04 d197779cb9a7d3d33eb6afb2ad6d1a7ea7cf6e3f
不符合时间规则的逻辑修改
1个文件已修改
79 ■■■■ 已修改文件
ruleserver/ruleToformula.go 79 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
ruleserver/ruleToformula.go
@@ -113,7 +113,7 @@
    IsStatic    bool       // 是否静止
    ImageWidth  int        // 摄像机拍摄的图像宽 像素
    ImageHeight int        // 摄像机拍摄的图像高 像素
    AreaMapList []AreaMap // 本sdk提取的数据按照区域划分后的数据集合
    AreaMapList []AreaMap  // 本sdk提取的数据按照区域划分后的数据集合
}
// 从算法模块儿拿来的对一帧图像各个算法提取的数据集合
@@ -234,7 +234,7 @@
            fmt.Println("比对规则有误", err.(string))
        }
    }()
    log.Println("+++++++++++规则开始运行+++++++++++++++++当前规则--:",(*groupRule).GroupText)
    log.Println("+++++++++++规则开始运行+++++++++++++++++当前规则--:", (*groupRule).GroupText)
    resultSplice := []*LittleRuleResult{}
    // 先过完条件规则
    for j := 0; j < len(groupRule.Rules); j++ {
@@ -253,7 +253,7 @@
    for _, sdkData := range args.Sdkdata {
        for _, areaMap := range sdkData.AreaMapList {
            for j := 0; j < len(groupRule.Rules); j++ {
                faces = append(faces,putFaceToResult(groupRule.Rules[j], &areaMap)...)
                faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...)
            }
        }
    }
@@ -309,10 +309,10 @@
    if completeFormula != "" {
        log.Println("结果公式-----------:", completeFormula)
        expression, _ := govaluate.NewEvaluableExpression(completeFormula)
        if strings.HasPrefix(completeFormula,"&&") || strings.HasPrefix(completeFormula,"||"){
        if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") {
            panic("规则有误,得到的数学公式不可解析")
        }
        result,_ := expression.Evaluate(nil) // 得到数学公式的结果
        result, _ := expression.Evaluate(nil) // 得到数学公式的结果
        if !result.(bool) {
            // 给这帧数据打上规则组标签
            //args.RuleResult = append(args.RuleResult, Result{TaskId: taskId, RuleGroupId: groupRule.GroupId})
@@ -339,17 +339,17 @@
                sdkName := ""
                for j := 0; j < len(groupRule.Rules); j++ {
                    for _, sdkData := range args.Sdkdata {
                        if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName,sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                        if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName, sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                            sdkName = sdkData.SdkName + " "
                        }
                    }
                }
                if sdkName != "" {
                    args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
                    log.Println("-------------------yolo结果标签",args.RuleResult["yolo"].([]Result))
                    log.Println("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
                }
                if args.RuleResult["face"] != nil {
                    log.Println("-------------------face结果标签",args.RuleResult["face"].([]Arg))
                    log.Println("-------------------face结果标签", args.RuleResult["face"].([]Arg))
                }
                return true
            } else {
@@ -363,7 +363,7 @@
func putFaceToResult(rule *protomsg.Rule, am *AreaMap) []Arg {
    faces := []Arg{}
    if rule.SdkId == am.sdkId && rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a971"{
    if rule.SdkId == am.sdkId && rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
        if len(am.filterData) > 0 {
            for _, data := range am.filterData {
                faces = append(faces, data)
@@ -392,7 +392,7 @@
    // }
    for _, obj := range arg.Photo {
        log.Println("------------------看看sdkData:",arg.SdkName,"的Photo数据----------------", obj, "----顺便看看占比-----:", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale))
        log.Println("------------------看看sdkData:", arg.SdkName, "的Photo数据----------------", obj, "----顺便看看占比-----:", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale))
        if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) {
            // 这步要备齐表达式里所需要的所有参数
            a.targetNum++
@@ -448,7 +448,7 @@
// 联动任务的处理
func LinkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) {
    // new一个定时器,如果以此groupId为标志的定时器不存在的话
    log.Println("------------------------------------------当前是联动任务,规则是:",groupRule.GroupText)
    log.Println("------------------------------------------当前是联动任务,规则是:", groupRule.GroupText)
    var flag bool = true
    var timeEle = TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId}
    for k, timeEle1 := range TimeEleList {
@@ -501,7 +501,7 @@
            for _, va := range timeEle.RuleResults {
                completeFormula = completeFormula + va.RuleWithPre + "" + va.Result
            }
            log.Println("---------------------------联动任务的公式",completeFormula)
            log.Println("---------------------------联动任务的公式", completeFormula)
            if completeFormula != "" {
                expression, _ := govaluate.NewEvaluableExpression(completeFormula)
                result, _ := expression.Evaluate(nil) // 得到数学公式的结果
@@ -510,17 +510,17 @@
                    sdkName := ""
                    for j := 0; j < len(groupRule.Rules); j++ {
                        for _, sdkData := range args.Sdkdata {
                            if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName,sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                            if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName, sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                                sdkName = sdkData.SdkName + " "
                            }
                        }
                    }
                    if sdkName != "" {
                        args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
                        log.Println("-------------------yolo结果标签",args.RuleResult["yolo"].([]Result))
                        log.Println("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
                    }
                    if args.RuleResult["face"] != nil {
                        log.Println("-------------------face结果标签",args.RuleResult["face"].([]Arg))
                        log.Println("-------------------face结果标签", args.RuleResult["face"].([]Arg))
                    }
                    os.Exit(0)
                }
@@ -598,13 +598,13 @@
                            var formula string
                            if rule.SdkArgAlias == "score" {
                                formula = strconv.FormatFloat(arg.Score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                                log.Println("当前相似度小公式:",formula)
                                log.Println("当前相似度小公式:", formula)
                            } else if rule.SdkArgAlias == "proportion" {
                                formula = strconv.FormatFloat(arg.Proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                                log.Println("当前占比小公式:",formula)
                                log.Println("当前占比小公式:", formula)
                            } else {
                                formula = strconv.FormatFloat(arg.Size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                                log.Println("当前尺寸小公式:",formula)
                                log.Println("当前尺寸小公式:", formula)
                            }
                            expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                            result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -633,7 +633,7 @@
        // 处理的都是yolo数据
        if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则,其次规则所对应的区域id要跟区域数据的id对的上
            if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值,占比,尺寸等过滤条件,如果再有,还可以再加
                log.Println("----------当前条件规则:---------",rule)
                log.Println("----------当前条件规则:---------", rule)
                var args []Arg
                if rule.RuleWithPre == "&&" {
                    args = am.filterData
@@ -646,13 +646,13 @@
                    var formula string
                    if rule.SdkArgAlias == "score" {
                        formula = strconv.FormatFloat(arg.Score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前相似度小公式:",formula)
                        log.Println("当前相似度小公式:", formula)
                    } else if rule.SdkArgAlias == "proportion" {
                        formula = strconv.FormatFloat(arg.Proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前占比小公式:",formula)
                        log.Println("当前占比小公式:", formula)
                    } else {
                        formula = strconv.FormatFloat(arg.Size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前尺寸小公式:",formula)
                        log.Println("当前尺寸小公式:", formula)
                    }
                    expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                    result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -753,20 +753,21 @@
    return LittleRuleResult{}
}
func timeRuleResult(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
    if rule.PolygonId == am.areaId && rule.SdkId == am.sdkId{ // 首先规则所对应的区域id要跟区域数据的id对的上
    if rule.PolygonId == am.areaId && rule.SdkId == am.sdkId { // 首先规则所对应的区域id要跟区域数据的id对的上
        if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规则
            log.Println("----------当前时间规则:---------",rule)
            log.Println("----------当前时间规则:---------", rule)
            // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段(比如当前时间是周三,应根据区域id查出其周三的几个布防时间段,数组)
            //fmt.Println("时间规则的测试")
            now := time.Now()
            index := getIndexOfWeek(now.Weekday().String())
            timeList := GetTimeById(rule.SdkArgValue, index)
            log.Println("当天的时间段集合:----------",timeList)
            log.Println("当天的时间段集合:----------", timeList)
            //fmt.Println("从数据库中查出的时间规则:", timeList)
            // 判断图片数据的时间是否符合当前规则 在一个即为true,全不在为false
            flag := "false"
            for _, timeSlot := range timeList {
                if rule.Operator == "satisfy" || rule.Operator == "=="{ // 满足所选的时间规则
            if rule.Operator == "satisfy" || rule.Operator == "==" { // 满足所选的时间规则
                flag := "false"
                for _, timeSlot := range timeList {
                    formula := "'" + timeSlot.Start + "'" + "<" + "'" + am.time + "'"
                    expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                    result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -780,23 +781,29 @@
                        break
                    }
                }
                if rule.Operator == "unsatisfy" || rule.Operator == "!="{ // 不满足所选的时间规则
                    formula := "'"+timeSlot.Start + "'" + " < " + "'" + am.time + "'"
                    log.Println("-----------------时间规则不满足的公式start:",formula)
                return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
            }
            if rule.Operator == "unsatisfy" || rule.Operator == "!=" { // 不满足所选的时间规则
                flag := "true"
                for _, timeSlot := range timeList {
                    formula := "'" + timeSlot.Start + "'" + " < " + "'" + am.time + "'"
                    log.Println("-----------------时间规则不满足的公式start:", formula)
                    expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                    result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
                    formula1 := "'"+timeSlot.End +"'"+ " > " +"'" + am.time+ "'"
                    log.Println("-----------------时间规则不满足的公式end:",formula1)
                    formula1 := "'" + timeSlot.End + "'" + " > " + "'" + am.time + "'"
                    log.Println("-----------------时间规则不满足的公式end:", formula1)
                    expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式
                    result1, _ := expression1.Evaluate(nil)                      // 得到数学公式的结果
                    if result.(bool) && result1.(bool) {
                        flag = "true"
                        flag = "false"
                        break
                    }
                }
                return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
            }
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
        }
    }
    return LittleRuleResult{}
@@ -832,7 +839,7 @@
    //        cameraTimeRule = rule
    //    }
    //}
    _,cameraTimeRule := cache.GetTimeRuleById(id)
    _, cameraTimeRule := cache.GetTimeRuleById(id)
    var timeRangeList []day
    json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList)
    for _, timerange := range timeRangeList {