装配目标

panlei

2019-12-17 eb6f08d1eea2430383ad486eca9d3a509db344bf

 algorithm/personTrack/personTrack.go

@@ -1,4 +1,4 @@
package personTrack
package main

import (
   "encoding/json"
@@ -8,11 +8,11 @@
   "strconv"
   "time"

   "basic.com/valib/logger.git"
   "github.com/knetic/govaluate"
   "nanomsg.org/go-mangos"
   "nanomsg.org/go-mangos/protocol/req"
   "nanomsg.org/go-mangos/transport/tcp"
   "github.com/knetic/govaluate"
   "basic.com/valib/logger.git"

   "basic.com/pubsub/protomsg.git"
   "ruleprocess/structure"
@@ -20,7 +20,7 @@

var sender chan []byte = make(chan []byte)
var receiver chan string = make(chan string)
var initFlag bool = false

var sock mangos.Socket
func Init() {
   go Push()
@@ -28,14 +28,12 @@

// 人体跟踪算法
func Entrance(rule *protomsg.Rule, am *structure.AreaMap, lable *structure.Others, args *structure.SdkDatas, message *protomsg.SdkMessage) structure.LittleRuleResult {
   if !initFlag {
      Init()
   }

   if rule.PolygonId == am.AreaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
      //logger.Debug("---------走了人员异常算法", rule.Id, rule.SdkArgAlias, rule.Operator, rule.SdkArgValue, am.AreaId)
      if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" || rule.SdkArgAlias == "" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
         return filterRule(rule, am)
      } else if rule.SdkArgAlias == "threshold"{
      } else if rule.SdkArgAlias == "cmpThreshold"{
         return track(rule,am)
      }
      return structure.LittleRuleResult{}
@@ -78,16 +76,16 @@
      }
      am.TargetNum = len(am.FilterData) // 把符合条件的目标数量更新到targetNum字段
      if am.TargetNum > 0 {
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort,am.FilterData}
      } else {
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort,am.FilterData}
      }

   } else if rule.SdkArgAlias == "" {
      if am.TargetNum > 0 {
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort,am.FilterData}
      } else {
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
         return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort,am.FilterData}
      }
   }
   return structure.LittleRuleResult{}
@@ -107,9 +105,9 @@
      logger.Error("json序列化错误", err)
   }
   dataId := getCompareMsg(b)
   if esId == dataId { // 说明没比到相似的人
   if esId == dataId || dataId == ""{ // 说明没比到相似的人
      logger.Info("没比到人")
      arg.Uuid = dataId
      arg.Uuid = esId
   } else {
      logger.Info("比对出来的dataId:",dataId)
      base := structure.BaseInfo{TargetId:dataId}
@@ -118,12 +116,13 @@
}

func track (rule *protomsg.Rule,am *structure.AreaMap) structure.LittleRuleResult{
   var threshold float32 = 70 // 默认阈值为70
   var threshold float32 = 60 // 默认阈值为70
   if th,err := strconv.ParseFloat(rule.SdkArgValue,32); err == nil {
      threshold = float32(th)
   }
   for _, arg := range am.FilterData {
      fillLiker(threshold,arg,am)
      //logger.Info("比完之后的数据追踪数据是：",arg)
      //if len(arg.Liker) == 0 {
      //   // 如果没有相似者则删除本目标数据
      //   am.FilterData = append(am.FilterData[:i],am.FilterData[i+1:]...)
@@ -135,32 +134,43 @@
   //   return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "false", rule.Sort}
   //}
   // 无论有没有相似者都要返回true
   return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort}
   return structure.LittleRuleResult{am.SdkName, rule.RuleWithPre + "" + "true", rule.Sort,nil}
}

func Push(){
   var err error
   var msg []byte

   logger.Info("追踪请求通道建立")
   if sock, err = req.NewSocket(); err != nil {
      logger.Error("创建请求socket失败: %s", err.Error())
   }
   errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024)
   errSize := sock.SetOption(mangos.OptionMaxRecvSize,20*1024*1024)
   if errSize != nil {
      logger.Error("Failed set MaxRecvSize: %v", err)
   }
   errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 1500)
   if errTimeOut != nil {
      logger.Error("接收响应超时")
      return
   }
   errTimeOut1 := sock.SetOption(mangos.OptionSendDeadline,time.Millisecond * 1500)
   if errTimeOut1 != nil {
      logger.Error("发送超时")
      return
   }
   //sock.AddTransport(ipc.NewTransport())
   sock.AddTransport(tcp.NewTransport())
   serverIP, _ := GetLocalIP()
   if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
   if err = sock.Dial("tcp://"+serverIP+":4012"); err != nil {
      logger.Error("请求socket拨号失败: %s", err.Error())
   }
   logger.Info("序列化数据")
   initFlag = true

   for {
      select {
      case data := <- sender:
         logger.Debug("推送数据:",len(data))
         //trackMap := make(map[string]interface{})
         //_ = json.Unmarshal(data, &trackMap)
         //logger.Info("请求张增飞比对数据参数:",trackMap)
         if err = sock.Send(data); err != nil {
            logger.Error("推送socket发送数据失败: %s", err.Error())
         }