UASWaypointManager.cc 26.4 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

lm's avatar
lm committed
5
(c) 2009, 2010 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
//#include "MainWindow.h"
36

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

41
UASWaypointManager::UASWaypointManager(UAS &_uas)
pixhawk's avatar
pixhawk committed
42
        : uas(_uas),
43
        current_retries(0),
pixhawk's avatar
pixhawk committed
44 45 46 47
        current_wp_id(0),
        current_count(0),
        current_state(WP_IDLE),
        current_partner_systemid(0),
pixhawk's avatar
pixhawk committed
48 49 50 51 52 53 54
        current_partner_compid(0),
        protocol_timer(this)
{
    connect(&protocol_timer, SIGNAL(timeout()), this, SLOT(timeout()));
}

void UASWaypointManager::timeout()
55
{
56 57 58 59
    if (current_retries > 0)
    {
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries--;
60
        emit updateStatusString(tr("Timeout, retrying (retries left: %1)").arg(current_retries));
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
        qDebug() << "Timeout, retrying (retries left:" << current_retries << ")";
        if (current_state == WP_GETLIST)
        {
            sendWaypointRequestList();
        }
        else if (current_state == WP_GETLIST_GETWPS)
        {
            sendWaypointRequest(current_wp_id);
        }
        else if (current_state == WP_SENDLIST)
        {
            sendWaypointCount();
        }
        else if (current_state == WP_SENDLIST_SENDWPS)
        {
            sendWaypoint(current_wp_id);
        }
        else if (current_state == WP_CLEARLIST)
        {
            sendWaypointClearAll();
        }
        else if (current_state == WP_SETCURRENT)
        {
            sendWaypointSetCurrent(current_wp_id);
        }
    }
    else
    {
        protocol_timer.stop();
pixhawk's avatar
pixhawk committed
90

91
        qDebug() << "Waypoint transaction (state=" << current_state << ") timed out going to state WP_IDLE";
pixhawk's avatar
pixhawk committed
92

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

95 96 97 98 99 100
        current_state = WP_IDLE;
        current_count = 0;
        current_wp_id = 0;
        current_partner_systemid = 0;
        current_partner_compid = 0;
    }
101 102 103 104 105 106
}

void UASWaypointManager::handleWaypointCount(quint8 systemId, quint8 compId, quint16 count)
{
    if (current_state == WP_GETLIST && systemId == current_partner_systemid && compId == current_partner_compid)
    {
107 108 109
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

110
        qDebug() << "got waypoint count (" << count << ") from ID " << systemId;
pixhawk's avatar
pixhawk committed
111

pixhawk's avatar
pixhawk committed
112 113 114 115 116
        if (count > 0)
        {
            current_count = count;
            current_wp_id = 0;
            current_state = WP_GETLIST_GETWPS;
117

pixhawk's avatar
pixhawk committed
118 119 120 121
            sendWaypointRequest(current_wp_id);
        }
        else
        {
122
            protocol_timer.stop();
pixhawk's avatar
pixhawk committed
123 124
            emit updateStatusString("done.");
            qDebug() << "No waypoints on UAS " << systemId;
125 126 127 128 129
            current_state = WP_IDLE;
            current_count = 0;
            current_wp_id = 0;
            current_partner_systemid = 0;
            current_partner_compid = 0;
pixhawk's avatar
pixhawk committed
130
        }
131
    }
132 133 134 135
    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);
    }
136 137
}

pixhawk's avatar
pixhawk committed
138 139 140 141
void UASWaypointManager::handleWaypoint(quint8 systemId, quint8 compId, mavlink_waypoint_t *wp)
{
    if (systemId == current_partner_systemid && compId == current_partner_compid && current_state == WP_GETLIST_GETWPS && wp->seq == current_wp_id)
    {
142 143 144
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

pixhawk's avatar
pixhawk committed
145 146
        if(wp->seq == current_wp_id)
        {
147
            //qDebug() << "Got WP: " << wp->seq << wp->x <<  wp->y << wp->z << wp->param4 << "auto:" << wp->autocontinue << "curr:" << wp->current << wp->param1 << wp->param2 << "Frame:"<< (MAV_FRAME) wp->frame << "Command:" << (MAV_CMD) wp->command;
lm's avatar
lm committed
148
            Waypoint *lwp = 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);
pixhawk's avatar
pixhawk committed
149
            addWaypoint(lwp, false);
pixhawk's avatar
pixhawk committed
150 151 152 153 154 155

            //get next waypoint
            current_wp_id++;

            if(current_wp_id < current_count)
            {
156
                sendWaypointRequest(current_wp_id);
pixhawk's avatar
pixhawk committed
157 158 159
            }
            else
            {
160 161
                sendWaypointAck(0);

pixhawk's avatar
pixhawk committed
162
                // all waypoints retrieved, change state to idle
pixhawk's avatar
pixhawk committed
163 164 165 166 167
                current_state = WP_IDLE;
                current_count = 0;
                current_wp_id = 0;
                current_partner_systemid = 0;
                current_partner_compid = 0;
168

pixhawk's avatar
pixhawk committed
169
                protocol_timer.stop();
170
                emit readGlobalWPFromUAS(false);
171 172
                emit updateStatusString("done.");

pixhawk's avatar
pixhawk committed
173
                qDebug() << "got all waypoints from ID " << systemId;
pixhawk's avatar
pixhawk committed
174 175 176 177
            }
        }
        else
        {
178
            emit updateStatusString(tr("Waypoint ID mismatch, rejecting waypoint"));
pixhawk's avatar
pixhawk committed
179 180
        }
    }
181 182 183 184
    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);
    }
pixhawk's avatar
pixhawk committed
185 186
}

187 188
void UASWaypointManager::handleWaypointAck(quint8 systemId, quint8 compId, mavlink_waypoint_ack_t *wpa)
{
189
    if (systemId == current_partner_systemid && compId == current_partner_compid)
190
    {
191
        if((current_state == WP_SENDLIST || current_state == WP_SENDLIST_SENDWPS) && (current_wp_id == waypoint_buffer.count()-1 && wpa->type == 0))
192 193 194
        {
            //all waypoints sent and ack received
            protocol_timer.stop();
195
            current_state = WP_IDLE;
196 197 198
            emit updateStatusString("done.");
            qDebug() << "sent all waypoints to ID " << systemId;
        }
199 200 201 202 203 204 205
        else if(current_state == WP_CLEARLIST)
        {
            protocol_timer.stop();
            current_state = WP_IDLE;
            emit updateStatusString("done.");
            qDebug() << "cleared waypoint list of ID " << systemId;
        }
206 207 208
    }
}

pixhawk's avatar
pixhawk committed
209
void UASWaypointManager::handleWaypointRequest(quint8 systemId, quint8 compId, mavlink_waypoint_request_t *wpr)
210
{
211
    if (systemId == current_partner_systemid && compId == current_partner_compid && ((current_state == WP_SENDLIST && wpr->seq == 0) || (current_state == WP_SENDLIST_SENDWPS && (wpr->seq == current_wp_id || wpr->seq == current_wp_id + 1))))
pixhawk's avatar
pixhawk committed
212
    {
213 214
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
215

216
        if (wpr->seq < waypoint_buffer.count())
pixhawk's avatar
pixhawk committed
217
        {
pixhawk's avatar
pixhawk committed
218 219 220
            current_state = WP_SENDLIST_SENDWPS;
            current_wp_id = wpr->seq;
            sendWaypoint(current_wp_id);
pixhawk's avatar
pixhawk committed
221 222 223 224 225 226
        }
        else
        {
            //TODO: Error message or something
        }
    }
227 228 229 230
    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);
    }
231 232
}

pixhawk's avatar
pixhawk committed
233
void UASWaypointManager::handleWaypointReached(quint8 systemId, quint8 compId, mavlink_waypoint_reached_t *wpr)
234
{
pixhawk's avatar
pixhawk committed
235 236 237 238
    if (systemId == uas.getUASID() && compId == MAV_COMP_ID_WAYPOINTPLANNER)
    {
        emit updateStatusString(QString("Reached waypoint %1").arg(wpr->seq));
    }
pixhawk's avatar
pixhawk committed
239 240
}

241
void UASWaypointManager::handleWaypointCurrent(quint8 systemId, quint8 compId, mavlink_waypoint_current_t *wpc)
pixhawk's avatar
pixhawk committed
242
{
pixhawk's avatar
pixhawk committed
243 244
    if (systemId == uas.getUASID() && compId == MAV_COMP_ID_WAYPOINTPLANNER)
    {
245
        // FIXME Petri
246 247 248 249
        if (current_state == WP_SETCURRENT)
        {
            protocol_timer.stop();
            current_state = WP_IDLE;
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266

            // update the local main storage
            if (wpc->seq < waypoints.size())
            {
                for(int i = 0; i < waypoints.size(); i++)
                {
                    if (waypoints[i]->getId() == wpc->seq)
                    {
                        waypoints[i]->setCurrent(true);
                    }
                    else
                    {
                        waypoints[i]->setCurrent(false);
                    }
                }
            }

267
            //qDebug() << "Updated waypoints list";
268
        }
269 270 271
        emit updateStatusString(QString("New current waypoint %1").arg(wpc->seq));
        //emit update to UI widgets
        emit currentWaypointChanged(wpc->seq);
272
        //qDebug() << "new current waypoint" << wpc->seq;
pixhawk's avatar
pixhawk committed
273
    }
pixhawk's avatar
pixhawk committed
274 275
}

276 277
void UASWaypointManager::notifyOfChange(Waypoint* wp)
{
278
    qDebug() << "WAYPOINT CHANGED: ID:" << wp->getId();
279 280 281 282 283 284 285 286 287 288
    // If only one waypoint was changed, emit only WP signal
    if (wp != NULL)
    {
        emit waypointChanged(uas.getUASID(), wp);
    }
    else
    {
        emit waypointListChanged();
        emit waypointListChanged(uas.getUASID());
    }
289 290
}

291
int UASWaypointManager::setCurrentWaypoint(quint16 seq)
pixhawk's avatar
pixhawk committed
292
{
293
    if (seq < waypoints.size())
294
    {
295 296 297 298 299 300 301 302 303 304 305 306 307 308
        if(current_state == WP_IDLE)
        {
            //update local main storage
            for(int i = 0; i < waypoints.size(); i++)
            {
                if (waypoints[i]->getId() == seq)
                {
                    waypoints[i]->setCurrent(true);
                }
                else
                {
                    waypoints[i]->setCurrent(false);
                }
            }
309

310
            //TODO: signal changed waypoint list
311

312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
            //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;
            current_partner_systemid = uas.getUASID();
            current_partner_compid = MAV_COMP_ID_WAYPOINTPLANNER;

            sendWaypointSetCurrent(current_wp_id);

            //emit waypointListChanged();

            return 0;
        }
    }
    return -1;
}

pixhawk's avatar
pixhawk committed
331 332 333 334
/**
 * @param enforceFirstActive Enforces that the first waypoint is set as active
 */
void UASWaypointManager::addWaypoint(Waypoint *wp, bool enforceFirstActive)
335 336 337 338
{
    if (wp)
    {
        wp->setId(waypoints.size());
pixhawk's avatar
pixhawk committed
339
        if (enforceFirstActive && waypoints.size() == 0) wp->setCurrent(true);
340
        waypoints.insert(waypoints.size(), wp);
341
        connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChange(Waypoint*)));
342

343
        emit waypointListChanged();
344
        emit waypointListChanged(uas.getUASID());
345
    }
pixhawk's avatar
pixhawk committed
346 347
}

348
int UASWaypointManager::removeWaypoint(quint16 seq)
349 350 351 352 353 354 355 356 357 358 359
{
    if (seq < waypoints.size())
    {
        Waypoint *t = waypoints[seq];
        waypoints.remove(seq);
        delete t;

        for(int i = seq; i < waypoints.size(); i++)
        {
            waypoints[i]->setId(i);
        }
Alejandro's avatar
Alejandro committed
360

361
        emit waypointListChanged();
362
        emit waypointListChanged(uas.getUASID());
363 364 365 366 367
        return 0;
    }
    return -1;
}

368
void UASWaypointManager::moveWaypoint(quint16 cur_seq, quint16 new_seq)
369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392
{
    if (cur_seq != new_seq && cur_seq < waypoints.size() && new_seq < waypoints.size())
    {
        Waypoint *t = waypoints[cur_seq];
        if (cur_seq < new_seq)
        {
            for (int i = cur_seq; i < new_seq; i++)
            {
                waypoints[i] = waypoints[i+1];
                //waypoints[i]->setId(i);       // let the local IDs stay the same so that they can be found more easily by the user
            }
        }
        else
        {
            for (int i = cur_seq; i > new_seq; i--)
            {
                waypoints[i] = waypoints[i-1];
               // waypoints[i]->setId(i);
            }
        }
        waypoints[new_seq] = t;
        //waypoints[new_seq]->setId(new_seq);

        emit waypointListChanged();
393
        emit waypointListChanged(uas.getUASID());
394 395 396
    }
}

397
void UASWaypointManager::saveWaypoints(const QString &saveFile)
398 399 400 401 402 403
{
    QFile file(saveFile);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
        return;

    QTextStream out(&file);
404 405

    //write the waypoint list version to the first line for compatibility check
lm's avatar
lm committed
406
    out << "QGC WPL 110\r\n";
407

408 409
    for (int i = 0; i < waypoints.size(); i++)
    {
410
        waypoints[i]->setId(i);
411 412 413 414 415
        waypoints[i]->save(out);
    }
    file.close();
}

416
void UASWaypointManager::loadWaypoints(const QString &loadFile)
417 418 419 420 421 422 423 424 425 426 427 428 429
{
    QFile file(loadFile);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
        return;

    while(waypoints.size()>0)
    {
        Waypoint *t = waypoints[0];
        waypoints.remove(0);
        delete t;
    }

    QTextStream in(&file);
430 431 432

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

lm's avatar
lm committed
433
    if (!(version.size() == 3 && version[0] == "QGC" && version[1] == "WPL" && version[2] == "110"))
434
    {
435 436
        emit updateStatusString(tr("The waypoint file is not compatible with the current version of QGroundControl."));
        //MainWindow::instance()->showCriticalMessage(tr("Error loading waypoint file"),tr("The waypoint file is not compatible with the current version of QGroundControl."));
437 438 439 440
    }
    else
    {
        while (!in.atEnd())
441
        {
442 443 444 445 446 447 448 449
            Waypoint *t = new Waypoint();
            if(t->load(in))
            {
                t->setId(waypoints.size());
                waypoints.insert(waypoints.size(), t);
            }
            else
            {
450 451
                emit updateStatusString(tr("The waypoint file is corrupted. Load operation only partly succesful."));
                //MainWindow::instance()->showCriticalMessage(tr("Error loading waypoint file"),);
452 453
                break;
            }
454 455
        }
    }
456

457 458
    file.close();

459
    emit loadWPFile();
460
    emit waypointListChanged();
461
    emit waypointListChanged(uas.getUASID());
462 463
}

464

465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482
//void UASWaypointManager::globalAddWaypoint(Waypoint *wp)
//{
//    // FIXME Will be removed
//    Q_UNUSED(wp);
//}

//int UASWaypointManager::globalRemoveWaypoint(quint16 seq)
//{
//    // FIXME Will be removed
//    Q_UNUSED(seq);
//    return 0;
//}

//void UASWaypointManager::waypointUpdated(Waypoint*)
//{
//    // FIXME Add rate limiter
//    emit waypointListChanged(uas.getUASID());
//}
pixhawk's avatar
pixhawk committed
483

484
void UASWaypointManager::clearWaypointList()
pixhawk's avatar
pixhawk committed
485 486 487
{
    if(current_state == WP_IDLE)
    {
pixhawk's avatar
pixhawk committed
488
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
489
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
490

491 492
        current_state = WP_CLEARLIST;
        current_wp_id = 0;
493 494
        current_partner_systemid = uas.getUASID();
        current_partner_compid = MAV_COMP_ID_WAYPOINTPLANNER;
pixhawk's avatar
pixhawk committed
495

496
        sendWaypointClearAll();
497 498 499
    }
}

500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515
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;
    foreach (Waypoint* wp, waypoints)
    {
        if (wp->getFrame() == MAV_FRAME_GLOBAL)
        {
            wps.append(wp);
        }
    }
    return wps;
}

lm's avatar
lm committed
516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531
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;
    foreach (Waypoint* wp, waypoints)
    {
        if (wp->getFrame() == MAV_FRAME_GLOBAL && wp->isNavigationType())
        {
            wps.append(wp);
        }
    }
    return wps;
}

532 533 534 535 536
int UASWaypointManager::getIndexOf(Waypoint* wp)
{
    return waypoints.indexOf(wp);
}

537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
int UASWaypointManager::getGlobalFrameIndexOf(Waypoint* wp)
{
    // Search through all waypoints,
    // counting only those in global frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_GLOBAL)
        {
            if (p == wp)
            {
                return i;
            }
            i++;
        }
    }

    return -1;
}

lm's avatar
lm committed
557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576
int UASWaypointManager::getGlobalFrameAndNavTypeIndexOf(Waypoint* wp)
{
    // Search through all waypoints,
    // counting only those in global frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_GLOBAL && p->isNavigationType())
        {
            if (p == wp)
            {
                return i;
            }
            i++;
        }
    }

    return -1;
}

577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592
int UASWaypointManager::getGlobalFrameCount()
{
    // Search through all waypoints,
    // counting only those in global frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_GLOBAL)
        {
            i++;
        }
    }

    return i;
}

lm's avatar
lm committed
593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608
int UASWaypointManager::getGlobalFrameAndNavTypeCount()
{
    // Search through all waypoints,
    // counting only those in global frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_GLOBAL && p->isNavigationType())
        {
            i++;
        }
    }

    return i;
}

609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 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 664
int UASWaypointManager::getLocalFrameCount()
{
    // Search through all waypoints,
    // counting only those in global frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_GLOBAL)
        {
            i++;
        }
    }

    return i;
}

int UASWaypointManager::getLocalFrameIndexOf(Waypoint* wp)
{
    // Search through all waypoints,
    // counting only those in local frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_LOCAL)
        {
            if (p == wp)
            {
                return i;
            }
            i++;
        }
    }

    return -1;
}

int UASWaypointManager::getMissionFrameIndexOf(Waypoint* wp)
{
    // Search through all waypoints,
    // counting only those in mission frame
    int i = 0;
    foreach (Waypoint* p, waypoints)
    {
        if (p->getFrame() == MAV_FRAME_MISSION)
        {
            if (p == wp)
            {
                return i;
            }
            i++;
        }
    }

    return -1;
}

665 666
void UASWaypointManager::readWaypoints()
{
667
    emit readGlobalWPFromUAS(true);
668 669
    if(current_state == WP_IDLE)
    {
670 671
        while(waypoints.size()>0)
        {
672
            delete waypoints.back();
673
            waypoints.pop_back();
674

675 676
        }

677 678
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
679 680 681 682 683 684

        current_state = WP_GETLIST;
        current_wp_id = 0;
        current_partner_systemid = uas.getUASID();
        current_partner_compid = MAV_COMP_ID_WAYPOINTPLANNER;

685
        sendWaypointRequestList();
686

pixhawk's avatar
pixhawk committed
687 688 689
    }
}

690
void UASWaypointManager::writeWaypoints()
pixhawk's avatar
pixhawk committed
691
{
692 693
    if (current_state == WP_IDLE)
    {
pixhawk's avatar
pixhawk committed
694
        // Send clear all if count == 0
695
        if (waypoints.count() > 0)
696
        {
pixhawk's avatar
pixhawk committed
697
            protocol_timer.start(PROTOCOL_TIMEOUT_MS);
698
            current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed
699

700
            current_count = waypoints.count();
pixhawk's avatar
pixhawk committed
701 702 703 704 705 706
            current_state = WP_SENDLIST;
            current_wp_id = 0;
            current_partner_systemid = uas.getUASID();
            current_partner_compid = MAV_COMP_ID_WAYPOINTPLANNER;

            //clear local buffer
pixhawk's avatar
pixhawk committed
707 708 709
            // 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.
pixhawk's avatar
pixhawk committed
710 711 712 713 714
            while(!waypoint_buffer.empty())
            {
                delete waypoint_buffer.back();
                waypoint_buffer.pop_back();
            }
pixhawk's avatar
pixhawk committed
715

716 717
            bool noCurrent = true;

pixhawk's avatar
pixhawk committed
718 719 720 721 722 723
            //copy waypoint data to local buffer
            for (int i=0; i < current_count; i++)
            {
                waypoint_buffer.push_back(new mavlink_waypoint_t);
                mavlink_waypoint_t *cur_d = waypoint_buffer.back();
                memset(cur_d, 0, sizeof(mavlink_waypoint_t));   //initialize with zeros
724
                const Waypoint *cur_s = waypoints.at(i);
pixhawk's avatar
pixhawk committed
725 726

                cur_d->autocontinue = cur_s->getAutoContinue();
727
                cur_d->current = cur_s->getCurrent() & noCurrent;   //make sure only one current waypoint is selected, the first selected will be chosen
728 729
                cur_d->param1 = cur_s->getParam1();
                cur_d->param2 = cur_s->getParam2();
lm's avatar
lm committed
730 731
                cur_d->param3 = cur_s->getParam3();
                cur_d->param4 = cur_s->getParam4();
732
                cur_d->frame = cur_s->getFrame();
lm's avatar
lm committed
733
                cur_d->command = cur_s->getAction();
734
                cur_d->seq = i;     // don't read out the sequence number of the waypoint class
pixhawk's avatar
pixhawk committed
735 736 737
                cur_d->x = cur_s->getX();
                cur_d->y = cur_s->getY();
                cur_d->z = cur_s->getZ();
738 739 740

                if (cur_s->getCurrent() && noCurrent)
                    noCurrent = false;
741 742
                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
743
            }
744

745 746 747



pixhawk's avatar
pixhawk committed
748
            //send the waypoint count to UAS (this starts the send transaction)
749
            sendWaypointCount();
pixhawk's avatar
pixhawk committed
750
        }
751
    }
752 753 754 755
    else if (waypoints.count() == 0)
    {
        sendWaypointClearAll();
    }
756 757 758 759 760
    else
    {
        //we're in another transaction, ignore command
        qDebug() << "UASWaypointManager::sendWaypoints() doing something else ignoring command";
    }
pixhawk's avatar
pixhawk committed
761 762
}

763 764 765 766 767 768 769 770
void UASWaypointManager::sendWaypointClearAll()
{
    mavlink_message_t message;
    mavlink_waypoint_clear_all_t wpca;

    wpca.target_system = uas.getUASID();
    wpca.target_component = MAV_COMP_ID_WAYPOINTPLANNER;

771
    emit updateStatusString(QString("Clearing waypoint list..."));
772 773 774

    mavlink_msg_waypoint_clear_all_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wpca);
    uas.sendMessage(message);
775
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
776 777 778 779 780 781 782 783 784 785 786 787 788

    qDebug() << "sent waypoint clear all to ID " << wpca.target_system;
}

void UASWaypointManager::sendWaypointSetCurrent(quint16 seq)
{
    mavlink_message_t message;
    mavlink_waypoint_set_current_t wpsc;

    wpsc.target_system = uas.getUASID();
    wpsc.target_component = MAV_COMP_ID_WAYPOINTPLANNER;
    wpsc.seq = seq;

789
    emit updateStatusString(QString("Updating target waypoint..."));
790 791 792

    mavlink_msg_waypoint_set_current_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wpsc);
    uas.sendMessage(message);
793
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
794 795 796 797 798 799 800 801 802 803 804 805 806

    qDebug() << "sent waypoint set current (" << wpsc.seq << ") to ID " << wpsc.target_system;
}

void UASWaypointManager::sendWaypointCount()
{
    mavlink_message_t message;
    mavlink_waypoint_count_t wpc;

    wpc.target_system = uas.getUASID();
    wpc.target_component = MAV_COMP_ID_WAYPOINTPLANNER;
    wpc.count = current_count;

807
    qDebug() << "sent waypoint count (" << wpc.count << ") to ID " << wpc.target_system;
808
    emit updateStatusString(QString("Starting to transmit waypoints..."));
809 810 811

    mavlink_msg_waypoint_count_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wpc);
    uas.sendMessage(message);
812
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
813 814 815 816 817 818 819 820 821 822 823 824

    qDebug() << "sent waypoint count (" << wpc.count << ") to ID " << wpc.target_system;
}

void UASWaypointManager::sendWaypointRequestList()
{
    mavlink_message_t message;
    mavlink_waypoint_request_list_t wprl;

    wprl.target_system = uas.getUASID();
    wprl.target_component = MAV_COMP_ID_WAYPOINTPLANNER;

825
    emit updateStatusString(QString("Requesting waypoint list..."));
826 827 828

    mavlink_msg_waypoint_request_list_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wprl);
    uas.sendMessage(message);
829
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
830 831

    qDebug() << "sent waypoint list request to ID " << wprl.target_system;
832 833


834 835
}

836
void UASWaypointManager::sendWaypointRequest(quint16 seq)
pixhawk's avatar
pixhawk committed
837
{
838 839 840 841 842 843 844
    mavlink_message_t message;
    mavlink_waypoint_request_t wpr;

    wpr.target_system = uas.getUASID();
    wpr.target_component = MAV_COMP_ID_WAYPOINTPLANNER;
    wpr.seq = seq;

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

847 848
    mavlink_msg_waypoint_request_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wpr);
    uas.sendMessage(message);
849
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
850 851

    qDebug() << "sent waypoint request (" << wpr.seq << ") to ID " << wpr.target_system;
pixhawk's avatar
pixhawk committed
852
}
pixhawk's avatar
pixhawk committed
853

pixhawk's avatar
pixhawk committed
854 855 856
void UASWaypointManager::sendWaypoint(quint16 seq)
{
    mavlink_message_t message;
857
    qDebug() <<" WP Buffer count: "<<waypoint_buffer.count();
pixhawk's avatar
pixhawk committed
858 859 860

    if (seq < waypoint_buffer.count())
    {
861

pixhawk's avatar
pixhawk committed
862 863
        mavlink_waypoint_t *wp;

864

pixhawk's avatar
pixhawk committed
865 866 867 868
        wp = waypoint_buffer.at(seq);
        wp->target_system = uas.getUASID();
        wp->target_component = MAV_COMP_ID_WAYPOINTPLANNER;

869
        emit updateStatusString(QString("Sending waypoint ID %1 of %2 total").arg(wp->seq).arg(current_count));
pixhawk's avatar
pixhawk committed
870

871 872
        qDebug() << "sent waypoint (" << wp->seq << ") to ID " << wp->target_system<<" WP Buffer count: "<<waypoint_buffer.count();

pixhawk's avatar
pixhawk committed
873 874
        mavlink_msg_waypoint_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, wp);
        uas.sendMessage(message);
875
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
pixhawk's avatar
pixhawk committed
876 877

    }
pixhawk's avatar
pixhawk committed
878
}
879 880 881 882 883 884 885 886 887 888 889 890

void UASWaypointManager::sendWaypointAck(quint8 type)
{
    mavlink_message_t message;
    mavlink_waypoint_ack_t wpa;

    wpa.target_system = uas.getUASID();
    wpa.target_component = MAV_COMP_ID_WAYPOINTPLANNER;
    wpa.type = type;

    mavlink_msg_waypoint_ack_encode(uas.mavlink->getSystemId(), uas.mavlink->getComponentId(), &message, &wpa);
    uas.sendMessage(message);
891
    MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
892 893 894

    qDebug() << "sent waypoint ack (" << wpa.type << ") to ID " << wpa.target_system;
}