UAS.cc 81.7 KB
Newer Older
1
/*===================================================================
pixhawk's avatar
pixhawk committed
2 3 4 5 6 7 8 9 10 11 12 13 14
======================================================================*/

/**
 * @file
 *   @brief Represents one unmanned aerial vehicle
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QList>
#include <QMessageBox>
#include <QTimer>
15
#include <QSettings>
pixhawk's avatar
pixhawk committed
16 17 18
#include <iostream>
#include <QDebug>
#include <cmath>
19
#include <qmath.h>
pixhawk's avatar
pixhawk committed
20 21 22
#include "UAS.h"
#include "LinkInterface.h"
#include "UASManager.h"
23
#include "QGC.h"
pixhawk's avatar
pixhawk committed
24
#include "GAudioOutput.h"
25
#include "MAVLinkProtocol.h"
pixhawk's avatar
pixhawk committed
26
#include "QGCMAVLink.h"
27 28
#include "LinkManager.h"
#include "SerialLink.h"
pixhawk's avatar
pixhawk committed
29

30
UAS::UAS(MAVLinkProtocol* protocol, int id) : UASInterface(),
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
    uasId(id),
    startTime(QGC::groundTimeMilliseconds()),
    commStatus(COMM_DISCONNECTED),
    name(""),
    autopilot(-1),
    links(new QList<LinkInterface*>()),
    unknownPackets(),
    mavlink(protocol),
    waypointManager(*this),
    thrustSum(0),
    thrustMax(10),
    startVoltage(0),
    warnVoltage(9.5f),
    warnLevelPercent(20.0f),
    currentVoltage(12.0f),
    lpVoltage(12.0f),
    batteryRemainingEstimateEnabled(false),
    mode(-1),
    status(-1),
    navMode(-1),
    onboardTimeOffset(0),
    controlRollManual(true),
    controlPitchManual(true),
    controlYawManual(true),
    controlThrustManual(true),
    manualRollAngle(0),
    manualPitchAngle(0),
    manualYawAngle(0),
    manualThrust(0),
    receiveDropRate(0),
    sendDropRate(0),
    lowBattAlarm(false),
    positionLock(false),
    localX(0.0),
    localY(0.0),
    localZ(0.0),
    latitude(0.0),
    longitude(0.0),
    altitude(0.0),
    roll(0.0),
    pitch(0.0),
    yaw(0.0),
    statusTimeout(new QTimer(this)),
    paramsOnceRequested(false),
    airframe(0),
    attitudeKnown(false),
    paramManager(NULL)
pixhawk's avatar
pixhawk committed
78
{
79
    color = UASInterface::getNextColor();
pixhawk's avatar
pixhawk committed
80
    setBattery(LIPOLY, 3);
81
    connect(statusTimeout, SIGNAL(timeout()), this, SLOT(updateState()));
82
    connect(this, SIGNAL(systemSpecsChanged(int)), this, SLOT(writeSettings()));
83
    statusTimeout->start(500);
84
    readSettings();
pixhawk's avatar
pixhawk committed
85 86 87 88
}

UAS::~UAS()
{
89
    writeSettings();
pixhawk's avatar
pixhawk committed
90
    delete links;
91
    links=NULL;
pixhawk's avatar
pixhawk committed
92 93
}

94 95 96 97 98
void UAS::writeSettings()
{
    QSettings settings;
    settings.beginGroup(QString("MAV%1").arg(uasId));
    settings.setValue("NAME", this->name);
99 100
    settings.setValue("AIRFRAME", this->airframe);
    settings.setValue("AP_TYPE", this->autopilot);
101
    settings.setValue("BATTERY_SPECS", getBatterySpecs());
102 103 104 105 106 107 108 109 110
    settings.endGroup();
    settings.sync();
}

void UAS::readSettings()
{
    QSettings settings;
    settings.beginGroup(QString("MAV%1").arg(uasId));
    this->name = settings.value("NAME", this->name).toString();
111 112
    this->airframe = settings.value("AIRFRAME", this->airframe).toInt();
    this->autopilot = settings.value("AP_TYPE", this->autopilot).toInt();
113
    if (settings.contains("BATTERY_SPECS")) {
114 115
        setBatterySpecs(settings.value("BATTERY_SPECS").toString());
    }
116 117 118
    settings.endGroup();
}

119
int UAS::getUASID() const
pixhawk's avatar
pixhawk committed
120 121 122 123
{
    return uasId;
}

124 125
void UAS::updateState()
{
126 127
    // Check if heartbeat timed out
    quint64 heartbeatInterval = QGC::groundTimeUsecs() - lastHeartbeat;
128
    if (heartbeatInterval > timeoutIntervalHeartbeat) {
129 130 131 132
        emit heartbeatTimeout(heartbeatInterval);
        emit heartbeatTimeout();
    }

133 134
    // Position lock is set by the MAVLink message handler
    // if no position lock is available, indicate an error
135
    if (positionLock) {
136
        positionLock = false;
137 138
    } else {
        if (mode > (uint8_t)MAV_MODE_LOCKED && positionLock) {
139 140 141 142 143
            GAudioOutput::instance()->notifyNegative();
        }
    }
}

pixhawk's avatar
pixhawk committed
144 145
void UAS::setSelected()
{
146
    if (UASManager::instance()->getActiveUAS() != this) {
147 148 149 150 151 152 153 154
        UASManager::instance()->setActiveUAS(this);
        emit systemSelected(true);
    }
}

bool UAS::getSelected() const
{
    return (UASManager::instance()->getActiveUAS() == this);
pixhawk's avatar
pixhawk committed
155 156
}

157 158
void UAS::receiveMessageNamedValue(const mavlink_message_t& message)
{
159
    if (message.msgid == MAVLINK_MSG_ID_NAMED_VALUE_FLOAT) {
pixhawk's avatar
pixhawk committed
160 161
        mavlink_named_value_float_t val;
        mavlink_msg_named_value_float_decode(&message, &val);
lm's avatar
lm committed
162 163
        QByteArray bytes(val.name, MAVLINK_MSG_NAMED_VALUE_FLOAT_FIELD_NAME_LEN);
        emit valueChanged(this->getUASID(), QString(bytes), tr("raw"), val.value, getUnixTime());
164
    } else if (message.msgid == MAVLINK_MSG_ID_NAMED_VALUE_INT) {
pixhawk's avatar
pixhawk committed
165 166
        mavlink_named_value_int_t val;
        mavlink_msg_named_value_int_decode(&message, &val);
lm's avatar
lm committed
167 168
        QByteArray bytes(val.name, MAVLINK_MSG_NAMED_VALUE_INT_FIELD_NAME_LEN);
        emit valueChanged(this->getUASID(), QString(bytes), tr("raw"), val.value, getUnixTime());
169 170 171
    }
}

pixhawk's avatar
pixhawk committed
172 173
void UAS::receiveMessage(LinkInterface* link, mavlink_message_t message)
{
174
    if (!link) return;
175
    if (!links->contains(link)) {
pixhawk's avatar
pixhawk committed
176
        addLink(link);
177
        //        qDebug() << __FILE__ << __LINE__ << "ADDED LINK!" << link->getName();
pixhawk's avatar
pixhawk committed
178
    }
179 180 181 182
    //    else
    //    {
    //        qDebug() << __FILE__ << __LINE__ << "DID NOT ADD LINK" << link->getName() << "ALREADY IN LIST";
    //    }
pixhawk's avatar
pixhawk committed
183

184
    //    qDebug() << "UAS RECEIVED from" << message.sysid << "component" << message.compid << "msg id" << message.msgid << "seq no" << message.seq;
pixhawk's avatar
pixhawk committed
185

186
    if (message.sysid == uasId) {
pixhawk's avatar
pixhawk committed
187 188
        QString uasState;
        QString stateDescription;
pixhawk's avatar
pixhawk committed
189

190
        switch (message.msgid) {
pixhawk's avatar
pixhawk committed
191
        case MAVLINK_MSG_ID_HEARTBEAT:
192
            lastHeartbeat = QGC::groundTimeUsecs();
pixhawk's avatar
pixhawk committed
193 194
            emit heartbeat(this);
            // Set new type if it has changed
195
            if (this->type != mavlink_msg_heartbeat_get_type(&message)) {
pixhawk's avatar
pixhawk committed
196
                this->type = mavlink_msg_heartbeat_get_type(&message);
197 198
                if (airframe == 0) {
                    switch (type) {
199 200 201 202 203 204 205 206 207 208 209
                    case MAV_FIXED_WING:
                        setAirframe(UASInterface::QGC_AIRFRAME_EASYSTAR);
                        break;
                    case MAV_QUADROTOR:
                        setAirframe(UASInterface::QGC_AIRFRAME_CHEETAH);
                        break;
                    default:
                        // Do nothing
                        break;
                    }
                }
210
                this->autopilot = mavlink_msg_heartbeat_get_autopilot(&message);
pixhawk's avatar
pixhawk committed
211 212
                emit systemTypeSet(this, type);
            }
213

214 215 216 217 218
            break;
        case MAVLINK_MSG_ID_NAMED_VALUE_FLOAT:
        case MAVLINK_MSG_ID_NAMED_VALUE_INT:
            // Receive named value message
            receiveMessageNamedValue(message);
pixhawk's avatar
pixhawk committed
219 220 221 222 223 224
            break;
        case MAVLINK_MSG_ID_BOOT:
            getStatusForCode((int)MAV_STATE_BOOT, uasState, stateDescription);
            emit statusChanged(this, uasState, stateDescription);
            onboardTimeOffset = 0; // Reset offset measurement
            break;
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246
        case MAVLINK_MSG_ID_SYS_STATUS: {
            mavlink_sys_status_t state;
            mavlink_msg_sys_status_decode(&message, &state);

            // FIXME
            //qDebug() << "1 SYSTEM STATUS:" << state.status;

            QString audiostring = "System " + getUASName();
            QString stateAudio = "";
            QString modeAudio = "";
            bool statechanged = false;
            bool modechanged = false;

            if (state.status != this->status) {
                statechanged = true;
                this->status = state.status;
                getStatusForCode((int)state.status, uasState, stateDescription);
                emit statusChanged(this, uasState, stateDescription);
                emit statusChanged(this->status);

                stateAudio = " changed status to " + uasState;
            }
247

248 249 250 251
            if (navMode != state.nav_mode) {
                emit navModeChanged(uasId, state.nav_mode, getNavModeText(state.nav_mode));
                navMode = state.nav_mode;
            }
lm's avatar
lm committed
252

253 254
            emit loadChanged(this,state.load/10.0f);
            emit valueChanged(uasId, "Load", "%", ((float)state.load)/10.0f, getUnixTime());
pixhawk's avatar
pixhawk committed
255

256 257 258 259
            if (this->mode != static_cast<int>(state.mode)) {
                modechanged = true;
                this->mode = static_cast<int>(state.mode);
                QString mode;
260

261 262 263 264 265 266 267
                switch (state.mode) {
                case (uint8_t)MAV_MODE_LOCKED:
                    mode = "LOCKED MODE";
                    break;
                case (uint8_t)MAV_MODE_MANUAL:
                    mode = "MANUAL MODE";
                    break;
268

269 270 271 272 273 274 275
#ifdef MAVLINK_ENABLED_SLUGS
                case (uint8_t)MAV_MODE_AUTO:
                    mode = "WAYPOINT MODE";
                    break;
                case (uint8_t)MAV_MODE_GUIDED:
                    mode = "MID-L CMDS MODE";
                    break;
276

277 278 279 280 281 282 283 284 285 286 287 288 289
                case (uint8_t)MAV_MODE_TEST1:
                    mode = "PASST MODE";
                    break;
                case (uint8_t)MAV_MODE_TEST2:
                    mode = "SEL PT MODE";
                    break;
#else
                case (uint8_t)MAV_MODE_AUTO:
                    mode = "AUTO MODE";
                    break;
                case (uint8_t)MAV_MODE_GUIDED:
                    mode = "GUIDED MODE";
                    break;
290

291 292 293 294 295 296 297 298 299
                case (uint8_t)MAV_MODE_TEST1:
                    mode = "TEST1 MODE";
                    break;
                case (uint8_t)MAV_MODE_TEST2:
                    mode = "TEST2 MODE";
                    break;
#endif
                case (uint8_t)MAV_MODE_READY:
                    mode = "READY MODE";
300 301
                    break;

302 303 304
                case (uint8_t)MAV_MODE_TEST3:
                    mode = "TEST3 MODE";
                    break;
pixhawk's avatar
pixhawk committed
305

306 307 308 309 310 311 312
                case (uint8_t)MAV_MODE_RC_TRAINING:
                    mode = "RC TRAINING MODE";
                    break;
                default:
                    mode = "UNINIT MODE";
                    break;
                }
313

314
                emit modeChanged(this->getUASID(), mode, "");
315

316
                //qDebug() << "2 SYSTEM MODE:" << mode;
lm's avatar
lm committed
317

318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
                modeAudio = " is now in " + mode;
            }
            currentVoltage = state.vbat/1000.0f;
            lpVoltage = filterVoltage(currentVoltage);
            if (startVoltage == 0) startVoltage = currentVoltage;
            timeRemaining = calculateTimeRemaining();
            if (!batteryRemainingEstimateEnabled) {
                chargeLevel = state.battery_remaining/10.0f;
            }
            //qDebug() << "Voltage: " << currentVoltage << " Chargelevel: " << getChargeLevel() << " Time remaining " << timeRemaining;
            emit batteryChanged(this, lpVoltage, getChargeLevel(), timeRemaining);
            emit voltageChanged(message.sysid, state.vbat/1000.0f);

            // LOW BATTERY ALARM
            if (lpVoltage < warnVoltage) {
                startLowBattAlarm();
            } else {
                stopLowBattAlarm();
            }
337

338 339
            // COMMUNICATIONS DROP RATE
            emit dropRateChanged(this->getUASID(), state.packet_drop/1000.0f);
340 341


342 343
            //add for development
            //emit remoteControlRSSIChanged(state.packet_drop/1000.0f);
344

345 346 347
            //float en = state.packet_drop/1000.0f;
            //emit remoteControlChannelRawChanged(0, en);//MAVLINK_MSG_ID_RC_CHANNELS_RAW
            //emit remoteControlChannelScaledChanged(0, en/100.0f);//MAVLINK_MSG_ID_RC_CHANNELS_SCALED
348

lm's avatar
lm committed
349

350
            //qDebug() << __FILE__ << __LINE__ << "RCV LOSS: " << state.packet_drop;
351

352 353 354 355 356 357 358 359 360 361 362 363 364
            // AUDIO
            if (modechanged && statechanged) {
                // Output both messages
                audiostring += modeAudio + " and " + stateAudio;
            } else {
                // Output the one message
                audiostring += modeAudio + stateAudio;
            }
            if ((int)state.status == (int)MAV_STATE_CRITICAL || state.status == (int)MAV_STATE_EMERGENCY) {
                GAudioOutput::instance()->startEmergency();
            } else if (modechanged || statechanged) {
                GAudioOutput::instance()->stopEmergency();
                GAudioOutput::instance()->say(audiostring);
pixhawk's avatar
pixhawk committed
365
            }
James Goppert's avatar
James Goppert committed
366

367 368 369
            if (state.status == MAV_STATE_POWEROFF) {
                emit systemRemoved(this);
                emit systemRemoved();
370
            }
371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389
        }
        break;

#ifdef MAVLINK_ENABLED_PIXHAWK
        case MAVLINK_MSG_ID_CONTROL_STATUS: {
            mavlink_control_status_t status;
            mavlink_msg_control_status_decode(&message, &status);
            // Emit control status vector
            emit attitudeControlEnabled(static_cast<bool>(status.control_att));
            emit positionXYControlEnabled(static_cast<bool>(status.control_pos_xy));
            emit positionZControlEnabled(static_cast<bool>(status.control_pos_z));
            emit positionYawControlEnabled(static_cast<bool>(status.control_pos_yaw));

            // Emit localization status vector
            emit localizationChanged(this, status.position_fix);
            emit visionLocalizationChanged(this, status.vision_fix);
            emit gpsLocalizationChanged(this, status.gps_fix);
        }
        break;
lm's avatar
lm committed
390
#endif // PIXHAWK
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422
        case MAVLINK_MSG_ID_RAW_IMU: {
            mavlink_raw_imu_t raw;
            mavlink_msg_raw_imu_decode(&message, &raw);
            quint64 time = getUnixTime(raw.usec);

            emit valueChanged(uasId, "accel x", "raw", static_cast<double>(raw.xacc), time);
            emit valueChanged(uasId, "accel y", "raw", static_cast<double>(raw.yacc), time);
            emit valueChanged(uasId, "accel z", "raw", static_cast<double>(raw.zacc), time);
            emit valueChanged(uasId, "gyro roll", "raw", static_cast<double>(raw.xgyro), time);
            emit valueChanged(uasId, "gyro pitch", "raw", static_cast<double>(raw.ygyro), time);
            emit valueChanged(uasId, "gyro yaw", "raw", static_cast<double>(raw.zgyro), time);
            emit valueChanged(uasId, "mag x", "raw", static_cast<double>(raw.xmag), time);
            emit valueChanged(uasId, "mag y", "raw", static_cast<double>(raw.ymag), time);
            emit valueChanged(uasId, "mag z", "raw", static_cast<double>(raw.zmag), time);
        }
        break;
        case MAVLINK_MSG_ID_SCALED_IMU: {
            mavlink_scaled_imu_t scaled;
            mavlink_msg_scaled_imu_decode(&message, &scaled);
            quint64 time = getUnixTime(scaled.usec);

            emit valueChanged(uasId, "accel x", "g", scaled.xacc/1000.0f, time);
            emit valueChanged(uasId, "accel y", "g", scaled.yacc/1000.0f, time);
            emit valueChanged(uasId, "accel z", "g", scaled.zacc/1000.0f, time);
            emit valueChanged(uasId, "gyro roll", "rad/s", scaled.xgyro/1000.0f, time);
            emit valueChanged(uasId, "gyro pitch", "rad/s", scaled.ygyro/1000.0f, time);
            emit valueChanged(uasId, "gyro yaw", "rad/s", scaled.zgyro/1000.0f, time);
            emit valueChanged(uasId, "mag x", "tesla", scaled.xmag/1000.0f, time);
            emit valueChanged(uasId, "mag y", "tesla", scaled.ymag/1000.0f, time);
            emit valueChanged(uasId, "mag z", "tesla", scaled.zmag/1000.0f, time);
        }
        break;
pixhawk's avatar
pixhawk committed
423 424
        case MAVLINK_MSG_ID_ATTITUDE:
            //std::cerr << std::endl;
pixhawk's avatar
pixhawk committed
425
            //std::cerr << "Decoded attitude message:" << " roll: " << std::dec << mavlink_msg_attitude_get_roll(message.payload) << " pitch: " << mavlink_msg_attitude_get_pitch(message.payload) << " yaw: " << mavlink_msg_attitude_get_yaw(message.payload) << std::endl;
426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
        {
            mavlink_attitude_t attitude;
            mavlink_msg_attitude_decode(&message, &attitude);
            quint64 time = getUnixTime(attitude.usec);
            roll = QGC::limitAngleToPMPIf(attitude.roll);
            pitch = QGC::limitAngleToPMPIf(attitude.pitch);
            yaw = QGC::limitAngleToPMPIf(attitude.yaw);
            emit valueChanged(uasId, "roll", "rad", roll, time);
            emit valueChanged(uasId, "pitch", "rad", pitch, time);
            emit valueChanged(uasId, "yaw", "rad", yaw, time);
            emit valueChanged(uasId, "rollspeed", "rad/s", attitude.rollspeed, time);
            emit valueChanged(uasId, "pitchspeed", "rad/s", attitude.pitchspeed, time);
            emit valueChanged(uasId, "yawspeed", "rad/s", attitude.yawspeed, time);

            // Emit in angles

            // Convert yaw angle to compass value
            // in 0 - 360 deg range
            float compass = (yaw/M_PI)*180.0+360.0f;
            while (compass > 360.0f) {
                compass -= 360.0f;
            }
lm's avatar
lm committed
448

449
            attitudeKnown = true;
lm's avatar
lm committed
450

451 452 453 454 455 456
            emit valueChanged(uasId, "roll deg", "deg", (roll/M_PI)*180.0, time);
            emit valueChanged(uasId, "pitch deg", "deg", (pitch/M_PI)*180.0, time);
            emit valueChanged(uasId, "heading deg", "deg", compass, time);
            emit valueChanged(uasId, "rollspeed d/s", "deg/s", (attitude.rollspeed/M_PI)*180.0, time);
            emit valueChanged(uasId, "pitchspeed d/s", "deg/s", (attitude.pitchspeed/M_PI)*180.0, time);
            emit valueChanged(uasId, "yawspeed d/s", "deg/s", (attitude.yawspeed/M_PI)*180.0, time);
457

458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
            emit attitudeChanged(this, roll, pitch, yaw, time);
            emit attitudeSpeedChanged(uasId, attitude.rollspeed, attitude.pitchspeed, attitude.yawspeed, time);
        }
        break;
        case MAVLINK_MSG_ID_VFR_HUD: {
            mavlink_vfr_hud_t hud;
            mavlink_msg_vfr_hud_decode(&message, &hud);
            quint64 time = getUnixTime();
            // Display updated values
            emit valueChanged(uasId, "airspeed", "m/s", hud.airspeed, time);
            emit valueChanged(uasId, "groundspeed", "m/s", hud.groundspeed, time);
            emit valueChanged(uasId, "altitude", "m", hud.alt, time);
            emit valueChanged(uasId, "heading", "deg", hud.heading, time);
            emit valueChanged(uasId, "climbrate", "m/s", hud.climb, time);
            emit valueChanged(uasId, "throttle", "%", hud.throttle, time);
            emit thrustChanged(this, hud.throttle/100.0);

            if (!attitudeKnown) {
                yaw = QGC::limitAngleToPMPId((((double)hud.heading-180.0)/360.0)*M_PI);
477
                emit attitudeChanged(this, roll, pitch, yaw, time);
pixhawk's avatar
pixhawk committed
478
            }
lm's avatar
lm committed
479

480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500
            emit altitudeChanged(uasId, hud.alt);
            //yaw = (hud.heading-180.0f/360.0f)*M_PI;
            //emit attitudeChanged(this, roll, pitch, yaw, getUnixTime());
            emit speedChanged(this, hud.airspeed, 0.0f, hud.climb, getUnixTime());
        }
        break;
        case MAVLINK_MSG_ID_NAV_CONTROLLER_OUTPUT: {
            mavlink_nav_controller_output_t nav;
            mavlink_msg_nav_controller_output_decode(&message, &nav);
            quint64 time = getUnixTime();
            // Update UI
            emit valueChanged(uasId, "nav roll", "deg", nav.nav_roll, time);
            emit valueChanged(uasId, "nav pitch", "deg", nav.nav_pitch, time);
            emit valueChanged(uasId, "nav bearing", "deg", nav.nav_bearing, time);
            emit valueChanged(uasId, "target bearing", "deg", nav.target_bearing, time);
            emit valueChanged(uasId, "wp dist", "m", nav.wp_dist, time);
            emit valueChanged(uasId, "alt err", "m", nav.alt_error, time);
            emit valueChanged(uasId, "airspeed err", "m/s", nav.alt_error, time);
            emit valueChanged(uasId, "xtrack err", "m", nav.xtrack_error, time);
        }
        break;
501
        case MAVLINK_MSG_ID_LOCAL_POSITION:
lm's avatar
lm committed
502
            //std::cerr << std::endl;
pixhawk's avatar
pixhawk committed
503
            //std::cerr << "Decoded attitude message:" << " roll: " << std::dec << mavlink_msg_attitude_get_roll(message.payload) << " pitch: " << mavlink_msg_attitude_get_pitch(message.payload) << " yaw: " << mavlink_msg_attitude_get_yaw(message.payload) << std::endl;
504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527
        {
            mavlink_local_position_t pos;
            mavlink_msg_local_position_decode(&message, &pos);
            quint64 time = getUnixTime(pos.usec);
            localX = pos.x;
            localY = pos.y;
            localZ = pos.z;
            emit valueChanged(uasId, "x", "m", pos.x, time);
            emit valueChanged(uasId, "y", "m", pos.y, time);
            emit valueChanged(uasId, "z", "m", pos.z, time);
            emit valueChanged(uasId, "x speed", "m/s", pos.vx, time);
            emit valueChanged(uasId, "y speed", "m/s", pos.vy, time);
            emit valueChanged(uasId, "z speed", "m/s", pos.vz, time);
            emit localPositionChanged(this, pos.x, pos.y, pos.z, time);
            emit speedChanged(this, pos.vx, pos.vy, pos.vz, time);

            //                qDebug()<<"Local Position = "<<pos.x<<" - "<<pos.y<<" - "<<pos.z;
            //                qDebug()<<"Speed Local Position = "<<pos.vx<<" - "<<pos.vy<<" - "<<pos.vz;

            //emit attitudeChanged(this, pos.roll, pos.pitch, pos.yaw, time);
            // Set internal state
            if (!positionLock) {
                // If position was not locked before, notify positive
                GAudioOutput::instance()->notifyPositive();
lm's avatar
lm committed
528
            }
529 530 531
            positionLock = true;
        }
        break;
532
        case MAVLINK_MSG_ID_GLOBAL_POSITION_INT:
533 534
            //std::cerr << std::endl;
            //std::cerr << "Decoded attitude message:" << " roll: " << std::dec << mavlink_msg_attitude_get_roll(message.payload) << " pitch: " << mavlink_msg_attitude_get_pitch(message.payload) << " yaw: " << mavlink_msg_attitude_get_yaw(message.payload) << std::endl;
535 536 537
        {
            mavlink_global_position_int_t pos;
            mavlink_msg_global_position_int_decode(&message, &pos);
538
            quint64 time = getUnixTime();
539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555
            latitude = pos.lat/(double)1E7;
            longitude = pos.lon/(double)1E7;
            altitude = pos.alt/1000.0;
            speedX = pos.vx/100.0;
            speedY = pos.vy/100.0;
            speedZ = pos.vz/100.0;
            emit valueChanged(uasId, "latitude", "deg", latitude, time);
            emit valueChanged(uasId, "longitude", "deg", longitude, time);
            emit valueChanged(uasId, "altitude", "m", altitude, time);
            double totalSpeed = sqrt(speedX*speedX + speedY*speedY + speedZ*speedZ);
            emit valueChanged(uasId, "gps speed", "m/s", totalSpeed, time);
            emit globalPositionChanged(this, latitude, longitude, altitude, time);
            emit speedChanged(this, speedX, speedY, speedZ, time);
            // Set internal state
            if (!positionLock) {
                // If position was not locked before, notify positive
                GAudioOutput::instance()->notifyPositive();
556
            }
557 558 559 560 561 562 563 564
            positionLock = true;
            //TODO fix this hack for forwarding of global position for patch antenna tracking
            forwardMessage(message);
        }
        break;
        case MAVLINK_MSG_ID_GLOBAL_POSITION: {
            mavlink_global_position_t pos;
            mavlink_msg_global_position_decode(&message, &pos);
565
            quint64 time = getUnixTime();
566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582
            latitude = pos.lat;
            longitude = pos.lon;
            altitude = pos.alt;
            speedX = pos.vx;
            speedY = pos.vy;
            speedZ = pos.vz;
            emit valueChanged(uasId, "latitude", "deg", latitude, time);
            emit valueChanged(uasId, "longitude", "deg", longitude, time);
            emit valueChanged(uasId, "altitude", "m", altitude, time);
            double totalSpeed = sqrt(speedX*speedX + speedY*speedY + speedZ*speedZ);
            emit valueChanged(uasId, "gps speed", "m/s", totalSpeed, time);
            emit globalPositionChanged(this, latitude, longitude, altitude, time);
            emit speedChanged(this, speedX, speedY, speedZ, time);
            // Set internal state
            if (!positionLock) {
                // If position was not locked before, notify positive
                GAudioOutput::instance()->notifyPositive();
583
            }
584 585 586 587 588
            positionLock = true;
            //TODO fix this hack for forwarding of global position for patch antenna tracking
            forwardMessage(message);
        }
        break;
589 590 591
        case MAVLINK_MSG_ID_GPS_RAW:
            //std::cerr << std::endl;
            //std::cerr << "Decoded attitude message:" << " roll: " << std::dec << mavlink_msg_attitude_get_roll(message.payload) << " pitch: " << mavlink_msg_attitude_get_pitch(message.payload) << " yaw: " << mavlink_msg_attitude_get_yaw(message.payload) << std::endl;
592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616
        {
            mavlink_gps_raw_t pos;
            mavlink_msg_gps_raw_decode(&message, &pos);

            // SANITY CHECK
            // only accept values in a realistic range
            // quint64 time = getUnixTime(pos.usec);
            quint64 time = getUnixTime();

            emit valueChanged(uasId, "latitude", "deg", pos.lat, time);
            emit valueChanged(uasId, "longitude", "deg", pos.lon, time);

            if (pos.fix_type > 0) {
                emit globalPositionChanged(this, pos.lat, pos.lon, pos.alt, time);
                emit valueChanged(uasId, "gps speed", "m/s", pos.v, time);
                latitude = pos.lat;
                longitude = pos.lon;
                altitude = pos.alt;
                positionLock = true;

                // Check for NaN
                int alt = pos.alt;
                if (alt != alt) {
                    alt = 0;
                    emit textMessageReceived(uasId, message.compid, 255, "GCS ERROR: RECEIVED NaN FOR ALTITUDE");
617
                }
618 619 620 621 622 623 624 625
                emit valueChanged(uasId, "altitude", "m", pos.alt, time);
                // Smaller than threshold and not NaN
                if (pos.v < 1000000 && pos.v == pos.v) {
                    emit valueChanged(uasId, "speed", "m/s", pos.v, time);
                    //qDebug() << "GOT GPS RAW";
                    // emit speedChanged(this, (double)pos.v, 0.0, 0.0, time);
                } else {
                    emit textMessageReceived(uasId, message.compid, 255, QString("GCS ERROR: RECEIVED INVALID SPEED OF %1 m/s").arg(pos.v));
626 627
                }
            }
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663
        }
        break;
        case MAVLINK_MSG_ID_GPS_RAW_INT: {
            mavlink_gps_raw_int_t pos;
            mavlink_msg_gps_raw_int_decode(&message, &pos);

            // SANITY CHECK
            // only accept values in a realistic range
            // quint64 time = getUnixTime(pos.usec);
            quint64 time = getUnixTime();

            emit valueChanged(uasId, "latitude", "deg", pos.lat/(double)1E7, time);
            emit valueChanged(uasId, "longitude", "deg", pos.lon/(double)1E7, time);

            if (pos.fix_type > 0) {
                emit globalPositionChanged(this, pos.lat/(double)1E7, pos.lon/(double)1E7, pos.alt/1000.0, time);
                emit valueChanged(uasId, "gps speed", "m/s", pos.v, time);
                latitude = pos.lat/(double)1E7;
                longitude = pos.lon/(double)1E7;
                altitude = pos.alt/1000.0;
                positionLock = true;

                // Check for NaN
                int alt = pos.alt;
                if (alt != alt) {
                    alt = 0;
                    emit textMessageReceived(uasId, message.compid, 255, "GCS ERROR: RECEIVED NaN FOR ALTITUDE");
                }
                emit valueChanged(uasId, "altitude", "m", pos.alt/(double)1E3, time);
                // Smaller than threshold and not NaN
                if (pos.v < 1000000 && pos.v == pos.v) {
                    emit valueChanged(uasId, "speed", "m/s", pos.v, time);
                    //qDebug() << "GOT GPS RAW";
                    // emit speedChanged(this, (double)pos.v, 0.0, 0.0, time);
                } else {
                    emit textMessageReceived(uasId, message.compid, 255, QString("GCS ERROR: RECEIVED INVALID SPEED OF %1 m/s").arg(pos.v));
lm's avatar
lm committed
664 665
                }
            }
666 667 668 669 670 671 672
        }
        break;
        case MAVLINK_MSG_ID_GPS_STATUS: {
            mavlink_gps_status_t pos;
            mavlink_msg_gps_status_decode(&message, &pos);
            for(int i = 0; i < (int)pos.satellites_visible; i++) {
                emit gpsSatelliteStatusChanged(uasId, (unsigned char)pos.satellite_prn[i], (unsigned char)pos.satellite_elevation[i], (unsigned char)pos.satellite_azimuth[i], (unsigned char)pos.satellite_snr[i], static_cast<bool>(pos.satellite_used[i]));
673
            }
674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
        }
        break;
        case MAVLINK_MSG_ID_GPS_LOCAL_ORIGIN_SET: {
            mavlink_gps_local_origin_set_t pos;
            mavlink_msg_gps_local_origin_set_decode(&message, &pos);
            // FIXME Emit to other components
        }
        break;
        case MAVLINK_MSG_ID_RAW_PRESSURE: {
            mavlink_raw_pressure_t pressure;
            mavlink_msg_raw_pressure_decode(&message, &pressure);
            quint64 time = this->getUnixTime(pressure.usec);
            emit valueChanged(uasId, "abs pressure", "raw", pressure.press_abs, time);
            emit valueChanged(uasId, "diff pressure 1", "raw", pressure.press_diff1, time);
            emit valueChanged(uasId, "diff pressure 2", "raw", pressure.press_diff2, time);
            emit valueChanged(uasId, "temperature", "raw", pressure.temperature, time);
        }
        break;
692

James Goppert's avatar
James Goppert committed
693 694 695 696 697 698 699 700 701
        case MAVLINK_MSG_ID_SCALED_PRESSURE: {
            mavlink_scaled_pressure_t pressure;
            mavlink_msg_scaled_pressure_decode(&message, &pressure);
            quint64 time = this->getUnixTime(pressure.usec);
            emit valueChanged(uasId, "abs pressure", "hPa", pressure.press_abs, time);
            emit valueChanged(uasId, "diff pressure", "hPa", pressure.press_diff, time);
            emit valueChanged(uasId, "temperature", "C", pressure.temperature/100.0, time);
        }
        break;
702

703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741
        case MAVLINK_MSG_ID_RC_CHANNELS_RAW: {
            mavlink_rc_channels_raw_t channels;
            mavlink_msg_rc_channels_raw_decode(&message, &channels);
            emit remoteControlRSSIChanged(channels.rssi/255.0f);
            emit remoteControlChannelRawChanged(0, channels.chan1_raw);
            emit remoteControlChannelRawChanged(1, channels.chan2_raw);
            emit remoteControlChannelRawChanged(2, channels.chan3_raw);
            emit remoteControlChannelRawChanged(3, channels.chan4_raw);
            emit remoteControlChannelRawChanged(4, channels.chan5_raw);
            emit remoteControlChannelRawChanged(5, channels.chan6_raw);
            emit remoteControlChannelRawChanged(6, channels.chan7_raw);
            emit remoteControlChannelRawChanged(7, channels.chan8_raw);
        }
        break;
        case MAVLINK_MSG_ID_RC_CHANNELS_SCALED: {
            mavlink_rc_channels_scaled_t channels;
            mavlink_msg_rc_channels_scaled_decode(&message, &channels);
            emit remoteControlRSSIChanged(channels.rssi/255.0f);
            emit remoteControlChannelScaledChanged(0, channels.chan1_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(1, channels.chan2_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(2, channels.chan3_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(3, channels.chan4_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(4, channels.chan5_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(5, channels.chan6_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(6, channels.chan7_scaled/10000.0f);
            emit remoteControlChannelScaledChanged(7, channels.chan8_scaled/10000.0f);
        }
        break;
        case MAVLINK_MSG_ID_PARAM_VALUE: {
            mavlink_param_value_t value;
            mavlink_msg_param_value_decode(&message, &value);
            QByteArray bytes((char*)value.param_id, MAVLINK_MSG_PARAM_VALUE_FIELD_PARAM_ID_LEN);
            QString parameterName = QString(bytes);
            int component = message.compid;
            float val = value.param_value;

            // Insert component if necessary
            if (!parameters.contains(component)) {
                parameters.insert(component, new QMap<QString, float>());
lm's avatar
lm committed
742
            }
743

744 745 746
            // Insert parameter into registry
            if (parameters.value(component)->contains(parameterName)) parameters.value(component)->remove(parameterName);
            parameters.value(component)->insert(parameterName, val);
747

748 749 750 751 752
            // Emit change
            emit parameterChanged(uasId, message.compid, parameterName, val);
            emit parameterChanged(uasId, message.compid, value.param_count, value.param_index, parameterName, val);
        }
        break;
lm's avatar
lm committed
753 754 755
        case MAVLINK_MSG_ID_ACTION_ACK:
            mavlink_action_ack_t ack;
            mavlink_msg_action_ack_decode(&message, &ack);
756
            if (ack.result == 1) {
lm's avatar
lm committed
757
                emit textMessageReceived(uasId, message.compid, 0, tr("SUCCESS: Executed action: %1").arg(ack.action));
758
            } else {
lm's avatar
lm committed
759
                emit textMessageReceived(uasId, message.compid, 0, tr("FAILURE: Rejected action: %1").arg(ack.action));
760 761
            }
            break;
pixhawk's avatar
pixhawk committed
762
        case MAVLINK_MSG_ID_DEBUG:
763
            emit valueChanged(uasId, QString("debug ") + QString::number(mavlink_msg_debug_get_ind(&message)), "raw", mavlink_msg_debug_get_value(&message), MG::TIME::getGroundTimeNow());
pixhawk's avatar
pixhawk committed
764
            break;
765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789
        case MAVLINK_MSG_ID_ATTITUDE_CONTROLLER_OUTPUT: {
            mavlink_attitude_controller_output_t out;
            mavlink_msg_attitude_controller_output_decode(&message, &out);
            quint64 time = MG::TIME::getGroundTimeNowUsecs();
            emit attitudeThrustSetPointChanged(this, out.roll/127.0f, out.pitch/127.0f, out.yaw/127.0f, (uint8_t)out.thrust, time);
            emit valueChanged(uasId, "att control roll", "raw", out.roll, time/1000.0f);
            emit valueChanged(uasId, "att control pitch", "raw", out.pitch, time/1000.0f);
            emit valueChanged(uasId, "att control yaw", "raw", out.yaw, time/1000.0f);
        }
        break;
        case MAVLINK_MSG_ID_POSITION_CONTROLLER_OUTPUT: {
            mavlink_position_controller_output_t out;
            mavlink_msg_position_controller_output_decode(&message, &out);
            quint64 time = MG::TIME::getGroundTimeNow();
            //emit positionSetPointsChanged(uasId, out.x/127.0f, out.y/127.0f, out.z/127.0f, out.yaw, time);
            emit valueChanged(uasId, "pos control x", "raw", out.x, time);
            emit valueChanged(uasId, "pos control y", "raw", out.y, time);
            emit valueChanged(uasId, "pos control z", "raw", out.z, time);
        }
        break;
        case MAVLINK_MSG_ID_WAYPOINT_COUNT: {
            mavlink_waypoint_count_t wpc;
            mavlink_msg_waypoint_count_decode(&message, &wpc);
            if (wpc.target_system == mavlink->getSystemId() && wpc.target_component == mavlink->getComponentId()) {
                waypointManager.handleWaypointCount(message.sysid, message.compid, wpc.count);
pixhawk's avatar
pixhawk committed
790
            }
791 792
        }
        break;
pixhawk's avatar
pixhawk committed
793

794 795 796 797 798 799
        case MAVLINK_MSG_ID_WAYPOINT: {
            mavlink_waypoint_t wp;
            mavlink_msg_waypoint_decode(&message, &wp);
            //qDebug() << "got waypoint (" << wp.seq << ") from ID " << message.sysid << " x=" << wp.x << " y=" << wp.y << " z=" << wp.z;
            if(wp.target_system == mavlink->getSystemId() && wp.target_component == mavlink->getComponentId()) {
                waypointManager.handleWaypoint(message.sysid, message.compid, &wp);
800
            }
801 802
        }
        break;
pixhawk's avatar
pixhawk committed
803

804 805 806 807 808
        case MAVLINK_MSG_ID_WAYPOINT_ACK: {
            mavlink_waypoint_ack_t wpa;
            mavlink_msg_waypoint_ack_decode(&message, &wpa);
            if(wpa.target_system == mavlink->getSystemId() && wpa.target_component == mavlink->getComponentId()) {
                waypointManager.handleWaypointAck(message.sysid, message.compid, &wpa);
809
            }
810 811
        }
        break;
812

813 814 815 816 817
        case MAVLINK_MSG_ID_WAYPOINT_REQUEST: {
            mavlink_waypoint_request_t wpr;
            mavlink_msg_waypoint_request_decode(&message, &wpr);
            if(wpr.target_system == mavlink->getSystemId() && wpr.target_component == mavlink->getComponentId()) {
                waypointManager.handleWaypointRequest(message.sysid, message.compid, &wpr);
pixhawk's avatar
pixhawk committed
818
            }
819 820
        }
        break;
pixhawk's avatar
pixhawk committed
821

822 823 824 825 826 827 828 829 830
        case MAVLINK_MSG_ID_WAYPOINT_REACHED: {
            mavlink_waypoint_reached_t wpr;
            mavlink_msg_waypoint_reached_decode(&message, &wpr);
            waypointManager.handleWaypointReached(message.sysid, message.compid, &wpr);
            QString text = QString("System %1 reached waypoint %2").arg(getUASName()).arg(wpr.seq);
            GAudioOutput::instance()->say(text);
            emit textMessageReceived(message.sysid, message.compid, 0, text);
        }
        break;
pixhawk's avatar
pixhawk committed
831

832 833 834 835 836 837
        case MAVLINK_MSG_ID_WAYPOINT_CURRENT: {
            mavlink_waypoint_current_t wpc;
            mavlink_msg_waypoint_current_decode(&message, &wpc);
            waypointManager.handleWaypointCurrent(message.sysid, message.compid, &wpc);
        }
        break;
pixhawk's avatar
pixhawk committed
838

839 840 841 842 843 844 845 846 847
        case MAVLINK_MSG_ID_LOCAL_POSITION_SETPOINT: {
            mavlink_local_position_setpoint_t p;
            mavlink_msg_local_position_setpoint_decode(&message, &p);
            emit positionSetPointsChanged(uasId, p.x, p.y, p.z, p.yaw, QGC::groundTimeUsecs());
        }
        break;
        case MAVLINK_MSG_ID_SERVO_OUTPUT_RAW: {
            mavlink_servo_output_raw_t servos;
            mavlink_msg_servo_output_raw_decode(&message, &servos);
848
            quint64 time = getUnixTime();
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870
            emit valueChanged(uasId, "servo #1", "us", servos.servo1_raw, time);
            emit valueChanged(uasId, "servo #2", "us", servos.servo2_raw, time);
            emit valueChanged(uasId, "servo #3", "us", servos.servo3_raw, time);
            emit valueChanged(uasId, "servo #4", "us", servos.servo4_raw, time);
            emit valueChanged(uasId, "servo #5", "us", servos.servo5_raw, time);
            emit valueChanged(uasId, "servo #6", "us", servos.servo6_raw, time);
            emit valueChanged(uasId, "servo #7", "us", servos.servo7_raw, time);
            emit valueChanged(uasId, "servo #8", "us", servos.servo8_raw, time);
        }
        break;
        case MAVLINK_MSG_ID_STATUSTEXT: {
            QByteArray b;
            b.resize(MAVLINK_MSG_STATUSTEXT_FIELD_TEXT_LEN);
            mavlink_msg_statustext_get_text(&message, (int8_t*)b.data());
            //b.append('\0');
            QString text = QString(b);
            int severity = mavlink_msg_statustext_get_severity(&message);
            //qDebug() << "RECEIVED STATUS:" << text;false
            //emit statusTextReceived(severity, text);
            emit textMessageReceived(uasId, message.compid, severity, text);
        }
        break;
871
#ifdef MAVLINK_ENABLED_PIXHAWK
872 873 874 875 876 877 878 879 880 881 882
        case MAVLINK_MSG_ID_DATA_TRANSMISSION_HANDSHAKE: {
            qDebug() << "RECIEVED ACK TO GET IMAGE";
            mavlink_data_transmission_handshake_t p;
            mavlink_msg_data_transmission_handshake_decode(&message, &p);
            imageSize = p.size;
            imagePackets = p.packets;
            imagePayload = p.payload;
            imageQuality = p.jpg_quality;
            imageStart = QGC::groundTimeMilliseconds();
        }
        break;
883

884 885 886 887 888 889 890 891 892
        case MAVLINK_MSG_ID_ENCAPSULATED_DATA: {
            mavlink_encapsulated_data_t img;
            mavlink_msg_encapsulated_data_decode(&message, &img);
            int seq = img.seqnr;
            int pos = seq * imagePayload;

            for (int i = 0; i < imagePayload; ++i) {
                if (pos <= imageSize) {
                    imageRecBuffer[pos] = img.data[i];
893
                }
894 895
                ++pos;
            }
896

897
            ++imagePacketsArrived;
898

899
            // emit signal if all packets arrived
900 901
            if ((imagePacketsArrived == imagePackets))
            {
902 903
                // Restart statemachine
                imagePacketsArrived = 0;
904
                emit imageReady(this);
905
                qDebug() << "imageReady emitted. all packets arrived";
906

907 908
                //this->requestImage();
                //qDebug() << "SENDING REQUEST TO GET NEW IMAGE FROM SYSTEM" << uasId;
909
            }
910 911
        }
        break;
912
#endif
913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938
        case MAVLINK_MSG_ID_DEBUG_VECT: {
            mavlink_debug_vect_t vect;
            mavlink_msg_debug_vect_decode(&message, &vect);
            QString str((const char*)vect.name);
            quint64 time = getUnixTime(vect.usec);
            emit valueChanged(uasId, str+".x", "raw", vect.x, time);
            emit valueChanged(uasId, str+".y", "raw", vect.y, time);
            emit valueChanged(uasId, str+".z", "raw", vect.z, time);
        }
        break;
        //#ifdef MAVLINK_ENABLED_PIXHAWK
        //            case MAVLINK_MSG_ID_POINT_OF_INTEREST:
        //            {
        //                mavlink_point_of_interest_t poi;
        //                mavlink_msg_point_of_interest_decode(&message, &poi);
        //                emit poiFound(this, poi.type, poi.color, QString((QChar*)poi.name, MAVLINK_MSG_POINT_OF_INTEREST_FIELD_NAME_LEN), poi.x, poi.y, poi.z);
        //            }
        //            break;
        //            case MAVLINK_MSG_ID_POINT_OF_INTEREST_CONNECTION:
        //            {
        //                mavlink_point_of_interest_connection_t poi;
        //                mavlink_msg_point_of_interest_connection_decode(&message, &poi);
        //                emit poiConnectionFound(this, poi.type, poi.color, QString((QChar*)poi.name, MAVLINK_MSG_POINT_OF_INTEREST_CONNECTION_FIELD_NAME_LEN), poi.x1, poi.y1, poi.z1, poi.x2, poi.y2, poi.z2);
        //            }
        //            break;
        //#endif
lm's avatar
lm committed
939
#ifdef MAVLINK_ENABLED_UALBERTA
940 941 942
        case MAVLINK_MSG_ID_NAV_FILTER_BIAS: {
            mavlink_nav_filter_bias_t bias;
            mavlink_msg_nav_filter_bias_decode(&message, &bias);
943
            quint64 time = getUnixTime();
944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979
            emit valueChanged(uasId, "b_f[0]", "raw", bias.accel_0, time);
            emit valueChanged(uasId, "b_f[1]", "raw", bias.accel_1, time);
            emit valueChanged(uasId, "b_f[2]", "raw", bias.accel_2, time);
            emit valueChanged(uasId, "b_w[0]", "raw", bias.gyro_0, time);
            emit valueChanged(uasId, "b_w[1]", "raw", bias.gyro_1, time);
            emit valueChanged(uasId, "b_w[2]", "raw", bias.gyro_2, time);
        }
        break;
        case MAVLINK_MSG_ID_RADIO_CALIBRATION: {
            mavlink_radio_calibration_t radioMsg;
            mavlink_msg_radio_calibration_decode(&message, &radioMsg);
            QVector<float> aileron;
            QVector<float> elevator;
            QVector<float> rudder;
            QVector<float> gyro;
            QVector<float> pitch;
            QVector<float> throttle;

            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_AILERON_LEN; ++i)
                aileron << radioMsg.aileron[i];
            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_ELEVATOR_LEN; ++i)
                elevator << radioMsg.elevator[i];
            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_RUDDER_LEN; ++i)
                rudder << radioMsg.rudder[i];
            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_GYRO_LEN; ++i)
                gyro << radioMsg.gyro[i];
            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_PITCH_LEN; ++i)
                pitch << radioMsg.pitch[i];
            for (int i=0; i<MAVLINK_MSG_RADIO_CALIBRATION_FIELD_THROTTLE_LEN; ++i)
                throttle << radioMsg.throttle[i];

            QPointer<RadioCalibrationData> radioData = new RadioCalibrationData(aileron, elevator, rudder, gyro, pitch, throttle);
            emit radioCalibrationReceived(radioData);
            delete radioData;
        }
        break;
980

981
#endif
982 983
        // Messages to ignore
        case MAVLINK_MSG_ID_LOCAL_POSITION_SETPOINT_SET:
984
            break;
985 986 987 988
        default: {
            if (!unknownPackets.contains(message.msgid)) {
                unknownPackets.append(message.msgid);
                QString errString = tr("UNABLE TO DECODE MESSAGE NUMBER %1").arg(message.msgid);
989
                GAudioOutput::instance()->say(errString+tr(", please check console for details."));
990 991 992
                emit textMessageReceived(uasId, message.compid, 255, errString);
                std::cout << "Unable to decode message from system " << std::dec << static_cast<int>(message.sysid) << " with message id:" << static_cast<int>(message.msgid) << std::endl;
                //qDebug() << std::cerr << "Unable to decode message from system " << std::dec << static_cast<int>(message.acid) << " with message id:" << static_cast<int>(message.msgid) << std::endl;
lm's avatar
lm committed
993
            }
994 995
        }
        break;
pixhawk's avatar
pixhawk committed
996 997 998 999
        }
    }
}

1000 1001 1002 1003 1004 1005 1006 1007 1008
void UAS::setHomePosition(double lat, double lon, double alt)
{
    // Send new home position to UAS
    mavlink_gps_set_global_origin_t home;
    home.target_system = uasId;
    home.target_component = 0; // ALL components
    home.latitude = lat*1E7;
    home.longitude = lon*1E7;
    home.altitude = alt*1000;
1009
    qDebug() << "lat:" << home.latitude << " lon:" << home.longitude;
1010 1011 1012 1013 1014
    mavlink_message_t msg;
    mavlink_msg_gps_set_global_origin_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &home);
    sendMessage(msg);
}

1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
void UAS::setLocalOriginAtCurrentGPSPosition()
{
    bool result = false;
    QMessageBox msgBox;
    msgBox.setIcon(QMessageBox::Warning);
    msgBox.setText("Setting new World Coordinate Frame Origin");
    msgBox.setInformativeText("Do you want to set a new origin? Waypoints defined in the local frame will be shifted in their physical location");
    msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::Cancel);
    msgBox.setDefaultButton(QMessageBox::Cancel);
    int ret = msgBox.exec();

    // Close the message box shortly after the click to prevent accidental clicks
    QTimer::singleShot(5000, &msgBox, SLOT(reject()));


1030
    if (ret == QMessageBox::Yes) {
1031
        mavlink_message_t msg;
1032
        mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), 0, MAV_ACTION_SET_ORIGIN);
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
        // Send message twice to increase chance that it reaches its goal
        sendMessage(msg);
        // Wait 5 ms
        MG::SLEEP::usleep(5000);
        // Send again
        sendMessage(msg);
        result = true;
    }
}

pixhawk's avatar
pixhawk committed
1043 1044
void UAS::setLocalPositionSetpoint(float x, float y, float z, float yaw)
{
1045
#ifdef MAVLINK_ENABLED_PIXHAWK
pixhawk's avatar
pixhawk committed
1046
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1047 1048
    mavlink_msg_position_control_setpoint_set_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, 0, 0, x, y, z, yaw);
    sendMessage(msg);
1049
#else
lm's avatar
lm committed
1050 1051 1052 1053
    Q_UNUSED(x);
    Q_UNUSED(y);
    Q_UNUSED(z);
    Q_UNUSED(yaw);
1054
#endif
pixhawk's avatar
pixhawk committed
1055 1056
}

pixhawk's avatar
pixhawk committed
1057 1058
void UAS::setLocalPositionOffset(float x, float y, float z, float yaw)
{
lm's avatar
lm committed
1059
#ifdef MAVLINK_ENABLED_PIXHAWK
1060 1061 1062
    mavlink_message_t msg;
    mavlink_msg_position_control_offset_set_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, 0, x, y, z, yaw);
    sendMessage(msg);
lm's avatar
lm committed
1063
#else
1064 1065 1066 1067
    Q_UNUSED(x);
    Q_UNUSED(y);
    Q_UNUSED(z);
    Q_UNUSED(yaw);
pixhawk's avatar
pixhawk committed
1068 1069 1070 1071 1072
#endif
}

void UAS::startRadioControlCalibration()
{
1073 1074 1075
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_CALIBRATE_RC);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1076 1077
}

lm's avatar
lm committed
1078 1079
void UAS::startDataRecording()
{
1080 1081 1082
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_REC_START);
    sendMessage(msg);
lm's avatar
lm committed
1083 1084 1085 1086
}

void UAS::pauseDataRecording()
{
1087 1088 1089
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_REC_PAUSE);
    sendMessage(msg);
lm's avatar
lm committed
1090 1091 1092 1093
}

void UAS::stopDataRecording()
{
1094 1095 1096
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_REC_STOP);
    sendMessage(msg);
lm's avatar
lm committed
1097 1098
}

pixhawk's avatar
pixhawk committed
1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
void UAS::startMagnetometerCalibration()
{
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_CALIBRATE_MAG);
    sendMessage(msg);
}

void UAS::startGyroscopeCalibration()
{
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_CALIBRATE_GYRO);
    sendMessage(msg);
}

void UAS::startPressureCalibration()
{
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, MAV_COMP_ID_IMU, MAV_ACTION_CALIBRATE_PRESSURE);
    sendMessage(msg);
}

1120 1121
quint64 UAS::getUnixTime(quint64 time)
{
1122
    if (time == 0) {
LM's avatar
LM committed
1123 1124
//        qDebug() << "XNEW time:" <<QGC::groundTimeMilliseconds();
        return QGC::groundTimeMilliseconds();
1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
    }
    // Check if time is smaller than 40 years,
    // assuming no system without Unix timestamp
    // runs longer than 40 years continuously without
    // reboot. In worst case this will add/subtract the
    // communication delay between GCS and MAV,
    // it will never alter the timestamp in a safety
    // critical way.
    //
    // Calculation:
    // 40 years
    // 365 days
    // 24 hours
    // 60 minutes
    // 60 seconds
    // 1000 milliseconds
    // 1000 microseconds
1142
#ifndef _MSC_VER
1143
    else if (time < 1261440000000000LLU)
1144
#else
1145
    else if (time < 1261440000000000)
1146
#endif
1147
    {
LM's avatar
LM committed
1148
//        qDebug() << "GEN time:" << time/1000 + onboardTimeOffset;
1149
        if (onboardTimeOffset == 0) {
LM's avatar
LM committed
1150
            onboardTimeOffset = QGC::groundTimeMilliseconds() - time/1000;
1151
        }
1152
        return time/1000 + onboardTimeOffset;
1153
    } else {
1154 1155
        // Time is not zero and larger than 40 years -> has to be
        // a Unix epoch timestamp. Do nothing.
1156
        return time/1000;
1157 1158 1159
    }
}

1160 1161
QList<QString> UAS::getParameterNames(int component)
{
1162
    if (parameters.contains(component)) {
1163
        return parameters.value(component)->keys();
1164
    } else {
1165 1166 1167 1168 1169 1170 1171 1172 1173
        return QList<QString>();
    }
}

QList<int> UAS::getComponentIds()
{
    return parameters.keys();
}

pixhawk's avatar
pixhawk committed
1174
void UAS::setMode(int mode)
pixhawk's avatar
pixhawk committed
1175
{
1176
    if ((uint8_t)mode >= MAV_MODE_LOCKED && (uint8_t)mode <= MAV_MODE_RC_TRAINING) {
1177
        //this->mode = mode; //no call assignament, update receive message from UAS
pixhawk's avatar
pixhawk committed
1178
        mavlink_message_t msg;
lm's avatar
lm committed
1179
        mavlink_msg_set_mode_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, (uint8_t)uasId, (uint8_t)mode);
pixhawk's avatar
pixhawk committed
1180
        sendMessage(msg);
lm's avatar
lm committed
1181
        qDebug() << "SENDING REQUEST TO SET MODE TO SYSTEM" << uasId << ", REQUEST TO SET MODE " << (uint8_t)mode;
1182
    } else {
1183 1184
        qDebug() << "uas Mode not assign: " << mode;
    }
pixhawk's avatar
pixhawk committed
1185 1186 1187 1188 1189
}

void UAS::sendMessage(mavlink_message_t message)
{
    // Emit message on all links that are currently connected
1190 1191
    foreach (LinkInterface* link, *links) {
        if (link) {
1192
            sendMessage(link, message);
1193
        } else {
1194 1195 1196
            // Remove from list
            links->removeAt(links->indexOf(link));
        }
pixhawk's avatar
pixhawk committed
1197 1198 1199
    }
}

1200 1201 1202 1203
void UAS::forwardMessage(mavlink_message_t message)
{
    // Emit message on all links that are currently connected
    QList<LinkInterface*>link_list = LinkManager::instance()->getLinksForProtocol(mavlink);
1204

1205 1206
    foreach(LinkInterface* link, link_list) {
        if (link) {
1207
            SerialLink* serial = dynamic_cast<SerialLink*>(link);
1208
            if(serial != 0) {
1209

1210 1211
                for(int i=0; i<links->size(); i++) {
                    if(serial != links->at(i)) {
1212
                        qDebug()<<"Antenna tracking: Forwarding Over link: "<<serial->getName()<<" "<<serial;
1213 1214
                        sendMessage(serial, message);
                    }
1215 1216 1217 1218 1219 1220
                }
            }
        }
    }
}

pixhawk's avatar
pixhawk committed
1221 1222
void UAS::sendMessage(LinkInterface* link, mavlink_message_t message)
{
1223
    if(!link) return;
pixhawk's avatar
pixhawk committed
1224 1225 1226
    // Create buffer
    uint8_t buffer[MAVLINK_MAX_PACKET_LEN];
    // Write message into buffer, prepending start sign
pixhawk's avatar
pixhawk committed
1227
    int len = mavlink_msg_to_send_buffer(buffer, &message);
1228
    mavlink_finalize_message_chan(&message, mavlink->getSystemId(), mavlink->getComponentId(), link->getId(), message.len);
pixhawk's avatar
pixhawk committed
1229
    // If link is connected
1230
    if (link->isConnected()) {
pixhawk's avatar
pixhawk committed
1231 1232 1233 1234 1235 1236 1237 1238
        // Send the portion of the buffer now occupied by the message
        link->writeBytes((const char*)buffer, len);
    }
}

/**
 * @param value battery voltage
 */
1239
float UAS::filterVoltage(float value) const
pixhawk's avatar
pixhawk committed
1240
{
1241
    return lpVoltage * 0.7f + value * 0.3f;
pixhawk's avatar
pixhawk committed
1242 1243
}

1244 1245
QString UAS::getNavModeText(int mode)
{
1246
    switch (mode) {
1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278
    case MAV_NAV_GROUNDED:
        return QString("GROUNDED");
        break;
    case MAV_NAV_HOLD:
        return QString("HOLD");
        break;
    case MAV_NAV_LANDING:
        return QString("LANDING");
        break;
    case MAV_NAV_LIFTOFF:
        return QString("LIFTOFF");
        break;
    case MAV_NAV_LOITER:
        return QString("LOITER");
        break;
    case MAV_NAV_LOST:
        return QString("LOST");
        break;
    case MAV_NAV_RETURNING:
        return QString("RETURNING");
        break;
    case MAV_NAV_VECTOR:
        return QString("VECTOR");
        break;
    case MAV_NAV_WAYPOINT:
        return QString("WAYPOINT");
        break;
    default:
        return QString("UNKNOWN");
    }
}

pixhawk's avatar
pixhawk committed
1279 1280
void UAS::getStatusForCode(int statusCode, QString& uasState, QString& stateDescription)
{
1281
    switch (statusCode) {
lm's avatar
lm committed
1282
    case MAV_STATE_UNINIT:
pixhawk's avatar
pixhawk committed
1283
        uasState = tr("UNINIT");
1284
        stateDescription = tr("Unitialized, booting up.");
pixhawk's avatar
pixhawk committed
1285
        break;
lm's avatar
lm committed
1286
    case MAV_STATE_BOOT:
pixhawk's avatar
pixhawk committed
1287
        uasState = tr("BOOT");
1288
        stateDescription = tr("Booting system, please wait.");
pixhawk's avatar
pixhawk committed
1289
        break;
lm's avatar
lm committed
1290
    case MAV_STATE_CALIBRATING:
pixhawk's avatar
pixhawk committed
1291
        uasState = tr("CALIBRATING");
1292
        stateDescription = tr("Calibrating sensors, please wait.");
pixhawk's avatar
pixhawk committed
1293
        break;
lm's avatar
lm committed
1294
    case MAV_STATE_ACTIVE:
pixhawk's avatar
pixhawk committed
1295
        uasState = tr("ACTIVE");
1296
        stateDescription = tr("Active, normal operation.");
pixhawk's avatar
pixhawk committed
1297
        break;
lm's avatar
lm committed
1298 1299
    case MAV_STATE_STANDBY:
        uasState = tr("STANDBY");
1300
        stateDescription = tr("Standby mode, ready for liftoff.");
lm's avatar
lm committed
1301 1302
        break;
    case MAV_STATE_CRITICAL:
pixhawk's avatar
pixhawk committed
1303
        uasState = tr("CRITICAL");
1304
        stateDescription = tr("FAILURE: Continuing operation.");
pixhawk's avatar
pixhawk committed
1305
        break;
lm's avatar
lm committed
1306
    case MAV_STATE_EMERGENCY:
pixhawk's avatar
pixhawk committed
1307
        uasState = tr("EMERGENCY");
1308
        stateDescription = tr("EMERGENCY: Land Immediately!");
pixhawk's avatar
pixhawk committed
1309
        break;
James Goppert's avatar
James Goppert committed
1310
        //case MAV_STATE_HILSIM:
James Goppert's avatar
James Goppert committed
1311 1312 1313
        //uasState = tr("HIL SIM");
        //stateDescription = tr("HIL Simulation, Sensors read from SIM");
        //break;
1314

lm's avatar
lm committed
1315
    case MAV_STATE_POWEROFF:
pixhawk's avatar
pixhawk committed
1316
        uasState = tr("SHUTDOWN");
1317
        stateDescription = tr("Powering off system.");
pixhawk's avatar
pixhawk committed
1318
        break;
1319

lm's avatar
lm committed
1320
    default:
pixhawk's avatar
pixhawk committed
1321
        uasState = tr("UNKNOWN");
1322
        stateDescription = tr("Unknown system state");
pixhawk's avatar
pixhawk committed
1323 1324 1325 1326
        break;
    }
}

1327 1328
QImage UAS::getImage()
{
1329 1330
    #ifdef MAVLINK_ENABLED_PIXHAWK
    image.loadFromData(imageRecBuffer);
1331 1332
    // Restart statemachine
    imagePacketsArrived = 0;
1333
    return image;
1334 1335
    #endif

1336 1337 1338 1339
}

void UAS::requestImage()
{
1340
#ifdef MAVLINK_ENABLED_PIXHAWK
1341 1342
    qDebug() << "trying to get an image from the uas...";

1343 1344 1345
    // check if there is already an image transmission going on
    if (imagePacketsArrived == 0)
    {
1346 1347 1348 1349
        mavlink_message_t msg;
        mavlink_msg_data_transmission_handshake_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, DATA_TYPE_JPEG_IMAGE, 0, 0, 0, 50);
        sendMessage(msg);
    }
1350
#endif
1351
}
pixhawk's avatar
pixhawk committed
1352 1353 1354 1355 1356 1357 1358 1359 1360


/* MANAGEMENT */

/*
 *
 * @return The uptime in milliseconds
 *
 **/
1361
quint64 UAS::getUptime() const
pixhawk's avatar
pixhawk committed
1362 1363 1364 1365 1366 1367 1368 1369
{
    if(startTime == 0) {
        return 0;
    } else {
        return MG::TIME::getGroundTimeNow() - startTime;
    }
}

1370
int UAS::getCommunicationStatus() const
pixhawk's avatar
pixhawk committed
1371 1372 1373 1374
{
    return commStatus;
}

1375 1376 1377
void UAS::requestParameters()
{
    mavlink_message_t msg;
1378
    mavlink_msg_param_request_list_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), 25);
1379 1380
    // Send message twice to increase chance of reception
    sendMessage(msg);
1381 1382
}

1383
void UAS::writeParametersToStorage()
1384
{
1385 1386 1387
    mavlink_message_t msg;
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(),MAV_COMP_ID_IMU, (uint8_t)MAV_ACTION_STORAGE_WRITE);
unknown's avatar
unknown committed
1388
    //mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(),(uint8_t)MAV_ACTION_STORAGE_WRITE);
1389 1390 1391 1392 1393 1394 1395 1396 1397
    sendMessage(msg);
}

void UAS::readParametersFromStorage()
{
    mavlink_message_t msg;
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,(uint8_t)MAV_ACTION_STORAGE_READ);
    sendMessage(msg);
lm's avatar
lm committed
1398 1399
}

1400
void UAS::enableAllDataTransmission(int rate)
lm's avatar
lm committed
1401 1402
{
    // Buffers to write data to
1403
    mavlink_message_t msg;
1404
    mavlink_request_data_stream_t stream;
lm's avatar
lm committed
1405 1406
    // Select the message to request from now on
    // 0 is a magic ID and will enable/disable the standard message set except for heartbeat
1407
    stream.req_stream_id = MAV_DATA_STREAM_ALL;
lm's avatar
lm committed
1408 1409
    // Select the update rate in Hz the message should be send
    // All messages will be send with their default rate
1410 1411
    // TODO: use 0 to turn off and get rid of enable/disable? will require
    //  a different magic flag for turning on defaults, possibly something really high like 1111 ?
lm's avatar
lm committed
1412 1413
    stream.req_message_rate = 0;
    // Start / stop the message
1414
    stream.start_stop = (rate) ? 1 : 0;
lm's avatar
lm committed
1415 1416 1417 1418 1419
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
1420
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1421 1422
    // Send message twice to increase chance of reception
    sendMessage(msg);
lm's avatar
lm committed
1423 1424 1425
    sendMessage(msg);
}

1426
void UAS::enableRawSensorDataTransmission(int rate)
lm's avatar
lm committed
1427 1428 1429
{
    // Buffers to write data to
    mavlink_message_t msg;
1430
    mavlink_request_data_stream_t stream;
lm's avatar
lm committed
1431
    // Select the message to request from now on
1432
    stream.req_stream_id = MAV_DATA_STREAM_RAW_SENSORS;
lm's avatar
lm committed
1433
    // Select the update rate in Hz the message should be send
1434
    stream.req_message_rate = rate;
lm's avatar
lm committed
1435
    // Start / stop the message
1436
    stream.start_stop = (rate) ? 1 : 0;
lm's avatar
lm committed
1437 1438 1439 1440 1441
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
1442
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1443 1444
    // Send message twice to increase chance of reception
    sendMessage(msg);
lm's avatar
lm committed
1445 1446 1447
    sendMessage(msg);
}

1448
void UAS::enableExtendedSystemStatusTransmission(int rate)
lm's avatar
lm committed
1449
{
1450 1451 1452 1453
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1454
    stream.req_stream_id = MAV_DATA_STREAM_EXTENDED_STATUS;
1455
    // Select the update rate in Hz the message should be send
1456
    stream.req_message_rate = rate;
1457
    // Start / stop the message
1458
    stream.start_stop = (rate) ? 1 : 0;
1459 1460 1461 1462 1463 1464
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1465 1466
    // Send message twice to increase chance of reception
    sendMessage(msg);
1467
    sendMessage(msg);
lm's avatar
lm committed
1468
}
1469

1470
void UAS::enableRCChannelDataTransmission(int rate)
lm's avatar
lm committed
1471
{
1472 1473 1474 1475 1476
#if defined(MAVLINK_ENABLED_UALBERTA_MESSAGES)
    mavlink_message_t msg;
    mavlink_msg_request_rc_channels_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, enabled);
    sendMessage(msg);
#else
1477 1478 1479
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1480
    stream.req_stream_id = MAV_DATA_STREAM_RC_CHANNELS;
1481
    // Select the update rate in Hz the message should be send
1482
    stream.req_message_rate = rate;
1483
    // Start / stop the message
1484
    stream.start_stop = (rate) ? 1 : 0;
1485 1486 1487 1488 1489 1490
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1491 1492
    // Send message twice to increase chance of reception
    sendMessage(msg);
1493
    sendMessage(msg);
1494
#endif
lm's avatar
lm committed
1495 1496
}

1497
void UAS::enableRawControllerDataTransmission(int rate)
lm's avatar
lm committed
1498
{
1499 1500 1501 1502
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1503
    stream.req_stream_id = MAV_DATA_STREAM_RAW_CONTROLLER;
1504
    // Select the update rate in Hz the message should be send
1505
    stream.req_message_rate = rate;
1506
    // Start / stop the message
1507
    stream.start_stop = (rate) ? 1 : 0;
1508 1509 1510 1511 1512 1513
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1514 1515
    // Send message twice to increase chance of reception
    sendMessage(msg);
1516
    sendMessage(msg);
lm's avatar
lm committed
1517 1518
}

lm's avatar
lm committed
1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
//void UAS::enableRawSensorFusionTransmission(int rate)
//{
//    // Buffers to write data to
//    mavlink_message_t msg;
//    mavlink_request_data_stream_t stream;
//    // Select the message to request from now on
//    stream.req_stream_id = MAV_DATA_STREAM_RAW_SENSOR_FUSION;
//    // Select the update rate in Hz the message should be send
//    stream.req_message_rate = rate;
//    // Start / stop the message
//    stream.start_stop = (rate) ? 1 : 0;
//    // The system which should take this command
//    stream.target_system = uasId;
//    // The component / subsystem which should take this command
//    stream.target_component = 0;
//    // Encode and send the message
//    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
//    // Send message twice to increase chance of reception
//    sendMessage(msg);
//    sendMessage(msg);
//}
1540

1541
void UAS::enablePositionTransmission(int rate)
pixhawk's avatar
pixhawk committed
1542 1543 1544 1545 1546
{
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1547
    stream.req_stream_id = MAV_DATA_STREAM_POSITION;
pixhawk's avatar
pixhawk committed
1548
    // Select the update rate in Hz the message should be send
1549
    stream.req_message_rate = rate;
pixhawk's avatar
pixhawk committed
1550
    // Start / stop the message
1551
    stream.start_stop = (rate) ? 1 : 0;
pixhawk's avatar
pixhawk committed
1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
}

1563
void UAS::enableExtra1Transmission(int rate)
pixhawk's avatar
pixhawk committed
1564 1565 1566 1567 1568
{
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1569
    stream.req_stream_id = MAV_DATA_STREAM_EXTRA1;
pixhawk's avatar
pixhawk committed
1570
    // Select the update rate in Hz the message should be send
1571
    stream.req_message_rate = rate;
pixhawk's avatar
pixhawk committed
1572
    // Start / stop the message
1573
    stream.start_stop = (rate) ? 1 : 0;
pixhawk's avatar
pixhawk committed
1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
}

1585
void UAS::enableExtra2Transmission(int rate)
pixhawk's avatar
pixhawk committed
1586 1587 1588 1589 1590
{
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1591
    stream.req_stream_id = MAV_DATA_STREAM_EXTRA2;
pixhawk's avatar
pixhawk committed
1592
    // Select the update rate in Hz the message should be send
1593
    stream.req_message_rate = rate;
pixhawk's avatar
pixhawk committed
1594
    // Start / stop the message
1595
    stream.start_stop = (rate) ? 1 : 0;
pixhawk's avatar
pixhawk committed
1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
}

1607
void UAS::enableExtra3Transmission(int rate)
pixhawk's avatar
pixhawk committed
1608 1609 1610 1611 1612
{
    // Buffers to write data to
    mavlink_message_t msg;
    mavlink_request_data_stream_t stream;
    // Select the message to request from now on
1613
    stream.req_stream_id = MAV_DATA_STREAM_EXTRA3;
pixhawk's avatar
pixhawk committed
1614
    // Select the update rate in Hz the message should be send
1615
    stream.req_message_rate = rate;
1616
    // Start / stop the message
1617
    stream.start_stop = (rate) ? 1 : 0;
1618 1619 1620 1621 1622 1623
    // The system which should take this command
    stream.target_system = uasId;
    // The component / subsystem which should take this command
    stream.target_component = 0;
    // Encode and send the message
    mavlink_msg_request_data_stream_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &stream);
1624 1625
    // Send message twice to increase chance of reception
    sendMessage(msg);
1626
    sendMessage(msg);
1627 1628
}

lm's avatar
lm committed
1629 1630 1631 1632 1633 1634 1635
/**
 * Set a parameter value onboard
 *
 * @param component The component to set the parameter
 * @param id Name of the parameter
 * @param value Parameter value
 */
lm's avatar
lm committed
1636 1637
void UAS::setParameter(const int component, const QString& id, const float value)
{
1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650
    if (!id.isNull()) {
        mavlink_message_t msg;
        mavlink_param_set_t p;
        p.param_value = value;
        p.target_system = (uint8_t)uasId;
        p.target_component = (uint8_t)component;

        // Copy string into buffer, ensuring not to exceed the buffer size
        for (unsigned int i = 0; i < sizeof(p.param_id); i++) {
            // String characters
            if ((int)i < id.length() && i < (sizeof(p.param_id) - 1)) {
                p.param_id[i] = id.toAscii()[i];
            }
1651 1652 1653 1654 1655
//        // Null termination at end of string or end of buffer
//        else if ((int)i == id.length() || i == (sizeof(p.param_id) - 1))
//        {
//            p.param_id[i] = '\0';
//        }
1656 1657 1658 1659
            // Zero fill
            else {
                p.param_id[i] = 0;
            }
lm's avatar
lm committed
1660
        }
1661 1662
        mavlink_msg_param_set_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &p);
        sendMessage(msg);
lm's avatar
lm committed
1663
    }
1664 1665
}

1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677
void UAS::requestParameter(int component, int parameter)
{
    mavlink_message_t msg;
    mavlink_param_request_read_t read;
    read.param_index = parameter;
    read.target_system = uasId;
    read.target_component = component;
    mavlink_msg_param_request_read_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &read);
    sendMessage(msg);
    qDebug() << __FILE__ << __LINE__ << "REQUESTING PARAM RETRANSMISSION FROM COMPONENT" << component << "FOR PARAM ID" << parameter;
}

1678 1679 1680 1681
void UAS::setSystemType(int systemType)
{
    type = systemType;
    // If the airframe is still generic, change it to a close default type
1682 1683
    if (airframe == 0) {
        switch (systemType) {
1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
        case MAV_FIXED_WING:
            airframe = QGC_AIRFRAME_EASYSTAR;
            break;
        case MAV_QUADROTOR:
            airframe = QGC_AIRFRAME_MIKROKOPTER;
            break;
        }
    }
    emit systemSpecsChanged(uasId);
}

1695 1696 1697
void UAS::setUASName(const QString& name)
{
    this->name = name;
1698
    writeSettings();
1699
    emit nameChanged(name);
1700
    emit systemSpecsChanged(uasId);
1701 1702
}

lm's avatar
lm committed
1703 1704 1705 1706
void UAS::executeCommand(MAV_CMD command)
{
    mavlink_message_t msg;
    mavlink_command_t cmd;
1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731
    cmd.command = (uint8_t)command;
    cmd.confirmation = 0;
    cmd.param1 = 0.0f;
    cmd.param2 = 0.0f;
    cmd.param3 = 0.0f;
    cmd.param4 = 0.0f;
    cmd.target_system = uasId;
    cmd.target_component = 0;
    mavlink_msg_command_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &cmd);
    sendMessage(msg);
}

void UAS::executeCommand(MAV_CMD command, int confirmation, float param1, float param2, float param3, float param4, int component)
{
    mavlink_message_t msg;
    mavlink_command_t cmd;
    cmd.command = (uint8_t)command;
    cmd.confirmation = confirmation;
    cmd.param1 = param1;
    cmd.param2 = param2;
    cmd.param3 = param3;
    cmd.param4 = param4;
    cmd.target_system = uasId;
    cmd.target_component = component;
    mavlink_msg_command_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &cmd);
lm's avatar
lm committed
1732 1733 1734
    sendMessage(msg);
}

1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747
/**
 * Sets an action
 *
 **/
void UAS::setAction(MAV_ACTION action)
{
    mavlink_message_t msg;
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), 0, action);
    // Send message twice to increase chance that it reaches its goal
    sendMessage(msg);
    sendMessage(msg);
}

pixhawk's avatar
pixhawk committed
1748
/**
lm's avatar
lm committed
1749
 * Launches the system
pixhawk's avatar
pixhawk committed
1750 1751 1752 1753
 *
 **/
void UAS::launch()
{
1754
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1755
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1756
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), MAV_COMP_ID_IMU, (uint8_t)MAV_ACTION_TAKEOFF);
1757 1758 1759
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1760 1761 1762 1763 1764 1765 1766 1767
}

/**
 * Depending on the UAS, this might make the rotors of a helicopter spinning
 *
 **/
void UAS::enable_motors()
{
1768
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1769
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1770
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), MAV_COMP_ID_IMU, (uint8_t)MAV_ACTION_MOTORS_START);
1771 1772 1773
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1774 1775 1776 1777 1778 1779 1780 1781
}

/**
 * @warning Depending on the UAS, this might completely stop all motors.
 *
 **/
void UAS::disable_motors()
{
1782
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1783
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1784
    mavlink_msg_action_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, this->getUASID(), MAV_COMP_ID_IMU, (uint8_t)MAV_ACTION_MOTORS_STOP);
1785 1786 1787
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801
}

void UAS::setManualControlCommands(double roll, double pitch, double yaw, double thrust)
{
    // Scale values
    double rollPitchScaling = 0.2f;
    double yawScaling = 0.5f;
    double thrustScaling = 1.0f;

    manualRollAngle = roll * rollPitchScaling;
    manualPitchAngle = pitch * rollPitchScaling;
    manualYawAngle = yaw * yawScaling;
    manualThrust = thrust * thrustScaling;

1802
    if(mode == (int)MAV_MODE_MANUAL) {
1803 1804 1805 1806
        mavlink_message_t message;
        mavlink_msg_manual_control_pack(mavlink->getSystemId(), mavlink->getComponentId(), &message, this->uasId, (float)manualRollAngle, (float)manualPitchAngle, (float)manualYawAngle, (float)manualThrust, controlRollManual, controlPitchManual, controlYawManual, controlThrustManual);
        sendMessage(message);
        qDebug() << __FILE__ << __LINE__ << ": SENT MANUAL CONTROL MESSAGE: roll" << manualRollAngle << " pitch: " << manualPitchAngle << " yaw: " << manualYawAngle << " thrust: " << manualThrust;
pixhawk's avatar
pixhawk committed
1807

1808
        emit attitudeThrustSetPointChanged(this, roll, pitch, yaw, thrust, MG::TIME::getGroundTimeNow());
1809
    } else {
1810 1811
        qDebug() << "JOYSTICK/MANUAL CONTROL: IGNORING COMMANDS: Set mode to MANUAL to send joystick commands first";
    }
pixhawk's avatar
pixhawk committed
1812 1813
}

1814 1815 1816 1817 1818
int UAS::getSystemType()
{
    return this->type;
}

pixhawk's avatar
pixhawk committed
1819 1820
void UAS::receiveButton(int buttonIndex)
{
1821
    switch (buttonIndex) {
pixhawk's avatar
pixhawk committed
1822
    case 0:
pixhawk's avatar
pixhawk committed
1823

pixhawk's avatar
pixhawk committed
1824 1825
        break;
    case 1:
pixhawk's avatar
pixhawk committed
1826

pixhawk's avatar
pixhawk committed
1827 1828 1829 1830 1831
        break;
    default:

        break;
    }
1832
    //    qDebug() << __FILE__ << __LINE__ << ": Received button clicked signal (button # is: " << buttonIndex << "), UNIMPLEMENTED IN MAVLINK!";
pixhawk's avatar
pixhawk committed
1833 1834 1835

}

1836

1837
/*void UAS::requestWaypoints()
1838 1839 1840 1841 1842
{
//    mavlink_message_t msg;
//    mavlink_msg_waypoint_request_list_pack(mavlink->getSystemId(), mavlink->getComponentId(), &msg, uasId, 25);
//    // Send message twice to increase chance of reception
//    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1843 1844
    waypointManager.requestWaypoints();
    qDebug() << "UAS Request WPs";
1845 1846
}

pixhawk's avatar
pixhawk committed
1847 1848
void UAS::setWaypoint(Waypoint* wp)
{
1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866
//    mavlink_message_t msg;
//    mavlink_waypoint_set_t set;
//    set.id = wp->id;
//    //QString name = wp->name;
//    // FIXME Check if this works properly
//    //name.truncate(MAVLINK_MSG_WAYPOINT_SET_FIELD_NAME_LEN);
//    //strcpy((char*)set.name, name.toStdString().c_str());
//    set.autocontinue = wp->autocontinue;
//    set.target_component = 25; // FIXME
//    set.target_system = uasId;
//    set.active = wp->current;
//    set.x = wp->x;
//    set.y = wp->y;
//    set.z = wp->z;
//    set.yaw = wp->yaw;
//    mavlink_msg_waypoint_set_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &set);
//    // Send message twice to increase chance of reception
//    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1867 1868 1869 1870
}

void UAS::setWaypointActive(int id)
{
1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893
//    mavlink_message_t msg;
//    mavlink_waypoint_set_active_t active;
//    active.id = id;
//    active.target_system = uasId;
//    active.target_component = 25; // FIXME
//    mavlink_msg_waypoint_set_active_encode(mavlink->getSystemId(), mavlink->getComponentId(), &msg, &active);
//    // Send message twice to increase chance of reception
//    sendMessage(msg);
//    sendMessage(msg);
//    // TODO This should be not directly emitted, but rather being fed back from the UAS
//    emit waypointSelected(getUASID(), id);
}

void UAS::clearWaypointList()
{
//    mavlink_message_t msg;
//    // FIXME
//    mavlink_waypoint_clear_list_t clist;
//    clist.target_system = uasId;
//    clist.target_component = 25;  // FIXME
//    mavlink_msg_waypoint_clear_list_encode(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, &clist);
//    sendMessage(msg);
//    qDebug() << "UAS clears Waypoints!";
1894
}*/
pixhawk's avatar
pixhawk committed
1895 1896 1897 1898


void UAS::halt()
{
1899

1900
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1901
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1902
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU, (int)MAV_ACTION_HALT);
1903 1904 1905
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
1906

pixhawk's avatar
pixhawk committed
1907 1908 1909 1910
}

void UAS::go()
{
1911
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1912
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1913
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,  (int)MAV_ACTION_CONTINUE);
1914 1915 1916
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1917 1918 1919 1920 1921
}

/** Order the robot to return home / to land on the runway **/
void UAS::home()
{
1922
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1923
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1924
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,  (int)MAV_ACTION_RETURN);
1925 1926 1927
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1928 1929 1930 1931 1932 1933 1934 1935
}

/**
 * The MAV starts the emergency landing procedure. The behaviour depends on the onboard implementation
 * and might differ between systems.
 */
void UAS::emergencySTOP()
{
1936

1937
    mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1938
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
1939
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU, (int)MAV_ACTION_EMCY_LAND);
1940 1941 1942
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
pixhawk's avatar
pixhawk committed
1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965
}

/**
 * All systems are immediately shut down (e.g. the main power line is cut).
 * @warning This might lead to a crash
 *
 * The command will not be executed until emergencyKILLConfirm is issues immediately afterwards
 */
bool UAS::emergencyKILL()
{
    bool result = false;
    QMessageBox msgBox;
    msgBox.setIcon(QMessageBox::Critical);
    msgBox.setText("EMERGENCY: KILL ALL MOTORS ON UAS");
    msgBox.setInformativeText("Do you want to cut power on all systems?");
    msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::Cancel);
    msgBox.setDefaultButton(QMessageBox::Cancel);
    int ret = msgBox.exec();

    // Close the message box shortly after the click to prevent accidental clicks
    QTimer::singleShot(5000, &msgBox, SLOT(reject()));


1966
    if (ret == QMessageBox::Yes) {
1967
        mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
1968
        // TODO Replace MG System ID with static function call and allow to change ID in GUI
1969
        mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU, (int)MAV_ACTION_EMCY_KILL);
1970 1971 1972
        // Send message twice to increase chance of reception
        sendMessage(msg);
        sendMessage(msg);
pixhawk's avatar
pixhawk committed
1973 1974 1975 1976 1977
        result = true;
    }
    return result;
}

1978 1979
void UAS::startHil()
{
1980

James Goppert's avatar
James Goppert committed
1981 1982 1983 1984 1985 1986
    mavlink_message_t msg;
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,(int)MAV_ACTION_START_HILSIM);
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
1987 1988 1989

}

1990 1991
void UAS::stopHil()
{
1992

James Goppert's avatar
James Goppert committed
1993 1994 1995 1996 1997 1998
    mavlink_message_t msg;
    // TODO Replace MG System ID with static function call and allow to change ID in GUI
    mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,(int)MAV_ACTION_STOP_HILSIM);
    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
1999 2000 2001 2002

}


pixhawk's avatar
pixhawk committed
2003 2004 2005 2006 2007 2008 2009
void UAS::shutdown()
{
    bool result = false;
    QMessageBox msgBox;
    msgBox.setIcon(QMessageBox::Critical);
    msgBox.setText("Shutting down the UAS");
    msgBox.setInformativeText("Do you want to shut down the onboard computer?");
2010

pixhawk's avatar
pixhawk committed
2011 2012 2013 2014 2015 2016 2017 2018
    msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::Cancel);
    msgBox.setDefaultButton(QMessageBox::Cancel);
    int ret = msgBox.exec();

    // Close the message box shortly after the click to prevent accidental clicks
    QTimer::singleShot(5000, &msgBox, SLOT(reject()));


2019
    if (ret == QMessageBox::Yes) {
pixhawk's avatar
pixhawk committed
2020
        // If the active UAS is set, execute command
2021
        mavlink_message_t msg;
pixhawk's avatar
pixhawk committed
2022
        // TODO Replace MG System ID with static function call and allow to change ID in GUI
2023
        mavlink_msg_action_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, this->getUASID(), MAV_COMP_ID_IMU,(int)MAV_ACTION_SHUTDOWN);
2024 2025 2026
        // Send message twice to increase chance of reception
        sendMessage(msg);
        sendMessage(msg);
2027

pixhawk's avatar
pixhawk committed
2028 2029 2030 2031
        result = true;
    }
}

2032 2033
void UAS::setTargetPosition(float x, float y, float z, float yaw)
{
2034 2035 2036 2037 2038 2039
    mavlink_message_t msg;
    mavlink_msg_position_target_pack(MG::SYSTEM::ID, MG::SYSTEM::COMPID, &msg, x, y, z, yaw);

    // Send message twice to increase chance of reception
    sendMessage(msg);
    sendMessage(msg);
2040 2041
}

pixhawk's avatar
pixhawk committed
2042 2043 2044
/**
 * @return The name of this system as string in human-readable form
 */
2045
QString UAS::getUASName(void) const
pixhawk's avatar
pixhawk committed
2046 2047
{
    QString result;
2048
    if (name == "") {
pixhawk's avatar
pixhawk committed
2049
        result = tr("MAV ") + result.sprintf("%03d", getUASID());
2050
    } else {
pixhawk's avatar
pixhawk committed
2051 2052 2053 2054 2055 2056 2057
        result = name;
    }
    return result;
}

void UAS::addLink(LinkInterface* link)
{
2058
    if (!links->contains(link)) {
pixhawk's avatar
pixhawk committed
2059
        links->append(link);
2060
        connect(link, SIGNAL(destroyed(QObject*)), this, SLOT(removeLink(QObject*)));
pixhawk's avatar
pixhawk committed
2061 2062 2063
    }
}

2064 2065 2066
void UAS::removeLink(QObject* object)
{
    LinkInterface* link = dynamic_cast<LinkInterface*>(object);
2067
    if (link) {
2068 2069 2070 2071
        links->removeAt(links->indexOf(link));
    }
}

pixhawk's avatar
pixhawk committed
2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086
/**
 * @brief Get the links associated with this robot
 *
 **/
QList<LinkInterface*>* UAS::getLinks()
{
    return links;
}



void UAS::setBattery(BatteryType type, int cells)
{
    this->batteryType = type;
    this->cells = cells;
2087
    switch (batteryType) {
lm's avatar
lm committed
2088
    case NICD:
pixhawk's avatar
pixhawk committed
2089
        break;
lm's avatar
lm committed
2090
    case NIMH:
pixhawk's avatar
pixhawk committed
2091
        break;
lm's avatar
lm committed
2092
    case LIION:
pixhawk's avatar
pixhawk committed
2093
        break;
lm's avatar
lm committed
2094
    case LIPOLY:
pixhawk's avatar
pixhawk committed
2095 2096 2097
        fullVoltage = this->cells * UAS::lipoFull;
        emptyVoltage = this->cells * UAS::lipoEmpty;
        break;
lm's avatar
lm committed
2098
    case LIFE:
pixhawk's avatar
pixhawk committed
2099
        break;
lm's avatar
lm committed
2100
    case AGZN:
pixhawk's avatar
pixhawk committed
2101 2102 2103 2104
        break;
    }
}

2105 2106
void UAS::setBatterySpecs(const QString& specs)
{
2107
    if (specs.length() == 0 || specs.contains("%")) {
2108
        batteryRemainingEstimateEnabled = false;
2109
        bool ok;
2110 2111 2112
        QString percent = specs;
        percent = percent.remove("%");
        float temp = percent.toFloat(&ok);
2113
        if (ok) {
2114
            warnLevelPercent = temp;
2115
        } else {
2116 2117
            emit textMessageReceived(0, 0, 0, "Could not set battery options, format is wrong");
        }
2118
    } else {
2119 2120 2121 2122 2123
        batteryRemainingEstimateEnabled = true;
        QString stringList = specs;
        stringList = stringList.remove("V");
        stringList = stringList.remove("v");
        QStringList parts = stringList.split(",");
2124
        if (parts.length() == 3) {
2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135
            float temp;
            bool ok;
            // Get the empty voltage
            temp = parts.at(0).toFloat(&ok);
            if (ok) emptyVoltage = temp;
            // Get the warning voltage
            temp = parts.at(1).toFloat(&ok);
            if (ok) warnVoltage = temp;
            // Get the full voltage
            temp = parts.at(2).toFloat(&ok);
            if (ok) fullVoltage = temp;
2136
        } else {
2137 2138
            emit textMessageReceived(0, 0, 0, "Could not set battery options, format is wrong");
        }
2139 2140 2141 2142 2143
    }
}

QString UAS::getBatterySpecs()
{
2144
    if (batteryRemainingEstimateEnabled) {
2145
        return QString("%1V,%2V,%3V").arg(emptyVoltage).arg(warnVoltage).arg(fullVoltage);
2146
    } else {
2147 2148
        return QString("%1%").arg(warnLevelPercent);
    }
2149 2150
}

pixhawk's avatar
pixhawk committed
2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163
int UAS::calculateTimeRemaining()
{
    quint64 dt = MG::TIME::getGroundTimeNow() - startTime;
    double seconds = dt / 1000.0f;
    double voltDifference = startVoltage - currentVoltage;
    if (voltDifference <= 0) voltDifference = 0.00000000001f;
    double dischargePerSecond = voltDifference / seconds;
    int remaining = static_cast<int>((currentVoltage - emptyVoltage) / dischargePerSecond);
    // Can never be below 0
    if (remaining < 0) remaining = 0;
    return remaining;
}

lm's avatar
lm committed
2164 2165 2166
/**
 * @return charge level in percent - 0 - 100
 */
2167
float UAS::getChargeLevel()
pixhawk's avatar
pixhawk committed
2168
{
2169 2170
    if (batteryRemainingEstimateEnabled) {
        if (lpVoltage < emptyVoltage) {
2171
            chargeLevel = 0.0f;
2172
        } else if (lpVoltage > fullVoltage) {
2173
            chargeLevel = 100.0f;
2174
        } else {
2175 2176
            chargeLevel = 100.0f * ((lpVoltage - emptyVoltage)/(fullVoltage - emptyVoltage));
        }
2177 2178
    }
    return chargeLevel;
pixhawk's avatar
pixhawk committed
2179 2180
}

lm's avatar
lm committed
2181 2182
void UAS::startLowBattAlarm()
{
2183
    if (!lowBattAlarm) {
2184
        GAudioOutput::instance()->alert(tr("SYSTEM %1 HAS LOW BATTERY").arg(getUASName()));
2185
        QTimer::singleShot(2000, GAudioOutput::instance(), SLOT(startEmergency()));
lm's avatar
lm committed
2186 2187 2188 2189 2190 2191
        lowBattAlarm = true;
    }
}

void UAS::stopLowBattAlarm()
{
2192
    if (lowBattAlarm) {
lm's avatar
lm committed
2193 2194 2195 2196
        GAudioOutput::instance()->stopEmergency();
        lowBattAlarm = false;
    }
}