ardupilot/libraries/AP_VisualOdom/AP_VisualOdom.cpp

253 lines
7.3 KiB
C++

/*
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 <http://www.gnu.org/licenses/>.
*/
#include "AP_VisualOdom.h"
#if HAL_VISUALODOM_ENABLED
#include "AP_VisualOdom_Backend.h"
#include "AP_VisualOdom_MAV.h"
#include "AP_VisualOdom_IntelT265.h"
#include <AP_AHRS/AP_AHRS.h>
#include <AP_Logger/AP_Logger.h>
extern const AP_HAL::HAL &hal;
// table of user settable parameters
const AP_Param::GroupInfo AP_VisualOdom::var_info[] = {
// @Param: _TYPE
// @DisplayName: Visual odometry camera connection type
// @Description: Visual odometry camera connection type
// @Values: 0:None,1:MAVLink,2:IntelT265
// @User: Advanced
// @RebootRequired: True
AP_GROUPINFO_FLAGS("_TYPE", 0, AP_VisualOdom, _type, 0, AP_PARAM_FLAG_ENABLE),
// @Param: _POS_X
// @DisplayName: Visual odometry camera X position offset
// @Description: X position of the camera in body frame. Positive X is forward of the origin.
// @Units: m
// @Range: -5 5
// @Increment: 0.01
// @User: Advanced
// @Param: _POS_Y
// @DisplayName: Visual odometry camera Y position offset
// @Description: Y position of the camera in body frame. Positive Y is to the right of the origin.
// @Units: m
// @Range: -5 5
// @Increment: 0.01
// @User: Advanced
// @Param: _POS_Z
// @DisplayName: Visual odometry camera Z position offset
// @Description: Z position of the camera in body frame. Positive Z is down from the origin.
// @Units: m
// @Range: -5 5
// @Increment: 0.01
// @User: Advanced
AP_GROUPINFO("_POS", 1, AP_VisualOdom, _pos_offset, 0.0f),
// @Param: _ORIENT
// @DisplayName: Visual odometery camera orientation
// @Description: Visual odometery camera orientation
// @Values: 0:Forward, 2:Right, 4:Back, 6:Left, 24:Up, 25:Down
// @User: Advanced
AP_GROUPINFO("_ORIENT", 2, AP_VisualOdom, _orientation, ROTATION_NONE),
// @Param: _SCALE
// @DisplayName: Visual odometry scaling factor
// @Description: Visual odometry scaling factor applied to position estimates from sensor
// @User: Advanced
AP_GROUPINFO("_SCALE", 3, AP_VisualOdom, _pos_scale, 1.0f),
// @Param: _DELAY_MS
// @DisplayName: Visual odometry sensor delay
// @Description: Visual odometry sensor delay relative to inertial measurements
// @Units: ms
// @Range: 0 250
// @User: Advanced
AP_GROUPINFO("_DELAY_MS", 4, AP_VisualOdom, _delay_ms, 10),
// @Param: _VEL_M_NSE
// @DisplayName: Visual odometry velocity measurement noise
// @Description: Visual odometry velocity measurement noise in m/s
// @Units: m/s
// @Range: 0.05 5.0
// @User: Advanced
AP_GROUPINFO("_VEL_M_NSE", 5, AP_VisualOdom, _vel_noise, 0.1),
AP_GROUPEND
};
AP_VisualOdom::AP_VisualOdom()
{
AP_Param::setup_object_defaults(this, var_info);
#if CONFIG_HAL_BOARD == HAL_BOARD_SITL
if (_singleton != nullptr) {
AP_HAL::panic("must be singleton");
}
#endif
_singleton = this;
}
// detect and initialise any sensors
void AP_VisualOdom::init()
{
// create backend
switch (_type) {
case AP_VisualOdom_Type_None:
// do nothing
break;
case AP_VisualOdom_Type_MAV:
_driver = new AP_VisualOdom_MAV(*this);
break;
case AP_VisualOdom_Type_IntelT265:
_driver = new AP_VisualOdom_IntelT265(*this);
break;
}
}
// return true if sensor is enabled
bool AP_VisualOdom::enabled() const
{
return ((_type != AP_VisualOdom_Type_None));
}
// return true if sensor is basically healthy (we are receiving data)
bool AP_VisualOdom::healthy() const
{
if (!enabled()) {
return false;
}
if (_driver == nullptr) {
return false;
}
return _driver->healthy();
}
// consume vision_position_delta mavlink messages
void AP_VisualOdom::handle_vision_position_delta_msg(const mavlink_message_t &msg)
{
// exit immediately if not enabled
if (!enabled()) {
return;
}
// call backend
if (_driver != nullptr) {
_driver->handle_vision_position_delta_msg(msg);
}
}
// general purpose method to consume position estimate data and send to EKF
// distances in meters, roll, pitch and yaw are in radians
void AP_VisualOdom::handle_vision_position_estimate(uint64_t remote_time_us, uint32_t time_ms, float x, float y, float z, float roll, float pitch, float yaw, uint8_t reset_counter)
{
// exit immediately if not enabled
if (!enabled()) {
return;
}
// call backend
if (_driver != nullptr) {
// convert attitude to quaternion and call backend
Quaternion attitude;
attitude.from_euler(roll, pitch, yaw);
_driver->handle_vision_position_estimate(remote_time_us, time_ms, x, y, z, attitude, reset_counter);
}
}
// general purpose method to consume position estimate data and send to EKF
void AP_VisualOdom::handle_vision_position_estimate(uint64_t remote_time_us, uint32_t time_ms, float x, float y, float z, const Quaternion &attitude, uint8_t reset_counter)
{
// exit immediately if not enabled
if (!enabled()) {
return;
}
// call backend
if (_driver != nullptr) {
_driver->handle_vision_position_estimate(remote_time_us, time_ms, x, y, z, attitude, reset_counter);
}
}
void AP_VisualOdom::handle_vision_speed_estimate(uint64_t remote_time_us, uint32_t time_ms, const Vector3f &vel, uint8_t reset_counter)
{
// exit immediately if not enabled
if (!enabled()) {
return;
}
// call backend
if (_driver != nullptr) {
_driver->handle_vision_speed_estimate(remote_time_us, time_ms, vel, reset_counter);
}
}
// calibrate camera attitude to align with vehicle's AHRS/EKF attitude
void AP_VisualOdom::align_sensor_to_vehicle()
{
// exit immediately if not enabled
if (!enabled()) {
return;
}
// call backend
if (_driver != nullptr) {
_driver->align_sensor_to_vehicle();
}
}
// returns false if we fail arming checks, in which case the buffer will be populated with a failure message
bool AP_VisualOdom::pre_arm_check(char *failure_msg, uint8_t failure_msg_len) const
{
// exit immediately if not enabled
if (!enabled()) {
return true;
}
// check healthy
if (!healthy()) {
hal.util->snprintf(failure_msg, failure_msg_len, "not healthy");
return false;
}
// if no backend we must have failed to create because out of memory
if (_driver == nullptr) {
hal.util->snprintf(failure_msg, failure_msg_len, "out of memory");
return false;
}
// call backend specific arming check
return _driver->pre_arm_check(failure_msg, failure_msg_len);
}
// singleton instance
AP_VisualOdom *AP_VisualOdom::_singleton;
namespace AP {
AP_VisualOdom *visualodom()
{
return AP_VisualOdom::get_singleton();
}
}
#endif