Skip to content
UASWaypointManager.cc 31.9 KiB
Newer Older
pixhawk's avatar
pixhawk committed
/*=====================================================================

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

lm's avatar
lm committed
(c) 2009, 2010 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
pixhawk's avatar
pixhawk committed

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

lm's avatar
lm committed
    QGROUNDCONTROL is free software: you can redistribute it and/or modify
pixhawk's avatar
pixhawk committed
    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
    QGROUNDCONTROL is distributed in the hope that it will be useful,
pixhawk's avatar
pixhawk committed
    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
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
pixhawk's avatar
pixhawk committed

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

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

#include "UASWaypointManager.h"
#include "UAS.h"
#include "mavlink_types.h"
#define PROTOCOL_TIMEOUT_MS 2000    ///< maximum time to wait for pending messages until timeout
lm's avatar
lm committed
#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

lm's avatar
lm committed
UASWaypointManager::UASWaypointManager(UAS* _uas)
    : uas(_uas),
      current_retries(0),
      current_wp_id(0),
      current_count(0),
      current_state(WP_IDLE),
      current_partner_systemid(0),
      current_partner_compid(0),
pixhawk's avatar
pixhawk committed
{
lm's avatar
lm committed
        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;
pixhawk's avatar
pixhawk committed
}

void UASWaypointManager::timeout()
    if (current_retries > 0) {
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries--;
        emit updateStatusString(tr("Timeout, retrying (retries left: %1)").arg(current_retries));
        // // 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) {
        } 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);
        }
pixhawk's avatar
pixhawk committed

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

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

        current_state = WP_IDLE;
        current_count = 0;
        current_wp_id = 0;
        current_partner_systemid = 0;
        current_partner_compid = 0;
    }
void UASWaypointManager::handleLocalPositionChanged(UASInterface* mav, double x, double y, double z, quint64 time)
{
    Q_UNUSED(mav);
    Q_UNUSED(time);
pixhawk's avatar
pixhawk committed
    if (waypointsEditable.count() > 0 && currentWaypointEditable && (currentWaypointEditable->getFrame() == MAV_FRAME_LOCAL_NED || currentWaypointEditable->getFrame() == MAV_FRAME_LOCAL_ENU))
pixhawk's avatar
pixhawk committed
        double xdiff = x-currentWaypointEditable->getX();
        double ydiff = y-currentWaypointEditable->getY();
        double zdiff = z-currentWaypointEditable->getZ();
        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)
{
    Q_UNUSED(mav);
    Q_UNUSED(time);
    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);
    }
void UASWaypointManager::handleWaypointCount(quint8 systemId, quint8 compId, quint16 count)
{
    if (current_state == WP_GETLIST && systemId == current_partner_systemid && compId == current_partner_compid) {
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

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

        if (count > 0) {
pixhawk's avatar
pixhawk committed
            current_count = count;
            current_wp_id = 0;
            current_state = WP_GETLIST_GETWPS;
pixhawk's avatar
pixhawk committed
            sendWaypointRequest(current_wp_id);
            protocol_timer.stop();
pixhawk's avatar
pixhawk committed
            emit updateStatusString("done.");
            // // qDebug() << "No waypoints on UAS " << systemId;
            current_state = WP_IDLE;
            current_count = 0;
            current_wp_id = 0;
            current_partner_systemid = 0;
            current_partner_compid = 0;
pixhawk's avatar
pixhawk committed
        }
    } 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);
lm's avatar
lm committed
void UASWaypointManager::handleWaypoint(quint8 systemId, quint8 compId, mavlink_mission_item_t *wp)
pixhawk's avatar
pixhawk committed
{
    if (systemId == current_partner_systemid && compId == current_partner_compid && current_state == WP_GETLIST_GETWPS && wp->seq == current_wp_id) {
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;

        if(wp->seq == current_wp_id) {
            //// // 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;
pixhawk's avatar
pixhawk committed

            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

            //get next waypoint
            current_wp_id++;

            if(current_wp_id < current_count) {
                sendWaypointRequest(current_wp_id);
pixhawk's avatar
pixhawk committed
                // all waypoints retrieved, change state to idle
pixhawk's avatar
pixhawk committed
                current_state = WP_IDLE;
                current_count = 0;
                current_wp_id = 0;
                current_partner_systemid = 0;
                current_partner_compid = 0;
pixhawk's avatar
pixhawk committed
                protocol_timer.stop();
                emit readGlobalWPFromUAS(false);
                //if (currentWaypointEditable) emit currentWaypointChanged(currentWaypointEditable->getId());
                emit updateStatusString("done.");

                // // qDebug() << "got all waypoints from ID " << systemId;
pixhawk's avatar
pixhawk committed
            }
            emit updateStatusString(tr("Waypoint ID mismatch, rejecting waypoint"));
pixhawk's avatar
pixhawk committed
        }
    } 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
}

lm's avatar
lm committed
void UASWaypointManager::handleWaypointAck(quint8 systemId, quint8 compId, mavlink_mission_ack_t *wpa)
    if (systemId == current_partner_systemid && compId == current_partner_compid) {
        if((current_state == WP_SENDLIST || current_state == WP_SENDLIST_SENDWPS) && (current_wp_id == waypoint_buffer.count()-1 && wpa->type == 0)) {
            //all waypoints sent and ack received
            protocol_timer.stop();
            current_state = WP_IDLE;
            emit updateStatusString("done.");
            // // qDebug() << "sent all waypoints to ID " << systemId;
        } else if(current_state == WP_CLEARLIST) {
            protocol_timer.stop();
            current_state = WP_IDLE;
            emit updateStatusString("done.");
            // // qDebug() << "cleared waypoint list of ID " << systemId;
lm's avatar
lm committed
void UASWaypointManager::handleWaypointRequest(quint8 systemId, quint8 compId, mavlink_mission_request_t *wpr)
    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)))) {
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
        if (wpr->seq < waypoint_buffer.count()) {
pixhawk's avatar
pixhawk committed
            current_state = WP_SENDLIST_SENDWPS;
            current_wp_id = wpr->seq;
            sendWaypoint(current_wp_id);
pixhawk's avatar
pixhawk committed
            //TODO: Error message or something
        }
    } 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);
lm's avatar
lm committed
void UASWaypointManager::handleWaypointReached(quint8 systemId, quint8 compId, mavlink_mission_item_reached_t *wpr)
lm's avatar
lm committed
    if (!uas) return;
    if (systemId == uasid) {
pixhawk's avatar
pixhawk committed
        emit updateStatusString(QString("Reached waypoint %1").arg(wpr->seq));
    }
pixhawk's avatar
pixhawk committed
}

lm's avatar
lm committed
void UASWaypointManager::handleWaypointCurrent(quint8 systemId, quint8 compId, mavlink_mission_current_t *wpc)
pixhawk's avatar
pixhawk committed
{
lm's avatar
lm committed
    if (!uas) return;
    if (systemId == uasid) {
        if (current_state == WP_SETCURRENT) {
            protocol_timer.stop();
            current_state = WP_IDLE;

            // update the local main storage
pixhawk's avatar
pixhawk committed
            if (wpc->seq < waypointsViewOnly.size()) {
                for(int i = 0; i < waypointsViewOnly.size(); i++) {
                    if (waypointsViewOnly[i]->getId() == wpc->seq) {
                        waypointsViewOnly[i]->setCurrent(true);
                        //currentWaypointEditable = waypoints[i];
pixhawk's avatar
pixhawk committed
                        waypointsViewOnly[i]->setCurrent(false);
        emit updateStatusString(QString("New current waypoint %1").arg(wpc->seq));
        //emit update to UI widgets
        emit currentWaypointChanged(wpc->seq);
pixhawk's avatar
pixhawk committed
    }
pixhawk's avatar
pixhawk committed
}

void UASWaypointManager::notifyOfChangeEditable(Waypoint* wp)
    // // qDebug() << "WAYPOINT CHANGED: ID:" << wp->getId();
    // If only one waypoint was changed, emit only WP signal
    if (wp != NULL) {
        emit waypointEditableChanged(uasid, wp);
        emit waypointEditableListChanged(uasid);
void UASWaypointManager::notifyOfChangeViewOnly(Waypoint* wp)
{
    if (wp != NULL) {
        emit waypointViewOnlyChanged(uasid, wp);
    } else {
        emit waypointViewOnlyListChanged();
        emit waypointViewOnlyListChanged(uasid);
int UASWaypointManager::setCurrentWaypoint(quint16 seq)
pixhawk's avatar
pixhawk committed
{
pixhawk's avatar
pixhawk committed
    if (seq < waypointsViewOnly.size()) {
        if(current_state == WP_IDLE) {
            //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 = uasid;
lm's avatar
lm committed
            current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;

            sendWaypointSetCurrent(current_wp_id);

            //emit waypointListChanged();

            return 0;
        }
    }
    return -1;
}

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);
                    //currentWaypointEditable = waypoints[i];
                } else {
                    waypointsEditable[i]->setCurrent(false);
                }
            }

            return 0;
        }
    }
    return -1;
}
pixhawk's avatar
pixhawk committed

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

        emit waypointViewOnlyListChanged();
        emit waypointViewOnlyListChanged(uasid);
pixhawk's avatar
pixhawk committed
/**
 * @warning Make sure the waypoint stays valid for the whole application lifecycle!
pixhawk's avatar
pixhawk committed
 * @param enforceFirstActive Enforces that the first waypoint is set as active
 * @see createWaypoint() is more suitable for most use cases
pixhawk's avatar
pixhawk committed
 */
void UASWaypointManager::addWaypointEditable(Waypoint *wp, bool enforceFirstActive)
lm's avatar
lm committed
    if (wp)
    {
pixhawk's avatar
pixhawk committed
        wp->setId(waypointsEditable.size());
        if (enforceFirstActive && waypointsEditable.size() == 0)
pixhawk's avatar
pixhawk committed
            currentWaypointEditable = wp;
pixhawk's avatar
pixhawk committed
        waypointsEditable.insert(waypointsEditable.size(), wp);
        connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChangeEditable(Waypoint*)));
        emit waypointEditableListChanged(uasid);
pixhawk's avatar
pixhawk committed
}

/**
 * @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
    wp->setId(waypointsEditable.size());
    if (enforceFirstActive && waypointsEditable.size() == 0)
pixhawk's avatar
pixhawk committed
        currentWaypointEditable = wp;
pixhawk's avatar
pixhawk committed
    waypointsEditable.insert(waypointsEditable.size(), wp);
    connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChangeEditable(Waypoint*)));
    emit waypointEditableListChanged(uasid);
int UASWaypointManager::removeWaypoint(quint16 seq)
pixhawk's avatar
pixhawk committed
    if (seq < waypointsEditable.size())
pixhawk's avatar
pixhawk committed
        Waypoint *t = waypointsEditable[seq];

        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
        waypointsEditable.remove(seq);
pixhawk's avatar
pixhawk committed
        for(int i = seq; i < waypointsEditable.size(); i++)
pixhawk's avatar
pixhawk committed
            waypointsEditable[i]->setId(i);
Alejandro's avatar
Alejandro committed

        emit waypointEditableListChanged(uasid);
void UASWaypointManager::moveWaypoint(quint16 cur_seq, quint16 new_seq)
pixhawk's avatar
pixhawk committed
    if (cur_seq != new_seq && cur_seq < waypointsEditable.size() && new_seq < waypointsEditable.size())
pixhawk's avatar
pixhawk committed
        Waypoint *t = waypointsEditable[cur_seq];
        if (cur_seq < new_seq) {
            for (int i = cur_seq; i < new_seq; i++)
            {
pixhawk's avatar
pixhawk committed
                waypointsEditable[i] = waypointsEditable[i+1];
        }
        else
        {
            for (int i = cur_seq; i > new_seq; i--)
            {
pixhawk's avatar
pixhawk committed
                waypointsEditable[i] = waypointsEditable[i-1];
pixhawk's avatar
pixhawk committed
        waypointsEditable[new_seq] = t;
        waypointsEditable[new_seq]->setId(new_seq);
        emit waypointEditableListChanged(uasid);
void UASWaypointManager::saveWaypoints(const QString &saveFile)
{
    QFile file(saveFile);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
        return;

    QTextStream out(&file);

    //write the waypoint list version to the first line for compatibility check
pixhawk's avatar
pixhawk committed
    for (int i = 0; i < waypointsEditable.size(); i++)
pixhawk's avatar
pixhawk committed
        waypointsEditable[i]->setId(i);
        waypointsEditable[i]->save(out);
void UASWaypointManager::loadWaypoints(const QString &loadFile)
{
    QFile file(loadFile);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
        return;

pixhawk's avatar
pixhawk committed
    while(waypointsEditable.size()>0) {
        Waypoint *t = waypointsEditable[0];
        waypointsEditable.remove(0);

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

    if (!(version.size() == 3 && version[0] == "QGC" && version[1] == "WPL" && version[2] == "120"))
        emit updateStatusString(tr("The waypoint file is not compatible with the current version of QGroundControl."));
pixhawk's avatar
pixhawk committed
                t->setId(waypointsEditable.size());
                waypointsEditable.insert(waypointsEditable.size(), t);
                emit updateStatusString(tr("The waypoint file is corrupted. Load operation only partly succesful."));
                //MainWindow::instance()->showCriticalMessage(tr("Error loading waypoint file"),);
    emit loadWPFile();
    emit waypointEditableListChanged(uasid);
}

void UASWaypointManager::clearWaypointList()
pixhawk's avatar
pixhawk committed
{
    if(current_state == WP_IDLE)
    {
pixhawk's avatar
pixhawk committed
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed

        current_state = WP_CLEARLIST;
        current_wp_id = 0;
        current_partner_systemid = uasid;
lm's avatar
lm committed
        current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed

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
    foreach (Waypoint* wp, waypointsEditable)
    {
        if (wp->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
lm's avatar
lm committed
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
    foreach (Waypoint* wp, waypointsEditable)
    {
        if ((wp->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && wp->isNavigationType())
        {
lm's avatar
lm committed
            wps.append(wp);
        }
    }
    return wps;
}

LM's avatar
LM committed
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
    foreach (Waypoint* wp, waypointsEditable)
    {
        if (wp->isNavigationType())
        {
LM's avatar
LM committed
            wps.append(wp);
        }
    }
    return wps;
}

int UASWaypointManager::getIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    return waypointsEditable.indexOf(wp);
int UASWaypointManager::getGlobalFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable) {
        if (p->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
            if (p == wp)
            {
lm's avatar
lm committed
int UASWaypointManager::getGlobalFrameAndNavTypeIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
lm's avatar
lm committed
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable) {
        if ((p->getFrame() == MAV_FRAME_GLOBAL || wp->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && p->isNavigationType())
lm's avatar
lm committed
                return i;
            }
            i++;
        }
    }

    return -1;
}

LM's avatar
LM committed
int UASWaypointManager::getNavTypeIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
LM's avatar
LM committed
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable)
    {
        if (p->isNavigationType())
        {
            if (p == wp)
            {
LM's avatar
LM committed
                return i;
            }
            i++;
        }
    }

    return -1;
}

int UASWaypointManager::getGlobalFrameCount()
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable)
        if (p->getFrame() == MAV_FRAME_GLOBAL || p->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT)
        {
lm's avatar
lm committed
int UASWaypointManager::getGlobalFrameAndNavTypeCount()
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
lm's avatar
lm committed
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable) {
        if ((p->getFrame() == MAV_FRAME_GLOBAL || p->getFrame() == MAV_FRAME_GLOBAL_RELATIVE_ALT) && p->isNavigationType())
lm's avatar
lm committed
            i++;
        }
    }

    return i;
}

LM's avatar
LM committed
int UASWaypointManager::getNavTypeCount()
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
LM's avatar
LM committed
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable) {
LM's avatar
LM committed
        if (p->isNavigationType()) {
            i++;
        }
    }

    return i;
}

int UASWaypointManager::getLocalFrameCount()
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
    // counting only those in global frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
    {
        if (p->getFrame() == MAV_FRAME_LOCAL_NED || p->getFrame() == MAV_FRAME_LOCAL_ENU)
lm's avatar
lm committed
        {
            i++;
        }
    }

    return i;
}

int UASWaypointManager::getLocalFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
    // counting only those in local frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
    {
        if (p->getFrame() == MAV_FRAME_LOCAL_NED || p->getFrame() == MAV_FRAME_LOCAL_ENU)
        {
                return i;
            }
            i++;
        }
    }

    return -1;
}

int UASWaypointManager::getMissionFrameIndexOf(Waypoint* wp)
{
pixhawk's avatar
pixhawk committed
    // Search through all waypointsEditable,
    // counting only those in mission frame
    int i = 0;
pixhawk's avatar
pixhawk committed
    foreach (Waypoint* p, waypointsEditable)
lm's avatar
lm committed
    {
        if (p->getFrame() == MAV_FRAME_MISSION)
        {
            if (p == wp)
            {
pixhawk's avatar
pixhawk committed

/**
 * @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)
pixhawk's avatar
pixhawk committed
    read_to_edit = readToEdit;
    emit readGlobalWPFromUAS(true);
    if(current_state == WP_IDLE) {
        //Clear the old view-list before receiving the new one
        while(waypointsViewOnly.size()>0) {
            delete waypointsViewOnly.back();
            waypointsViewOnly.pop_back();
        }

        //Clear the old edit-list before receiving the new one
        if (read_to_edit == true){
            while(waypointsEditable.size()>0) {
                delete waypointsEditable.back();
                waypointsEditable.pop_back();
            }
        protocol_timer.start(PROTOCOL_TIMEOUT_MS);
        current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed

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

        sendWaypointRequestList();
pixhawk's avatar
pixhawk committed
    }
}

void UASWaypointManager::writeWaypoints()
pixhawk's avatar
pixhawk committed
{
    if (current_state == WP_IDLE) {
pixhawk's avatar
pixhawk committed
        // Send clear all if count == 0
pixhawk's avatar
pixhawk committed
        if (waypointsEditable.count() > 0) {
pixhawk's avatar
pixhawk committed
            protocol_timer.start(PROTOCOL_TIMEOUT_MS);
            current_retries = PROTOCOL_MAX_RETRIES;
pixhawk's avatar
pixhawk committed

pixhawk's avatar
pixhawk committed
            current_count = waypointsEditable.count();
pixhawk's avatar
pixhawk committed
            current_state = WP_SENDLIST;
            current_wp_id = 0;
            current_partner_systemid = uasid;
lm's avatar
lm committed
            current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
pixhawk's avatar
pixhawk committed

            //clear local buffer
pixhawk's avatar
pixhawk committed
            // 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.
            while(!waypoint_buffer.empty()) {
pixhawk's avatar
pixhawk committed
                delete waypoint_buffer.back();
                waypoint_buffer.pop_back();
            }
pixhawk's avatar
pixhawk committed

pixhawk's avatar
pixhawk committed
            //copy waypoint data to local buffer
            for (int i=0; i < current_count; i++) {
lm's avatar
lm committed
                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
                const Waypoint *cur_s = waypointsEditable.at(i);
pixhawk's avatar
pixhawk committed

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

                if (cur_s->getCurrent() && noCurrent)
                    noCurrent = false;
                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
            }
pixhawk's avatar
pixhawk committed
            //send the waypoint count to UAS (this starts the send transaction)
pixhawk's avatar
pixhawk committed
        }
pixhawk's avatar
pixhawk committed
    } else if (waypointsEditable.count() == 0) {
        //we're in another transaction, ignore command
        qDebug() << "UASWaypointManager::sendWaypoints() doing something else ignoring command";
pixhawk's avatar
pixhawk committed
}

void UASWaypointManager::sendWaypointClearAll()
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_clear_all_t wpca;
    wpca.target_system = uasid;
lm's avatar
lm committed
    wpca.target_component = MAV_COMP_ID_MISSIONPLANNER;
    emit updateStatusString(QString("Clearing waypoint list..."));
lm's avatar
lm committed
    mavlink_msg_mission_clear_all_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpca);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint clear all to ID " << wpca.target_system;
}

void UASWaypointManager::sendWaypointSetCurrent(quint16 seq)
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_set_current_t wpsc;
    wpsc.target_system = uasid;
lm's avatar
lm committed
    wpsc.target_component = MAV_COMP_ID_MISSIONPLANNER;
    emit updateStatusString(QString("Updating target waypoint..."));
lm's avatar
lm committed
    mavlink_msg_mission_set_current_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpsc);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint set current (" << wpsc.seq << ") to ID " << wpsc.target_system;
}

void UASWaypointManager::sendWaypointCount()
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_count_t wpc;
    wpc.target_system = uasid;
lm's avatar
lm committed
    wpc.target_component = MAV_COMP_ID_MISSIONPLANNER;
    // // qDebug() << "sent waypoint count (" << wpc.count << ") to ID " << wpc.target_system;
    emit updateStatusString(QString("Starting to transmit waypoints..."));
lm's avatar
lm committed
    mavlink_msg_mission_count_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpc);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint count (" << wpc.count << ") to ID " << wpc.target_system;
}

void UASWaypointManager::sendWaypointRequestList()
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_request_list_t wprl;
    wprl.target_system = uasid;
lm's avatar
lm committed
    wprl.target_component = MAV_COMP_ID_MISSIONPLANNER;
    emit updateStatusString(QString("Requesting waypoint list..."));
lm's avatar
lm committed
    mavlink_msg_mission_request_list_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wprl);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint list request to ID " << wprl.target_system;
void UASWaypointManager::sendWaypointRequest(quint16 seq)
pixhawk's avatar
pixhawk committed
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_request_t wpr;
    wpr.target_system = uasid;
lm's avatar
lm committed
    wpr.target_component = MAV_COMP_ID_MISSIONPLANNER;
    emit updateStatusString(QString("Retrieving waypoint ID %1 of %2 total").arg(wpr.seq).arg(current_count));
lm's avatar
lm committed
    mavlink_msg_mission_request_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpr);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint request (" << wpr.seq << ") to ID " << wpr.target_system;
pixhawk's avatar
pixhawk committed
}
pixhawk's avatar
pixhawk committed

pixhawk's avatar
pixhawk committed
void UASWaypointManager::sendWaypoint(quint16 seq)
{
lm's avatar
lm committed
    if (!uas) return;
pixhawk's avatar
pixhawk committed
    mavlink_message_t message;
    // // qDebug() <<" WP Buffer count: "<<waypoint_buffer.count();
    if (seq < waypoint_buffer.count()) {
lm's avatar
lm committed
        mavlink_mission_item_t *wp;
pixhawk's avatar
pixhawk committed

        wp = waypoint_buffer.at(seq);
        wp->target_system = uasid;
lm's avatar
lm committed
        wp->target_component = MAV_COMP_ID_MISSIONPLANNER;
        emit updateStatusString(QString("Sending waypoint ID %1 of %2 total").arg(wp->seq).arg(current_count));
        // // qDebug() << "sent waypoint (" << wp->seq << ") to ID " << wp->target_system<<" WP Buffer count: "<<waypoint_buffer.count();
lm's avatar
lm committed
        mavlink_msg_mission_item_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, wp);
        if (uas) uas->sendMessage(message);
        QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
pixhawk's avatar
pixhawk committed
    }
pixhawk's avatar
pixhawk committed
}

void UASWaypointManager::sendWaypointAck(quint8 type)
{
lm's avatar
lm committed
    if (!uas) return;
    mavlink_message_t message;
lm's avatar
lm committed
    mavlink_mission_ack_t wpa;
    wpa.target_system = uasid;
lm's avatar
lm committed
    wpa.target_component = MAV_COMP_ID_MISSIONPLANNER;
lm's avatar
lm committed
    mavlink_msg_mission_ack_encode(uas->mavlink->getSystemId(), uas->mavlink->getComponentId(), &message, &wpa);
    if (uas) uas->sendMessage(message);
    QGC::SLEEP::msleep(PROTOCOL_DELAY_MS);
    // // qDebug() << "sent waypoint ack (" << wpa.type << ") to ID " << wpa.target_system;