Update controlsd.py

This commit is contained in:
infiniteCable
2025-04-06 14:48:29 +02:00
committed by GitHub
parent 3b46b401a8
commit 9a0b4fd155
+8 -1
View File
@@ -18,6 +18,7 @@ from openpilot.selfdrive.controls.lib.latcontrol_angle import LatControlAngle, S
from openpilot.selfdrive.controls.lib.latcontrol_torque import LatControlTorque
from openpilot.selfdrive.controls.lib.longcontrol import LongControl
from openpilot.selfdrive.locationd.helpers import PoseCalibrator, Pose
from openpilot.selfdrive.controls.lib.longitudinal_mpc_lib.long_mpc import get_T_FOLLOW
State = log.SelfdriveState.OpenpilotState
LaneChangeState = log.LaneChangeState
@@ -48,6 +49,7 @@ class Controls:
self.steer_limited_by_controls = False
self.curvature = 0.0
self.desired_curvature = 0.0
self.roll = 0.0
self.pose_calibrator = PoseCalibrator()
self.calibrated_pose: Pose | None = None
@@ -80,7 +82,7 @@ class Controls:
self.VM.update_params(x, sr)
steer_angle_without_offset = math.radians(CS.steeringAngleDeg - lp.angleOffsetDeg)
self.curvature = -self.VM.calc_curvature(steer_angle_without_offset, CS.vEgo, lp.roll)
self.curvature = -self.VM.calc_curvature(steer_angle_without_offset, CS.vEgo, 0)
# Update Torque Params
if self.CP.lateralTuning.which() == 'torque':
@@ -156,6 +158,9 @@ class Controls:
def publish(self, CC, CC_SP, lac_log):
CS = self.sm['carState']
CC.currentCurvature = self.curvature
CC.rollDEPRECATED = self.roll
# Orientation and angle rates can be useful for carcontroller
# Only calibrated (car) frame is relevant for the carcontroller
if self.calibrated_pose is not None:
@@ -174,7 +179,9 @@ class Controls:
hudControl.speedVisible = CC.enabled
hudControl.lanesVisible = CC.enabled
hudControl.leadVisible = self.sm['longitudinalPlan'].hasLead
hudControl.leadDistance = self.sm['longitudinalPlan'].leadDistance
hudControl.leadDistanceBars = self.sm['selfdriveState'].personality.raw + 1
hudControl.leadFollowTime = get_T_FOLLOW(hudControl.leadDistanceBars - 1)
hudControl.visualAlert = self.sm['selfdriveState'].alertHudVisual
hudControl.rightLaneVisible = True