]> sigrok.org Git - pulseview.git/blobdiff - pv/mainwindow.cpp
Added Ctrl-Q and Ctrl-W shortcuts.
[pulseview.git] / pv / mainwindow.cpp
index e1073144f7f97ca931b4733b27eead0dad3afaa0..662d68af2f29bdba5ceff5f68931855d18708026 100644 (file)
@@ -34,6 +34,7 @@
 #include <QMessageBox>
 #include <QSettings>
 #include <QWidget>
+#include <QShortcut>
 
 #include "mainwindow.hpp"
 
@@ -81,6 +82,7 @@ MainWindow::MainWindow(DeviceManager &device_manager,
        icon_grey_(":/icons/status-grey.svg")
 {
        qRegisterMetaType<util::Timestamp>("util::Timestamp");
+       qRegisterMetaType<uint64_t>("uint64_t");
 
        setup_ui();
        restore_ui_settings();
@@ -213,6 +215,12 @@ shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
                                        this, SLOT(on_new_view(Session*)));
 
                                main_bar->action_view_show_cursors()->setChecked(v->cursors_shown());
+
+                               /* For the main view we need to prevent the dock widget from
+                                * closing itself when its close button is clicked. This is
+                                * so we can confirm with the user first. Regular views don't
+                                * need this */
+                               close_btn->disconnect(SIGNAL(clicked()), dock, SLOT(close()));
                        } else {
                                /* Additional view, create a standard bar */
                                pv::views::trace::StandardBar *standard_bar =
@@ -229,6 +237,34 @@ shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
        return nullptr;
 }
 
+void MainWindow::remove_view(shared_ptr<views::ViewBase> view)
+{
+       for (shared_ptr<Session> session : sessions_) {
+               if (!session->has_view(view))
+                       continue;
+
+               // Find the dock the view is contained in and remove it
+               for (auto entry : view_docks_)
+                       if (entry.second == view) {
+                               // Remove the view from the session
+                               session->deregister_view(view);
+
+                               // Remove the view from its parent; otherwise, Qt will
+                               // call deleteLater() on it, which causes a double free
+                               // since the shared_ptr in view_docks_ doesn't know
+                               // that Qt keeps a pointer to the view around
+                               view->setParent(0);
+
+                               // Delete the view's dock widget and all widgets inside it
+                               entry.first->deleteLater();
+
+                               // Remove the dock widget from the list and stop iterating
+                               view_docks_.erase(entry.first);
+                               break;
+                       }
+       }
+}
+
 shared_ptr<Session> MainWindow::add_session()
 {
        static int last_session_id = 1;
@@ -266,24 +302,8 @@ void MainWindow::remove_session(shared_ptr<Session> session)
 {
        int h = new_session_button_->height();
 
-       for (shared_ptr<views::ViewBase> view : session->views()) {
-               // Find the dock the view is contained in and remove it
-               for (auto entry : view_docks_)
-                       if (entry.second == view) {
-                               // Remove the view from the session
-                               session->deregister_view(view);
-
-                               // Remove the view from its parent; otherwise, Qt will
-                               // call deleteLater() on it, which causes a double free
-                               // since the shared_ptr in view_docks_ doesn't know
-                               // that Qt keeps a pointer to the view around
-                               entry.second->setParent(0);
-
-                               // Remove this entry from the container and stop iterating.
-                               view_docks_.erase(entry.first);
-                               break;
-                       }
-       }
+       for (shared_ptr<views::ViewBase> view : session->views())
+               remove_view(view);
 
        QMainWindow *window = session_windows_.at(session);
        session_selector_.removeTab(session_selector_.indexOf(window));
@@ -351,9 +371,11 @@ void MainWindow::setup_ui()
        run_stop_button_ = new QToolButton();
        run_stop_button_->setAutoRaise(true);
        run_stop_button_->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
-       run_stop_button_->setShortcut(QKeySequence(Qt::Key_Space));
        run_stop_button_->setToolTip(tr("Start/Stop Acquisition"));
 
+       run_stop_shortcut_ = new QShortcut(QKeySequence(Qt::Key_Space), run_stop_button_, SLOT(click()));
+       run_stop_shortcut_->setAutoRepeat(false);
+
        settings_button_ = new QToolButton();
        settings_button_->setIcon(QIcon::fromTheme("configure",
                QIcon(":/icons/configure.png")));
@@ -379,6 +401,11 @@ void MainWindow::setup_ui()
        session_selector_.setCornerWidget(static_tab_widget_, Qt::TopLeftCorner);
        session_selector_.setTabsClosable(true);
 
+       close_application_shortcut_ = new QShortcut(QKeySequence(Qt::CTRL + Qt::Key_Q), this, SLOT(close()));
+       close_application_shortcut_->setAutoRepeat(false);
+
+       close_current_tab_shortcut_ = new QShortcut(QKeySequence(Qt::CTRL + Qt::Key_W), this, SLOT(on_close_current_tab()));
+
        connect(new_session_button_, SIGNAL(clicked(bool)),
                this, SLOT(on_new_session_clicked()));
        connect(run_stop_button_, SIGNAL(clicked(bool)),
@@ -465,8 +492,20 @@ std::shared_ptr<Session> MainWindow::get_tab_session(int index) const
 
 void MainWindow::closeEvent(QCloseEvent *event)
 {
-       save_ui_settings();
-       event->accept();
+       bool data_saved = true;
+
+       for (auto entry : session_windows_)
+               if (!entry.first->data_saved())
+                       data_saved = false;
+
+       if (!data_saved && (QMessageBox::question(this, tr("Confirmation"),
+               tr("There is unsaved data. Close anyway?"),
+               QMessageBox::Yes | QMessageBox::No) == QMessageBox::No)) {
+               event->ignore();
+       } else {
+               save_ui_settings();
+               event->accept();
+       }
 }
 
 QMenu* MainWindow::createPopupMenu()
@@ -650,12 +689,17 @@ void MainWindow::on_view_close_clicked()
                if (!session->has_view(view))
                        continue;
 
-               // Also destroy the entire session if its main view is closing
+               // Also destroy the entire session if its main view is closing...
                if (view == session->main_view()) {
-                       remove_session(session);
+                       // ...but only if data is saved or the user confirms closing
+                       if (session->data_saved() || (QMessageBox::question(this, tr("Confirmation"),
+                               tr("This session contains unsaved data. Close it anyway?"),
+                               QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes))
+                               remove_session(session);
                        break;
                } else
-                       session->deregister_view(view);
+                       // All other views can be closed at any time as no data will be lost
+                       remove_view(view);
        }
 }
 
@@ -669,11 +713,13 @@ void MainWindow::on_tab_changed(int index)
 
 void MainWindow::on_tab_close_requested(int index)
 {
-       // TODO Ask user if this is intended in case data is unsaved
-
        shared_ptr<Session> session = get_tab_session(index);
 
-       if (session)
+       assert(session);
+
+       if (session->data_saved() || (QMessageBox::question(this, tr("Confirmation"),
+               tr("This session contains unsaved data. Close it anyway?"),
+               QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes))
                remove_session(session);
 }
 
@@ -701,4 +747,11 @@ void MainWindow::on_actionAbout_triggered()
        dlg.exec();
 }
 
+void MainWindow::on_close_current_tab()
+{
+       int tab = session_selector_.currentIndex();
+
+       on_tab_close_requested(tab);
+}
+
 } // namespace pv