Skip to content
Snippets Groups Projects
UASWaypointManager.cc 28.3 KiB
Newer Older
  • Learn to ignore specific revisions
  • 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
    
    #define PROTOCOL_DELAY_MS 40        ///< minimum delay between sent messages
    
    #define PROTOCOL_MAX_RETRIES 3      ///< maximum number of send retries (after timeout)
    
    pixhawk's avatar
    pixhawk 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),
          protocol_timer(this)
    
    pixhawk's avatar
    pixhawk committed
    {
        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)));
    
    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);
        if (waypoints.count() > 0 && (waypoints[current_wp_id]->getFrame() == MAV_FRAME_LOCAL || waypoints[current_wp_id]->getFrame() == MAV_FRAME_LOCAL_ENU))
        {
            double xdiff = x-waypoints[current_wp_id]->getX();
            double ydiff = y-waypoints[current_wp_id]->getY();
            double zdiff = z-waypoints[current_wp_id]->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)
    {
    
    }
    
    
    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;
    
    lm's avatar
    lm committed
                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
                addWaypoint(lwp, false);
    
    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);
    
                    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 (systemId == uas.getUASID()) {
    
    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 (systemId == uas.getUASID()) {
    
            if (current_state == WP_SETCURRENT) {
    
                protocol_timer.stop();
                current_state = WP_IDLE;
    
    
                // 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);
    
                //// qDebug() << "Updated waypoints list";
    
            emit updateStatusString(QString("New current waypoint %1").arg(wpc->seq));
            //emit update to UI widgets
            emit currentWaypointChanged(wpc->seq);
    
            //// qDebug() << "new current waypoint" << wpc->seq;
    
    pixhawk's avatar
    pixhawk committed
        }
    
    pixhawk's avatar
    pixhawk committed
    }
    
    
    void UASWaypointManager::notifyOfChange(Waypoint* wp)
    {
    
        // qDebug() << "WAYPOINT CHANGED: ID:" << wp->getId();
    
        // If only one waypoint was changed, emit only WP signal
    
        if (wp != NULL) {
    
            emit waypointChanged(uas.getUASID(), wp);
    
            emit waypointListChanged();
            emit waypointListChanged(uas.getUASID());
        }
    
    int UASWaypointManager::setCurrentWaypoint(quint16 seq)
    
    pixhawk's avatar
    pixhawk committed
    {
    
        if (seq < waypoints.size()) {
            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);
    
                        waypoints[i]->setCurrent(false);
                    }
                }
    
                //TODO: signal changed waypoint list
    
                //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();
    
    lm's avatar
    lm committed
                current_partner_compid = MAV_COMP_ID_MISSIONPLANNER;
    
    
                sendWaypointSetCurrent(current_wp_id);
    
                //emit waypointListChanged();
    
                return 0;
            }
        }
        return -1;
    }
    
    
    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::addWaypoint(Waypoint *wp, bool enforceFirstActive)
    
            wp->setId(waypoints.size());
    
    pixhawk's avatar
    pixhawk committed
            if (enforceFirstActive && waypoints.size() == 0) wp->setCurrent(true);
    
            waypoints.insert(waypoints.size(), wp);
    
            connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChange(Waypoint*)));
    
            emit waypointListChanged();
    
    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();
        wp->setId(waypoints.size());
        if (enforceFirstActive && waypoints.size() == 0) wp->setCurrent(true);
        waypoints.insert(waypoints.size(), wp);
        connect(wp, SIGNAL(changed(Waypoint*)), this, SLOT(notifyOfChange(Waypoint*)));
    
        emit waypointListChanged();
        emit waypointListChanged(uas.getUASID());
        return wp;
    }
    
    
    int UASWaypointManager::removeWaypoint(quint16 seq)
    
        if (seq < waypoints.size()) {
    
            Waypoint *t = waypoints[seq];
            waypoints.remove(seq);
            delete t;
    
            for(int i = seq; i < waypoints.size(); i++) {
    
    Alejandro's avatar
    Alejandro committed
    
    
            emit waypointListChanged();
    
    void UASWaypointManager::moveWaypoint(quint16 cur_seq, quint16 new_seq)
    
        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();
    
    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
    
    lm's avatar
    lm committed
        out << "QGC WPL 110\r\n";
    
        for (int i = 0; i < waypoints.size(); i++) {
    
            waypoints[i]->setId(i);
    
    void UASWaypointManager::loadWaypoints(const QString &loadFile)
    
    {
        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);
    
    
        const QStringList &version = in.readLine().split(" ");
    
    
        if (!(version.size() == 3 && version[0] == "QGC" && version[1] == "WPL" && version[2] == "110")) {
    
            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."));
    
        } else {
            while (!in.atEnd()) {
    
                if(t->load(in)) {
    
                    t->setId(waypoints.size());
                    waypoints.insert(waypoints.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 waypointListChanged();
    
    //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());
    //}
    
    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 = uas.getUASID();
    
    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;
    
        foreach (Waypoint* wp, waypoints) {
            if (wp->getFrame() == MAV_FRAME_GLOBAL) {
    
    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;
    
        foreach (Waypoint* wp, waypoints) {
            if (wp->getFrame() == MAV_FRAME_GLOBAL && 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;
        foreach (Waypoint* wp, waypoints) {
            if (wp->isNavigationType()) {
                wps.append(wp);
            }
        }
        return wps;
    }
    
    
    int UASWaypointManager::getIndexOf(Waypoint* wp)
    {
        return waypoints.indexOf(wp);
    }
    
    
    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) {
    
    lm's avatar
    lm committed
    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) {
    
    lm's avatar
    lm committed
                    return i;
                }
                i++;
            }
        }
    
        return -1;
    }
    
    
    LM's avatar
    LM committed
    int UASWaypointManager::getNavTypeIndexOf(Waypoint* wp)
    {
        // Search through all waypoints,
        // counting only those in global frame
        int i = 0;
        foreach (Waypoint* p, waypoints) {
            if (p->isNavigationType()) {
                if (p == wp) {
                    return i;
                }
                i++;
            }
        }
    
        return -1;
    }
    
    
    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) {
    
    lm's avatar
    lm committed
    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()) {
    
    lm's avatar
    lm committed
                i++;
            }
        }
    
        return i;
    }
    
    
    LM's avatar
    LM committed
    int UASWaypointManager::getNavTypeCount()
    {
        // Search through all waypoints,
        // counting only those in global frame
        int i = 0;
        foreach (Waypoint* p, waypoints) {
            if (p->isNavigationType()) {
                i++;
            }
        }
    
        return i;
    }
    
    
    int UASWaypointManager::getLocalFrameCount()
    {
        // Search through all waypoints,
        // counting only those in global frame
        int i = 0;
    
    lm's avatar
    lm committed
        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;
    
    lm's avatar
    lm committed
        foreach (Waypoint* p, waypoints)
        {
            if (p->getFrame() == MAV_FRAME_LOCAL_NED || p->getFrame() == MAV_FRAME_LOCAL_ENU)
            {
    
                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;
    
    lm's avatar
    lm committed
        foreach (Waypoint* p, waypoints)
        {
    
            if (p->getFrame() == MAV_FRAME_MISSION) {
                if (p == wp) {
    
    void UASWaypointManager::readWaypoints()
    {
    
        emit readGlobalWPFromUAS(true);
    
        if(current_state == WP_IDLE) {
            while(waypoints.size()>0) {
    
            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 = uas.getUASID();
    
    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
    
            if (waypoints.count() > 0) {
    
    pixhawk's avatar
    pixhawk committed
                protocol_timer.start(PROTOCOL_TIMEOUT_MS);
    
                current_retries = PROTOCOL_MAX_RETRIES;
    
    pixhawk's avatar
    pixhawk committed
    
    
                current_count = waypoints.count();
    
    pixhawk's avatar
    pixhawk committed
                current_state = WP_SENDLIST;
                current_wp_id = 0;
                current_partner_systemid = uas.getUASID();
    
    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
    
                    const Waypoint *cur_s = waypoints.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
            }
    
        } else if (waypoints.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()
    {
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_clear_all_t wpca;
    
    
        wpca.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // qDebug() << "sent waypoint clear all to ID " << wpca.target_system;
    
    }
    
    void UASWaypointManager::sendWaypointSetCurrent(quint16 seq)
    {
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_set_current_t wpsc;
    
    
        wpsc.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // qDebug() << "sent waypoint set current (" << wpsc.seq << ") to ID " << wpsc.target_system;
    
    }
    
    void UASWaypointManager::sendWaypointCount()
    {
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_count_t wpc;
    
    
        wpc.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // qDebug() << "sent waypoint count (" << wpc.count << ") to ID " << wpc.target_system;
    
    }
    
    void UASWaypointManager::sendWaypointRequestList()
    {
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_request_list_t wprl;
    
    
        wprl.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // qDebug() << "sent waypoint list request to ID " << wprl.target_system;
    
    void UASWaypointManager::sendWaypointRequest(quint16 seq)
    
    pixhawk's avatar
    pixhawk committed
    {
    
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_request_t wpr;
    
    
        wpr.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // 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)
    {
        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 = uas.getUASID();
    
    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);
    
    pixhawk's avatar
    pixhawk committed
            uas.sendMessage(message);
    
            MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
    pixhawk's avatar
    pixhawk committed
    }
    
    
    void UASWaypointManager::sendWaypointAck(quint8 type)
    {
        mavlink_message_t message;
    
    lm's avatar
    lm committed
        mavlink_mission_ack_t wpa;
    
    
        wpa.target_system = uas.getUASID();
    
    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);
    
        uas.sendMessage(message);
    
        MG::SLEEP::usleep(PROTOCOL_DELAY_MS * 1000);
    
        // qDebug() << "sent waypoint ack (" << wpa.type << ") to ID " << wpa.target_system;