换log

panlei

2019-11-29 087bf92d707de1a838451719da505f69b6934087

 algorithm/middleware/middleware.go

@@ -2,23 +2,31 @@

import (
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/logger.git"
   logger "github.com/jeanphorn/log4go"
   "encoding/json"
   "github.com/knetic/govaluate"
   "plugin"
   "ruleprocess/cache"
   "ruleprocess/ruleserver"
   "ruleprocess/structure"
   "sort"
   "strings"
   "time"
)

func Entrance (args *structure.SdkDatas,groupRule protomsg.GroupRule) ([]*structure.LittleRuleResult,string,string){
func Entrance (args *structure.SdkDatas,groupRule protomsg.GroupRule,lable *structure.Others,message *protomsg.SdkMessage) (bool,string,string){
   resultSplice := []*structure.LittleRuleResult{}
   sdkNames := ""
   polygonId := ""
   // 先过完条件规则
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkArgAlias == "score" || groupRule.Rules[j].SdkArgAlias == "proportion" || groupRule.Rules[j].SdkArgAlias == "size" || (groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && groupRule.Rules[j].SdkArgAlias != "time_rule"){

      carFlag := groupRule.Rules[j].SdkArgAlias == "license" ||  groupRule.Rules[j].SdkArgAlias == "nConfidence" || groupRule.Rules[j].SdkArgAlias == "fConfidence" ||
         groupRule.Rules[j].SdkArgAlias == "nType" || groupRule.Rules[j].SdkArgAlias == "nVehicleColor1" ||  groupRule.Rules[j].SdkArgAlias == "nVehicleBright" || groupRule.Rules[j].SdkArgAlias == "nCarColor" ||
      groupRule.Rules[j].SdkArgAlias == "fVDConf" || groupRule.Rules[j].SdkArgAlias == "nVehicleColor2" ||  groupRule.Rules[j].SdkArgAlias == "nColor" || groupRule.Rules[j].SdkArgAlias == "nPlateFlag"

      if groupRule.Rules[j].SdkArgAlias == "score" || groupRule.Rules[j].SdkArgAlias == "proportion" || groupRule.Rules[j].SdkArgAlias == "size" ||
         (groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && groupRule.Rules[j].SdkArgAlias != "time_rule") || carFlag{
         for _, sdkData := range args.Sdkdata {
            // 根据规则的sdkId查出其对应的ipcId，用ipcId去找该比对的数据
            sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId)
@@ -29,9 +37,9 @@
            sdkName := sdk.SdkName
            //logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId)
            if ipcId == sdkData.IpcId {
               //logger.Info("当前走的规则是--：", sdkName, "---","")
               logger.Info("当前走的规则是--：", groupRule.Rules[j],sdkName,groupRule.Rules[j].SdkArgAlias, "---","")
               for _, areaMap := range sdkData.AreaMapList {
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap)
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message)
                  if ruleResult.Result != "" {
                     logger.Info("条件规则结果：", ruleResult.Result)
                     // 如果结果为真，把这条规则中的区域置为有效
@@ -52,6 +60,7 @@
         }
      }
   }
   // 人脸
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && groupRule.Rules[j].SdkArgAlias != "time_rule"{
         for _, sdkData := range args.Sdkdata {
@@ -66,9 +75,81 @@
            if ipcId == sdkData.IpcId {
               //logger.Info("当前走的规则是--：", sdkName, "---","")
               for _, areaMap := range sdkData.AreaMapList {
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap)
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message)
                  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 + " "
                     }
                     if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(polygonId, groupRule.Rules[j].PolygonId) {
                        polygonId += groupRule.Rules[j].PolygonId + ","
                     }
                     resultSplice = append(resultSplice, &ruleResult)
                  }
               }
            }
         }
      }
   }
   // 个体静止
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a977" && groupRule.Rules[j].SdkArgAlias == "duration"{
         for _, sdkData := range args.Sdkdata {
            // 根据规则的sdkId查出其对应的ipcId，用ipcId去找该比对的数据
            sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId)
            if err != nil {
               logger.Error("没查到sdk的信息---", err)
            }
            ipcId := sdk.IpcId
            sdkName := sdk.SdkName
            //logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId)
            if ipcId == sdkData.IpcId {
               logger.Info("当前走的规则是--：", groupRule.Rules[j],sdkName,groupRule.Rules[j].SdkArgAlias, "---","")
               for _, areaMap := range sdkData.AreaMapList {
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message)
                  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 + " "
                     }
                     if strings.Contains(ruleResult.Result, "true") && ipcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && !strings.Contains(polygonId, groupRule.Rules[j].PolygonId) {
                        polygonId += groupRule.Rules[j].PolygonId + ","
                     }
                     resultSplice = append(resultSplice, &ruleResult)
                  }
               }
            }
         }
      }
   }
   // 车辆
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a978" && (groupRule.Rules[j].SdkArgAlias == "nCarCount" || groupRule.Rules[j].SdkArgAlias == "nCarLogoCount"){
         for _, sdkData := range args.Sdkdata {
            // 根据规则的sdkId查出其对应的ipcId，用ipcId去找该比对的数据
            sdk, err := cache.GetSdkById(groupRule.Rules[j].SdkId)
            if err != nil {
               logger.Error("没查到sdk的信息---", err)
            }
            ipcId := sdk.IpcId
            sdkName := sdk.SdkName
            //logger.Info("规则的ipcId与sdkData的IpcId:", ipcId, "===", sdkData.IpcId)
            if ipcId == sdkData.IpcId {
               logger.Info("当前走的规则是--：", groupRule.Rules[j],sdkName,groupRule.Rules[j].SdkArgAlias, "---","")
               for _, areaMap := range sdkData.AreaMapList {
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message)
                  if ruleResult.Result != "" {
                     logger.Info("车牌结果：", ruleResult.Result)
                     // 如果结果为真，把这条规则中的区域置为有效
                     if strings.Contains(ruleResult.Result, "true") {
                        areaMap.IsEffective = true
@@ -100,7 +181,7 @@
            sdkName := sdk.SdkName
            if ipcId == sdkData.IpcId {
               for _, areaMap := range sdkData.AreaMapList {
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap)
                  ruleResult := CallSo(sdk.Id, groupRule.Rules[j], areaMap,lable,args,message)
                  if ruleResult.Result != "" {
                     if strings.Contains(ruleResult.Result, "true") {
                        areaMap.IsEffective = true
@@ -148,7 +229,37 @@
         }
      }
   }
   return resultSplice,sdkNames,polygonId
   // 将数组按sort排序
   sort.Sort(ruleserver.ResultList(resultSplice))
   // 排序后取各自的结果和连接符拼出规则表达式得出结果
   completeFormula := ""
   for _, va := range resultSplice {
      completeFormula = completeFormula + va.Result
   }
   if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") || strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") {
      // 以这些开头的基本是联动任务
      if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") {
         completeFormula = completeFormula[2:]
      }
      if strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") {
         completeFormula = completeFormula[1:]
      }
      logger.Info("-------------------看看拔毛后的表达式：", completeFormula)
      //expression, _ := govaluate.NewEvaluableExpression(completeFormula)
      //result, _ := expression.Evaluate(nil) // 得到数学公式的结果
      //return result.(bool)
   }
   if completeFormula != "" {
      logger.Info("结果公式-----------：", completeFormula)
      expression, err2 := govaluate.NewEvaluableExpression(completeFormula)
      if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || err2 != nil {
         panic("规则有误，得到的数学公式不可解析")
      }
      result, _ := expression.Evaluate(nil) // 得到数学公式的结果
      return result.(bool),sdkNames,polygonId
   } else {
      return false,sdkNames,polygonId
   }
}

func timeRuleResult(rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult {
@@ -208,7 +319,7 @@
   return structure.LittleRuleResult{}
}

func CallSo(sdkId string,rule *protomsg.Rule, am *structure.AreaMap) structure.LittleRuleResult{
func CallSo(sdkId string,rule *protomsg.Rule, am *structure.AreaMap,lable *structure.Others,args *structure.SdkDatas,message *protomsg.SdkMessage) structure.LittleRuleResult{
   // 根据sdkId查出其对应的sdk的soName，调用相应so的Entrance方法
   var soName = ""
   if sdkId == "812b674b-2375-4589-919a-5c1c3278a97e" {
@@ -219,6 +330,10 @@
      soName = "personUnsual.so"
   } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
      soName = "faceCompare.so"
   } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a977" {
      soName = "static.so"
   } else if sdkId == "812b674b-2375-4589-919a-5c1c3278a978" {
      soName = "plate.so"
   }
   //soInfo,errr := cache.GetSoInfoById(sdkId)
   //if errr != nil {
@@ -233,7 +348,8 @@
   if err1 != nil {
      panic("没有找到入口函数")
   }
   ruleResult := f.(func(rule *protomsg.Rule, am *structure.AreaMap)structure.LittleRuleResult)(rule,am)
   logger.Info("发给so的数据：",sdkId,rule.SdkArgValue,rule.Operator,rule.SdkArgAlias)
   ruleResult := f.(func(rule *protomsg.Rule, am *structure.AreaMap,lable *structure.Others, args *structure.SdkDatas,message *protomsg.SdkMessage)structure.LittleRuleResult)(rule,am,lable,args,message)
   return ruleResult
}