From 5913d9cfae97f9b15721c4facf9ceb8749da350e Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 25 六月 2019 13:52:08 +0800
Subject: [PATCH] 占比乘以100
---
ruleserver/ruleToformula.go | 105 ++++++++++++++++++++++++++++++++++++----------------
1 files changed, 72 insertions(+), 33 deletions(-)
diff --git a/ruleserver/ruleToformula.go b/ruleserver/ruleToformula.go
index dd8b75f..820432a 100644
--- a/ruleserver/ruleToformula.go
+++ b/ruleserver/ruleToformula.go
@@ -3,6 +3,7 @@
import (
"encoding/json"
"fmt"
+ "log"
"sort"
"strconv"
"strings"
@@ -60,6 +61,12 @@
score float64 // 鍖哄煙鍐呯殑鐩爣鐨勭浉浼煎害
proportion float64 // 鍖哄煙鍐呯殑鐩爣鐨勫崰姣�
size float64 // 鍖哄煙鍐呯殑鐩爣鐨勫昂瀵�
+ liker []LikePerson
+}
+
+type LikePerson struct {
+ Id string // 涓庝箣鐩镐技鐨勫簳搴撲汉鍛樼殑id
+ Score float64 // 涓庡簳搴撲汉鍛樼殑鐩镐技鍊�
}
// 姣忎釜鍖哄煙鍐呯殑鍥剧墖鏁版嵁闆嗗悎
@@ -83,19 +90,20 @@
// sdk杈撳嚭鐨勫浘鐗囦笂鍗曚釜鐩爣鐨勬暟鎹�
type PhotoMap struct {
Rects Rect // 鐭╁舰鍖哄煙鍙傛暟
- Score float64 // 鐩镐技搴﹀緱鍒�
+ Score float64 // 鐩镐技搴﹀緱鍒嗭紙鏈夊澶х▼搴﹀儚涓�涓洰鏍囥�備汉鑴革紝浜轰綋鎴栬溅绛夌瓑锛�
+ Liker []LikePerson // 濡傛灉鏄汉鑴哥殑璇濆挨鍏舵槸姣斿锛屽簲瀛樹笅浠栬窡搴曞簱鐨勪汉鍛樼殑鐩镐技鎯呭喌 yolo鐨勮瘽缁檔il灏辫
}
// 浠庨�氶亾涓幏鍙栫殑sdk杈撳嚭鐨勫浘鍍忔暟鎹�(鐩墠涓昏鏄痽olo绠楁硶鐨勬暟鎹�)
type ArgsFromSdk struct {
- Photo []PhotoMap
CameraId string
TaskId string
- KeepRight bool // 鏄惁闈犲彸琛� 绠楁硶鍒ゆ柇鐨勪笌涓婁竴甯у浘鍍忕殑姣旇緝缁撴灉
- IsStatic bool // 鏄惁闈欐
- ImageWidth int // 鎽勫儚鏈烘媿鎽勭殑鍥惧儚瀹� 鍍忕礌
- ImageHeight int // 鎽勫儚鏈烘媿鎽勭殑鍥惧儚楂� 鍍忕礌
- RuleResult []Result // 杩囧畬瑙勫垯鍚庢墦鐨勬爣绛�
+ Photo []PhotoMap // yolo绠楁硶缁撴瀯锛屼篃鍙互瀛樹汉鑴哥殑鏁版嵁锛屾瘯绔熶汉鑴镐腑鑳界敤瑙勫垯鏉ユ祴鐨勮繕鏄偅浜涘弬鏁�
+ KeepRight bool // 鏄惁闈犲彸琛� 绠楁硶鍒ゆ柇鐨勪笌涓婁竴甯у浘鍍忕殑姣旇緝缁撴灉
+ IsStatic bool // 鏄惁闈欐
+ ImageWidth int // 鎽勫儚鏈烘媿鎽勭殑鍥惧儚瀹� 鍍忕礌
+ ImageHeight int // 鎽勫儚鏈烘媿鎽勭殑鍥惧儚楂� 鍍忕礌
+ RuleResult []Result // 杩囧畬瑙勫垯鍚庢墦鐨勬爣绛�
}
// 灏嗕紶閫掕繃鏉ョ殑鍙傛暟杞寲涓�
@@ -110,22 +118,14 @@
type Result struct {
TaskId string // 浠诲姟id
RuleGroupId string // 瑙勫垯缁刬d
+ AlarmLevel int32 // 鎶ヨ绛夌骇
+ RuleText string // 鏂囧瓧鐗堣鍒欑粍
}
// 鍖呭惈N鏉¤鍒欏厓绱犵殑涓�鏁存潯瑙勫垯
type CompleteRule struct {
rule string
}
-
-// 鎽勫儚鏈哄尯鍩� 璺熸暟鎹簱鏄犲皠鐨�
-// type CameraPolygon struct {
-// Id string `json:"id"`
-// CameraId string `json:"camera_id"`
-// Name string `json:"name"`
-// Polygon string `json:"polygon"` // 鍧愭爣鐐瑰尯鍩�
-// TriggerLine string `json:"trigger_line"`
-// DirectionLine string `json:"direction_line"`
-// }
// 鏍规嵁鎽勫儚鏈篿d鎷垮埌鎽勫儚鏈烘墍鏈夊尯鍩�
func GetPolygons(cameraId string) []protomsg.CameraPolygon {
@@ -196,11 +196,17 @@
// 灏嗗瓧绗︿覆鏍煎紡鐨勫潗鏍囧簭鍒楀寲涓篜oint鏍煎紡
func Json2points(areaPoints string) []Point {
var pts []Point
-
- err := json.Unmarshal([]byte(areaPoints), &pts)
- if err != nil {
- fmt.Println("json.Unmarshal閿欒", err)
- panic("搴忓垪鍖栧潗鏍囧紓甯革紝绋嬪簭閫�鍑�")
+ if areaPoints == "" {
+ pts = append(pts, Point{0, 0})
+ pts = append(pts, Point{0, 540})
+ pts = append(pts, Point{960, 540})
+ pts = append(pts, Point{960, 0})
+ } else {
+ err := json.Unmarshal([]byte(areaPoints), &pts)
+ if err != nil {
+ fmt.Println("json.Unmarshal閿欒", err)
+ panic("搴忓垪鍖栧潗鏍囧紓甯革紝绋嬪簭閫�鍑�")
+ }
}
return pts
}
@@ -303,19 +309,26 @@
func singleTask(aml *AreaMapList, arg *ArgsFromSdk, groupRule *protomsg.GroupRule, taskId string) bool {
var completeFormula string = ""
for _, areaMap := range aml.areaMapList {
+ //fmt.Println("褰撳墠瑙勫垯缁勪负---------锛�",groupRule)
for j := 0; j < len(groupRule.Rules); j++ {
// 鍏堣繃瀹屾潯浠舵暟鎹�
filterRule(groupRule.Rules[j], &areaMap)
}
for j := 0; j < len(groupRule.Rules); j++ {
- // 鍐嶈繃鍏朵粬鏁版嵁 杩欐鐩存帴寰楀埌缁撴灉锛堢湡鎴栧亣锛�
+ // 鍐嶈繃鍏朵粬鏁版嵁 杩欐鐩存帴寰楀埌缁撴灉锛堢湡鎴栧亣锛� 杩囩洰鏍囨暟閲�
flag := transferParameters(groupRule.Rules[j], &areaMap)
if flag != "" {
fmt.Println("寰楀嚭鐨勭粨鏋�", flag)
completeFormula = completeFormula + groupRule.Rules[j].RuleWithPre + "" + flag
}
}
-
+ if completeFormula == "" {
+ flag := splice1(&areaMap)
+ if flag != "" {
+ fmt.Println("寮鸿鎷煎噾涓�涓汉鏁版槸鍚﹀ぇ浜�0鐨勭粨鏋�", flag)
+ completeFormula = flag
+ }
+ }
for j := 0; j < len(groupRule.Rules); j++ {
// 杩欐杩囩殑鏄椂闂磋鍒欙紙鏃堕棿娈电瓑锛�
flag := timeRuleResult(groupRule.Rules[j], &areaMap)
@@ -324,16 +337,18 @@
completeFormula = completeFormula + groupRule.Rules[j].RuleWithPre + "" + flag
}
}
+
for j := 0; j < len(groupRule.Rules); j++ {
// 鏈�鍚庤繃鎸佺画鏃堕棿绛夋椂闂寸淮搴︾殑鏉′欢
duration(groupRule.Rules[j], &areaMap)
}
+
}
fmt.Println("鎷煎嚭鐨勬暟瀛﹀叕寮忎负:==== ", completeFormula)
if completeFormula != "" {
expression, _ := govaluate.NewEvaluableExpression(completeFormula)
result, _ := expression.Evaluate(nil) // 寰楀埌鏁板鍏紡鐨勭粨鏋�
- fmt.Println("杩欏抚鍥惧儚鍦ㄤ换鍔′笅鐨勯櫎浜嗘寔缁椂闂村鐨勪竴鏁存潯瑙勫垯涓嬬殑鍒ゆ柇缁撴灉", result)
+ //fmt.Println("杩欏抚鍥惧儚鍦ㄤ换鍔′笅鐨勯櫎浜嗘寔缁椂闂村鐨勪竴鏁存潯瑙勫垯涓嬬殑鍒ゆ柇缁撴灉", result)
// 鐢变簬澶╃劧鎴栫殑鍏崇郴锛屾弧瓒充竴涓氨璇ユ姤璀�,鍗宠甯ф暟鎹浜庢煇涓换鍔＄殑鏌愪釜瑙勫垯缁勫簲璇ユ姤璀�
if !result.(bool) { // 濡傛灉涓嶇鍚堟潯浠讹紝搴旇閲嶇疆瀹氭椂鍣ㄥ厓绱狅紝绛夌鍚堟椂鍐嶅紑鍚�,鎶妅ey涓寘鍚换鍔d鐨則imeEle閮介噸缃�
for k, timeEle := range TimeEleList {
@@ -355,7 +370,7 @@
if flag {
fmt.Println("瀹氭椂鍣ㄦ姤璀︿簡")
// 杩囧畬瑙勫垯鍚庢墦涓爣绛撅紝鍛婅瘔璋冪敤鑰呮湰甯ф暟鎹拡瀵瑰摢涓换鍔″摢缁勮鍒欐姤璀︿簡
- arg.RuleResult = append(arg.RuleResult, Result{TaskId: taskId, RuleGroupId: groupRule.GroupId})
+ arg.RuleResult = append(arg.RuleResult, Result{taskId, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
return true
} else {
return false
@@ -377,9 +392,11 @@
for i := 0; i < len(ruleList); i++ {
temp := ruleList[i].Rules // temp涓轰竴缁勫畬鏁磋鍒� 鍦ㄦ闇�瑕佸垽鏂鍒欐槸鍚︽槸鑱斿姩瑙勫垯
if len(temp) > 0 {
- if strings.Contains(ruleList[i].GroupId, "link") { // groupId涓惈鏈塴ink鍒欎负鑱斿姩浠诲姟
+ if strings.Contains(ruleList[i].GroupId, "link") {
+ // groupId涓惈鏈塴ink鍒欎负鑱斿姩浠诲姟
linkTask(aml, arg, ruleList[i], taskId)
- } else { // 鐙珛浠诲姟鐨勫鐞�
+ } else {
+ // 鐙珛浠诲姟鐨勫鐞�
singleTask(aml, arg, ruleList[i], taskId)
}
}
@@ -388,7 +405,7 @@
}
}
-// 杩囨护瑙勫垯鍏堢瓫閫変汉鏁�
+// 杩囨护瑙勫垯鍏堢瓫閫夊嚭绗﹀悎鏉′欢鐨勭洰鏍囨暟閲�
func filterRule(rule *protomsg.Rule, am *AreaMap) {
if rule.PolygonId == am.areaId { // 棣栧厛瑙勫垯鎵�瀵瑰簲鐨勫尯鍩焛d瑕佽窡鍖哄煙鏁版嵁鐨刬d瀵圭殑涓�
if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 鍒ゆ柇鐨勬槸鐩镐技鍊硷紝鍗犳瘮锛屽昂瀵哥瓑杩囨护鏉′欢锛屽鏋滃啀鏈夛紝杩樺彲浠ュ啀鍔�
@@ -401,12 +418,14 @@
}
// 鍏堟竻绌鸿繃婊ゅ悗鐨勬暟鎹紝鍐嶅線閲屽鏈杩囨护鍚庣殑鏁版嵁
am.filterData = am.filterData[0:0]
+ log.Println("鐪嬩竴涓嬪綋鍓嶅皬瑙勫垯锛�",*rule)
for _, arg := range args {
var formula string
if rule.SdkArgAlias == "score" {
formula = strconv.FormatFloat(arg.score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 寰楀埌瀛楃涓插叕寮�
} else if rule.SdkArgAlias == "proportion" {
formula = strconv.FormatFloat(arg.proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 寰楀埌瀛楃涓插叕寮�
+ fmt.Println("鍗犳瘮鐨勫瓧绗︿覆鍏紡锛�--------",formula)
} else {
formula = strconv.FormatFloat(arg.size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 寰楀埌瀛楃涓插叕寮�
}
@@ -417,7 +436,6 @@
}
}
am.targetNum = len(am.filterData) // 鎶婄鍚堟潯浠剁殑鐩爣鏁伴噺鏇存柊鍒皌argetNum瀛楁
- //fmt.Println("绛涢�夊畬鍚庣殑鍐呭锛�", am)
}
}
}
@@ -446,10 +464,20 @@
}
}
+// 鍐椾綑鎷兼帴
+func splice1(am *AreaMap) string {
+ args := am.targetNum
+ log.Println("鐪嬬湅鍖哄煙鍐呯洰鏍囨暟閲忥細----------",args)
+ formula := strconv.Itoa(args) + " " + ">" + "0"
+ expression, _ := govaluate.NewEvaluableExpression(formula) // 寰楀埌鏁板鍏紡
+ result, _ := expression.Evaluate(nil) // 寰楀埌鏁板鍏紡鐨勭粨鏋�
+ return strconv.FormatBool(result.(bool))
+}
+
// 缁欐暟鎹簱鐨勮鍒欒〃杈惧紡浠ｅ弬 args: 涓�鏉″瓙瑙勫垯锛屽尯鍩熸暟鎹�
func transferParameters(rule *protomsg.Rule, am *AreaMap) string {
if rule.PolygonId == am.areaId { // 棣栧厛瑙勫垯鎵�瀵瑰簲鐨勫尯鍩焛d瑕佽窡鍖哄煙鏁版嵁鐨刬d瀵圭殑涓�
- if rule.SdkArgAlias == "targetNum" { // 濡傛灉鍙傛暟鏄鍖哄煙鍐呯洰鏍囨暟閲�
+ if rule.SdkArgAlias == "targetNum" { // 濡傛灉鍙傛暟鏄鍖哄煙鍐呯洰鏍囨暟閲� 鍗硑olo
//fmt.Println("寰楀嚭缁撴灉闃舵", "姣旇緝鐨勮鍒欐槸锛�", rule)
if rule.Operator == "" {
return strconv.Itoa(am.targetNum) // 濡傛灉鍚庨潰涓嶈窡鎿嶄綔绗﹀氨鐩存帴杩斿洖鏁伴噺 姣斿瑕佽窡涓嬩竴涓尯鍩熸瘮杈冩暟閲忕殑灏辩洿鎺ヨ繑鍥炴湰鍖哄煙鐨勬暟閲�
@@ -460,18 +488,29 @@
result, _ := expression.Evaluate(nil) // 寰楀埌鏁板鍏紡鐨勭粨鏋�
return strconv.FormatBool(result.(bool))
// 鍔犱笂鍏充簬绠楁硶鐨勫垽鏂潯浠讹紝涓嶈兘鍙湁鍏充簬瑙勫垯鐨勶紝鏈夌殑绠楁硶鏈韩灏辨槸涓�涓鍒欙紝濡備釜浣撻潤姝紝闈犲彸琛�,鎵�浠ワ紝鎷垮埌褰撳墠瀛愯鍒欑殑sdkid鏉ュ垽鏂槸鍚︽槸閭ｄ簺鐗规畩鐨勮鍒�
- } else if rule.SdkId == "涓綋闈欐" { // 鏆傛椂鐢ㄦ眽瀛椾唬鏇垮暒锛屾櫄浜涙浛鎹㈡垚姝ｅ紡鐨刬d
+ } else if rule.SdkId == "IsStatic" { // 闈欐绠楁硶
if am.isStatic {
return "true"
} else {
return "false"
}
- } else if rule.SdkId == "闈犲彸琛�" { // 鏆傛椂鐢ㄦ眽瀛椾唬鏇垮暒锛屾櫄浜涙浛鎹㈡垚姝ｅ紡鐨刬d
+ } else if rule.SdkId == "KeepRight" { // 闈犲彸琛岀畻娉�
if am.keepRight {
return "true"
} else {
return "false"
}
+ } else if rule.SdkId == "FaceDetect" { // 浜鸿劯妫�娴�
+ if rule.Operator == "==" || rule.Operator == ">=" || rule.Operator == "<=" || rule.Operator == "<" || rule.Operator == ">" || rule.Operator == "!=" {
+ // 濡傛灉鏄笉瑙勭煩鐨勮繛鎺ョ缁熺粺杩斿洖false 瑙勫垯涔熷彧鑳藉垽鏂汉鑴哥殑鐩镐技搴︼紝鎵�浠ヤ笉瀛樺湪鍒殑杩炴帴绗�
+
+ return "false"
+ } else {
+ return "false"
+ }
+ } else if rule.SdkId == "FaceCompare"{
+ // 鍙渶瑕佽繃婊ら槇鍊硷紝杩囨护瀹屽悗鏁扮粍闀垮害澶т簬0鍗充负鎶ヨ锛屼絾濡備綍瀵规瘡涓�寮犻兘鎶ヨ鍛�
+
}
}
--
Gitblit v1.8.0