OutOfLane.go 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. package main
  2. //触发器名称:冲出车道
  3. import (
  4. "cicv-data-closedloop/pjisuv_msgs"
  5. "fmt"
  6. "math"
  7. "sync"
  8. )
  9. func Topic() string {
  10. return "/map_polygon"
  11. }
  12. func Label() string {
  13. return "OutOfLane"
  14. }
  15. func Rule(shareVars *sync.Map, data *pjisuv_msgs.PolygonStamped) string {
  16. defer func() {
  17. if r := recover(); r != nil {
  18. fmt.Println("Recovered from panic:", r)
  19. }
  20. }()
  21. positionXOfCicvLocation, ok1 := shareVars.Load("PositionXOfCicvLocation")
  22. positionYOfCicvLocation, ok2 := shareVars.Load("PositionYOfCicvLocation")
  23. yawOfCicvLocation, ok3 := shareVars.Load("YawOfCicvLocation")
  24. automodeOfPjVehicleFdbPub, ok4 := shareVars.Load("AutomodeOfPjVehicleFdbPub")
  25. OutsideWorkshopFlag, _ := shareVars.Load("OutsideWorkshopFlag")
  26. OutsideWorkshopFlag = OutsideWorkshopFlag.(bool)
  27. if ok1 && ok2 && ok3 && ok4 {
  28. Points := data.Polygon.Points
  29. D1, D2, D3 := 1.0, 0.4, 0.4 //这是实车缩小后的尺寸
  30. corners := getVehicleCorners(positionXOfCicvLocation.(float64), positionYOfCicvLocation.(float64), D1, D2, D3, yawOfCicvLocation.(float64))
  31. for i := 0; i < len(Points)-1; i++ {
  32. A := Points[i]
  33. B := Points[i+1]
  34. if polygonLineIntersect(corners, A, B) && automodeOfPjVehicleFdbPub.(int16) == 1 && OutsideWorkshopFlag == true {
  35. return Label()
  36. }
  37. }
  38. }
  39. return ""
  40. }
  41. type Point struct {
  42. X, Y float64
  43. }
  44. func getVehicleCorners(x0, y0, D1, D2, D3, h0 float64) []pjisuv_msgs.Point64 {
  45. h0Rad := h0 * math.Pi / 180 // 转换角度到弧度
  46. cosH0, sinH0 := math.Cos(h0Rad), math.Sin(h0Rad)
  47. // 车辆四个角点相对于后轴中心点的坐标(局部坐标系)
  48. cornersLocal := []Point{
  49. {D1, D3}, // 前左角
  50. {D1, -D3}, // 前右角
  51. {-D2, D3}, // 后左角
  52. {-D2, -D3}, // 后右角
  53. }
  54. // 旋转矩阵并计算全局坐标
  55. rotationMatrix := [2][2]float64{{cosH0, -sinH0}, {sinH0, cosH0}}
  56. cornersGlobal := make([]pjisuv_msgs.Point64, len(cornersLocal))
  57. for i, corner := range cornersLocal {
  58. cornersGlobal[i].X = corner.X*rotationMatrix[0][0] + corner.Y*rotationMatrix[1][0] + x0
  59. cornersGlobal[i].Y = corner.X*rotationMatrix[0][1] + corner.Y*rotationMatrix[1][1] + y0
  60. }
  61. return cornersGlobal
  62. }
  63. func ccw(A, B, C pjisuv_msgs.Point64) bool {
  64. return (C.Y-A.Y)*(B.X-A.X) > (C.X-A.X)*(B.Y-A.Y)
  65. }
  66. func lineIntersect(A, B, C, D pjisuv_msgs.Point64) bool {
  67. return ccw(A, C, D) != ccw(B, C, D) && ccw(A, B, C) != ccw(A, B, D)
  68. }
  69. func polygonLineIntersect(polygon []pjisuv_msgs.Point64, A, B pjisuv_msgs.Point64) bool {
  70. for i := 0; i < len(polygon)-1; i++ {
  71. C := polygon[i]
  72. D := polygon[(i+1)%len(polygon)]
  73. if lineIntersect(A, B, C, D) {
  74. return true
  75. }
  76. }
  77. return false
  78. }