ardupilot/ArduCopter/system.pde

646 lines
18 KiB
Plaintext
Raw Normal View History

// -*- tab-width: 4; Mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*-
/*****************************************************************************
2012-08-21 23:19:50 -03:00
* The init_ardupilot function processes everything we need for an in - air restart
* We will determine later if we are actually on the ground and process a
* ground start in that case.
*
*****************************************************************************/
#if CLI_ENABLED == ENABLED
// Functions called from the top-level menu
2012-08-21 23:19:50 -03:00
static int8_t process_logs(uint8_t argc, const Menu::arg *argv); // in Log.pde
static int8_t setup_mode(uint8_t argc, const Menu::arg *argv); // in setup.pde
static int8_t test_mode(uint8_t argc, const Menu::arg *argv); // in test.cpp
static int8_t reboot_board(uint8_t argc, const Menu::arg *argv);
// This is the help function
// PSTR is an AVR macro to read strings from flash memory
// printf_P is a version of print_f that reads from flash memory
2012-08-21 23:19:50 -03:00
static int8_t main_menu_help(uint8_t argc, const Menu::arg *argv)
{
cliSerial->printf_P(PSTR("Commands:\n"
2012-08-21 23:19:50 -03:00
" logs\n"
" setup\n"
" test\n"
" reboot\n"
2012-08-21 23:19:50 -03:00
"\n"));
return(0);
}
// Command/function table for the top-level menu.
const struct Menu::command main_menu_commands[] PROGMEM = {
// command function called
// ======= ===============
2012-08-21 23:19:50 -03:00
{"logs", process_logs},
{"setup", setup_mode},
{"test", test_mode},
{"reboot", reboot_board},
2012-08-21 23:19:50 -03:00
{"help", main_menu_help},
};
// Create the top-level menu object.
MENU(main_menu, THISFIRMWARE, main_menu_commands);
static int8_t reboot_board(uint8_t argc, const Menu::arg *argv)
{
reboot_apm();
return 0;
}
// the user wants the CLI. It never exits
static void run_cli(AP_HAL::UARTDriver *port)
{
cliSerial = port;
Menu::set_port(port);
port->set_blocking_writes(true);
// disable the mavlink delay callback
hal.scheduler->register_delay_callback(NULL, 5);
// disable main_loop failsafe
failsafe_disable();
// cut the engines
if(motors.armed()) {
motors.armed(false);
motors.output();
}
while (1) {
main_menu.run();
}
}
#endif // CLI_ENABLED
static void init_ardupilot()
{
#if USB_MUX_PIN > 0
2011-11-25 19:11:25 -04:00
// on the APM2 board we have a mux thet switches UART0 between
// USB and the board header. If the right ArduPPM firmware is
// installed we can detect if USB is connected using the
// USB_MUX_PIN
pinMode(USB_MUX_PIN, INPUT);
ap_system.usb_connected = !digitalRead(USB_MUX_PIN);
if (!ap_system.usb_connected) {
// USB is not connected, this means UART0 may be a Xbee, with
// its darned bricking problem. We can't write to it for at
// least one second after powering up. Simplest solution for
// now is to delay for 1 second. Something more elegant may be
// added later
delay(1000);
}
#endif
2012-08-21 23:19:50 -03:00
// Console serial port
//
// The console port buffers are defined to be sufficiently large to support
// the MAVLink protocol efficiently
//
#if HIL_MODE != HIL_MODE_DISABLED
// we need more memory for HIL, as we get a much higher packet rate
hal.uartA->begin(SERIAL0_BAUD, 256, 256);
#else
// use a bit less for non-HIL operation
hal.uartA->begin(SERIAL0_BAUD, 128, 128);
#endif
2012-08-21 23:19:50 -03:00
// GPS serial port.
//
#if GPS_PROTOCOL != GPS_PROTOCOL_IMU
// standard gps running. Note that we need a 256 byte buffer for some
// GPS types (eg. UBLOX)
2012-12-13 15:48:01 -04:00
hal.uartB->begin(38400, 256, 16);
2012-08-21 23:19:50 -03:00
#endif
cliSerial->printf_P(PSTR("\n\nInit " THISFIRMWARE
2012-08-21 23:19:50 -03:00
"\n\nFree RAM: %u\n"),
memcheck_available_memory());
2012-08-21 23:19:50 -03:00
//
// Report firmware version code expect on console (check of actual EEPROM format version is done in load_parameters function)
2012-08-21 23:19:50 -03:00
//
report_version();
2012-08-21 23:19:50 -03:00
// setup IO pins
pinMode(A_LED_PIN, OUTPUT); // GPS status LED
digitalWrite(A_LED_PIN, LED_OFF);
2012-08-21 23:19:50 -03:00
pinMode(B_LED_PIN, OUTPUT); // GPS status LED
digitalWrite(B_LED_PIN, LED_OFF);
2012-08-21 23:19:50 -03:00
pinMode(C_LED_PIN, OUTPUT); // GPS status LED
digitalWrite(C_LED_PIN, LED_OFF);
2012-12-13 15:48:01 -04:00
relay.init();
#if COPTER_LEDS == ENABLED
copter_leds_init();
#endif
2012-02-12 07:27:51 -04:00
// load parameters from EEPROM
load_parameters();
2013-05-25 00:21:29 -03:00
#if HIL_MODE != HIL_MODE_ATTITUDE
barometer.init();
#endif
2012-08-21 23:19:50 -03:00
// init the GCS
2012-12-13 15:48:01 -04:00
gcs0.init(hal.uartA);
// Register the mavlink service callback. This will run
// anytime there are more than 5ms remaining in a call to
// hal.scheduler->delay.
hal.scheduler->register_delay_callback(mavlink_delay_cb, 5);
#if USB_MUX_PIN > 0
if (!ap_system.usb_connected) {
// we are not connected via USB, re-init UART0 with right
// baud rate
hal.uartA->begin(map_baudrate(g.serial3_baud, SERIAL3_BAUD));
}
#else
// we have a 2nd serial port for telemetry
hal.uartC->begin(map_baudrate(g.serial3_baud, SERIAL3_BAUD), 128, 128);
2012-12-13 15:48:01 -04:00
gcs3.init(hal.uartC);
#endif
// identify ourselves correctly with the ground station
2012-08-21 23:19:50 -03:00
mavlink_system.sysid = g.sysid_this_mav;
mavlink_system.type = 2; //MAV_QUADROTOR;
2012-02-11 02:25:24 -04:00
2011-12-28 00:53:05 -04:00
#if LOGGING_ENABLED == ENABLED
DataFlash.Init();
if (!DataFlash.CardInserted()) {
gcs_send_text_P(SEVERITY_LOW, PSTR("No dataflash inserted"));
g.log_bitmask.set(0);
} else if (DataFlash.NeedErase()) {
gcs_send_text_P(SEVERITY_LOW, PSTR("ERASING LOGS"));
2012-08-21 23:19:50 -03:00
do_erase_logs();
gcs0.reset_cli_timeout();
2012-08-21 23:19:50 -03:00
}
2011-12-28 00:53:05 -04:00
#endif
2012-08-21 23:19:50 -03:00
#if FRAME_CONFIG == HELI_FRAME
motors.servo_manual = false;
motors.init_swash(); // heli initialisation
#endif
2012-08-21 23:19:50 -03:00
init_rc_in(); // sets up rc channels from radio
init_rc_out(); // sets up motors and output to escs
/*
* setup the 'main loop is dead' check. Note that this relies on
* the RC library being initialised.
*/
2012-12-13 15:48:01 -04:00
hal.scheduler->register_timer_failsafe(failsafe_check, 1000);
2011-11-12 22:47:54 -04:00
#if HIL_MODE != HIL_MODE_ATTITUDE
2012-08-21 23:19:50 -03:00
#if CONFIG_ADC == ENABLED
// begin filtering the ADC Gyros
2012-12-13 15:48:01 -04:00
adc.Init(); // APM ADC library initialization
2012-08-21 23:19:50 -03:00
#endif // CONFIG_ADC
#endif // HIL_MODE
2012-08-21 23:19:50 -03:00
// Do GPS init
g_gps = &g_gps_driver;
// GPS Initialization
2012-12-17 22:36:57 -04:00
g_gps->init(hal.uartB, GPS::GPS_ENGINE_AIRBORNE_1G);
2012-08-21 23:19:50 -03:00
if(g.compass_enabled)
init_compass();
2012-08-21 23:19:50 -03:00
// init the optical flow sensor
if(g.optflow_enabled) {
init_optflow();
}
// initialise inertial nav
inertial_nav.init();
2011-10-18 03:51:47 -03:00
#ifdef USERHOOK_INIT
2012-08-21 23:19:50 -03:00
USERHOOK_INIT
2011-10-18 03:51:47 -03:00
#endif
2013-03-18 02:07:04 -03:00
#if CLI_ENABLED == ENABLED
const prog_char_t *msg = PSTR("\nPress ENTER 3 times to start interactive setup\n");
cliSerial->println_P(msg);
#if USB_MUX_PIN == 0
2012-12-13 15:48:01 -04:00
hal.uartC->println_P(msg);
#endif
#endif // CLI_ENABLED
#if HIL_MODE != HIL_MODE_DISABLED
while (!barometer.healthy) {
// the barometer becomes healthy when we get the first
// HIL_STATE message
gcs_send_text_P(SEVERITY_LOW, PSTR("Waiting for first HIL_STATE message"));
delay(1000);
}
#endif
2012-08-21 23:19:50 -03:00
#if HIL_MODE != HIL_MODE_ATTITUDE
// read Baro pressure at ground
//-----------------------------
init_barometer();
#endif
// initialise sonar
#if CONFIG_SONAR == ENABLED
init_sonar();
#endif
#if FRAME_CONFIG == HELI_FRAME
2013-05-25 00:21:29 -03:00
// initialise controller filters
init_rate_controllers();
#endif // HELI_FRAME
2012-08-21 23:19:50 -03:00
// initialize commands
// -------------------
init_commands();
// set the correct flight mode
// ---------------------------
reset_control_switch();
startup_ground();
#if LOGGING_ENABLED == ENABLED
2012-08-21 23:19:50 -03:00
Log_Write_Startup();
#endif
2012-12-13 15:48:01 -04:00
cliSerial->print_P(PSTR("\nReady to FLY "));
}
2012-12-13 15:48:01 -04:00
//******************************************************************************
//This function does all the calibrations, etc. that we need during a ground start
2012-12-13 15:48:01 -04:00
//******************************************************************************
static void startup_ground(void)
{
2012-08-21 23:19:50 -03:00
gcs_send_text_P(SEVERITY_LOW,PSTR("GROUND START"));
2013-01-13 01:04:04 -04:00
// initialise ahrs (may push imu calibration into the mpu6000 if using that device).
ahrs.init();
2012-08-21 23:19:50 -03:00
// Warm up and read Gyro offsets
// -----------------------------
2012-12-14 00:12:39 -04:00
ins.init(AP_InertialSensor::COLD_START,
ins_sample_rate,
2012-12-13 15:48:01 -04:00
flash_leds);
2012-08-21 23:19:50 -03:00
#if CLI_ENABLED == ENABLED
report_ins();
2012-08-21 23:19:50 -03:00
#endif
// setup fast AHRS gains to get right attitude
ahrs.set_fast_gains(true);
#if SECONDARY_DMP_ENABLED == ENABLED
ahrs2.init(&timer_scheduler);
ahrs2.set_as_secondary(true);
ahrs2.set_fast_gains(true);
#endif
2012-08-21 23:19:50 -03:00
// reset the leds
// ---------------------------
clear_leds();
// when we re-calibrate the gyros,
// all previous I values are invalid
reset_I_all();
}
// returns true or false whether mode requires GPS
static bool mode_requires_GPS(uint8_t mode) {
switch(mode) {
case AUTO:
case GUIDED:
case LOITER:
case RTL:
case CIRCLE:
case POSITION:
return true;
default:
return false;
}
return false;
}
// set_mode - change flight mode and perform any necessary initialisation
static void set_mode(uint8_t mode)
2012-08-21 23:19:50 -03:00
{
// Switch to stabilize mode if requested mode requires a GPS lock
if(g_gps->status() != GPS::GPS_OK_FIX_3D && mode_requires_GPS(mode)) {
mode = STABILIZE;
} else if(mode == RTL && !ap.home_is_set) {
2012-08-21 23:19:50 -03:00
mode = STABILIZE;
}
// Switch to stabilize if OF_LOITER requested but no optical flow sensor
if (mode == OF_LOITER && !g.optflow_enabled ) {
mode = STABILIZE;
2012-08-21 23:19:50 -03:00
}
2012-12-14 00:12:39 -04:00
control_mode = mode;
2013-04-21 10:44:57 -03:00
control_mode = constrain_int16(control_mode, 0, NUM_MODES - 1);
2012-08-21 23:19:50 -03:00
// if we change modes, we must clear landed flag
2012-11-10 01:55:51 -04:00
set_land_complete(false);
2012-08-21 23:19:50 -03:00
// report the GPS and Motor arming status
led_mode = NORMAL_LEDS;
switch(control_mode)
{
case ACRO:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = true;
ap.manual_attitude = true;
set_yaw_mode(ACRO_YAW);
set_roll_pitch_mode(ACRO_RP);
set_throttle_mode(ACRO_THR);
set_nav_mode(NAV_NONE);
// reset acro axis targets to current attitude
2012-11-10 01:55:51 -04:00
if(g.axis_enabled){
2012-10-27 18:13:46 -03:00
roll_axis = ahrs.roll_sensor;
pitch_axis = ahrs.pitch_sensor;
nav_yaw = ahrs.yaw_sensor;
}
2012-08-21 23:19:50 -03:00
break;
case STABILIZE:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = true;
ap.manual_attitude = true;
set_yaw_mode(YAW_HOLD);
set_roll_pitch_mode(ROLL_PITCH_STABLE);
set_throttle_mode(THROTTLE_MANUAL_TILT_COMPENSATED);
set_nav_mode(NAV_NONE);
2012-08-21 23:19:50 -03:00
break;
case ALT_HOLD:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = true;
set_yaw_mode(ALT_HOLD_YAW);
set_roll_pitch_mode(ALT_HOLD_RP);
set_throttle_mode(ALT_HOLD_THR);
set_nav_mode(NAV_NONE);
2012-08-21 23:19:50 -03:00
break;
case AUTO:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
// roll-pitch, throttle and yaw modes will all be set by the first nav command
init_commands(); // clear the command queues. will be reloaded when "run_autopilot" calls "update_commands" function
2012-08-21 23:19:50 -03:00
break;
case CIRCLE:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
set_roll_pitch_mode(CIRCLE_RP);
set_throttle_mode(CIRCLE_THR);
set_nav_mode(CIRCLE_NAV);
set_yaw_mode(CIRCLE_YAW);
2012-08-21 23:19:50 -03:00
break;
case LOITER:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
set_yaw_mode(LOITER_YAW);
set_roll_pitch_mode(LOITER_RP);
set_throttle_mode(LOITER_THR);
set_nav_mode(LOITER_NAV);
2012-08-21 23:19:50 -03:00
break;
case POSITION:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = true;
ap.manual_attitude = false;
set_yaw_mode(POSITION_YAW);
set_roll_pitch_mode(POSITION_RP);
set_throttle_mode(POSITION_THR);
set_nav_mode(POSITION_NAV);
wp_nav.clear_angle_limit(); // ensure there are no left over angle limits from throttle controller. To-Do: move this to the exit routine of throttle controller
2012-08-21 23:19:50 -03:00
break;
case GUIDED:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
set_yaw_mode(get_wp_yaw_mode(false));
set_roll_pitch_mode(GUIDED_RP);
set_throttle_mode(GUIDED_THR);
set_nav_mode(GUIDED_NAV);
2012-08-21 23:19:50 -03:00
break;
case LAND:
// To-Do: it is messy to set manual_attitude here because the do_land function is reponsible for setting the roll_pitch_mode
if( ap.home_is_set && g_gps->status() == GPS::GPS_OK_FIX_3D ) {
// switch to loiter if we have gps
ap.manual_attitude = false;
}else{
// otherwise remain with stabilize roll and pitch
ap.manual_attitude = true;
}
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
do_land(NULL); // land at current location
2012-08-21 23:19:50 -03:00
break;
case RTL:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
do_RTL();
2012-08-21 23:19:50 -03:00
break;
case OF_LOITER:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = false;
set_yaw_mode(OF_LOITER_YAW);
set_roll_pitch_mode(OF_LOITER_RP);
set_throttle_mode(OF_LOITER_THR);
set_nav_mode(OF_LOITER_NAV);
2012-08-21 23:19:50 -03:00
break;
// THOR
// These are the flight modes for Toy mode
// See the defines for the enumerated values
case TOY_A:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = true;
set_yaw_mode(YAW_TOY);
set_roll_pitch_mode(ROLL_PITCH_TOY);
set_throttle_mode(THROTTLE_AUTO);
set_nav_mode(NAV_NONE);
2012-08-21 23:19:50 -03:00
// save throttle for fast exit of Alt hold
saved_toy_throttle = g.rc_3.control_in;
break;
case TOY_M:
2012-11-10 01:55:51 -04:00
ap.manual_throttle = false;
ap.manual_attitude = true;
set_yaw_mode(YAW_TOY);
set_roll_pitch_mode(ROLL_PITCH_TOY);
set_nav_mode(NAV_NONE);
set_throttle_mode(THROTTLE_HOLD);
2012-08-21 23:19:50 -03:00
break;
default:
break;
}
2012-11-10 01:55:51 -04:00
if(ap.manual_attitude) {
2012-08-21 23:19:50 -03:00
// We are under manual attitude control
// remove the navigation from roll and pitch command
reset_nav_params();
}
Log_Write_Mode(control_mode);
}
static void
init_simple_bearing()
{
2012-08-21 23:19:50 -03:00
initial_simple_bearing = ahrs.yaw_sensor;
if (g.log_bitmask != 0) {
Log_Write_Data(DATA_INIT_SIMPLE_BEARING, initial_simple_bearing);
}
}
// update_auto_armed - update status of auto_armed flag
static void update_auto_armed()
{
// disarm checks
if(ap.auto_armed){
// if motors are disarmed, auto_armed should also be false
if(!motors.armed()) {
set_auto_armed(false);
return;
}
// if in stabilize or acro flight mode and throttle is zero, auto-armed should become false
if(control_mode <= ACRO && g.rc_3.control_in == 0 && !ap.failsafe_radio) {
set_auto_armed(false);
}
}else{
// arm checks
// if motors are armed and throttle is above zero auto_armed should be true
if(motors.armed() && g.rc_3.control_in != 0) {
set_auto_armed(true);
}
}
}
/*
2012-08-21 23:19:50 -03:00
* map from a 8 bit EEPROM baud rate to a real baud rate
*/
static uint32_t map_baudrate(int8_t rate, uint32_t default_baud)
{
switch (rate) {
case 1: return 1200;
case 2: return 2400;
case 4: return 4800;
case 9: return 9600;
case 19: return 19200;
case 38: return 38400;
case 57: return 57600;
case 111: return 111100;
case 115: return 115200;
}
//cliSerial->println_P(PSTR("Invalid SERIAL3_BAUD"));
return default_baud;
}
#if USB_MUX_PIN > 0
static void check_usb_mux(void)
{
bool usb_check = !digitalRead(USB_MUX_PIN);
if (usb_check == ap_system.usb_connected) {
return;
}
// the user has switched to/from the telemetry port
ap_system.usb_connected = usb_check;
if (ap_system.usb_connected) {
hal.uartA->begin(SERIAL0_BAUD);
} else {
hal.uartA->begin(map_baudrate(g.serial3_baud, SERIAL3_BAUD));
}
}
#endif
/*
2012-08-21 23:19:50 -03:00
* called by gyro/accel init to flash LEDs so user
* has some mesmerising lights to watch while waiting
*/
void flash_leds(bool on)
{
2012-08-21 23:19:50 -03:00
digitalWrite(A_LED_PIN, on ? LED_OFF : LED_ON);
digitalWrite(C_LED_PIN, on ? LED_ON : LED_OFF);
}
2012-03-01 08:36:51 -04:00
/*
* Read Vcc vs 1.1v internal reference
*/
uint16_t board_voltage(void)
{
2012-12-13 15:48:01 -04:00
return board_vcc_analog_source->read_latest();
2012-03-01 08:36:51 -04:00
}
/*
force a software reset of the APM
*/
static void reboot_apm(void) {
hal.scheduler->reboot();
}
//
// print_flight_mode - prints flight mode to serial port.
//
static void
2013-04-20 02:18:22 -03:00
print_flight_mode(AP_HAL::BetterStream *port, uint8_t mode)
{
switch (mode) {
case STABILIZE:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("STABILIZE"));
break;
case ACRO:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("ACRO"));
break;
case ALT_HOLD:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("ALT_HOLD"));
break;
case AUTO:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("AUTO"));
break;
case GUIDED:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("GUIDED"));
break;
case LOITER:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("LOITER"));
break;
case RTL:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("RTL"));
break;
case CIRCLE:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("CIRCLE"));
break;
case POSITION:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("POSITION"));
break;
case LAND:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("LAND"));
break;
case OF_LOITER:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("OF_LOITER"));
break;
case TOY_M:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("TOY_M"));
break;
case TOY_A:
2013-04-20 02:18:22 -03:00
port->print_P(PSTR("TOY_A"));
break;
default:
2013-04-20 02:18:22 -03:00
port->printf_P(PSTR("Mode(%u)"), (unsigned)mode);
break;
}
}