def main(argv):
try:
opts, args = getopt.getopt(argv,"hp:c:",["Planning=","Control="])
except getopt.GetoptError:
print('<ERROR> main.py -p <PlanningMethod 0:AStar 1:RRTStar> -c <ControlMethod 0:PurePersuit 1:Stanley>')
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print('main.py -p <PlanningMethod 0:AStar 1:RRTStar> -c <ControlMethod 0:PurePersuit 1:Stanley>')
sys.exit()
elif opt in ("-p", "--Planning"):
plan_type = arg
elif opt in ("-c", "--Control"):
control_type = arg
# Path Tracking Controller
if control_type == '0':
from PathTracking.bicycle_pure_pursuit import PurePursuitControl
controller = PurePursuitControl(kp=0.7,Lfc=10)
elif control_type == '1':
from PathTracking.bicycle_stanley import StanleyControl
controller = StanleyControl(kp=0.75)
else:
print('ERROR')
sys.exit(2)
# Path Planning Planner
if plan_type == '0':
from PathPlanning.dijkstra import AStar
planner = AStar(m_dilate)
elif plan_type == '1':
from PathPlanning.rrt_star import RRTStar
planner = RRTStar(m_dilate)
#from PathPlanning.cubic_spline import *
global nav_pos, path, init_pos, pos, planFlag, smooth, last_nav, canControl
cv2.namedWindow(window_name)
cv2.setMouseCallback(window_name, mouse_click)
# Main Loop
while(True):
# Update State
car.update()
pos = (car.x, car.y, car.yaw)
print("\rState: "+car.state_str(), "| Goal:", nav_pos, end="\t")
img_ = img.copy()
#####################################
# Control and Path Planning
## Planning ##
if nav_pos != None and planFlag == False and last_nav != nav_pos:
path = planner.planning(start=(int(car.x),int(car.y)),goal=nav_pos,img=img)
last_nav = nav_pos
canControl = True
img_ = img.copy()
cv2.imshow(window_name,img_)
planFlag = True
# Extract Path
if not smooth:
for i in range(len(path)-1):
cv2.line(img, path[i], path[i+1], (1,0,0), 2)
else:
path = np.array(cubic_spline_2d(path, interval=1))
for i in range(len(path)-1):
cv2.line(img, pos_int(path[i]), pos_int(path[i+1]), (1,0,0), 1)
## Control ##
if canControl == True:
controller.set_path(path)
# ================= Control Algorithm =================
# PID Longitude Control
end_dist = np.hypot(path[-1,0]-car.x, path[-1,1]-car.y)
target_v = 20 if end_dist > 180 else 0
next_a = 0.1*(target_v - car.v)
# Lateral Control
state = {"x":car.x, "y":car.y, "yaw":car.yaw, "v":car.v, "l":car.l, "delta":car.delta}
next_delta, target = controller.feedback(state)
car.control(next_a, next_delta)
# =====================================================
#####################################
# Collision Simulation
if collision_detect(car, m):
car.redo()
car.v = -0.5*car.v
# Environment Rendering
if nav_pos is not None:
cv2.circle(img_,nav_pos,5,(0.5,0.5,1.0),3)
img_ = car.render(img_)
img_ = cv2.flip(img_, 0)
cv2.imshow(window_name ,img_)
# Remove Path
Path = None
planFlag = False
# Keyboard
k = cv2.waitKey(1)
if k == ord("a"):
car.delta += 5
elif k == ord("d"):
car.delta -= 5
elif k == ord("w"):
car.v += 4
elif k == ord("s"):
car.v -= 4
elif k == ord("r"):
car.init_state(init_pos)
nav_pos = None
path = None
print("Reset!!")
if k == 27:
print()
break
# Simulation Model
from bicycle_model import KinematicModel
car = KinematicModel(l=20, d=5, wu=5, wv=2, car_w=14, car_f=25, car_r=5)
car.init_state(init_pos)
# Path Tracking Controller
if control_type == 0:
from PathTracking.bicycle_pid import PidControl
controller = PidControl(kp=0.03, ki=0.00005, kd=0.08)
elif control_type == 1:
from PathTracking.bicycle_pure_pursuit import PurePursuitControl
controller = PurePursuitControl(kp=0.7,Lfc=10)
elif control_type == 2:
from PathTracking.bicycle_stanley import StanleyControl
controller = StanleyControl(kp=0.5)
elif control_type == 3:
from PathTracking.bicycle_lqr import LQRControl
controller = LQRControl()
# Path Planning Planner
if plan_type == 0:
from PathPlanning.astar import AStar
planner = AStar(m_dilate)
elif plan_type == 1:
from PathPlanning.rrt_star import RRTStar
planner = RRTStar(m_dilate)
from PathPlanning.cubic_spline import *
##############################
# Util Function
def Path_Tracking_Controller(image=None, control_type=0):
global controller
path_Controller = []
if control_type == 0:
from PathTracking.bicycle_pure_pursuit import PurePursuitControl
controller = PurePursuitControl(kp=0.7, Lfc=10)
#print("\rpath 1=")
#print(path)
controller.set_path(path)
#while (True):
print("\rState: " + car.state_str(), end="\t")
# ================= Control Algorithm =================
# PID Longitude Control
end_dist = np.hypot(path[-1, 0] - car.x, path[-1, 1] - car.y)
target_v = 15 if end_dist > 25 else 0
next_a = 0.5 * (target_v - car.v)
# Pure Pursuit Lateral Control
state = {
"x": car.x,
"y": car.y,
"yaw": car.yaw,
"v": car.v,
"l": car.l
}
next_delta, target = controller.feedback(state)
car.control(next_a, next_delta)
# =====================================================
print("\rState: " + car.state_str(), "| Goal:", nav_pos, end="\t")
"""
# Update & Render
car.update()
img_ = image.copy()
cv2.circle(image, (int(target[0]), int(target[1])), 3, (1, 0.3, 0.7), 2) # target points
img_ = car.render(img_)
img_ = cv2.flip(img_, 0)
cv2.imshow("Pure-Pursuit Control Test", img_)
k = cv2.waitKey(1)
if k == ord('r'):
car.init_state(pos)
if k == 27:
print()
break #[346, 380, 347, 380]
"""
elif control_type == 1:
from PathTracking.bicycle_stanley import StanleyControl
controller = StanleyControl(kp=0.5)
#print("\rpath 1=")
#print(path)
controller.set_path(path)
#while (True):
print("\rState: " + car.state_str(), end="\t")
# PID Longitude Control
end_dist = np.hypot(path[-1, 0] - car.x, path[-1, 1] - car.y)
target_v = 23 if end_dist > 30 else 0
next_a = 1 * (target_v - car.v)
# Stanley Lateral Control
state = {
"x": car.x,
"y": car.y,
"yaw": car.yaw,
"delta": car.delta,
"v": car.v,
"l": car.l
}
next_delta, target = controller.feedback(state)
car.control(next_a, next_delta)
"""