From c0d0a8df1bbe1cf0cb663d9a2732f5575572c4d7 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 10 七月 2019 11:44:43 +0800
Subject: [PATCH] 重要更新,塞符合条件的人脸标签、

---
 ruleserver/ruleToformula.go |   70 ++++++++++++++--------------------
 1 files changed, 29 insertions(+), 41 deletions(-)

diff --git a/ruleserver/ruleToformula.go b/ruleserver/ruleToformula.go
index 68f5380..8be94d4 100644
--- a/ruleserver/ruleToformula.go
+++ b/ruleserver/ruleToformula.go
@@ -185,8 +185,8 @@
 // 鎶妔dk浠庢暟鎹抚涓婃彁鍙栫殑鎸夌収鍖哄煙鍒嗙被褰掔疆
 func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) {
 	logger.Info("==================================鏈瑂dkData涓В鍑烘潵鐨勭洰鏍囨暟鎹�=======================================")
-	for _,photo := range arg.Photo{
-		logger.Info("--------鐪嬬湅瑙f瀽鍑烘潵鐨勬暟鎹�---", cameraId,arg.IpcId,photo.Rects,photo.Score)
+	for _, photo := range arg.Photo {
+		logger.Info("--------鐪嬬湅瑙f瀽鍑烘潵鐨勬暟鎹�---", cameraId, arg.IpcId, photo.Rects, photo.Score)
 	}
 	for _, polygon := range cameraPolygons {
 		//logger.Println("++++++鍦ㄨ繖鍎跨湅涓�涓嬪尯鍩熷晩:", polygon.Polygon)
@@ -258,7 +258,7 @@
 	}
 	for j := 0; j < len(groupRule.Rules); j++ {
 		// 鍙湁鏄汉鑴稿姣斿苟涓旈厤缃殑鍙傛暟鏄姣斿簳搴撴墠璧帮紙鑻ラ厤鐨勫姣旈槇鍊煎垯涓嶈蛋锛�
-		if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972"{
+		if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
 			for _, sdkData := range args.Sdkdata {
 				if sdkData.IpcId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 鎼滅储鍒颁汉鑴告娴嬬殑sdkdata 鐜板湪鍏充簬浜鸿劯鐨勫彧鏈変粬涓�涓猻dk锛屽叏鎵句粬
 					for _, areaMap := range sdkData.AreaMapList {
@@ -273,9 +273,9 @@
 										logger.Error("getBaseInfo瑙e帇閿欒", err1)
 									}
 									for key, val := range m {
-										baseinfo,err1 := esutil.Dbpersoninfosbyid(key)
+										baseinfo, err1 := esutil.Dbpersoninfosbyid(key)
 										if err1 != nil {
-											logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊",err1)
+											logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1)
 										}
 										baseinfo.CompareScore = val
 										arg.Liker = append(arg.Liker, &baseinfo)
@@ -293,7 +293,7 @@
 										for key, val := range m {
 											baseinfo, err1 := esutil.Dbpersoninfosbyid(key)
 											if err1 != nil {
-												logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊",err1)
+												logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1)
 											}
 											baseinfo.CompareScore = val
 											arg.Liker = append(arg.Liker, &baseinfo)
@@ -449,21 +449,21 @@
 		} else {
 			// 鏈�鍚庢垚鍔熸姤璀︽墠鎶婄鍚堟潯浠剁殑浜鸿劯鏁版嵁濉炶繘缁撴灉鏍囩閲�
 			faces := []Arg{}
+			faceFlag := false
+			for j := 0; j < len(groupRule.Rules); j++ {
+				if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
+					faceFlag = true
+				}
+			}
 			for _, sdkData := range args.Sdkdata {
-				for _, areaMap := range sdkData.AreaMapList {
-					for j := 0; j < len(groupRule.Rules); j++ {
-						faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...)
+				if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData閲屾湁浜鸿劯鏁版嵁涓旈厤缃簡绠楁硶鎵嶆妸绗﹀悎鏉′欢鐨勬暟鎹杩涙爣绛鹃噷鍘�
+					for _, areaMap := range sdkData.AreaMapList {
+						faces = append(faces, putFaceToResult(&areaMap)...)
 					}
 				}
 			}
 			// 閰嶄簡浜鸿劯鐨勭畻娉曟墠鎶婁汉鑴哥殑鏁版嵁鐢╁嚭鏉ユ墦鏍囩
-			flag11 := false
-			for j := 0; j < len(groupRule.Rules); j++ {
-				if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
-					flag11 = true
-				}
-			}
-			if flag11 {
+			if faceFlag {
 				args.RuleResult["face"] = faces
 			}
 			// 鍘荤湅姹犲瓙閲屾槸鍚︽湁涓庢湰甯ф暟鎹湁鍏崇殑瀹氭椂鍣紝濡傛灉鏈夛紝鐪嬫鏃舵槸鍚﹁蛋鍒�0锛屾病鏈夋瀹氭椂鍣ㄦ垨鏈夊畾鏃跺櫒璧板埌0鐨勮瘽杩斿洖鎴愬姛鎶ヨ
@@ -481,22 +481,12 @@
 				// 鎶婁粬浠殑浣嶇疆鏁版嵁涔熶紶涓嬪幓
 				locations := []Rect{}
 				for _, sdkData := range args.Sdkdata {
-					for _, areaMap := range sdkData.AreaMapList {
-						for j := 0; j < len(groupRule.Rules); j++ {
-							locations = append(locations, putYolosToResult(groupRule.Rules[j], &areaMap)...)
+					if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 鎶妝olo鏁版嵁鐨勫悇涓洰鏍囩殑鍧愭爣杈撳嚭鏂逛究鍚庨潰鐢绘
+						for _, areaMap := range sdkData.AreaMapList {
+							locations = append(locations, putYolosToResult(&areaMap)...)
 						}
 					}
 				}
-				//// 閰嶄簡浜鸿劯鐨勭畻娉曟墠鎶婁汉鑴哥殑鏁版嵁鐢╁嚭鏉ユ墦鏍囩
-				//flag11 := false
-				//for j := 0; j < len(groupRule.Rules); j++ {
-				//	if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
-				//		flag11 = true
-				//	}
-				//}
-				//if flag11 {
-				//	args.RuleResult["face"] = faces
-				//}
 				if sdkNames != "" {
 					args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText, locations})
 					logger.Info("-------------------yolo缁撴灉鏍囩", args.RuleResult["yolo"].([]Result))
@@ -514,26 +504,23 @@
 	}
 }
 
-func putFaceToResult(rule *protomsg.Rule, am *AreaMap) []Arg {
+func putFaceToResult(am *AreaMap) []Arg {
 	faces := []Arg{}
-	if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" {
-		if len(am.filterData) > 0 {
-			for _, data := range am.filterData {
-				faces = append(faces, data)
-			}
+	if len(am.filterData) > 0 {
+		for _, data := range am.filterData {
+			faces = append(faces, data)
 		}
 	}
+
 	//logger.Println("-----------------------------------------------鍚浣犳槸绌虹殑锛�",faces)
 	return faces
 }
 
-func putYolosToResult(rule *protomsg.Rule, am *AreaMap) []Rect {
+func putYolosToResult(am *AreaMap) []Rect {
 	locations := []Rect{}
-	if rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a97e" && rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a972" {
-		if len(am.filterData) > 0 {
-			for _, data := range am.filterData {
-				locations = append(locations, data.Location)
-			}
+	if len(am.filterData) > 0 {
+		for _, data := range am.filterData {
+			locations = append(locations, data.Location)
 		}
 	}
 	//logger.Println("-----------------------------------------------鍚浣犳槸绌虹殑锛�",faces)
@@ -752,6 +739,7 @@
 						}
 						// 鍏堟竻绌鸿繃婊ゅ悗鐨勬暟鎹紝鍐嶅線閲屽鏈杩囨护鍚庣殑鏁版嵁
 						am.filterData = am.filterData[0:0]
+						logger.Info("-----------------------浜鸿劯杩囨护鐨刟rgs锛�", args)
 						for _, arg := range args {
 							var formula string
 							if rule.SdkArgAlias == "score" {

--
Gitblit v1.8.0