当前位置:   article > 正文

四、Hybrid_astar.py文件中Hybrid A * 算法程序的详细介绍

四、Hybrid_astar.py文件中Hybrid A * 算法程序的详细介绍

hybrid_astar.py文件

  1. # 导入库和模块
  2. import os
  3. import sys
  4. import math # heapq和heapdict用于构建优先级队列;numpy和matplotlib用于数值计算和绘图,scipy.spatial.kdtree用于创建K-D树来进行碰撞检测
  5. import heapq
  6. from heapdict import heapdict
  7. import time
  8. import numpy as np
  9. import matplotlib.pyplot as plt
  10. import scipy.spatial.kdtree as kd
  11. sys.path.append(os.path.dirname(os.path.abspath(__file__)) +
  12. "/../../HybridAstarPlanner/")
  13. import astar as astar
  14. import draw as draw
  15. import reeds_shepp as rs
  16. # 定义参数配置类 C;用于存储路径规划中使用的各种参数
  17. class C: # Parameter config
  18. PI = math.pi
  19. XY_RESO = 2.0 # 坐标分辨率,表示每个坐标点在X和Y方向上的间隔
  20. YAW_RESO = np.deg2rad(15.0) # 角度分辨率,表示每个角度的间隔且将角度转换为弧度
  21. MOVE_STEP = 0.4 # 插值分辨率,表示在路径中每个节点之间的距离
  22. N_STEER = 20.0 # 计算转向角的离散化值,即将最大转向角范围分成多少个不同的离散值
  23. COLLISION_CHECK_STEP = 5 # 碰撞检测时,跳过的节点数,即每隔多少个节点进行一次碰撞检测,用于减少碰撞检测的计算量
  24. EXTEND_BOUND = 1 # 碰撞检测范围的扩展值,用于在实际碰撞检测时,将检测范围扩展一定距离。
  25. GEAR_COST = 100.0 # 切换方向的惩罚成本,当车辆需要切换行驶方向时,会增加这个成本
  26. BACKWARD_COST = 5.0 # 后退行驶的惩罚成本
  27. STEER_CHANGE_COST = 5.0 # 转向角变化的惩罚成本
  28. STEER_ANGLE_COST = 1.0 # 转向角度的惩罚成本
  29. H_COST = 15.0 # 启发式成本的惩罚成本,用于引导启发式搜索算法的探索方向
  30. RF = 4.5 # 车辆从后部到车辆前部的距离
  31. RB = 1.0 # 车辆从后部到车辆后部的距离
  32. W = 3.0 # 车辆的宽度
  33. WD = 0.7 * W # 左右车轮之间的距离,用于计算车辆的转向角
  34. WB = 3.5 # 车辆的轴距,表示前后轮之间的距离
  35. TR = 0.5 # 车轮的半径
  36. TW = 1 # 车轮的宽度
  37. MAX_STEER = 0.6 # 最大转向角,表示车辆的最大转向角度
  38. # 定义节点类 Node表示路径规划中的节点
  39. class Node:
  40. def __init__(self, xind, yind, yawind, direction, x, y,
  41. yaw, directions, steer, cost, pind):
  42. self.xind = xind # xind:节点所在的 X 坐标索引
  43. self.yind = yind # yind:节点所在的 Y 坐标索引
  44. self.yawind = yawind # 节点的航向角索引
  45. self.direction = direction # 车辆的行驶方向,值为 1 表示正向,-1 表示反向;
  46. self.x = x
  47. self.y = y # 车辆路径中的坐标x,y
  48. self.yaw = yaw # 车辆路径中的航向角列表
  49. self.directions = directions # 车辆在每个路径节点上的行驶方向列表
  50. self.steer = steer # 转向角
  51. self.cost = cost # 从起始节点到当前节点的路径代价
  52. self.pind = pind # 父节点在路径规划中的索引,用于追踪路径的连接
  53. # 存储路径规划所需的各种参数和计算结果
  54. class Para:
  55. def __init__(self, minx, miny, minyaw, maxx, maxy, maxyaw,
  56. xw, yw, yaww, xyreso, yawreso, ox, oy, kdtree):
  57. self.minx = minx # 地图中 X 坐标的最小值
  58. self.miny = miny # 地图中 Y 坐标的最小值
  59. self.minyaw = minyaw # 地图中航向角的最小值
  60. self.maxx = maxx # 地图中 X 坐标的最大值
  61. self.maxy = maxy # 地图中 Y 坐标的最大值
  62. self.maxyaw = maxyaw # 地图中航向角的最大值
  63. self.xw = xw # X 方向上的网格数
  64. self.yw = yw # Y 方向上的网格数
  65. self.yaww = yaww # 航向角方向上的网格数
  66. self.xyreso = xyreso # XY 坐标分辨率
  67. self.yawreso = yawreso # 航向角分辨率
  68. self.ox = ox # 障碍物的 X 坐标列表
  69. self.oy = oy # 障碍物的 Y 坐标列表
  70. self.kdtree = kdtree # 使用 k-d 树表示的障碍物数据结构,用于快速查找最近的障碍物点
  71. # Path 的类,用于存储路径规划的结果;描述路径节点属性
  72. class Path:
  73. def __init__(self, x, y, yaw, direction, cost):
  74. self.x = x
  75. self.y = y
  76. self.yaw = yaw
  77. self.direction = direction
  78. self.cost = cost
  79. # 优先级队列
  80. class QueuePrior:
  81. def __init__(self): # 初始化优先级队列的构造函数
  82. self.queue = heapdict() # 创建一个 heapdict 数据结构作为队列的实现,该数据结构是基于堆和字典的混合数据结构,允许在 O(1) 时间内执行插入和删除操作
  83. def empty(self): # 判断优先级队列是否为空
  84. return len(self.queue) == 0 # if Q is empty;如果队列中没有元素,返回 True;否则,返回 False
  85. def put(self, item, priority): # 将元素 item 插入队列,
  86. self.queue[item] = priority # 指定它的优先级为 priority
  87. def get(self):
  88. return self.queue.popitem()[0] # 具有最小优先级的弹出元素
  89. # hybrid_astar_planning函数实现了混合A*算法
  90. def hybrid_astar_planning(sx, sy, syaw, gx, gy, gyaw, ox, oy, xyreso, yawreso):
  91. # 将连续坐标转换为离散坐标
  92. sxr, syr = round(sx / xyreso), round(sy / xyreso)
  93. gxr, gyr = round(gx / xyreso), round(gy / xyreso)
  94. syawr = round(rs.pi_2_pi(syaw) / yawreso)
  95. gyawr = round(rs.pi_2_pi(gyaw) / yawreso)
  96. # 创建起始节点和目标节点
  97. nstart = Node(sxr, syr, syawr, 1, [sx], [sy], [syaw], [1], 0.0, 0.0, -1)
  98. ngoal = Node(gxr, gyr, gyawr, 1, [gx], [gy], [gyaw], [1], 0.0, 0.0, -1)
  99. # 创建KD树以加速障碍物查询
  100. kdtree = kd.KDTree([[x, y] for x, y in zip(ox, oy)])
  101. P = calc_parameters(ox, oy, xyreso, yawreso, kdtree) # 计算规划参数P
  102. hmap = astar.calc_holonomic_heuristic_with_obstacle(ngoal, P.ox, P.oy, P.xyreso, 1.0) # 使用A*算法计算启发式地图
  103. steer_set, direc_set = calc_motion_set() # 计算转向和方向的集合
  104. open_set, closed_set = {calc_index(nstart, P): nstart}, {} # 初始化open_set、closed_set,将起始节点添加到open_set中
  105. qp = QueuePrior() # 初始化优先队列qp
  106. qp.put(calc_index(nstart, P), calc_hybrid_cost(nstart, hmap, P))
  107. # 开始主循环
  108. while True:
  109. if not open_set:
  110. return None # 无解情况
  111. ind = qp.get()
  112. n_curr = open_set[ind]
  113. closed_set[ind] = n_curr
  114. open_set.pop(ind)
  115. update, fpath = update_node_with_analystic_expantion(n_curr, ngoal, P) # 尝试使用解析扩展更新节点
  116. if update:
  117. fnode = fpath
  118. break
  119. # 遍历转向和方向的集合
  120. for i in range(len(steer_set)):
  121. # 计算下一个节点
  122. node = calc_next_node(n_curr, ind, steer_set[i], direc_set[i], P)
  123. if not node:
  124. continue
  125. node_ind = calc_index(node, P)
  126. if node_ind in closed_set:
  127. continue
  128. if node_ind not in open_set:
  129. open_set[node_ind] = node
  130. qp.put(node_ind, calc_hybrid_cost(node, hmap, P))
  131. else:
  132. if open_set[node_ind].cost > node.cost:
  133. open_set[node_ind] = node
  134. qp.put(node_ind, calc_hybrid_cost(node, hmap, P))
  135. return extract_path(closed_set, fnode, nstart) # 提取并返回路径
  136. # extract_path函数用于从封闭集(closed set)中提取并构造最终的路径
  137. def extract_path(closed, ngoal, nstart):
  138. rx, ry, ryaw, direc = [], [], [], [] # 创建空列表用于存储路径信息
  139. cost = 0.0 # 初始化路径成本
  140. node = ngoal # 从目标节点开始回溯路径
  141. while True:
  142. rx += node.x[::-1] # 将当前节点的x坐标列表反向加入rx列表
  143. ry += node.y[::-1] # 将当前节点的y坐标列表反向加入ry列表
  144. ryaw += node.yaw[::-1] # 将当前节点的航向角列表反向加入ryaw列表
  145. direc += node.directions[::-1] # 将当前节点的方向列表反向加入direc列表
  146. cost += node.cost # 将当前节点的成本累加到总成本中
  147. if is_same_grid(node, nstart): # 如果当前节点已经回溯到起始节点
  148. break # 停止回溯
  149. node = closed[node.pind] # 如果当前节点不是起始节点,获取上一节点,继续回溯
  150. rx = rx[::-1] # 将rx列表反向,得到正确的路径顺序
  151. ry = ry[::-1] # 将ry列表反向,得到正确的路径顺序
  152. ryaw = ryaw[::-1] # 将ryaw列表反向,得到正确的路径顺序
  153. direc = direc[::-1] # 将direc列表反向,得到正确的路径顺序
  154. direc[0] = direc[1] # 将起始节点的方向设置为与第二个节点的方向相同
  155. path = Path(rx, ry, ryaw, direc, cost) # 创建Path对象,表示完整路径
  156. return path # 返回构造的路径对象
  157. # calc_next_node函数实现了计算下一个路径规划节点的功能
  158. def calc_next_node(n_curr, c_id, u, d, P): # 当前节点、索引、转向角、进行方向、参数对象
  159. step = C.XY_RESO * 2 # 计算一个步长
  160. nlist = math.ceil(step / C.MOVE_STEP) # 计算下一个节点数量
  161. xlist = [n_curr.x[-1] + d * C.MOVE_STEP * math.cos(n_curr.yaw[-1])] # 初始化列表,储存下一个节点的x坐标、y坐标、航向角;在 d 方向上行驶 C.MOVE_STEP 距离
  162. ylist = [n_curr.y[-1] + d * C.MOVE_STEP * math.sin(n_curr.yaw[-1])]
  163. yawlist = [rs.pi_2_pi(n_curr.yaw[-1] + d * C.MOVE_STEP / C.WB * math.tan(u))]
  164. # 不断累加进行插值,计算出剩余的节点的 x、y 坐标和航向角
  165. for i in range(nlist - 1):
  166. xlist.append(xlist[i] + d * C.MOVE_STEP * math.cos(yawlist[i]))
  167. ylist.append(ylist[i] + d * C.MOVE_STEP * math.sin(yawlist[i]))
  168. yawlist.append(rs.pi_2_pi(yawlist[i] + d * C.MOVE_STEP / C.WB * math.tan(u)))
  169. # 最后一个节点的坐标和航向角,将其转换为在地图上的索引坐标
  170. xind = round(xlist[-1] / P.xyreso)
  171. yind = round(ylist[-1] / P.xyreso)
  172. yawind = round(yawlist[-1] / P.yawreso)
  173. # 调用 is_index_ok 函数检查这些索引坐标是否合法,即是否在地图范围内且不与障碍物冲突。如果不合法,说明生成的节点在地图外或与障碍物冲突,此时会返回 None。
  174. if not is_index_ok(xind, yind, xlist, ylist, yawlist, P):
  175. return None
  176. cost = 0.0
  177. if d > 0:
  178. direction = 1
  179. cost += abs(step) # 前进成本
  180. else:
  181. direction = -1
  182. cost += abs(step) * C.BACKWARD_COST # 后退成本
  183. if direction != n_curr.direction: # 生成的节点的行进方向和父节点的行进方向不一致
  184. cost += C.GEAR_COST # 切换车辆行进方向加上切换成本
  185. cost += C.STEER_ANGLE_COST * abs(u) # 累加转向角度的惩罚成本
  186. cost += C.STEER_CHANGE_COST * abs(n_curr.steer - u) # 转向角度的变化成本
  187. cost = n_curr.cost + cost
  188. directions = [direction for _ in range(len(xlist))] # 将生成的行进方向信息复制多次,以匹配生成的节点数量,存储在 directions 列表中
  189. node = Node(xind, yind, yawind, direction, xlist, ylist, # 创建一个新的节点对象 Node
  190. yawlist, directions, u, cost, c_id)
  191. return node
  192. # is_index_ok函数用于判断生成的路径规划节点的索引坐标是否在地图范围内,并且检查这些节点是否与障碍物发生碰撞
  193. def is_index_ok(xind, yind, xlist, ylist, yawlist, P):
  194. if xind <= P.minx or \
  195. xind >= P.maxx or \
  196. yind <= P.miny or \
  197. yind >= P.maxy:
  198. return False # 如果任何一个条件成立,说明节点位于地图边界外,此时返回 False,表示索引不合法
  199. ind = range(0, len(xlist), C.COLLISION_CHECK_STEP) # 计算一个索引范围 ind;步长是 C.COLLISION_CHECK_STEP
  200. nodex = [xlist[k] for k in ind] # 选择在 xlist、ylist 和 yawlist 中的哪些值需要进行碰撞检查
  201. nodey = [ylist[k] for k in ind] # 将对应索引处的值从 xlist、ylist 和 yawlist 中提取出来,分别存储在 nodex、nodey 和 nodeyaw 列表中
  202. nodeyaw = [yawlist[k] for k in ind]
  203. if is_collision(nodex, nodey, nodeyaw, P):
  204. return False # 调用函数 is_collision,将 nodex、nodey 和 nodeyaw 传递给它,以及参数对象 P,用于检查这些节点是否与障碍物发生碰撞。如果发生碰撞,则返回 False,否则返回 True
  205. return True
  206. def update_node_with_analystic_expantion(n_curr, ngoal, P): # 当前节点 、目标节点、 以及参数对象
  207. path = analystic_expantion(n_curr, ngoal, P) # 分析路径规划,当前节点到目标节点是否有更优的路径
  208. if not path:
  209. return False, None # 如果没有找到更优的路径,则返回 False 表示无法更新节点。
  210. # 如果 analystic_expantion 找到了从当前节点到目标节点的可行Reeds shee曲线路径,它会返回一个路径对象 path,然后代码继续执行
  211. fx = path.x[1:-1] # 提取出除去起始和终点的路径信息、分别存储在 fx、fy、fyaw 和 fd 列表中
  212. fy = path.y[1:-1]
  213. fyaw = path.yaw[1:-1]
  214. fd = path.directions[1:-1]
  215. fcost = n_curr.cost + calc_rs_path_cost(path) # 新节点的成本由当前节点的成本加上从当前节点到目标节点的路径 path 的成本计算得到
  216. fpind = calc_index(n_curr, P) # 计算新节点的索引
  217. fsteer = 0.0
  218. fpath = Node(n_curr.xind, n_curr.yind, n_curr.yawind, n_curr.direction,
  219. fx, fy, fyaw, fd, fsteer, fcost, fpind) # 创建一个新的节点对象 fpath
  220. return True, fpath # 成功更新节点
  221. # analystic_expantion函数会尝试调用Reeds_sheep算法查找48种可行曲线
  222. def analystic_expantion(node, ngoal, P):
  223. sx, sy, syaw = node.x[-1], node.y[-1], node.yaw[-1]
  224. gx, gy, gyaw = ngoal.x[-1], ngoal.y[-1], ngoal.yaw[-1]
  225. maxc = math.tan(C.MAX_STEER) / C.WB # 车辆最大转向角对应的转弯半径
  226. paths = rs.calc_all_paths(sx, sy, syaw, gx, gy, gyaw, maxc, step_size=C.MOVE_STEP) # 根据 Reeds-Shepp 路径类型计算从起始状态到目标状态 的所有可能路径
  227. if not paths:
  228. return None # 无法进行路径扩展
  229. pq = QueuePrior() # 如果找到了路径,则创建一个优先级队列 pq
  230. for path in paths:
  231. pq.put(path, calc_rs_path_cost(path))
  232. while not pq.empty():
  233. path = pq.get() # 优先级队列中依次弹出路径
  234. ind = range(0, len(path.x), C.COLLISION_CHECK_STEP) # 使用间隔为 C.COLLISION_CHECK_STEP 的索引,从路径中提取出部分点
  235. pathx = [path.x[k] for k in ind]
  236. pathy = [path.y[k] for k in ind]
  237. pathyaw = [path.yaw[k] for k in ind]
  238. if not is_collision(pathx, pathy, pathyaw, P): # 调用 is_collision 函数检查这些点是否与障碍物发生碰撞
  239. return path
  240. return None
  241. # is_collision函数用于检测车辆是否与障碍物发生碰撞。
  242. def is_collision(x, y, yaw, P):
  243. for ix, iy, iyaw in zip(x, y, yaw):
  244. d = 1 # 扩展碰撞检测范围
  245. dl = (C.RF - C.RB) / 2.0 # 车辆前后轮之间的距离差的一半
  246. r = (C.RF + C.RB) / 2.0 + d # 车辆的半径
  247. cx = ix + dl * math.cos(iyaw) # 车辆的前轴中心点坐标 (cx, cy)
  248. cy = iy + dl * math.sin(iyaw)
  249. ids = P.kdtree.query_ball_point([cx, cy], r) # kd数近邻点查询,以车辆前部中心为中心,以 r 为半径的范围内的所有障碍物点的索引
  250. if not ids:
  251. continue
  252. for i in ids:
  253. xo = P.ox[i] - cx # 车辆前部中心点的相对坐标 (xo, yo)
  254. yo = P.oy[i] - cy
  255. dx = xo * math.cos(iyaw) + yo * math.sin(iyaw) # 相对坐标通过旋转变换(根据车辆航向角)映射到车辆坐标系
  256. dy = -xo * math.sin(iyaw) + yo * math.cos(iyaw)
  257. if abs(dx) < r and abs(dy) < C.W / 2 + d:
  258. return True # 发生碰撞
  259. return False # 未发生碰撞
  260. # calc_rs_path_cost函数用于计算路径的成本,即评估给定路径的优劣程度
  261. def calc_rs_path_cost(rspath):
  262. cost = 0.0 # 初始化路径成本为0
  263. for lr in rspath.lengths:
  264. if lr >= 0:
  265. cost += 1
  266. else:
  267. cost += abs(lr) * C.BACKWARD_COST # 后退行驶
  268. for i in range(len(rspath.lengths) - 1):
  269. if rspath.lengths[i] * rspath.lengths[i + 1] < 0.0: # 检查相邻的路径段是否具有不同的行驶方向
  270. cost += C.GEAR_COST # 增加换挡的代价
  271. for ctype in rspath.ctypes:
  272. if ctype != "S": # 不是直线
  273. cost += C.STEER_ANGLE_COST * abs(C.MAX_STEER)
  274. nctypes = len(rspath.ctypes)
  275. ulist = [0.0 for _ in range(nctypes)] # 存储路径每个曲线段对应的转向角度
  276. for i in range(nctypes):
  277. if rspath.ctypes[i] == "R": # 右转
  278. ulist[i] = -C.MAX_STEER
  279. elif rspath.ctypes[i] == "WB": # 左转
  280. ulist[i] = C.MAX_STEER
  281. for i in range(nctypes - 1): # 遍历转向角度列表
  282. cost += C.STEER_CHANGE_COST * abs(ulist[i + 1] - ulist[i])
  283. return cost
  284. # calc_hybrid_cost函数用于计算Hybrid A * 算法中节点的成本
  285. def calc_hybrid_cost(node, hmap, P): # 节点、启发式地图、包含地图和路径规划参数的对象
  286. cost = node.cost + \
  287. C.H_COST * hmap[node.xind - P.minx][node.yind - P.miny] # 启发式地图的成本
  288. return cost
  289. # calc_motion_set函数用于计算可用的车辆操控动作集合,包括转向角度和方向
  290. def calc_motion_set():
  291. s = np.arange(C.MAX_STEER / C.N_STEER, # 生成转向角度值
  292. C.MAX_STEER, C.MAX_STEER / C.N_STEER)
  293. steer = list(s) + [0.0] + list(-s)
  294. direc = [1.0 for _ in range(len(steer))] + [-1.0 for _ in range(len(steer))]
  295. steer = steer + steer
  296. return steer, direc
  297. # is_same_grid函数用于判断两个节点是否位于相同的栅格(格子)上
  298. def is_same_grid(node1, node2):
  299. if node1.xind != node2.xind or \
  300. node1.yind != node2.yind or \
  301. node1.yawind != node2.yawind:
  302. return False # 不在相同的栅格上
  303. return True
  304. # calc_index(node, P)函数用于计算给定节点在一维数组(通常用于表示二维栅格地图或状态空间)中的索引
  305. def calc_index(node, P):
  306. ind = (node.yawind - P.minyaw) * P.xw * P.yw + \
  307. (node.yind - P.miny) * P.xw + \
  308. (node.xind - P.minx)
  309. return ind
  310. # calc_parameters函数用于计算参数并返回一个包含计算结果的参数对象
  311. def calc_parameters(ox, oy, xyreso, yawreso, kdtree):
  312. minx = round(min(ox) / xyreso)
  313. miny = round(min(oy) / xyreso)
  314. maxx = round(max(ox) / xyreso)
  315. maxy = round(max(oy) / xyreso)
  316. xw, yw = maxx - minx, maxy - miny
  317. minyaw = round(-C.PI / yawreso) - 1
  318. maxyaw = round(C.PI / yawreso)
  319. yaww = maxyaw - minyaw
  320. return Para(minx, miny, minyaw, maxx, maxy, maxyaw,
  321. xw, yw, yaww, xyreso, yawreso, ox, oy, kdtree)
  322. # 用于在一个matplotlib图中绘制一个表示汽车的简化形状,包括车身和车轮
  323. def draw_car(x, y, yaw, steer, color='black'):
  324. # 定义车辆轮廓
  325. car = np.array([[-C.RB, -C.RB, C.RF, C.RF, -C.RB],
  326. [C.W / 2, -C.W / 2, -C.W / 2, C.W / 2, C.W / 2]])
  327. # 定义车轮形状
  328. wheel = np.array([[-C.TR, -C.TR, C.TR, C.TR, -C.TR],
  329. [C.TW / 4, -C.TW / 4, -C.TW / 4, C.TW / 4, C.TW / 4]])
  330. # 复制车轮形状,用于各个轮子
  331. rlWheel = wheel.copy()
  332. rrWheel = wheel.copy()
  333. frWheel = wheel.copy()
  334. flWheel = wheel.copy()
  335. # 构建旋转矩阵
  336. Rot1 = np.array([[math.cos(yaw), -math.sin(yaw)],
  337. [math.sin(yaw), math.cos(yaw)]])
  338. Rot2 = np.array([[math.cos(steer), math.sin(steer)],
  339. [-math.sin(steer), math.cos(steer)]])
  340. # 根据转向角度旋转前轮
  341. frWheel = np.dot(Rot2, frWheel)
  342. flWheel = np.dot(Rot2, flWheel)
  343. # 调整前轮位置
  344. frWheel += np.array([[C.WB], [-C.WD / 2]])
  345. flWheel += np.array([[C.WB], [C.WD / 2]])
  346. rrWheel[1, :] -= C.WD / 2
  347. rlWheel[1, :] += C.WD / 2
  348. # 根据车辆朝向旋转前轮和车轮轮廓
  349. frWheel = np.dot(Rot1, frWheel)
  350. flWheel = np.dot(Rot1, flWheel)
  351. rrWheel = np.dot(Rot1, rrWheel)
  352. rlWheel = np.dot(Rot1, rlWheel)
  353. car = np.dot(Rot1, car)
  354. # 平移车辆和车轮位置
  355. frWheel += np.array([[x], [y]])
  356. flWheel += np.array([[x], [y]])
  357. rrWheel += np.array([[x], [y]])
  358. rlWheel += np.array([[x], [y]])
  359. car += np.array([[x], [y]])
  360. # 绘制车辆和车轮
  361. plt.plot(car[0, :], car[1, :], color)
  362. plt.plot(frWheel[0, :], frWheel[1, :], color)
  363. plt.plot(rrWheel[0, :], rrWheel[1, :], color)
  364. plt.plot(flWheel[0, :], flWheel[1, :], color)
  365. plt.plot(rlWheel[0, :], rlWheel[1, :], color)
  366. draw.Arrow(x, y, yaw, C.WB * 0.8, color)
  367. # design_obstacles函数用于生成障碍物的坐标
  368. def design_obstacles(x, y): # 地图的宽度和高度
  369. ox, oy = [], []
  370. # 添加四条边界:通过循环,将地图的四条边界添加到障碍物坐标中,以防止车辆越过地图边界
  371. for i in range(x):
  372. ox.append(i)
  373. oy.append(0)
  374. for i in range(x):
  375. ox.append(i)
  376. oy.append(y - 1)
  377. for i in range(y):
  378. ox.append(0)
  379. oy.append(i)
  380. for i in range(y):
  381. ox.append(x - 1)
  382. oy.append(i)
  383. for i in range(10, 21): # 例子
  384. ox.append(i)
  385. oy.append(15)
  386. for i in range(15):
  387. ox.append(20)
  388. oy.append(i)
  389. for i in range(15, 30):
  390. ox.append(30)
  391. oy.append(i)
  392. for i in range(16):
  393. ox.append(40)
  394. oy.append(i)
  395. return ox, oy
  396. # 用于调用Hybrid A*算法来规划车辆的路径,并在matplotlib中绘制车辆的轨迹和障碍物
  397. def main():
  398. print("start!")
  399. x, y = 51, 31 # 地图的宽度和高度
  400. sx, sy, syaw0 = 10.0, 7.0, np.deg2rad(120.0) # 起始点的x,y坐标和偏航角
  401. gx, gy, gyaw0 = 45.0, 20.0, np.deg2rad(90.0) # 目标点的x,y坐标和偏航角
  402. # 生成障碍物的坐标
  403. ox, oy = design_obstacles(x, y)
  404. t0 = time.time()
  405. # 使用Hybrid A*算法进行路径规划
  406. path = hybrid_astar_planning(sx, sy, syaw0, gx, gy, gyaw0,
  407. ox, oy, C.XY_RESO, C.YAW_RESO)
  408. t1 = time.time()
  409. print("running T: ", t1 - t0)
  410. if not path:
  411. print("Searching failed!")
  412. return
  413. # 提取路径的信息
  414. x = path.x
  415. y = path.y
  416. yaw = path.yaw
  417. direction = path.direction
  418. # 在每个时刻绘制车辆的轨迹和障碍物
  419. for k in range(len(x)):
  420. plt.cla() # 清空当前图形
  421. plt.plot(ox, oy, "sk") # 绘制障碍物和规划路径
  422. plt.plot(x, y, linewidth=1.5, color='r')
  423. # 计算当前时刻的方向和转向角
  424. if k < len(x) - 2:
  425. dy = (yaw[k + 1] - yaw[k]) / C.MOVE_STEP
  426. steer = rs.pi_2_pi(math.atan(-C.WB * dy / direction[k]))
  427. else:
  428. steer = 0.0
  429. # 绘制车辆
  430. draw_car(gx, gy, gyaw0, 0.0, 'dimgray') # 绘制目标点
  431. draw_car(x[k], y[k], yaw[k], steer) # 绘制当前车辆状态
  432. plt.title("Hybrid A*")
  433. plt.axis("equal")
  434. plt.pause(0.0001)
  435. plt.show()
  436. print("Done!")
  437. if __name__ == '__main__':
  438. main()

1、导入库和模块

    这部分代码导入了所需的库和模块,其中heapqheapdict用于构建优先级队列numpymatplotlib用于数值计算和绘图,scipy.spatial.kdtree用于创建K-D树来进行碰撞检测。

2、定义参数配置类 C

  定义了一个名为 C 的参数配置类,用于存储路径规划中使用的各种参数。这些参数会在路径规划的不同阶段用于计算代价、转向角、碰撞检测等,以实现车辆的路径规划和运动控制。

3、定义节点类 Node

       用于表示路径规划中的节点,这个 Node 类用于在路径规划过程中存储节点的信息,以便计算路径代价、生成路径、进行碰撞检测等操作。每个节点都包含了一系列与路径规划相关的属性,用于描述车辆在路径中的状态和轨迹。

4. 定义参数类 Para

       用于封装路径规划所需的各种参数和计算结果,以便在算法的不同阶段方便地获取和传递这些数据。

5. 定义路径类 Path

        Path 类用于封装路径规划的结果,以便在算法完成后方便地获取和传递路径的各个属性,以及计算路径的总代价。它用于存储车辆的路径信息,包括坐标、航向角、行驶方向和代价等。

6. 定义优先级队列类 QueuePrior

   实现了优先级队列的基本功能,它用于在路径规划的过程中管理待处理的节点,并根据节点的代价或优先级进行排序和选择。

7 .hybrid_astar_planning函数

     hybrid_astar_planning函数实现了混合A*算法,根据起始点、目标点、障碍物、规划分辨率等信息进行路径规划。主循环中,它通过不断扩展节点、更新优先队列来寻找一条最优路径,如果成功找到路径,将返回路径  。

8、extract_path函数

      extract_path 函数的作用是从目标节点往回回溯,提取路径上的节点信息,并将这些信息组合成一个 Path 对象,表示一条路径。

9、calc_next_node函数

      个函数的作用是根据当前节点、转向角、行进方向等信息,计算并返回下一个可能的路径规划节点。

10、 is_index_ok函数

       这个函数用于检查生成的路径规划节点的索引是否在地图范围内,并且对一部分节点进行了碰撞检查,以确保生成的节点不会与障碍物发生碰撞。

11、update_node_with_analystic_expantion函数

       用于分析判断是否存在从当前拓展点到目标点的可行reeds sheep曲线路径,若存在则返回TRUE及Reeds sheep路径信息,若不存在则返回FALSE和None。

12、analystic_expantion函数

         analystic_expantion 函数的目的是通过使用 Reeds-Shepp 路径类型计算从当前状态到目标状态的所有可能路径,并尝试找到一条不与障碍物发生碰撞的路径作为扩展路径。

13、is_collision函数

      用于检测车辆是否与障碍物发生碰撞。

14、calc_rs_path_cost函数

      用于计算路径的成本,即评估给定路径的优劣程度。该函数基于路径的长度、曲线类型以及转向角变化等因素来评估路径的优劣程度,从而帮助路径规划算法选择更优的路径。

15、calc_hybrid_cost函数

      calc_hybrid_cost函数用于计算Hybrid A * 算法中节点的成本,结合了路径规划成本和启发式地图成本,用于评估节点的优劣程度。

      该函数的目的是为了在路径规划时综合考虑路径的实际规划成本和启发式地图提供的信息,以更好地指导路径搜索算法选择节点,从而获得更优的路径。

16、calc_motion_set函数

17、is_same_grid函数

18、 calc_index函数

19、calc_parameters函数

20、draw_car函数

21、design_obstacles函数

22、主函数main

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号