77"""
88import math
99import matplotlib .pyplot as plt
10+ import numpy as np
1011
1112
1213def mod2pi (theta ):
@@ -224,7 +225,7 @@ def generate_course(length, mode, c):
224225
225226 for m , l in zip (mode , length ):
226227 pd = 0.0
227- if m is "S" :
228+ if m == "S" :
228229 d = 1.0 / c
229230 else : # turning couse
230231 d = np .deg2rad (3.0 )
@@ -234,23 +235,23 @@ def generate_course(length, mode, c):
234235 px .append (px [- 1 ] + d * c * math .cos (pyaw [- 1 ]))
235236 py .append (py [- 1 ] + d * c * math .sin (pyaw [- 1 ]))
236237
237- if m is "L" : # left turn
238+ if m == "L" : # left turn
238239 pyaw .append (pyaw [- 1 ] + d )
239- elif m is "S" : # Straight
240+ elif m == "S" : # Straight
240241 pyaw .append (pyaw [- 1 ])
241- elif m is "R" : # right turn
242+ elif m == "R" : # right turn
242243 pyaw .append (pyaw [- 1 ] - d )
243244 pd += d
244245
245246 d = l - pd
246247 px .append (px [- 1 ] + d * c * math .cos (pyaw [- 1 ]))
247248 py .append (py [- 1 ] + d * c * math .sin (pyaw [- 1 ]))
248249
249- if m is "L" : # left turn
250+ if m == "L" : # left turn
250251 pyaw .append (pyaw [- 1 ] + d )
251- elif m is "S" : # Straight
252+ elif m == "S" : # Straight
252253 pyaw .append (pyaw [- 1 ])
253- elif m is "R" : # right turn
254+ elif m == "R" : # right turn
254255 pyaw .append (pyaw [- 1 ] - d )
255256 pd += d
256257
0 commit comments