MissionController.cc 39.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
/*=====================================================================

QGroundControl Open Source Ground Control Station

(c) 2009, 2015 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>

This file is part of the QGROUNDCONTROL project

    QGROUNDCONTROL is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    QGROUNDCONTROL is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.

======================================================================*/

#include "MissionController.h"
#include "MultiVehicleManager.h"
#include "MissionManager.h"
#include "CoordinateVector.h"
28
#include "FirmwarePlugin.h"
29
#include "QGCApplication.h"
30 31
#include "SimpleMissionItem.h"
#include "ComplexMissionItem.h"
32
#include "JsonHelper.h"
33

34 35 36 37
#ifndef __mobile__
#include "QGCFileDialog.h"
#endif

38 39
QGC_LOGGING_CATEGORY(MissionControllerLog, "MissionControllerLog")

40 41
const char* MissionController::jsonSimpleItemsKey = "items";

42 43 44 45
const char* MissionController::_settingsGroup =                 "MissionController";
const char* MissionController::_jsonVersionKey =                "version";
const char* MissionController::_jsonGroundStationKey =          "groundStation";
const char* MissionController::_jsonMavAutopilotKey =           "MAV_AUTOPILOT";
46
const char* MissionController::_jsonComplexItemsKey =           "complexItems";
47
const char* MissionController::_jsonPlannedHomePositionKey =    "plannedHomePosition";
48 49 50

MissionController::MissionController(QObject *parent)
    : QObject(parent)
51
    , _editMode(false)
52 53
    , _visualItems(NULL)
    , _complexItems(NULL)
54
    , _activeVehicle(NULL)
55
    , _autoSync(false)
56
    , _firstItemsFromVehicle(false)
57 58
    , _missionItemsRequested(false)
    , _queuedSend(false)
59
{
60 61 62 63 64

}

MissionController::~MissionController()
{
Don Gagne's avatar
Don Gagne committed
65

66 67 68 69
}

void MissionController::start(bool editMode)
{
70 71
    qCDebug(MissionControllerLog) << "start editMode" << editMode;

72 73
    _editMode = editMode;

74
    MultiVehicleManager* multiVehicleMgr = qgcApp()->toolbox()->multiVehicleManager();
75

76
    connect(multiVehicleMgr, &MultiVehicleManager::activeVehicleChanged, this, &MissionController::_activeVehicleChanged);
77
    _activeVehicleChanged(multiVehicleMgr->activeVehicle());
78

79
    // We start with an empty mission
80 81 82
    _visualItems = new QmlObjectListModel(this);
    _addPlannedHomePosition(_visualItems, false /* addToCenter */);
    _initAllVisualItems();
83 84
}

85
// Called when new mission items have completed downloading from Vehicle
86
void MissionController::_newMissionItemsAvailableFromVehicle(void)
87
{
88 89
    qCDebug(MissionControllerLog) << "_newMissionItemsAvailableFromVehicle";

90
    if (!_editMode || _missionItemsRequested || _visualItems->count() == 1) {
91 92 93 94 95
        // Fly Mode:
        //      - Always accepts new items fromthe vehicle so Fly view is kept up to date
        // Edit Mode:
        //      - Either a load from vehicle was manually requested or
        //      - The initial automatic load from a vehicle completed and the current editor it empty
96

97 98
        QmlObjectListModel* newControllerMissionItems = new QmlObjectListModel(this);
        const QList<MissionItem*>& newMissionItems = _activeVehicle->missionManager()->missionItems();
99

100 101 102 103 104 105 106 107 108 109 110 111
        qCDebug(MissionControllerLog) << "loading from vehicle: count"<< _visualItems->count();
        foreach(const MissionItem* missionItem, newMissionItems) {
            newControllerMissionItems->append(new SimpleMissionItem(_activeVehicle, *missionItem, this));
        }

        _deinitAllVisualItems();

        _visualItems->deleteListAndContents();
        _visualItems = newControllerMissionItems;

        if (!_activeVehicle->firmwarePlugin()->sendHomePositionToVehicle() || _visualItems->count() == 0) {
            _addPlannedHomePosition(_visualItems,true /* addToCenter */);
112 113
        }

114
        _missionItemsRequested = false;
115

116
        _initAllVisualItems();
117
        emit newItemsFromVehicle();
118 119 120
    }
}

121
void MissionController::getMissionItems(void)
122
{
123
    Vehicle* activeVehicle = qgcApp()->toolbox()->multiVehicleManager()->activeVehicle();
124

125 126 127 128 129 130 131 132
    if (activeVehicle) {
        _missionItemsRequested = true;
        activeVehicle->missionManager()->requestMissionItems();
    }
}

void MissionController::sendMissionItems(void)
{
133 134
    if (_activeVehicle) {
        // Convert to MissionItems so we can send to vehicle
135

136 137 138 139 140 141 142 143 144 145
        QList<MissionItem*> missionItems;

        int sequenceNumber = 0;
        for (int i=0; i<_visualItems->count(); i++) {
            VisualMissionItem* visualItem = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
            if (visualItem->isSimpleItem()) {
                MissionItem& missionItem = qobject_cast<SimpleMissionItem*>(visualItem)->missionItem();
                missionItem.setSequenceNumber(sequenceNumber++);
                missionItems.append(&missionItem);
            } else {
146 147 148
                ComplexMissionItem* complexItem = qobject_cast<ComplexMissionItem*>(visualItem);
                for (int j=0; j<complexItem->missionItems().count(); j++) {
                    MissionItem* missionItem = complexItem->missionItems()[i];
149 150 151 152 153 154 155 156
                    missionItem->setSequenceNumber(sequenceNumber++);
                    missionItems.append(missionItem);
                }
            }
        }

        _activeVehicle->missionManager()->writeMissionItems(missionItems);
        _visualItems->setDirty(false);
157 158
    }
}
159

160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
int MissionController::_nextSequenceNumber(void)
{
    if (_visualItems->count() == 0) {
        qWarning() << "Internal error: Empty visual item list";
        return 0;
    } else {
        VisualMissionItem* lastItem = qobject_cast<VisualMissionItem*>(_visualItems->get(_visualItems->count() - 1));

        if (lastItem->isSimpleItem()) {
            return lastItem->sequenceNumber() + 1;
        } else {
            return qobject_cast<ComplexMissionItem*>(lastItem)->lastSequenceNumber() + 1;
        }
    }
}

176
int MissionController::insertSimpleMissionItem(QGeoCoordinate coordinate, int i)
177
{
178
    SimpleMissionItem * newItem = new SimpleMissionItem(_activeVehicle, this);
179
    newItem->setSequenceNumber(_nextSequenceNumber());
180
    newItem->setCoordinate(coordinate);
181 182 183
    newItem->setCommand(MavlinkQmlSingleton::MAV_CMD_NAV_WAYPOINT);
    _initVisualItem(newItem);
    if (_visualItems->count() == 1) {
184 185
        newItem->setCommand(MavlinkQmlSingleton::MAV_CMD_NAV_TAKEOFF);
    }
186
    newItem->setDefaultsForCommand();
Don Gagne's avatar
Don Gagne committed
187
    if ((MAV_CMD)newItem->command() == MAV_CMD_NAV_WAYPOINT) {
188 189 190
        double lastValue;

        if (_findLastAcceptanceRadius(&lastValue)) {
191
            newItem->missionItem().setParam2(lastValue);
192 193
        }
        if (_findLastAltitude(&lastValue)) {
194
            newItem->missionItem().setParam7(lastValue);
195
        }
196
    }
197
    _visualItems->insert(i, newItem);
198 199 200

    _recalcAll();

201
    return _visualItems->count() - 1;
202 203
}

204 205
int MissionController::insertComplexMissionItem(QGeoCoordinate coordinate, int i)
{
206
    ComplexMissionItem* newItem = new ComplexMissionItem(_activeVehicle, this);
207
    newItem->setSequenceNumber(_nextSequenceNumber());
208
    newItem->setCoordinate(coordinate);
209
    _initVisualItem(newItem);
210

211 212
    _visualItems->insert(i, newItem);
    _complexItems->append(newItem);
213 214 215

    _recalcAll();

216
    return _visualItems->count() - 1;
217 218
}

219 220
void MissionController::removeMissionItem(int index)
{
221
    VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->removeAt(index));
222

223 224 225 226 227 228 229 230 231
    _deinitVisualItem(item);
    if (!item->isSimpleItem()) {
        ComplexMissionItem* complexItem = qobject_cast<ComplexMissionItem*>(_complexItems->removeOne(item));
        if (complexItem) {
            complexItem->deleteLater();
        } else {
            qWarning() << "Complex item missing";
        }
    }
232
    item->deleteLater();
233 234 235 236 237

    _recalcAll();

    // Set the new current item

238
    if (index >= _visualItems->count()) {
239 240
        index--;
    }
241 242
    for (int i=0; i<_visualItems->count(); i++) {
        MissionItem* item =  qobject_cast<MissionItem*>(_visualItems->get(i));
243 244
        item->setIsCurrentItem(i == index);
    }
245
    _visualItems->setDirty(true);
246 247
}

248 249
void MissionController::removeAllMissionItems(void)
{
250 251 252 253 254 255
    if (_visualItems) {
        _deinitAllVisualItems();
        _visualItems->deleteListAndContents();
        _visualItems = new QmlObjectListModel(this);
        _addPlannedHomePosition(_visualItems, false /* addToCenter */);
        _initAllVisualItems();
256
    }
257
}
258

259
bool MissionController::_loadJsonMissionFile(const QByteArray& bytes, QmlObjectListModel* visualItems, QmlObjectListModel* complexItems, QString& errorString)
260 261 262 263 264 265 266 267 268 269 270
{
    QJsonParseError jsonParseError;
    QJsonDocument   jsonDoc(QJsonDocument::fromJson(bytes, &jsonParseError));

    if (jsonParseError.error != QJsonParseError::NoError) {
        errorString = jsonParseError.errorString();
        return false;
    }

    QJsonObject json = jsonDoc.object();

271 272 273 274 275 276 277 278 279 280 281 282 283
    // Check for required keys
    QStringList requiredKeys;
    requiredKeys << _jsonVersionKey << _jsonPlannedHomePositionKey;
    if (!JsonHelper::validateRequiredKeys(json, requiredKeys, errorString)) {
        return false;
    }

    // Validate base key types
    QStringList             keyList;
    QList<QJsonValue::Type> typeList;
    keyList << jsonSimpleItemsKey << _jsonVersionKey << _jsonGroundStationKey << _jsonMavAutopilotKey << _jsonComplexItemsKey << _jsonPlannedHomePositionKey;
    typeList << QJsonValue::Array << QJsonValue::String << QJsonValue::String << QJsonValue::Double << QJsonValue::Array << QJsonValue::Object;
    if (!JsonHelper::validateKeyTypes(json, keyList, typeList, errorString)) {
284 285
        return false;
    }
286 287

    // Version check
288 289 290 291 292
    if (json[_jsonVersionKey].toString() != "1.0") {
        errorString = QStringLiteral("QGroundControl does not support this file version");
        return false;
    }

293 294 295 296 297
    // Read complex items
    QJsonArray complexArray(json[_jsonComplexItemsKey].toArray());
    qCDebug(MissionControllerLog) << "Json load: complex item count" << complexArray.count();
    for (int i=0; i<complexArray.count(); i++) {
        const QJsonValue& itemValue = complexArray[i];
298

299 300 301 302 303 304 305 306 307 308 309
        if (!itemValue.isObject()) {
            errorString = QStringLiteral("Mission item is not an object");
            return false;
        }

        ComplexMissionItem* item = new ComplexMissionItem(_activeVehicle, this);
        if (item->load(itemValue.toObject(), errorString)) {
            qCDebug(MissionControllerLog) << "Json load: complex item start:stop" << item->sequenceNumber() << item->lastSequenceNumber();
            complexItems->append(item);
        } else {
            return false;
310
        }
311
    }
312

313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
    // Read simple items, interspersing complex items into the full list

    int nextSimpleItemIndex= 0;
    int nextComplexItemIndex= 0;
    int nextSequenceNumber = 1; // Start with 1 since home is in 0
    QJsonArray itemArray(json[jsonSimpleItemsKey].toArray());

    qCDebug(MissionControllerLog) << "Json load: simple item loop start simpleItemCount:ComplexItemCount" << itemArray.count() << complexItems->count();
    do {
        qCDebug(MissionControllerLog) << "Json load: simple item loop nextSimpleItemIndex:nextComplexItemIndex:nextSequenceNumber" << nextSimpleItemIndex << nextComplexItemIndex << nextSequenceNumber;

        // If there is a complex item that should be next in sequence add it in
        if (nextComplexItemIndex < complexItems->count()) {
            ComplexMissionItem* complexItem = qobject_cast<ComplexMissionItem*>(complexItems->get(nextComplexItemIndex));

            if (complexItem->sequenceNumber() == nextSequenceNumber) {
                qCDebug(MissionControllerLog) << "Json load: injecting complex item expectedSequence:actualSequence:" << nextSequenceNumber << complexItem->sequenceNumber();
                visualItems->append(complexItem);
                nextSequenceNumber = complexItem->lastSequenceNumber() + 1;
                nextComplexItemIndex++;
                continue;
            }
        }

        // Add the next available simple item
        if (nextSimpleItemIndex < itemArray.count()) {
            const QJsonValue& itemValue = itemArray[nextSimpleItemIndex++];

341 342 343 344 345
            if (!itemValue.isObject()) {
                errorString = QStringLiteral("Mission item is not an object");
                return false;
            }

346
            SimpleMissionItem* item = new SimpleMissionItem(_activeVehicle, this);
347
            if (item->load(itemValue.toObject(), errorString)) {
348 349
                qCDebug(MissionControllerLog) << "Json load: adding simple item expectedSequence:actualSequence" << nextSequenceNumber << item->sequenceNumber();
                visualItems->append(item);
350 351 352
            } else {
                return false;
            }
353 354

            nextSequenceNumber++;
355
        }
356
    } while (nextSimpleItemIndex < itemArray.count() | nextComplexItemIndex < complexItems->count());
357 358

    if (json.contains(_jsonPlannedHomePositionKey)) {
359
        SimpleMissionItem* item = new SimpleMissionItem(_activeVehicle, this);
360 361

        if (item->load(json[_jsonPlannedHomePositionKey].toObject(), errorString)) {
362
            visualItems->insert(0, item);
363 364 365 366
        } else {
            return false;
        }
    } else {
367
        _addPlannedHomePosition(visualItems, true /* addToCenter */);
368 369 370 371 372
    }

    return true;
}

373
bool MissionController::_loadTextMissionFile(QTextStream& stream, QmlObjectListModel* visualItems, QString& errorString)
374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
{
    bool addPlannedHomePosition = false;

    QString firstLine = stream.readLine();
    const QStringList& version = firstLine.split(" ");

    bool versionOk = false;
    if (version.size() == 3 && version[0] == "QGC" && version[1] == "WPL") {
        if (version[2] == "110") {
            // ArduPilot file, planned home position is already in position 0
            versionOk = true;
        } else if (version[2] == "120") {
            // Old QGC file, no planned home position
            versionOk = true;
            addPlannedHomePosition = true;
        }
    }

    if (versionOk) {
        while (!stream.atEnd()) {
394
            SimpleMissionItem* item = new SimpleMissionItem(_activeVehicle, this);
395 396

            if (item->load(stream)) {
397
                visualItems->append(item);
398 399 400 401 402 403 404 405 406 407
            } else {
                errorString = QStringLiteral("The mission file is corrupted.");
                return false;
            }
        }
    } else {
        errorString = QStringLiteral("The mission file is not compatible with this version of QGroundControl.");
        return false;
    }

408 409
    if (addPlannedHomePosition || visualItems->count() == 0) {
        _addPlannedHomePosition(visualItems, true /* addToCenter */);
Don Gagne's avatar
Don Gagne committed
410

411 412 413 414 415
        // Update sequence numbers in DO_JUMP commands to take into account added home position in index 0
        for (int i=1; i<visualItems->count(); i++) {
            SimpleMissionItem* item = qobject_cast<SimpleMissionItem*>(visualItems->get(i));
            if (item && item->command() == MavlinkQmlSingleton::MAV_CMD_DO_JUMP) {
                item->missionItem().setParam1((int)item->missionItem().param1() + 1);
Don Gagne's avatar
Don Gagne committed
416 417
            }
        }
418 419 420
    }

    return true;
421 422
}

423
void MissionController::_loadMissionFromFile(const QString& filename)
424 425 426 427 428 429 430
{
    QString errorString;

    if (filename.isEmpty()) {
        return;
    }

431 432
    QmlObjectListModel* newVisualItems = new QmlObjectListModel(this);
    QmlObjectListModel* newComplexItems = new QmlObjectListModel(this);
433 434 435 436 437 438

    QFile file(filename);

    if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        errorString = file.errorString();
    } else {
439 440
        QByteArray  bytes = file.readAll();
        QTextStream stream(&bytes);
441

442 443 444
        QString firstLine = stream.readLine();
        if (firstLine.contains(QRegExp("QGC.*WPL"))) {
            stream.seek(0);
445
            _loadTextMissionFile(stream, newVisualItems, errorString);
446
        } else {
447
            _loadJsonMissionFile(bytes, newVisualItems, newComplexItems, errorString);
448 449
        }
    }
450

451
    if (!errorString.isEmpty()) {
452 453 454 455 456 457 458 459 460
        for (int i=0; i<newVisualItems->count(); i++) {
            newVisualItems->get(i)->deleteLater();
        }
        for (int i=0; i<newComplexItems->count(); i++) {
            newComplexItems->get(i)->deleteLater();
        }
        delete newVisualItems;
        delete newComplexItems;

461
        qgcApp()->showMessage(errorString);
462
        return;
463 464
    }

465 466 467
    if (_visualItems) {
        _deinitAllVisualItems();
        _visualItems->deleteListAndContents();
468
    }
469 470 471 472 473 474 475
    if (_complexItems) {
        _complexItems->deleteLater();
    }

    _visualItems = newVisualItems;
    _complexItems = newComplexItems;

476 477
    if (_visualItems->count() == 0) {
        _addPlannedHomePosition(_visualItems, true /* addToCenter */);
478 479
    }

480
    _initAllVisualItems();
481 482
}

483
void MissionController::loadMissionFromFile(void)
484
{
485
#ifndef __mobile__
486 487 488 489 490 491 492 493 494 495 496 497
    QString filename = QGCFileDialog::getOpenFileName(NULL, "Select Mission File to load", QString(), "Mission file (*.mission);;All Files (*.*)");

    if (filename.isEmpty()) {
        return;
    }
    _loadMissionFromFile(filename);
#endif
}

void MissionController::_saveMissionToFile(const QString& filename)
{
    qDebug() << filename;
498 499 500 501

    if (filename.isEmpty()) {
        return;
    }
502

503
    QString missionFilename = filename;
504
    if (!QFileInfo(filename).fileName().contains(".")) {
505
        missionFilename += ".mission";
506 507
    }

508
    QFile file(missionFilename);
509

510
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
511
        qgcApp()->showMessage(file.errorString());
512
    } else {
513 514 515
        QJsonObject missionFileObject;      // top level json object
        QJsonArray  simpleItemsObject;
        QJsonArray  complexItemsObject;
516

517 518
        missionFileObject[_jsonVersionKey] =        "1.0";
        missionFileObject[_jsonGroundStationKey] =  "QGroundControl";
519

520 521 522 523 524 525 526 527 528 529
        MAV_AUTOPILOT firmwareType = MAV_AUTOPILOT_GENERIC;
        if (_activeVehicle) {
            firmwareType = _activeVehicle->firmwareType();
        } else {
            // FIXME: Hack duplicated code from QGroundControlQmlGlobal. Had to do this for now since
            // QGroundControlQmlGlobal is not available from C++ side.

            QSettings settings;
            firmwareType = (MAV_AUTOPILOT)settings.value("OfflineEditingFirmwareType", MAV_AUTOPILOT_ARDUPILOTMEGA).toInt();
        }
530
        missionFileObject[_jsonMavAutopilotKey] = firmwareType;
531

532
        // Save planned home position
533
        QJsonObject homePositionObject;
534 535 536 537 538 539 540
        SimpleMissionItem* homeItem = qobject_cast<SimpleMissionItem*>(_visualItems->get(0));
        if (homeItem) {
            homeItem->missionItem().save(homePositionObject);
        } else {
            qgcApp()->showMessage(QStringLiteral("Internal error: VisualMissionItem at index 0 not SimpleMissionItem"));
            return;
        }
541
        missionFileObject[_jsonPlannedHomePositionKey] = homePositionObject;
542

543
        // Save the visual items
544
        for (int i=1; i<_visualItems->count(); i++) {
545 546
            QJsonObject itemObject;

547
            VisualMissionItem* visualItem = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
548 549 550 551 552 553
            visualItem->save(itemObject);

            if (visualItem->isSimpleItem()) {
                simpleItemsObject.append(itemObject);
            } else {
                complexItemsObject.append(itemObject);
554
            }
555
        }
556

557 558 559 560
        missionFileObject[jsonSimpleItemsKey] = simpleItemsObject;
        missionFileObject[_jsonComplexItemsKey] = complexItemsObject;

        QJsonDocument saveDoc(missionFileObject);
561
        file.write(saveDoc.toJson());
562 563
    }

564
    _visualItems->setDirty(false);
565 566 567 568 569 570 571 572 573 574
}

void MissionController::saveMissionToFile(void)
{
#ifndef __mobile__
    QString filename = QGCFileDialog::getSaveFileName(NULL, "Select file to save mission to", QString(), "Mission file (*.mission);;All Files (*.*)");

    if (filename.isEmpty()) {
        return;
    }
Don Gagne's avatar
Don Gagne committed
575
    _saveMissionToFile(filename);
576
#endif
577 578
}

579 580 581 582 583 584 585 586
void MissionController::saveMobileMissionToFile(const QString& filename)
{
    QStringList docDirs = QStandardPaths::standardLocations(QStandardPaths::DocumentsLocation);
    if (docDirs.count() <= 0) {
        qWarning() << "No Documents location";
        return;
    }

Don Gagne's avatar
Don Gagne committed
587
    _saveMissionToFile(docDirs.at(0) + QDir::separator() + filename);
588 589 590 591 592 593 594 595 596 597 598 599
}

void MissionController::loadMobileMissionFromFile(const QString& filename)
{
    QStringList docDirs = QStandardPaths::standardLocations(QStandardPaths::DocumentsLocation);
    if (docDirs.count() <= 0) {
        qWarning() << "No Documents location";
        return;
    }
    _loadMissionFromFile(docDirs.at(0) + QDir::separator() + filename);
}

600
void MissionController::_calcPrevWaypointValues(double homeAlt, VisualMissionItem* currentItem, VisualMissionItem* prevItem, double* azimuth, double* distance, double* altDifference)
601
{
Don Gagne's avatar
Don Gagne committed
602
    QGeoCoordinate  currentCoord =  currentItem->coordinate();
603
    QGeoCoordinate  prevCoord =     prevItem->exitCoordinate();
604 605 606 607
    bool            distanceOk =    false;

    // Convert to fixed altitudes

608
    qCDebug(MissionControllerLog) << homeAlt
609 610
                                  << currentItem->coordinateHasRelativeAltitude() << currentItem->coordinate().altitude()
                                  << prevItem->exitCoordinateHasRelativeAltitude() << prevItem->exitCoordinate().altitude();
611

612
    distanceOk = true;
613
    if (currentItem->coordinateHasRelativeAltitude()) {
614 615
        currentCoord.setAltitude(homeAlt + currentCoord.altitude());
    }
616
    if (prevItem->exitCoordinateHasRelativeAltitude()) {
617
        prevCoord.setAltitude(homeAlt + prevCoord.altitude());
618 619 620 621 622
    }

    qCDebug(MissionControllerLog) << "distanceOk" << distanceOk;

    if (distanceOk) {
Don Gagne's avatar
Don Gagne committed
623 624 625
        *altDifference = currentCoord.altitude() - prevCoord.altitude();
        *distance = prevCoord.distanceTo(currentCoord);
        *azimuth = prevCoord.azimuthTo(currentCoord);
626
    } else {
Don Gagne's avatar
Don Gagne committed
627
        *altDifference = 0.0;
628
        *azimuth = 0.0;
Don Gagne's avatar
Don Gagne committed
629
        *distance = -1.0;   // Signals no values
630 631 632
    }
}

633 634
void MissionController::_recalcWaypointLines(void)
{
635 636 637 638 639 640 641 642 643 644 645
    bool                firstCoordinateItem =   true;
    VisualMissionItem*  lastCoordinateItem =    qobject_cast<VisualMissionItem*>(_visualItems->get(0));

    SimpleMissionItem*  homeItem = qobject_cast<SimpleMissionItem*>(lastCoordinateItem);

    if (!homeItem) {
        qWarning() << "Home item is not SimpleMissionItem";
    }

    bool    showHomePosition =  homeItem->showHomePosition();
    double  homeAlt =           homeItem->coordinate().altitude();
646 647 648 649 650 651

    qCDebug(MissionControllerLog) << "_recalcWaypointLines";

    // If home position is valid we can calculate distances between all waypoints.
    // If home position is not valid we can only calculate distances between waypoints which are
    // both relative altitude.
652

653
    // No values for first item
654
    lastCoordinateItem->setAltDifference(0.0);
655
    lastCoordinateItem->setAzimuth(0.0);
656 657
    lastCoordinateItem->setDistance(-1.0);

658 659 660 661 662
    double minAltSeen = 0.0;
    double maxAltSeen = 0.0;
    double homePositionAltitude = homeItem->coordinate().altitude();
    minAltSeen = maxAltSeen = homeItem->coordinate().altitude();

663 664
    _waypointLines.clear();

Don Gagne's avatar
Don Gagne committed
665
    bool linkBackToHome = false;
666 667
    for (int i=1; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
668

669 670 671
        // Assume the worst
        item->setAzimuth(0.0);
        item->setDistance(-1.0);
672

673 674 675 676
        // If we still haven't found the first coordinate item and we hit a a takeoff command link back to home
        if (firstCoordinateItem &&
                item->isSimpleItem() &&
                qobject_cast<SimpleMissionItem*>(item)->command() == MavlinkQmlSingleton::MAV_CMD_NAV_TAKEOFF) {
Don Gagne's avatar
Don Gagne committed
677 678 679
            linkBackToHome = true;
        }

680
        if (item->specifiesCoordinate()) {
681 682
            // Keep track of the min/max altitude for all waypoints so we can show altitudes as a percentage

683
            double absoluteAltitude = item->coordinate().altitude();
684
            if (item->coordinateHasRelativeAltitude()) {
685 686 687 688 689
                absoluteAltitude += homePositionAltitude;
            }
            minAltSeen = std::min(minAltSeen, absoluteAltitude);
            maxAltSeen = std::max(maxAltSeen, absoluteAltitude);

690 691 692 693 694 695 696 697 698 699
            if (!item->exitCoordinateSameAsEntry()) {
                absoluteAltitude = item->exitCoordinate().altitude();
                if (item->exitCoordinateHasRelativeAltitude()) {
                    absoluteAltitude += homePositionAltitude;
                }
                minAltSeen = std::min(minAltSeen, absoluteAltitude);
                maxAltSeen = std::max(maxAltSeen, absoluteAltitude);
            }

            if (!item->isStandaloneCoordinate()) {
Don Gagne's avatar
Don Gagne committed
700
                firstCoordinateItem = false;
701
                if (lastCoordinateItem != homeItem || (showHomePosition && linkBackToHome)) {
702 703 704 705
                    double azimuth, distance, altDifference;

                    // Subsequent coordinate items link to last coordinate item. If the last coordinate item
                    // is an invalid home position we skip the line
706
                    _calcPrevWaypointValues(homeAlt, item, lastCoordinateItem, &azimuth, &distance, &altDifference);
707 708 709
                    item->setAltDifference(altDifference);
                    item->setAzimuth(azimuth);
                    item->setDistance(distance);
710
                    _waypointLines.append(new CoordinateVector(lastCoordinateItem->isSimpleItem() ? lastCoordinateItem->coordinate() : lastCoordinateItem->exitCoordinate(), item->coordinate()));
711
                }
712 713 714 715 716 717 718
                lastCoordinateItem = item;
            }
        }
    }

    // Walk the list again calculating altitude percentages
    double altRange = maxAltSeen - minAltSeen;
719 720
    for (int i=0; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
721 722 723

        if (item->specifiesCoordinate()) {
            double absoluteAltitude = item->coordinate().altitude();
724
            if (item->coordinateHasRelativeAltitude()) {
725 726 727 728 729 730
                absoluteAltitude += homePositionAltitude;
            }
            if (altRange == 0.0) {
                item->setAltPercent(0.0);
            } else {
                item->setAltPercent((absoluteAltitude - minAltSeen) / altRange);
731
            }
732 733
        }
    }
734 735

    emit waypointLinesChanged();
736 737
}

738 739 740
// This will update the sequence numbers to be sequential starting from 0
void MissionController::_recalcSequence(void)
{
741 742 743 744 745 746 747 748 749
    // Setup ascending sequence numbers for all visual items

    int sequenceNumber = 0;
    for (int i=0; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));

        item->setSequenceNumber(sequenceNumber++);
        if (!item->isSimpleItem()) {
            ComplexMissionItem* complexItem = qobject_cast<ComplexMissionItem*>(item);
750

751
            if (complexItem) {
752
                 sequenceNumber = complexItem->lastSequenceNumber() + 1;
753 754 755 756
            } else {
                qWarning() << "isSimpleItem == false, yet not ComplexMissionItem";
            }
        }
757 758 759
    }
}

760 761 762
// This will update the child item hierarchy
void MissionController::_recalcChildItems(void)
{
763
    VisualMissionItem* currentParentItem = qobject_cast<VisualMissionItem*>(_visualItems->get(0));
764 765 766

    currentParentItem->childItems()->clear();

767 768
    for (int i=1; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
769 770 771 772 773 774 775 776 777 778 779

        // Set up non-coordinate item child hierarchy
        if (item->specifiesCoordinate()) {
            item->childItems()->clear();
            currentParentItem = item;
        } else {
            currentParentItem->childItems()->append(item);
        }
    }
}

780 781
void MissionController::_recalcAll(void)
{
782
    _recalcSequence();
783 784 785 786
    _recalcChildItems();
    _recalcWaypointLines();
}

787
/// Initializes a new set of mission items
788
void MissionController::_initAllVisualItems(void)
789
{
790 791 792 793 794 795 796 797 798
    SimpleMissionItem* homeItem = NULL;

    // Setup home position at index 0

    homeItem = qobject_cast<SimpleMissionItem*>(_visualItems->get(0));
    if (!homeItem) {
        qWarning() << "homeItem not SimpleMissionItem";
        return;
    }
799

Don Gagne's avatar
Don Gagne committed
800
    homeItem->setHomePositionSpecialCase(true);
801
    homeItem->setShowHomePosition(_editMode);
802 803
    homeItem->missionItem().setCommand(MAV_CMD_NAV_WAYPOINT);
    homeItem->missionItem().setFrame(MAV_FRAME_GLOBAL);
804 805 806 807 808
    homeItem->setIsCurrentItem(true);

    if (!_editMode && _activeVehicle && _activeVehicle->homePositionAvailable()) {
        homeItem->setCoordinate(_activeVehicle->homePosition());
        homeItem->setShowHomePosition(true);
809
    }
810

811
    QmlObjectListModel* newComplexItems = new QmlObjectListModel(this);
812 813 814
    for (int i=0; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
        _initVisualItem(item);
815

816 817 818
        // Set up complex item list
        if (!item->isSimpleItem()) {
            ComplexMissionItem* complexItem = qobject_cast<ComplexMissionItem*>(item);
819

820 821 822 823 824
            if (complexItem) {
                newComplexItems->append(item);
            } else {
                qWarning() << "isSimpleItem == false, but not ComplexMissionItem";
            }
825
        }
826 827
    }

828 829 830 831
    if (_complexItems) {
        _complexItems->deleteLater();
    }
    _complexItems = newComplexItems;
832

833
    _recalcAll();
834

835 836
    emit visualItemsChanged();
    emit complexVisualItemsChanged();
837

838
    _visualItems->setDirty(false);
839

840
    connect(_visualItems, &QmlObjectListModel::dirtyChanged, this, &MissionController::_dirtyChanged);
841 842
}

843
void MissionController::_deinitAllVisualItems(void)
844
{
845 846
    for (int i=0; i<_visualItems->count(); i++) {
        _deinitVisualItem(qobject_cast<VisualMissionItem*>(_visualItems->get(i)));
847 848
    }

849
    connect(_visualItems, &QmlObjectListModel::dirtyChanged, this, &MissionController::_dirtyChanged);
850 851
}

852
void MissionController::_initVisualItem(VisualMissionItem* visualItem)
853
{
854 855 856
    _visualItems->setDirty(false);

    connect(visualItem, &VisualMissionItem::coordinateChanged,                          this, &MissionController::_itemCoordinateChanged);
857
    connect(visualItem, &VisualMissionItem::specifiesCoordinateChanged,                 this, &MissionController::_recalcWaypointLines);
858 859
    connect(visualItem, &VisualMissionItem::coordinateHasRelativeAltitudeChanged,       this, &MissionController::_recalcWaypointLines);
    connect(visualItem, &VisualMissionItem::exitCoordinateHasRelativeAltitudeChanged,   this, &MissionController::_recalcWaypointLines);
860

861 862 863 864 865 866 867 868 869 870
    if (visualItem->isSimpleItem()) {
        // We need to track commandChanged on simple item since recalc has special handling for takeoff command

        SimpleMissionItem* simpleItem = qobject_cast<SimpleMissionItem*>(visualItem);
        if (simpleItem) {
            connect(&simpleItem->missionItem()._commandFact, &Fact::valueChanged, this, &MissionController::_itemCommandChanged);
        } else {
            qWarning() << "isSimpleItem == true, yet not SimpleMissionItem";
        }
    }
871 872
}

873
void MissionController::_deinitVisualItem(VisualMissionItem* visualItem)
874
{
875 876
    // Disconnect all signals
    disconnect(visualItem, 0, 0, 0);
877 878 879 880 881 882 883 884
}

void MissionController::_itemCoordinateChanged(const QGeoCoordinate& coordinate)
{
    Q_UNUSED(coordinate);
    _recalcWaypointLines();
}

885
void MissionController::_itemCommandChanged(void)
886
{
887 888
    _recalcChildItems();
    _recalcWaypointLines();
889 890 891 892
}

void MissionController::_activeVehicleChanged(Vehicle* activeVehicle)
{
893 894
    qCDebug(MissionControllerLog) << "_activeVehicleChanged activeVehicle" << activeVehicle;

895 896
    if (_activeVehicle) {
        MissionManager* missionManager = _activeVehicle->missionManager();
897

898
        disconnect(missionManager, &MissionManager::newMissionItemsAvailable,   this, &MissionController::_newMissionItemsAvailableFromVehicle);
899
        disconnect(missionManager, &MissionManager::inProgressChanged,          this, &MissionController::_inProgressChanged);
900
        disconnect(missionManager, &MissionManager::currentItemChanged,         this, &MissionController::_currentMissionItemChanged);
901 902
        disconnect(_activeVehicle, &Vehicle::homePositionAvailableChanged,      this, &MissionController::_activeVehicleHomePositionAvailableChanged);
        disconnect(_activeVehicle, &Vehicle::homePositionChanged,               this, &MissionController::_activeVehicleHomePositionChanged);
903
        _activeVehicle = NULL;
Don Gagne's avatar
Don Gagne committed
904 905
    }

906
    _activeVehicle = activeVehicle;
907

908 909
    if (_activeVehicle) {
        MissionManager* missionManager = activeVehicle->missionManager();
910

911
        connect(missionManager, &MissionManager::newMissionItemsAvailable,  this, &MissionController::_newMissionItemsAvailableFromVehicle);
912
        connect(missionManager, &MissionManager::inProgressChanged,         this, &MissionController::_inProgressChanged);
913
        connect(missionManager, &MissionManager::currentItemChanged,        this, &MissionController::_currentMissionItemChanged);
914 915 916
        connect(_activeVehicle, &Vehicle::homePositionAvailableChanged,     this, &MissionController::_activeVehicleHomePositionAvailableChanged);
        connect(_activeVehicle, &Vehicle::homePositionChanged,              this, &MissionController::_activeVehicleHomePositionChanged);

917 918 919
        if (!_editMode) {
            removeAllMissionItems();
        }
920 921 922 923

        _activeVehicleHomePositionChanged(_activeVehicle->homePosition());
        _activeVehicleHomePositionAvailableChanged(_activeVehicle->homePositionAvailable());
    }
924 925 926

    // Whenever vehicle changes we need to update syncInProgress
    emit syncInProgressChanged(syncInProgress());
927 928 929 930
}

void MissionController::_activeVehicleHomePositionAvailableChanged(bool homePositionAvailable)
{
931 932 933 934 935 936 937 938 939
    if (!_editMode && _visualItems) {
        SimpleMissionItem* homeItem = qobject_cast<SimpleMissionItem*>(_visualItems->get(0));

        if (homeItem) {
            homeItem->setShowHomePosition(homePositionAvailable);
            _recalcWaypointLines();
        } else {
            qWarning() << "Unabled to cast home item to SimpleMissionItem";
        }
940
    }
941 942 943 944
}

void MissionController::_activeVehicleHomePositionChanged(const QGeoCoordinate& homePosition)
{
945 946
    if (!_editMode && _visualItems) {
        qobject_cast<VisualMissionItem*>(_visualItems->get(0))->setCoordinate(homePosition);
947 948
        _recalcWaypointLines();
    }
949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968
}

void MissionController::setAutoSync(bool autoSync)
{
    // FIXME: AutoSync temporarily turned off
#if 0
    _autoSync = autoSync;
    emit autoSyncChanged(_autoSync);

    if (_autoSync) {
        _dirtyChanged(true);
    }
#else
    Q_UNUSED(autoSync)
#endif
}

void MissionController::_dirtyChanged(bool dirty)
{
    if (dirty && _autoSync) {
969
        Vehicle* activeVehicle = qgcApp()->toolbox()->multiVehicleManager()->activeVehicle();
970 971 972 973 974 975 976 977 978 979 980 981 982 983

        if (activeVehicle && !activeVehicle->armed()) {
            if (_activeVehicle->missionManager()->inProgress()) {
                _queuedSend = true;
            } else {
                _autoSyncSend();
            }
        }
    }
}

void MissionController::_autoSyncSend(void)
{
    _queuedSend = false;
984
    if (_visualItems) {
985
        sendMissionItems();
986
        _visualItems->setDirty(false);
987 988
    }
}
989

990
void MissionController::_inProgressChanged(bool inProgress)
991
{
992
    emit syncInProgressChanged(inProgress);
993 994 995 996
    if (!inProgress && _queuedSend) {
        _autoSyncSend();
    }
}
997

998
bool MissionController::_findLastAltitude(double* lastAltitude)
999
{
1000 1001
    bool found = false;
    double foundAltitude;
1002

1003
    // Don't use home position
1004 1005
    for (int i=1; i<_visualItems->count(); i++) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
1006

1007 1008
        if (item->specifiesCoordinate() && !item->isStandaloneCoordinate()) {
            foundAltitude = item->exitCoordinate().altitude();
1009
            found = true;
1010 1011 1012
        }
    }

1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024
    if (found) {
        *lastAltitude = foundAltitude;
    }

    return found;
}

bool MissionController::_findLastAcceptanceRadius(double* lastAcceptanceRadius)
{
    bool found = false;
    double foundAcceptanceRadius;

1025 1026
    for (int i=0; i<_visualItems->count(); i++) {
        VisualMissionItem* visualItem = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
1027

1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038
        if (visualItem->isSimpleItem()) {
            SimpleMissionItem* simpleItem = qobject_cast<SimpleMissionItem*>(visualItem);

            if (simpleItem) {
                if ((MAV_CMD)simpleItem->command() == MAV_CMD_NAV_WAYPOINT) {
                    foundAcceptanceRadius = simpleItem->missionItem().param2();
                    found = true;
                }
            } else {
                qWarning() << "isSimpleItem == true, yet not SimpleMissionItem";
            }
1039 1040 1041 1042 1043 1044 1045 1046
        }
    }

    if (found) {
        *lastAcceptanceRadius = foundAcceptanceRadius;
    }

    return found;
1047
}
1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061

double MissionController::_normalizeLat(double lat)
{
    // Normalize latitude to range: 0 to 180, S to N
    return lat + 90.0;
}

double MissionController::_normalizeLon(double lon)
{
    // Normalize longitude to range: 0 to 360, W to E
    return lon  + 180.0;
}

/// Add the home position item to the front of the list
1062
void MissionController::_addPlannedHomePosition(QmlObjectListModel* visualItems, bool addToCenter)
1063
{
1064 1065
    SimpleMissionItem* homeItem = new SimpleMissionItem(_activeVehicle, this);
    visualItems->insert(0, homeItem);
1066

1067 1068
    if (visualItems->count() > 1  && addToCenter) {
        VisualMissionItem* item = qobject_cast<VisualMissionItem*>(visualItems->get(1));
1069 1070 1071 1072 1073 1074

        double north = _normalizeLat(item->coordinate().latitude());
        double south = north;
        double east = _normalizeLon(item->coordinate().longitude());
        double west = east;

1075 1076
        for (int i=2; i<visualItems->count(); i++) {
            item = qobject_cast<VisualMissionItem*>(visualItems->get(i));
1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091

            double lat = _normalizeLat(item->coordinate().latitude());
            double lon = _normalizeLon(item->coordinate().longitude());

            north = fmax(north, lat);
            south = fmin(south, lat);
            east = fmax(east, lon);
            west = fmin(west, lon);
        }

        homeItem->setCoordinate(QGeoCoordinate((south + ((north - south) / 2)) - 90.0, (west + ((east - west) / 2)) - 180.0, 0.0));
    } else {
        homeItem->setCoordinate(qgcApp()->lastKnownHomePosition());
    }
}
1092 1093 1094 1095 1096 1097 1098 1099

void MissionController::_currentMissionItemChanged(int sequenceNumber)
{
    if (!_editMode) {
        if (!_activeVehicle->firmwarePlugin()->sendHomePositionToVehicle()) {
            sequenceNumber++;
        }

1100 1101
        for (int i=0; i<_visualItems->count(); i++) {
            VisualMissionItem* item = qobject_cast<VisualMissionItem*>(_visualItems->get(i));
1102 1103 1104 1105
            item->setIsCurrentItem(item->sequenceNumber() == sequenceNumber);
        }
    }
}
1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125

QStringList MissionController::getMobileMissionFiles(void)
{
    QStringList missionFiles;

    QStringList docDirs = QStandardPaths::standardLocations(QStandardPaths::DocumentsLocation);
    if (docDirs.count() <= 0) {
        qWarning() << "No Documents location";
        return QStringList();
    }
    QDir missionDir = docDirs.at(0);

    QFileInfoList missionFileInfoList = missionDir.entryInfoList(QStringList(QStringLiteral("*.mission")),  QDir::Files, QDir::Name);

    foreach (const QFileInfo& missionFileInfo, missionFileInfoList) {
        missionFiles << missionFileInfo.baseName() + ".mission";
    }

    return missionFiles;
}
1126 1127 1128

bool MissionController::syncInProgress(void)
{
1129 1130 1131 1132 1133
    if (_activeVehicle) {
        return _activeVehicle->missionManager()->inProgress();
    } else {
        return false;
    }
1134
}