code clean up for LGTM

This commit is contained in:
Atsushi Sakai
2019-02-03 10:00:09 +09:00
parent 138f478f4d
commit f0b6b7a94d
8 changed files with 20 additions and 21 deletions

View File

@@ -5,15 +5,18 @@ Path tracking simulation with LQR steering control and PID speed control.
author Atsushi Sakai (@Atsushi_twi)
"""
import scipy.linalg as la
import matplotlib.pyplot as plt
import math
import numpy as np
import sys
sys.path.append("../../PathPlanning/CubicSpline/")
import numpy as np
import math
import matplotlib.pyplot as plt
import scipy.linalg as la
import cubic_spline_planner
try:
import cubic_spline_planner
except:
raise
Kp = 1.0 # speed proportional gain
@@ -76,7 +79,6 @@ def solve_DARE(A, B, Q, R):
Xn = A.T @ X @ A - A.T @ X @ B @ \
la.inv(R + B.T @ X @ B) @ B.T @ X @ A + Q
if (abs(Xn - X)).max() < eps:
X = Xn
break
X = Xn
@@ -206,8 +208,8 @@ def closed_loop_prediction(cx, cy, cyaw, ck, speed_profile, goal):
plt.plot(cx[target_ind], cy[target_ind], "xg", label="target")
plt.axis("equal")
plt.grid(True)
plt.title("speed[km/h]:" + str(round(state.v * 3.6, 2)) +
",target index:" + str(target_ind))
plt.title("speed[km/h]:" + str(round(state.v * 3.6, 2))
+ ",target index:" + str(target_ind))
plt.pause(0.0001)
return t, x, y, yaw, v

View File

@@ -140,6 +140,7 @@ def main():
assert lastIndex >= target_ind, "Cannot goal"
if show_animation:
plt.cla()
plt.plot(cx, cy, ".r", label="course")
plt.plot(x, y, "-b", label="trajectory")
plt.legend()
@@ -148,7 +149,7 @@ def main():
plt.axis("equal")
plt.grid(True)
flg, ax = plt.subplots(1)
plt.subplots(1)
plt.plot(t, [iv * 3.6 for iv in v], "-r")
plt.xlabel("Time[s]")
plt.ylabel("Speed[km/h]")