UASWaypointManager.cc 30.7 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2
/*=====================================================================

lm's avatar
lm committed
3
QGroundControl Open Source Ground Control Station
pixhawk's avatar
pixhawk committed
4

Lorenz Meier's avatar
Lorenz Meier committed
5
(c) 2009-2012 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
pixhawk's avatar
pixhawk committed
6

lm's avatar
lm committed
7
This file is part of the QGROUNDCONTROL project
pixhawk's avatar
pixhawk committed
8

lm's avatar
lm committed
9
    QGROUNDCONTROL is free software: you can redistribute it and/or modify
pixhawk's avatar
pixhawk committed
10 11 12 13
    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.

lm's avatar
lm committed
14
    QGROUNDCONTROL is distributed in the hope that it will be useful,
pixhawk's avatar
pixhawk committed
15 16 17 18 19
    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
lm's avatar
lm committed
20
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
pixhawk's avatar
pixhawk committed
21 22 23 24 25 26 27 28 29 30 31

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

/**
 * @file
 *   @brief Implementation of the waypoint protocol handler
 *
 *   @author Petri Tanskanen <mavteam@student.ethz.ch>
 *
 */

32 33
#include "UASWaypointManager.h"
#include "UAS.h"
34
#include "mavlink_types.h"
35

36
#define PROTOCOL_TIMEOUT_MS 2000    ///< maximum time to wait for pending messages until timeout
lm's avatar
lm committed
37 38
#define PROTOCOL_DELAY_MS 20        ///< minimum delay between sent messages
#define PROTOCOL_MAX_RETRIES 5      ///< maximum number of send retries (after timeout)
pixhawk's avatar
pixhawk committed
39

lm's avatar
lm committed
40
UASWaypointManager::UASWaypointManager(UAS* _uas)
41 42 43 44 45 46 47
    : uas(_uas),
      current_retries(0),
      current_wp_id(0),
      current_count(0),
      current_state(WP_IDLE),
      current_partner_systemid(0),
      current_partner_compid(0),
48 49
      currentWaypointEditable(NULL),
      protocol_timer(this)
pixhawk's avatar
pixhawk committed
50
{
51 52
    if (uas)
    {
lm's avatar
lm committed
53 54 55 56 57 58 59 60
        uasid = uas->getUASID();
        connect(&protocol_timer, SIGNAL(timeout()), this, SLOT(timeout()));
        connect(uas, SIGNAL(localPositionChanged(UASInterface*,double,double,double,quint64)), this, SLOT(handleLocalPositionChanged(UASInterface*,double,double,double,quint64)));
        connect(uas, SIGNAL(globalPositionChanged(UASInterface*,double,double,double,quint64)), this, SLOT(handleGlobalPositionChanged(UASInterface*,double,double,double,quint64)));
    }
    else
    {
        uasid = 0;
61
    }
pixhawk's avatar
pixhawk committed
62 63
}

LM's avatar
LM committed
64 65 66 67 68
UASWaypointManager::~UASWaypointManager()
{

}

pixhawk's avatar
pixhawk committed
69
void UASWaypointManager::timeout()
70
{
71
    if (current_retries > 0) {
72 73
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries--;
74
        emit updateStatusString(tr("Timeout, retrying (retries left: %1)").arg(current_retries));
75

76
        if (current_state == WP_GETLIST) {
77
            sendWaypointRequestList();
78
        } else if (current_state == WP_GETLIST_GETWPS) {
79
            sendWaypointRequest(current_wp_id);
80
        } else if (current_state == WP_SENDLIST) {
81
            sendWaypointCount();
82
        } else if (current_state == WP_SENDLIST_SENDWPS) {
83
            sendWaypoint(current_wp_id);
84
        } else if (current_state == WP_CLEARLIST) {
85
            sendWaypointClearAll();
86
        } else if (current_state == WP_SETCURRENT) {
87 88
            sendWaypointSetCurrent(current_wp_id);
        }
89
    } else {
90
        protocol_timer.stop();
pixhawk's avatar
pixhawk committed
91

92
        emit updateStatusString("Operation timed out.");
pixhawk's avatar
pixhawk committed
93

94 95 96 97 98 99
        current_state = WP_IDLE;
        current_count = 0;
        current_wp_id = 0;
        current_partner_systemid = 0;
        current_partner_compid = 0;
    }
100 101
}

102 103 104 105
void UASWaypointManager::handleLocalPositionChanged(UASInterface* mav, double x, double y, double z, quint64 time)
{
    Q_UNUSED(mav);
    Q_UNUSED(time);
pixhawk's avatar
pixhawk committed
106
    if (waypointsEditable.count() > 0 && currentWaypointEditable && (currentWaypointEditable->getFrame() == MAV_FRAME_LOCAL_NED || currentWaypointEditable->getFrame() == MAV_FRAME_LOCAL_ENU))
107
    {
pixhawk's avatar
pixhawk committed
108 109 110
        double xdiff = x-currentWaypointEditable->getX();
        double ydiff = y-currentWaypointEditable->getY();
        double zdiff = z-currentWaypointEditable->getZ();
111 112 113 114 115 116 117
        double dist = sqrt(xdiff*xdiff + ydiff*ydiff + zdiff*zdiff);
        emit waypointDistanceChanged(dist);
    }
}

void UASWaypointManager::handleGlobalPositionChanged(UASInterface* mav, double lat, double lon, double alt, quint64 time)
{
118 119
    Q_UNUSED(mav);
    Q_UNUSED(time);
120 121 122
	Q_UNUSED(alt);
	Q_UNUSED(lon);
	Q_UNUSED(lat);
123 124 125 126 127 128
    if (waypointsEditable.count() > 0 && currentWaypointEditable && (currentWaypointEditable->getFrame() == MAV_FRAME_GLOBAL || currentWaypointEditable->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT))
    {
        // TODO FIXME Calculate distance
        double dist = 0;
        emit waypointDistanceChanged(dist);
    }
129 130
}

131 132
void UASWaypointManager::handleWaypointCount(quint8 systemId, quint8 compId, quint16 count)
{
133
    if (current_state == WP_GETLIST && systemId == current_partner_systemid) {
134 135 136
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

137 138 139 140 141 142 143 144 145 146
        //Clear the old edit-list before receiving the new one
        if (read_to_edit == true){
            while(waypointsEditable.size()>0) {
                Waypoint *t = waypointsEditable[0];
                waypointsEditable.remove(0);
                delete t;
            }
            emit waypointEditableListChanged();
        }

147
        if (count > 0) {
pixhawk's avatar
pixhawk committed
148 149 150 151
            current_count = count;
            current_wp_id = 0;
            current_state = WP_GETLIST_GETWPS;
            sendWaypointRequest(current_wp_id);
152
        } else {
153
            protocol_timer.stop();
pixhawk's avatar
pixhawk committed
154
            emit updateStatusString("done.");
155 156 157 158 159
            current_state = WP_IDLE;
            current_count = 0;
            current_wp_id = 0;
            current_partner_systemid = 0;
            current_partner_compid = 0;
pixhawk's avatar
pixhawk committed
160
        }
161 162


163 164
    } else {
        qDebug("Rejecting message, check mismatch: current_state: %d == %d, system id %d == %d, comp id %d == %d", current_state, WP_GETLIST, current_partner_systemid, systemId, current_partner_compid, compId);
165
    }
166 167
}

lm's avatar
lm committed
168
void UASWaypointManager::handleWaypoint(quint8 systemId, quint8 compId, mavlink_mission_item_t *wp)
pixhawk's avatar
pixhawk committed
169
{
170
    if (systemId == current_partner_systemid && current_state == WP_GETLIST_GETWPS && wp->seq == current_wp_id) {
171 172 173
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

174
        if(wp->seq == current_wp_id) {
pixhawk's avatar
pixhawk committed
175 176 177 178 179 180 181 182 183 184 185

            Waypoint *lwp_vo = new Waypoint(wp->seq, wp->x, wp->y, wp->z, wp->param1, wp->param2, wp->param3, wp->param4, wp->autocontinue, wp->current, (MAV_FRAME) wp->frame, (MAV_CMD) wp->command);
            addWaypointViewOnly(lwp_vo);


            if (read_to_edit == true) {
                Waypoint *lwp_ed = new Waypoint(wp->seq, wp->x, wp->y, wp->z, wp->param1, wp->param2, wp->param3, wp->param4, wp->autocontinue, wp->current, (MAV_FRAME) wp->frame, (MAV_CMD) wp->command);
                addWaypointEditable(lwp_ed, false);
                if (wp->current == 1) currentWaypointEditable = lwp_ed;
            }

pixhawk's avatar
pixhawk committed
186 187 188 189

            //get next waypoint
            current_wp_id++;

190
            if(current_wp_id < current_count) {
191
                sendWaypointRequest(current_wp_id);
192
            } else {
193 194
                sendWaypointAck(0);

pixhawk's avatar
pixhawk committed
195
                // all waypoints retrieved, change state to idle
pixhawk's avatar
pixhawk committed
196 197 198 199 200
                current_state = WP_IDLE;
                current_count = 0;
                current_wp_id = 0;
                current_partner_systemid = 0;
                current_partner_compid = 0;
201

pixhawk's avatar
pixhawk committed
202
                protocol_timer.stop();
203
                emit readGlobalWPFromUAS(false);
204 205 206
                QTime time = QTime::currentTime();
                QString timeString = time.toString();
                emit updateStatusString(tr("done. (updated at %1)").arg(timeString));
207

pixhawk's avatar
pixhawk committed
208
            }
209
        } else {
210
            emit updateStatusString(tr("Waypoint ID mismatch, rejecting waypoint"));
pixhawk's avatar
pixhawk committed
211
        }
212 213
    } else {
        qDebug("Rejecting message, check mismatch: current_state: %d == %d, system id %d == %d, comp id %d == %d", current_state, WP_GETLIST, current_partner_systemid, systemId, current_partner_compid, compId);
214
    }
pixhawk's avatar
pixhawk committed
215 216
}

lm's avatar
lm committed
217
void UASWaypointManager::handleWaypointAck(quint8 systemId, quint8 compId, mavlink_mission_ack_t *wpa)
218
{
219
    if (systemId == current_partner_systemid && (compId == current_partner_compid || compId == MAV_COMP_ID_ALL)) {
220
        if((current_state == WP_SENDLIST || current_state == WP_SENDLIST_SENDWPS) && (current_wp_id == waypoint_buffer.count()-1 && wpa->type == 0)) {
221 222
            //all waypoints sent and ack received
            protocol_timer.stop();
223
            current_state = WP_IDLE;
224
            readWaypoints(false); //Update "Onboard Waypoints"-tab immidiately after the waypoint list has been sent.
225
            emit updateStatusString("done.");
226
        } else if(current_state == WP_CLEARLIST) {
227 228 229 230
            protocol_timer.stop();
            current_state = WP_IDLE;
            emit updateStatusString("done.");
        }
231 232 233
    }
}

lm's avatar
lm committed
234
void UASWaypointManager::handleWaypointRequest(quint8 systemId, quint8 compId, mavlink_mission_request_t *wpr)
235
{
236
    if (systemId == current_partner_systemid && ((current_state == WP_SENDLIST && wpr->seq == 0) || (current_state == WP_SENDLIST_SENDWPS && (wpr->seq == current_wp_id || wpr->seq == current_wp_id + 1)))) {
237 238
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
239

240
        if (wpr->seq < waypoint_buffer.count()) {
pixhawk's avatar
pixhawk committed
241 242 243
            current_state = WP_SENDLIST_SENDWPS;
            current_wp_id = wpr->seq;
            sendWaypoint(current_wp_id);
244
        } else {
pixhawk's avatar
pixhawk committed
245 246
            //TODO: Error message or something
        }
247 248
    } else {
        qDebug("Rejecting message, check mismatch: current_state: %d == %d, system id %d == %d, comp id %d == %d", current_state, WP_GETLIST, current_partner_systemid, systemId, current_partner_compid, compId);
249
    }
250 251
}

lm's avatar
lm committed
252
void UASWaypointManager::handleWaypointReached(quint8 systemId, quint8 compId, mavlink_mission_item_reached_t *wpr)
253
{
254
	Q_UNUSED(compId);
lm's avatar
lm committed
255
    if (!uas) return;
256
    if (systemId == uasid) {
pixhawk's avatar
pixhawk committed
257 258
        emit updateStatusString(QString("Reached waypoint %1").arg(wpr->seq));
    }
pixhawk's avatar
pixhawk committed
259 260
}

lm's avatar
lm committed
261
void UASWaypointManager::handleWaypointCurrent(quint8 systemId, quint8 compId, mavlink_mission_current_t *wpc)
pixhawk's avatar
pixhawk committed
262
{
263
    Q_UNUSED(compId);
lm's avatar
lm committed
264
    if (!uas) return;
265
    if (systemId == uasid) {
266
        // FIXME Petri
267
        if (current_state == WP_SETCURRENT) {
268 269
            protocol_timer.stop();
            current_state = WP_IDLE;
270 271

            // update the local main storage
pixhawk's avatar
pixhawk committed
272 273 274 275
            if (wpc->seq < waypointsViewOnly.size()) {
                for(int i = 0; i < waypointsViewOnly.size(); i++) {
                    if (waypointsViewOnly[i]->getId() == wpc->seq) {
                        waypointsViewOnly[i]->setCurrent(true);
276
                    } else {
pixhawk's avatar
pixhawk committed
277
                        waypointsViewOnly[i]->setCurrent(false);
278 279 280
                    }
                }
            }
281
        }
282 283 284
        emit updateStatusString(QString("New current waypoint %1").arg(wpc->seq));
        //emit update to UI widgets
        emit currentWaypointChanged(wpc->seq);
pixhawk's avatar
pixhawk committed
285
    }
pixhawk's avatar
pixhawk committed
286 287
}

288
void UASWaypointManager::notifyOfChangeEditable(Waypoint* wp)
289
{
290
    // If only one waypoint was changed, emit only WP signal
291
    if (wp != NULL) {
292
        emit waypointEditableChanged(uasid, wp);
293
    } else {
294
        emit waypointEditableListChanged();
295
        emit waypointEditableListChanged(uasid);
296
    }
297 298
}

299 300 301
void UASWaypointManager::notifyOfChangeViewOnly(Waypoint* wp)
{
    if (wp != NULL) {
302
        emit waypointViewOnlyChanged(uasid, wp);
303 304
    } else {
        emit waypointViewOnlyListChanged();
305
        emit waypointViewOnlyListChanged(uasid);
306 307
    }
}
308 309


310
int UASWaypointManager::setCurrentWaypoint(quint16 seq)
pixhawk's avatar
pixhawk committed
311
{
pixhawk's avatar
pixhawk committed
312
    if (seq < waypointsViewOnly.size()) {
313
        if(current_state == WP_IDLE) {
314

315 316 317 318 319 320
            //send change to UAS - important to note: if the transmission fails, we have inconsistencies
            protocol_timer.start(PROTOCOL_TIMEOUT_MS);
            current_retries = PROTOCOL_MAX_RETRIES;

            current_state = WP_SETCURRENT;
            current_wp_id = seq;
321
            current_partner_systemid = uasid;
lm's avatar
lm committed
322
            current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
323 324 325 326 327 328 329 330 331

            sendWaypointSetCurrent(current_wp_id);

            return 0;
        }
    }
    return -1;
}

332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349
int UASWaypointManager::setCurrentEditable(quint16 seq)
{
    if (seq < waypointsEditable.size()) {
        if(current_state == WP_IDLE) {
            //update local main storage
            for(int i = 0; i < waypointsEditable.size(); i++) {
                if (waypointsEditable[i]->getId() == seq) {
                    waypointsEditable[i]->setCurrent(true);
                } else {
                    waypointsEditable[i]->setCurrent(false);
                }
            }

            return 0;
        }
    }
    return -1;
}
pixhawk's avatar
pixhawk committed
350 351 352 353

void UASWaypointManager::addWaypointViewOnly(Waypoint *wp)
{
    if (wp)
354
    {
pixhawk's avatar
pixhawk committed
355 356 357 358
        waypointsViewOnly.insert(waypointsViewOnly.size(), wp);
        connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChangeViewOnly(Waypoint*)));

        emit waypointViewOnlyListChanged();
359
        emit waypointViewOnlyListChanged(uasid);
pixhawk's avatar
pixhawk committed
360 361 362 363
    }
}


pixhawk's avatar
pixhawk committed
364
/**
365
 * @warning Make sure the waypoint stays valid for the whole application lifecycle!
pixhawk's avatar
pixhawk committed
366
 * @param enforceFirstActive Enforces that the first waypoint is set as active
367
 * @see createWaypoint() is more suitable for most use cases
pixhawk's avatar
pixhawk committed
368
 */
369
void UASWaypointManager::addWaypointEditable(Waypoint *wp, bool enforceFirstActive)
370
{
lm's avatar
lm committed
371 372
    if (wp)
    {
pixhawk's avatar
pixhawk committed
373 374
        wp->setId(waypointsEditable.size());
        if (enforceFirstActive && waypointsEditable.size() == 0)
375 376
        {
            wp->setCurrent(true);
pixhawk's avatar
pixhawk committed
377
            currentWaypointEditable = wp;
378
        }
pixhawk's avatar
pixhawk committed
379
        waypointsEditable.insert(waypointsEditable.size(), wp);
380
        connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChangeEditable(Waypoint*)));
381

382
        emit waypointEditableListChanged();
383
        emit waypointEditableListChanged(uasid);
384
    }
pixhawk's avatar
pixhawk committed
385 386
}

387 388 389 390 391 392
/**
 * @param enforceFirstActive Enforces that the first waypoint is set as active
 */
Waypoint* UASWaypointManager::createWaypoint(bool enforceFirstActive)
{
    Waypoint* wp = new Waypoint();
pixhawk's avatar
pixhawk committed
393 394
    wp->setId(waypointsEditable.size());
    if (enforceFirstActive && waypointsEditable.size() == 0)
395 396
    {
        wp->setCurrent(true);
pixhawk's avatar
pixhawk committed
397
        currentWaypointEditable = wp;
398
    }
pixhawk's avatar
pixhawk committed
399
    waypointsEditable.insert(waypointsEditable.size(), wp);
400
    connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChangeEditable(Waypoint*)));
401

402
    emit waypointEditableListChanged();
403
    emit waypointEditableListChanged(uasid);
404 405 406
    return wp;
}

407
int UASWaypointManager::removeWaypoint(quint16 seq)
408
{
pixhawk's avatar
pixhawk committed
409
    if (seq < waypointsEditable.size())
410
    {
pixhawk's avatar
pixhawk committed
411
        Waypoint *t = waypointsEditable[seq];
412 413 414 415 416 417 418 419 420 421 422 423 424

        if (t->getCurrent() == true) //trying to remove the current waypoint
        {
            if (seq+1 < waypointsEditable.size()) // setting the next waypoint as current
            {
                waypointsEditable[seq+1]->setCurrent(true);
            }
            else if (seq-1 >= 0) //if deleting the last on the list, then setting the previous waypoint as current
            {
                waypointsEditable[seq-1]->setCurrent(true);
            }
        }

pixhawk's avatar
pixhawk committed
425
        waypointsEditable.remove(seq);
426
        delete t;
427
        t = NULL;
428

pixhawk's avatar
pixhawk committed
429
        for(int i = seq; i < waypointsEditable.size(); i++)
430
        {
pixhawk's avatar
pixhawk committed
431
            waypointsEditable[i]->setId(i);
432
        }
Alejandro's avatar
Alejandro committed
433

434
        emit waypointEditableListChanged();
435
        emit waypointEditableListChanged(uasid);
436 437 438 439 440
        return 0;
    }
    return -1;
}

441
void UASWaypointManager::moveWaypoint(quint16 cur_seq, quint16 new_seq)
442
{
pixhawk's avatar
pixhawk committed
443
    if (cur_seq != new_seq && cur_seq < waypointsEditable.size() && new_seq < waypointsEditable.size())
444
    {
pixhawk's avatar
pixhawk committed
445
        Waypoint *t = waypointsEditable[cur_seq];
446
        if (cur_seq < new_seq) {
447 448
            for (int i = cur_seq; i < new_seq; i++)
            {
pixhawk's avatar
pixhawk committed
449
                waypointsEditable[i] = waypointsEditable[i+1];
450
                waypointsEditable[i]->setId(i);
451
            }
452 453 454 455 456
        }
        else
        {
            for (int i = cur_seq; i > new_seq; i--)
            {
pixhawk's avatar
pixhawk committed
457
                waypointsEditable[i] = waypointsEditable[i-1];
458
                waypointsEditable[i]->setId(i);
459 460
            }
        }
pixhawk's avatar
pixhawk committed
461
        waypointsEditable[new_seq] = t;
462
        waypointsEditable[new_seq]->setId(new_seq);
463

464
        emit waypointEditableListChanged();
465
        emit waypointEditableListChanged(uasid);
466 467 468
    }
}

469
void UASWaypointManager::saveWaypoints(const QString &saveFile)
470 471 472 473 474 475
{
    QFile file(saveFile);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
        return;

    QTextStream out(&file);
476 477

    //write the waypoint list version to the first line for compatibility check
478
    out << "QGC WPL 120\r\n";
479

pixhawk's avatar
pixhawk committed
480
    for (int i = 0; i < waypointsEditable.size(); i++)
481
    {
pixhawk's avatar
pixhawk committed
482 483
        waypointsEditable[i]->setId(i);
        waypointsEditable[i]->save(out);
484 485 486 487
    }
    file.close();
}

488
void UASWaypointManager::loadWaypoints(const QString &loadFile)
489 490 491 492 493
{
    QFile file(loadFile);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
        return;

pixhawk's avatar
pixhawk committed
494 495 496
    while(waypointsEditable.size()>0) {
        Waypoint *t = waypointsEditable[0];
        waypointsEditable.remove(0);
497 498 499 500
        delete t;
    }

    QTextStream in(&file);
501 502 503

    const QStringList &version = in.readLine().split(" ");

504
    if (!(version.size() == 3 && version[0] == "QGC" && version[1] == "WPL" && version[2] == "120"))
505
    {
506
        emit updateStatusString(tr("The waypoint file is not compatible with the current version of QGroundControl."));
507 508 509 510 511
    }
    else
    {
        while (!in.atEnd())
        {
512
            Waypoint *t = new Waypoint();
513 514
            if(t->load(in))
            {
pixhawk's avatar
pixhawk committed
515 516
                t->setId(waypointsEditable.size());
                waypointsEditable.insert(waypointsEditable.size(), t);
517 518 519
            }
            else
            {
520
                emit updateStatusString(tr("The waypoint file is corrupted. Load operation only partly succesful."));
521 522
                break;
            }
523 524
        }
    }
525

526 527
    file.close();

528
    emit loadWPFile();
529
    emit waypointEditableListChanged();
530
    emit waypointEditableListChanged(uasid);
531 532 533
}

void UASWaypointManager::clearWaypointList()
pixhawk's avatar
pixhawk committed
534
{
535 536
    if(current_state == WP_IDLE)
    {
pixhawk's avatar
pixhawk committed
537
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
538
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
539

540 541
        current_state = WP_CLEARLIST;
        current_wp_id = 0;
542
        current_partner_systemid = uasid;
lm's avatar
lm committed
543
        current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed
544

545
        sendWaypointClearAll();
546 547 548
    }
}

549 550 551 552 553 554
const QVector<Waypoint *> UASWaypointManager::getGlobalFrameWaypointList()
{
    // TODO Keep this global frame list up to date
    // with complete waypoint list
    // instead of filtering on each request
    QVector<Waypoint*> wps;
pixhawk's avatar
pixhawk committed
555
    foreach (Waypoint* wp, waypointsEditable)
556 557 558
    {
        if (wp->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
559 560 561 562 563 564
            wps.append(wp);
        }
    }
    return wps;
}

lm's avatar
lm committed
565 566 567 568 569 570
const QVector<Waypoint *> UASWaypointManager::getGlobalFrameAndNavTypeWaypointList()
{
    // TODO Keep this global frame list up to date
    // with complete waypoint list
    // instead of filtering on each request
    QVector<Waypoint*> wps;
pixhawk's avatar
pixhawk committed
571
    foreach (Waypoint* wp, waypointsEditable)
572 573 574
    {
        if ((wp->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && wp->isNavigationType())
        {
lm's avatar
lm committed
575 576 577 578 579 580
            wps.append(wp);
        }
    }
    return wps;
}

LM's avatar
LM committed
581 582 583 584 585 586
const QVector<Waypoint *> UASWaypointManager::getNavTypeWaypointList()
{
    // TODO Keep this global frame list up to date
    // with complete waypoint list
    // instead of filtering on each request
    QVector<Waypoint*> wps;
pixhawk's avatar
pixhawk committed
587
    foreach (Waypoint* wp, waypointsEditable)
588 589 590
    {
        if (wp->isNavigationType())
        {
LM's avatar
LM committed
591 592 593 594 595 596
            wps.append(wp);
        }
    }
    return wps;
}

597 598
int UASWaypointManager::getIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
599
    return waypointsEditable.indexOf(wp);
600 601
}

602 603
int UASWaypointManager::getGlobalFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
604
    // Search through all waypointsEditable,
605 606
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
607
    foreach (Waypoint* p, waypointsEditable) {
608 609 610 611
        if (p->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
            if (p == wp)
            {
612 613 614 615 616 617 618 619 620
                return i;
            }
            i++;
        }
    }

    return -1;
}

lm's avatar
lm committed
621 622
int UASWaypointManager::getGlobalFrameAndNavTypeIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
623
    // Search through all waypointsEditable,
lm's avatar
lm committed
624 625
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
626
    foreach (Waypoint* p, waypointsEditable) {
627
        if ((p->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && p->isNavigationType())
628 629 630
        {
            if (p == wp)
            {
lm's avatar
lm committed
631 632 633 634 635 636 637 638 639
                return i;
            }
            i++;
        }
    }

    return -1;
}

LM's avatar
LM committed
640 641
int UASWaypointManager::getNavTypeIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
642
    // Search through all waypointsEditable,
LM's avatar
LM committed
643 644
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
645
    foreach (Waypoint* p, waypointsEditable)
646 647 648 649 650
    {
        if (p->isNavigationType())
        {
            if (p == wp)
            {
LM's avatar
LM committed
651 652 653 654 655 656 657 658 659
                return i;
            }
            i++;
        }
    }

    return -1;
}

660 661
int UASWaypointManager::getGlobalFrameCount()
{
pixhawk's avatar
pixhawk committed
662
    // Search through all waypointsEditable,
663 664
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
665
    foreach (Waypoint* p, waypointsEditable)
666
    {
667 668
        if (p->getFrame() == MAV_FRAME_GLOBAL || p->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
669 670 671 672 673 674 675
            i++;
        }
    }

    return i;
}

lm's avatar
lm committed
676 677
int UASWaypointManager::getGlobalFrameAndNavTypeCount()
{
pixhawk's avatar
pixhawk committed
678
    // Search through all waypointsEditable,
lm's avatar
lm committed
679 680
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
681
    foreach (Waypoint* p, waypointsEditable) {
682
        if ((p->getFrame() == MAV_FRAME_GLOBAL || p->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && p->isNavigationType())
683
        {
lm's avatar
lm committed
684 685 686 687 688 689 690
            i++;
        }
    }

    return i;
}

LM's avatar
LM committed
691 692
int UASWaypointManager::getNavTypeCount()
{
pixhawk's avatar
pixhawk committed
693
    // Search through all waypointsEditable,
LM's avatar
LM committed
694 695
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
696
    foreach (Waypoint* p, waypointsEditable) {
LM's avatar
LM committed
697 698 699 700 701 702 703 704
        if (p->isNavigationType()) {
            i++;
        }
    }

    return i;
}

705 706
int UASWaypointManager::getLocalFrameCount()
{
pixhawk's avatar
pixhawk committed
707
    // Search through all waypointsEditable,
708 709
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
710
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
711
    {
712
        if (p->getFrame() == MAV_FRAME_LOCAL_NED || p->getFrame() == MAV_FRAME_LOCAL_ENU)
lm's avatar
lm committed
713
        {
714 715 716 717 718 719 720 721 722
            i++;
        }
    }

    return i;
}

int UASWaypointManager::getLocalFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
723
    // Search through all waypointsEditable,
724 725
    // counting only those in local frame
    int i = 0;
pixhawk's avatar
pixhawk committed
726
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
727 728 729
    {
        if (p->getFrame() == MAV_FRAME_LOCAL_NED || p->getFrame() == MAV_FRAME_LOCAL_ENU)
        {
730 731
            if (p == wp)
            {
732 733 734 735 736 737 738 739 740 741 742
                return i;
            }
            i++;
        }
    }

    return -1;
}

int UASWaypointManager::getMissionFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
743
    // Search through all waypointsEditable,
744 745
    // counting only those in mission frame
    int i = 0;
pixhawk's avatar
pixhawk committed
746
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
747
    {
748 749 750 751
        if (p->getFrame() == MAV_FRAME_MISSION)
        {
            if (p == wp)
            {
752 753 754 755 756 757 758 759 760
                return i;
            }
            i++;
        }
    }

    return -1;
}

pixhawk's avatar
pixhawk committed
761 762 763 764 765

/**
 * @param readToEdit If true, incoming waypoints will be copied both to "edit"-tab and "view"-tab. Otherwise, only to "view"-tab.
 */
void UASWaypointManager::readWaypoints(bool readToEdit)
766
{
pixhawk's avatar
pixhawk committed
767
    read_to_edit = readToEdit;
768
    emit readGlobalWPFromUAS(true);
769
    if(current_state == WP_IDLE) {
770

771

772 773
        //Clear the old view-list before receiving the new one
        while(waypointsViewOnly.size()>0) {
774 775 776
            Waypoint *t = waypointsViewOnly[0];
            waypointsViewOnly.remove(0);
            delete t;
777
        }
778 779
        emit waypointViewOnlyListChanged();
        /* THIS PART WAS MOVED TO handleWaypointCount. THE EDIT-LIST SHOULD NOT BE CLEARED UNLESS THERE IS A RESPONSE FROM UAV.
780 781 782
        //Clear the old edit-list before receiving the new one
        if (read_to_edit == true){
            while(waypointsEditable.size()>0) {
783 784 785
                Waypoint *t = waypointsEditable[0];
                waypointsEditable.remove(0);
                delete t;
786
            }
787
            emit waypointEditableListChanged();
788
        }
789
        */
790 791
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
792 793 794

        current_state = WP_GETLIST;
        current_wp_id = 0;
795
        current_partner_systemid = uasid;
lm's avatar
lm committed
796
        current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed
797

798
        sendWaypointRequestList();
799

pixhawk's avatar
pixhawk committed
800 801 802
    }
}

803
void UASWaypointManager::writeWaypoints()
pixhawk's avatar
pixhawk committed
804
{
805
    if (current_state == WP_IDLE) {
pixhawk's avatar
pixhawk committed
806
        // Send clear all if count == 0
pixhawk's avatar
pixhawk committed
807
        if (waypointsEditable.count() > 0) {
pixhawk's avatar
pixhawk committed
808
            protocol_timer.start(PROTOCOL_TIMEOUT_MS);
809
            current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
810

pixhawk's avatar
pixhawk committed
811
            current_count = waypointsEditable.count();
pixhawk's avatar
pixhawk committed
812 813
            current_state = WP_SENDLIST;
            current_wp_id = 0;
814
            current_partner_systemid = uasid;
lm's avatar
lm committed
815
            current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed
816 817

            //clear local buffer
pixhawk's avatar
pixhawk committed
818 819 820
            // Why not replace with waypoint_buffer.clear() ?
            // because this will lead to memory leaks, the waypoint-structs
            // have to be deleted, clear() would only delete the pointers.
821
            while(!waypoint_buffer.empty()) {
pixhawk's avatar
pixhawk committed
822 823 824
                delete waypoint_buffer.back();
                waypoint_buffer.pop_back();
            }
pixhawk's avatar
pixhawk committed
825

826 827
            bool noCurrent = true;

pixhawk's avatar
pixhawk committed
828
            //copy waypoint data to local buffer
829
            for (int i=0; i < current_count; i++) {
lm's avatar
lm committed
830 831 832
                waypoint_buffer.push_back(new mavlink_mission_item_t);
                mavlink_mission_item_t *cur_d = waypoint_buffer.back();
                memset(cur_d, 0, sizeof(mavlink_mission_item_t));   //initialize with zeros
pixhawk's avatar
pixhawk committed
833
                const Waypoint *cur_s = waypointsEditable.at(i);
pixhawk's avatar
pixhawk committed
834 835

                cur_d->autocontinue = cur_s->getAutoContinue();
836
                cur_d->current = cur_s->getCurrent() & noCurrent;   //make sure only one current waypoint is selected, the first selected will be chosen
837 838
                cur_d->param1 = cur_s->getParam1();
                cur_d->param2 = cur_s->getParam2();
lm's avatar
lm committed
839 840
                cur_d->param3 = cur_s->getParam3();
                cur_d->param4 = cur_s->getParam4();
841
                cur_d->frame = cur_s->getFrame();
lm's avatar
lm committed
842
                cur_d->command = cur_s->getAction();
843
                cur_d->seq = i;     // don't read out the sequence number of the waypoint class
pixhawk's avatar
pixhawk committed
844 845 846
                cur_d->x = cur_s->getX();
                cur_d->y = cur_s->getY();
                cur_d->z = cur_s->getZ();
847 848 849

                if (cur_s->getCurrent() && noCurrent)
                    noCurrent = false;
850 851
                if (i == (current_count - 1) && noCurrent == true) //not a single waypoint was set as "current"
                    cur_d->current = true; // set the last waypoint as current. Or should it better be the first waypoint ?
pixhawk's avatar
pixhawk committed
852
            }
853

854 855 856



pixhawk's avatar
pixhawk committed
857
            //send the waypoint count to UAS (this starts the send transaction)
858
            sendWaypointCount();
Lorenz Meier's avatar
Lorenz Meier committed
859
        } else if (waypointsEditable.count() == 0)
860 861 862
        {
            sendWaypointClearAll();
        }
Lorenz Meier's avatar
Lorenz Meier committed
863 864 865
    }
    else
    {
866
        //we're in another transaction, ignore command
867
        qDebug() << "UASWaypointManager::sendWaypoints() doing something else ignoring command";
868
    }
pixhawk's avatar
pixhawk committed
869 870
}

871 872
void UASWaypointManager::sendWaypointClearAll()
{
lm's avatar
lm committed
873
    if (!uas) return;
874
    mavlink_message_t message;
lm's avatar
lm committed
875
    mavlink_mission_clear_all_t wpca;
876

877
    wpca.target_system = uasid;
lm's avatar
lm committed
878
    wpca.target_component = MAV_COMP_ID_MISSIONPLANNER;
879

880
    emit updateStatusString(QString("Clearing waypoint list..."));
881

lm's avatar
lm committed
882
    mavlink_msg_mission_clear_all_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpca);
883

884
    uas->sendMessage(message);
lm's avatar
lm committed
885
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
886 887 888 889
}

void UASWaypointManager::sendWaypointSetCurrent(quint16 seq)
{
lm's avatar
lm committed
890
    if (!uas) return;
891
    mavlink_message_t message;
lm's avatar
lm committed
892
    mavlink_mission_set_current_t wpsc;
893

894
    wpsc.target_system = uasid;
lm's avatar
lm committed
895
    wpsc.target_component = MAV_COMP_ID_MISSIONPLANNER;
896 897
    wpsc.seq = seq;

898
    emit updateStatusString(QString("Updating target waypoint..."));
899

lm's avatar
lm committed
900
    mavlink_msg_mission_set_current_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpsc);
901
    uas->sendMessage(message);
lm's avatar
lm committed
902
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
903 904 905 906
}

void UASWaypointManager::sendWaypointCount()
{
lm's avatar
lm committed
907
    if (!uas) return;
908
    mavlink_message_t message;
lm's avatar
lm committed
909
    mavlink_mission_count_t wpc;
910

911
    wpc.target_system = uasid;
lm's avatar
lm committed
912
    wpc.target_component = MAV_COMP_ID_MISSIONPLANNER;
913 914
    wpc.count = current_count;

915
    emit updateStatusString(QString("Starting to transmit waypoints..."));
916

lm's avatar
lm committed
917
    mavlink_msg_mission_count_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpc);
918
    uas->sendMessage(message);
lm's avatar
lm committed
919
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
920 921 922 923
}

void UASWaypointManager::sendWaypointRequestList()
{
lm's avatar
lm committed
924
    if (!uas) return;
925
    mavlink_message_t message;
lm's avatar
lm committed
926
    mavlink_mission_request_list_t wprl;
927

928
    wprl.target_system = uasid;
lm's avatar
lm committed
929
    wprl.target_component = MAV_COMP_ID_MISSIONPLANNER;
930

931
    emit updateStatusString(QString("Requesting waypoint list..."));
932

lm's avatar
lm committed
933
    mavlink_msg_mission_request_list_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wprl);
934
    uas->sendMessage(message);
lm's avatar
lm committed
935
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
936 937
}

938
void UASWaypointManager::sendWaypointRequest(quint16 seq)
pixhawk's avatar
pixhawk committed
939
{
lm's avatar
lm committed
940
    if (!uas) return;
941
    mavlink_message_t message;
lm's avatar
lm committed
942
    mavlink_mission_request_t wpr;
943

944
    wpr.target_system = uasid;
lm's avatar
lm committed
945
    wpr.target_component = MAV_COMP_ID_MISSIONPLANNER;
946 947
    wpr.seq = seq;

948
    emit updateStatusString(QString("Retrieving waypoint ID %1 of %2 total").arg(wpr.seq).arg(current_count));
pixhawk's avatar
pixhawk committed
949

lm's avatar
lm committed
950
    mavlink_msg_mission_request_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpr);
951
    uas->sendMessage(message);
lm's avatar
lm committed
952
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
pixhawk's avatar
pixhawk committed
953
}
pixhawk's avatar
pixhawk committed
954

pixhawk's avatar
pixhawk committed
955 956
void UASWaypointManager::sendWaypoint(quint16 seq)
{
lm's avatar
lm committed
957
    if (!uas) return;
pixhawk's avatar
pixhawk committed
958
    mavlink_message_t message;
959

960
    if (seq < waypoint_buffer.count()) {
pixhawk's avatar
pixhawk committed
961

lm's avatar
lm committed
962
        mavlink_mission_item_t *wp;
963

pixhawk's avatar
pixhawk committed
964 965

        wp = waypoint_buffer.at(seq);
966
        wp->target_system = uasid;
lm's avatar
lm committed
967
        wp->target_component = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed
968

969
        emit updateStatusString(QString("Sending waypoint ID %1 of %2 total").arg(wp->seq).arg(current_count));
970

pixhawk's avatar
pixhawk committed
971

lm's avatar
lm committed
972
        mavlink_msg_mission_item_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, wp);
973
        uas->sendMessage(message);
lm's avatar
lm committed
974
        QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
pixhawk's avatar
pixhawk committed
975
    }
pixhawk's avatar
pixhawk committed
976
}
977 978 979

void UASWaypointManager::sendWaypointAck(quint8 type)
{
lm's avatar
lm committed
980
    if (!uas) return;
981
    mavlink_message_t message;
lm's avatar
lm committed
982
    mavlink_mission_ack_t wpa;
983

984
    wpa.target_system = uasid;
lm's avatar
lm committed
985
    wpa.target_component = MAV_COMP_ID_MISSIONPLANNER;
986 987
    wpa.type = type;

lm's avatar
lm committed
988
    mavlink_msg_mission_ack_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpa);
989
    uas->sendMessage(message);
lm's avatar
lm committed
990
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
991
}