---

panlei

2019-07-03 940117bea5848f7a09fe937eb2f0a7e2701ea55e

 ruleserver/ruleToformula.go

@@ -4,12 +4,13 @@
   "encoding/json"
   "fmt"
   "log"
   "os"
   "ruleprocess/cache"
   "sort"
   "strconv"
   "strings"
   "time"

   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/knetic/govaluate"
)
@@ -139,7 +140,7 @@
   RuleText    string // 文字版规则组
}
type LittleRuleResult struct {
   SdkName string //记录下此结果是哪个sdk的结果
   SdkName string // 记录下此结果是哪个sdk的结果
   Result  string // 已包含了前置连接符
   Sort    int32
}
@@ -159,19 +160,21 @@

// 根据摄像机id拿到摄像机所有区域
func GetPolygons(cameraId string) []protomsg.CameraPolygon {
   var api dbapi.CameraApi
   data := api.FindAllPolygons()
   //fmt.Println("查到的所有区域：", data)
   // 根据id从map中拿到区域
   //var api dbapi.CameraApi
   //data := api.FindAllPolygons()
   ////fmt.Println("查到的所有区域：", data)
   //// 根据id从map中拿到区域
   var cameraPolygons []protomsg.CameraPolygon
   for _, item := range data {
      if item.CameraId == cameraId && item.Type != "line" {
         // 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小   x坐标分别*image.width/页面区域宽  y坐标分别*image.height/页面区域高
         // 前台宽高固定
         //log.Println("------------符合条件的区域：",item)
         cameraPolygons = append(cameraPolygons, item)
      }
   }
   //for _, item := range data {
   //   if item.CameraId == cameraId && item.Type != "line" {
   //      // 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小   x坐标分别*image.width/页面区域宽  y坐标分别*image.height/页面区域高
   //      // 前台宽高固定
   //      //log.Println("------------符合条件的区域：",item)
   //      cameraPolygons = append(cameraPolygons, item)
   //   }
   //}
   cameraPolygons = cache.GetPolygonsByCameraId(cameraId)

   //log.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为：", len(cameraPolygons))
   return cameraPolygons
}
@@ -179,7 +182,7 @@
// 把sdk从数据帧上提取的按照区域分类归置
func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) {
   for _, polygon := range cameraPolygons {
      //log.Println("++++++在这儿看一下区域啊", polygon)
      //log.Println("++++++在这儿看一下区域啊:", polygon.Polygon)
      areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine}
      // 为每个摄像机区域填充数据
      areaMap.CountAreaObjs(arg)
@@ -205,14 +208,16 @@
      args.RuleResult["face"] = []Arg{}
      if len(taskRuleList) > 0 {
         for _, taskRule := range taskRuleList {
            //log.Println("------------------------------任务规则：",taskRule)
            ruleList := taskRule.GroupRules // 获取的是task下面的任务组
            taskId := taskRule.TaskId
            //log.Println("------------本组任务下的规则组的数量：",len(ruleList))
            for i := 0; i < len(ruleList); i++ {
               temp := ruleList[i].Rules // temp为一组完整规则 在此需要判断规则是否是联动规则
               if len(temp) > 0 {
                  if ruleList[i].SetType == "linkTask" {
                     // groupId中含有link则为联动任务
                     linkTask(args, ruleList[i], taskId)
                     LinkTask(args, ruleList[i], taskId)
                  } else {
                     // 独立任务的处理
                     RunRule(args, ruleList[i], taskId)
@@ -228,8 +233,8 @@
      if err := recover(); err != nil {
         fmt.Println("比对规则有误", err.(string))
      }

   }()
   log.Println("+++++++++++规则开始运行+++++++++++++++++当前规则--：",(*groupRule).GroupText)
   resultSplice := []*LittleRuleResult{}
   // 先过完条件规则
   for j := 0; j < len(groupRule.Rules); j++ {
@@ -248,11 +253,20 @@
   for _, sdkData := range args.Sdkdata {
      for _, areaMap := range sdkData.AreaMapList {
         for j := 0; j < len(groupRule.Rules); j++ {
            putFaceToResult(groupRule.Rules[j], &areaMap, faces)
            faces = append(faces,putFaceToResult(groupRule.Rules[j], &areaMap)...)
         }
      }
   }
   args.RuleResult["face"] = faces
   // 配了人脸的算法才把人脸的数据甩出来打标签
   flag := false
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
         flag = true
      }
   }
   if flag {
      args.RuleResult["face"] = faces
   }
   // 再过其他数据 这步直接得到结果（真或假） 过目标数量
   for j := 0; j < len(groupRule.Rules); j++ {
      for _, sdkData := range args.Sdkdata {
@@ -298,10 +312,7 @@
      if strings.HasPrefix(completeFormula,"&&") || strings.HasPrefix(completeFormula,"||"){
         panic("规则有误，得到的数学公式不可解析")
      }
      result, err := expression.Evaluate(nil) // 得到数学公式的结果
      if err != nil {
         panic("得到的数学公式不可解析")
      }
      result,_ := expression.Evaluate(nil) // 得到数学公式的结果
      if !result.(bool) {
         // 给这帧数据打上规则组标签
         //args.RuleResult = append(args.RuleResult, Result{TaskId: taskId, RuleGroupId: groupRule.GroupId})
@@ -317,24 +328,29 @@
         for k, timeEle := range TimeEleList {
            if strings.Contains(k, taskId) {
               if timeEle.N != 0 { // 跟这个任务有关的定时器要全部等于0
                  log.Println("———————————-------------不冤，你是被定时器打败的：")
                  flag = false
               }
            }
         }
         if flag {
            fmt.Println("----------------------------终于走完万里长征")
            fmt.Println("___________________________________________________________________________终于走完万里长征")
            // 如果成功了我应该找到规则中涉及到的sdk，记录下他们的sdkname
            sdkName := ""
            for j := 0; j < len(groupRule.Rules); j++ {
               for _, sdkData := range args.Sdkdata {
                  if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk
                     sdkName = sdkName + "," + sdkData.SdkName
                  if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName,sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                     sdkName = sdkData.SdkName + " "
                  }
               }
            }
            // 过完规则后打个标签，告诉调用者本帧数据针对哪个任务哪组规则报警了     后加：可能还不够，还需要区分触发报警的对象，后面往es数据库插数据时要用
            args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
            log.Println("-------------------结果标签",args.RuleResult["yolo"].([]Result))
            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))
            }
            if args.RuleResult["face"] != nil {
               log.Println("-------------------face结果标签",args.RuleResult["face"].([]Arg))
            }
            return true
         } else {
            return false
@@ -345,14 +361,17 @@
   }
}

func putFaceToResult(rule *protomsg.Rule, am *AreaMap, faces []Arg) {
   if rule.SdkId == am.sdkId {
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 len(am.filterData) > 0 {
         for _, data := range am.filterData {
            faces = append(faces, data)
         }
      }
   }
   //log.Println("-----------------------------------------------听说你是空的？",faces)
   return faces
}

// 计算区域内的目标数量以及将相似度、占比、尺寸等打包
@@ -373,7 +392,7 @@

   // }
   for _, obj := range arg.Photo {
      log.Println("------------------看看sdkData的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++
@@ -388,6 +407,7 @@
   a.time = time.Unix(time.Now().Unix(), 0).String()[11:16]
   a.keepRight = arg.KeepRight
   a.isStatic = arg.IsStatic
   //log.Println("--------------------看看区域数据：",*a)
}

// 将字符串格式的坐标序列化为Point格式
@@ -412,21 +432,23 @@
func GetRuleGroup(cameraId string) []*protomsg.TaskGroupArgs {
   // 查询数据库
   // 第一步查出跟这个摄像机相关的group_id(大规则)
   var api dbapi.CameraTaskArgsApi
   all := api.FindAll()
   //fmt.Println("所有规则:", all)
   var taskArgs []*protomsg.TaskGroupArgs
   for _, camArg := range all {
      if camArg.CameraId == cameraId {
         taskArgs = camArg.TaskArgs
      }
   }
   return taskArgs
   //var api dbapi.CameraTaskArgsApi
   //
   ////fmt.Println("所有规则:", all)
   //var taskArgs []*protomsg.TaskGroupArgs
   //for _, camArg := range all {
   //   if camArg.CameraId == cameraId {
   //      taskArgs = camArg.TaskArgs
   //   }
   //}
   all := cache.GetCameraTaskRulesByCameraId(cameraId)
   return all
}

// 联动任务的处理
func linkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) {
func LinkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) {
   // new一个定时器，如果以此groupId为标志的定时器不存在的话
   log.Println("------------------------------------------当前是联动任务，规则是：",groupRule.GroupText)
   var flag bool = true
   var timeEle = TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId}
   for k, timeEle1 := range TimeEleList {
@@ -439,7 +461,7 @@
      timeEle := TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId} // 扔进去一个定时器元素
      //TimeEleList = make(map[string]timeElement)
      TimeEleList[groupRule.GroupId] = &timeEle // 定时器元素以规则组id为键
      fmt.Println("创建了计数器并且计数器集合为：", TimeEleList)
      fmt.Println("---------------------------------------------联动任务创建了计数器并且计数器集合为：", TimeEleList)
      // 得出这组完整规则里涉及到几个摄像机，决定着数组里有几个结构体,去重添加方式
      for j := 0; j < len(groupRule.Rules); j++ {
         var flag1 bool = true
@@ -483,16 +505,23 @@
            expression, _ := govaluate.NewEvaluableExpression(completeFormula)
            result, _ := expression.Evaluate(nil) // 得到数学公式的结果
            if result.(bool) {
               log.Println("___________________________________________________________________联动任务报警")
               sdkName := ""
               for j := 0; j < len(groupRule.Rules); j++ {
                  for _, sdkData := range args.Sdkdata {
                     if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk
                        sdkName = sdkName + "," + sdkData.SdkName
                     if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName,sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                        sdkName = sdkData.SdkName + " "
                     }
                  }
               }
               // 过完规则后打个标签，告诉调用者本帧数据针对哪个任务哪组规则报警了
               args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
               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))
               }
               if args.RuleResult["face"] != nil {
                  log.Println("-------------------face结果标签",args.RuleResult["face"].([]Arg))
               }
               os.Exit(0)
            }
         }
      } else {
@@ -512,12 +541,12 @@

// 过滤规则先筛选出符合条件的目标数量
func filterRule(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
   if rule.SdkId == "FaceDetect" || rule.SdkId == "FaceCompare" {
   if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
      // 处理的是人脸算法 如果这条规则配置的是人脸算法，过滤完条件之后直接得出结果，因为肯定没有数量条件，自己拼接
      log.Println("------------------------------------------还能进这儿了不成？？？")

      if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 算法和区域都得对的上

         if rule.SdkId == "FaceCompare" {
         if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
            // 只需要过滤阈值，过滤完后数组长度大于0即为报警，但要考虑如何对每一张都报警呢
            argValue, err := strconv.ParseFloat(rule.SdkArgValue, 64)
            if err != nil {
@@ -551,7 +580,8 @@
            }
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
         }
         if rule.SdkId == "FaceDetect" { // 人脸检测
         if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 人脸检测
            log.Println("------------------------------------------------------------------------------------------------------------------------------------终于等到你")
            if rule.Operator == "==" || rule.Operator == ">=" || rule.Operator == "<=" || rule.Operator == "<" || rule.Operator == ">" || rule.Operator == "!=" {
               // 如果是不规矩的连接符统统返回false 规则也只能判断人脸的相似度，所以不存在别的连接符
               if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
@@ -581,6 +611,7 @@
                  am.targetNum = len(am.filterData) // 把符合条件的目标数量更新到targetNum字段
               }
               if am.targetNum > 0 {
                  log.Println("！！！！！！！！！人脸检测成功")
                  return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort}
               } else {
                  return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort}
@@ -611,10 +642,13 @@
               var formula string
               if rule.SdkArgAlias == "score" {
                  formula = strconv.FormatFloat(arg.Score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前相似度小公式：",formula)
               } else if rule.SdkArgAlias == "proportion" {
                  formula = strconv.FormatFloat(arg.Proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前占比小公式：",formula)
               } else {
                  formula = strconv.FormatFloat(arg.Size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前尺寸小公式：",formula)
               }
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -715,7 +749,7 @@
   return LittleRuleResult{}
}
func timeRuleResult(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
   if rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上
   if rule.PolygonId == am.areaId && rule.SdkId == am.sdkId{ // 首先规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规则
         log.Println("----------当前时间规则：---------",rule)
         // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段（比如当前时间是周三，应根据区域id查出其周三的几个布防时间段，数组）
@@ -723,11 +757,12 @@
         now := time.Now()
         index := getIndexOfWeek(now.Weekday().String())
         timeList := GetTimeById(rule.SdkArgValue, index)
         log.Println("当天的时间段集合：----------",timeList)
         //fmt.Println("从数据库中查出的时间规则：", timeList)
         // 判断图片数据的时间是否符合当前规则 在一个即为true，全不在为false
         flag := "false"
         for _, timeSlot := range timeList {
            if rule.Operator == "satisfy" { // 满足所选的时间规则
            if rule.Operator == "satisfy" || rule.Operator == "=="{ // 满足所选的时间规则
               formula := "'" + timeSlot.Start + "'" + "<" + "'" + am.time + "'"
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -741,7 +776,7 @@
                  break
               }
            }
            if rule.Operator == "unsatisfy" { // 不满足所选的时间规则
            if rule.Operator == "unsatisfy" || rule.Operator == "!="{ // 不满足所选的时间规则
               formula := timeSlot.Start + "<" + am.time
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -783,14 +818,15 @@

// 取出某个时间规则的第几天的规则段集合
func GetTimeById(id string, index int) []TimeRange {
   var cameraTimeRule protomsg.CameraTimerule
   var api dbapi.CameraApi
   _, rules := api.FindAllTimeRules()
   for _, rule := range rules {
      if rule.Id == id {
         cameraTimeRule = rule
      }
   }
   //var cameraTimeRule protomsg.CameraTimerule
   //var api dbapi.CameraApi
   //_, rules := api.FindAllTimeRules()
   //for _, rule := range rules {
   //   if rule.Id == id {
   //      cameraTimeRule = rule
   //   }
   //}
   _,cameraTimeRule := cache.GetTimeRuleById(id)
   var timeRangeList []day
   json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList)
   for _, timerange := range timeRangeList {