抽出入侵和人员异常算法

panlei

2019-08-15 09da5bfd5f39f6e49e35f4c08a425680b317861b

 algorithm/intrusion/intrusion.go

@@ -7,16 +7,22 @@
   "ruleprocess/structure"
   "strconv"
)

func Entrance() {

//入侵算法
func Entrance(rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult {
   if rule.PolygonId == am.AreaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
         return filterRule(rule,am)
      } else {
         return structure.LittleRuleResult{}
      }
   } else {
      return structure.LittleRuleResult{}
   }
}

// 过滤规则先筛选出符合条件的目标数量
func filterRule(rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult {
   // 处理的都是yolo数据
   if rule.PolygonId == am.AreaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
         var args []*structure.Arg
         if rule.RuleWithPre == "&&" {
            args = am.FilterData
@@ -50,38 +56,4 @@
         } else {
            return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
         }

      } else {
         return structure.LittleRuleResult{}
      }

   } else {
      return structure.LittleRuleResult{}
   }
}

// 给数据库的规则表达式代参 args: 一条子规则，区域数据
func transferParameters(rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult {
   if rule.PolygonId == am.AreaId { // 首先规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "objCount" { // 如果参数是要区域内目标数量 即yolo 人脸不会有数量
         //logger.Info("当前小规则是：---------", rule)
         //logger.Info("得出结果阶段", "比较的规则是：", rule)
         if rule.Operator == "" {
            return structure.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 {
            if data.IsYolo {
               num++
            }
         }
         formula := strconv.Itoa(num) + " " + rule.Operator + " " + rule.SdkArgValue
         expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
         result, _ := expression.Evaluate(nil)          // 得到数学公式的结果
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort}
         // 加上关于算法的判断条件，不能只有关于规则的，有的算法本身就是一个规则，如个体静止，靠右行,所以，拿到当前子规则的sdkid来判断是否是那些特殊的规则
      }
   }
   return structure.LittleRuleResult{}
}