Add temporary snapping disable menu to markers and cursors
[pulseview.git] / pv / mainwindow.cpp
1 /*
2  * This file is part of the PulseView project.
3  *
4  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #ifdef ENABLE_DECODE
21 #include <libsigrokdecode/libsigrokdecode.h>
22 #endif
23
24 #include <algorithm>
25 #include <cassert>
26 #include <cstdarg>
27 #include <cstdint>
28 #include <iterator>
29
30 #include <QAction>
31 #include <QApplication>
32 #include <QCloseEvent>
33 #include <QDebug>
34 #include <QDockWidget>
35 #include <QHBoxLayout>
36 #include <QMessageBox>
37 #include <QSettings>
38 #include <QShortcut>
39 #include <QWidget>
40
41 #include "mainwindow.hpp"
42
43 #include "devicemanager.hpp"
44 #include "devices/hardwaredevice.hpp"
45 #include "dialogs/settings.hpp"
46 #include "globalsettings.hpp"
47 #include "toolbars/mainbar.hpp"
48 #include "util.hpp"
49 #include "views/trace/view.hpp"
50 #include "views/trace/standardbar.hpp"
51
52 #ifdef ENABLE_DECODE
53 #include "subwindows/decoder_selector/subwindow.hpp"
54 #endif
55
56 #include <libsigrokcxx/libsigrokcxx.hpp>
57
58 using std::dynamic_pointer_cast;
59 using std::make_shared;
60 using std::shared_ptr;
61 using std::string;
62
63 namespace pv {
64
65 namespace view {
66 class ViewItem;
67 }
68
69 using toolbars::MainBar;
70
71 const QString MainWindow::WindowTitle = tr("PulseView");
72
73 MainWindow::MainWindow(DeviceManager &device_manager, QWidget *parent) :
74         QMainWindow(parent),
75         device_manager_(device_manager),
76         session_selector_(this),
77         session_state_mapper_(this),
78         icon_red_(":/icons/status-red.svg"),
79         icon_green_(":/icons/status-green.svg"),
80         icon_grey_(":/icons/status-grey.svg")
81 {
82         GlobalSettings::add_change_handler(this);
83
84         setup_ui();
85         restore_ui_settings();
86 }
87
88 MainWindow::~MainWindow()
89 {
90         GlobalSettings::remove_change_handler(this);
91
92         // Make sure we no longer hold any shared pointers to widgets after the
93         // destructor finishes (goes for sessions and sub windows alike)
94
95         while (!sessions_.empty())
96                 remove_session(sessions_.front());
97
98         sub_windows_.clear();
99 }
100
101 void MainWindow::show_session_error(const QString text, const QString info_text)
102 {
103         // TODO Emulate noquote()
104         qDebug() << "Notifying user of session error:" << info_text;
105
106         QMessageBox msg;
107         msg.setText(text + "\n\n" + info_text);
108         msg.setStandardButtons(QMessageBox::Ok);
109         msg.setIcon(QMessageBox::Warning);
110         msg.exec();
111 }
112
113 shared_ptr<views::ViewBase> MainWindow::get_active_view() const
114 {
115         // If there's only one view, use it...
116         if (view_docks_.size() == 1)
117                 return view_docks_.begin()->second;
118
119         // ...otherwise find the dock widget the widget with focus is contained in
120         QObject *w = QApplication::focusWidget();
121         QDockWidget *dock = nullptr;
122
123         while (w) {
124                 dock = qobject_cast<QDockWidget*>(w);
125                 if (dock)
126                         break;
127                 w = w->parent();
128         }
129
130         // Get the view contained in the dock widget
131         for (auto& entry : view_docks_)
132                 if (entry.first == dock)
133                         return entry.second;
134
135         return nullptr;
136 }
137
138 shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
139         views::ViewType type, Session &session)
140 {
141         GlobalSettings settings;
142         shared_ptr<views::ViewBase> v;
143
144         QMainWindow *main_window = nullptr;
145         for (auto& entry : session_windows_)
146                 if (entry.first.get() == &session)
147                         main_window = entry.second;
148
149         assert(main_window);
150
151         shared_ptr<MainBar> main_bar = session.main_bar();
152
153         QDockWidget* dock = new QDockWidget(title, main_window);
154         dock->setObjectName(title);
155         main_window->addDockWidget(Qt::TopDockWidgetArea, dock);
156
157         // Insert a QMainWindow into the dock widget to allow for a tool bar
158         QMainWindow *dock_main = new QMainWindow(dock);
159         dock_main->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
160
161         if (type == views::ViewTypeTrace)
162                 // This view will be the main view if there's no main bar yet
163                 v = make_shared<views::trace::View>(session,
164                         (main_bar ? false : true), dock_main);
165
166         if (!v)
167                 return nullptr;
168
169         view_docks_[dock] = v;
170         session.register_view(v);
171
172         dock_main->setCentralWidget(v.get());
173         dock->setWidget(dock_main);
174
175         dock->setContextMenuPolicy(Qt::PreventContextMenu);
176         dock->setFeatures(QDockWidget::DockWidgetMovable |
177                 QDockWidget::DockWidgetFloatable | QDockWidget::DockWidgetClosable);
178
179         QAbstractButton *close_btn =
180                 dock->findChildren<QAbstractButton*>("qt_dockwidget_closebutton")  // clazy:exclude=detaching-temporary
181                         .front();
182
183         connect(close_btn, SIGNAL(clicked(bool)),
184                 this, SLOT(on_view_close_clicked()));
185
186         connect(&session, SIGNAL(trigger_event(int, util::Timestamp)),
187                 qobject_cast<views::ViewBase*>(v.get()),
188                 SLOT(trigger_event(int, util::Timestamp)));
189
190         if (type == views::ViewTypeTrace) {
191                 views::trace::View *tv =
192                         qobject_cast<views::trace::View*>(v.get());
193
194                 tv->enable_colored_bg(settings.value(GlobalSettings::Key_View_ColoredBG).toBool());
195                 tv->enable_show_sampling_points(settings.value(GlobalSettings::Key_View_ShowSamplingPoints).toBool());
196                 tv->enable_show_analog_minor_grid(settings.value(GlobalSettings::Key_View_ShowAnalogMinorGrid).toBool());
197
198                 if (!main_bar) {
199                         /* Initial view, create the main bar */
200                         main_bar = make_shared<MainBar>(session, this, tv);
201                         dock_main->addToolBar(main_bar.get());
202                         session.set_main_bar(main_bar);
203
204                         connect(main_bar.get(), SIGNAL(new_view(Session*)),
205                                 this, SLOT(on_new_view(Session*)));
206                         connect(main_bar.get(), SIGNAL(show_decoder_selector(Session*)),
207                                 this, SLOT(on_show_decoder_selector(Session*)));
208
209                         main_bar->action_view_show_cursors()->setChecked(tv->cursors_shown());
210
211                         /* For the main view we need to prevent the dock widget from
212                          * closing itself when its close button is clicked. This is
213                          * so we can confirm with the user first. Regular views don't
214                          * need this */
215                         close_btn->disconnect(SIGNAL(clicked()), dock, SLOT(close()));
216                 } else {
217                         /* Additional view, create a standard bar */
218                         pv::views::trace::StandardBar *standard_bar =
219                                 new pv::views::trace::StandardBar(session, this, tv);
220                         dock_main->addToolBar(standard_bar);
221
222                         standard_bar->action_view_show_cursors()->setChecked(tv->cursors_shown());
223                 }
224         }
225
226         return v;
227 }
228
229 void MainWindow::remove_view(shared_ptr<views::ViewBase> view)
230 {
231         for (shared_ptr<Session> session : sessions_) {
232                 if (!session->has_view(view))
233                         continue;
234
235                 // Find the dock the view is contained in and remove it
236                 for (auto& entry : view_docks_)
237                         if (entry.second == view) {
238                                 // Remove the view from the session
239                                 session->deregister_view(view);
240
241                                 // Remove the view from its parent; otherwise, Qt will
242                                 // call deleteLater() on it, which causes a double free
243                                 // since the shared_ptr in view_docks_ doesn't know
244                                 // that Qt keeps a pointer to the view around
245                                 view->setParent(nullptr);
246
247                                 // Delete the view's dock widget and all widgets inside it
248                                 entry.first->deleteLater();
249
250                                 // Remove the dock widget from the list and stop iterating
251                                 view_docks_.erase(entry.first);
252                                 break;
253                         }
254         }
255 }
256
257 shared_ptr<subwindows::SubWindowBase> MainWindow::add_subwindow(
258         subwindows::SubWindowType type, Session &session)
259 {
260         GlobalSettings settings;
261         shared_ptr<subwindows::SubWindowBase> v;
262
263         QMainWindow *main_window = nullptr;
264         for (auto entry : session_windows_)
265                 if (entry.first.get() == &session)
266                         main_window = entry.second;
267
268         assert(main_window);
269
270         QString title = "";
271
272         switch (type) {
273 #ifdef ENABLE_DECODE
274                 case subwindows::SubWindowTypeDecoderSelector:
275                         title = tr("Decoder Selector");
276                         break;
277 #endif
278                 default:
279                         break;
280         }
281
282         QDockWidget* dock = new QDockWidget(title, main_window);
283         dock->setObjectName(title);
284         main_window->addDockWidget(Qt::TopDockWidgetArea, dock);
285
286         // Insert a QMainWindow into the dock widget to allow for a tool bar
287         QMainWindow *dock_main = new QMainWindow(dock);
288         dock_main->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
289
290 #ifdef ENABLE_DECODE
291         if (type == subwindows::SubWindowTypeDecoderSelector)
292                 v = make_shared<subwindows::decoder_selector::SubWindow>(session, dock_main);
293 #endif
294
295         if (!v)
296                 return nullptr;
297
298         sub_windows_[dock] = v;
299         dock_main->setCentralWidget(v.get());
300         dock->setWidget(dock_main);
301
302         dock->setContextMenuPolicy(Qt::PreventContextMenu);
303         dock->setFeatures(QDockWidget::DockWidgetMovable |
304                 QDockWidget::DockWidgetFloatable | QDockWidget::DockWidgetClosable);
305
306         QAbstractButton *close_btn =
307                 dock->findChildren<QAbstractButton*>
308                         ("qt_dockwidget_closebutton").front();
309
310         connect(close_btn, SIGNAL(clicked(bool)),
311                 this, SLOT(on_sub_window_close_clicked()));
312
313         if (v->has_toolbar())
314                 dock_main->addToolBar(v->create_toolbar(dock_main));
315
316         if (v->minimum_width() > 0)
317                 dock->setMinimumSize(v->minimum_width(), 0);
318
319         return v;
320 }
321
322 shared_ptr<Session> MainWindow::add_session()
323 {
324         static int last_session_id = 1;
325         QString name = tr("Session %1").arg(last_session_id++);
326
327         shared_ptr<Session> session = make_shared<Session>(device_manager_, name);
328
329         connect(session.get(), SIGNAL(add_view(const QString&, views::ViewType, Session*)),
330                 this, SLOT(on_add_view(const QString&, views::ViewType, Session*)));
331         connect(session.get(), SIGNAL(name_changed()),
332                 this, SLOT(on_session_name_changed()));
333         session_state_mapper_.setMapping(session.get(), session.get());
334         connect(session.get(), SIGNAL(capture_state_changed(int)),
335                 &session_state_mapper_, SLOT(map()));
336
337         sessions_.push_back(session);
338
339         QMainWindow *window = new QMainWindow();
340         window->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
341         session_windows_[session] = window;
342
343         int index = session_selector_.addTab(window, name);
344         session_selector_.setCurrentIndex(index);
345         last_focused_session_ = session;
346
347         window->setDockNestingEnabled(true);
348
349         shared_ptr<views::ViewBase> main_view =
350                 add_view(name, views::ViewTypeTrace, *session);
351
352         return session;
353 }
354
355 void MainWindow::remove_session(shared_ptr<Session> session)
356 {
357         // Determine the height of the button before it collapses
358         int h = new_session_button_->height();
359
360         // Stop capture while the session still exists so that the UI can be
361         // updated in case we're currently running. If so, this will schedule a
362         // call to our on_capture_state_changed() slot for the next run of the
363         // event loop. We need to have this executed immediately or else it will
364         // be dismissed since the session object will be deleted by the time we
365         // leave this method and the event loop gets a chance to run again.
366         session->stop_capture();
367         QApplication::processEvents();
368
369         for (const shared_ptr<views::ViewBase>& view : session->views())
370                 remove_view(view);
371
372         QMainWindow *window = session_windows_.at(session);
373         session_selector_.removeTab(session_selector_.indexOf(window));
374
375         session_windows_.erase(session);
376
377         if (last_focused_session_ == session)
378                 last_focused_session_.reset();
379
380         // Remove the session from our list of sessions (which also destroys it)
381         sessions_.remove_if([&](shared_ptr<Session> s) {
382                 return s == session; });
383
384         if (sessions_.empty()) {
385                 // When there are no more tabs, the height of the QTabWidget
386                 // drops to zero. We must prevent this to keep the static
387                 // widgets visible
388                 for (QWidget *w : static_tab_widget_->findChildren<QWidget*>())  // clazy:exclude=range-loop
389                         w->setMinimumHeight(h);
390
391                 int margin = static_tab_widget_->layout()->contentsMargins().bottom();
392                 static_tab_widget_->setMinimumHeight(h + 2 * margin);
393                 session_selector_.setMinimumHeight(h + 2 * margin);
394
395                 // Update the window title if there is no view left to
396                 // generate focus change events
397                 setWindowTitle(WindowTitle);
398         }
399 }
400
401 void MainWindow::add_session_with_file(string open_file_name,
402         string open_file_format, string open_setup_file_name)
403 {
404         shared_ptr<Session> session = add_session();
405         session->load_init_file(open_file_name, open_file_format, open_setup_file_name);
406 }
407
408 void MainWindow::add_default_session()
409 {
410         // Only add the default session if there would be no session otherwise
411         if (sessions_.size() > 0)
412                 return;
413
414         shared_ptr<Session> session = add_session();
415
416         // Check the list of available devices. Prefer the one that was
417         // found with user supplied scan specs (if applicable). Then try
418         // one of the auto detected devices that are not the demo device.
419         // Pick demo in the absence of "genuine" hardware devices.
420         shared_ptr<devices::HardwareDevice> user_device, other_device, demo_device;
421         for (const shared_ptr<devices::HardwareDevice>& dev : device_manager_.devices()) {
422                 if (dev == device_manager_.user_spec_device()) {
423                         user_device = dev;
424                 } else if (dev->hardware_device()->driver()->name() == "demo") {
425                         demo_device = dev;
426                 } else {
427                         other_device = dev;
428                 }
429         }
430         if (user_device)
431                 session->select_device(user_device);
432         else if (other_device)
433                 session->select_device(other_device);
434         else
435                 session->select_device(demo_device);
436 }
437
438 void MainWindow::save_sessions()
439 {
440         QSettings settings;
441         int id = 0;
442
443         for (shared_ptr<Session>& session : sessions_) {
444                 // Ignore sessions using the demo device or no device at all
445                 if (session->device()) {
446                         shared_ptr<devices::HardwareDevice> device =
447                                 dynamic_pointer_cast< devices::HardwareDevice >
448                                 (session->device());
449
450                         if (device &&
451                                 device->hardware_device()->driver()->name() == "demo")
452                                 continue;
453
454                         settings.beginGroup("Session" + QString::number(id++));
455                         settings.remove("");  // Remove all keys in this group
456                         session->save_settings(settings);
457                         settings.endGroup();
458                 }
459         }
460
461         settings.setValue("sessions", id);
462 }
463
464 void MainWindow::restore_sessions()
465 {
466         QSettings settings;
467         int i, session_count;
468
469         session_count = settings.value("sessions", 0).toInt();
470
471         for (i = 0; i < session_count; i++) {
472                 settings.beginGroup("Session" + QString::number(i));
473                 shared_ptr<Session> session = add_session();
474                 session->restore_settings(settings);
475                 settings.endGroup();
476         }
477 }
478
479 void MainWindow::on_setting_changed(const QString &key, const QVariant &value)
480 {
481         if (key == GlobalSettings::Key_View_ColoredBG)
482                 on_settingViewColoredBg_changed(value);
483
484         if (key == GlobalSettings::Key_View_ShowSamplingPoints)
485                 on_settingViewShowSamplingPoints_changed(value);
486
487         if (key == GlobalSettings::Key_View_ShowAnalogMinorGrid)
488                 on_settingViewShowAnalogMinorGrid_changed(value);
489 }
490
491 void MainWindow::setup_ui()
492 {
493         setObjectName(QString::fromUtf8("MainWindow"));
494
495         setCentralWidget(&session_selector_);
496
497         // Set the window icon
498         QIcon icon;
499         icon.addFile(QString(":/icons/pulseview.png"));
500         setWindowIcon(icon);
501
502         view_sticky_scrolling_shortcut_ = new QShortcut(QKeySequence(Qt::Key_S), this, SLOT(on_view_sticky_scrolling_shortcut()));
503         view_sticky_scrolling_shortcut_->setAutoRepeat(false);
504
505         view_show_sampling_points_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Period), this, SLOT(on_view_show_sampling_points_shortcut()));
506         view_show_sampling_points_shortcut_->setAutoRepeat(false);
507
508         view_show_analog_minor_grid_shortcut_ = new QShortcut(QKeySequence(Qt::Key_G), this, SLOT(on_view_show_analog_minor_grid_shortcut()));
509         view_show_analog_minor_grid_shortcut_->setAutoRepeat(false);
510
511         view_colored_bg_shortcut_ = new QShortcut(QKeySequence(Qt::Key_B), this, SLOT(on_view_colored_bg_shortcut()));
512         view_colored_bg_shortcut_->setAutoRepeat(false);
513
514         zoom_in_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Plus), this, SLOT(on_zoom_in_shortcut_triggered()));
515         zoom_in_shortcut_->setAutoRepeat(false);
516
517         zoom_in_shortcut_2_ = new QShortcut(QKeySequence(Qt::Key_Up), this, SLOT(on_zoom_in_shortcut_triggered()));
518
519         zoom_out_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Minus), this, SLOT(on_zoom_out_shortcut_triggered()));
520         zoom_out_shortcut_->setAutoRepeat(false);
521
522         zoom_out_shortcut_2_ = new QShortcut(QKeySequence(Qt::Key_Down), this, SLOT(on_zoom_out_shortcut_triggered()));
523
524         home_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Home), this, SLOT(on_scroll_to_start_triggered()));
525         home_shortcut_->setAutoRepeat(false);
526
527         end_shortcut_ = new QShortcut(QKeySequence(Qt::Key_End), this, SLOT(on_scroll_to_end_triggered()));
528         end_shortcut_->setAutoRepeat(false);
529
530         // Set up the tab area
531         new_session_button_ = new QToolButton();
532         new_session_button_->setIcon(QIcon::fromTheme("document-new",
533                 QIcon(":/icons/document-new.png")));
534         new_session_button_->setToolTip(tr("Create New Session"));
535         new_session_button_->setAutoRaise(true);
536
537         run_stop_button_ = new QToolButton();
538         run_stop_button_->setAutoRaise(true);
539         run_stop_button_->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
540         run_stop_button_->setToolTip(tr("Start/Stop Acquisition"));
541
542         run_stop_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Space), run_stop_button_, SLOT(click()));
543         run_stop_shortcut_->setAutoRepeat(false);
544
545         settings_button_ = new QToolButton();
546         settings_button_->setIcon(QIcon::fromTheme("preferences-system",
547                 QIcon(":/icons/preferences-system.png")));
548         settings_button_->setToolTip(tr("Settings"));
549         settings_button_->setAutoRaise(true);
550
551         QFrame *separator1 = new QFrame();
552         separator1->setFrameStyle(QFrame::VLine | QFrame::Raised);
553         QFrame *separator2 = new QFrame();
554         separator2->setFrameStyle(QFrame::VLine | QFrame::Raised);
555
556         QHBoxLayout* layout = new QHBoxLayout();
557         layout->setContentsMargins(2, 2, 2, 2);
558         layout->addWidget(new_session_button_);
559         layout->addWidget(separator1);
560         layout->addWidget(run_stop_button_);
561         layout->addWidget(separator2);
562         layout->addWidget(settings_button_);
563
564         static_tab_widget_ = new QWidget();
565         static_tab_widget_->setLayout(layout);
566
567         session_selector_.setCornerWidget(static_tab_widget_, Qt::TopLeftCorner);
568         session_selector_.setTabsClosable(true);
569
570         close_application_shortcut_ = new QShortcut(QKeySequence(Qt::CTRL + Qt::Key_Q), this, SLOT(close()));
571         close_application_shortcut_->setAutoRepeat(false);
572
573         close_current_tab_shortcut_ = new QShortcut(QKeySequence(Qt::CTRL + Qt::Key_W), this, SLOT(on_close_current_tab()));
574
575         connect(new_session_button_, SIGNAL(clicked(bool)),
576                 this, SLOT(on_new_session_clicked()));
577         connect(run_stop_button_, SIGNAL(clicked(bool)),
578                 this, SLOT(on_run_stop_clicked()));
579         connect(&session_state_mapper_, SIGNAL(mapped(QObject*)),
580                 this, SLOT(on_capture_state_changed(QObject*)));
581         connect(settings_button_, SIGNAL(clicked(bool)),
582                 this, SLOT(on_settings_clicked()));
583
584         connect(&session_selector_, SIGNAL(tabCloseRequested(int)),
585                 this, SLOT(on_tab_close_requested(int)));
586         connect(&session_selector_, SIGNAL(currentChanged(int)),
587                 this, SLOT(on_tab_changed(int)));
588
589
590         connect(static_cast<QApplication *>(QCoreApplication::instance()),
591                 SIGNAL(focusChanged(QWidget*, QWidget*)),
592                 this, SLOT(on_focus_changed()));
593 }
594
595 void MainWindow::save_ui_settings()
596 {
597         QSettings settings;
598
599         settings.beginGroup("MainWindow");
600         settings.setValue("state", saveState());
601         settings.setValue("geometry", saveGeometry());
602         settings.endGroup();
603 }
604
605 void MainWindow::restore_ui_settings()
606 {
607         QSettings settings;
608
609         settings.beginGroup("MainWindow");
610
611         if (settings.contains("geometry")) {
612                 restoreGeometry(settings.value("geometry").toByteArray());
613                 restoreState(settings.value("state").toByteArray());
614         } else
615                 resize(1000, 720);
616
617         settings.endGroup();
618 }
619
620 void MainWindow::zoom_current_view(double steps)
621 {
622         shared_ptr<Session> session = get_tab_session(session_selector_.currentIndex());
623
624         if (!session)
625                 return;
626
627         shared_ptr<views::ViewBase> v = session.get()->main_view();
628         views::trace::View *tv =
629                 qobject_cast<views::trace::View*>(v.get());
630         tv->zoom(steps);
631 }
632
633 shared_ptr<Session> MainWindow::get_tab_session(int index) const
634 {
635         // Find the session that belongs to the tab's main window
636         for (auto& entry : session_windows_)
637                 if (entry.second == session_selector_.widget(index))
638                         return entry.first;
639
640         return nullptr;
641 }
642
643 void MainWindow::closeEvent(QCloseEvent *event)
644 {
645         bool data_saved = true;
646
647         for (auto& entry : session_windows_)
648                 if (!entry.first->data_saved())
649                         data_saved = false;
650
651         if (!data_saved && (QMessageBox::question(this, tr("Confirmation"),
652                 tr("There is unsaved data. Close anyway?"),
653                 QMessageBox::Yes | QMessageBox::No) == QMessageBox::No)) {
654                 event->ignore();
655         } else {
656                 save_ui_settings();
657                 save_sessions();
658                 event->accept();
659         }
660 }
661
662 QMenu* MainWindow::createPopupMenu()
663 {
664         return nullptr;
665 }
666
667 bool MainWindow::restoreState(const QByteArray &state, int version)
668 {
669         (void)state;
670         (void)version;
671
672         // Do nothing. We don't want Qt to handle this, or else it
673         // will try to restore all the dock widgets and create havoc.
674
675         return false;
676 }
677
678 void MainWindow::on_add_view(const QString &title, views::ViewType type,
679         Session *session)
680 {
681         // We get a pointer and need a reference
682         for (shared_ptr<Session>& s : sessions_)
683                 if (s.get() == session)
684                         add_view(title, type, *s);
685 }
686
687 void MainWindow::on_focus_changed()
688 {
689         shared_ptr<views::ViewBase> view = get_active_view();
690
691         if (view) {
692                 for (shared_ptr<Session> session : sessions_) {
693                         if (session->has_view(view)) {
694                                 if (session != last_focused_session_) {
695                                         // Activate correct tab if necessary
696                                         shared_ptr<Session> tab_session = get_tab_session(
697                                                 session_selector_.currentIndex());
698                                         if (tab_session != session)
699                                                 session_selector_.setCurrentWidget(
700                                                         session_windows_.at(session));
701
702                                         on_focused_session_changed(session);
703                                 }
704
705                                 break;
706                         }
707                 }
708         }
709
710         if (sessions_.empty())
711                 setWindowTitle(WindowTitle);
712 }
713
714 void MainWindow::on_focused_session_changed(shared_ptr<Session> session)
715 {
716         last_focused_session_ = session;
717
718         setWindowTitle(session->name() + " - " + WindowTitle);
719
720         // Update the state of the run/stop button, too
721         on_capture_state_changed(session.get());
722 }
723
724 void MainWindow::on_new_session_clicked()
725 {
726         add_session();
727 }
728
729 void MainWindow::on_run_stop_clicked()
730 {
731         shared_ptr<Session> session = last_focused_session_;
732
733         if (!session)
734                 return;
735
736         switch (session->get_capture_state()) {
737         case Session::Stopped:
738                 session->start_capture([&](QString message) {
739                         show_session_error("Capture failed", message); });
740                 break;
741         case Session::AwaitingTrigger:
742         case Session::Running:
743                 session->stop_capture();
744                 break;
745         }
746 }
747
748 void MainWindow::on_settings_clicked()
749 {
750         dialogs::Settings dlg(device_manager_);
751         dlg.exec();
752 }
753
754 void MainWindow::on_session_name_changed()
755 {
756         // Update the corresponding dock widget's name(s)
757         Session *session = qobject_cast<Session*>(QObject::sender());
758         assert(session);
759
760         for (const shared_ptr<views::ViewBase>& view : session->views()) {
761                 // Get the dock that contains the view
762                 for (auto& entry : view_docks_)
763                         if (entry.second == view) {
764                                 entry.first->setObjectName(session->name());
765                                 entry.first->setWindowTitle(session->name());
766                         }
767         }
768
769         // Update the tab widget by finding the main window and the tab from that
770         for (auto& entry : session_windows_)
771                 if (entry.first.get() == session) {
772                         QMainWindow *window = entry.second;
773                         const int index = session_selector_.indexOf(window);
774                         session_selector_.setTabText(index, session->name());
775                 }
776
777         // Refresh window title if the affected session has focus
778         if (session == last_focused_session_.get())
779                 setWindowTitle(session->name() + " - " + WindowTitle);
780 }
781
782 void MainWindow::on_capture_state_changed(QObject *obj)
783 {
784         Session *caller = qobject_cast<Session*>(obj);
785
786         // Ignore if caller is not the currently focused session
787         // unless there is only one session
788         if ((sessions_.size() > 1) && (caller != last_focused_session_.get()))
789                 return;
790
791         int state = caller->get_capture_state();
792
793         const QIcon *icons[] = {&icon_grey_, &icon_red_, &icon_green_};
794         run_stop_button_->setIcon(*icons[state]);
795         run_stop_button_->setText((state == pv::Session::Stopped) ?
796                 tr("Run") : tr("Stop"));
797 }
798
799 void MainWindow::on_new_view(Session *session)
800 {
801         // We get a pointer and need a reference
802         for (shared_ptr<Session>& s : sessions_)
803                 if (s.get() == session)
804                         add_view(session->name(), views::ViewTypeTrace, *s);
805 }
806
807 void MainWindow::on_view_close_clicked()
808 {
809         // Find the dock widget that contains the close button that was clicked
810         QObject *w = QObject::sender();
811         QDockWidget *dock = nullptr;
812
813         while (w) {
814             dock = qobject_cast<QDockWidget*>(w);
815             if (dock)
816                 break;
817             w = w->parent();
818         }
819
820         // Get the view contained in the dock widget
821         shared_ptr<views::ViewBase> view;
822
823         for (auto& entry : view_docks_)
824                 if (entry.first == dock)
825                         view = entry.second;
826
827         // Deregister the view
828         for (shared_ptr<Session> session : sessions_) {
829                 if (!session->has_view(view))
830                         continue;
831
832                 // Also destroy the entire session if its main view is closing...
833                 if (view == session->main_view()) {
834                         // ...but only if data is saved or the user confirms closing
835                         if (session->data_saved() || (QMessageBox::question(this, tr("Confirmation"),
836                                 tr("This session contains unsaved data. Close it anyway?"),
837                                 QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes))
838                                 remove_session(session);
839                         break;
840                 } else
841                         // All other views can be closed at any time as no data will be lost
842                         remove_view(view);
843         }
844 }
845
846 void MainWindow::on_tab_changed(int index)
847 {
848         shared_ptr<Session> session = get_tab_session(index);
849
850         if (session)
851                 on_focused_session_changed(session);
852 }
853
854 void MainWindow::on_tab_close_requested(int index)
855 {
856         shared_ptr<Session> session = get_tab_session(index);
857
858         if (!session)
859                 return;
860
861         if (session->data_saved() || (QMessageBox::question(this, tr("Confirmation"),
862                 tr("This session contains unsaved data. Close it anyway?"),
863                 QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes))
864                 remove_session(session);
865 }
866
867 void MainWindow::on_show_decoder_selector(Session *session)
868 {
869 #ifdef ENABLE_DECODE
870         // Close dock widget if it's already showing and return
871         for (auto entry : sub_windows_) {
872                 QDockWidget* dock = entry.first;
873                 if (dynamic_pointer_cast<subwindows::decoder_selector::SubWindow>(entry.second)) {
874                         sub_windows_.erase(dock);
875                         dock->close();
876                         return;
877                 }
878         }
879
880         // We get a pointer and need a reference
881         for (shared_ptr<Session> s : sessions_)
882                 if (s.get() == session)
883                         add_subwindow(subwindows::SubWindowTypeDecoderSelector, *s);
884 #endif
885 }
886
887 void MainWindow::on_sub_window_close_clicked()
888 {
889         // Find the dock widget that contains the close button that was clicked
890         QObject *w = QObject::sender();
891         QDockWidget *dock = nullptr;
892
893         while (w) {
894             dock = qobject_cast<QDockWidget*>(w);
895             if (dock)
896                 break;
897             w = w->parent();
898         }
899
900         sub_windows_.erase(dock);
901         dock->close();
902 }
903
904 void MainWindow::on_view_colored_bg_shortcut()
905 {
906         GlobalSettings settings;
907
908         bool state = settings.value(GlobalSettings::Key_View_ColoredBG).toBool();
909         settings.setValue(GlobalSettings::Key_View_ColoredBG, !state);
910 }
911
912 void MainWindow::on_view_sticky_scrolling_shortcut()
913 {
914         GlobalSettings settings;
915
916         bool state = settings.value(GlobalSettings::Key_View_StickyScrolling).toBool();
917         settings.setValue(GlobalSettings::Key_View_StickyScrolling, !state);
918 }
919
920 void MainWindow::on_view_show_sampling_points_shortcut()
921 {
922         GlobalSettings settings;
923
924         bool state = settings.value(GlobalSettings::Key_View_ShowSamplingPoints).toBool();
925         settings.setValue(GlobalSettings::Key_View_ShowSamplingPoints, !state);
926 }
927
928 void MainWindow::on_view_show_analog_minor_grid_shortcut()
929 {
930         GlobalSettings settings;
931
932         bool state = settings.value(GlobalSettings::Key_View_ShowAnalogMinorGrid).toBool();
933         settings.setValue(GlobalSettings::Key_View_ShowAnalogMinorGrid, !state);
934 }
935
936 void MainWindow::on_settingViewColoredBg_changed(const QVariant new_value)
937 {
938         bool state = new_value.toBool();
939
940         for (auto& entry : view_docks_) {
941                 shared_ptr<views::ViewBase> viewbase = entry.second;
942
943                 // Only trace views have this setting
944                 views::trace::View* view =
945                                 qobject_cast<views::trace::View*>(viewbase.get());
946                 if (view)
947                         view->enable_colored_bg(state);
948         }
949 }
950
951 void MainWindow::on_settingViewShowSamplingPoints_changed(const QVariant new_value)
952 {
953         bool state = new_value.toBool();
954
955         for (auto& entry : view_docks_) {
956                 shared_ptr<views::ViewBase> viewbase = entry.second;
957
958                 // Only trace views have this setting
959                 views::trace::View* view =
960                                 qobject_cast<views::trace::View*>(viewbase.get());
961                 if (view)
962                         view->enable_show_sampling_points(state);
963         }
964 }
965
966 void MainWindow::on_settingViewShowAnalogMinorGrid_changed(const QVariant new_value)
967 {
968         bool state = new_value.toBool();
969
970         for (auto& entry : view_docks_) {
971                 shared_ptr<views::ViewBase> viewbase = entry.second;
972
973                 // Only trace views have this setting
974                 views::trace::View* view =
975                                 qobject_cast<views::trace::View*>(viewbase.get());
976                 if (view)
977                         view->enable_show_analog_minor_grid(state);
978         }
979 }
980
981 void MainWindow::on_zoom_out_shortcut_triggered()
982 {
983         zoom_current_view(-1);
984 }
985
986 void MainWindow::on_zoom_in_shortcut_triggered()
987 {
988         zoom_current_view(1);
989 }
990
991 void MainWindow::on_scroll_to_start_triggered()
992 {
993         scroll_to_start_or_end(true);
994 }
995
996 void MainWindow::on_scroll_to_end_triggered()
997 {
998         scroll_to_start_or_end(false);
999 }
1000
1001 void MainWindow::scroll_to_start_or_end(bool start)
1002 {
1003         shared_ptr<Session> session = get_tab_session(session_selector_.currentIndex());
1004
1005         if (!session)
1006                 return;
1007
1008         shared_ptr<views::ViewBase> v = session.get()->main_view();
1009         views::trace::View *tv =
1010                 qobject_cast<views::trace::View*>(v.get());
1011         tv->set_h_offset(start ? 0 : tv->get_h_scrollbar_maximum());
1012 }
1013
1014 void MainWindow::on_close_current_tab()
1015 {
1016         int tab = session_selector_.currentIndex();
1017
1018         on_tab_close_requested(tab);
1019 }
1020
1021 } // namespace pv