multiTargetAhead.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. package main
  2. import (
  3. "cicv-data-closedloop/pjisuv_msgs"
  4. "fmt"
  5. "math"
  6. "sync"
  7. )
  8. func Topic() string {
  9. return "/tpperception"
  10. }
  11. // Label todo 禁止存在下划线_
  12. // MultiPedestrianAhead
  13. // MultiCarAhead
  14. // MultiTruckAhead
  15. // MultiBicycleAhead
  16. // MultiTricycleAhead
  17. // MultiTrafficConeAhead
  18. func Label() string {
  19. return "MultiPedestrianAhead"
  20. }
  21. type Record struct {
  22. StableFrame int
  23. LastExistingTime float64
  24. }
  25. var record *Record = nil
  26. // typeCheck 目标物类型检测
  27. // × 金龙车:CAR_TYPE=0, TRUCK_TYPE=1, PEDESTRIAN_TYPE=2, CYCLIST_TYPE=3, UNKNOWN_TYPE=4, UNKNOWN_MOVABLE_TYPE=5, UNKNOWN_UNMOVABLE_TYPE=6
  28. // √ 多功能车:UNKNOWN TYPE=O, PEDESTRIAN TYPE=1, CAR TYPE=2, TRUCK TYPE=3, Bicycle TYPE=4, Tricycle TYPE=5, Traffic Cone TYPE=6
  29. func typeCheck(obj *pjisuv_msgs.PerceptionObject) bool {
  30. const targetType uint8 = 1
  31. return targetType == obj.Type
  32. }
  33. // posCheck 判断目标物位置关系
  34. func posCheck(obj *pjisuv_msgs.PerceptionObject) bool {
  35. const laneWidth = 3.5 // m
  36. return obj.X > 0 && math.Abs(float64(obj.Y)) < laneWidth*1.5
  37. }
  38. // Rule 检测前方指定目标物数量
  39. func Rule(shareVars *sync.Map, msg *pjisuv_msgs.PerceptionObjects) string {
  40. defer func() {
  41. if r := recover(); r != nil {
  42. fmt.Println("Recovered from panic:", r)
  43. }
  44. }()
  45. const Cgcs2000X = 456256.260152
  46. const Cgcs2000Y = 4397809.886833
  47. const MinStableFrameCount = 5 // frame
  48. const MaxTimeBetweenFrame = 0.5 // second
  49. const minTargetNum = 4
  50. OutsideWorkshopFlag, _ := shareVars.Load("OutsideWorkshopFlag")
  51. OutsideWorkshopFlag = OutsideWorkshopFlag.(bool)
  52. if len(msg.Objs) >= minTargetNum {
  53. currTargetNum := 0
  54. for _, obj := range msg.Objs {
  55. // 判断目标物是否满足筛选条件
  56. if typeCheck(&obj) && posCheck(&obj) {
  57. currTargetNum++
  58. }
  59. }
  60. if currTargetNum >= minTargetNum {
  61. if nil == record {
  62. record = &Record{
  63. StableFrame: 1,
  64. LastExistingTime: msg.Header.TimeStamp,
  65. }
  66. } else {
  67. newRecord := Record{
  68. StableFrame: 1,
  69. LastExistingTime: msg.Header.TimeStamp,
  70. }
  71. if newRecord.LastExistingTime-record.LastExistingTime <= MaxTimeBetweenFrame {
  72. newRecord.StableFrame = record.StableFrame + 1
  73. if newRecord.StableFrame == MinStableFrameCount && OutsideWorkshopFlag == true {
  74. record = nil
  75. return Label()
  76. } else {
  77. }
  78. } else {
  79. }
  80. record = &newRecord
  81. }
  82. }
  83. }
  84. return ""
  85. }