UASListWidget.cc 7.65 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 35
/*=====================================================================

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 <QFileDialog>
#include <QDebug>
36
#include <QApplication>
pixhawk's avatar
pixhawk committed
37 38 39 40 41 42

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

48
UASListWidget::UASListWidget(QWidget *parent) : QWidget(parent),
49
    uWidget(NULL),
50
    m_ui(new Ui::UASList)
pixhawk's avatar
pixhawk committed
51
{
52 53 54 55
    // 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
56
    m_ui->setupUi(this);
57
    m_ui->verticalLayout->setAlignment(Qt::AlignTop);
pixhawk's avatar
pixhawk committed
58

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

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

    this->setVisible(false);
66

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

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

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

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

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

Lorenz Meier's avatar
Lorenz Meier committed
101 102 103 104 105 106 107 108 109 110 111 112 113
// 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);
    }
}

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

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

125 126 127
        if (!link)
            continue;

128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
        ProtocolInterface* p = LinkManager::instance()->getProtocolForLink(link);

        // Build the tooltip out of the protocol parsing data: received, dropped, and parsing errors.
        QString displayString("");
        int c;
        if ((c = p->getReceivedPacketCount(link)) != -1)
        {
            displayString += QString(tr("<br/>Received: %2")).arg(QString::number(c));
        }
        if ((c = p->getDroppedPacketCount(link)) != -1)
        {
            displayString += QString(tr("<br/>Dropped: %2")).arg(QString::number(c));
        }
        if ((c = p->getParsingErrorCount(link)) != -1)
        {
            displayString += QString(tr("<br/>Errors: %2")).arg(QString::number(c));
        }
        if (!displayString.isEmpty())
        {
            displayString = QString("<b>%1</b>").arg(i.key()->getName()) + displayString;
        }
149
//        qDebug() << p << ": " + displayString;
150 151 152
        i.value()->setToolTip(displayString);
    }
}
153

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

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

            // 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;
190
                updateStatus(); // Update the link status for this GroupBox.
191 192 193 194 195
            }

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

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

210 211 212 213
/**
 * If the UAS was removed, check to see if it was the last one in the QGroupBox and delete
 * the QGroupBox if so.
 */
214
void UASListWidget::removeUAS(UASInterface* uas)
pixhawk's avatar
pixhawk committed
215
{
216 217 218 219
    // 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);
220
    uasViews.remove(uas);
221 222
    int otherViews = 0;
    foreach (UASView* view, box->findChildren<UASView*>())
223
    {
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
        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());
241
                break;
242 243 244 245
            }
            ++i;
        }
        box->deleteLater();
246

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