#pragma once /* This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . */ /* * NavEKF based AHRS (Attitude Heading Reference System) interface for * ArduPilot * */ #include #ifndef HAL_NAVEKF2_AVAILABLE // only default to EK2 enabled on boards with over 1M flash #define HAL_NAVEKF2_AVAILABLE (BOARD_FLASH_SIZE>1024) #endif #ifndef HAL_NAVEKF3_AVAILABLE #define HAL_NAVEKF3_AVAILABLE 1 #endif #include "AP_AHRS.h" #if CONFIG_HAL_BOARD == HAL_BOARD_SITL #include #endif #include #include #include #include // definitions shared by inertial and ekf nav filters #include "AP_AHRS_DCM.h" // forward declare view class class AP_AHRS_View; #define AP_AHRS_NAVEKF_SETTLE_TIME_MS 20000 // time in milliseconds the ekf needs to settle after being started #include // fwd declare GSF estimator class EKFGSF_yaw; class AP_AHRS { friend class AP_AHRS_View; public: enum Flags { FLAG_ALWAYS_USE_EKF = 0x1, }; // Constructor AP_AHRS(uint8_t flags = 0); // initialise void init(void); /* Do not allow copies */ CLASS_NO_COPY(AP_AHRS); // get singleton instance static AP_AHRS *get_singleton() { return _singleton; } // allow for runtime change of orientation // this makes initial config easier void update_orientation(); // allow threads to lock against AHRS update HAL_Semaphore &get_semaphore(void) { return _rsem; } // return the smoothed gyro vector corrected for drift const Vector3f &get_gyro(void) const; // return the current drift correction integrator value const Vector3f &get_gyro_drift(void) const; // reset the current gyro drift estimate // should be called if gyro offsets are recalculated void reset_gyro_drift(); void update(bool skip_ins_update=false); void reset(); // dead-reckoning support bool get_position(struct Location &loc) const; // get latest altitude estimate above ground level in meters and validity flag bool get_hagl(float &hagl) const WARN_IF_UNUSED; // status reporting of estimated error float get_error_rp() const; float get_error_yaw() const; /* * wind estimation support */ // enable wind estimation void set_wind_estimation_enabled(bool b) { wind_estimation_enabled = b; } // wind_estimation_enabled returns true if wind estimation is enabled bool get_wind_estimation_enabled() const { return wind_estimation_enabled; } // return a wind estimation vector, in m/s Vector3f wind_estimate() const; // instruct DCM to update its wind estimate: void estimate_wind() { dcm.estimate_wind(); } // return the parameter AHRS_WIND_MAX in metres per second uint8_t get_max_wind() const { return _wind_max; } /* * airspeed support */ // get apparent to true airspeed ratio float get_EAS2TAS(void) const { // FIXME: make this is a method on the active backend return dcm.get_EAS2TAS(); } // return an airspeed estimate if available. return true // if we have an estimate bool airspeed_estimate(float &airspeed_ret) const; // return a true airspeed estimate (navigation airspeed) if // available. return true if we have an estimate bool airspeed_estimate_true(float &airspeed_ret) const; // return estimate of true airspeed vector in body frame in m/s // returns false if estimate is unavailable bool airspeed_vector_true(Vector3f &vec) const; // return true if airspeed comes from an airspeed sensor, as // opposed to an IMU estimate bool airspeed_sensor_enabled(void) const { // FIXME: make this "using_airspeed_sensor"? return dcm.airspeed_sensor_enabled(); } // return true if airspeed comes from a specific airspeed sensor, as // opposed to an IMU estimate bool airspeed_sensor_enabled(uint8_t airspeed_index) const { // FIXME: make this a method on the active backend return dcm.airspeed_sensor_enabled(airspeed_index); } // return a synthetic airspeed estimate (one derived from sensors // other than an actual airspeed sensor), if available. return // true if we have a synthetic airspeed. ret will not be modified // on failure. bool synthetic_airspeed(float &ret) const WARN_IF_UNUSED; // true if compass is being used bool use_compass(); // return the quaternion defining the rotation from NED to XYZ (body) axes bool get_quaternion(Quaternion &quat) const WARN_IF_UNUSED; // return secondary attitude solution if available, as eulers in radians bool get_secondary_attitude(Vector3f &eulers) const; // return secondary attitude solution if available, as quaternion bool get_secondary_quaternion(Quaternion &quat) const; // return secondary position solution if available bool get_secondary_position(struct Location &loc) const; // EKF has a better ground speed vector estimate Vector2f groundspeed_vector(); // return ground speed estimate in meters/second. Used by ground vehicles. float groundspeed(void); const Vector3f &get_accel_ef(uint8_t i) const; const Vector3f &get_accel_ef() const; // Retrieves the corrected NED delta velocity in use by the inertial navigation void getCorrectedDeltaVelocityNED(Vector3f& ret, float& dt) const; // blended accelerometer values in the earth frame in m/s/s const Vector3f &get_accel_ef_blended() const; // set the EKF's origin location in 10e7 degrees. This should only // be called when the EKF has no absolute position reference (i.e. GPS) // from which to decide the origin on its own bool set_origin(const Location &loc) WARN_IF_UNUSED; // returns the inertial navigation origin in lat/lon/alt bool get_origin(Location &ret) const WARN_IF_UNUSED; bool have_inertial_nav() const; bool get_velocity_NED(Vector3f &vec) const; // return the relative position NED to either home or origin // return true if the estimate is valid bool get_relative_position_NED_home(Vector3f &vec) const; bool get_relative_position_NED_origin(Vector3f &vec) const; // return the relative position NE to either home or origin // return true if the estimate is valid bool get_relative_position_NE_home(Vector2f &posNE) const; bool get_relative_position_NE_origin(Vector2f &posNE) const; // return the relative position down to either home or origin // baro will be used for the _home relative one if the EKF isn't void get_relative_position_D_home(float &posD) const; bool get_relative_position_D_origin(float &posD) const; // Get a derivative of the vertical position in m/s which is kinematically consistent with the vertical position is required by some control loops. // This is different to the vertical velocity from the EKF which is not always consistent with the vertical position due to the various errors that are being corrected for. bool get_vert_pos_rate(float &velocity) const; // write optical flow measurements to EKF void writeOptFlowMeas(const uint8_t rawFlowQuality, const Vector2f &rawFlowRates, const Vector2f &rawGyroRates, const uint32_t msecFlowMeas, const Vector3f &posOffset); // write body odometry measurements to the EKF void writeBodyFrameOdom(float quality, const Vector3f &delPos, const Vector3f &delAng, float delTime, uint32_t timeStamp_ms, uint16_t delay_ms, const Vector3f &posOffset); // Writes the default equivalent airspeed and its 1-sigma uncertainty in m/s to be used in forward flight if a measured airspeed is required and not available. void writeDefaultAirSpeed(float airspeed, float uncertainty); // Write position and quaternion data from an external navigation system void writeExtNavData(const Vector3f &pos, const Quaternion &quat, float posErr, float angErr, uint32_t timeStamp_ms, uint16_t delay_ms, uint32_t resetTime_ms); // Write velocity data from an external navigation system void writeExtNavVelData(const Vector3f &vel, float err, uint32_t timeStamp_ms, uint16_t delay_ms); // get speed limit void getControlLimits(float &ekfGndSpdLimit, float &controlScaleXY) const; float getControlScaleZ(void) const; // is the AHRS subsystem healthy? bool healthy() const; // returns false if we fail arming checks, in which case the buffer will be populated with a failure message // requires_position should be true if horizontal position configuration should be checked bool pre_arm_check(bool requires_position, char *failure_msg, uint8_t failure_msg_len) const; // true if the AHRS has completed initialisation bool initialised() const; // return true if *DCM* yaw has been initialised bool dcm_yaw_initialised(void) const { return dcm.yaw_initialised(); } // get_filter_status - returns filter status as a series of flags bool get_filter_status(nav_filter_status &status) const; // get compass offset estimates // true if offsets are valid bool getMagOffsets(uint8_t mag_idx, Vector3f &magOffsets) const; // check all cores providing consistent attitudes for prearm checks bool attitudes_consistent(char *failure_msg, const uint8_t failure_msg_len) const; // return the amount of yaw angle change due to the last yaw angle reset in radians // returns the time of the last yaw angle reset or 0 if no reset has ever occurred uint32_t getLastYawResetAngle(float &yawAng); // return the amount of NE position change in meters due to the last reset // returns the time of the last reset or 0 if no reset has ever occurred uint32_t getLastPosNorthEastReset(Vector2f &pos); // return the amount of NE velocity change in meters/sec due to the last reset // returns the time of the last reset or 0 if no reset has ever occurred uint32_t getLastVelNorthEastReset(Vector2f &vel) const; // return the amount of vertical position change due to the last reset in meters // returns the time of the last reset or 0 if no reset has ever occurred uint32_t getLastPosDownReset(float &posDelta); // Resets the baro so that it reads zero at the current height // Resets the EKF height to zero // Adjusts the EKf origin height so that the EKF height + origin height is the same as before // Returns true if the height datum reset has been performed // If using a range finder for height no reset is performed and it returns false bool resetHeightDatum(); // send a EKF_STATUS_REPORT for current EKF void send_ekf_status_report(mavlink_channel_t chan) const; // get_hgt_ctrl_limit - get maximum height to be observed by the control loops in meters and a validity flag // this is used to limit height during optical flow navigation // it will return invalid when no limiting is required bool get_hgt_ctrl_limit(float &limit) const; // Set to true if the terrain underneath is stable enough to be used as a height reference // this is not related to terrain following void set_terrain_hgt_stable(bool stable); // return the innovations for the specified instance // An out of range instance (eg -1) returns data for the primary instance bool get_innovations(Vector3f &velInnov, Vector3f &posInnov, Vector3f &magInnov, float &tasInnov, float &yawInnov) const; // returns true when the state estimates are significantly degraded by vibration bool is_vibration_affected() const; // get_variances - provides the innovations normalised using the innovation variance where a value of 0 // indicates perfect consistency between the measurement and the EKF solution and a value of of 1 is the maximum // inconsistency that will be accepted by the filter // boolean false is returned if variances are not available bool get_variances(float &velVar, float &posVar, float &hgtVar, Vector3f &magVar, float &tasVar) const; // get a source's velocity innovations // returns true on success and results are placed in innovations and variances arguments bool get_vel_innovations_and_variances_for_source(uint8_t source, Vector3f &innovations, Vector3f &variances) const WARN_IF_UNUSED; // returns the expected NED magnetic field bool get_mag_field_NED(Vector3f& ret) const; // returns the estimated magnetic field offsets in body frame bool get_mag_field_correction(Vector3f &ret) const; // return the index of the airspeed we should use for airspeed measurements // with multiple airspeed sensors and airspeed affinity in EKF3, it is possible to have switched // over to a lane not using the primary airspeed sensor, so AHRS should know which airspeed sensor // to use, i.e, the one being used by the primary lane. A lane switch could have happened due to an // airspeed sensor fault, which makes this even more necessary uint8_t get_active_airspeed_index() const; // return the index of the primary core or -1 if no primary core selected int8_t get_primary_core_index() const; // get the index of the current primary accelerometer sensor uint8_t get_primary_accel_index(void) const; // get the index of the current primary gyro sensor uint8_t get_primary_gyro_index(void) const; // see if EKF lane switching is possible to avoid EKF failsafe void check_lane_switch(void); // request EKF yaw reset to try and avoid the need for an EKF lane switch or failsafe void request_yaw_reset(void); // set position, velocity and yaw sources to either 0=primary, 1=secondary, 2=tertiary void set_posvelyaw_source_set(uint8_t source_set_idx); void Log_Write(); // check if non-compass sensor is providing yaw. Allows compass pre-arm checks to be bypassed bool using_noncompass_for_yaw(void) const; // check if external nav is providing yaw bool using_extnav_for_yaw(void) const; // set and save the ALT_M_NSE parameter value void set_alt_measurement_noise(float noise); // active EKF type for logging uint8_t get_active_AHRS_type(void) const { return uint8_t(active_EKF_type()); } // get the selected ekf type, for allocation decisions int8_t get_ekf_type(void) const { return _ekf_type; } // these are only out here so vehicles can reference them for parameters #if HAL_NAVEKF2_AVAILABLE NavEKF2 EKF2; #endif #if HAL_NAVEKF3_AVAILABLE NavEKF3 EKF3; #endif // for holding parameters static const struct AP_Param::GroupInfo var_info[]; // create a view AP_AHRS_View *create_view(enum Rotation rotation, float pitch_trim_deg=0); // write AOA and SSA information to dataflash logs: void Write_AOA_SSA(void) const; // return AOA float getAOA(void) const { return _AOA; } // return SSA float getSSA(void) const { return _SSA; } /* * trim-related functions */ // get trim const Vector3f &get_trim() const { return _trim.get(); } // set trim void set_trim(const Vector3f &new_trim); // add_trim - adjust the roll and pitch trim up to a total of 10 degrees void add_trim(float roll_in_radians, float pitch_in_radians, bool save_to_eeprom = true); // trim rotation matrices: const Matrix3f& get_rotation_autopilot_body_to_vehicle_body(void) const { return _rotation_autopilot_body_to_vehicle_body; } const Matrix3f& get_rotation_vehicle_body_to_autopilot_body(void) const { return _rotation_vehicle_body_to_autopilot_body; } // Logging functions void Log_Write_Home_And_Origin(); void Write_AHRS2(void) const; void Write_Attitude(const Vector3f &targets) const; void Write_Origin(uint8_t origin_type, const Location &loc) const; void Write_POS(void) const; // return a smoothed and corrected gyro vector in radians/second // using the latest ins data (which may not have been consumed by // the EKF yet) Vector3f get_gyro_latest(void) const; // get yaw rate in earth frame in radians/sec float get_yaw_rate_earth(void) const { return get_gyro() * get_rotation_body_to_ned().c; } /* * home-related functionality */ // get the home location. This is const to prevent any changes to // home without telling AHRS about the change const struct Location &get_home(void) const { return _home; } // functions to handle locking of home. Some vehicles use this to // allow GCS to lock in a home location. void lock_home() { _home_locked = true; } bool home_is_locked() const { return _home_locked; } // returns true if home is set bool home_is_set(void) const { return _home_is_set; } // set the home location in 10e7 degrees. This should be called // when the vehicle is at this position. It is assumed that the // current barometer and GPS altitudes correspond to this altitude bool set_home(const Location &loc) WARN_IF_UNUSED; /* * Attitude-related public methods and attributes: */ // roll/pitch/yaw euler angles, all in radians float roll; float pitch; float yaw; float get_roll() const { return roll; } float get_pitch() const { return pitch; } float get_yaw() const { return yaw; } // helper trig value accessors float cos_roll() const { return _cos_roll; } float cos_pitch() const { return _cos_pitch; } float cos_yaw() const { return _cos_yaw; } float sin_roll() const { return _sin_roll; } float sin_pitch() const { return _sin_pitch; } float sin_yaw() const { return _sin_yaw; } // integer Euler angles (Degrees * 100) int32_t roll_sensor; int32_t pitch_sensor; int32_t yaw_sensor; const Matrix3f &get_rotation_body_to_ned(void) const; // return a Quaternion representing our current attitude in NED frame void get_quat_body_to_ned(Quaternion &quat) const { quat.from_rotation_matrix(get_rotation_body_to_ned()); } // get rotation matrix specifically from DCM backend (used for // compass calibrator) const Matrix3f &get_DCM_rotation_body_to_ned(void) const { return dcm_estimates.dcm_matrix; } // rotate a 2D vector from earth frame to body frame // in result, x is forward, y is right Vector2f earth_to_body2D(const Vector2f &ef_vector) const; // rotate a 2D vector from earth frame to body frame // in input, x is forward, y is right Vector2f body_to_earth2D(const Vector2f &bf) const; // convert a vector from body to earth frame Vector3f body_to_earth(const Vector3f &v) const { return v * get_rotation_body_to_ned(); } // convert a vector from earth to body frame Vector3f earth_to_body(const Vector3f &v) const { return get_rotation_body_to_ned().mul_transpose(v); } /* * methods for the benefit of LUA bindings */ // return current vibration vector for primary IMU Vector3f get_vibration(void) const; // return primary accels, for lua const Vector3f &get_accel(void) const { return AP::ins().get_accel(); } /* * AHRS is used as a transport for vehicle-takeoff-expected and * vehicle-landing-expected: */ void set_takeoff_expected(bool b); bool get_takeoff_expected(void) const { return takeoff_expected; } void set_touchdown_expected(bool b); bool get_touchdown_expected(void) const { return touchdown_expected; } /* * fly_forward is set by the vehicles to indicate the vehicle * should generally be moving in the direction of its heading. * It is an additional piece of information that the backends can * use to provide additional and/or improved estimates. */ void set_fly_forward(bool b) { fly_forward = b; } bool get_fly_forward(void) const { return fly_forward; } /* we modify our behaviour based on what sort of vehicle the * vehicle code tells us we are. This information is also pulled * from AP_AHRS by other libraries */ enum class VehicleClass : uint8_t { UNKNOWN, GROUND, COPTER, FIXED_WING, SUBMARINE, }; VehicleClass get_vehicle_class(void) const { return _vehicle_class; } void set_vehicle_class(VehicleClass vclass) { _vehicle_class = vclass; } // get the view's rotation, or ROTATION_NONE enum Rotation get_view_rotation(void) const; // get access to an EKFGSF_yaw estimator const EKFGSF_yaw *get_yaw_estimator(void) const; private: // optional view class AP_AHRS_View *_view; static AP_AHRS *_singleton; /* we modify our behaviour based on what sort of vehicle the * vehicle code tells us we are. This information is also pulled * from AP_AHRS by other libraries */ VehicleClass _vehicle_class{VehicleClass::UNKNOWN}; // multi-thread access support HAL_Semaphore _rsem; /* * Parameters */ AP_Int8 _wind_max; AP_Int8 _board_orientation; AP_Int8 _ekf_type; AP_Float _custom_roll; AP_Float _custom_pitch; AP_Float _custom_yaw; /* * support for custom AHRS orientation, replacing _board_orientation */ Matrix3f _custom_rotation; /* * DCM-backend parameters; it takes references to these */ // settable parameters AP_Float _kp_yaw; AP_Float _kp; AP_Float gps_gain; AP_Float beta; AP_Enum _gps_use; AP_Int8 _gps_minsats; enum class EKFType { NONE = 0 #if HAL_NAVEKF3_AVAILABLE ,THREE = 3 #endif #if HAL_NAVEKF2_AVAILABLE ,TWO = 2 #endif #if CONFIG_HAL_BOARD == HAL_BOARD_SITL ,SIM = 10 #endif #if HAL_EXTERNAL_AHRS_ENABLED ,EXTERNAL = 11 #endif }; EKFType active_EKF_type(void) const; // if successful returns true and sets secondary_ekf_type to None (for DCM), EKF3 or EKF3 // returns false if no secondary (i.e. only using DCM) bool get_secondary_EKF_type(EKFType &secondary_ekf_type) const; bool always_use_EKF() const { return _ekf_flags & FLAG_ALWAYS_USE_EKF; } /* * Attitude-related private methods and attributes: */ // calculate sin/cos of roll/pitch/yaw from rotation void calc_trig(const Matrix3f &rot, float &cr, float &cp, float &cy, float &sr, float &sp, float &sy) const; // update_trig - recalculates _cos_roll, _cos_pitch, etc based on latest attitude // should be called after _dcm_matrix is updated void update_trig(void); // update roll_sensor, pitch_sensor and yaw_sensor void update_cd_values(void); // helper trig variables float _cos_roll{1.0f}; float _cos_pitch{1.0f}; float _cos_yaw{1.0f}; float _sin_roll; float _sin_pitch; float _sin_yaw; #if HAL_NAVEKF2_AVAILABLE void update_EKF2(void); bool _ekf2_started; #endif #if HAL_NAVEKF3_AVAILABLE bool _ekf3_started; void update_EKF3(void); #endif // rotation from vehicle body to NED frame Matrix3f _dcm_matrix; Vector3f _gyro_drift; Vector3f _gyro_estimate; Vector3f _accel_ef_ekf[INS_MAX_INSTANCES]; Vector3f _accel_ef_ekf_blended; const uint16_t startup_delay_ms = 1000; uint32_t start_time_ms; uint8_t _ekf_flags; // bitmask from Flags enumeration EKFType ekf_type(void) const; void update_DCM(); // get the index of the current primary IMU uint8_t get_primary_IMU_index(void) const; /* * home-related state */ void load_watchdog_home(); bool _checked_watchdog_home; struct Location _home; bool _home_is_set :1; bool _home_locked :1; // avoid setting current state repeatedly across all cores on all EKFs: enum class TriState { False = 0, True = 1, UNKNOWN = 3, }; TriState terrainHgtStableState = TriState::UNKNOWN; /* * private AOA and SSA-related state and methods */ float _AOA, _SSA; uint32_t _last_AOA_update_ms; void update_AOA_SSA(void); EKFType last_active_ekf_type; #if CONFIG_HAL_BOARD == HAL_BOARD_SITL SITL::SIM *_sitl; uint32_t _last_body_odm_update_ms; void update_SITL(void); #endif #if HAL_EXTERNAL_AHRS_ENABLED void update_external(void); #endif /* * trim-related state and private methods: */ // a vector to capture the difference between the controller and body frames AP_Vector3f _trim; // cached trim rotations Vector3f _last_trim; Matrix3f _rotation_autopilot_body_to_vehicle_body; Matrix3f _rotation_vehicle_body_to_autopilot_body; // updates matrices responsible for rotating vectors from vehicle body // frame to autopilot body frame from _trim variables void update_trim_rotation_matrices(); /* * AHRS is used as a transport for vehicle-takeoff-expected and * vehicle-landing-expected: */ // update takeoff/touchdown flags void update_flags(); bool takeoff_expected; // true if the vehicle is in a state that takeoff might be expected. Ground effect may be in play. uint32_t takeoff_expected_start_ms; bool touchdown_expected; // true if the vehicle is in a state that touchdown might be expected. Ground effect may be in play. uint32_t touchdown_expected_start_ms; /* * wind estimation support */ bool wind_estimation_enabled; /* * fly_forward is set by the vehicles to indicate the vehicle * should generally be moving in the direction of its heading. * It is an additional piece of information that the backends can * use to provide additional and/or improved estimates. */ bool fly_forward; // true if we can assume the vehicle will be flying forward on its X axis // poke AP_Notify based on values from status void update_notify_from_filter_status(const nav_filter_status &status); /* * backends (and their results) */ AP_AHRS_DCM dcm{_kp_yaw, _kp, gps_gain, beta, _gps_use, _gps_minsats}; struct AP_AHRS_Backend::Estimates dcm_estimates; /* * copy results from a backend over AP_AHRS canonical results. * This updates member variables like roll and pitch, as well as * updating derived values like sin_roll and sin_pitch. */ void copy_estimates_from_backend_estimates(const AP_AHRS_Backend::Estimates &results); #if HAL_NMEA_OUTPUT_ENABLED class AP_NMEA_Output* _nmea_out; #endif }; namespace AP { AP_AHRS &ahrs(); };