|
|
|
@ -4,7 +4,9 @@ from collections import deque |
|
|
|
|
|
|
|
|
|
from cereal import log |
|
|
|
|
from opendbc.car.lateral import FRICTION_THRESHOLD, get_friction |
|
|
|
|
from opendbc.car.tests.test_lateral_limits import MAX_LAT_JERK_UP |
|
|
|
|
from openpilot.common.constants import ACCELERATION_DUE_TO_GRAVITY |
|
|
|
|
from openpilot.common.filter_simple import FirstOrderFilter |
|
|
|
|
from openpilot.selfdrive.controls.lib.drive_helpers import MIN_SPEED |
|
|
|
|
from openpilot.selfdrive.controls.lib.latcontrol import LatControl |
|
|
|
|
from openpilot.common.pid import PIDController |
|
|
|
@ -36,6 +38,8 @@ class LatControlTorque(LatControl): |
|
|
|
|
self.steering_angle_deadzone_deg = self.torque_params.steeringAngleDeadzoneDeg |
|
|
|
|
self.LATACCEL_REQUEST_BUFFER_NUM_FRAMES = int(1 / self.dt) |
|
|
|
|
self.requested_lateral_accel_buffer = deque([0.] * self.LATACCEL_REQUEST_BUFFER_NUM_FRAMES , maxlen=self.LATACCEL_REQUEST_BUFFER_NUM_FRAMES) |
|
|
|
|
self.previous_measurement = 0.0 |
|
|
|
|
self.measurement_rate_filter = FirstOrderFilter(0.0, 1 / (2 * np.pi * (MAX_LAT_JERK_UP - 0.5)), self.dt) |
|
|
|
|
|
|
|
|
|
def update_live_torque_params(self, latAccelFactor, latAccelOffset, friction): |
|
|
|
|
self.torque_params.latAccelFactor = latAccelFactor |
|
|
|
@ -53,9 +57,10 @@ class LatControlTorque(LatControl): |
|
|
|
|
output_torque = 0.0 |
|
|
|
|
pid_log.active = False |
|
|
|
|
else: |
|
|
|
|
actual_curvature = -VM.calc_curvature(math.radians(CS.steeringAngleDeg - params.angleOffsetDeg), CS.vEgo, params.roll) |
|
|
|
|
measured_curvature = -VM.calc_curvature(math.radians(CS.steeringAngleDeg - params.angleOffsetDeg), CS.vEgo, params.roll) |
|
|
|
|
roll_compensation = params.roll * ACCELERATION_DUE_TO_GRAVITY |
|
|
|
|
curvature_deadzone = abs(VM.calc_curvature(math.radians(self.steering_angle_deadzone_deg), CS.vEgo, 0.0)) |
|
|
|
|
lateral_accel_deadzone = curvature_deadzone * CS.vEgo ** 2 |
|
|
|
|
|
|
|
|
|
delay_frames = int(np.clip(lat_delay / self.dt, 1, self.LATACCEL_REQUEST_BUFFER_NUM_FRAMES)) |
|
|
|
|
expected_lateral_accel = self.requested_lateral_accel_buffer[-delay_frames] |
|
|
|
@ -64,12 +69,13 @@ class LatControlTorque(LatControl): |
|
|
|
|
self.requested_lateral_accel_buffer.append(future_desired_lateral_accel) |
|
|
|
|
gravity_adjusted_future_lateral_accel = future_desired_lateral_accel - roll_compensation |
|
|
|
|
desired_lateral_jerk = (future_desired_lateral_accel - expected_lateral_accel) / lat_delay |
|
|
|
|
actual_lateral_accel = actual_curvature * CS.vEgo ** 2 |
|
|
|
|
lateral_accel_deadzone = curvature_deadzone * CS.vEgo ** 2 |
|
|
|
|
|
|
|
|
|
measurement = measured_curvature * CS.vEgo ** 2 |
|
|
|
|
measurement_rate = self.measurement_rate_filter.update((measurement - self.previous_measurement) / self.dt) |
|
|
|
|
self.previous_measurement = measurement |
|
|
|
|
|
|
|
|
|
low_speed_factor = (np.interp(CS.vEgo, LOW_SPEED_X, LOW_SPEED_Y) / max(CS.vEgo, MIN_SPEED)) ** 2 |
|
|
|
|
setpoint = lat_delay * desired_lateral_jerk + expected_lateral_accel |
|
|
|
|
measurement = actual_lateral_accel |
|
|
|
|
error = setpoint - measurement |
|
|
|
|
error_lsf = error + low_speed_factor * error |
|
|
|
|
|
|
|
|
@ -83,6 +89,7 @@ class LatControlTorque(LatControl): |
|
|
|
|
|
|
|
|
|
freeze_integrator = steer_limited_by_safety or CS.steeringPressed or CS.vEgo < 5 |
|
|
|
|
output_lataccel = self.pid.update(pid_log.error, |
|
|
|
|
-measurement_rate, |
|
|
|
|
feedforward=ff, |
|
|
|
|
speed=CS.vEgo, |
|
|
|
|
freeze_integrator=freeze_integrator) |
|
|
|
@ -94,8 +101,8 @@ class LatControlTorque(LatControl): |
|
|
|
|
pid_log.d = float(self.pid.d) |
|
|
|
|
pid_log.f = float(self.pid.f) |
|
|
|
|
pid_log.output = float(-output_torque) # TODO: log lat accel? |
|
|
|
|
pid_log.actualLateralAccel = float(actual_lateral_accel) |
|
|
|
|
pid_log.desiredLateralAccel = float(expected_lateral_accel) |
|
|
|
|
pid_log.actualLateralAccel = float(measurement) |
|
|
|
|
pid_log.desiredLateralAccel = float(setpoint) |
|
|
|
|
pid_log.saturated = bool(self._check_saturation(self.steer_max - abs(output_torque) < 1e-3, CS, steer_limited_by_safety, curvature_limited)) |
|
|
|
|
|
|
|
|
|
# TODO left is positive in this convention |
|
|
|
|