把数据定义模块儿提取出来

panlei

2019-08-15 6f1ae979d13028006ccdca01e310d2b5d2d53c5c

 algorithm/intrusion/intrusion.go

@@ -4,7 +4,7 @@
   "basic.com/pubsub/protomsg.git"
   "github.com/knetic/govaluate"
   "ruleprocess/logger"
   "ruleprocess/ruleserver"
   "ruleprocess/structure"
   "strconv"
)

@@ -13,11 +13,11 @@
}

// 过滤规则先筛选出符合条件的目标数量
func filterRule(rule *protomsg.Rule, am *ruleserver.AreaMap) ruleserver.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 []*ruleserver.Arg
         var args []*structure.Arg
         if rule.RuleWithPre == "&&" {
            args = am.FilterData
         } else {
@@ -46,28 +46,28 @@
         }
         am.TargetNum = len(am.FilterData) // 把符合条件的目标数量更新到targetNum字段
         if am.TargetNum > 0 {
            return ruleserver.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
            return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
         } else {
            return ruleserver.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
            return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
         }

      } else {
         return ruleserver.LittleRuleResult{}
         return structure.LittleRuleResult{}
      }

   } else {
      return ruleserver.LittleRuleResult{}
      return structure.LittleRuleResult{}
   }
}

// 给数据库的规则表达式代参 args: 一条子规则，区域数据
func transferParameters(rule *protomsg.Rule, am *ruleserver.AreaMap) ruleserver.LittleRuleResult {
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 ruleserver.LittleRuleResult{am.SdkName, strconv.Itoa(am.TargetNum) + "" + rule.RuleWithPre, rule.Sort} // 如果后面不跟操作符就直接返回数量  比如要跟下一个区域比较数量的就直接返回本区域的数量
            return structure.LittleRuleResult{am.SdkName, strconv.Itoa(am.TargetNum) + "" + rule.RuleWithPre, rule.Sort} // 如果后面不跟操作符就直接返回数量  比如要跟下一个区域比较数量的就直接返回本区域的数量
         }
         //args := am.targetNum     targetNum 已成所有目标的总数量，这里只算yolo的
         var num int = 0
@@ -79,9 +79,9 @@
         formula := strconv.Itoa(num) + " " + rule.Operator + " " + rule.SdkArgValue
         expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
         result, _ := expression.Evaluate(nil)          // 得到数学公式的结果
         return ruleserver.LittleRuleResult{am.SdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort}
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + " " + strconv.FormatBool(result.(bool)), rule.Sort}
         // 加上关于算法的判断条件，不能只有关于规则的，有的算法本身就是一个规则，如个体静止，靠右行,所以，拿到当前子规则的sdkid来判断是否是那些特殊的规则
      }
   }
   return ruleserver.LittleRuleResult{}
   return structure.LittleRuleResult{}
}