| | |
| | | |
| | | import ( |
| | | "encoding/json" |
| | | "fmt" |
| | | "ruleprocess/cache" |
| | | "ruleprocess/logger" |
| | | "sort" |
| | |
| | | func GetPolygons(cameraId string) []protomsg.CameraPolygon { |
| | | //var api dbapi.CameraApi |
| | | //data := api.FindAllPolygons() |
| | | ////fmt.Println("查到的所有区域:", data) |
| | | ////logger.Info("查到的所有区域:", data) |
| | | //// 根据id从map中拿到区域 |
| | | var cameraPolygons []protomsg.CameraPolygon |
| | | //for _, item := range data { |
| | |
| | | func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) bool { |
| | | defer func() { |
| | | if err := recover(); err != nil { |
| | | fmt.Println("比对规则有误", err.(string)) |
| | | logger.Error("比对规则有误", err.(string)) |
| | | } |
| | | }() |
| | | logger.Info("+++++++++++规则开始运行+++++++++++++++++当前规则--:", (*groupRule).GroupText) |
| | |
| | | } |
| | | } |
| | | if flag { |
| | | fmt.Println("___________________________________________________________________________终于走完万里长征") |
| | | logger.Info("___________________________________________________________________________终于走完万里长征") |
| | | // 如果成功了我应该找到规则中涉及到的sdk,记录下他们的sdkname |
| | | sdkName := "" |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | |
| | | } else { |
| | | err := json.Unmarshal([]byte(areaPoints), &pts) |
| | | if err != nil { |
| | | fmt.Println("json.Unmarshal错误", err) |
| | | logger.Error("json.Unmarshal错误", err) |
| | | panic("序列化坐标异常,程序退出") |
| | | } |
| | | } |
| | |
| | | // 第一步查出跟这个摄像机相关的group_id(大规则) |
| | | //var api dbapi.CameraTaskArgsApi |
| | | // |
| | | ////fmt.Println("所有规则:", all) |
| | | ////logger.Info("所有规则:", all) |
| | | //var taskArgs []*protomsg.TaskGroupArgs |
| | | //for _, camArg := range all { |
| | | // if camArg.CameraId == cameraId { |
| | |
| | | timeEle := TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId} // 扔进去一个定时器元素 |
| | | //TimeEleList = make(map[string]timeElement) |
| | | TimeEleList[groupRule.GroupId] = &timeEle // 定时器元素以规则组id为键 |
| | | fmt.Println("---------------------------------------------联动任务创建了计数器并且计数器集合为:", TimeEleList) |
| | | logger.Info("---------------------------------------------联动任务创建了计数器并且计数器集合为:", TimeEleList) |
| | | // 得出这组完整规则里涉及到几个摄像机,决定着数组里有几个结构体,去重添加方式 |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | var flag1 bool = true |
| | |
| | | // 往数组里赋值 |
| | | isOk := RunRule(args, groupRule, taskId) |
| | | if isOk { |
| | | fmt.Println("这帧图像在任务下的一整条规则下(联动任务下就是跟本摄像机像相关的小规则)的判断结果为true") |
| | | logger.Info("这帧图像在任务下的一整条规则下(联动任务下就是跟本摄像机像相关的小规则)的判断结果为true") |
| | | // 根据cameraId去更新或者插入结果,然后判断是否数组是否可以得出报警的结论 |
| | | // 往联动任务的结果数组里放值或更新 |
| | | for _, va := range timeEle.RuleResults { |
| | |
| | | } |
| | | } |
| | | } else { |
| | | fmt.Println("数组不圆满不打标签") |
| | | logger.Warn("数组不圆满不打标签") |
| | | } |
| | | } else { // 没有报警, |
| | | fmt.Println("这帧图像在任务下的一整条规则下(联动任务下就是跟本摄像机像相关的小规则)的判断结果为false") |
| | | logger.Info("这帧图像在任务下的一整条规则下(联动任务下就是跟本摄像机像相关的小规则)的判断结果为false") |
| | | // 所以也要去结果数组里放值或更新 |
| | | for _, va := range timeEle.RuleResults { |
| | | if args.CameraId != "" && va.CameraId == args.CameraId { // arg.CameraId 随便找一个数据 |
| | |
| | | for k, _ := range TimeEleList { |
| | | if k == am.taskId+" "+am.sdkId+" "+am.areaId { |
| | | flag = false // 有就置为false |
| | | fmt.Println("有这个定时器,不再创建了:") |
| | | logger.Info("有这个定时器,不再创建了:") |
| | | } |
| | | } |
| | | |
| | |
| | | timeEle := TimeElement{N: timeLength, InitN: timeLength} // 扔进去一个定时器元素 |
| | | //TimeEleList = make(map[string]timeElement) |
| | | TimeEleList[am.taskId+" "+am.sdkId+" "+am.areaId] = &timeEle // 定时器元素以摄像机id拼接区域id为键 |
| | | fmt.Println("创建了计数器并且计数器集合为:", TimeEleList) |
| | | logger.Info("创建了计数器并且计数器集合为:", TimeEleList) |
| | | } |
| | | } |
| | | } |
| | |
| | | func transferParameters(rule *protomsg.Rule, am *AreaMap) LittleRuleResult { |
| | | if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 首先规则所对应的区域id要跟区域数据的id对的上 |
| | | if rule.SdkArgAlias == "targetNum" { // 如果参数是要区域内目标数量 即yolo 人脸不会有数量 |
| | | //fmt.Println("得出结果阶段", "比较的规则是:", rule) |
| | | //logger.Info("得出结果阶段", "比较的规则是:", rule) |
| | | if rule.Operator == "" { |
| | | return LittleRuleResult{am.sdkName, strconv.Itoa(am.targetNum) + "" + rule.RuleWithPre, rule.Sort} // 如果后面不跟操作符就直接返回数量 比如要跟下一个区域比较数量的就直接返回本区域的数量 |
| | | } |
| | |
| | | if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规则 |
| | | logger.Info("----------当前时间规则:---------", rule) |
| | | // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段(比如当前时间是周三,应根据区域id查出其周三的几个布防时间段,数组) |
| | | //fmt.Println("时间规则的测试") |
| | | //logger.Info("时间规则的测试") |
| | | now := time.Now() |
| | | index := getIndexOfWeek(now.Weekday().String()) |
| | | timeList := GetTimeById(rule.SdkArgValue, index) |
| | | logger.Info("当天的时间段集合:----------", timeList) |
| | | //fmt.Println("从数据库中查出的时间规则:", timeList) |
| | | //logger.Info("从数据库中查出的时间规则:", timeList) |
| | | // 判断图片数据的时间是否符合当前规则 在一个即为true,全不在为false |
| | | |
| | | if rule.Operator == "satisfy" || rule.Operator == "==" { // 满足所选的时间规则 |
| | |
| | | formula1 := "'" + timeSlot.End + "'" + ">" + "'" + am.time + "'" |
| | | expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式 |
| | | result1, _ := expression1.Evaluate(nil) // 得到数学公式的结果 |
| | | //fmt.Println("看看这两尊大神", result, result1) |
| | | //logger.Info("看看这两尊大神", result, result1) |
| | | if result.(bool) && result1.(bool) { |
| | | flag = "true" |
| | | break |