派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -89,6 +89,12 @@ Location []Rect // 目标的坐标 AlarmPolygon string // 触发的报警框 } // 过规则库打上的标签 type FaceResult struct { Result Args []*Arg } type LittleRuleResult struct { SdkName string // 记录下此结果是哪个sdk的结果 Result string // 已包含了前置连接符 @@ -101,6 +107,7 @@ // 拿到本摄像机的区域 cameraPolygons := GetPolygons(args.CameraId) // 把所有的sdk提取的数据都按所属摄像机的区域归置 logger.Debug("当前摄像机id为：",message.Cid,"当前摄像机执行的任务是：",message.Tasklab.Taskname) for _, arg := range args.Sdkdata { SdkDataFormat(args.CameraId, arg, cameraPolygons) } @@ -111,7 +118,7 @@ args.RuleResult = make(map[string]interface{}) args.RuleResult["yolo"] = []Result{} args.RuleResult["face"] = []Arg{} if len(taskGroup.GroupRules) > 0 { if taskGroup != nil && len(taskGroup.GroupRules) > 0 { for _, group := range taskGroup.GroupRules { //logger.Println("------------------------------任务规则：",taskRule) taskId := taskGroup.TaskId @@ -154,11 +161,7 @@ sdkName := sdk.SdkName logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId) if ipcId == sdkData.IpcId { logger.Info("当前走的规则的算法是--：", sdkName, "---") logger.Info("========================有几个区域：", len(sdkData.AreaMapList)) for _, areaMap := range sdkData.AreaMapList { logger.Info("检查一下区域的具体数据：", areaMap.areaJson) } logger.Info("当前走的规则是--：", sdkName, "---","") for _, areaMap := range sdkData.AreaMapList { ruleResult := filterRule(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { @@ -254,8 +257,8 @@ } if completeFormula != "" { logger.Info("结果公式-----------：", completeFormula) expression, _ := govaluate.NewEvaluableExpression(completeFormula) if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") { expression, err := govaluate.NewEvaluableExpression(completeFormula) if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || err != nil{ panic("规则有误，得到的数学公式不可解析") } result, _ := expression.Evaluate(nil) // 得到数学公式的结果 @@ -450,24 +453,8 @@ if rule.PolygonId == am.areaId { // 算法和区域都得对的上 if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" && rule.SdkArgAlias != "compareBase" { argValue, err := strconv.ParseFloat(rule.SdkArgValue, 64) if err != nil { logger.Error("规则配置的阈值非法") return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort} } logger.Debug("当前小规则是：",rule) flag := "false" for _, obj := range am.filterData { for index := 0; index < len(obj.Liker); { // 将达不到阈值的相似者从相似者数组中删除 logger.Debug("=======================相似值：", float64(obj.Liker[index].CompareScore*100)) if float64(obj.Liker[index].CompareScore*100) < argValue { // Go 语言中切片删除元素的本质是：以被删除元素为分界点，将前后两个部分的内存重新连接起来。不用怀疑，数组删除元素就这么坑爹 obj.Liker = append(obj.Liker[:index], obj.Liker[index+1:]...) } else { index++ } } } // 把没有相似者的人脸从filterData中删除 for index := 0; index < len(am.filterData); { // 将达不到阈值的相似者从相似者数组中删除 @@ -484,6 +471,7 @@ return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort} } if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 人脸检测 logger.Debug("当前小规则是：",rule) if rule.Operator == "==" || rule.Operator == ">=" || rule.Operator == "<=" || rule.Operator == "<" || rule.Operator == ">" || rule.Operator == "!=" { // 如果是不规矩的连接符统统返回false 规则也只能判断人脸的相似度，所以不存在别的连接符 if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加 @@ -536,7 +524,6 @@ // 处理的都是yolo数据 if rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上 if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加 logger.Info("----------当前条件规则：---------", rule) var args []*Arg if rule.RuleWithPre == "&&" { args = am.filterData @@ -585,6 +572,7 @@ func duration(rule *protomsg.Rule, groupId string, am *AreaMap, args *SdkDatas, message *protomsg.SdkMessage) { if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 配置的算法要对的上 if rule.SdkArgAlias == "duration" { // logger.Info("当前小规则是：---------", rule) // 先看看定时器元素队列中是否有这个摄像机这个区域的定时器,如果有就不能再次创建了 var flag bool = true for k, _ := range TimeEleList { @@ -608,7 +596,8 @@ // 给数据库的规则表达式代参 args: 一条子规则，区域数据 func transferParameters(rule *protomsg.Rule, am *AreaMap) LittleRuleResult { if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 if rule.SdkArgAlias == "targetNum" { // 如果参数是要区域内目标数量 即yolo 人脸不会有数量 if rule.SdkArgAlias == "objCount" { // 如果参数是要区域内目标数量 即yolo 人脸不会有数量 logger.Info("当前小规则是：---------", rule) //logger.Info("得出结果阶段", "比较的规则是：", rule) if rule.Operator == "" { return LittleRuleResult{am.sdkName, strconv.Itoa(am.targetNum) + "" + rule.RuleWithPre, rule.Sort} // 如果后面不跟操作符就直接返回数量 比如要跟下一个区域比较数量的就直接返回本区域的数量 @@ -623,7 +612,7 @@ formula := strconv.Itoa(num) + " " + rule.Operator + " " + rule.SdkArgValue expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式 result, _ := expression.Evaluate(nil) // 得到数学公式的结果 return LittleRuleResult{am.sdkName, strconv.FormatBool(result.(bool)) + "" + rule.RuleWithPre, rule.Sort} return LittleRuleResult{am.sdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort} // 加上关于算法的判断条件，不能只有关于规则的，有的算法本身就是一个规则，如个体静止，靠右行,所以，拿到当前子规则的sdkid来判断是否是那些特殊的规则 } else if rule.SdkId == "IsStatic" { // 静止算法 if am.isStatic { @@ -643,7 +632,7 @@ } func timeRuleResult(rule *protomsg.Rule, am *AreaMap) LittleRuleResult { if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规则 if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规 logger.Info("----------当前时间规则：---------", rule) // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段（比如当前时间是周三，应根据区域id查出其周三的几个布防时间段，数组） //logger.Info("时间规则的测试")
