If you have any comments/suggestions, encountered issues with the code, or have questions about the document, please contact Sarah (vexforum @sarah_97963a, discord @Sarah | iCRY | 97963A#2509)
Wiki dev's note: GitBook doesn't have a way to run code natively (at least that I know of), but there are some references to running/writing the code throughout this article. If you'd like to try pure pursuit yourself, or just see what the output of the code should look like, here's a link to Sarah's original Google Colab.
Import Necessary Libraries
import numpy as npimport matplotlib.pyplot as pltimport mathimport matplotlib.animation as animationfrom IPython import display
Helper Functions for Graphing
Below are some helper functions to help visualize the output of pure pursuit and path generator
Feel free to modify these functions to graph lines in the style you like
defadd_line (path) : for i inrange (0,len(path)): plt.plot(path[i][0],path[i][1],'.',color='red',markersize=10)for i inrange(0,len(path)-1): plt.plot([path[i][0],path[i+1][0]],[path[i][1],path[i+1][1]],color='b') plt.axis('scaled')# plt.show()defadd_complicated_line (path,lineStyle,lineColor,lineLabel) :for i inrange (0,len(path)): plt.plot(path[i][0],path[i][1],'.',color='red',markersize=10)for i inrange(0,len(path)-1):if(i ==0):# plt.plot([path[i][0],path[i+1][0]],[path[i][1],path[i+1][1]],color='b') plt.plot([path[i][0],path[i+1][0]],[path[i][1],path[i+1][1]],lineStyle,color=lineColor,label=lineLabel)else: plt.plot([path[i][0],path[i+1][0]],[path[i][1],path[i+1][1]],lineStyle,color=lineColor) plt.axis('scaled')defhighlight_points (points,pointColor):for point in points : plt.plot(point[0], point[1], '.', color = pointColor, markersize =10)defdraw_circle (x,y,r,circleColor): xs = [] ys = [] angles = np.arange(0, 2.2*np.pi, 0.2)for angle in angles : xs.append(r*np.cos(angle) + x) ys.append(r*np.sin(angle) + y) plt.plot(xs, ys, '-', color = circleColor)
What is the Pure Pursuit Controller?
The pure pursuit controller is an automatic steering method for wheeled mobile robots. It is a steering method, which means it computes the necessary angular velocity for a wheeled robot to stay on pre-computed paths. Linear velocity is assumed to be constant. Therefore, an additional velocity controller of your choice is needed if you wish to slow down the robot as it approaches the target (something as simple as a proportional controller could work). During the 2020-2021 VRC season, our team used the pure pursuit controller for the programming skills challenge and achieved great results. Videos of our robot in action can be found here.
Read a brief introduction of the pure pursuit controller by MathWorks here.
A quick demonstration of path tracking using the pure pursuit controller:
In the animation shown above, the dotted gray line is the pre-computed path that the robot needs to follow and the solid black line is the robot's trajectory. The big yellow dot and the short red line represent the robot's current position and heading. There is another solid line connecting the point the robot is "chasing after" and the robot's current position, but it's very hard to see in this demo. Although the robot's trajectory does not match with the path perfectly, the pure pursuit controller still performed decently well.
In the sections below, we will implement a basic pure pursuit controller for a differential drive (NOT the type of differential commonly used for power sharing in VRC), a robot model fairly similar to the 4 wheel tank drives in VRC.
Limitations
As shown above, the pure pursuit controller cannot trace a path perfectly due to the non-zero look ahead distance. The more sharp corners the path contains, the worse the performance. There are two ways to achieve better performance under the presence of sharp corners:
Optimize the path during the path generation process
Make improvements to the pure pursuit controller itself
For most circumstances in the VRC competition, the basic pure pursuit controller will suffice even without the above two improvements.
How does it work?
The key of the pure pursuit controller involves calculating a goal point on the path that is a certain distance ld, which is called the look ahead distance, from the robot's current position. Then, the goal point is used to calculate the appropriate angular velocity needed for the robot to move toward that point. As the robot moves forward, a different goal point on the path is chosen and the robot's angular velocity gets updated. Since the robot can never reach this goal point and the goal point stays on the path, the end result would be the robot tracing the path.
To find the appropriate goal point for the robot to follow mathematically, a circle centered at the robot's current position with radius ld is made. The circle's intersections with the path are the potential goal points. We will return to this part of the algorithm later in the document.
A more intuitive way of understanding and visualizing the pure pursuit controller would be the GIF below:
The carrot is always at a constant distance away from the donkey so the donkey moves in the direction the rider wants by chasing after the carrot.
In our implementation of the pure pursuit controller, the path is a 2D array represented by equally spaced points (1D array). The basic algorithm is to use a for loop to go through each pair of points to determine if there are any intersections within each line segments. If goal points are found, follow the most appropriate one. These steps will be repeated until the end of the path is reached.
Line-Circle Intersection
Let's start by looking at the basic math behind choosing goal points -- the line-circle intersection. There are multiple ways to find the intersection, we have found that this method (the screenshot below) was the easiest to implement and debug. (Side note: the math here might look a bit terrifying, but it's okay if you don't fully understand how it works. This statement only applies to this section.)
In the method presented above, the circle is assumed to be centered at the the origin. In our application, the circle would be centered at the robot's current position [currentX, currentY] and the two points that define the line would be arbitrary points pt1 = [x1, y1] and pt2 = [x2, y2]. Therefore, to apply this method, we first need to subtract currentX and currentY from [x1, y1] and [x2, y2] to center the system at origin. Again, we perform such offsets so that we can use the method from the screenshot above, which requires the circle to be centered at the origin.
In the code cell below, you are encouraged to code such line-circle intersection algorithm yourself. If you do not know python at all, complete code with comments is provided.
# helper function: sgn(num)# returns -1 if num is negative, 1 otherwisedefsgn (num): if num >=0:return1else:return-1# currentPos: [currentX, currentY]# pt1: [x1, y1]# pt2: [x2, y2]defline_circle_intersection (currentPos,pt1,pt2,lookAheadDis):# extract currentX, currentY, x1, x2, y1, and y2 from input arrays # DO NOT modified these variables! currentX = currentPos[0] currentY = currentPos[1] x1 = pt1[0] y1 = pt1[1] x2 = pt2[0] y2 = pt2[1]# boolean variable to keep track of if intersections are found# remember to set this correctly for the graphing functions to work! intersectFound =False# your code goes here# output (intersections found) should be stored in arrays sol1 and sol2 in the form of sol1 = [sol1_x, sol1_y]# if two solutions are the same, store the same values in both sol1 and sol2# graphing functions to visualize the outcome # ---------------------------------------------------------------------------------------------------------------------------------------
plt.plot ([x1, x2], [y1, y2], '--', color='grey')draw_circle (currentX, currentY, lookAheadDis, 'orange')if intersectFound ==False:print ('No intersection Found!')else:print ('Solution 1 found at [{}, {}]'.format(sol1[0], sol1[1]))print ('Solution 2 found at [{}, {}]'.format(sol2[0], sol2[1])) plt.plot ([sol1[0]], [sol1[1]], '.', weight=8, color='red', label='sol1') plt.plot ([sol2[0]], [sol2[1]], '.', weight=8, color='blue', label='sol2') plt.legend() plt.axis('scaled') plt.show()# now call this function and see the results!line_circle_intersection ([0, 1], [2, 3], [-2, -4], 0.9)
Line-Circle Intersection: Commented Code Example
# helper function: sgn(num)# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1# currentPos: [currentX, currentY]# pt1: [x1, y1]# pt2: [x2, y2]defline_circle_intersection (currentPos,pt1,pt2,lookAheadDis):# extract currentX, currentY, x1, x2, y1, and y2 from input arrays currentX = currentPos[0] currentY = currentPos[1] x1 = pt1[0] y1 = pt1[1] x2 = pt2[0] y2 = pt2[1]# boolean variable to keep track of if intersections are found intersectFound =False# output (intersections found) should be stored in arrays sol1 and sol2 # if two solutions are the same, store the same values in both sol1 and sol2 # subtract currentX and currentY from [x1, y1] and [x2, y2] to offset the system to origin x1_offset = x1 - currentX y1_offset = y1 - currentY x2_offset = x2 - currentX y2_offset = y2 - currentY # calculate the discriminant using equations from the wolframalpha article dx = x2_offset - x1_offset dy = y2_offset - y1_offset dr = math.sqrt (dx**2+ dy**2) D = x1_offset*y2_offset - x2_offset*y1_offset discriminant = (lookAheadDis**2) * (dr**2) - D**2# if discriminant is >= 0, there exist solutionsif discriminant >=0: intersectFound =True# calculate the solutions sol_x1 = (D * dy +sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_x2 = (D * dy -sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_y1 = (- D * dx +abs(dy)* np.sqrt(discriminant)) / dr**2 sol_y2 = (- D * dx -abs(dy)* np.sqrt(discriminant)) / dr**2# add currentX and currentY back to the solutions, offset the system back to its original position sol1 = [sol_x1 + currentX, sol_y1 + currentY] sol2 = [sol_x2 + currentX, sol_y2 + currentY] # graphing functions to visualize the outcome # ---------------------------------------------------------------------------------------------------------------------------------------
plt.plot ([x1, x2], [y1, y2], '--', color='grey')draw_circle (currentX, currentY, lookAheadDis, 'orange')if intersectFound ==False:print ('No intersection Found!')else:print ('Solution 1 found at [{}, {}]'.format(sol1[0], sol1[1]))print ('Solution 2 found at [{}, {}]'.format(sol2[0], sol2[1])) plt.plot (sol1[0], sol1[1], '.', markersize=10, color='red', label='sol1') plt.plot (sol2[0], sol2[1], '.', markersize=10, color='blue', label='sol2') plt.legend() plt.axis('scaled') plt.show()# now call this function and see the results!line_circle_intersection ([0, 1], [2, 3], [-2, -4], 1)
Line-Circle Intersection with Bounds
You might have already noticed something not quite right when messing with the algorithm above. Under certain condition, although the intersections found are on the infinite line defined by pt1 and pt2, they are not exactly within the range of [x1, y1] and [x2, y2]. Consider the situation illustrated below:
Although intersections are found, they are not within any of the line segments in the path. Those are not the points we want the robot to follow! So how do we prevent such situation from happening?
The solution is simple. After intersections are found, we check to see if their x values are within the range of [min(x1, x2), max(x1, x2)] and their y values are within the range of [min(y1, y2), max(y1, y2)]. That is, for a solution to be valid and intersectFound to be True, the following conditions need to be satisfied:
You can modify your line_circle_intersection function in the code cell below. Completed and commented code is also provided.
(Side note: use (a >= x) && (x >= b) in VEXCode.)
# helper function: sgn(num)# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1# currentPos: [currentX, currentY]# pt1: [x1, y1]# pt2: [x2, y2]defline_circle_intersection (currentPos,pt1,pt2,lookAheadDis):# extract currentX, currentY, x1, x2, y1, and y2 from input arrays# DO NOT modify these variables! currentX = currentPos[0] currentY = currentPos[1] x1 = pt1[0] y1 = pt1[1] x2 = pt2[0] y2 = pt2[1]# boolean variable to keep track of if intersections are found# remember to set this correctly for the graphing functions to work! intersectFound =False# your code goes here# output (intersections found) should be stored in arrays sol1 and sol2 in the form of sol1 = [sol1_x, sol1_y]# hint: if two solutions are the same, store the same values in both sol1 and sol2# add boundary checking code here # for now, the function should print out 'solution _ is valid!' ( _ can be 1 or 2) if a solution point is within the range
# we will use this outcome more in the next section# graphing functions to visualize the outcome # ---------------------------------------------------------------------------------------------------------------------------------------
plt.plot ([x1, x2], [y1, y2], '--', color='grey')draw_circle (currentX, currentY, lookAheadDis, 'orange')if intersectFound ==False:print ('No intersection Found!')else:print ('Solution 1 found at [{}, {}]'.format(sol1[0], sol1[1]))print ('Solution 2 found at [{}, {}]'.format(sol2[0], sol2[1])) plt.plot (sol1[0], sol1[1], '.', markersize=10, color='red', label='sol1') plt.plot (sol2[0], sol2[1], '.', markersize=10, color='blue', label='sol2') plt.legend() plt.axis('scaled') plt.show()# now call this function and see the results!line_circle_intersection ([0, 1], [2, 3], [-2, -4], 0.9)
Line-Circle Intersection with Bounds: Commented Code Example
# helper function: sgn(num)# returns -1 if num is negative, 1 otherwisedefsgn (num): if num >=0:return1else:return-1# currentPos: [currentX, currentY]# pt1: [x1, y1]# pt2: [x2, y2]defline_circle_intersection (currentPos,pt1,pt2,lookAheadDis):# extract currentX, currentY, x1, x2, y1, and y2 from input arrays currentX = currentPos[0] currentY = currentPos[1] x1 = pt1[0] y1 = pt1[1] x2 = pt2[0] y2 = pt2[1]# boolean variable to keep track of if intersections are found intersectFound =False# output (intersections found) should be stored in arrays sol1 and sol2# if two solutions are the same, store the same values in both sol1 and sol2# subtract currentX and currentY from [x1, y1] and [x2, y2] to offset the system to origin x1_offset = x1 - currentX y1_offset = y1 - currentY x2_offset = x2 - currentX y2_offset = y2 - currentY # calculate the discriminant using equations from the wolframalpha article dx = x2_offset - x1_offset dy = y2_offset - y1_offset dr = math.sqrt (dx**2+ dy**2) D = x1_offset*y2_offset - x2_offset*y1_offset discriminant = (lookAheadDis**2) * (dr**2) - D**2# if discriminant is >= 0, there exist solutionsif discriminant >=0:# calculate the solutions sol_x1 = (D * dy +sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_x2 = (D * dy -sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_y1 = (- D * dx +abs(dy)* np.sqrt(discriminant)) / dr**2 sol_y2 = (- D * dx -abs(dy)* np.sqrt(discriminant)) / dr**2# add currentX and currentY back to the solutions, offset the system back to its original position sol1 = [sol_x1 + currentX, sol_y1 + currentY] sol2 = [sol_x2 + currentX, sol_y2 + currentY]# find min and max x y values minX =min(x1, x2) maxX =max(x1, x2) minY =min(y1, y2) maxY =max(y1, y2)# check to see if any of the two solution points are within the correct range # for a solution point to be considered valid, its x value needs to be within minX and maxX AND its y value needs to be between minY and maxY
# if sol1 OR sol2 are within the range, intersection is found if (minX <= sol1[0] <= maxX and minY <= sol1[1] <= maxY) or (minX <= sol2[0] <= maxX and minY <= sol2[1] <= maxY) :
intersectFound =True# now do a more detailed check to determine which point is valid, which is notif (minX <= sol1[0]<= maxX and minY <= sol1[1]<= maxY) :print ('solution 1 is valid!')if (minX <= sol2[0]<= maxX and minY <= sol2[1]<= maxY) :print ('solution 1 is valid!')# graphing functions to visualize the outcome # ----------------------------------------------------------------------------------------------------------------------------------------
plt.plot ([x1, x2], [y1, y2], '--', color='grey')draw_circle (currentX, currentY, lookAheadDis, 'orange')if intersectFound ==False:print ('No intersection Found!')else:print ('Solution 1 found at [{}, {}]'.format(sol1[0], sol1[1]))print ('Solution 2 found at [{}, {}]'.format(sol2[0], sol2[1])) plt.plot (sol1[0], sol1[1], '.', markersize=10, color='red', label='sol1') plt.plot (sol2[0], sol2[1], '.', markersize=10, color='blue', label='sol2') plt.legend() plt.axis('scaled') plt.show()# now call this function and see the results!line_circle_intersection ([0, 1], [2, 3], [0, 1], 1)
Choosing the Goal Point
All possible outcomes of the line-circle intersections (with a single line segment) are the following:
Situation 1:
No intersection. If this is the case, the discriminant will be negative.
Situation 2:
Intersections are found, but they are not in between (x1, y1) and (x2, y2). The discriminant is positive, but we should still consider this situation as “no intersection”.
Situation 3:
There is one and only one intersection inside the range of (x1, y1) and (x2, y2). The discriminant is positive, and the robot should follow the intersection found.
Situation 4:
There are two intersections and both are in between (x1, y1) and (x2, y2). In this situation, we have to determine which point is better for the robot to follow. One method we can to use is to calculate the distance between the intersections and the second point (x2, y2)(the path goes in the direction of (x1, y1) -> (x2, y2)) and pick the point that is closer to the second point (in other words, the point closer to the end of the path).
In some extreme cases where the robot is traveling through sharp corners, it might create multiple intersections with multiple line segments (as shown in the picture below). We can code our program in a way that the robot would follow the first valid point it found. This way, in extreme cases where the path overlaps itself or there exists multiple sharp corners, the robot would not skip a portion of the path altogether. In order to prevent the robot from going backwards in the path, we can create a variable lastFoundIndex to store the index of the point it just passed. Every time the loop runs, it will only check the points that are located after path[lastFoundIndex]. This way, the segments the robot has already traveled through will not be checked again for intersections. In the cases that no new intersection has been found (robot deviates from the path), the robot will follow the point at lastFoundIndex.
Let's take a closer look at how lastFoundIndex and the goal point choosing function should work.
When the robot first entered the path (the 1st loop iteration), lastFoundIndex = 0. The line-circle intersection search starts from path[0], an intersection is found and selected as the goal point for the robot to move toward.
At the end of the 1st loop iteration, lastFoundIndex is updated to 0 since the goal point was found in between path[0] and path[1]. When the 2nd loop iteration starts, the robot has moved closer to path[1]. Since lastFoundIndex is still 0, the line-circle intersection starts from path[0] again. This time, there are two intersections with the path: one located in between path[0] and path[1] and the other located in between path[1] and path[2].
Following normal procedure, the algorithm would choose the intersection in between path[0] and path[1] as the goal point and break out of the search loop. As we can tell from the picture below, it's not a very good choice since it will cause the robot to go backward. To avoid such bad goal point choice, we can add an additional condition to evaluate the goal point found by the algorithm: the search loop break statement can only be reached if the distance between the goal point chosen and the next point in path is shorter than the distance between the robot's current position and the next point in path. If the above statement is not true, the search loop continues. We can also increment lastFoundIndex in case the robot fails to find intersections in the next line segment. This will prevent the robot from going backward since path[lastFoundIndex] will become the goal point when no intersection can be found.
In the situation illustrated below, it is obvious that the distance between the goalPt (the left intersection) and path[1] is greater that the distance between currentPos and path[1] (both distances are marked with dotted red line). Hence, the search loop continues and the intersection between path[1] and path[2] will be chosen in the next search loop iteration.
At the end of the 2nd loop iteration, lastFoundIndex is updated to 1. When the 3rd loop iteration starts, the next search will start from path[1] so the intersection in between path[0] and path[1] will be omitted. The portion of the path omitted for goal point searching is marked in brown.
The code cell below can be used for implementing the goal_pt_search algorithm yourself (for people that are not familiar enough with python, commented code is provided as before). A sample path has been provided.
# implement the goal_pt_search algorithm# last section of the visualization code needs to be uncommented when the function is done!!# you will need the line-circle intersection with bounds algorithm completed in previous sectionspath1 = [[0.0, 0.0], [0.011580143395790051, 0.6570165243709267], [0.07307496243411533, 1.2724369146199181], [0.3136756819515748, 1.7385910188236868], [0.8813313906933087, 1.9320292911046681], [1.6153051608455251, 1.9849785681091774], [2.391094224224885, 1.9878393390954208], [3.12721333474683, 1.938831731115573], [3.685011039017028, 1.7396821576569221], [3.9068092597113266, 1.275245079016133], [3.9102406525571713, 0.7136897450501469], [3.68346383786099, 0.2590283720040381], [3.1181273273535957, 0.06751996250999465], [2.3832776875784316, 0.013841087641154892], [1.5971423891000605, 0.0023698980178599423], [0.7995795475309813, 0.0003490964043320208], [0, 0]]
# helper functionsdefpt_to_pt_distance (pt1,pt2): distance = np.sqrt((pt2[0] - pt1[0])**2+ (pt2[1] - pt1[1])**2)return distance# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1defgoal_pt_search (path,currentPos,lookAheadDis,lastFoundIndex) :# initialize goalPt in case no intersection is found goalPt = [None,None]# use for loop to search intersections startingIndex = lastFoundIndexfor i inrange (startingIndex, len(path1)):# your code goes here# store the final goal point in array goalPt = [goalPt_x, goalPt_y]# right now you can just run the function as is to see the visualization of the path# but when you are ready to code, the pass below needs to be removed pass# visualize outcome # ---------------------------------------------------------------------------------------------------------------------------------------
# traveled path (path omitted for searching) will be marked in brown field = plt.figure() xscale,yscale = (1.5,1.5) # <- modify these values to scale the plot path_ax = field.add_axes([0,0,xscale,yscale])add_complicated_line(path[0:lastFoundIndex+1],'--','brown','traveled path')add_complicated_line(path[lastFoundIndex:len(path)],'--','grey','remaining path')highlight_points(path[0:lastFoundIndex], 'brown')highlight_points(path[lastFoundIndex:len(path)], 'grey') xMin, yMin, xMax, yMax = (-1,-1,5,3) # <- modify these values to set plot boundaries# (minX, minY, maxX, maxY)# plot field path_ax.plot([xMin,xMax],[yMin,yMin],color='black') path_ax.plot([xMin,xMin],[yMin,yMax],color='black') path_ax.plot([xMax,xMax],[yMin,yMax],color='black') path_ax.plot([xMax,xMin],[yMax,yMax],color='black')# set grid xTicks = np.arange(xMin, xMax+1, 2) yTicks = np.arange(yMin, yMax+1, 2) path_ax.set_xticks(xTicks) path_ax.set_yticks(yTicks) path_ax.grid(True) path_ax.set_xlim(xMin-0.25,xMax+0.25) path_ax.set_ylim(yMin-0.25,yMax+0.25)# plot start and end path_ax.plot(path[0][0],path[0][1],'.',color='blue',markersize=15,label='start') path_ax.plot(path[-1][0],path[-1][1],'.',color='green',markersize=15,label='end')# plot current position and goal pointdraw_circle (currentPos[0], currentPos[1], lookAheadDis, 'orange') plt.plot (currentPos[0], currentPos[1], '.', markersize=15, color='orange', label='current position')if goalPt != [None,None] : plt.plot (goalPt[0], goalPt[1], '.', markersize=15, color='red', label='goal point')add_complicated_line([currentPos, goalPt], '-', 'black', 'look ahead distance') path_ax.legend()# call the function to see the resultsgoal_pt_search (path1, [1, 2.2], 0.8, 3)
Choosing the Goal Point: Commented Code Example
path1 = [[0.0, 0.0], [0.011580143395790051, 0.6570165243709267], [0.07307496243411533, 1.2724369146199181], [0.3136756819515748, 1.7385910188236868], [0.8813313906933087, 1.9320292911046681], [1.6153051608455251, 1.9849785681091774], [2.391094224224885, 1.9878393390954208], [3.12721333474683, 1.938831731115573], [3.685011039017028, 1.7396821576569221], [3.9068092597113266, 1.275245079016133], [3.9102406525571713, 0.7136897450501469], [3.68346383786099, 0.2590283720040381], [3.1181273273535957, 0.06751996250999465], [2.3832776875784316, 0.013841087641154892], [1.5971423891000605, 0.0023698980178599423], [0.7995795475309813, 0.0003490964043320208], [0, 0]]
# helper functionsdefpt_to_pt_distance (pt1,pt2) : distance = np.sqrt((pt2[0] - pt1[0])**2+ (pt2[1] - pt1[1])**2)return distance# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1defgoal_pt_search (path,currentPos,lookAheadDis,lastFoundIndex) :# extract currentX and currentY currentX = currentPos[0] currentY = currentPos[1]# initialize goalPt in case no intersection is found goalPt = [None,None]# use for loop to search intersections intersectFound =False startingIndex = lastFoundIndexfor i inrange (startingIndex, len(path)-1):# beginning of line-circle intersection code x1 = path[i][0] - currentX y1 = path[i][1] - currentY x2 = path[i+1][0] - currentX y2 = path[i+1][1] - currentY dx = x2 - x1 dy = y2 - y1 dr = math.sqrt (dx**2+ dy**2) D = x1*y2 - x2*y1 discriminant = (lookAheadDis**2) * (dr**2) - D**2if discriminant >=0: sol_x1 = (D * dy +sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_x2 = (D * dy -sgn(dy)* dx * np.sqrt(discriminant)) / dr**2 sol_y1 = (- D * dx +abs(dy)* np.sqrt(discriminant)) / dr**2 sol_y2 = (- D * dx -abs(dy)* np.sqrt(discriminant)) / dr**2 sol_pt1 = [sol_x1 + currentX, sol_y1 + currentY] sol_pt2 = [sol_x2 + currentX, sol_y2 + currentY]# end of line-circle intersection code minX =min(path[i][0], path[i+1][0]) minY =min(path[i][1], path[i+1][1]) maxX =max(path[i][0], path[i+1][0]) maxY =max(path[i][1], path[i+1][1])# if one or both of the solutions are in range if ((minX <= sol_pt1[0] <= maxX) and (minY <= sol_pt1[1] <= maxY)) or ((minX <= sol_pt2[0] <= maxX) and (minY <= sol_pt2[1] <= maxY)):
foundIntersection =True# if both solutions are in range, check which one is better if ((minX <= sol_pt1[0] <= maxX) and (minY <= sol_pt1[1] <= maxY)) and ((minX <= sol_pt2[0] <= maxX) and (minY <= sol_pt2[1] <= maxY)):
# make the decision by comparing the distance between the intersections and the next point in pathifpt_to_pt_distance(sol_pt1, path[i+1])<pt_to_pt_distance(sol_pt2, path[i+1]): goalPt = sol_pt1else: goalPt = sol_pt2# if not both solutions are in range, take the one that's in rangeelse:# if solution pt1 is in range, set that as goal pointif (minX <= sol_pt1[0]<= maxX) and (minY <= sol_pt1[1]<= maxY): goalPt = sol_pt1else: goalPt = sol_pt2 # only exit loop if the solution pt found is closer to the next pt in path than the current posifpt_to_pt_distance (goalPt, path[i+1])<pt_to_pt_distance ([currentX, currentY], path[i+1]):# update lastFoundIndex and exit lastFoundIndex = ibreakelse: # in case for some reason the robot cannot find intersection in the next path segment, but we also don't want it to go backward
lastFoundIndex = i+1# if no solutions are in rangeelse: foundIntersection =False# no new intersection found, potentially deviated from the path# follow path[lastFoundIndex] goalPt = [path[lastFoundIndex][0], path[lastFoundIndex][1]]# visualize outcome # ---------------------------------------------------------------------------------------------------------------------------------------
# traveled path (path omitted for searching) will be marked in brown field = plt.figure() xscale,yscale = (1.5,1.5) # <- modify these values to scale the plot path_ax = field.add_axes([0,0,xscale,yscale])add_complicated_line(path[0:lastFoundIndex+1],'--','brown','traveled path')add_complicated_line(path[lastFoundIndex:len(path)],'--','grey','remaining path')highlight_points(path[0:lastFoundIndex], 'brown')highlight_points(path[lastFoundIndex:len(path)], 'grey') xMin, yMin, xMax, yMax = (-1,-1,5,3) # <- modify these values to set plot boundaries# (minX, minY, maxX, maxY)# plot field path_ax.plot([xMin,xMax],[yMin,yMin],color='black') path_ax.plot([xMin,xMin],[yMin,yMax],color='black') path_ax.plot([xMax,xMax],[yMin,yMax],color='black') path_ax.plot([xMax,xMin],[yMax,yMax],color='black')# set grid xTicks = np.arange(xMin, xMax+1, 2) yTicks = np.arange(yMin, yMax+1, 2) path_ax.set_xticks(xTicks) path_ax.set_yticks(yTicks) path_ax.grid(True) path_ax.set_xlim(xMin-0.25,xMax+0.25) path_ax.set_ylim(yMin-0.25,yMax+0.25)# plot start and end path_ax.plot(path[0][0],path[0][1],'.',color='blue',markersize=15,label='start') path_ax.plot(path[-1][0],path[-1][1],'.',color='green',markersize=15,label='end')# plot current position and goal pointdraw_circle (currentX, currentY, lookAheadDis, 'orange') plt.plot (currentX,currentY, '.', markersize=15, color='orange', label='current position')if goalPt != [None,None] : plt.plot (goalPt[0], goalPt[1], '.', markersize=15, color='red', label='goal point')add_complicated_line([currentPos, goalPt], '-', 'black', 'look ahead distance')print('Goal point found at {}'.format(goalPt))else:print('No intersection found!') path_ax.legend()# call the function to see the resultsgoal_pt_search (path1, [1, 2.2], 0.6, 3)
Move Toward Target Point - Part 1
Now that we have a goal point determined, the next step would be to make the robot move toward that point. Let's first look at how to move the robot to a fixed target point. Consider the situation illustrated below, where the robot is located at [currentX, currentY] and we would like to make it move to point [targetX, targetY]:
The robot would need to perform two actions at the same time: moving toward the target point and turning toward the target point. To achieve this, we need to calculate linear and turn velocity separately and add them together to obtain the final speed. Obtaining linear error is easy in this case. In the picture above, the total distance the robot needs to travel is marked with the dotted gray line and has the value sqrt(pow(targetY - currentY, 2) + pow(targetX - currentX, 2)).
Obtaining turn error, however, is slightly more complicated. The total amount the robot needs to turn is marked in blue. Since the direction of the turn is counterclockwise, the blue angle (which would be referred to as turnError from this point on) should be positive. We can obtain the magnitude of this angle by performing subtraction:
The turnError can be calculated by subtracting the orange angle (which is just currentHeading) from the grey angle, which is the angle vector [targetX-currentX, targetY-currentY] makes with the global X axis. From this point on, we will call this grey angle absTargetAngle. We can use the built-in atan2 function to calculate its value. Since atan2's output ranges from -180 to 180 but our Cartesian coordinate system's range is 0 to 360, we can simply add 360 to the atan2 output if its negative.
Before we move on, there are some special cases that are worth mentioning. If currentHeading is 1 degrees while absTargetAngle is 359 degrees, we would get turnError = absTargetHeading - currentHeading = 1 - 359 = -358 deg, which is totally nonsense. To make the function as efficient as possible, the robot should just turn to the target from another direction if turnError turns out to be greater than 180 degrees.
The code cells below is a space for you to write a find_min_angle function that computes the minimum turn error the robot would need to turn toward the target point. If your implementation is correct, the angle returned should stay in the range of [-180, 180] under all circumstances. If you are more familiar with using rotation and modulo (%) instead of heading, feel free to implement the function that way. The animation function provided in later sections uses heading by default, but code that's compatible with rotation and modulo will also be provided.
# helper function# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1# this function should subtract currentHeading from absTargetAngle# the correct output should be stored in variable minAngledeffind_min_angle (absTargetAngle,currentHeading) : # your code goes here minAngle =# visualization # ---------------------------------------------------------------------------------------------------------------------------------------
print ('The minimum angle between {}(absTargetHeading) and {}(currentHeading) is {}.'.format(absTargetAngle, currentHeading, minAngle))
add_complicated_line([[0, 0], [5*np.cos(absTargetAngle*np.pi/180), 5*np.sin(absTargetAngle*np.pi/180)]], '--', 'red', 'absTargetHeading')
add_complicated_line([[0, 0], [5*np.cos(currentHeading*np.pi/180), 5*np.sin(currentHeading*np.pi/180)]], '--', 'blue', 'currentHeading')
plt.plot(5*np.cos(absTargetAngle*np.pi/180), 5*np.sin(absTargetAngle*np.pi/180), '.', markersize=15, color='red') plt.plot(5*np.cos(currentHeading*np.pi/180), 5*np.sin(currentHeading*np.pi/180), '.', markersize=15, color='blue') plt.axis('scaled') plt.legend() plt.show()# call the function to see resultfind_min_angle(0, 95)
Find Minimum Turn Error: Commented Code Example
# helper function# returns -1 if num is negative, 1 otherwisedefsgn (num):if num >=0:return1else:return-1# this function should subtract currentHeading from absTargetAngle# the correct output should be stored in variable minAngledeffind_min_angle (absTargetAngle,currentHeading) : minAngle = absTargetAngle - currentHeading # idk why this works but it worksif minAngle >180or minAngle <-180: minAngle =-1*sgn(minAngle)* (360-abs(minAngle))# for visualization # ---------------------------------------------------------------------------------------------------------------------------------------
print ('The minimum angle between {}(absTargetHeading) and {}(currentHeading) is {}.'.format(absTargetAngle, currentHeading, minAngle))
add_complicated_line([[0, 0], [5*np.cos(absTargetAngle*np.pi/180), 5*np.sin(absTargetAngle*np.pi/180)]], '--', 'red', 'absTargetHeading')
add_complicated_line([[0, 0], [5*np.cos(currentHeading*np.pi/180), 5*np.sin(currentHeading*np.pi/180)]], '--', 'blue', 'currentHeading')
plt.plot(5*np.cos(absTargetAngle*np.pi/180), 5*np.sin(absTargetAngle*np.pi/180), '.', markersize=15, color='red') plt.plot(5*np.cos(currentHeading*np.pi/180), 5*np.sin(currentHeading*np.pi/180), '.', markersize=15, color='blue') plt.axis('scaled') plt.legend() plt.show()# call the function to see resultfind_min_angle(345, 45)
Moving Toward Target Point - Part 2
With the linear error and the turn error obtained, we can calculate the robot's left and right side velocity based on those. Since the turn error is positive when the robot turns counterclockwise, the left side should have velocity linearVel - turnVel and the right side should have velocity linearVel + turnVel. In the code example below, a basic proportional controller is used to compute these two velocities. Please note that the animation code assumes linearVel and turnVel are in velocityUnits::pct.
To recap, the move to target point algorithm goes as follows (not real code):
// compute linear error
linearVel = sqrt(pow(targetY - currentY, 2) + pow(targetX - currentX, 2))
// compute turn error
absTargetAngle = atan2(targetY - currentY, targetX - currentX)
if absTargetAngle < 0 : absTargetAngle += 360
turnError = find_min_angle(absTargetAngle, currentHeading)
// compute linear and turn velocities using controller of your choice
// send commend to motors
left side motor speed = linearVel - turnVel
right side motor speed = linearVel + turnVel
defsgn (num): if num >=0:return1else:return-1# set this to true if you use rotationsusing_rotation =False# return the computed turn velocity/turnVel and linear velocity/linVel (both in pct). positive turn vel means turning counterclockwise, negative turn vel means turning clockwise
# the function below will be used to create a nice animation. # for the animation to work, please do not include any while loops in this function