Mirror
/
ardupilot
mirror of https://github.com/ArduPilot/ardupilot
5
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
ardupilot/libraries/AC_Avoidance/AP_OADatabase.cpp

491 lines
16 KiB
C++
Raw Blame History

/*
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 "AC_Avoidance_config.h"
#if AP_OADATABASE_ENABLED
#include "AP_OADatabase.h"
#include <AP_AHRS/AP_AHRS.h>
#include <GCS_MAVLink/GCS.h>
#include <AP_Math/AP_Math.h>
#include <AP_Vehicle/AP_Vehicle_Type.h>
extern const AP_HAL::HAL& hal;
#ifndef AP_OADATABASE_TIMEOUT_SECONDS_DEFAULT
#define AP_OADATABASE_TIMEOUT_SECONDS_DEFAULT 10
#endif
#ifndef AP_OADATABASE_SIZE_DEFAULT
#define AP_OADATABASE_SIZE_DEFAULT 100
#endif
#ifndef AP_OADATABASE_QUEUE_SIZE_DEFAULT
#define AP_OADATABASE_QUEUE_SIZE_DEFAULT 80
#endif
#ifndef AP_OADATABASE_DISTANCE_FROM_HOME
#define AP_OADATABASE_DISTANCE_FROM_HOME 3
#endif
const AP_Param::GroupInfo AP_OADatabase::var_info[] = {
// @Param: SIZE
// @DisplayName: OADatabase maximum number of points
// @Description: OADatabase maximum number of points. Set to 0 to disable the OA Database. Larger means more points but is more cpu intensive to process
// @Range: 0 10000
// @User: Advanced
// @RebootRequired: True
AP_GROUPINFO("SIZE", 1, AP_OADatabase, _database_size_param, AP_OADATABASE_SIZE_DEFAULT),
// @Param: EXPIRE
// @DisplayName: OADatabase item timeout
// @Description: OADatabase item timeout. The time an item will linger without any updates before it expires. Zero means never expires which is useful for a sent-once static environment but terrible for dynamic ones.
// @Units: s
// @Range: 0 127
// @Increment: 1
// @User: Advanced
AP_GROUPINFO("EXPIRE", 2, AP_OADatabase, _database_expiry_seconds, AP_OADATABASE_TIMEOUT_SECONDS_DEFAULT),
// @Param: QUEUE_SIZE
// @DisplayName: OADatabase queue maximum number of points
// @Description: OADatabase queue maximum number of points. This in an input buffer size. Larger means it can handle larger bursts of incoming data points to filter into the database. No impact on cpu, only RAM. Recommend larger for faster datalinks or for sensors that generate a lot of data.
// @Range: 1 200
// @User: Advanced
// @RebootRequired: True
AP_GROUPINFO("QUEUE_SIZE", 3, AP_OADatabase, _queue_size_param, AP_OADATABASE_QUEUE_SIZE_DEFAULT),
// @Param: OUTPUT
// @DisplayName: OADatabase output level
// @Description: OADatabase output level to configure which database objects are sent to the ground station. All data is always available internally for avoidance algorithms.
// @Values: 0:Disabled,1:Send only HIGH importance items,2:Send HIGH and NORMAL importance items,3:Send all items
// @User: Advanced
AP_GROUPINFO("OUTPUT", 4, AP_OADatabase, _output_level, (float)OutputLevel::HIGH),
// @Param: BEAM_WIDTH
// @DisplayName: OADatabase beam width
// @Description: Beam width of incoming lidar data
// @Units: deg
// @Range: 1 10
// @User: Advanced
// @RebootRequired: True
AP_GROUPINFO("BEAM_WIDTH", 5, AP_OADatabase, _beam_width, 5.0f),
// @Param: RADIUS_MIN
// @DisplayName: OADatabase Minimum radius
// @Description: Minimum radius of objects held in database
// @Units: m
// @Range: 0 10
// @User: Advanced
AP_GROUPINFO("RADIUS_MIN", 6, AP_OADatabase, _radius_min, 0.01f),
// @Param: DIST_MAX
// @DisplayName: OADatabase Distance Maximum
// @Description: Maximum distance of objects held in database. Set to zero to disable the limits
// @Units: m
// @Range: 0 10
// @User: Advanced
AP_GROUPINFO("DIST_MAX", 7, AP_OADatabase, _dist_max, 0.0f),
// @Param{Copter}: ALT_MIN
// @DisplayName: OADatabase minimum altitude above home before storing obstacles
// @Description: OADatabase will reject obstacles if vehicle's altitude above home is below this parameter, in a 3 meter radius around home. Set 0 to disable this feature.
// @Units: m
// @Range: 0 4
// @User: Advanced
AP_GROUPINFO_FRAME("ALT_MIN", 8, AP_OADatabase, _min_alt, 0.0f, AP_PARAM_FRAME_COPTER | AP_PARAM_FRAME_HELI | AP_PARAM_FRAME_TRICOPTER),
AP_GROUPEND
};
AP_OADatabase::AP_OADatabase()
{
if (_singleton != nullptr) {
AP_HAL::panic("AP_OADatabase must be singleton");
}
_singleton = this;
AP_Param::setup_object_defaults(this, var_info);
}
void AP_OADatabase::init()
{
init_database();
init_queue();
// initialise scalar using beam width of at least 1deg
dist_to_radius_scalar = tanf(radians(MAX(_beam_width, 1.0f)));
if (!healthy()) {
GCS_SEND_TEXT(MAV_SEVERITY_INFO, "DB init failed . Sizes queue:%u, db:%u", (unsigned int)_queue.size, (unsigned int)_database.size);
delete _queue.items;
delete[] _database.items;
return;
}
}
void AP_OADatabase::update()
{
if (!healthy()) {
return;
}
process_queue();
database_items_remove_all_expired();
}
// push a location into the database
void AP_OADatabase::queue_push(const Vector3f &pos, uint32_t timestamp_ms, float distance)
{
if (!healthy()) {
return;
}
// check if this obstacle needs to be rejected from DB because of low altitude near home
#if APM_BUILD_COPTER_OR_HELI
if (!is_zero(_min_alt)) {
Vector3f current_pos;
if (!AP::ahrs().get_relative_position_NED_home(current_pos)) {
// we do not know where the vehicle is
return;
}
if (current_pos.xy().length() < AP_OADATABASE_DISTANCE_FROM_HOME) {
// vehicle is within a small radius of home
if (-current_pos.z < _min_alt) {
// vehicle is below the minimum alt
return;
}
}
}
#endif
// ignore objects that are far away
if ((_dist_max > 0.0f) && (distance > _dist_max)) {
return;
}
const OA_DbItem item = {pos, timestamp_ms, MAX(_radius_min, distance * dist_to_radius_scalar), 0, AP_OADatabase::OA_DbItemImportance::Normal};
{
WITH_SEMAPHORE(_queue.sem);
_queue.items->push(item);
}
}
void AP_OADatabase::init_queue()
{
_queue.size = _queue_size_param;
if (_queue.size == 0) {
return;
}
_queue.items = NEW_NOTHROW ObjectBuffer<OA_DbItem>(_queue.size);
if (_queue.items != nullptr && _queue.items->get_size() == 0) {
// allocation failed
delete _queue.items;
_queue.items = nullptr;
}
}
void AP_OADatabase::init_database()
{
_database.size = _database_size_param;
if (_database_size_param == 0) {
return;
}
_database.items = NEW_NOTHROW OA_DbItem[_database.size];
}
// get bitmask of gcs channels item should be sent to based on its importance
// returns 0xFF (send to all channels) if should be sent, 0 if it should not be sent
uint8_t AP_OADatabase::get_send_to_gcs_flags(const OA_DbItemImportance importance)
{
switch (importance) {
case OA_DbItemImportance::Low:
if (_output_level >= OutputLevel::ALL) {
return 0xFF;
}
break;
case OA_DbItemImportance::Normal:
if (_output_level >= OutputLevel::HIGH_AND_NORMAL) {
return 0xFF;
}
break;
case OA_DbItemImportance::High:
if (_output_level >= OutputLevel::HIGH) {
return 0xFF;
}
break;
}
return 0x0;
}
// returns true when there's more work in the queue to do
bool AP_OADatabase::process_queue()
{
if (!healthy()) {
return false;
}
// processing queue by moving those entries into the database
// Using a for with fixed size is better than while(!empty) because the
// while could get us stuck here longer than expected if we're getting
// a lot of values pushing into it while we're trying to empty it. With
// the for we know we will exit at an expected time
const uint16_t queue_available = MIN(_queue.items->available(), 100U);
if (queue_available == 0) {
return false;
}
for (uint16_t queue_index=0; queue_index<queue_available; queue_index++) {
OA_DbItem item;
bool pop_success;
{
WITH_SEMAPHORE(_queue.sem);
pop_success = _queue.items->pop(item);
}
if (!pop_success) {
return false;
}
item.send_to_gcs = get_send_to_gcs_flags(item.importance);
// compare item to all items in database. If found a similar item, update the existing, else add it as a new one
bool found = false;
for (uint16_t i=0; i<_database.count; i++) {
if (is_close_to_item_in_database(i, item)) {
database_item_refresh(i, item.timestamp_ms, item.radius);
found = true;
break;
}
}
if (!found) {
database_item_add(item);
}
}
return (_queue.items->available() > 0);
}
void AP_OADatabase::database_item_add(const OA_DbItem &item)
{
if (_database.count >= _database.size) {
return;
}
_database.items[_database.count] = item;
_database.items[_database.count].send_to_gcs = get_send_to_gcs_flags(_database.items[_database.count].importance);
_database.count++;
}
void AP_OADatabase::database_item_remove(const uint16_t index)
{
if (index >= _database.count || _database.count == 0) {
// index out of range
return;
}
// radius of 0 tells the GCS we don't care about it any more (aka it expired)
_database.items[index].radius = 0;
_database.items[index].send_to_gcs = get_send_to_gcs_flags(_database.items[index].importance);
_database.count--;
if (_database.count == 0) {
return;
}
if (index != _database.count) {
// copy last object in array over expired object
_database.items[index] = _database.items[_database.count];
_database.items[index].send_to_gcs = get_send_to_gcs_flags(_database.items[index].importance);
}
}
void AP_OADatabase::database_item_refresh(const uint16_t index, const uint32_t timestamp_ms, const float radius)
{
if (index >= _database.count) {
// index out of range
return;
}
const bool is_different =
(!is_equal(_database.items[index].radius, radius)) ||
(timestamp_ms - _database.items[index].timestamp_ms >= 500);
if (is_different) {
// update timestamp and radius on close object so it stays around longer
// and trigger resending to GCS
_database.items[index].timestamp_ms = timestamp_ms;
_database.items[index].radius = radius;
_database.items[index].send_to_gcs = get_send_to_gcs_flags(_database.items[index].importance);
}
}
void AP_OADatabase::database_items_remove_all_expired()
{
// calculate age of all items in the _database
if (_database_expiry_seconds <= 0) {
// zero means never expire. This is not normal behavior but perhaps you could send a static
// environment once that you don't want to have to constantly update
return;
}
const uint32_t now_ms = AP_HAL::millis();
const uint32_t expiry_ms = (uint32_t)_database_expiry_seconds * 1000;
uint16_t index = 0;
while (index < _database.count) {
if (now_ms - _database.items[index].timestamp_ms > expiry_ms) {
database_item_remove(index);
} else {
index++;
}
}
}
// returns true if a similar object already exists in database. When true, the object timer is also reset
bool AP_OADatabase::is_close_to_item_in_database(const uint16_t index, const OA_DbItem &item) const
{
if (index >= _database.count) {
// index out of range
return false;
}
const float distance_sq = (_database.items[index].pos - item.pos).length_squared();
return ((distance_sq < sq(item.radius)) || (distance_sq < sq(_database.items[index].radius)));
}
#if HAL_GCS_ENABLED
// send ADSB_VEHICLE mavlink messages
void AP_OADatabase::send_adsb_vehicle(mavlink_channel_t chan, uint16_t interval_ms)
{
// ensure database's send_to_gcs field is large enough
static_assert(MAVLINK_COMM_NUM_BUFFERS <= sizeof(OA_DbItem::send_to_gcs) * 8,
"AP_OADatabase's OA_DBItem.send_to_gcs bitmask must be large enough to hold MAVLINK_COMM_NUM_BUFFERS");
if ((_output_level <= OutputLevel::NONE) || !healthy()) {
return;
}
const uint8_t chan_as_bitmask = 1 << chan;
const char callsign[9] = "OA_DB";
// calculate how many messages we should send
const uint32_t now_ms = AP_HAL::millis();
uint16_t num_to_send = 1;
uint16_t num_sent = 0;
if ((_last_send_to_gcs_ms[chan] != 0) && (interval_ms > 0)) {
uint32_t diff_ms = now_ms - _last_send_to_gcs_ms[chan];
num_to_send = MAX(diff_ms / interval_ms, 1U);
}
_last_send_to_gcs_ms[chan] = now_ms;
// send unsent objects until output buffer is full or have sent enough
for (uint16_t i=0; i < _database.count; i++) {
if (!HAVE_PAYLOAD_SPACE(chan, ADSB_VEHICLE) || (num_sent >= num_to_send)) {
// all done for now
return;
}
const uint16_t idx = _next_index_to_send[chan];
// prepare to send next object
_next_index_to_send[chan]++;
if (_next_index_to_send[chan] >= _database.count) {
_next_index_to_send[chan] = 0;
}
if ((_database.items[idx].send_to_gcs & chan_as_bitmask) == 0) {
continue;
}
// convert object's position as an offset from EKF origin to Location
const Location item_loc(Vector3f(_database.items[idx].pos.x * 100.0f, _database.items[idx].pos.y * 100.0f, _database.items[idx].pos.z * 100.0f), Location::AltFrame::ABOVE_ORIGIN);
mavlink_msg_adsb_vehicle_send(chan,
idx,
item_loc.lat,
item_loc.lng,
0, // altitude_type
item_loc.alt,
0, // heading
0, // hor_velocity
0, // ver_velocity
callsign, // callsign
255, // emitter_type
0, // tslc
0, // flags
(uint16_t)(_database.items[idx].radius * 100.f)); // squawk
// unmark item for sending to gcs
_database.items[idx].send_to_gcs &= ~chan_as_bitmask;
// update highest index sent to GCS
_highest_index_sent[chan] = MAX(idx, _highest_index_sent[chan]);
// update count sent
num_sent++;
}
// clear expired items in case the database size shrank
while (_highest_index_sent[chan] > _database.count) {
if (!HAVE_PAYLOAD_SPACE(chan, ADSB_VEHICLE) || (num_sent >= num_to_send)) {
// all done for now
return;
}
const uint16_t idx = _highest_index_sent[chan];
_highest_index_sent[chan]--;
if (_database.items[idx].importance != OA_DbItemImportance::High) {
continue;
}
mavlink_msg_adsb_vehicle_send(chan,
idx, // id
0, // latitude
0, // longitude
0, // altitude_type
0, // altitude
0, // heading
0, // hor_velocity
0, // ver_velocity
callsign, // callsign
255, // emitter_type
0, // tslc
0, // flags
0); // squawk
// update count sent
num_sent++;
}
}
#endif // HAL_GCS_ENABLED
// singleton instance
AP_OADatabase *AP_OADatabase::_singleton;
namespace AP {
AP_OADatabase *oadatabase()
{
return AP_OADatabase::get_singleton();
}
}
#endif // AP_OADATABASE_ENABLED
Reference in New Issue View Git Blame Copy Permalink