ardupilot/libraries/AP_Beacon/AP_Beacon_Marvelmind.cpp

392 lines
16 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/>.
*/
/*
Original C Code by Marvelmind (https://github.com/MarvelmindRobotics/marvelmind.c)
Adapted into Ardupilot by Karthik Desai, Amilcar Lucas
April 2017
*/
#include "AP_Beacon_Marvelmind.h"
#if AP_BEACON_MARVELMIND_ENABLED
#include <AP_HAL/AP_HAL.h>
#include <AP_Math/crc.h>
#define AP_BEACON_MARVELMIND_POSITION_DATAGRAM_ID 0x0001
#define AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_ID 0x0002
#define AP_BEACON_MARVELMIND_DISTANCES_DATAGRAM_ID 0x0004
#define AP_BEACON_MARVELMIND_POSITION_DATAGRAM_HIGHRES_ID 0x0011
#define AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_HIGHRES_ID 0x0012
extern const AP_HAL::HAL& hal;
#define MM_DEBUG_LEVEL 0
#if MM_DEBUG_LEVEL
#include <GCS_MAVLink/GCS.h>
#define Debug(level, fmt, args ...) do { if (level <= MM_DEBUG_LEVEL) { gcs().send_text(MAV_SEVERITY_INFO, fmt, ## args); } } while (0)
#else
#define Debug(level, fmt, args ...)
#endif
void AP_Beacon_Marvelmind::process_position_datagram()
{
hedge.cur_position.address = input_buffer[16];
hedge.cur_position.timestamp = input_buffer[5]
| (((uint32_t) input_buffer[6]) << 8)
| (((uint32_t) input_buffer[7]) << 16)
| (((uint32_t) input_buffer[8]) << 24);
const int16_t vx = input_buffer[9] | (((uint16_t) input_buffer[10]) << 8);
hedge.cur_position.x__mm = vx * 10; // centimeters -> millimeters
const int16_t vy = input_buffer[11] | (((uint16_t) input_buffer[12]) << 8);
hedge.cur_position.y__mm = vy * 10; // centimeters -> millimeters
const int16_t vz = input_buffer[13] | (((uint16_t) input_buffer[14]) << 8);
hedge.cur_position.z__mm = vz * 10; // centimeters -> millimeters
hedge.cur_position.high_resolution = false;
hedge._have_new_values = true;
}
void AP_Beacon_Marvelmind::process_position_highres_datagram()
{
hedge.cur_position.address = input_buffer[22];
hedge.cur_position.timestamp = input_buffer[5]
| (((uint32_t) input_buffer[6]) << 8)
| (((uint32_t) input_buffer[7]) << 16)
| (((uint32_t) input_buffer[8]) << 24);
hedge.cur_position.x__mm = input_buffer[9] | (((uint32_t) input_buffer[10]) << 8)
| (((uint32_t) input_buffer[11]) << 16)
| (((uint32_t) input_buffer[12]) << 24);
hedge.cur_position.y__mm = input_buffer[13] | (((uint32_t) input_buffer[14]) << 8)
| (((uint32_t) input_buffer[15]) << 16)
| (((uint32_t) input_buffer[16]) << 24);
hedge.cur_position.z__mm = input_buffer[17] | (((uint32_t) input_buffer[18]) << 8)
| (((uint32_t) input_buffer[19]) << 16)
| (((uint32_t) input_buffer[20]) << 24);
hedge.cur_position.high_resolution = true;
hedge._have_new_values = true;
}
AP_Beacon_Marvelmind::StationaryBeaconPosition* AP_Beacon_Marvelmind::get_or_alloc_beacon(uint8_t address)
{
const uint8_t n_used = hedge.positions_beacons.num_beacons;
for (uint8_t i = 0; i < n_used; i++) {
if (hedge.positions_beacons.beacons[i].address == address) {
return &hedge.positions_beacons.beacons[i];
}
}
if (n_used >= AP_BEACON_MAX_BEACONS) {
return nullptr;
}
hedge.positions_beacons.num_beacons = (n_used + 1);
return &hedge.positions_beacons.beacons[n_used];
}
void AP_Beacon_Marvelmind::process_beacons_positions_datagram()
{
const uint8_t n = input_buffer[5]; // number of beacons in packet
StationaryBeaconPosition *stationary_beacon;
if ((1 + n * 8) != input_buffer[4]) {
Debug(1, "beacon pos lo pkt size %d != %d", input_buffer[4], (1 + n * 8));
return; // incorrect size
}
for (uint8_t i = 0; i < n; i++) {
const uint8_t ofs = 6 + i * 8;
const uint8_t address = input_buffer[ofs];
const int16_t x = input_buffer[ofs + 1]
| (((uint16_t) input_buffer[ofs + 2]) << 8);
const int16_t y = input_buffer[ofs + 3]
| (((uint16_t) input_buffer[ofs + 4]) << 8);
const int16_t z = input_buffer[ofs + 5]
| (((uint16_t) input_buffer[ofs + 6]) << 8);
stationary_beacon = get_or_alloc_beacon(address);
if (stationary_beacon != nullptr) {
stationary_beacon->address = address; //The instance and the address are the same
stationary_beacon->x__mm = x * 10; // centimeters -> millimeters
stationary_beacon->y__mm = y * 10; // centimeters -> millimeters
stationary_beacon->z__mm = z * 10; // centimeters -> millimeters
stationary_beacon->high_resolution = false;
hedge.positions_beacons.updated = true;
}
}
order_stationary_beacons();
}
void AP_Beacon_Marvelmind::process_beacons_positions_highres_datagram()
{
const uint8_t n = input_buffer[5]; // number of beacons in packet
StationaryBeaconPosition *stationary_beacon;
if ((1 + n * 14) != input_buffer[4]) {
Debug(1, "beacon pos hi pkt size %d != %d", input_buffer[4], (1 + n * 14));
return; // incorrect size
}
for (uint8_t i = 0; i < n; i++) {
const uint8_t ofs = 6 + i * 14;
const uint8_t address = input_buffer[ofs];
const int32_t x = input_buffer[ofs + 1]
| (((uint32_t) input_buffer[ofs + 2]) << 8)
| (((uint32_t) input_buffer[ofs + 3]) << 16)
| (((uint32_t) input_buffer[ofs + 4]) << 24);
const int32_t y = input_buffer[ofs + 5]
| (((uint32_t) input_buffer[ofs + 6]) << 8)
| (((uint32_t) input_buffer[ofs + 7]) << 16)
| (((uint32_t) input_buffer[ofs + 8]) << 24);
const int32_t z = input_buffer[ofs + 9]
| (((uint32_t) input_buffer[ofs + 10]) << 8)
| (((uint32_t) input_buffer[ofs + 11]) << 16)
| (((uint32_t) input_buffer[ofs + 12]) << 24);
stationary_beacon = get_or_alloc_beacon(address);
if (stationary_beacon != nullptr) {
stationary_beacon->address = address; //The instance and the address are the same
stationary_beacon->x__mm = x; // millimeters
stationary_beacon->y__mm = y; // millimeters
stationary_beacon->z__mm = z; // millimeters
stationary_beacon->high_resolution = true;
hedge.positions_beacons.updated = true;
}
}
order_stationary_beacons();
}
void AP_Beacon_Marvelmind::process_beacons_distances_datagram()
{
if (32 != input_buffer[4]) {
Debug(1, "beacon dist pkt size %d != 32", input_buffer[4]);
return; // incorrect size
}
bool set = false;
for (uint8_t i = 0; i < hedge.positions_beacons.num_beacons; i++) {
const uint8_t ofs = 6 + i * 6;
const uint8_t address = input_buffer[ofs];
const int8_t instance = find_beacon_instance(address);
if (instance != -1) {
const uint32_t distance = input_buffer[ofs + 1]
| (((uint32_t) input_buffer[ofs + 2]) << 8)
| (((uint32_t) input_buffer[ofs + 3]) << 16)
| (((uint32_t) input_buffer[ofs + 4]) << 24);
hedge.positions_beacons.beacons[instance].distance__m = distance * 0.001f; // millimeters -> meters
set_beacon_distance(instance, hedge.positions_beacons.beacons[instance].distance__m);
set = true;
Debug(2, "Beacon %d is %.2fm", instance, hedge.positions_beacons.beacons[instance].distance__m);
}
}
if (set) {
last_update_ms = AP_HAL::millis();
}
}
int8_t AP_Beacon_Marvelmind::find_beacon_instance(uint8_t address) const
{
for (uint8_t i = 0; i < hedge.positions_beacons.num_beacons; i++) {
if (hedge.positions_beacons.beacons[i].address == address) {
return i;
}
}
return -1;
}
void AP_Beacon_Marvelmind::update(void)
{
if (uart == nullptr) {
return;
}
// read any available characters
uint16_t num_bytes_read = MIN(uart->available(), 16384U);
while (num_bytes_read-- > 0) {
bool good_byte = false;
if (!uart->read(input_buffer[num_bytes_in_block_received])) {
break;
}
const uint8_t received_char = input_buffer[num_bytes_in_block_received];
switch (parse_state) {
case RECV_HDR:
switch (num_bytes_in_block_received) {
case 0:
good_byte = (received_char == 0xff);
break;
case 1:
good_byte = (received_char == 0x47);
break;
case 2:
good_byte = true;
break;
case 3:
data_id = (((uint16_t)received_char) << 8) + input_buffer[2];
good_byte = (data_id == AP_BEACON_MARVELMIND_POSITION_DATAGRAM_ID)
|| (data_id == AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_ID)
|| (data_id == AP_BEACON_MARVELMIND_DISTANCES_DATAGRAM_ID)
|| (data_id == AP_BEACON_MARVELMIND_POSITION_DATAGRAM_HIGHRES_ID)
|| (data_id == AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_HIGHRES_ID);
break;
case 4: {
switch (data_id) {
case AP_BEACON_MARVELMIND_POSITION_DATAGRAM_ID: {
good_byte = (received_char == 0x10);
break;
}
case AP_BEACON_MARVELMIND_DISTANCES_DATAGRAM_ID:
case AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_ID:
case AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_HIGHRES_ID:
good_byte = true;
break;
case AP_BEACON_MARVELMIND_POSITION_DATAGRAM_HIGHRES_ID: {
good_byte = (received_char == 0x16);
break;
}
}
if (good_byte) {
parse_state = RECV_DGRAM;
}
break;
}
}
if (good_byte) {
// correct header byte
num_bytes_in_block_received++;
} else {
// ...or incorrect
parse_state = RECV_HDR;
num_bytes_in_block_received = 0;
}
break;
case RECV_DGRAM:
num_bytes_in_block_received++;
if (num_bytes_in_block_received >= 7 + input_buffer[4]) {
// parse dgram
uint16_t block_crc = calc_crc_modbus(input_buffer, num_bytes_in_block_received);
if (block_crc == 0) {
switch (data_id) {
case AP_BEACON_MARVELMIND_POSITION_DATAGRAM_ID:
{
// add to position_buffer
process_position_datagram();
vehicle_position_initialized = true;
set_stationary_beacons_positions();
break;
}
case AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_ID:
{
process_beacons_positions_datagram();
beacon_position_initialized = true;
set_stationary_beacons_positions();
break;
}
case AP_BEACON_MARVELMIND_DISTANCES_DATAGRAM_ID:
{
process_beacons_distances_datagram();
break;
}
case AP_BEACON_MARVELMIND_POSITION_DATAGRAM_HIGHRES_ID:
{
process_position_highres_datagram();
vehicle_position_initialized = true;
set_stationary_beacons_positions();
break;
}
case AP_BEACON_MARVELMIND_POSITIONS_DATAGRAM_HIGHRES_ID:
{
process_beacons_positions_highres_datagram();
beacon_position_initialized = true;
set_stationary_beacons_positions();
break;
}
}
}
// and repeat
parse_state = RECV_HDR;
num_bytes_in_block_received = 0;
}
break;
}
}
}
bool AP_Beacon_Marvelmind::healthy()
{
// healthy if we have parsed a message within the past 300ms
return ((AP_HAL::millis() - last_update_ms) < AP_BEACON_TIMEOUT_MS);
}
void AP_Beacon_Marvelmind::set_stationary_beacons_positions()
{
bool set = false;
if (vehicle_position_initialized && beacon_position_initialized) {
if (hedge._have_new_values) {
vehicle_position_NED__m = Vector3f(hedge.cur_position.y__mm * 0.001f,
hedge.cur_position.x__mm * 0.001f,
-hedge.cur_position.z__mm * 0.001f); //Transform Marvelmind ENU to Ardupilot NED
//TODO: Calculate Accuracy of the received signal. Marvelmind *advertises* +/- 2cms
// But we are conservative here and use 20cm instead (until MM provides us with a proper accuracy value)
set_vehicle_position(vehicle_position_NED__m, 0.2f);
set = true;
Debug(2,
"Hedge is at N%.2f, E%.2f, D%.2f",
vehicle_position_NED__m[0],
vehicle_position_NED__m[1],
vehicle_position_NED__m[2]);
}
hedge._have_new_values = false;
for (uint8_t i = 0; i < hedge.positions_beacons.num_beacons; ++i) {
if (hedge.positions_beacons.updated) {
beacon_position_NED__m[i] = Vector3f(hedge.positions_beacons.beacons[i].y__mm * 0.001f,
hedge.positions_beacons.beacons[i].x__mm * 0.001f,
-hedge.positions_beacons.beacons[i].z__mm * 0.001f); //Transform Marvelmind ENU to Ardupilot NED
set_beacon_position(i, beacon_position_NED__m[i]);
set = true;
Debug(2,
"Beacon %d is at N%.2f, E%.2f, D%.2f",
i,
beacon_position_NED__m[i][0],
beacon_position_NED__m[i][1],
beacon_position_NED__m[i][2]);
}
}
hedge.positions_beacons.updated = false;
}
if (set) {
last_update_ms = AP_HAL::millis();
}
}
void AP_Beacon_Marvelmind::order_stationary_beacons()
{
if (hedge.positions_beacons.updated) {
bool swapped = false;
uint8_t j = hedge.positions_beacons.num_beacons;
do
{
swapped = false;
StationaryBeaconPosition beacon_to_swap;
for (uint8_t i = 1; i < j; i++) {
if (hedge.positions_beacons.beacons[i-1].address > hedge.positions_beacons.beacons[i].address) {
beacon_to_swap = hedge.positions_beacons.beacons[i];
hedge.positions_beacons.beacons[i] = hedge.positions_beacons.beacons[i-1];
hedge.positions_beacons.beacons[i-1] = beacon_to_swap;
swapped = true;
}
}
j--;
} while(swapped);
}
}
#endif // AP_BEACON_MARVELMIND_ENABLED