Skip to content

Commit

Permalink
Fix initial implementation
Browse files Browse the repository at this point in the history
  • Loading branch information
gzaragoza314 committed Feb 3, 2024
1 parent 646f434 commit a12d373
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion rb_ws/src/buggy/scripts/2d_sim/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,8 +137,9 @@ def step(self):
velocity = self.velocity
steering_angle = self.steering_angle

# RK4 to calculate next position
h = 1/self.rate
state = np.array([e_utm, n_utm, heading, steering_angle])
state = np.array([e_utm, n_utm, np.deg2rad(heading), np.deg2rad(steering_angle)])
k1 = self.dynamics(state, velocity)
k2 = self.dynamics(state + h/2 * k1, velocity)
k3 = self.dynamics(state + h/2 * k2, velocity)
Expand All @@ -147,6 +148,7 @@ def step(self):
final_state = state + h/6 * (k1 + 2 * k2 + 2 * k3 + k4)

e_utm_new, n_utm_new, heading_new, _ = final_state
heading_new = np.rad2deg(heading_new)

with self.lock:
self.e_utm = e_utm_new
Expand Down

0 comments on commit a12d373

Please sign in to comment.