mirror of https://github.com/ArduPilot/ardupilot
136 lines
4.4 KiB
C++
136 lines
4.4 KiB
C++
#include "SmartRTL_test.h"
|
|
#include <AP_AHRS/AP_AHRS.h>
|
|
#include <AP_Baro/AP_Baro.h>
|
|
#include <AP_Compass/AP_Compass.h>
|
|
#include <AP_GPS/AP_GPS.h>
|
|
#include <AP_InertialSensor/AP_InertialSensor.h>
|
|
#include <AP_NavEKF2/AP_NavEKF2.h>
|
|
#include <AP_NavEKF3/AP_NavEKF3.h>
|
|
#include <AP_RangeFinder/AP_RangeFinder.h>
|
|
#include <AP_SerialManager/AP_SerialManager.h>
|
|
|
|
const AP_HAL::HAL &hal = AP_HAL::get_HAL();
|
|
|
|
// INS and Baro declaration
|
|
AP_InertialSensor ins;
|
|
Compass compass;
|
|
AP_GPS gps;
|
|
AP_Baro barometer;
|
|
AP_SerialManager serial_manager;
|
|
|
|
class DummyVehicle {
|
|
public:
|
|
RangeFinder rangefinder {serial_manager, ROTATION_PITCH_270};
|
|
AP_AHRS_NavEKF ahrs{ins, barometer, gps, rangefinder, EKF2, EKF3,
|
|
AP_AHRS_NavEKF::FLAG_ALWAYS_USE_EKF};
|
|
NavEKF2 EKF2{&ahrs, barometer, rangefinder};
|
|
NavEKF3 EKF3{&ahrs, barometer, rangefinder};
|
|
};
|
|
|
|
static DummyVehicle vehicle;
|
|
|
|
AP_AHRS_NavEKF ahrs(vehicle.ahrs);
|
|
AP_SmartRTL smart_rtl {ahrs, true};
|
|
|
|
void setup();
|
|
void loop();
|
|
void reset();
|
|
void check_path(const std::vector<Vector3f> &correct_path, const char* test_name, uint32_t time_us);
|
|
|
|
void setup()
|
|
{
|
|
hal.console->printf("SmartRTL test\n");
|
|
AP_BoardConfig{}.init();
|
|
smart_rtl.init();
|
|
}
|
|
|
|
void loop()
|
|
{
|
|
if (!hal.console->is_initialized()) {
|
|
return;
|
|
}
|
|
uint32_t reference_time, run_time;
|
|
|
|
hal.console->printf("--------------------\n");
|
|
|
|
// reset path and upload "test_path_before" to smart_rtl
|
|
reference_time = AP_HAL::micros();
|
|
reset();
|
|
run_time = AP_HAL::micros() - reference_time;
|
|
|
|
// check path after initial load (no simplification or pruning)
|
|
check_path(test_path_after_adding, "append", run_time);
|
|
|
|
// test simplifications
|
|
reference_time = AP_HAL::micros();
|
|
while (!smart_rtl.request_thorough_cleanup(AP_SmartRTL::THOROUGH_CLEAN_SIMPLIFY_ONLY)) {
|
|
smart_rtl.run_background_cleanup();
|
|
}
|
|
run_time = AP_HAL::micros() - reference_time;
|
|
check_path(test_path_after_simplifying, "simplify", run_time);
|
|
|
|
// test both simplification and pruning
|
|
hal.scheduler->delay(5); // delay 5 milliseconds because request_through_cleanup uses millisecond timestamps
|
|
reset();
|
|
reference_time = AP_HAL::micros();
|
|
while (!smart_rtl.request_thorough_cleanup(AP_SmartRTL::THOROUGH_CLEAN_ALL)) {
|
|
smart_rtl.run_background_cleanup();
|
|
}
|
|
run_time = AP_HAL::micros() - reference_time;
|
|
check_path(test_path_complete, "simplify and pruning", run_time);
|
|
|
|
// delay before next display
|
|
hal.scheduler->delay(5e3); // 5 seconds
|
|
}
|
|
|
|
// reset path (i.e. clear path and add home) and upload "test_path_before" to smart_rtl
|
|
void reset()
|
|
{
|
|
smart_rtl.reset_path(true, Vector3f{0.0f, 0.0f, 0.0f});
|
|
for (Vector3f v : test_path_before) {
|
|
smart_rtl.update(true, v);
|
|
}
|
|
}
|
|
|
|
// compare the vector array passed in with the path held in the smart_rtl object
|
|
void check_path(const std::vector<Vector3f>& correct_path, const char* test_name, uint32_t time_us)
|
|
{
|
|
// check number of points
|
|
bool num_points_match = correct_path.size() == smart_rtl.get_num_points();
|
|
uint16_t points_to_compare = MIN(correct_path.size(), smart_rtl.get_num_points());
|
|
|
|
// check all points match
|
|
bool points_match = true;
|
|
uint16_t failure_index = 0;
|
|
for (uint16_t i = 0; i < points_to_compare; i++) {
|
|
if (smart_rtl.get_point(i) != correct_path[i]) {
|
|
failure_index = i;
|
|
points_match = false;
|
|
}
|
|
}
|
|
|
|
// display overall results
|
|
hal.console->printf("%s: %s time:%u us\n", test_name, (num_points_match && points_match) ? "success" : "fail", time_us);
|
|
|
|
// display number of points
|
|
hal.console->printf(" expected %u points, got %u\n", (unsigned)correct_path.size(), (unsigned)smart_rtl.get_num_points());
|
|
|
|
// display the first failed point and all subsequent points
|
|
if (!points_match) {
|
|
for (uint16_t j = failure_index; j < points_to_compare; j++) {
|
|
const Vector3f& smartrtl_point = smart_rtl.get_point(j);
|
|
hal.console->printf(" expected point %d to be %4.2f,%4.2f,%4.2f, got %4.2f,%4.2f,%4.2f\n",
|
|
(int)j,
|
|
(double)correct_path[j].x,
|
|
(double)correct_path[j].y,
|
|
(double)correct_path[j].z,
|
|
(double)smartrtl_point.x,
|
|
(double)smartrtl_point.y,
|
|
(double)smartrtl_point.z
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
AP_HAL_MAIN();
|