From 573a1e019fc00e171b7df7105fe69b414a490966 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期六, 09 十一月 2019 15:03:36 +0800 Subject: [PATCH] 把一部分判断放进中间件 --- ruleserver/readyDataForRule.go | 47 ++++++++++++++++------------------------------- 1 files changed, 16 insertions(+), 31 deletions(-) diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go index b547b9c..91b36da 100644 --- a/ruleserver/readyDataForRule.go +++ b/ruleserver/readyDataForRule.go @@ -3,6 +3,7 @@ import ( "basic.com/dbapi.git" "basic.com/pubsub/protomsg.git" + "basic.com/valib/logger.git" "encoding/json" "errors" "fmt" @@ -13,7 +14,6 @@ "nanomsg.org/go-mangos/transport/tcp" "net" "ruleprocess/cache" - "basic.com/valib/logger.git" "ruleprocess/structure" "strconv" "time" @@ -31,8 +31,6 @@ } func Init() { - //logger.Debug("鏈満淇℃伅鍜宻erver淇℃伅锛�", localConfig, serverIp, serverPort) - //bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId) go Push1() } @@ -63,7 +61,6 @@ taskGroup = task } } - logger.Debug("褰撳墠鏁版嵁甯ц鍖归厤鐨勮鍒欑粍锛�-------------------------","鎽勫儚鏈篿d:",cameraId,"浠诲姟id",taskId) if taskGroup == nil { return nil } else { @@ -78,7 +75,7 @@ func GetPolygons(cameraId string) []protomsg.CameraPolygon { var cameraPolygons []protomsg.CameraPolygon cameraPolygons = cache.GetPolygonsByCameraId(cameraId) - logger.Debug("------=======鏌ョ湅涓嬪叏閮ㄥ尯鍩燂細",cameraPolygons) + //logger.Debug("------=======鏌ョ湅涓嬪叏閮ㄥ尯鍩燂細",cameraPolygons) return cameraPolygons } @@ -89,16 +86,16 @@ } func Decimal(value float32) float64 { value1, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(value)), 64) - logger.Info("鍒濇淇濈暀涓や綅鎴恠tr:",value1) + //logger.Info("鍒濇淇濈暀涓や綅鎴恠tr:",value1) n10 := math.Pow10(4) value2 := math.Trunc((value1+0.5/n10)*n10) / n10 - logger.Info("鍒濇淇濈暀涓や綅鎴恠tr::::",value2) + //logger.Info("鍒濇淇濈暀涓や綅鎴恠tr::::",value2) return value2 } // 鍙栧嚭鏌愪釜鏃堕棿瑙勫垯鐨勭鍑犲ぉ鐨勮鍒欐闆嗗悎 -func GetTimeById(id string, index int) []TimeRange { +func GetTimeById(id string, index int) []structure.TimeRange { _, cameraTimeRule := cache.GetTimeRuleById(id) - var timeRangeList []day + var timeRangeList []structure.Day err := json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList) if err != nil { logger.Error("鍙栨椂闂磋鍒欐椂鍙嶅簭鍒楀寲閿欒锛�") @@ -123,24 +120,17 @@ return 0 } -type TimeRange struct { - Start string `json:"start"` - End string `json:"end"` -} -type day struct { - Day int `json:"day"` // 鏍囩ず褰撳墠鏄熸湡鍑� - TimeRange []TimeRange `json:"time_range"` // 褰撳ぉ鐨勫嚑涓椂闂存 -} + // 灏嗗瓧绗︿覆鏍煎紡鐨勫潗鏍囧簭鍒楀寲涓篜oint鏍煎紡 -func Json2points(areaPoints string) []Point { - var pts []Point +func Json2points(areaPoints string) []structure.Point { + var pts []structure.Point if areaPoints == "[]" || areaPoints == "" { logger.Error("=====================姝ゅ尯鍩熶负鍏ㄩ儴鍖哄煙") - pts = append(pts, Point{0, 0}) - pts = append(pts, Point{0, 540}) - pts = append(pts, Point{960, 540}) - pts = append(pts, Point{960, 0}) + pts = append(pts, structure.Point{0, 0}) + pts = append(pts, structure.Point{0, 540}) + pts = append(pts, structure.Point{960, 540}) + pts = append(pts, structure.Point{960, 0}) } else { err := json.Unmarshal([]byte(areaPoints), &pts) if err != nil { @@ -162,7 +152,6 @@ } bytes := getCompareMsg(comArg) - //bytes := []byte{} var scResult protomsg.SdkCompareResult err1 := proto.Unmarshal(bytes, &scResult) if err1 != nil { @@ -285,11 +274,11 @@ areaPoints := Json2points(a.AreaJson) logger.Info("鐪嬬湅鍥剧墖鐨剋idth鍜宧eight:",arg.ImageWidth,arg.ImageHeight) - widthScale := float64(arg.ImageWidth / 960) - heigthScale := float64(arg.ImageHeight / 540) + widthScale := float64(arg.ImageWidth) / 960 + heigthScale := float64(arg.ImageHeight) / 540 for _, obj := range arg.Photo { //logger.Info("------------------鐪嬬湅sdkData:", arg.SdkName, "鐨凱hoto鏁版嵁----------------", 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) { + if obj.Score >= threshold && float64(obj.Rects.Width*obj.Rects.Height) >= size && PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) >= intersectionper { // 杩欐瑕佸榻愯〃杈惧紡閲屾墍闇�瑕佺殑鎵�鏈夊弬鏁� a.TargetNum++ arg1 := structure.Arg{obj.Id,obj.Score, PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale), float64(obj.Rects.Width * obj.Rects.Height), obj.IsYolo, obj.Rects, obj.Feature, obj.ThftRes, []*structure.BaseInfo{}} @@ -299,9 +288,6 @@ } } logger.Info("鍖哄煙鏄細",areaPoints,"鍖哄煙鍐呯洰鏍囨暟閲忎负锛�",a.TargetNum,"---",len(a.FilterData)) - //for _,tar := range a.FilterData { - // logger.Info("鍏蜂綋鐩爣鐨勫潗鏍囦负",tar.Id,tar.Location) - //} a.Time = time.Unix(time.Now().Unix(), 0).String()[11:16] a.KeepRight = arg.KeepRight a.IsStatic = arg.IsStatic @@ -362,7 +348,6 @@ logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg)) return msg } - func Push1(){ //var sock mangos.Socket -- Gitblit v1.8.0