mirror of https://github.com/ArduPilot/ardupilot
108 lines
5.2 KiB
C++
108 lines
5.2 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/>.
|
|
*/
|
|
#pragma once
|
|
|
|
#include <AP_Common/AP_Common.h>
|
|
#include <AP_HAL/AP_HAL.h>
|
|
#include "AP_Proximity.h"
|
|
#include <AP_Common/Location.h>
|
|
|
|
#define PROXIMITY_NUM_SECTORS 8 // number of sectors
|
|
#define PROXIMITY_SECTOR_WIDTH_DEG 45.0f // width of sectors in degrees
|
|
#define PROXIMITY_BOUNDARY_DIST_MIN 0.6f // minimum distance for a boundary point. This ensures the object avoidance code doesn't think we are outside the boundary.
|
|
#define PROXIMITY_BOUNDARY_DIST_DEFAULT 100 // if we have no data for a sector, boundary is placed 100m out
|
|
|
|
class AP_Proximity_Backend
|
|
{
|
|
public:
|
|
// constructor. This incorporates initialisation as well.
|
|
AP_Proximity_Backend(AP_Proximity &_frontend, AP_Proximity::Proximity_State &_state);
|
|
|
|
// we declare a virtual destructor so that Proximity drivers can
|
|
// override with a custom destructor if need be
|
|
virtual ~AP_Proximity_Backend(void) {}
|
|
|
|
// update the state structure
|
|
virtual void update() = 0;
|
|
|
|
// get maximum and minimum distances (in meters) of sensor
|
|
virtual float distance_max() const = 0;
|
|
virtual float distance_min() const = 0;
|
|
|
|
// get distance upwards in meters. returns true on success
|
|
virtual bool get_upward_distance(float &distance) const { return false; }
|
|
|
|
// handle mavlink DISTANCE_SENSOR messages
|
|
virtual void handle_msg(const mavlink_message_t &msg) {}
|
|
|
|
// get boundary points around vehicle for use by avoidance
|
|
// returns nullptr and sets num_points to zero if no boundary can be returned
|
|
const Vector2f* get_boundary_points(uint16_t& num_points) const;
|
|
|
|
// get distance and angle to closest object (used for pre-arm check)
|
|
// returns true on success, false if no valid readings
|
|
bool get_closest_object(float& angle_deg, float &distance) const;
|
|
|
|
// get number of objects, angle and distance - used for non-GPS avoidance
|
|
uint8_t get_object_count() const;
|
|
bool get_object_angle_and_distance(uint8_t object_number, float& angle_deg, float &distance) const;
|
|
|
|
// get distances in 8 directions. used for sending distances to ground station
|
|
bool get_horizontal_distances(AP_Proximity::Proximity_Distance_Array &prx_dist_array) const;
|
|
|
|
protected:
|
|
|
|
// set status and update valid_count
|
|
void set_status(AP_Proximity::Status status);
|
|
|
|
// correct an angle (in degrees) based on the orientation and yaw correction parameters
|
|
float correct_angle_for_orientation(float angle_degrees) const;
|
|
|
|
// find which sector a given angle falls into
|
|
uint8_t convert_angle_to_sector(float angle_degrees) const;
|
|
|
|
// initialise the boundary and sector_edge_vector array used for object avoidance
|
|
// should be called if the sector_middle_deg or _setor_width_deg arrays are changed
|
|
void init_boundary();
|
|
|
|
// update boundary points used for object avoidance based on a single sector's distance changing
|
|
// the boundary points lie on the line between sectors meaning two boundary points may be updated based on a single sector's distance changing
|
|
// the boundary point is set to the shortest distance found in the two adjacent sectors, this is a conservative boundary around the vehicle
|
|
void update_boundary_for_sector(const uint8_t sector, const bool push_to_OA_DB);
|
|
|
|
// check if a reading should be ignored because it falls into an ignore area
|
|
// angles should be in degrees and in the range of 0 to 360
|
|
bool ignore_reading(uint16_t angle_deg) const;
|
|
|
|
// database helpers. all angles are in degrees
|
|
bool database_prepare_for_push(Vector3f ¤t_pos, Matrix3f &body_to_ned);
|
|
void database_push(float angle, float distance);
|
|
void database_push(float angle, float distance, uint32_t timestamp_ms, const Vector3f ¤t_pos, const Matrix3f &body_to_ned);
|
|
AP_Proximity &frontend;
|
|
AP_Proximity::Proximity_State &state; // reference to this instances state
|
|
|
|
// sectors
|
|
const uint16_t _sector_middle_deg[PROXIMITY_NUM_SECTORS] = {0, 45, 90, 135, 180, 225, 270, 315}; // middle angle of each sector
|
|
|
|
// sensor data
|
|
float _angle[PROXIMITY_NUM_SECTORS]; // angle to closest object within each sector
|
|
float _distance[PROXIMITY_NUM_SECTORS]; // distance to closest object within each sector
|
|
bool _distance_valid[PROXIMITY_NUM_SECTORS]; // true if a valid distance received for each sector
|
|
|
|
// fence boundary
|
|
Vector2f _sector_edge_vector[PROXIMITY_NUM_SECTORS]; // vector for right-edge of each sector, used to speed up calculation of boundary
|
|
Vector2f _boundary_point[PROXIMITY_NUM_SECTORS]; // bounding polygon around the vehicle calculated conservatively for object avoidance
|
|
};
|