~jinersong/ruleprocess.git

			@@ -112,13 +112,13 @@
			IsStatic bool // 是否静止
			ImageWidth int // 摄像机拍摄的图像宽像素
			ImageHeight int // 摄像机拍摄的图像高像素
			AreaMapList []*AreaMap // 本sdk提取的数据按照区域划分后的数据集合
			AreaMapList []AreaMap // 本sdk提取的数据按照区域划分后的数据集合
			}

			// 从算法模块儿拿来的对一帧图像各个算法提取的数据集合
			type SdkDatas struct {
			CameraId string
			Sdkdata []SdkData
			Sdkdata []*SdkData
			RuleResult map[string]interface{} // 过完规则后打的标签 face: []Arg yolo: []Result
			}

			@@ -158,17 +158,18 @@
			}

			// 根据摄像机id拿到摄像机所有区域
			func GetPolygons(cameraId string) []*protomsg.CameraPolygon {
			func GetPolygons(cameraId string) []protomsg.CameraPolygon {
			var api dbapi.CameraApi
			data := api.FindAllPolygons()
			//fmt.Println("查到的所有区域：", data)
			// 根据id从map中拿到区域
			var cameraPolygons []*protomsg.CameraPolygon
			var cameraPolygons []protomsg.CameraPolygon
			for _, item := range data {
			if item.CameraId == cameraId {
			if item.CameraId == cameraId && item.Type != "line" {
			// 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小 x坐标分别image.width/页面区域宽 y坐标分别image.height/页面区域高
			// 前台宽高固定
			cameraPolygons = append(cameraPolygons, &item)
			//log.Println("------------符合条件的区域：",item)
			cameraPolygons = append(cameraPolygons, item)
			}
			}
			//log.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为：", len(cameraPolygons))
			@@ -176,25 +177,26 @@
			}

			// 把sdk从数据帧上提取的按照区域分类归置
			func SdkDataFormat(cameraId string, arg SdkData, cameraPolygons []protomsg.CameraPolygon) {
			// cameraPolygons := GetPolygons(arg.CameraId)
			// list := AreaMapList{}
			func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) {
			for _, polygon := range cameraPolygons {
			//log.Println("++++++在这儿看一下区域啊", polygon)
			areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine}
			// 为每个摄像机区域填充数据
			(&areaMap).CountAreaObjs(arg)
			arg.AreaMapList = append(arg.AreaMapList, &areaMap)
			areaMap.CountAreaObjs(arg)
			log.Println("-------areaMap的内容",areaMap)
			arg.AreaMapList = append(arg.AreaMapList, areaMap)
			log.Println("-------AreaMapList的内容",arg.AreaMapList)
			}
			}

			// 对单帧图像的判断是舍弃（或者说对于某些需求可以放ES数据库一份）还是返回
			func Judge(args *SdkDatas) {
			func Judge(args SdkDatas) {
			if len(args.Sdkdata) > 0 {
			// 拿到本摄像机的区域
			cameraPolygons := GetPolygons(args.CameraId)
			// 把所有的sdk提取的数据都按所属摄像机的区域归置
			for _, arg := range args.Sdkdata {
			SdkDataFormat(args.CameraId, &arg, cameraPolygons)
			SdkDataFormat(args.CameraId, arg, cameraPolygons)
			}
			// 跑本摄像机的所有规则组一组一组跑
			taskRuleList := GetRuleGroup(args.CameraId) // 本摄像机下所有任务组
			@@ -224,17 +226,16 @@
			}
			}
			}
			func RunRule(args SdkDatas, groupRule protomsg.GroupRule, taskId string) bool {
			func RunRule(args SdkDatas, groupRule *protomsg.GroupRule, taskId string) bool {
			resultSplice := []*LittleRuleResult{}
			// 先过完条件规则
			log.Println("---------------------看下这个组规则：",groupRule.Rules)
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			if sdkData.SdkId == "812b674b-2375-4589-919a-5c1c3278a975" {
			log.Println("入侵的算法数据",sdkData)
			log.Println("入侵的算法数据", sdkData)
			}
			for _, areaMap := range sdkData.AreaMapList {
			ruleResult := filterRule(groupRule.Rules[j], areaMap)
			ruleResult := filterRule(groupRule.Rules[j], &areaMap)
			if ruleResult.Result != "" {
			log.Println("条件规则结果：", ruleResult.Result)
			resultSplice = append(resultSplice, &ruleResult)
			@@ -247,7 +248,7 @@
			for _, sdkData := range args.Sdkdata {
			for _, areaMap := range sdkData.AreaMapList {
			for j := 0; j < len(groupRule.Rules); j++ {
			putFaceToResult(groupRule.Rules[j], areaMap, faces)
			putFaceToResult(groupRule.Rules[j], &areaMap, faces)
			}
			}
			}
			@@ -256,7 +257,7 @@
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			for _, areaMap := range sdkData.AreaMapList {
			ruleResult := transferParameters(groupRule.Rules[j], areaMap)
			ruleResult := transferParameters(groupRule.Rules[j], &areaMap)
			if ruleResult.Result != "" {
			log.Println("数量规则结果：", ruleResult.Result)
			resultSplice = append(resultSplice, &ruleResult)
			@@ -268,7 +269,7 @@
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			for _, areaMap := range sdkData.AreaMapList {
			ruleResult := timeRuleResult(groupRule.Rules[j], areaMap)
			ruleResult := timeRuleResult(groupRule.Rules[j], &areaMap)
			if ruleResult.Result != "" {
			log.Println("时间规则结果：", ruleResult.Result)
			resultSplice = append(resultSplice, &ruleResult)
			@@ -280,7 +281,7 @@
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			for _, areaMap := range sdkData.AreaMapList {
			duration(groupRule.Rules[j], areaMap)
			duration(groupRule.Rules[j], &areaMap)
			}
			}
			}
			@@ -314,18 +315,19 @@
			}
			}
			if flag {
			fmt.Println("本帧数据符合规则")
			fmt.Println("----------------------------终于走完万里长征")
			// 如果成功了我应该找到规则中涉及到的sdk，记录下他们的sdkname
			sdkName := ""
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo{ // 只记录yolo符合规则的sdk
			if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk
			sdkName = sdkName + "," + sdkData.SdkName
			}
			}
			}
			// 过完规则后打个标签，告诉调用者本帧数据针对哪个任务哪组规则报警了后加：可能还不够，还需要区分触发报警的对象，后面往es数据库插数据时要用
			args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
			log.Println("-------------------结果标签",args.RuleResult["yolo"].([]Result))
			return true
			} else {
			return false
			@@ -364,11 +366,12 @@

			// }
			for _, obj := range arg.Photo {
			log.Println("------------------看看sdkData的Photo数据----------------", 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) {
			// 这步要备齐表达式里所需要的所有参数
			a.targetNum++
			arg1 := Arg{obj.Score, PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale), float64(obj.Rects.Width * obj.Rects.Height), obj.IsYolo, obj.Rects, obj.SdkName, obj.ThftRes, obj.Liker}
			//log.Println("放进去的arg：-------", arg1)
			log.Println("放进去的arg：-------", arg1)
			a.args = append(a.args, arg1)
			a.filterData = append(a.filterData, arg1)
			}
			@@ -415,7 +418,7 @@
			}

			// 联动任务的处理
			func linkTask(args SdkDatas, groupRule protomsg.GroupRule, taskId string) {
			func linkTask(args SdkDatas, groupRule *protomsg.GroupRule, taskId string) {
			// new一个定时器，如果以此groupId为标志的定时器不存在的话
			var flag bool = true
			var timeEle = TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId}
			@@ -476,7 +479,7 @@
			sdkName := ""
			for j := 0; j < len(groupRule.Rules); j++ {
			for _, sdkData := range args.Sdkdata {
			if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo{ // 只记录yolo符合规则的sdk
			if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk
			sdkName = sdkName + "," + sdkData.SdkName
			}
			}
			@@ -586,7 +589,7 @@
			}
			} else {
			// 处理的都是yolo数据
			log.Println("规则中的sdkId为：",rule.SdkId,"--区域数据集合中的sdkId为：",am.sdkId,"--规则中的区域Id为：",rule.PolygonId,"--区域数据集合中的区域Id为：",am.areaId,)
			log.Println("规则中的sdkId为：", rule.SdkId, "--区域数据集合中的sdkId为：", am.sdkId, "--规则中的区域Id为：", rule.PolygonId, "--区域数据集合中的区域Id为：", am.areaId, )
			if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
			if rule.SdkArgAlias == "score" \|\| rule.SdkArgAlias == "proportion" \|\| rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
			var args []Arg