mirror of https://github.com/ArduPilot/ardupilot
97 lines
2.9 KiB
C++
97 lines
2.9 KiB
C++
#include "Copter.h"
|
|
|
|
#if MODE_BRAKE_ENABLED == ENABLED
|
|
|
|
/*
|
|
* Init and run calls for brake flight mode
|
|
*/
|
|
|
|
// brake_init - initialise brake controller
|
|
bool ModeBrake::init(bool ignore_checks)
|
|
{
|
|
// set target to current position
|
|
init_target();
|
|
|
|
// initialize vertical speed and acceleration
|
|
pos_control->set_max_speed_z(BRAKE_MODE_SPEED_Z, BRAKE_MODE_SPEED_Z);
|
|
pos_control->set_max_accel_z(BRAKE_MODE_DECEL_RATE);
|
|
|
|
// initialise position and desired velocity
|
|
if (!pos_control->is_active_z()) {
|
|
pos_control->set_alt_target_to_current_alt();
|
|
pos_control->set_desired_velocity_z(inertial_nav.get_velocity_z());
|
|
}
|
|
|
|
_timeout_ms = 0;
|
|
|
|
return true;
|
|
}
|
|
|
|
// brake_run - runs the brake controller
|
|
// should be called at 100hz or more
|
|
void ModeBrake::run()
|
|
{
|
|
// if not armed set throttle to zero and exit immediately
|
|
if (is_disarmed_or_landed()) {
|
|
make_safe_spool_down();
|
|
init_target();
|
|
return;
|
|
}
|
|
|
|
// set motors to full range
|
|
motors->set_desired_spool_state(AP_Motors::DesiredSpoolState::THROTTLE_UNLIMITED);
|
|
|
|
// relax stop target if we might be landed
|
|
if (copter.ap.land_complete_maybe) {
|
|
loiter_nav->soften_for_landing();
|
|
}
|
|
|
|
// use position controller to stop
|
|
pos_control->set_desired_velocity_xy(0.0f, 0.0f);
|
|
pos_control->update_xy_controller();
|
|
|
|
// call attitude controller
|
|
attitude_control->input_thrust_vector_rate_heading(wp_nav->get_thrust_vector(), 0.0f);
|
|
|
|
// update altitude target and call position controller
|
|
// protects heli's from inflight motor interlock disable
|
|
if (motors->get_desired_spool_state() == AP_Motors::DesiredSpoolState::GROUND_IDLE && !copter.ap.land_complete) {
|
|
pos_control->set_alt_target_from_climb_rate(-abs(g.land_speed), G_Dt, false);
|
|
} else {
|
|
pos_control->set_alt_target_from_climb_rate_ff(0.0f, G_Dt, false);
|
|
}
|
|
pos_control->update_z_controller();
|
|
|
|
if (_timeout_ms != 0 && millis()-_timeout_start >= _timeout_ms) {
|
|
if (!copter.set_mode(Mode::Number::LOITER, ModeReason::BRAKE_TIMEOUT)) {
|
|
copter.set_mode(Mode::Number::ALT_HOLD, ModeReason::BRAKE_TIMEOUT);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ModeBrake::timeout_to_loiter_ms(uint32_t timeout_ms)
|
|
{
|
|
_timeout_start = millis();
|
|
_timeout_ms = timeout_ms;
|
|
}
|
|
|
|
void ModeBrake::init_target()
|
|
{
|
|
// initialise position controller
|
|
pos_control->set_desired_velocity_xy(0.0f,0.0f);
|
|
pos_control->set_desired_accel_xy(0.0f,0.0f);
|
|
pos_control->init_xy_controller();
|
|
|
|
// initialise pos controller speed and acceleration
|
|
pos_control->set_max_speed_xy(inertial_nav.get_velocity().length());
|
|
pos_control->set_max_accel_xy(BRAKE_MODE_DECEL_RATE);
|
|
pos_control->calc_leash_length_xy();
|
|
|
|
// set target position
|
|
Vector3f stopping_point;
|
|
pos_control->get_stopping_point_xy(stopping_point);
|
|
pos_control->set_xy_target(stopping_point.x, stopping_point.y);
|
|
}
|
|
|
|
#endif
|