提交了时间段的判断的独立文件

panlei

2019-08-15 258ab2a39235115c037394f62783fc9ec1f8fb44

 ruleserver/ruleToformula.go

@@ -27,21 +27,22 @@

// 每个区域内的图片数据集合
type AreaMap struct {
   cameraId      string
   areaId        string
   groupId       string
   taskId        string
   sdkId         string
   sdkName       string
   areaJson      string
   triggerLine   string
   directionLine string
   targetNum     int    // 区域内目标数量
   args          []*Arg // 区域内目标集合
   filterData    []*Arg // 过滤后区域内目标集合
   time          string // 当前时间（用以匹配时间规则）
   keepRight     bool   // 是否靠右行
   isStatic      bool   // 是否静止
   CameraId      string
   AreaId        string
   GroupId       string
   TaskId        string
   SdkId         string
   SdkName       string
   AreaJson      string
   TriggerLine   string
   DirectionLine string
   TargetNum     int    // 区域内目标数量
   Args          []*Arg // 区域内目标集合
   FilterData    []*Arg // 过滤后区域内目标集合
   Time          string // 当前时间（用以匹配时间规则）
   IsEffective   bool   // 规则中是否用到了此区域
   KeepRight     bool   // 是否靠右行
   IsStatic      bool   // 是否静止
}

// sdk输出的图片上单个目标的数据
@@ -90,6 +91,7 @@
   RuleText     string // 文字版规则组
   Location     []Rect // 目标的坐标
   AlarmPolygon string // 触发的报警框
   IsLink       bool   // 是否是联动任务
   Others
}
type Others struct {
@@ -120,7 +122,7 @@
         SdkDataFormat(args.CameraId, arg, cameraPolygons)
      }
      // 跑本摄像机的所有规则组 一组一组跑
      taskGroup := GetRuleGroup(args.CameraId, args.TaskId) // 本摄像机下所有任务组
      taskGroup := GetRuleGroup(args.CameraId, args.TaskId) // 本摄像机本任务下所有规则组
      //logger.Println("看下摄像机下的任务组：",taskRuleList)
      // 得到属于该摄像机的若干组任务的完整规则（跟每一条完整规则比较之后得出本张图像对于某个规则是否报警的结果。放进map，比如本帧图像的id，所碰撞成功的规则id）
      args.RuleResult = make(map[string]interface{})
@@ -162,7 +164,7 @@
func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string, message *protomsg.SdkMessage, label Others) bool {
   defer func() {
      if err := recover(); err != nil {
         logger.Error("比对规则有误", err.(string))
         logger.Error("比对规则有误", err)
      }
   }()
   logger.Info("+++++++++++规则开始运行+++++++++++++++++当前大规则--：", (*groupRule).GroupText)
@@ -188,6 +190,10 @@
               ruleResult := filterRule(groupRule.Rules[j], areaMap)
               if ruleResult.Result != "" {
                  logger.Info("条件规则结果：", ruleResult.Result)
                  // 如果结果为真，把这条规则中的区域置为有效
                  if strings.Contains(ruleResult.Result, "true") {
                     areaMap.IsEffective = true
                  }
                  // 如果此结果为真且当前过的是yolo算法，应记下此规则所对应的sdkName,另外，还要去重 (后加：把此条触碰的区域id也记录下来)
                  if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) {
                     sdkNames = sdkName + ","
@@ -217,6 +223,9 @@
            for _, areaMap := range sdkData.AreaMapList {
               ruleResult := transferParameters(groupRule.Rules[j], areaMap)
               if ruleResult.Result != "" {
                  if strings.Contains(ruleResult.Result, "true") {
                     areaMap.IsEffective = true
                  }
                  logger.Info("数量规则结果：", ruleResult.Result)
                  if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) {
                     sdkNames = sdkName + ","
@@ -243,6 +252,9 @@
            for _, areaMap := range sdkData.AreaMapList {
               ruleResult := timeRuleResult(groupRule.Rules[j], areaMap)
               if ruleResult.Result != "" {
                  if strings.Contains(ruleResult.Result, "true") {
                     areaMap.IsEffective = true
                  }
                  logger.Info("时间规则结果：", ruleResult.Result)
                  if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(sdkNames, sdkName) {
                     sdkNames = sdkName + ","
@@ -318,7 +330,9 @@
            for _, sdkData := range args.Sdkdata {
               if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData里有人脸数据且配置了算法才把符合条件的数据塞进标签里去
                  for _, areaMap := range sdkData.AreaMapList {
                     faces = append(faces, putFaceToResult(areaMap, faces)...)
                     if areaMap.IsEffective {
                        faces = append(faces, putFaceToResult(areaMap, faces)...)
                     }
                  }
               }
            }
@@ -336,12 +350,18 @@
               }
            }
            //logger.Debug("------locations的内容：", locations)
            var islink bool
            if groupRule.SetType == "linkTask" {
               islink = true
            } else {
               islink = false
            }
            if sdkNames != "" {
               args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, label})
               args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink, label,})
               //logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result)))
            }
            if faceFlag {
               args.RuleResult["face"] = append(args.RuleResult["face"].([]FaceResult), FaceResult{Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []Rect{}, polygonId, label}, faces})
               args.RuleResult["face"] = append(args.RuleResult["face"].([]FaceResult), FaceResult{Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, []Rect{}, polygonId, islink, label,}, faces})
               //logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]FaceResult)))
            }
            return true
@@ -362,8 +382,8 @@

func putFaceToResult(am *AreaMap, faceList []Arg) []Arg {
   faces := []Arg{}
   if len(am.filterData) > 0 {
      for _, data := range am.filterData {
   if len(am.FilterData) > 0 {
      for _, data := range am.FilterData {
         flag := true
         for _, face := range faceList {
            if data.Location.X == face.Location.X && data.Location.Y == face.Location.Y && data.Location.Width == face.Location.Width && data.Location.Height == face.Location.Height {
@@ -375,15 +395,14 @@
         }
      }
   }

   //logger.Println("-----------------------------------------------听说你是空的？",faces)
   return faces
}

func putYolosToResult(am *AreaMap) []Rect {
   locations := []Rect{}
   if len(am.filterData) > 0 {
      for _, data := range am.filterData {
   if len(am.FilterData) > 0 {
      for _, data := range am.FilterData {
         locations = append(locations, data.Location)
      }
   }
@@ -472,26 +491,31 @@
            result, _ := expression.Evaluate(nil) // 得到数学公式的结果
            if result.(bool) {
               logger.Info("___________________________________________________________________联动任务报警")
               // 把数组里缓存的数据取出来一起报警
               label.LinkCache = []ResultMsg{}
               for _, ruleRes := range TimeEleList[groupRule.GroupId].RuleResults {
                  label.LinkCache = append(label.LinkCache, ruleRes.CacheData)
               }
               logger.Debug("联动任务缓存了几个数据", len(label.LinkCache))
               for i := 0; i < len(args.RuleResult["yolo"].([]Result)); i++ {
                  if args.RuleResult["yolo"].([]Result)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上
                     args.RuleResult["yolo"].([]Result)[i].Others.LinkCache = label.LinkCache
               if TimeEleList[groupRule.GroupId] != nil { // 极偶尔有情况会等于nil，不知道为啥，做个判断以防宕机
                  // 把数组里缓存的数据取出来一起报警
                  label.LinkCache = []ResultMsg{}
                  for _, ruleRes := range TimeEleList[groupRule.GroupId].RuleResults {
                     label.LinkCache = append(label.LinkCache, ruleRes.CacheData)
                  }
               }
               for i := 0; i < len(args.RuleResult["face"].([]FaceResult)); i++ {
                  if args.RuleResult["face"].([]FaceResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上
                     args.RuleResult["face"].([]FaceResult)[i].Others.LinkCache = label.LinkCache
                  logger.Debug("联动任务缓存了几个数据", len(label.LinkCache))
                  for i := 0; i < len(args.RuleResult["yolo"].([]Result)); i++ {
                     if args.RuleResult["yolo"].([]Result)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上
                        args.RuleResult["yolo"].([]Result)[i].Others.LinkCache = label.LinkCache
                     }
                  }
                  for i := 0; i < len(args.RuleResult["face"].([]FaceResult)); i++ {
                     if args.RuleResult["face"].([]FaceResult)[i].RuleGroupId == groupRule.GroupId { // 把联动数据追加上
                        args.RuleResult["face"].([]FaceResult)[i].Others.LinkCache = label.LinkCache
                     }
                  }
               }
            }
         }
      } else {
         logger.Warn("数组不圆满不打标签")
         // 倒是把打的组规则标签给去掉了啊
         lens := len(args.RuleResult["yolo"].([]Result)) - 1
         args.RuleResult["yolo"] = args.RuleResult["yolo"].([]Result)[0:lens]
      }
   } else { // 没有报警，
      //logger.Info("这帧图像在任务下的一整条规则下（联动任务下就是跟本摄像机像相关的小规则）的判断结果为false")
@@ -517,42 +541,44 @@
   if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
      // 处理的是人脸算法 如果这条规则配置的是人脸算法，过滤完条件之后直接得出结果，因为肯定没有数量条件，自己拼接
      //logger.Info("规则的算法id和区域的算法id：", rule.SdkId, "===", am.sdkId)
      if rule.PolygonId == am.areaId { // 算法和区域都得对的上
      if rule.PolygonId == am.AreaId { // 算法和区域都得对的上

         if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" && rule.SdkArgAlias != "compareBase" {
         if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" {
            //logger.Debug("当前小规则是：",rule)
            flag := "false"
            // 把没有相似者的人脸从filterData中删除
            for index := 0; index < len(am.filterData); {
            for index := 0; index < len(am.FilterData); {
               // 将达不到阈值的相似者从相似者数组中删除
               if len(am.filterData[index].Liker) == 0 {
               logger.Info("看看相似者人数：", len(am.FilterData[index].Liker))
               if len(am.FilterData[index].Liker) == 0 {
                  // Go 语言中切片删除元素的本质是：以被删除元素为分界点，将前后两个部分的内存重新连接起来。不用怀疑，数组删除元素就这么坑爹
                  am.filterData = append(am.filterData[:index], am.filterData[index+1:]...)
                  am.FilterData = append(am.FilterData[:index], am.FilterData[index+1:]...)
               } else {
                  index++
               }
            }
            if len(am.filterData) > 0 {
            if len(am.FilterData) > 0 {
               flag = "true"
            }
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
            logger.Info("---------人脸比对符合条件的数量为：", len(am.FilterData))
            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" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
                  //logger.Info("---------------------------终于等到你")
                  logger.Info("-----------------------过规则之前区域内的人脸数量为：", am.TargetNum)
                  var args []*Arg
                  if rule.RuleWithPre == "&&" {
                     args = am.filterData
                     args = am.FilterData
                     //logger.Info("过滤后的args的长度为：",len(args))
                  } else {
                     args = am.args
                     args = am.Args
                     //logger.Info("没过滤的args的长度为：",len(args))
                  }
                  // 先清空过滤后的数据，再往里塞本次过滤后的数据
                  am.filterData = am.filterData[0:0]
                  am.FilterData = am.FilterData[0:0]
                  //logger.Info("-----------------------人脸过滤的args里的数量：", len(args))
                  for _, arg := range args {
                     var formula string
@@ -569,16 +595,16 @@
                     expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                     result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
                     if result.(bool) {
                        am.filterData = append(am.filterData, arg) // 得到符合条件的过滤数据
                        am.FilterData = append(am.FilterData, arg) // 得到符合条件的过滤数据
                     }
                  }
                  am.targetNum = len(am.filterData) // 把符合条件的目标数量更新到targetNum字段
                  logger.Info("过完条件后的目标数量为：",am.targetNum)
                  if am.targetNum > 0 {
                  am.TargetNum = len(am.FilterData) // 把符合条件的目标数量更新到targetNum字段
                  logger.Info("过完条件后的目标数量为：", am.TargetNum)
                  if am.TargetNum > 0 {
                     logger.Info("！！！！！！！！！人脸检测成功")
                     return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort}
                     return LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
                  } else {
                     return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort}
                     return LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
                  }
               } else {
                  return LittleRuleResult{}
@@ -594,16 +620,16 @@
      }
   } else {
      // 处理的都是yolo数据
      if rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
      if rule.PolygonId == am.AreaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
         if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
            var args []*Arg
            if rule.RuleWithPre == "&&" {
               args = am.filterData
               args = am.FilterData
            } else {
               args = am.args
               args = am.Args
            }
            // 先清空过滤后的数据，再往里塞本次过滤后的数据
            am.filterData = am.filterData[0:0]
            am.FilterData = am.FilterData[0:0]
            //logger.Debug("看看args：：：：：", args)
            for _, arg := range args {
               var formula string
@@ -620,14 +646,14 @@
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
               if result.(bool) {
                  am.filterData = append(am.filterData, arg) // 得到符合条件的过滤数据
                  am.FilterData = append(am.FilterData, arg) // 得到符合条件的过滤数据
               }
            }
            am.targetNum = len(am.filterData) // 把符合条件的目标数量更新到targetNum字段
            if am.targetNum > 0 {
               return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort}
            am.TargetNum = len(am.FilterData) // 把符合条件的目标数量更新到targetNum字段
            if am.TargetNum > 0 {
               return LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
            } else {
               return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort}
               return LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
            }

         } else {
@@ -642,7 +668,7 @@

// 如果有持续时间条件维护开启一个定时器
func duration(rule *protomsg.Rule, groupId string, am *AreaMap, args *SdkDatas, message *protomsg.SdkMessage) {
   if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上  配置的算法要对的上
   if rule.PolygonId == am.AreaId { // 首先规则所对应的区域id要跟区域数据的id对的上  配置的算法要对的上
      if rule.SdkArgAlias == "duration" { //
         logger.Info("当前小规则是：---------", rule)
         // 先看看定时器元素队列中是否有这条规则的定时器,如果有就不能再次创建了
@@ -667,16 +693,16 @@

// 给数据库的规则表达式代参 args: 一条子规则，区域数据
func transferParameters(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
   if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上
   if rule.PolygonId == am.AreaId { // 首先规则所对应的区域id要跟区域数据的id对的上
      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} // 如果后面不跟操作符就直接返回数量  比如要跟下一个区域比较数量的就直接返回本区域的数量
            return LittleRuleResult{am.SdkName, strconv.Itoa(am.TargetNum) + "" + rule.RuleWithPre, rule.Sort} // 如果后面不跟操作符就直接返回数量  比如要跟下一个区域比较数量的就直接返回本区域的数量
         }
         //args := am.targetNum     targetNum 已成所有目标的总数量，这里只算yolo的
         var num int = 0
         for _, data := range am.filterData {
         for _, data := range am.FilterData {
            if data.IsYolo {
               num++
            }
@@ -684,26 +710,14 @@
         formula := strconv.Itoa(num) + " " + rule.Operator + " " + rule.SdkArgValue
         expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
         result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
         return LittleRuleResult{am.sdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort}
         return LittleRuleResult{am.SdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort}
         // 加上关于算法的判断条件，不能只有关于规则的，有的算法本身就是一个规则，如个体静止，靠右行,所以，拿到当前子规则的sdkid来判断是否是那些特殊的规则
      } else if rule.SdkId == "IsStatic" { // 静止算法
         if am.isStatic {
            return LittleRuleResult{}
         } else {
            return LittleRuleResult{}
         }
      } else if rule.SdkId == "KeepRight" { // 靠右行算法
         if am.keepRight {
            return LittleRuleResult{}
         } else {
            return LittleRuleResult{}
         }
      }
   }
   return LittleRuleResult{}
}
func timeRuleResult(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
   if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上
   if rule.PolygonId == am.AreaId { // 首先规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规
         //logger.Info("----------当前时间规则：---------", rule)
         // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段（比如当前时间是周三，应根据区域id查出其周三的几个布防时间段，数组）
@@ -718,11 +732,11 @@
         if rule.Operator == "satisfy" || rule.Operator == "==" { // 满足所选的时间规则
            flag := "false"
            for _, timeSlot := range timeList {
               formula := "'" + timeSlot.Start + "'" + "<" + "'" + am.time + "'"
               formula := "'" + timeSlot.Start + "'" + "<" + "'" + am.Time + "'"
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果

               formula1 := "'" + timeSlot.End + "'" + ">" + "'" + am.time + "'"
               formula1 := "'" + timeSlot.End + "'" + ">" + "'" + am.Time + "'"
               expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式
               result1, _ := expression1.Evaluate(nil)                      // 得到数学公式的结果
               //logger.Info("看看这两尊大神", result, result1)
@@ -731,18 +745,18 @@
                  break
               }
            }
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
            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 + "'"
               formula := "'" + timeSlot.Start + "'" + " < " + "'" + am.Time + "'"
               //logger.Info("-----------------时间规则不满足的公式start：", formula)
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果

               formula1 := "'" + timeSlot.End + "'" + " > " + "'" + am.time + "'"
               formula1 := "'" + timeSlot.End + "'" + " > " + "'" + am.Time + "'"
               //logger.Info("-----------------时间规则不满足的公式end：", formula1)
               expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式
               result1, _ := expression1.Evaluate(nil)                      // 得到数学公式的结果
@@ -751,7 +765,7 @@
                  break
               }
            }
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
            return LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + flag, rule.Sort}
         }

      }