Skip to content

Commit dd43ae2

Browse files
authored
Full localizer: Use standard naming conventions (#25007)
Use standard naming conventions
1 parent c49f997 commit dd43ae2

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

selfdrive/locationd/models/loc_kf.py

+16-16
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,15 @@ class States():
2525
ODO_SCALE_UNUSED = slice(18, 19) # odometer scale
2626
ACCELERATION = slice(19, 22) # Acceleration in device frame in m/s**2
2727
FOCAL_SCALE_UNUSED = slice(22, 23) # focal length scale
28-
IMU_OFFSET = slice(23, 26) # imu offset angles in radians
28+
IMU_FROM_DEVICE_EULER = slice(23, 26) # imu offset angles in radians
2929
GLONASS_BIAS = slice(26, 27) # GLONASS bias in m expressed as bias + freq_num*freq_slope
3030
GLONASS_FREQ_SLOPE = slice(27, 28) # GLONASS bias in m expressed as bias + freq_num*freq_slope
3131
CLOCK_ACCELERATION = slice(28, 29) # clock acceleration in light-meters/s**2,
3232
ACCELEROMETER_SCALE_UNUSED = slice(29, 30) # scale of mems accelerometer
3333
ACCELEROMETER_BIAS = slice(30, 33) # bias of mems accelerometer
3434
# TODO the offset is likely a translation of the sensor, not a rotation of the camera
35-
WIDE_CAM_OFFSET = slice(33, 36) # wide camera offset angles in radians (tici only)
36-
# We curently do not use ACCELEROMETER_SCALE to avoid instability due to too many free variables (ACCELEROMETER_SCALE, ACCELEROMETER_BIAS, IMU_OFFSET).
35+
WIDE_FROM_DEVICE_EULER = slice(33, 36) # wide camera offset angles in radians (tici only)
36+
# We curently do not use ACCELEROMETER_SCALE to avoid instability due to too many free variables (ACCELEROMETER_SCALE, ACCELEROMETER_BIAS, IMU_FROM_DEVICE_EULER).
3737
# From experiments we see that ACCELEROMETER_BIAS is more correct than ACCELEROMETER_SCALE
3838

3939
# Error-state has different slices because it is an ESKF
@@ -47,13 +47,13 @@ class States():
4747
ODO_SCALE_ERR_UNUSED = slice(17, 18)
4848
ACCELERATION_ERR = slice(18, 21)
4949
FOCAL_SCALE_ERR_UNUSED = slice(21, 22)
50-
IMU_OFFSET_ERR = slice(22, 25)
50+
IMU_FROM_DEVICE_EULER_ERR = slice(22, 25)
5151
GLONASS_BIAS_ERR = slice(25, 26)
5252
GLONASS_FREQ_SLOPE_ERR = slice(26, 27)
5353
CLOCK_ACCELERATION_ERR = slice(27, 28)
5454
ACCELEROMETER_SCALE_ERR_UNUSED = slice(28, 29)
5555
ACCELEROMETER_BIAS_ERR = slice(29, 32)
56-
WIDE_CAM_OFFSET_ERR = slice(32, 35)
56+
WIDE_FROM_DEVICE_EULER_ERR = slice(32, 35)
5757

5858

5959
class LocKalman():
@@ -140,15 +140,15 @@ def generate_code(generated_dir, N=4):
140140
cd = state[States.CLOCK_DRIFT, :]
141141
roll_bias, pitch_bias, yaw_bias = state[States.GYRO_BIAS, :]
142142
acceleration = state[States.ACCELERATION, :]
143-
imu_angles = state[States.IMU_OFFSET, :]
144-
imu_angles[0, 0] = 0 # not observable enough
145-
imu_angles[2, 0] = 0 # not observable enough
143+
imu_from_device_euler = state[States.IMU_FROM_DEVICE_EULER, :]
144+
imu_from_device_euler[0, 0] = 0 # not observable enough
145+
imu_from_device_euler[2, 0] = 0 # not observable enough
146146
glonass_bias = state[States.GLONASS_BIAS, :]
147147
glonass_freq_slope = state[States.GLONASS_FREQ_SLOPE, :]
148148
ca = state[States.CLOCK_ACCELERATION, :]
149149
accel_bias = state[States.ACCELEROMETER_BIAS, :]
150-
wide_cam_angles = state[States.WIDE_CAM_OFFSET, :]
151-
wide_cam_angles[0, 0] = 0 # not observable enough
150+
wide_from_device_euler = state[States.WIDE_FROM_DEVICE_EULER, :]
151+
wide_from_device_euler[0, 0] = 0 # not observable enough
152152

153153
dt = sp.Symbol('dt')
154154

@@ -273,15 +273,15 @@ def generate_code(generated_dir, N=4):
273273
los_vector[2] * (sat_vz - vz) +
274274
cd[0]])
275275

276-
imu_rot = euler_rotate(*imu_angles)
277-
h_gyro_sym = imu_rot * sp.Matrix([vroll + roll_bias,
276+
imu_from_device = euler_rotate(*imu_from_device_euler)
277+
h_gyro_sym = imu_from_device * sp.Matrix([vroll + roll_bias,
278278
vpitch + pitch_bias,
279279
vyaw + yaw_bias])
280280

281281
pos = sp.Matrix([x, y, z])
282282
# add 1 for stability, prevent division by 0
283283
gravity = quat_rot.T * ((EARTH_GM / ((x**2 + y**2 + z**2 + 1)**(3.0 / 2.0))) * pos)
284-
h_acc_sym = imu_rot * (gravity + acceleration + accel_bias)
284+
h_acc_sym = imu_from_device * (gravity + acceleration + accel_bias)
285285
h_acc_stationary_sym = acceleration
286286
h_phone_rot_sym = sp.Matrix([vroll, vpitch, vyaw])
287287
h_relative_motion = sp.Matrix(quat_rot.T * v)
@@ -297,7 +297,7 @@ def generate_code(generated_dir, N=4):
297297
[h_phone_rot_sym, ObservationKind.CAMERA_ODO_ROTATION, None],
298298
[h_acc_stationary_sym, ObservationKind.NO_ACCEL, None]]
299299

300-
wide_cam_rot = euler_rotate(*wide_cam_angles)
300+
wide_from_device = euler_rotate(*wide_from_device_euler)
301301
# MSCKF configuration
302302
if N > 0:
303303
# experimentally found this is correct value for imx298 with 910 focal length
@@ -312,7 +312,7 @@ def generate_code(generated_dir, N=4):
312312

313313
track_pos_sym = sp.Matrix([track_x - x, track_y - y, track_z - z])
314314
track_pos_rot_sym = quat_rot.T * track_pos_sym
315-
track_pos_rot_wide_cam_sym = wide_cam_rot * track_pos_rot_sym
315+
track_pos_rot_wide_cam_sym = wide_from_device * track_pos_rot_sym
316316
h_track_sym[-2:, :] = sp.Matrix([focal_scale * (track_pos_rot_sym[1] / track_pos_rot_sym[0]),
317317
focal_scale * (track_pos_rot_sym[2] / track_pos_rot_sym[0])])
318318
h_track_wide_cam_sym[-2:, :] = sp.Matrix([focal_scale * (track_pos_rot_wide_cam_sym[1] / track_pos_rot_wide_cam_sym[0]),
@@ -329,7 +329,7 @@ def generate_code(generated_dir, N=4):
329329
quat_rot = quat_rotate(*q)
330330
track_pos_sym = sp.Matrix([track_x - x, track_y - y, track_z - z])
331331
track_pos_rot_sym = quat_rot.T * track_pos_sym
332-
track_pos_rot_wide_cam_sym = wide_cam_rot * track_pos_rot_sym
332+
track_pos_rot_wide_cam_sym = wide_from_device * track_pos_rot_sym
333333
h_track_sym[n * 2:n * 2 + 2, :] = sp.Matrix([focal_scale * (track_pos_rot_sym[1] / track_pos_rot_sym[0]),
334334
focal_scale * (track_pos_rot_sym[2] / track_pos_rot_sym[0])])
335335
h_track_wide_cam_sym[n * 2: n * 2 + 2, :] = sp.Matrix([focal_scale * (track_pos_rot_wide_cam_sym[1] / track_pos_rot_wide_cam_sym[0]),

0 commit comments

Comments
 (0)