sensors/vehicle_angular_velocity: accumualted notch filtering and reset improvements

- apply sensor scaling immediately to keep things simple (FIFO vs regular)
 - inline filter helpers (minor performance improvement)
 - dynamic notch filtering
    - reorder by axis (applied per axis)
    - don't remove notch filters immediately if ESC or FFT data times out
    - constrain notch filter frequency and bandwidth to safe range (minimum bandwidth for flaot precision, Nyquist, etc)
 - add safe constraint on dt
This commit is contained in:
Daniel Agar 2021-05-25 12:29:16 -04:00
parent 561cfca4f9
commit 3269ee8df1
3 changed files with 171 additions and 124 deletions

View File

@ -33,6 +33,8 @@
#pragma once #pragma once
#include <stdint.h>
namespace px4 namespace px4
{ {

View File

@ -144,37 +144,33 @@ bool VehicleAngularVelocity::UpdateSampleRate()
return PX4_ISFINITE(_filter_sample_rate_hz) && (_filter_sample_rate_hz > 0); return PX4_ISFINITE(_filter_sample_rate_hz) && (_filter_sample_rate_hz > 0);
} }
void VehicleAngularVelocity::ResetFilters(float new_scale) void VehicleAngularVelocity::ResetFilters()
{ {
if ((_filter_sample_rate_hz > 0) && PX4_ISFINITE(_filter_sample_rate_hz)) { if ((_filter_sample_rate_hz > 0) && PX4_ISFINITE(_filter_sample_rate_hz)) {
const Vector3f angular_velocity{GetResetAngularVelocity(new_scale)}; const Vector3f angular_velocity_uncalibrated{GetResetAngularVelocity()};
const Vector3f angular_acceleration{GetResetAngularAcceleration(new_scale)}; const Vector3f angular_acceleration_uncalibrated{GetResetAngularAcceleration()};
for (int axis = 0; axis < 3; axis++) { for (int axis = 0; axis < 3; axis++) {
// angular velocity low pass // angular velocity low pass
_lp_filter_velocity[axis].set_cutoff_frequency(_filter_sample_rate_hz, _param_imu_gyro_cutoff.get()); _lp_filter_velocity[axis].set_cutoff_frequency(_filter_sample_rate_hz, _param_imu_gyro_cutoff.get());
_lp_filter_velocity[axis].reset(angular_velocity(axis)); _lp_filter_velocity[axis].reset(angular_velocity_uncalibrated(axis));
// angular velocity notch // angular velocity notch
_notch_filter_velocity[axis].setParameters(_filter_sample_rate_hz, _param_imu_gyro_nf_freq.get(), _notch_filter_velocity[axis].setParameters(_filter_sample_rate_hz, _param_imu_gyro_nf_freq.get(),
_param_imu_gyro_nf_bw.get()); _param_imu_gyro_nf_bw.get());
_notch_filter_velocity[axis].reset(angular_velocity(axis)); _notch_filter_velocity[axis].reset(angular_velocity_uncalibrated(axis));
// angular acceleration low pass // angular acceleration low pass
_lp_filter_acceleration[axis].set_cutoff_frequency(_filter_sample_rate_hz, _param_imu_dgyro_cutoff.get()); _lp_filter_acceleration[axis].set_cutoff_frequency(_filter_sample_rate_hz, _param_imu_dgyro_cutoff.get());
_lp_filter_acceleration[axis].reset(angular_acceleration(axis)); _lp_filter_acceleration[axis].reset(angular_acceleration_uncalibrated(axis));
} }
// dynamic notch filters, first disable, then force update (if available) // force reset notch filters on any scale change
DisableDynamicNotchEscRpm(); UpdateDynamicNotchEscRpm(true);
DisableDynamicNotchFFT(); UpdateDynamicNotchFFT(true);
UpdateDynamicNotchEscRpm(new_scale, true); _angular_velocity_raw_prev = angular_velocity_uncalibrated;
UpdateDynamicNotchFFT(new_scale, true);
_angular_velocity_prev = angular_velocity;
_last_scale = new_scale;
_reset_filters = false; _reset_filters = false;
perf_count(_filter_reset_perf); perf_count(_filter_reset_perf);
@ -231,7 +227,6 @@ bool VehicleAngularVelocity::SensorSelectionUpdate(bool force)
_reset_filters = true; _reset_filters = true;
_bias.zero(); _bias.zero();
_fifo_available = true; _fifo_available = true;
_last_scale = 0.f;
perf_count(_selection_changed_perf); perf_count(_selection_changed_perf);
@ -260,7 +255,6 @@ bool VehicleAngularVelocity::SensorSelectionUpdate(bool force)
_reset_filters = true; _reset_filters = true;
_bias.zero(); _bias.zero();
_fifo_available = false; _fifo_available = false;
_last_scale = 1.f;
perf_count(_selection_changed_perf); perf_count(_selection_changed_perf);
@ -350,30 +344,30 @@ void VehicleAngularVelocity::ParametersUpdate(bool force)
} }
} }
Vector3f VehicleAngularVelocity::GetResetAngularVelocity(float new_scale) const Vector3f VehicleAngularVelocity::GetResetAngularVelocity() const
{ {
if ((_last_publish != 0) && (new_scale > 0.f)) { if (_last_publish != 0) {
// angular velocity filtering is performed on raw unscaled data // angular velocity filtering is performed on raw unscaled data
// start with last valid vehicle body frame angular velocity and compute equivalent raw data (for current sensor selection) // start with last valid vehicle body frame angular velocity and compute equivalent raw data (for current sensor selection)
Vector3f angular_velocity{_calibration.Uncorrect(_angular_velocity + _bias) / new_scale}; Vector3f angular_velocity_uncalibrated{_calibration.Uncorrect(_angular_velocity + _bias)};
if (PX4_ISFINITE(angular_velocity(0)) if (PX4_ISFINITE(angular_velocity_uncalibrated(0))
&& PX4_ISFINITE(angular_velocity(1)) && PX4_ISFINITE(angular_velocity_uncalibrated(1))
&& PX4_ISFINITE(angular_velocity(2))) { && PX4_ISFINITE(angular_velocity_uncalibrated(2))) {
return angular_velocity; return angular_velocity_uncalibrated;
} }
} }
return Vector3f{0.f, 0.f, 0.f}; return Vector3f{0.f, 0.f, 0.f};
} }
Vector3f VehicleAngularVelocity::GetResetAngularAcceleration(float new_scale) const Vector3f VehicleAngularVelocity::GetResetAngularAcceleration() const
{ {
if ((_last_publish != 0) && (new_scale > 0.f)) { if (_last_publish != 0) {
// angular acceleration filtering is performed on unscaled angular velocity data // angular acceleration filtering is performed on unscaled angular velocity data
// start with last valid vehicle body frame angular acceleration and compute equivalent raw data (for current sensor selection) // start with last valid vehicle body frame angular acceleration and compute equivalent raw data (for current sensor selection)
Vector3f angular_acceleration{_calibration.rotation().I() *_angular_acceleration / new_scale}; Vector3f angular_acceleration{_calibration.rotation().I() *_angular_acceleration};
if (PX4_ISFINITE(angular_acceleration(0)) if (PX4_ISFINITE(angular_acceleration(0))
&& PX4_ISFINITE(angular_acceleration(1)) && PX4_ISFINITE(angular_acceleration(1))
@ -390,16 +384,20 @@ void VehicleAngularVelocity::DisableDynamicNotchEscRpm()
{ {
#if !defined(CONSTRAINED_FLASH) #if !defined(CONSTRAINED_FLASH)
// device id mismatch, disable all if (_dynamic_notch_esc_rpm_available) {
for (auto &dnf : _dynamic_notch_filter_esc_rpm) { for (int axis = 0; axis < 3; axis++) {
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) { for (int esc = 0; esc < MAX_NUM_ESC_RPM; esc++) {
for (int axis = 0; axis < 3; axis++) { for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) {
dnf[harmonic][axis].setParameters(0, 0, 0); _dynamic_notch_filter_esc_rpm[axis][esc][harmonic].setParameters(0, 0, 0);
}
_esc_available.set(esc, false);
} }
} }
_dynamic_notch_esc_rpm_available = false;
} }
_dynamic_notch_esc_rpm_available = false;
#endif // !CONSTRAINED_FLASH #endif // !CONSTRAINED_FLASH
} }
@ -407,121 +405,166 @@ void VehicleAngularVelocity::DisableDynamicNotchFFT()
{ {
#if !defined(CONSTRAINED_FLASH) #if !defined(CONSTRAINED_FLASH)
// device id mismatch, disable all if (_dynamic_notch_fft_available) {
for (auto &dnf : _dynamic_notch_filter_fft) {
for (int axis = 0; axis < 3; axis++) { for (int axis = 0; axis < 3; axis++) {
dnf[axis].setParameters(0, 0, 0); for (int peak = 0; peak < MAX_NUM_FFT_PEAKS; peak++) {
_dynamic_notch_filter_fft[axis][peak].setParameters(0, 0, 0);
}
} }
_dynamic_notch_fft_available = false;
} }
_dynamic_notch_fft_available = false;
#endif // !CONSTRAINED_FLASH #endif // !CONSTRAINED_FLASH
} }
void VehicleAngularVelocity::UpdateDynamicNotchEscRpm(float new_scale, bool force) void VehicleAngularVelocity::UpdateDynamicNotchEscRpm(bool force)
{ {
#if !defined(CONSTRAINED_FLASH) #if !defined(CONSTRAINED_FLASH)
const bool enabled = _param_imu_gyro_dyn_nf.get() & DynamicNotch::EscRpm; const bool enabled = _param_imu_gyro_dyn_nf.get() & DynamicNotch::EscRpm;
if (enabled && (_esc_status_sub.updated() || force)) { if (enabled && (_esc_status_sub.updated() || force)) {
_dynamic_notch_esc_rpm_available = false;
if (!_dynamic_notch_esc_rpm_available) {
// force update filters if previously disabled
force = true;
}
esc_status_s esc_status; esc_status_s esc_status;
if (_esc_status_sub.copy(&esc_status)) { if (_esc_status_sub.copy(&esc_status) && (hrt_elapsed_time(&esc_status.timestamp) < DYNAMIC_NOTCH_FITLER_TIMEOUT)) {
for (size_t i = 0; i < MAX_NUM_ESC_RPM; i++) {
static constexpr int32_t MIN_ESC_RPM = 20 * 60; // 20 Hz safe minimum limit TODO: configurable
if ((esc_status.esc[i].timestamp != 0) && ((_timestamp_sample_last - esc_status.esc[i].timestamp) < 1_s) static constexpr int32_t ESC_RPM_MIN = 20 * 60; // TODO: configurable
&& (esc_status.esc[i].esc_rpm > MIN_ESC_RPM)) { const int32_t ESC_RPM_MAX = roundf(_filter_sample_rate_hz / 3.f * 60.f); // upper bound safety (well below Nyquist)
const float esc_hz = static_cast<float>(esc_status.esc[i].esc_rpm) / 60.f; for (size_t esc = 0; esc < math::min(esc_status.esc_count, (uint8_t)MAX_NUM_ESC_RPM); esc++) {
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) { const esc_report_s &esc_report = esc_status.esc[esc];
const float frequency_hz = esc_hz * (harmonic + 1);
auto &dnf0 = _dynamic_notch_filter_esc_rpm[i][harmonic][0]; // only update if ESC RPM range seems valid
const float change_percent = fabsf(dnf0.getNotchFreq() - frequency_hz) / frequency_hz; if ((esc_report.esc_rpm > ESC_RPM_MIN) && (esc_report.esc_rpm < ESC_RPM_MAX)
&& (hrt_elapsed_time(&esc_report.timestamp) < DYNAMIC_NOTCH_FITLER_TIMEOUT)) {
// for each ESC check determine if enabled/disabled from first notch (x axis, harmonic 0)
auto &nfx0 = _dynamic_notch_filter_esc_rpm[0][esc][0];
bool reset = (nfx0.getNotchFreq() <= FLT_EPSILON); // notch was previously disabled
const float esc_hz = static_cast<float>(esc_report.esc_rpm) / 60.f;
// update filter parameters if frequency changed or forced
if (force || reset || (fabsf(nfx0.getNotchFreq() - esc_hz) > FLT_EPSILON)) {
static constexpr float ESC_NOTCH_BW_HZ = 8.f; // TODO: configurable bandwidth
// force reset if the notch frequency jumps significantly
if (!reset || (fabsf(nfx0.getNotchFreq() - esc_hz) > ESC_NOTCH_BW_HZ)) {
reset = true;
}
for (int harmonic = MAX_NUM_ESC_RPM_HARMONICS; harmonic >= 0; harmonic--) {
const float frequency_hz = esc_hz * (harmonic + 1);
if (change_percent > 0.001f) {
// peak frequency changed by at least 0.1%
for (int axis = 0; axis < 3; axis++) { for (int axis = 0; axis < 3; axis++) {
auto &dnf = _dynamic_notch_filter_esc_rpm[i][harmonic][axis]; _dynamic_notch_filter_esc_rpm[axis][esc][harmonic].setParameters(_filter_sample_rate_hz, frequency_hz, ESC_NOTCH_BW_HZ);
dnf.setParameters(_filter_sample_rate_hz, frequency_hz, 1.f); // TODO: configurable bandwidth
} }
}
// only reset if there's sufficient change (> 1%) perf_count(_dynamic_notch_filter_esc_rpm_update_perf);
if (force || (change_percent > 0.01f)) { }
const Vector3f reset_angular_velocity{GetResetAngularVelocity(new_scale)};
for (int axis = 0; axis < 3; axis++) { if (force || reset) {
auto &dnf = _dynamic_notch_filter_esc_rpm[i][harmonic][axis]; const Vector3f reset_angular_velocity{GetResetAngularVelocity()};
dnf.reset(reset_angular_velocity(axis));
} for (int axis = 0; axis < 3; axis++) {
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) {
_dynamic_notch_filter_esc_rpm[axis][esc][harmonic].reset(reset_angular_velocity(axis));
} }
perf_count(_dynamic_notch_filter_esc_rpm_update_perf);
} }
} }
_dynamic_notch_esc_rpm_available = true; _dynamic_notch_esc_rpm_available = true;
_esc_available.set(esc, true);
_last_esc_rpm_notch_update[esc] = esc_report.timestamp;
} else { } else if (force || (hrt_elapsed_time(&_last_esc_rpm_notch_update[esc]) >= DYNAMIC_NOTCH_FITLER_TIMEOUT)) {
// disable all notch filters for this ESC // disable all notch filters for this ESC after timeout
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) { _esc_available.set(esc, false);
for (int axis = 0; axis < 3; axis++) {
_dynamic_notch_filter_esc_rpm[i][harmonic][axis].setParameters(0, 0, 0); for (int axis = 0; axis < 3; axis++) {
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) {
_dynamic_notch_filter_esc_rpm[axis][esc][harmonic].setParameters(0, 0, 0);
} }
} }
} }
} }
} else {
DisableDynamicNotchEscRpm();
} }
} }
#endif // !CONSTRAINED_FLASH #endif // !CONSTRAINED_FLASH
} }
void VehicleAngularVelocity::UpdateDynamicNotchFFT(float new_scale, bool force) void VehicleAngularVelocity::UpdateDynamicNotchFFT(bool force)
{ {
#if !defined(CONSTRAINED_FLASH) #if !defined(CONSTRAINED_FLASH)
const bool enabled = _param_imu_gyro_dyn_nf.get() & DynamicNotch::FFT; const bool enabled = _param_imu_gyro_dyn_nf.get() & DynamicNotch::FFT;
if (enabled && (_sensor_gyro_fft_sub.updated() || force)) { if (enabled && (_sensor_gyro_fft_sub.updated() || force)) {
_dynamic_notch_fft_available = false;
if (!_dynamic_notch_fft_available) {
// force update filters if previously disabled
force = true;
}
sensor_gyro_fft_s sensor_gyro_fft; sensor_gyro_fft_s sensor_gyro_fft;
if (_sensor_gyro_fft_sub.copy(&sensor_gyro_fft) && (sensor_gyro_fft.device_id == _selected_sensor_device_id) if (_sensor_gyro_fft_sub.copy(&sensor_gyro_fft)
&& (sensor_gyro_fft.device_id == _selected_sensor_device_id)
&& (hrt_elapsed_time(&sensor_gyro_fft.timestamp) < DYNAMIC_NOTCH_FITLER_TIMEOUT)
&& (fabsf(sensor_gyro_fft.sensor_sample_rate_hz - _filter_sample_rate_hz) < 10.f)) { && (fabsf(sensor_gyro_fft.sensor_sample_rate_hz - _filter_sample_rate_hz) < 10.f)) {
// ignore any peaks below half the gyro cutoff frequency
const float peak_freq_min = _param_imu_gyro_cutoff.get() / 2.f;
const float peak_freq_max = _filter_sample_rate_hz / 3.f; // upper bound safety (well below Nyquist)
const float bandwidth = math::constrain(sensor_gyro_fft.resolution_hz, 8.f, 30.f); // TODO: base on numerical limits?
float *peak_frequencies[] {sensor_gyro_fft.peak_frequencies_x, sensor_gyro_fft.peak_frequencies_y, sensor_gyro_fft.peak_frequencies_z}; float *peak_frequencies[] {sensor_gyro_fft.peak_frequencies_x, sensor_gyro_fft.peak_frequencies_y, sensor_gyro_fft.peak_frequencies_z};
for (int axis = 0; axis < 3; axis++) { for (int axis = 0; axis < 3; axis++) {
for (int i = 0; i < MAX_NUM_FFT_PEAKS; i++) { for (int peak = 0; peak < MAX_NUM_FFT_PEAKS; peak++) {
auto &dnf = _dynamic_notch_filter_fft[i][axis]; auto &nf = _dynamic_notch_filter_fft[axis][peak];
const float &peak_freq = peak_frequencies[axis][i];
if (PX4_ISFINITE(peak_freq) && (peak_freq > 1.f)) { bool reset = (nf.getNotchFreq() <= FLT_EPSILON); // notch was previously disabled
const float peak_diff_abs = fabsf(dnf.getNotchFreq() - peak_freq);
const float change_percent = peak_diff_abs / peak_freq;
if (force || (change_percent > 0.001f)) { const float peak_freq = peak_frequencies[axis][peak];
// peak frequency changed by at least 0.1%
dnf.setParameters(_filter_sample_rate_hz, peak_freq, sensor_gyro_fft.resolution_hz);
// only reset if there's sufficient change if (PX4_ISFINITE(peak_freq) && (peak_freq > peak_freq_min) && (peak_freq < peak_freq_max)) {
if (peak_diff_abs > sensor_gyro_fft.resolution_hz) { // force reset if the notch frequency jumps significantly
dnf.reset(GetResetAngularVelocity(new_scale)(axis)); if (fabsf(nf.getNotchFreq() - peak_freq) > bandwidth) {
} reset = true;
}
// update filter parameters if frequency changed or forced
if (force || (fabsf(nf.getNotchFreq() - peak_freq) > FLT_EPSILON)) {
nf.setParameters(_filter_sample_rate_hz, peak_freq, bandwidth);
perf_count(_dynamic_notch_filter_fft_update_perf); perf_count(_dynamic_notch_filter_fft_update_perf);
} }
if (force || reset) {
const Vector3f reset_angular_velocity{GetResetAngularVelocity()};
nf.reset(reset_angular_velocity(axis));
}
_dynamic_notch_fft_available = true; _dynamic_notch_fft_available = true;
} else { } else {
// disable this notch filter // disable this notch filter (if it isn't already)
dnf.setParameters(0, 0, 0); if (force || !reset) {
nf.setParameters(0, 0, 0);
}
} }
} }
} }
@ -542,13 +585,10 @@ float VehicleAngularVelocity::FilterAngularVelocity(int axis, float data[], int
if (_dynamic_notch_esc_rpm_available) { if (_dynamic_notch_esc_rpm_available) {
perf_begin(_dynamic_notch_filter_esc_rpm_perf); perf_begin(_dynamic_notch_filter_esc_rpm_perf);
for (auto &dnf : _dynamic_notch_filter_esc_rpm) { for (int esc = 0; esc < MAX_NUM_ESC_RPM; esc++) {
for (int harmonic = 0; harmonic < MAX_NUM_ESC_RPM_HARMONICS; harmonic++) { if (_esc_available[esc]) {
if (dnf[harmonic][axis].getNotchFreq() > 0.f) { for (int harmonic = MAX_NUM_ESC_RPM_HARMONICS - 1; harmonic >= 0; harmonic--) {
dnf[harmonic][axis].applyArray(data, N); _dynamic_notch_filter_esc_rpm[axis][esc][harmonic].applyArray(data, N);
} else {
break;
} }
} }
} }
@ -560,9 +600,9 @@ float VehicleAngularVelocity::FilterAngularVelocity(int axis, float data[], int
if (_dynamic_notch_fft_available) { if (_dynamic_notch_fft_available) {
perf_begin(_dynamic_notch_filter_fft_perf); perf_begin(_dynamic_notch_filter_fft_perf);
for (auto &dnf : _dynamic_notch_filter_fft) { for (int peak = MAX_NUM_FFT_PEAKS - 1; peak >= 0; peak--) {
if (dnf[axis].getNotchFreq() > 0.f) { if (_dynamic_notch_filter_fft[axis][peak].getNotchFreq() > 0.f) {
dnf[axis].applyArray(data, N); _dynamic_notch_filter_fft[axis][peak].applyArray(data, N);
} }
} }
@ -589,9 +629,9 @@ float VehicleAngularVelocity::FilterAngularAcceleration(int axis, float dt_s, fl
float angular_acceleration_filtered = 0.f; float angular_acceleration_filtered = 0.f;
for (int n = 0; n < N; n++) { for (int n = 0; n < N; n++) {
const float angular_acceleration = (data[n] - _angular_velocity_prev(axis)) / dt_s; const float angular_acceleration = (data[n] - _angular_velocity_raw_prev(axis)) / dt_s;
angular_acceleration_filtered = _lp_filter_acceleration[axis].apply(angular_acceleration); angular_acceleration_filtered = _lp_filter_acceleration[axis].apply(angular_acceleration);
_angular_velocity_prev(axis) = data[n]; _angular_velocity_raw_prev(axis) = data[n];
} }
return angular_acceleration_filtered; return angular_acceleration_filtered;
@ -622,25 +662,24 @@ void VehicleAngularVelocity::Run()
sensor_gyro_fifo_s sensor_fifo_data; sensor_gyro_fifo_s sensor_fifo_data;
while (_sensor_fifo_sub.update(&sensor_fifo_data)) { while (_sensor_fifo_sub.update(&sensor_fifo_data)) {
const float dt_s = sensor_fifo_data.dt * 1e-6f; const float dt_s = math::constrain(sensor_fifo_data.dt * 1e-6f, 0.00002f, 0.02f); // 20 us to 20 ms
_timestamp_sample_last = sensor_fifo_data.timestamp_sample;
// in FIFO mode the unscaled raw data is filtered, reset filters on any scale change // in FIFO mode the unscaled raw data is filtered, reset filters on any scale change
if (_reset_filters || (fabsf(sensor_fifo_data.scale - _last_scale) > FLT_EPSILON)) { if (_reset_filters) {
ResetFilters(sensor_fifo_data.scale); ResetFilters();
if (_reset_filters) { if (_reset_filters) {
continue; // not safe to run until filters configured continue; // not safe to run until filters configured
} }
} }
UpdateDynamicNotchEscRpm();
UpdateDynamicNotchFFT();
const int N = sensor_fifo_data.samples; const int N = sensor_fifo_data.samples;
static constexpr int FIFO_SIZE_MAX = sizeof(sensor_fifo_data.x) / sizeof(sensor_fifo_data.x[0]); static constexpr int FIFO_SIZE_MAX = sizeof(sensor_fifo_data.x) / sizeof(sensor_fifo_data.x[0]);
if ((N > 0) && (N <= FIFO_SIZE_MAX)) { if ((N > 0) && (N <= FIFO_SIZE_MAX)) {
UpdateDynamicNotchEscRpm();
UpdateDynamicNotchFFT();
Vector3f angular_velocity_unscaled; Vector3f angular_velocity_unscaled;
Vector3f angular_acceleration_unscaled; Vector3f angular_acceleration_unscaled;
@ -651,7 +690,7 @@ void VehicleAngularVelocity::Run()
float data[FIFO_SIZE_MAX]; float data[FIFO_SIZE_MAX];
for (int n = 0; n < N; n++) { for (int n = 0; n < N; n++) {
data[n] = raw_data_array[axis][n]; data[n] = sensor_fifo_data.scale * raw_data_array[axis][n];
} }
// save last filtered sample // save last filtered sample
@ -661,7 +700,7 @@ void VehicleAngularVelocity::Run()
// Publish // Publish
CalibrateAndPublish(!_sensor_fifo_sub.updated(), sensor_fifo_data.timestamp_sample, angular_velocity_unscaled, CalibrateAndPublish(!_sensor_fifo_sub.updated(), sensor_fifo_data.timestamp_sample, angular_velocity_unscaled,
angular_acceleration_unscaled, sensor_fifo_data.scale); angular_acceleration_unscaled);
} }
} }
@ -670,11 +709,11 @@ void VehicleAngularVelocity::Run()
sensor_gyro_s sensor_data; sensor_gyro_s sensor_data;
while (_sensor_sub.update(&sensor_data)) { while (_sensor_sub.update(&sensor_data)) {
if (_timestamp_sample_last == 0) { if (_timestamp_sample_last == 0 || (sensor_data.timestamp_sample <= _timestamp_sample_last)) {
_timestamp_sample_last = sensor_data.timestamp_sample - 1e6f / _filter_sample_rate_hz; _timestamp_sample_last = sensor_data.timestamp_sample - 1e6f / _filter_sample_rate_hz;
} }
const float dt_s = math::constrain(((sensor_data.timestamp_sample - _timestamp_sample_last) / 1e6f), 0.0002f, 0.02f); const float dt_s = math::constrain(((sensor_data.timestamp_sample - _timestamp_sample_last) * 1e-6f), 0.00002f, 0.02f);
_timestamp_sample_last = sensor_data.timestamp_sample; _timestamp_sample_last = sensor_data.timestamp_sample;
if (_reset_filters) { if (_reset_filters) {
@ -709,13 +748,14 @@ void VehicleAngularVelocity::Run()
} }
void VehicleAngularVelocity::CalibrateAndPublish(bool publish, const hrt_abstime &timestamp_sample, void VehicleAngularVelocity::CalibrateAndPublish(bool publish, const hrt_abstime &timestamp_sample,
const Vector3f &angular_velocity_unscaled, const Vector3f &angular_acceleration_unscaled, float scale) const Vector3f &angular_velocity_unscaled, const Vector3f &angular_acceleration_unscaled)
{ {
// Angular velocity: rotate sensor frame to board, scale raw data to SI, apply calibration, and remove in-run estimated bias // Angular velocity: rotate sensor frame to board, scale raw data to SI, apply calibration, and remove in-run estimated bias
_angular_velocity = _calibration.Correct(angular_velocity_unscaled * scale) - _bias; _angular_velocity_prev = _angular_velocity;
_angular_velocity = _calibration.Correct(angular_velocity_unscaled) - _bias;
// Angular acceleration: rotate sensor frame to board, scale raw data to SI, apply any additional configured rotation // Angular acceleration: rotate sensor frame to board, scale raw data to SI, apply any additional configured rotation
_angular_acceleration = _calibration.rotation() * angular_acceleration_unscaled * scale; _angular_acceleration = _calibration.rotation() * angular_acceleration_unscaled;
if (publish && (timestamp_sample >= _last_publish + _publish_interval_min_us)) { if (publish && (timestamp_sample >= _last_publish + _publish_interval_min_us)) {

View File

@ -33,6 +33,7 @@
#pragma once #pragma once
#include <containers/Bitset.hpp>
#include <lib/sensor_calibration/Gyroscope.hpp> #include <lib/sensor_calibration/Gyroscope.hpp>
#include <lib/mathlib/math/Limits.hpp> #include <lib/mathlib/math/Limits.hpp>
#include <lib/matrix/matrix/math.hpp> #include <lib/matrix/matrix/math.hpp>
@ -76,25 +77,25 @@ private:
void Run() override; void Run() override;
void CalibrateAndPublish(bool publish, const hrt_abstime &timestamp_sample, const matrix::Vector3f &angular_velocity, void CalibrateAndPublish(bool publish, const hrt_abstime &timestamp_sample, const matrix::Vector3f &angular_velocity,
const matrix::Vector3f &angular_acceleration, float scale = 1.f); const matrix::Vector3f &angular_acceleration);
float FilterAngularVelocity(int axis, float data[], int N = 1); inline float FilterAngularVelocity(int axis, float data[], int N = 1);
float FilterAngularAcceleration(int axis, float dt_s, float data[], int N = 1); inline float FilterAngularAcceleration(int axis, float dt_s, float data[], int N = 1);
void DisableDynamicNotchEscRpm(); void DisableDynamicNotchEscRpm();
void DisableDynamicNotchFFT(); void DisableDynamicNotchFFT();
void ParametersUpdate(bool force = false); void ParametersUpdate(bool force = false);
void ResetFilters(float new_scale = 1.f); void ResetFilters();
void SensorBiasUpdate(bool force = false); void SensorBiasUpdate(bool force = false);
bool SensorSelectionUpdate(bool force = false); bool SensorSelectionUpdate(bool force = false);
void UpdateDynamicNotchEscRpm(float new_scale = 1.f, bool force = false); void UpdateDynamicNotchEscRpm(bool force = false);
void UpdateDynamicNotchFFT(float new_scale = 1.f, bool force = false); void UpdateDynamicNotchFFT(bool force = false);
bool UpdateSampleRate(); bool UpdateSampleRate();
// scaled appropriately for current FIFO mode // scaled appropriately for current sensor
matrix::Vector3f GetResetAngularVelocity(float new_scale = 1.f) const; matrix::Vector3f GetResetAngularVelocity() const;
matrix::Vector3f GetResetAngularAcceleration(float new_scale = 1.f) const; matrix::Vector3f GetResetAngularAcceleration() const;
static constexpr int MAX_SENSOR_COUNT = 4; static constexpr int MAX_SENSOR_COUNT = 4;
@ -119,9 +120,10 @@ private:
matrix::Vector3f _bias{}; matrix::Vector3f _bias{};
matrix::Vector3f _angular_velocity{}; matrix::Vector3f _angular_velocity{};
matrix::Vector3f _angular_velocity_prev{};
matrix::Vector3f _angular_acceleration{}; matrix::Vector3f _angular_acceleration{};
matrix::Vector3f _angular_velocity_prev{}; matrix::Vector3f _angular_velocity_raw_prev{};
hrt_abstime _timestamp_sample_last{0}; hrt_abstime _timestamp_sample_last{0};
hrt_abstime _publish_interval_min_us{0}; hrt_abstime _publish_interval_min_us{0};
@ -142,14 +144,19 @@ private:
FFT = 2, FFT = 2,
}; };
static constexpr hrt_abstime DYNAMIC_NOTCH_FITLER_TIMEOUT = 1_s;
static constexpr int MAX_NUM_ESC_RPM = sizeof(esc_status_s::esc) / sizeof(esc_status_s::esc[0]); static constexpr int MAX_NUM_ESC_RPM = sizeof(esc_status_s::esc) / sizeof(esc_status_s::esc[0]);
static constexpr int MAX_NUM_ESC_RPM_HARMONICS = 3; static constexpr int MAX_NUM_ESC_RPM_HARMONICS = 3;
static constexpr int MAX_NUM_FFT_PEAKS = sizeof(sensor_gyro_fft_s::peak_frequencies_x) / sizeof( static constexpr int MAX_NUM_FFT_PEAKS = sizeof(sensor_gyro_fft_s::peak_frequencies_x) / sizeof(
sensor_gyro_fft_s::peak_frequencies_x[0]); sensor_gyro_fft_s::peak_frequencies_x[0]);
math::NotchFilter<float> _dynamic_notch_filter_esc_rpm[MAX_NUM_ESC_RPM][MAX_NUM_ESC_RPM_HARMONICS][3] {}; math::NotchFilter<float> _dynamic_notch_filter_esc_rpm[3][MAX_NUM_ESC_RPM][MAX_NUM_ESC_RPM_HARMONICS] {};
math::NotchFilter<float> _dynamic_notch_filter_fft[MAX_NUM_FFT_PEAKS][3] {}; math::NotchFilter<float> _dynamic_notch_filter_fft[3][MAX_NUM_FFT_PEAKS] {};
px4::Bitset<MAX_NUM_ESC_RPM> _esc_available{};
hrt_abstime _last_esc_rpm_notch_update[MAX_NUM_ESC_RPM] {};
perf_counter_t _dynamic_notch_filter_esc_rpm_update_perf{nullptr}; perf_counter_t _dynamic_notch_filter_esc_rpm_update_perf{nullptr};
perf_counter_t _dynamic_notch_filter_fft_update_perf{nullptr}; perf_counter_t _dynamic_notch_filter_fft_update_perf{nullptr};
@ -165,8 +172,6 @@ private:
uint32_t _selected_sensor_device_id{0}; uint32_t _selected_sensor_device_id{0};
float _last_scale{0.f};
bool _reset_filters{true}; bool _reset_filters{true};
bool _fifo_available{false}; bool _fifo_available{false};