MainWindow.cc 48.6 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

lm's avatar
lm committed
149 150
    buildCustomWidget();

151

lm's avatar
lm committed
152

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

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

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

178 179
    connect(LinkManager::instance(), SIGNAL(newLink(LinkInterface*)), this, SLOT(addLink(LinkInterface*)));

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

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

194 195 196
    // Set low power mode
    enableLowPowerMode(lowPowerMode);

197 198
    // Initialize window state
    windowStateVal = windowState();
199 200

    show();
201 202 203

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

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

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

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

LM's avatar
LM committed
233
    // Delete all UAS objects
234 235
}

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

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

    QMainWindow::resizeEvent(event);
lm's avatar
lm committed
258 259
}

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

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

lm's avatar
lm committed
271 272
void MainWindow::buildCustomWidget()
{
lm's avatar
lm committed
273 274
    // Create custom widgets
    QList<QGCToolWidget*> widgets = QGCToolWidget::createWidgetsFromSettings(this);
lm's avatar
lm committed
275

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

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

            // 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
303 304 305 306
        }
    }
}

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

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

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

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

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

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

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

357
        addTool(debugConsoleDockWidget, tr("Communication Console"), Qt::BottomDockWidgetArea);
358
    }
359

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

lm's avatar
lm committed
370 371 372 373 374
    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
375
        addTool(mavlinkInspectorWidget, tr("MAVLink Inspector"), Qt::RightDockWidgetArea);
lm's avatar
lm committed
376 377
    }

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

pixhawk's avatar
pixhawk committed
384 385 386
    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");
387
    acceptList->append("0,airspeed,m/s,30");
388 389
    acceptList->append("0,groundspeed,m/s,30");
    acceptList->append("0,climbrate,m/s,30");
390
    acceptList->append("0,throttle,%,100");
391

pixhawk's avatar
pixhawk committed
392 393
    //FIXME: memory of acceptList2 will never be freed again
    QStringList* acceptList2 = new QStringList();
394 395 396
    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
397
    acceptList2->append("0,abs pressure,hPa,65500");
398

lm's avatar
lm committed
399 400 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
    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");
    }

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

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

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

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

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

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


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

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

533
        QAction* tempAction = ui.menuMain->addAction(title);
534

535 536 537 538 539 540
        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
541 542
        connect(widget, SIGNAL(visibilityChanged(bool)), tempAction, SLOT(setChecked(bool)));
        tempAction->setChecked(widget->isVisible());
543 544 545 546
    }
}


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

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

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

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

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

    connect(tool, SIGNAL(destroyed()), dock, SLOT(deleteLater()));
587
    dock->setWidget(tool);
588

589 590 591 592 593 594 595 596 597 598 599 600 601 602
    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));
603
    if (fileName != "") loadCustomWidget(fileName);
LM's avatar
LM committed
604
}
605

LM's avatar
LM committed
606 607 608 609
void MainWindow::loadCustomWidget(const QString& fileName, bool singleinstance)
{
    QGCToolWidget* tool = new QGCToolWidget("", this);
    if (tool->loadSettings(fileName, true) || !singleinstance)
LM's avatar
LM committed
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625
    {
        // 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
626 627 628
    else
    {
        return;
629
    }
630 631
}

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

LM's avatar
LM committed
636 637 638 639 640 641
    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
642
        showStatusMessage(tr("Did not find any custom widgets in %1").arg(defaultsDir));
643
    }
644

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

659 660 661 662 663 664
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();
665
    lowPowerMode = settings.value("LOW_POWER_MODE", lowPowerMode).toBool();
666 667 668 669 670 671 672 673 674 675
    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
676 677 678 679 680 681 682 683 684 685 686
    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
    }
687
    settings.setValue("LOW_POWER_MODE", lowPowerMode);
688 689 690
    settings.sync();
}

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

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

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

pixhawk's avatar
pixhawk committed
710
    windowname.append(")");
711

pixhawk's avatar
pixhawk committed
712
    setWindowTitle(windowname);
713 714

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

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

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

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

    // TODO Convert raw images to PNG
}

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

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

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

759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
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)
{
776 777 778 779 780 781 782 783 784
    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."));
785
        }
786 787
    }
    break;
788 789 790 791 792 793 794 795 796 797 798 799 800 801
    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;
}

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

807
    if (!styleFileName.endsWith(".css")) {
808 809 810 811 812 813 814 815 816 817
        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
818
    // Load style sheet
819 820 821 822 823 824 825
    reloadStylesheet();
}

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

845 846 847 848 849 850
/**
 * 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
851
void MainWindow::showStatusMessage(const QString& status, int timeout)
pixhawk's avatar
pixhawk committed
852
{
lm's avatar
lm committed
853
    statusBar()->showMessage(status, timeout);
pixhawk's avatar
pixhawk committed
854 855
}

856 857 858 859 860 861
/**
 * 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
 */
862
void MainWindow::showStatusMessage(const QString& status)
pixhawk's avatar
pixhawk committed
863
{
lm's avatar
lm committed
864
    statusBar()->showMessage(status, 20000);
865 866 867 868
}

void MainWindow::showCriticalMessage(const QString& title, const QString& message)
{
869
    QMessageBox msgBox(this);
870 871 872 873 874 875
    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
876 877
}

lm's avatar
lm committed
878 879
void MainWindow::showInfoMessage(const QString& title, const QString& message)
{
880
    QMessageBox msgBox(this);
lm's avatar
lm committed
881 882 883 884 885 886 887 888
    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
889 890 891 892
/**
* @brief Create all actions associated to the main window
*
**/
893
void MainWindow::connectCommonActions()
pixhawk's avatar
pixhawk committed
894
{
895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917
    // 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
918 919 920 921 922
    // 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*)));
923
    connect(UASManager::instance(), SIGNAL(activeUASSet(UASInterface*)), this, SLOT(setActiveUAS(UASInterface*)));
pixhawk's avatar
pixhawk committed
924

925
    // Unmanned System controls
pixhawk's avatar
pixhawk committed
926 927 928 929
    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()));
930
    connect(ui.actionShutdownMAV, SIGNAL(triggered()), UASManager::instance(), SLOT(shutdownActiveUAS()));
pixhawk's avatar
pixhawk committed
931 932
    connect(ui.actionConfiguration, SIGNAL(triggered()), UASManager::instance(), SLOT(configureActiveUAS()));

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

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

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

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

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

    // 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);
963 964 965

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

971 972
void MainWindow::showHelp()
{
lm's avatar
lm committed
973
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/users/start"))) {
974 975 976 977 978 979 980 981 982 983 984 985
        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()
{
986
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/credits"))) {
987 988 989 990 991 992 993 994 995 996 997 998
        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
999
    if(!QDesktopServices::openUrl(QUrl("http://qgroundcontrol.org/dev/roadmap"))) {
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009
        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
1010
void MainWindow::configure()
pixhawk's avatar
pixhawk committed
1011
{
1012 1013
    if (!joystickWidget) {
        if (!joystick->isRunning()) {
1014 1015 1016 1017 1018
            joystick->start();
        }
        joystickWidget = new JoystickWidget(joystick);
    }
    joystickWidget->show();
pixhawk's avatar
pixhawk committed
1019 1020
}

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

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

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

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

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

pixhawk's avatar
pixhawk committed
1048 1049
void MainWindow::addLink(LinkInterface *link)
{
1050 1051 1052 1053
    // 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)
1054
    LinkManager::instance()->add(link);
1055
    LinkManager::instance()->addProtocol(link, mavlink);
1056

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

1060
    bool found = false;
1061

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

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

1070
    if (!found || udp) {
1071 1072 1073 1074 1075 1076 1077 1078
        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);
1079
        if (sim) {
1080 1081 1082
            //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
1083 1084 1085
    }
}

1086 1087 1088 1089 1090 1091 1092
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);
}

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

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

pixhawk's avatar
pixhawk committed
1106 1107
    // Connect the UAS to the full user interface

lm's avatar
lm committed
1108 1109
    if (uas != NULL)
    {
1110
        // The pilot, operator and engineer views were not available on startup, enable them now
1111
        ui.actionPilotsView->setEnabled(true);
1112 1113
        ui.actionOperatorsView->setEnabled(true);
        ui.actionEngineersView->setEnabled(true);
1114 1115 1116 1117 1118 1119
        // 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);
1120

1121 1122
        QIcon icon;
        // Set matching icon
1123
        switch (uas->getSystemType()) {
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146
        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;
        }

1147 1148 1149
        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()));
1150
        connect(uas, SIGNAL(systemSpecsChanged(int)), this, SLOT(UASSpecsChanged(int)));
1151 1152

        ui.menuConnected_Systems->addAction(uasAction);
1153 1154

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

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

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

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

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


lm's avatar
lm committed
1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
        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);
            }
1211 1212 1213 1214 1215 1216
        }

        // 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
1217 1218
        if (UASManager::instance()->getUASList().size() == 1)
        {
1219
            // Load last view if setting is present
1220 1221
            if (settings.contains("CURRENT_VIEW_WITH_UAS_CONNECTED"))
            {
1222
                int view = settings.value("CURRENT_VIEW_WITH_UAS_CONNECTED").toInt();
1223
                switch (view) {
1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
                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;
1240
                }
1241 1242 1243
            }
            else
            {
1244
                loadOperatorView();
1245
            }
1246
        }
1247

1248
    }
1249 1250

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

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

1259
/**
1260
 * Stores the current view state
1261
 */
1262
void MainWindow::storeViewState()
1263
{
pixhawk's avatar
pixhawk committed
1264
    if (!aboutToCloseFlag)
lm's avatar
lm committed
1265
    {
pixhawk's avatar
pixhawk committed
1266 1267 1268 1269 1270 1271 1272 1273
        // 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());
1274
    }
1275
}
1276

1277 1278 1279
void MainWindow::loadViewState()
{
    // Restore center stack state
lm's avatar
lm committed
1280
    int index = settings.value(getWindowStateKey()+"CENTER_WIDGET", -1).toInt();
LM's avatar
LM committed
1281 1282 1283 1284 1285 1286
    // 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
1287

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

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

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

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

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

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

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

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


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