UASListWidget.cc 7.63 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
/*=====================================================================

PIXHAWK Micro Air Vehicle Flying Robotics Toolkit

(c) 2009, 2010 PIXHAWK PROJECT  <http://pixhawk.ethz.ch>

This file is part of the PIXHAWK project

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

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

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

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

/**
 * @file
 *   @brief List of unmanned vehicles
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QString>
#include <QTimer>
#include <QLabel>
#include <QDebug>
35
#include <QApplication>
pixhawk's avatar
pixhawk committed
36 37 38 39 40 41

#include "MG.h"
#include "UASListWidget.h"
#include "UASManager.h"
#include "UAS.h"
#include "UASView.h"
42 43 44 45
#include "QGCUnconnectedInfoWidget.h"
#include "MainWindow.h"
#include "MAVLinkSimulationLink.h"
#include "LinkManager.h"
pixhawk's avatar
pixhawk committed
46

47
UASListWidget::UASListWidget(QWidget *parent) : QWidget(parent),
48
    uWidget(NULL),
49
    m_ui(new Ui::UASList)
pixhawk's avatar
pixhawk committed
50
{
51 52 53 54
    // Use a timer to update the link health display.
    updateTimer = new QTimer(this);
    connect(updateTimer,SIGNAL(timeout()),this,SLOT(updateStatus()));

pixhawk's avatar
pixhawk committed
55
    m_ui->setupUi(this);
56
    m_ui->verticalLayout->setAlignment(Qt::AlignTop);
pixhawk's avatar
pixhawk committed
57

58
    this->setMinimumWidth(262);
pixhawk's avatar
pixhawk committed
59

60
    linkToBoxMapping = QMap<LinkInterface*, QGroupBox*>();
61
    uasToBoxMapping = QMap<UASInterface*, QGroupBox*>();
pixhawk's avatar
pixhawk committed
62
    uasViews = QMap<UASInterface*, UASView*>();
63 64

    this->setVisible(false);
65

66
    connect(LinkManager::instance(), SIGNAL(linkDeleted(LinkInterface*)), this, SLOT(removeLink(LinkInterface*)));
Lorenz Meier's avatar
Lorenz Meier committed
67

68 69
    // Listen for when UASes are added or removed. This does not manage the UASView
    // widgets that are displayed within this widget.
70 71
    connect(UASManager::instance(), SIGNAL(UASCreated(UASInterface*)),
            this, SLOT(addUAS(UASInterface*)));
72 73
    connect(UASManager::instance(), SIGNAL(UASDeleted(UASInterface*)),
            this, SLOT(removeUAS(UASInterface*)));
74 75

    // Get a list of all existing UAS
76 77
    foreach (UASInterface* uas, UASManager::instance()->getUASList())
    {
78 79
        addUAS(uas);
    }
pixhawk's avatar
pixhawk committed
80 81 82 83 84 85 86 87 88 89
}

UASListWidget::~UASListWidget()
{
    delete m_ui;
}

void UASListWidget::changeEvent(QEvent *e)
{
    QWidget::changeEvent(e);
lm's avatar
lm committed
90 91
    switch (e->type())
    {
pixhawk's avatar
pixhawk committed
92 93 94 95 96 97 98 99
    case QEvent::LanguageChange:
        m_ui->retranslateUi(this);
        break;
    default:
        break;
    }
}

Lorenz Meier's avatar
Lorenz Meier committed
100 101 102 103 104 105 106 107 108 109 110 111 112
// XXX This is just to prevent
// upfront crashes, will probably need further inspection
void UASListWidget::removeLink(LinkInterface* link)
{
    QGroupBox* box = linkToBoxMapping.value(link, NULL);

    if (box) {
        // Just stop updating the status for now - we should
        // remove the UAS probably
        linkToBoxMapping.remove(link);
    }
}

113 114 115 116 117 118
void UASListWidget::updateStatus()
{
    QMapIterator<LinkInterface*, QGroupBox*> i(linkToBoxMapping);
    while (i.hasNext()) {
        i.next();
        LinkInterface* link = i.key();
Lorenz Meier's avatar
Lorenz Meier committed
119 120 121 122 123

        // Paranoid sanity check
        if (!LinkManager::instance()->getLinks().contains(link))
            continue;

124 125 126
        if (!link)
            continue;

127
        MAVLinkProtocol* mavlink = MAVLinkProtocol::instance();
128 129 130 131

        // Build the tooltip out of the protocol parsing data: received, dropped, and parsing errors.
        QString displayString("");
        int c;
132
        if ((c = mavlink->getReceivedPacketCount(link)) != -1)
133 134 135
        {
            displayString += QString(tr("<br/>Received: %2")).arg(QString::number(c));
        }
136
        if ((c = mavlink->getDroppedPacketCount(link)) != -1)
137 138 139
        {
            displayString += QString(tr("<br/>Dropped: %2")).arg(QString::number(c));
        }
140
        if ((c = mavlink->getParsingErrorCount(link)) != -1)
141 142 143 144 145 146 147
        {
            displayString += QString(tr("<br/>Errors: %2")).arg(QString::number(c));
        }
        if (!displayString.isEmpty())
        {
            displayString = QString("<b>%1</b>").arg(i.key()->getName()) + displayString;
        }
148
//        qDebug() << p << ": " + displayString;
149 150 151
        i.value()->setToolTip(displayString);
    }
}
152

pixhawk's avatar
pixhawk committed
153 154
void UASListWidget::addUAS(UASInterface* uas)
{
155
    // If the list was empty, remove the unconnected widget and start the update timer.
lm's avatar
lm committed
156 157
    if (uasViews.isEmpty())
    {
158
        updateTimer->start(5000);
Lorenz Meier's avatar
Lorenz Meier committed
159

160 161
        if (uWidget)
        {
162
            m_ui->verticalLayout->removeWidget(uWidget);
163 164 165
            delete uWidget;
            uWidget = NULL;
        }
pixhawk's avatar
pixhawk committed
166
    }
lm's avatar
lm committed
167 168
    if (!uasViews.contains(uas))
    {
169
        // Only display the UAS in a single link.
170 171
        QList<LinkInterface*> x = uas->getLinks();
        if (x.size())
172
        {
173
            LinkInterface* li = x.first();
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188

            // Find an existing QGroupBox for this LinkInterface or create a
            // new one.
            QGroupBox* newBox;
            if (linkToBoxMapping.contains(li))
            {
                newBox = linkToBoxMapping[li];
            }
            else
            {
                newBox = new QGroupBox(li->getName(), this);
                QVBoxLayout* boxLayout = new QVBoxLayout(newBox);
                newBox->setLayout(boxLayout);
                m_ui->verticalLayout->addWidget(newBox);
                linkToBoxMapping[li] = newBox;
189
                updateStatus(); // Update the link status for this GroupBox.
190 191 192 193 194
            }

            // And add the new UAS to the UASList
            UASView* newView = new UASView(uas, newBox);
            uasViews.insert(uas, newView);
195
            uasToBoxMapping[uas] = newBox;
196 197
            newBox->layout()->addWidget(newView);
        }
pixhawk's avatar
pixhawk committed
198 199 200 201 202
    }
}

void UASListWidget::activeUAS(UASInterface* uas)
{
203
    UASView* view = uasViews.value(uas, NULL);
204
    if (view) {
205 206
        view->setUASasActive(true);
    }
pixhawk's avatar
pixhawk committed
207 208
}

209 210 211 212
/**
 * If the UAS was removed, check to see if it was the last one in the QGroupBox and delete
 * the QGroupBox if so.
 */
213
void UASListWidget::removeUAS(UASInterface* uas)
pixhawk's avatar
pixhawk committed
214
{
215 216 217 218
    // Remove the UASView and check if its parent GroupBox has any other children,
    // delete it if it doesn't.
    QGroupBox* box = uasToBoxMapping[uas];
    uasToBoxMapping.remove(uas);
219
    uasViews.remove(uas);
220 221
    int otherViews = 0;
    foreach (UASView* view, box->findChildren<UASView*>())
222
    {
223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239
        if (view->uas == uas)
        {
            view->deleteLater();
        }
        else
        {
            ++otherViews;
        }
    }
    if (otherViews == 0)
    {
        // Delete the groupbox.
        QMap<LinkInterface*, QGroupBox*>::const_iterator i = linkToBoxMapping.constBegin();
        while (i != linkToBoxMapping.constEnd()) {
            if (i.value() == box)
            {
                linkToBoxMapping.remove(i.key());
240
                break;
241 242 243 244
            }
            ++i;
        }
        box->deleteLater();
245

246
        // And if no other QGroupBoxes are left, put the initial widget back.
247
        // We also stop the update timer as there's nothing to update at this point.
248 249
        int otherBoxes = 0;
        foreach (const QGroupBox* otherBox, findChildren<QGroupBox*>())
250
        {
251 252 253
            if (otherBox != box)
            {
                ++otherBoxes;
254 255
            }
        }
256 257 258 259
        if (otherBoxes == 0)
        {
            uWidget = new QGCUnconnectedInfoWidget(this);
            m_ui->verticalLayout->addWidget(uWidget);
260
            updateTimer->stop();
261
        }
262
    }
pixhawk's avatar
pixhawk committed
263
}