---

panlei

2019-07-02 0cbddb6f06c9ebbfe9151b5c3425906786b62a3e

 ruleserver/ruleToformula.go

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

   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/knetic/govaluate"
)
@@ -139,7 +139,7 @@
   RuleText    string // 文字版规则组
}
type LittleRuleResult struct {
   SdkName string //记录下此结果是哪个sdk的结果
   SdkName string // 记录下此结果是哪个sdk的结果
   Result  string // 已包含了前置连接符
   Sort    int32
}
@@ -159,19 +159,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 +181,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)
@@ -207,6 +209,7 @@
         for _, taskRule := range taskRuleList {
            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 {
@@ -230,6 +233,7 @@
      }

   }()
   log.Println("+++++++++++规则开始运行+++++++++++++++++当前规则--：",*groupRule)
   resultSplice := []*LittleRuleResult{}
   // 先过完条件规则
   for j := 0; j < len(groupRule.Rules); j++ {
@@ -314,6 +318,7 @@
         for k, timeEle := range TimeEleList {
            if strings.Contains(k, taskId) {
               if timeEle.N != 0 { // 跟这个任务有关的定时器要全部等于0
                  log.Println("———————————-------------不冤，你是被定时器打败的：")
                  flag = false
               }
            }
@@ -324,14 +329,14 @@
            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
                  if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo && !strings.Contains(sdkName,sdkData.SdkName) { // 只记录yolo符合规则的sdk 同时要避免重复拼接
                     sdkName = 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))
            log.Println("-------------------yolo结果标签",args.RuleResult["yolo"].([]Result))
            log.Println("-------------------face结果标签",args.RuleResult["faces"].([]Arg))
            return true
         } else {
            return false
@@ -370,7 +375,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++
@@ -385,6 +390,7 @@
   a.time = time.Unix(time.Now().Unix(), 0).String()[11:16]
   a.keepRight = arg.KeepRight
   a.isStatic = arg.IsStatic
   log.Println("--------------------看看区域数据：",*a)
}

// 将字符串格式的坐标序列化为Point格式
@@ -409,16 +415,17 @@
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
}

// 联动任务的处理
@@ -714,7 +721,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查出其周三的几个布防时间段，数组）
@@ -722,7 +729,7 @@
         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"
@@ -783,14 +790,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 {