MainWindow.cc 48.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*=====================================================================

QGroundControl Open Source Ground Control Station

(c) 2009 - 2011 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>

This file is part of the QGROUNDCONTROL project

    QGROUNDCONTROL 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.

    QGROUNDCONTROL 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 QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.

pixhawk's avatar
pixhawk committed
22 23 24 25
======================================================================*/

/**
 * @file
26
 *   @brief Implementation of class MainWindow
27
 *   @author Lorenz Meier <mail@qgroundcontrol.org>
pixhawk's avatar
pixhawk committed
28 29 30 31 32 33 34 35 36 37
 */

#include <QSettings>
#include <QDockWidget>
#include <QNetworkInterface>
#include <QMessageBox>
#include <QDebug>
#include <QTimer>
#include <QHostInfo>

38
#include "QGC.h"
pixhawk's avatar
pixhawk committed
39 40 41 42 43
#include "MAVLinkSimulationLink.h"
#include "SerialLink.h"
#include "UDPLink.h"
#include "MAVLinkProtocol.h"
#include "CommConfigurationWindow.h"
44
#include "QGCWaypointListMulti.h"
pixhawk's avatar
pixhawk committed
45 46
#include "MainWindow.h"
#include "JoystickWidget.h"
pixhawk's avatar
pixhawk committed
47
#include "GAudioOutput.h"
48
#include "QGCToolWidget.h"
49
#include "QGCMAVLinkLogPlayer.h"
50
#include "QGCSettingsWidget.h"
51
#include "QGCMapTool.h"
LM's avatar
LM committed
52
#include "MAVLinkDecoder.h"
53

54
#ifdef QGC_OSG_ENABLED
55
#include "Q3DWidgetFactory.h"
56
#endif
pixhawk's avatar
pixhawk committed
57

lm's avatar
lm committed
58 59 60 61
// FIXME Move
#include "PxQuadMAV.h"
#include "SlugsMAV.h"

pixhawk's avatar
pixhawk committed
62

63
#include "LogCompressor.h"
pixhawk's avatar
pixhawk committed
64

65 66
MainWindow* MainWindow::instance()
{
67
    static MainWindow* _instance = 0;
68 69
    if(_instance == 0)
    {
70
        _instance = new MainWindow();
71

72
        /* Set the application as parent to ensure that this object
73
                 * will be destroyed when the main application exits */
74 75 76
        //_instance->setParent(qApp);
    }
    return _instance;
77 78
}

pixhawk's avatar
pixhawk committed
79 80 81 82 83 84 85
/**
* Create new mainwindow. The constructor instantiates all parts of the user
* interface. It does NOT show the mainwindow. To display it, call the show()
* method.
*
* @see QMainWindow::show()
**/
86
MainWindow::MainWindow(QWidget *parent):
87 88 89 90 91 92
    QMainWindow(parent),
    currentView(VIEW_UNCONNECTED),
    aboutToCloseFlag(false),
    changingViewsFlag(false),
    styleFileName(QCoreApplication::applicationDirPath() + "/style-indoor.css"),
    autoReconnect(false),
93
    currentStyle(QGC_MAINWINDOW_STYLE_INDOOR),
94 95
    centerStackActionGroup(this),
    lowPowerMode(false)
pixhawk's avatar
pixhawk committed
96
{
97
    loadSettings();
98 99
    if (!settings.contains("CURRENT_VIEW"))
    {
100 101
        // Set this view as default view
        settings.setValue("CURRENT_VIEW", currentView);
102 103 104
    }
    else
    {
105 106 107
        // LOAD THE LAST VIEW
        VIEW_SECTIONS currentViewCandidate = (VIEW_SECTIONS) settings.value("CURRENT_VIEW", currentView).toInt();
        if (currentViewCandidate != VIEW_ENGINEER &&
108
                currentViewCandidate != VIEW_OPERATOR &&
109 110 111
                currentViewCandidate != VIEW_PILOT &&
                currentViewCandidate != VIEW_FULL)
        {
112
            currentView = currentViewCandidate;
113
        }
114 115
    }

116 117
    settings.sync();

118
    loadStyle(currentStyle);
119

pixhawk's avatar
pixhawk committed
120 121 122
    // Setup user interface
    ui.setupUi(this);

123 124 125 126 127 128 129 130 131
    // Set dock options
    setDockOptions(AnimatedDocks | AllowTabbedDocks | AllowNestedDocks);
    statusBar()->setSizeGripEnabled(true);

    configureWindowName();

    // Setup UI state machines
    centerStackActionGroup.setExclusive(true);

132 133
    centerStack = new QStackedWidget(this);
    setCentralWidget(centerStack);
134

lm's avatar
lm committed
135 136 137 138 139 140 141 142
    // Load Toolbar
    toolBar = new QGCToolBar(this);
    this->addToolBar(toolBar);
    // Add actions
    toolBar->addPerspectiveChangeAction(ui.actionOperatorsView);
    toolBar->addPerspectiveChangeAction(ui.actionEngineersView);
    toolBar->addPerspectiveChangeAction(ui.actionPilotsView);

143 144
    buildCommonWidgets();
    connectCommonWidgets();
145

146
    // Create actions
147
    connectCommonActions();
148

149

lm's avatar
lm committed
150

151 152 153 154 155 156
    // Restore the window setup
    if (settings.contains(getWindowStateKey()))
    {
        loadViewState();
    }

157
    // Restore the window position and size
158 159
    if (settings.contains(getWindowGeometryKey()))
    {
160
        // Restore the window geometry
161
        restoreGeometry(settings.value(getWindowGeometryKey()).toByteArray());
162 163 164
    }
    else
    {
165 166 167
        // Adjust the size
        adjustSize();
    }
pixhawk's avatar
pixhawk committed
168

169 170
    // Populate link menu
    QList<LinkInterface*> links = LinkManager::instance()->getLinks();
171 172
    foreach(LinkInterface* link, links)
    {
173 174
        this->addLink(link);
    }
175

176 177
    connect(LinkManager::instance(), SIGNAL(newLink(LinkInterface*)), this, SLOT(addLink(LinkInterface*)));

178
    // Connect user interface devices
179 180
    joystickWidget = 0;
    joystick = new JoystickInput();
181

182
    // Connect link
183 184
    if (autoReconnect)
    {
185 186 187 188 189 190
        SerialLink* link = new SerialLink();
        // Add to registry
        LinkManager::instance()->add(link);
        LinkManager::instance()->addProtocol(link, mavlink);
        link->connect();
    }
191

192 193 194
    // Set low power mode
    enableLowPowerMode(lowPowerMode);

195 196
    // Initialize window state
    windowStateVal = windowState();
197 198

    show();
199 200 201

    connect(&windowNameUpdateTimer, SIGNAL(timeout()), this, SLOT(configureWindowName()));
    windowNameUpdateTimer.start(15000);
pixhawk's avatar
pixhawk committed
202 203
}

pixhawk's avatar
pixhawk committed
204
MainWindow::~MainWindow()
pixhawk's avatar
pixhawk committed
205
{
206
    delete mavlink;
207
    delete joystick;
lm's avatar
lm committed
208

209 210 211 212 213 214
    // Get and delete all dockwidgets and contained
    // widgets
    QObjectList childList( this->children() );

    QObjectList::iterator i;
    QDockWidget* dockWidget;
215 216
    for (i = childList.begin(); i != childList.end(); ++i)
    {
217
        dockWidget = dynamic_cast<QDockWidget*>(*i);
218 219
        if (dockWidget)
        {
220 221 222 223 224
            // Remove dock widget from main window
            removeDockWidget(dockWidget);
            delete dockWidget->widget();
            delete dockWidget;
        }
LM's avatar
LM committed
225 226 227 228
        else
        {
            delete dynamic_cast<QObject*>(*i);
        }
229 230
    }

LM's avatar
LM committed
231
    // Delete all UAS objects
232 233
}

lm's avatar
lm committed
234 235
void MainWindow::resizeEvent(QResizeEvent * event)
{
236 237
    if (height() < 800)
    {
lm's avatar
lm committed
238
        ui.statusBar->setVisible(false);
239 240 241
    }
    else
    {
lm's avatar
lm committed
242
        ui.statusBar->setVisible(true);
243
        ui.statusBar->setSizeGripEnabled(true);
lm's avatar
lm committed
244
    }
245 246 247 248 249 250 251 252 253 254 255

    if (width() > 1200)
    {
        toolBar->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
    }
    else
    {
        toolBar->setToolButtonStyle(Qt::ToolButtonIconOnly);
    }

    QMainWindow::resizeEvent(event);
lm's avatar
lm committed
256 257
}

258 259
QString MainWindow::getWindowStateKey()
{
260
    return QString::number(currentView)+"_windowstate";
261 262 263 264
}

QString MainWindow::getWindowGeometryKey()
{
265 266
    //return QString::number(currentView)+"_geometry";
    return "_geometry";
267 268
}

lm's avatar
lm committed
269 270 271
void MainWindow::buildCustomWidget()
{
    // Show custom widgets only if UAS is connected
272 273
    if (UASManager::instance()->getActiveUAS() != NULL)
    {
lm's avatar
lm committed
274 275 276
        // Create custom widgets
        QList<QGCToolWidget*> widgets = QGCToolWidget::createWidgetsFromSettings(this);

277 278
        if (widgets.size() > 0)
        {
lm's avatar
lm committed
279 280 281
            ui.menuTools->addSeparator();
        }

282 283
        for(int i = 0; i < widgets.size(); ++i)
        {
lm's avatar
lm committed
284
            // Check if this widget already has a parent, do not create it in this case
285 286
            QGCToolWidget* tool = widgets.at(i);
            QDockWidget* dock = dynamic_cast<QDockWidget*>(tool->parentWidget());
287 288
            if (!dock)
            {
289 290 291 292
                QDockWidget* dock = new QDockWidget(tool->windowTitle(), this);
                dock->setObjectName(tool->objectName()+"_DOCK");
                dock->setWidget(tool);
                connect(tool, SIGNAL(destroyed()), dock, SLOT(deleteLater()));
lm's avatar
lm committed
293
                QAction* showAction = new QAction(widgets.at(i)->windowTitle(), this);
294
                showAction->setCheckable(true);
lm's avatar
lm committed
295 296 297 298
                connect(showAction, SIGNAL(triggered(bool)), dock, SLOT(setVisible(bool)));
                connect(dock, SIGNAL(visibilityChanged(bool)), showAction, SLOT(setChecked(bool)));
                widgets.at(i)->setMainMenuAction(showAction);
                ui.menuTools->addAction(showAction);
299 300 301 302 303

                // Load dock widget location (default is bottom)
                Qt::DockWidgetArea location = static_cast <Qt::DockWidgetArea>(tool->getDockWidgetArea(currentView));

                addDockWidget(location, dock);
lm's avatar
lm committed
304 305 306 307 308
            }
        }
    }
}

309 310 311 312
void MainWindow::buildCommonWidgets()
{
    //TODO:  move protocol outside UI
    mavlink     = new MAVLinkProtocol();
313
    connect(mavlink, SIGNAL(protocolStatusMessage(QString,QString)), this, SLOT(showCriticalMessage(QString,QString)), Qt::QueuedConnection);
LM's avatar
LM committed
314 315
    // Add generic MAVLink decoder
    mavlinkDecoder = new MAVLinkDecoder(mavlink, this);
316 317

    // Dock widgets
318 319
    if (!controlDockWidget)
    {
320
        controlDockWidget = new QDockWidget(tr("Control"), this);
321
        controlDockWidget->setObjectName("UNMANNED_SYSTEM_CONTROL_DOCKWIDGET");
322
        controlDockWidget->setWidget( new UASControlWidget(this) );
323
        addTool(controlDockWidget, tr("Control"), Qt::LeftDockWidgetArea);
324
    }
325

326 327
    if (!listDockWidget)
    {
328 329
        listDockWidget = new QDockWidget(tr("Unmanned Systems"), this);
        listDockWidget->setWidget( new UASListWidget(this) );
330
        listDockWidget->setObjectName("UNMANNED_SYSTEMS_LIST_DOCKWIDGET");
331
        addTool(listDockWidget, tr("Unmanned Systems"), Qt::RightDockWidgetArea);
332
    }
333

334 335
    if (!waypointsDockWidget)
    {
336
        waypointsDockWidget = new QDockWidget(tr("Mission Plan"), this);
337
        waypointsDockWidget->setWidget( new QGCWaypointListMulti(this) );
338
        waypointsDockWidget->setObjectName("WAYPOINT_LIST_DOCKWIDGET");
339
        addTool(waypointsDockWidget, tr("Mission Plan"), Qt::BottomDockWidgetArea);
340
    }
341

342 343
    if (!infoDockWidget)
    {
344 345
        infoDockWidget = new QDockWidget(tr("Status Details"), this);
        infoDockWidget->setWidget( new UASInfoWidget(this) );
pixhawk's avatar
pixhawk committed
346
        infoDockWidget->setObjectName("UAS_STATUS_DETAILS_DOCKWIDGET");
347
        addTool(infoDockWidget, tr("Status Details"), Qt::RightDockWidgetArea);
348
    }
349

350 351
    if (!debugConsoleDockWidget)
    {
352 353
        debugConsoleDockWidget = new QDockWidget(tr("Communication Console"), this);
        debugConsoleDockWidget->setWidget( new DebugConsole(this) );
354
        debugConsoleDockWidget->setObjectName("COMMUNICATION_DEBUG_CONSOLE_DOCKWIDGET");
LM's avatar
LM committed
355 356 357 358

        DebugConsole *debugConsole = dynamic_cast<DebugConsole*>(debugConsoleDockWidget->widget());
        connect(mavlinkDecoder, SIGNAL(textMessageReceived(int, int, int, const QString)), debugConsole, SLOT(receiveTextMessage(int, int, int, const QString)));

359
        addTool(debugConsoleDockWidget, tr("Communication Console"), Qt::BottomDockWidgetArea);
360
    }
361

362 363
    if (!logPlayerDockWidget)
    {
364
        logPlayerDockWidget = new QDockWidget(tr("MAVLink Log Player"), this);
lm's avatar
lm committed
365 366 367
        logPlayer = new QGCMAVLinkLogPlayer(mavlink, this);
        toolBar->setLogPlayer(logPlayer);
        logPlayerDockWidget->setWidget(logPlayer);
368
        logPlayerDockWidget->setObjectName("MAVLINK_LOG_PLAYER_DOCKWIDGET");
369
        addTool(logPlayerDockWidget, tr("MAVLink Log Replay"), Qt::RightDockWidgetArea);
370 371
    }

lm's avatar
lm committed
372 373 374 375 376
    if (!mavlinkInspectorWidget)
    {
        mavlinkInspectorWidget = new QDockWidget(tr("MAVLink Message Inspector"), this);
        mavlinkInspectorWidget->setWidget( new QGCMAVLinkInspector(mavlink, this) );
        mavlinkInspectorWidget->setObjectName("MAVLINK_INSPECTOR_DOCKWIDGET");
lm's avatar
lm committed
377
        addTool(mavlinkInspectorWidget, tr("MAVLink Inspector"), Qt::RightDockWidgetArea);
lm's avatar
lm committed
378 379
    }

pixhawk's avatar
pixhawk committed
380 381
    //FIXME: memory of acceptList will never be freed again
    QStringList* acceptList = new QStringList();
pixhawk's avatar
pixhawk committed
382 383 384
    acceptList->append("-105,roll deg,deg,+105,s");
    acceptList->append("-105,pitch deg,deg,+105,s");
    acceptList->append("-105,heading deg,deg,+105,s");
385

pixhawk's avatar
pixhawk committed
386 387 388
    acceptList->append("-60,rollspeed d/s,deg/s,+60,s");
    acceptList->append("-60,pitchspeed d/s,deg/s,+60,s");
    acceptList->append("-60,yawspeed d/s,deg/s,+60,s");
389
    acceptList->append("0,airspeed,m/s,30");
390 391
    acceptList->append("0,groundspeed,m/s,30");
    acceptList->append("0,climbrate,m/s,30");
392
    acceptList->append("0,throttle,%,100");
393

pixhawk's avatar
pixhawk committed
394 395
    //FIXME: memory of acceptList2 will never be freed again
    QStringList* acceptList2 = new QStringList();
396 397 398
    acceptList2->append("900,servo #6,us,2100,s");
    acceptList2->append("900,servo #7,us,2100,s");
    acceptList2->append("900,servo #8,us,2100,s");
lm's avatar
lm committed
399
    acceptList2->append("0,abs pressure,hPa,65500");
400

lm's avatar
lm committed
401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480
    if (!parametersDockWidget) {
        parametersDockWidget = new QDockWidget(tr("Calibration and Onboard Parameters"), this);
        parametersDockWidget->setWidget( new ParameterInterface(this) );
        parametersDockWidget->setObjectName("PARAMETER_INTERFACE_DOCKWIDGET");
        addTool(parametersDockWidget, tr("Calibration and Parameters"), Qt::RightDockWidgetArea);
    }

    if (!hsiDockWidget) {
        hsiDockWidget = new QDockWidget(tr("Horizontal Situation Indicator"), this);
        hsiDockWidget->setWidget( new HSIDisplay(this) );
        hsiDockWidget->setObjectName("HORIZONTAL_SITUATION_INDICATOR_DOCK_WIDGET");
        addTool(hsiDockWidget, tr("Horizontal Situation"), Qt::BottomDockWidgetArea);
    }

    if (!headDown1DockWidget) {
        headDown1DockWidget = new QDockWidget(tr("Flight Display"), this);
        headDown1DockWidget->setWidget( new HDDisplay(acceptList, "Flight Display", this) );
        headDown1DockWidget->setObjectName("HEAD_DOWN_DISPLAY_1_DOCK_WIDGET");
        addTool(headDown1DockWidget, tr("Flight Display"), Qt::RightDockWidgetArea);
    }

    if (!headDown2DockWidget) {
        headDown2DockWidget = new QDockWidget(tr("Actuator Status"), this);
        headDown2DockWidget->setWidget( new HDDisplay(acceptList2, "Actuator Status", this) );
        headDown2DockWidget->setObjectName("HEAD_DOWN_DISPLAY_2_DOCK_WIDGET");
        addTool(headDown2DockWidget, tr("Actuator Status"), Qt::RightDockWidgetArea);
    }

    if (!rcViewDockWidget) {
        rcViewDockWidget = new QDockWidget(tr("Radio Control"), this);
        rcViewDockWidget->setWidget( new QGCRemoteControlView(this) );
        rcViewDockWidget->setObjectName("RADIO_CONTROL_CHANNELS_DOCK_WIDGET");
        addTool(rcViewDockWidget, tr("Radio Control"), Qt::BottomDockWidgetArea);
    }

    if (!headUpDockWidget) {
        headUpDockWidget = new QDockWidget(tr("HUD"), this);
        headUpDockWidget->setWidget( new HUD(320, 240, this));
        headUpDockWidget->setObjectName("HEAD_UP_DISPLAY_DOCK_WIDGET");
        addTool(headUpDockWidget, tr("Head Up Display"), Qt::RightDockWidgetArea);
    }

    if (!video1DockWidget) {
        video1DockWidget = new QDockWidget(tr("Video Stream 1"), this);
        HUD* video1 =  new HUD(160, 120, this);
        video1->enableHUDInstruments(false);
        video1->enableVideo(true);
        // FIXME select video stream as well
        video1DockWidget->setWidget(video1);
        video1DockWidget->setObjectName("VIDEO_STREAM_1_DOCK_WIDGET");
        addTool(video1DockWidget, tr("Video Stream 1"), Qt::LeftDockWidgetArea);
    }

    if (!video2DockWidget) {
        video2DockWidget = new QDockWidget(tr("Video Stream 2"), this);
        HUD* video2 =  new HUD(160, 120, this);
        video2->enableHUDInstruments(false);
        video2->enableVideo(true);
        // FIXME select video stream as well
        video2DockWidget->setWidget(video2);
        video2DockWidget->setObjectName("VIDEO_STREAM_2_DOCK_WIDGET");
        addTool(video2DockWidget, tr("Video Stream 2"), Qt::LeftDockWidgetArea);
    }

    // Custom widgets, added last to all menus and layouts
    buildCustomWidget();

    // Center widgets
    if (!mapWidget)
    {
        mapWidget = new QGCMapTool(this);
        addCentralWidget(mapWidget, "Maps");
    }

    if (!protocolWidget)
    {
        protocolWidget    = new XMLCommProtocolWidget(this);
        addCentralWidget(protocolWidget, "Mavlink Generator");
    }

481
    if (!hudWidget) {
482
        hudWidget         = new HUD(320, 240, this);
483
        addCentralWidget(hudWidget, tr("Head Up Display"));
lm's avatar
lm committed
484 485
    }

486
    if (!dataplotWidget) {
487
        dataplotWidget    = new QGCDataPlot2D(this);
488
        addCentralWidget(dataplotWidget, tr("Logfile Plot"));
489
    }
490

491
#ifdef QGC_OSG_ENABLED
492
    if (!_3DWidget) {
493
        _3DWidget         = Q3DWidgetFactory::get("PIXHAWK");
494
        addCentralWidget(_3DWidget, tr("Local 3D"));
495
    }
496
#endif
497

498
#if (defined _MSC_VER) | (defined Q_OS_MAC)
499
    if (!gEarthWidget) {
500
        gEarthWidget = new QGCGoogleEarthView(this);
501
        addCentralWidget(gEarthWidget, tr("Google Earth"));
502
    }
503
#endif
lm's avatar
lm committed
504
}
505

506
void MainWindow::addTool(QDockWidget* widget, const QString& title, Qt::DockWidgetArea area)
lm's avatar
lm committed
507
{
508
    QAction* tempAction = ui.menuTools->addAction(title);
509

510
    tempAction->setCheckable(true);
511 512 513 514
    QVariant var;
    var.setValue((QWidget*)widget);
    tempAction->setData(var);
    connect(tempAction,SIGNAL(triggered(bool)),this, SLOT(showTool(bool)));
LM's avatar
LM committed
515 516
    connect(widget, SIGNAL(visibilityChanged(bool)), tempAction, SLOT(setChecked(bool)));
    tempAction->setChecked(widget->isVisible());
517
    addDockWidget(area, widget);
518 519 520
}


521
void MainWindow::showTool(bool show)
lm's avatar
lm committed
522
{
523 524 525
    QAction* act = qobject_cast<QAction *>(sender());
    QWidget* widget = qVariantValue<QWidget *>(act->data());
    widget->setVisible(show);
526
}
527

528
void MainWindow::addCentralWidget(QWidget* widget, const QString& title)
lm's avatar
lm committed
529
{
530 531 532 533
    // Check if this widget already has been added
    if (centerStack->indexOf(widget) == -1)
    {
        centerStack->addWidget(widget);
534

535
        QAction* tempAction = ui.menuMain->addAction(title);
536

537 538 539 540 541 542
        tempAction->setCheckable(true);
        QVariant var;
        var.setValue((QWidget*)widget);
        tempAction->setData(var);
        centerStackActionGroup.addAction(tempAction);
        connect(tempAction,SIGNAL(triggered()),this, SLOT(showCentralWidget()));
LM's avatar
LM committed
543 544
        connect(widget, SIGNAL(visibilityChanged(bool)), tempAction, SLOT(setChecked(bool)));
        tempAction->setChecked(widget->isVisible());
545 546 547 548
    }
}


549
void MainWindow::showCentralWidget()
550
{
551 552 553
    QAction* act = qobject_cast<QAction *>(sender());
    QWidget* widget = qVariantValue<QWidget *>(act->data());
    centerStack->setCurrentWidget(widget);
554 555
}

556 557
void MainWindow::closeEvent(QCloseEvent *event)
{
pixhawk's avatar
pixhawk committed
558
    if (isVisible()) storeViewState();
559
    storeSettings();
560
    aboutToCloseFlag = true;
561
    mavlink->storeSettings();
562
    UASManager::instance()->storeSettings();
563 564
    QMainWindow::closeEvent(event);
}
565 566 567 568

/**
 * Connect the signals and slots of the common window widgets
 */
569
void MainWindow::connectCommonWidgets()
570
{
571
    if (infoDockWidget && infoDockWidget->widget()) {
pixhawk's avatar
pixhawk committed
572 573 574
        connect(mavlink, SIGNAL(receiveLossChanged(int, float)),
                infoDockWidget->widget(), SLOT(updateSendLoss(int, float)));
    }
575
}
576

577 578
void MainWindow::createCustomWidget()
{
579 580
    QDockWidget* dock = new QDockWidget("Unnamed Tool", this);
    QGCToolWidget* tool = new QGCToolWidget("Unnamed Tool", dock);
lm's avatar
lm committed
581

LM's avatar
LM committed
582 583
    if (QGCToolWidget::instances()->size() < 2)
    {
lm's avatar
lm committed
584 585 586 587 588
        // This is the first widget
        ui.menuTools->addSeparator();
    }

    connect(tool, SIGNAL(destroyed()), dock, SLOT(deleteLater()));
589
    dock->setWidget(tool);
590

591 592 593 594 595 596 597 598 599 600 601 602 603 604
    QAction* showAction = new QAction(tool->getTitle(), this);
    showAction->setCheckable(true);
    connect(dock, SIGNAL(visibilityChanged(bool)), showAction, SLOT(setChecked(bool)));
    connect(showAction, SIGNAL(triggered(bool)), dock, SLOT(setVisible(bool)));
    tool->setMainMenuAction(showAction);
    ui.menuTools->addAction(showAction);
    this->addDockWidget(Qt::BottomDockWidgetArea, dock);
    dock->setVisible(true);
}

void MainWindow::loadCustomWidget()
{
    QString widgetFileExtension(".qgw");
    QString fileName = QFileDialog::getOpenFileName(this, tr("Specify Widget File Name"), QDesktopServices::storageLocation(QDesktopServices::DesktopLocation), tr("QGroundControl Widget (*%1);;").arg(widgetFileExtension));
605
    if (fileName != "") loadCustomWidget(fileName);
LM's avatar
LM committed
606
}
607

LM's avatar
LM committed
608 609 610 611
void MainWindow::loadCustomWidget(const QString& fileName, bool singleinstance)
{
    QGCToolWidget* tool = new QGCToolWidget("", this);
    if (tool->loadSettings(fileName, true) || !singleinstance)
LM's avatar
LM committed
612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
    {
        // Add widget to UI
        QDockWidget* dock = new QDockWidget(tool->getTitle(), this);
        connect(tool, SIGNAL(destroyed()), dock, SLOT(deleteLater()));
        dock->setWidget(tool);
        tool->setParent(dock);

        QAction* showAction = new QAction(tool->getTitle(), this);
        showAction->setCheckable(true);
        connect(dock, SIGNAL(visibilityChanged(bool)), showAction, SLOT(setChecked(bool)));
        connect(showAction, SIGNAL(triggered(bool)), dock, SLOT(setVisible(bool)));
        tool->setMainMenuAction(showAction);
        ui.menuTools->addAction(showAction);
        this->addDockWidget(Qt::BottomDockWidgetArea, dock);
        dock->setVisible(true);
    }
LM's avatar
LM committed
628 629 630
    else
    {
        return;
631
    }
632 633
}

LM's avatar
LM committed
634
void MainWindow::loadCustomWidgetsFromDefaults(const QString& systemType, const QString& autopilotType)
635
{
636
    QString defaultsDir = qApp->applicationDirPath() + "/files/" + autopilotType.toLower() + "/" + systemType.toLower() + "/widgets/";
637

LM's avatar
LM committed
638 639 640 641 642 643
    QDir widgets(defaultsDir);
    QStringList files = widgets.entryList();
    if (files.count() == 0)
    {
        qDebug() << "No default custom widgets for system " << systemType << "autopilot" << autopilotType << " found";
        qDebug() << "Tried with path: " << defaultsDir;
lm's avatar
lm committed
644
        showStatusMessage(tr("Did not find any custom widgets in %1").arg(defaultsDir));
645
    }
646

LM's avatar
LM committed
647 648
    // Load all custom widgets found in the AP folder
    for(int i = 0; i < files.count(); ++i)
lm's avatar
lm committed
649
    {
LM's avatar
LM committed
650 651 652 653 654
        QString file = files[i];
        if (file.endsWith(".qgw"))
        {
            // Will only be loaded if not already a custom widget with
            // the same name is present
655
            loadCustomWidget(defaultsDir+"/"+file, true);
lm's avatar
lm committed
656
            showStatusMessage(tr("Loaded custom widget %1").arg(defaultsDir+"/"+file));
LM's avatar
LM committed
657
        }
lm's avatar
lm committed
658
    }
659 660
}

661 662 663 664 665 666
void MainWindow::loadSettings()
{
    QSettings settings;
    settings.beginGroup("QGC_MAINWINDOW");
    autoReconnect = settings.value("AUTO_RECONNECT", autoReconnect).toBool();
    currentStyle = (QGC_MAINWINDOW_STYLE)settings.value("CURRENT_STYLE", currentStyle).toInt();
667
    lowPowerMode = settings.value("LOW_POWER_MODE", lowPowerMode).toBool();
668 669 670 671 672 673 674 675 676 677
    settings.endGroup();
}

void MainWindow::storeSettings()
{
    QSettings settings;
    settings.beginGroup("QGC_MAINWINDOW");
    settings.setValue("AUTO_RECONNECT", autoReconnect);
    settings.setValue("CURRENT_STYLE", currentStyle);
    settings.endGroup();
pixhawk's avatar
pixhawk committed
678 679 680 681 682 683 684 685 686 687 688
    if (!aboutToCloseFlag && isVisible())
    {
        settings.setValue(getWindowGeometryKey(), saveGeometry());
        // Save the last current view in any case
        settings.setValue("CURRENT_VIEW", currentView);
        // Save the current window state, but only if a system is connected (else no real number of widgets would be present)
        if (UASManager::instance()->getUASList().length() > 0) settings.setValue(getWindowStateKey(), saveState(QGC::applicationVersion()));
        // Save the current view only if a UAS is connected
        if (UASManager::instance()->getUASList().length() > 0) settings.setValue("CURRENT_VIEW_WITH_UAS_CONNECTED", currentView);
        // Save the current power mode
    }
689
    settings.setValue("LOW_POWER_MODE", lowPowerMode);
690 691 692
    settings.sync();
}

693 694
void MainWindow::configureWindowName()
{
pixhawk's avatar
pixhawk committed
695 696 697
    QList<QHostAddress> hostAddresses = QNetworkInterface::allAddresses();
    QString windowname = qApp->applicationName() + " " + qApp->applicationVersion();
    bool prevAddr = false;
698

pixhawk's avatar
pixhawk committed
699
    windowname.append(" (" + QHostInfo::localHostName() + ": ");
700

701 702
    for (int i = 0; i < hostAddresses.size(); i++)
    {
pixhawk's avatar
pixhawk committed
703
        // Exclude loopback IPv4 and all IPv6 addresses
704 705
        if (hostAddresses.at(i) != QHostAddress("127.0.0.1") && !hostAddresses.at(i).toString().contains(":"))
        {
pixhawk's avatar
pixhawk committed
706 707 708 709 710
            if(prevAddr) windowname.append("/");
            windowname.append(hostAddresses.at(i).toString());
            prevAddr = true;
        }
    }
711

pixhawk's avatar
pixhawk committed
712
    windowname.append(")");
713

pixhawk's avatar
pixhawk committed
714
    setWindowTitle(windowname);
715 716

#ifndef Q_WS_MAC
pixhawk's avatar
pixhawk committed
717
    //qApp->setWindowIcon(QIcon(":/core/images/qtcreator_logo_128.png"));
718 719 720
#endif
}

pixhawk's avatar
pixhawk committed
721
void MainWindow::startVideoCapture()
pixhawk's avatar
pixhawk committed
722 723 724 725 726
{
    QString format = "bmp";
    QString initialPath = QDir::currentPath() + tr("/untitled.") + format;

    QString screenFileName = QFileDialog::getSaveFileName(this, tr("Save As"),
lm's avatar
lm committed
727 728 729 730
                                                          initialPath,
                                                          tr("%1 Files (*.%2);;All Files (*)")
                                                          .arg(format.toUpper())
                                                          .arg(format));
pixhawk's avatar
pixhawk committed
731 732
    delete videoTimer;
    videoTimer = new QTimer(this);
733 734 735
    //videoTimer->setInterval(40);
    //connect(videoTimer, SIGNAL(timeout()), this, SLOT(saveScreen()));
    //videoTimer->stop();
pixhawk's avatar
pixhawk committed
736 737
}

pixhawk's avatar
pixhawk committed
738
void MainWindow::stopVideoCapture()
pixhawk's avatar
pixhawk committed
739 740 741 742 743 744
{
    videoTimer->stop();

    // TODO Convert raw images to PNG
}

pixhawk's avatar
pixhawk committed
745
void MainWindow::saveScreen()
pixhawk's avatar
pixhawk committed
746 747 748 749
{
    QPixmap window = QPixmap::grabWindow(this->winId());
    QString format = "bmp";

750 751
    if (!screenFileName.isEmpty())
    {
pixhawk's avatar
pixhawk committed
752 753 754 755
        window.save(screenFileName, format.toAscii());
    }
}

756
void MainWindow::enableAutoReconnect(bool enabled)
pixhawk's avatar
pixhawk committed
757
{
758 759
    autoReconnect = enabled;
}
760

761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777
void MainWindow::loadNativeStyle()
{
    loadStyle(QGC_MAINWINDOW_STYLE_NATIVE);
}

void MainWindow::loadIndoorStyle()
{
    loadStyle(QGC_MAINWINDOW_STYLE_INDOOR);
}

void MainWindow::loadOutdoorStyle()
{
    loadStyle(QGC_MAINWINDOW_STYLE_OUTDOOR);
}

void MainWindow::loadStyle(QGC_MAINWINDOW_STYLE style)
{
778 779 780 781 782 783 784 785 786
    switch (style) {
    case QGC_MAINWINDOW_STYLE_NATIVE: {
        // Native mode means setting no style
        // so if we were already in native mode
        // take no action
        // Only if a style was set, remove it.
        if (style != currentStyle) {
            qApp->setStyleSheet("");
            showInfoMessage(tr("Please restart QGroundControl"), tr("Please restart QGroundControl to switch to fully native look and feel. Currently you have loaded Qt's plastique style."));
787
        }
788 789
    }
    break;
790 791 792 793 794 795 796 797 798 799 800 801 802 803
    case QGC_MAINWINDOW_STYLE_INDOOR:
        qApp->setStyle("plastique");
        styleFileName = ":/images/style-mission.css";
        reloadStylesheet();
        break;
    case QGC_MAINWINDOW_STYLE_OUTDOOR:
        qApp->setStyle("plastique");
        styleFileName = ":/images/style-outdoor.css";
        reloadStylesheet();
        break;
    }
    currentStyle = style;
}

804
void MainWindow::selectStylesheet()
pixhawk's avatar
pixhawk committed
805
{
806
    // Let user select style sheet
807
    styleFileName = QFileDialog::getOpenFileName(this, tr("Specify stylesheet"), styleFileName, tr("CSS Stylesheet (*.css);;"));
808

809
    if (!styleFileName.endsWith(".css")) {
810 811 812 813 814 815 816 817 818 819
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Information);
        msgBox.setText(tr("QGroundControl did lot load a new style"));
        msgBox.setInformativeText(tr("No suitable .css file selected. Please select a valid .css file."));
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
        return;
    }

pixhawk's avatar
pixhawk committed
820
    // Load style sheet
821 822 823 824 825 826 827
    reloadStylesheet();
}

void MainWindow::reloadStylesheet()
{
    // Load style sheet
    QFile* styleSheet = new QFile(styleFileName);
828
    if (!styleSheet->exists()) {
829 830
        styleSheet = new QFile(":/images/style-mission.css");
    }
831
    if (styleSheet->open(QIODevice::ReadOnly | QIODevice::Text)) {
832 833
        QString style = QString(styleSheet->readAll());
        style.replace("ICONDIR", QCoreApplication::applicationDirPath()+ "/images/");
pixhawk's avatar
pixhawk committed
834
        qApp->setStyleSheet(style);
835
    } else {
836 837 838
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Information);
        msgBox.setText(tr("QGroundControl did lot load a new style"));
839
        msgBox.setInformativeText(tr("Stylesheet file %1 was not readable").arg(styleFileName));
840 841 842
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
pixhawk's avatar
pixhawk committed
843
    }
844
    delete styleSheet;
pixhawk's avatar
pixhawk committed
845 846
}

847 848 849 850 851 852
/**
 * The status message will be overwritten if a new message is posted to this function
 *
 * @param status message text
 * @param timeout how long the status should be displayed
 */
pixhawk's avatar
pixhawk committed
853
void MainWindow::showStatusMessage(const QString& status, int timeout)
pixhawk's avatar
pixhawk committed
854
{
lm's avatar
lm committed
855
    statusBar()->showMessage(status, timeout);
pixhawk's avatar
pixhawk committed
856 857
}

858 859 860 861 862 863
/**
 * The status message will be overwritten if a new message is posted to this function.
 * it will be automatically hidden after 5 seconds.
 *
 * @param status message text
 */
864
void MainWindow::showStatusMessage(const QString& status)
pixhawk's avatar
pixhawk committed
865
{
lm's avatar
lm committed
866
    statusBar()->showMessage(status, 20000);
867 868 869 870
}

void MainWindow::showCriticalMessage(const QString& title, const QString& message)
{
871
    QMessageBox msgBox(this);
872 873 874 875 876 877
    msgBox.setIcon(QMessageBox::Critical);
    msgBox.setText(title);
    msgBox.setInformativeText(message);
    msgBox.setStandardButtons(QMessageBox::Ok);
    msgBox.setDefaultButton(QMessageBox::Ok);
    msgBox.exec();
pixhawk's avatar
pixhawk committed
878 879
}

lm's avatar
lm committed
880 881
void MainWindow::showInfoMessage(const QString& title, const QString& message)
{
882
    QMessageBox msgBox(this);
lm's avatar
lm committed
883 884 885 886 887 888 889 890
    msgBox.setIcon(QMessageBox::Information);
    msgBox.setText(title);
    msgBox.setInformativeText(message);
    msgBox.setStandardButtons(QMessageBox::Ok);
    msgBox.setDefaultButton(QMessageBox::Ok);
    msgBox.exec();
}

pixhawk's avatar
pixhawk committed
891 892 893 894
/**
* @brief Create all actions associated to the main window
*
**/
895
void MainWindow::connectCommonActions()
pixhawk's avatar
pixhawk committed
896
{
897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
    // Bind together the perspective actions
    QActionGroup* perspectives = new QActionGroup(ui.menuPerspectives);
    perspectives->addAction(ui.actionEngineersView);
    perspectives->addAction(ui.actionMavlinkView);
    perspectives->addAction(ui.actionPilotsView);
    perspectives->addAction(ui.actionOperatorsView);
    perspectives->addAction(ui.actionUnconnectedView);
    perspectives->setExclusive(true);

    // Mark the right one as selected
    if (currentView == VIEW_ENGINEER) ui.actionEngineersView->setChecked(true);
    if (currentView == VIEW_MAVLINK) ui.actionMavlinkView->setChecked(true);
    if (currentView == VIEW_PILOT) ui.actionPilotsView->setChecked(true);
    if (currentView == VIEW_OPERATOR) ui.actionOperatorsView->setChecked(true);
    if (currentView == VIEW_UNCONNECTED) ui.actionUnconnectedView->setChecked(true);

    // The UAS actions are not enabled without connection to system
    ui.actionLiftoff->setEnabled(false);
    ui.actionLand->setEnabled(false);
    ui.actionEmergency_Kill->setEnabled(false);
    ui.actionEmergency_Land->setEnabled(false);
    ui.actionShutdownMAV->setEnabled(false);

pixhawk's avatar
pixhawk committed
920 921 922 923 924
    // Connect actions from ui
    connect(ui.actionAdd_Link, SIGNAL(triggered()), this, SLOT(addLink()));

    // Connect internal actions
    connect(UASManager::instance(), SIGNAL(UASCreated(UASInterface*)), this, SLOT(UASCreated(UASInterface*)));
925
    connect(UASManager::instance(), SIGNAL(activeUASSet(UASInterface*)), this, SLOT(setActiveUAS(UASInterface*)));
pixhawk's avatar
pixhawk committed
926

927
    // Unmanned System controls
pixhawk's avatar
pixhawk committed
928 929 930 931
    connect(ui.actionLiftoff, SIGNAL(triggered()), UASManager::instance(), SLOT(launchActiveUAS()));
    connect(ui.actionLand, SIGNAL(triggered()), UASManager::instance(), SLOT(returnActiveUAS()));
    connect(ui.actionEmergency_Land, SIGNAL(triggered()), UASManager::instance(), SLOT(stopActiveUAS()));
    connect(ui.actionEmergency_Kill, SIGNAL(triggered()), UASManager::instance(), SLOT(killActiveUAS()));
932
    connect(ui.actionShutdownMAV, SIGNAL(triggered()), UASManager::instance(), SLOT(shutdownActiveUAS()));
pixhawk's avatar
pixhawk committed
933 934
    connect(ui.actionConfiguration, SIGNAL(triggered()), UASManager::instance(), SLOT(configureActiveUAS()));

935
    // Views actions
936
    connect(ui.actionPilotsView, SIGNAL(triggered()), this, SLOT(loadPilotView()));
937
    connect(ui.actionEngineersView, SIGNAL(triggered()), this, SLOT(loadEngineerView()));
938
    connect(ui.actionOperatorsView, SIGNAL(triggered()), this, SLOT(loadOperatorView()));
939
    connect(ui.actionUnconnectedView, SIGNAL(triggered()), this, SLOT(loadUnconnectedView()));
940

941
    connect(ui.actionMavlinkView, SIGNAL(triggered()), this, SLOT(loadMAVLinkView()));
942 943
    connect(ui.actionReloadStylesheet, SIGNAL(triggered()), this, SLOT(reloadStylesheet()));
    connect(ui.actionSelectStylesheet, SIGNAL(triggered()), this, SLOT(selectStylesheet()));
944

945
    // Help Actions
946
    connect(ui.actionOnline_Documentation, SIGNAL(triggered()), this, SLOT(showHelp()));
947
    connect(ui.actionDeveloper_Credits, SIGNAL(triggered()), this, SLOT(showCredits()));
948
    connect(ui.actionProject_Roadmap_2, SIGNAL(triggered()), this, SLOT(showRoadMap()));
949 950 951

    // Custom widget actions
    connect(ui.actionNewCustomWidget, SIGNAL(triggered()), this, SLOT(createCustomWidget()));
952
    connect(ui.actionLoadCustomWidgetFile, SIGNAL(triggered()), this, SLOT(loadCustomWidget()));
953 954 955

    // Audio output
    ui.actionMuteAudioOutput->setChecked(GAudioOutput::instance()->isMuted());
956
    connect(GAudioOutput::instance(), SIGNAL(mutedChanged(bool)), ui.actionMuteAudioOutput, SLOT(setChecked(bool)));
957
    connect(ui.actionMuteAudioOutput, SIGNAL(triggered(bool)), GAudioOutput::instance(), SLOT(mute(bool)));
958 959 960 961 962 963 964

    // User interaction
    // NOTE: Joystick thread is not started and
    // configuration widget is not instantiated
    // unless it is actually used
    // so no ressources spend on this.
    ui.actionJoystickSettings->setVisible(true);
965 966 967

    // Configuration
    // Joystick
968
    connect(ui.actionJoystickSettings, SIGNAL(triggered()), this, SLOT(configure()));
969 970
    // Application Settings
    connect(ui.actionSettings, SIGNAL(triggered()), this, SLOT(showSettings()));
971 972
}

973 974
void MainWindow::showHelp()
{
lm's avatar
lm committed
975
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/users/start"))) {
976 977 978 979 980 981 982 983 984 985 986 987
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Critical);
        msgBox.setText("Could not open help in browser");
        msgBox.setInformativeText("To get to the online help, please open http://qgroundcontrol.org/user_guide in a browser.");
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
    }
}

void MainWindow::showCredits()
{
988
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/credits"))) {
989 990 991 992 993 994 995 996 997 998 999 1000
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Critical);
        msgBox.setText("Could not open credits in browser");
        msgBox.setInformativeText("To get to the online help, please open http://qgroundcontrol.org/credits in a browser.");
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
    }
}

void MainWindow::showRoadMap()
{
lm's avatar
lm committed
1001
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/dev/roadmap"))) {
1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Critical);
        msgBox.setText("Could not open roadmap in browser");
        msgBox.setInformativeText("To get to the online help, please open http://qgroundcontrol.org/roadmap in a browser.");
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
    }
}

pixhawk's avatar
pixhawk committed
1012
void MainWindow::configure()
pixhawk's avatar
pixhawk committed
1013
{
1014 1015
    if (!joystickWidget) {
        if (!joystick->isRunning()) {
1016 1017 1018 1019 1020
            joystick->start();
        }
        joystickWidget = new JoystickWidget(joystick);
    }
    joystickWidget->show();
pixhawk's avatar
pixhawk committed
1021 1022
}

1023 1024 1025 1026 1027 1028
void MainWindow::showSettings()
{
    QGCSettingsWidget* settings = new QGCSettingsWidget(this);
    settings->show();
}

pixhawk's avatar
pixhawk committed
1029
void MainWindow::addLink()
pixhawk's avatar
pixhawk committed
1030 1031 1032 1033
{
    SerialLink* link = new SerialLink();
    // TODO This should be only done in the dialog itself

1034
    LinkManager::instance()->add(link);
pixhawk's avatar
pixhawk committed
1035 1036
    LinkManager::instance()->addProtocol(link, mavlink);

1037 1038
    // Go fishing for this link's configuration window
    QList<QAction*> actions = ui.menuNetwork->actions();
pixhawk's avatar
pixhawk committed
1039

lm's avatar
lm committed
1040 1041 1042 1043
    foreach (QAction* act, actions)
    {
        if (act->data().toInt() == LinkManager::instance()->getLinks().indexOf(link))
        {
1044 1045 1046 1047
            act->trigger();
            break;
        }
    }
pixhawk's avatar
pixhawk committed
1048 1049
}

pixhawk's avatar
pixhawk committed
1050 1051
void MainWindow::addLink(LinkInterface *link)
{
1052 1053 1054 1055
    // IMPORTANT! KEEP THESE TWO LINES
    // THEY MAKE SURE THE LINK IS PROPERLY REGISTERED
    // BEFORE LINKING THE UI AGAINST IT
    // Register (does nothing if already registered)
1056
    LinkManager::instance()->add(link);
1057
    LinkManager::instance()->addProtocol(link, mavlink);
1058

1059 1060
    // Go fishing for this link's configuration window
    QList<QAction*> actions = ui.menuNetwork->actions();
pixhawk's avatar
pixhawk committed
1061

1062
    bool found = false;
1063

1064 1065
    foreach (QAction* act, actions) {
        if (act->data().toInt() == LinkManager::instance()->getLinks().indexOf(link)) {
1066 1067 1068 1069 1070
            found = true;
        }
    }

    UDPLink* udp = dynamic_cast<UDPLink*>(link);
1071

1072
    if (!found || udp) {
1073 1074 1075 1076 1077 1078 1079 1080
        CommConfigurationWindow* commWidget = new CommConfigurationWindow(link, mavlink, this);
        QAction* action = commWidget->getAction();
        ui.menuNetwork->addAction(action);

        // Error handling
        connect(link, SIGNAL(communicationError(QString,QString)), this, SLOT(showCriticalMessage(QString,QString)), Qt::QueuedConnection);
        // Special case for simulationlink
        MAVLinkSimulationLink* sim = dynamic_cast<MAVLinkSimulationLink*>(link);
1081
        if (sim) {
1082 1083 1084
            //connect(sim, SIGNAL(valueChanged(int,QString,double,quint64)), linechart, SLOT(appendData(int,QString,double,quint64)));
            connect(ui.actionSimulate, SIGNAL(triggered(bool)), sim, SLOT(connectLink(bool)));
        }
pixhawk's avatar
pixhawk committed
1085 1086 1087
    }
}

1088 1089 1090 1091 1092 1093 1094
void MainWindow::setActiveUAS(UASInterface* uas)
{
    // Enable and rename menu
    ui.menuUnmanned_System->setTitle(uas->getUASName());
    if (!ui.menuUnmanned_System->isEnabled()) ui.menuUnmanned_System->setEnabled(true);
}

1095 1096 1097
void MainWindow::UASSpecsChanged(int uas)
{
    UASInterface* activeUAS = UASManager::instance()->getActiveUAS();
1098 1099
    if (activeUAS) {
        if (activeUAS->getUASID() == uas) {
1100 1101 1102 1103 1104
            ui.menuUnmanned_System->setTitle(activeUAS->getUASName());
        }
    }
}

pixhawk's avatar
pixhawk committed
1105
void MainWindow::UASCreated(UASInterface* uas)
pixhawk's avatar
pixhawk committed
1106
{
1107

pixhawk's avatar
pixhawk committed
1108 1109
    // Connect the UAS to the full user interface

lm's avatar
lm committed
1110 1111
    if (uas != NULL)
    {
1112
        // The pilot, operator and engineer views were not available on startup, enable them now
1113
        ui.actionPilotsView->setEnabled(true);
1114 1115
        ui.actionOperatorsView->setEnabled(true);
        ui.actionEngineersView->setEnabled(true);
1116 1117 1118 1119 1120 1121
        // The UAS actions are not enabled without connection to system
        ui.actionLiftoff->setEnabled(true);
        ui.actionLand->setEnabled(true);
        ui.actionEmergency_Kill->setEnabled(true);
        ui.actionEmergency_Land->setEnabled(true);
        ui.actionShutdownMAV->setEnabled(true);
1122

1123 1124
        QIcon icon;
        // Set matching icon
1125
        switch (uas->getSystemType()) {
1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148
        case 0:
            icon = QIcon(":/images/mavs/generic.svg");
            break;
        case 1:
            icon = QIcon(":/images/mavs/fixed-wing.svg");
            break;
        case 2:
            icon = QIcon(":/images/mavs/quadrotor.svg");
            break;
        case 3:
            icon = QIcon(":/images/mavs/coaxial.svg");
            break;
        case 4:
            icon = QIcon(":/images/mavs/helicopter.svg");
            break;
        case 5:
            icon = QIcon(":/images/mavs/groundstation.svg");
            break;
        default:
            icon = QIcon(":/images/mavs/unknown.svg");
            break;
        }

1149 1150 1151
        QAction* uasAction = new QAction(icon, tr("Select %1 for control").arg(uas->getUASName()), ui.menuConnected_Systems);
        connect(uas, SIGNAL(systemRemoved()), uasAction, SLOT(deleteLater()));
        connect(uasAction, SIGNAL(triggered()), uas, SLOT(setSelected()));
1152
        connect(uas, SIGNAL(systemSpecsChanged(int)), this, SLOT(UASSpecsChanged(int)));
1153 1154

        ui.menuConnected_Systems->addAction(uasAction);
1155 1156

        // FIXME Should be not inside the mainwindow
1157
        if (debugConsoleDockWidget) {
pixhawk's avatar
pixhawk committed
1158
            DebugConsole *debugConsole = dynamic_cast<DebugConsole*>(debugConsoleDockWidget->widget());
1159
            if (debugConsole) {
pixhawk's avatar
pixhawk committed
1160 1161 1162
                connect(uas, SIGNAL(textMessageReceived(int,int,int,QString)),
                        debugConsole, SLOT(receiveTextMessage(int,int,int,QString)));
            }
1163
        }
1164 1165

        // Health / System status indicator
1166
        if (infoDockWidget) {
pixhawk's avatar
pixhawk committed
1167
            UASInfoWidget *infoWidget = dynamic_cast<UASInfoWidget*>(infoDockWidget->widget());
1168
            if (infoWidget) {
pixhawk's avatar
pixhawk committed
1169 1170
                infoWidget->addUAS(uas);
            }
1171
        }
1172 1173

        // UAS List
lm's avatar
lm committed
1174 1175
        if (listDockWidget)
        {
pixhawk's avatar
pixhawk committed
1176
            UASListWidget *listWidget = dynamic_cast<UASListWidget*>(listDockWidget->widget());
lm's avatar
lm committed
1177 1178
            if (listWidget)
            {
pixhawk's avatar
pixhawk committed
1179 1180
                listWidget->addUAS(uas);
            }
1181
        }
1182

1183 1184
        // Line chart
        if (!linechartWidget)
lm's avatar
lm committed
1185
        {
1186
            // Center widgets
LM's avatar
LM committed
1187
            linechartWidget = new Linecharts(this);
1188
            linechartWidget->addSource(mavlinkDecoder);
1189
            addCentralWidget(linechartWidget, tr("Realtime Plot"));
1190
        }
1191

LM's avatar
LM committed
1192
        // Load default custom widgets for this autopilot type
lm's avatar
lm committed
1193
        loadCustomWidgetsFromDefaults(uas->getSystemTypeName(), uas->getAutopilotTypeName());
1194 1195


lm's avatar
lm committed
1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
        if (uas->getAutopilotType() == MAV_AUTOPILOT_PIXHAWK)
        {
            // Dock widgets
            if (!detectionDockWidget)
            {
                detectionDockWidget = new QDockWidget(tr("Object Recognition"), this);
                detectionDockWidget->setWidget( new ObjectDetectionView("images/patterns", this) );
                detectionDockWidget->setObjectName("OBJECT_DETECTION_DOCK_WIDGET");
                addTool(detectionDockWidget, tr("Object Recognition"), Qt::RightDockWidgetArea);
            }

            if (!watchdogControlDockWidget) {
                watchdogControlDockWidget = new QDockWidget(tr("Process Control"), this);
                watchdogControlDockWidget->setWidget( new WatchdogControl(this) );
                watchdogControlDockWidget->setObjectName("WATCHDOG_CONTROL_DOCKWIDGET");
                addTool(watchdogControlDockWidget, tr("Process Control"), Qt::BottomDockWidgetArea);
            }
1213 1214 1215 1216 1217 1218
        }

        // Change the view only if this is the first UAS

        // If this is the first connected UAS, it is both created as well as
        // the currently active UAS
1219 1220
        if (UASManager::instance()->getUASList().size() == 1)
        {
1221
            // Load last view if setting is present
1222 1223
            if (settings.contains("CURRENT_VIEW_WITH_UAS_CONNECTED"))
            {
1224
                int view = settings.value("CURRENT_VIEW_WITH_UAS_CONNECTED").toInt();
1225
                switch (view) {
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241
                case VIEW_ENGINEER:
                    loadEngineerView();
                    break;
                case VIEW_MAVLINK:
                    loadMAVLinkView();
                    break;
                case VIEW_PILOT:
                    loadPilotView();
                    break;
                case VIEW_UNCONNECTED:
                    loadUnconnectedView();
                    break;
                case VIEW_OPERATOR:
                default:
                    loadOperatorView();
                    break;
1242
                }
1243 1244 1245
            }
            else
            {
1246
                loadOperatorView();
1247
            }
1248
        }
1249

1250
    }
1251 1252

    if (!ui.menuConnected_Systems->isEnabled()) ui.menuConnected_Systems->setEnabled(true);
lm's avatar
lm committed
1253

1254
    // Restore the mainwindow size
LM's avatar
LM committed
1255 1256
    if (settings.contains(getWindowGeometryKey()))
    {
1257 1258
        restoreGeometry(settings.value(getWindowGeometryKey()).toByteArray());
    }
pixhawk's avatar
pixhawk committed
1259 1260
}

1261
/**
1262
 * Stores the current view state
1263
 */
1264
void MainWindow::storeViewState()
1265
{
pixhawk's avatar
pixhawk committed
1266
    if (!aboutToCloseFlag)
lm's avatar
lm committed
1267
    {
pixhawk's avatar
pixhawk committed
1268 1269 1270 1271 1272 1273 1274 1275
        // Save current state
        settings.setValue(getWindowStateKey(), saveState(QGC::applicationVersion()));
        settings.setValue(getWindowStateKey()+"CENTER_WIDGET", centerStack->currentIndex());
        // Although we want save the state of the window, we do not want to change the top-leve state (minimized, maximized, etc)
        // therefore this state is stored here and restored after applying the rest of the settings in the new
        // perspective.
        windowStateVal = this->windowState();
        settings.setValue(getWindowGeometryKey(), saveGeometry());
1276
    }
1277
}
1278

1279 1280 1281
void MainWindow::loadViewState()
{
    // Restore center stack state
lm's avatar
lm committed
1282
    int index = settings.value(getWindowStateKey()+"CENTER_WIDGET", -1).toInt();
LM's avatar
LM committed
1283 1284 1285 1286 1287 1288
    // The offline plot view is usually the consequence of a logging run, always show the realtime view first
    if (centerStack->indexOf(dataplotWidget) == index)
    {
        // Rewrite to realtime plot
        index = centerStack->indexOf(linechartWidget);
    }
pixhawk's avatar
pixhawk committed
1289

lm's avatar
lm committed
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300
    if (index != -1)
    {
        centerStack->setCurrentIndex(index);
    }
    else
    {
        // Load defaults
        switch (currentView)
        {
        case VIEW_ENGINEER:
            centerStack->setCurrentWidget(linechartWidget);
lm's avatar
lm committed
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
            controlDockWidget->hide();
            listDockWidget->hide();
            waypointsDockWidget->hide();
            infoDockWidget->hide();
            debugConsoleDockWidget->show();
            logPlayerDockWidget->show();
            mavlinkInspectorWidget->show();
            parametersDockWidget->show();
            hsiDockWidget->hide();
            headDown1DockWidget->hide();
            headDown2DockWidget->hide();
            rcViewDockWidget->hide();
            headUpDockWidget->hide();
            video1DockWidget->hide();
            video2DockWidget->hide();
lm's avatar
lm committed
1316 1317 1318
            break;
        case VIEW_PILOT:
            centerStack->setCurrentWidget(hudWidget);
lm's avatar
lm committed
1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333
            controlDockWidget->hide();
            listDockWidget->hide();
            waypointsDockWidget->hide();
            infoDockWidget->hide();
            debugConsoleDockWidget->hide();
            logPlayerDockWidget->hide();
            mavlinkInspectorWidget->hide();
            parametersDockWidget->hide();
            hsiDockWidget->show();
            headDown1DockWidget->show();
            headDown2DockWidget->show();
            rcViewDockWidget->hide();
            headUpDockWidget->hide();
            video1DockWidget->show();
            video2DockWidget->hide();
lm's avatar
lm committed
1334 1335 1336
            break;
        case VIEW_MAVLINK:
            centerStack->setCurrentWidget(protocolWidget);
lm's avatar
lm committed
1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351
            controlDockWidget->hide();
            listDockWidget->hide();
            waypointsDockWidget->hide();
            infoDockWidget->hide();
            debugConsoleDockWidget->hide();
            logPlayerDockWidget->hide();
            mavlinkInspectorWidget->hide();
            parametersDockWidget->hide();
            hsiDockWidget->hide();
            headDown1DockWidget->hide();
            headDown2DockWidget->hide();
            rcViewDockWidget->hide();
            headUpDockWidget->hide();
            video1DockWidget->hide();
            video2DockWidget->hide();
lm's avatar
lm committed
1352 1353 1354 1355 1356 1357
            break;
        case VIEW_OPERATOR:
        case VIEW_UNCONNECTED:
        case VIEW_FULL:
        default:
            centerStack->setCurrentWidget(mapWidget);
lm's avatar
lm committed
1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372
            controlDockWidget->show();
            listDockWidget->show();
            waypointsDockWidget->show();
            infoDockWidget->show();
            debugConsoleDockWidget->show();
            logPlayerDockWidget->show();
            mavlinkInspectorWidget->show();
            parametersDockWidget->hide();
            hsiDockWidget->show();
            headDown1DockWidget->hide();
            headDown2DockWidget->hide();
            rcViewDockWidget->hide();
            headUpDockWidget->show();
            video1DockWidget->hide();
            video2DockWidget->hide();
lm's avatar
lm committed
1373
            break;
pixhawk's avatar
pixhawk committed
1374 1375 1376
        }
    }

1377 1378 1379 1380
    // Restore the widget positions and size
    if (settings.contains(getWindowStateKey()))
    {
        restoreState(settings.value(getWindowStateKey()).toByteArray(), QGC::applicationVersion());
pixhawk's avatar
pixhawk committed
1381 1382 1383
    }
}

1384
void MainWindow::loadEngineerView()
lm's avatar
lm committed
1385
{
1386
    if (currentView != VIEW_ENGINEER) {
1387
        storeViewState();
1388 1389
        currentView = VIEW_ENGINEER;
        ui.actionEngineersView->setChecked(true);
1390
        loadViewState();
1391
    }
lm's avatar
lm committed
1392 1393
}

1394
void MainWindow::loadOperatorView()
lm's avatar
lm committed
1395
{
1396
    if (currentView != VIEW_OPERATOR) {
1397
        storeViewState();
1398 1399
        currentView = VIEW_OPERATOR;
        ui.actionOperatorsView->setChecked(true);
1400
        loadViewState();
1401 1402
    }
}
lm's avatar
lm committed
1403

1404 1405
void MainWindow::loadUnconnectedView()
{
1406 1407 1408
    if (currentView != VIEW_UNCONNECTED)
    {
        storeViewState();
1409 1410
        currentView = VIEW_UNCONNECTED;
        ui.actionUnconnectedView->setChecked(true);
1411
        loadViewState();
1412
    }
lm's avatar
lm committed
1413 1414
}

1415
void MainWindow::loadPilotView()
1416
{
1417 1418 1419
    if (currentView != VIEW_PILOT)
    {
        storeViewState();
1420 1421
        currentView = VIEW_PILOT;
        ui.actionPilotsView->setChecked(true);
1422
        loadViewState();
1423
    }
1424 1425
}

1426
void MainWindow::loadMAVLinkView()
1427
{
1428 1429 1430
    if (currentView != VIEW_MAVLINK)
    {
        storeViewState();
1431 1432
        currentView = VIEW_MAVLINK;
        ui.actionMavlinkView->setChecked(true);
1433
        loadViewState();
1434
    }
1435 1436
}

1437 1438
void MainWindow::loadDataView(QString fileName)
{
1439
    // Plot is now selected, now load data from file
1440 1441 1442 1443 1444 1445 1446 1447
    if (dataplotWidget)
    {
        dataplotWidget->loadFile(fileName);
    }
    QStackedWidget *centerStack = dynamic_cast<QStackedWidget*>(centralWidget());
    if (centerStack)
    {
        centerStack->setCurrentWidget(dataplotWidget);
1448
        dataplotWidget->loadFile(fileName);
1449
    }
1450 1451 1452
}


oberion's avatar
oberion committed
1453 1454
QList<QAction*> MainWindow::listLinkMenuActions(void)
{
lm's avatar
lm committed
1455
    return ui.menuNetwork->actions();
1456
}