]> sigrok.org Git - pulseview.git/blobdiff - pv/mainwindow.cpp
MainWindow: Update session tab when focusing a different view
[pulseview.git] / pv / mainwindow.cpp
index 8787073d5875e1d82c64d826eb845b26e09c9b3c..39ce7ed6f4b0034ad3b0adfc7241cdcd76340779 100644 (file)
@@ -31,6 +31,7 @@
 #include <QApplication>
 #include <QCloseEvent>
 #include <QDockWidget>
+#include <QHBoxLayout>
 #include <QSettings>
 #include <QWidget>
 
@@ -62,11 +63,14 @@ class ViewItem;
 
 using toolbars::MainBar;
 
+const QString MainWindow::WindowTitle = tr("PulseView");
+
 MainWindow::MainWindow(DeviceManager &device_manager,
        string open_file_name, string open_file_format,
        QWidget *parent) :
        QMainWindow(parent),
        device_manager_(device_manager),
+       session_selector_(this),
        action_view_sticky_scrolling_(new QAction(this)),
        action_view_coloured_bg_(new QAction(this)),
        action_about_(new QAction(this))
@@ -105,23 +109,8 @@ MainWindow::MainWindow(DeviceManager &device_manager,
 
 MainWindow::~MainWindow()
 {
-       for (auto entry : view_docks_) {
-
-               const std::shared_ptr<QDockWidget> dock = entry.first;
-
-               // Remove view from the dock widget's QMainWindow
-               QMainWindow *dock_main = dynamic_cast<QMainWindow*>(dock->widget());
-               dock_main->setCentralWidget(0);
-
-               // Remove the QMainWindow
-               dock->setWidget(0);
-
-               const std::shared_ptr<pv::view::View> view = entry.second;
-
-               for (shared_ptr<Session> session : sessions_)
-                       if (session->has_view(view))
-                               session->deregister_view(view);
-       }
+       while (!sessions_.empty())
+               remove_session(sessions_.front());
 }
 
 QAction* MainWindow::action_view_sticky_scrolling() const
@@ -139,7 +128,7 @@ QAction* MainWindow::action_about() const
        return action_about_;
 }
 
-shared_ptr<pv::view::View> MainWindow::get_active_view() const
+shared_ptr<views::ViewBase> MainWindow::get_active_view() const
 {
        // If there's only one view, use it...
        if (view_docks_.size() == 1)
@@ -158,27 +147,33 @@ shared_ptr<pv::view::View> MainWindow::get_active_view() const
 
        // Get the view contained in the dock widget
        for (auto entry : view_docks_)
-               if (entry.first.get() == dock)
+               if (entry.first == dock)
                        return entry.second;
 
-       return shared_ptr<pv::view::View>();
+       return nullptr;
 }
 
-shared_ptr<pv::view::View> MainWindow::add_view(const QString &title,
-       view::ViewType type, Session &session)
+shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
+       views::ViewType type, Session &session)
 {
-       shared_ptr<pv::view::View> v;
+       QMainWindow *main_window;
+       for (auto entry : session_windows_)
+               if (entry.first.get() == &session)
+                       main_window = entry.second;
 
-       if (type == pv::view::TraceView) {
-               shared_ptr<QDockWidget> dock = make_shared<QDockWidget>(title, this);
+       assert(main_window);
+
+       if (type == views::ViewTypeTrace) {
+               QDockWidget* dock = new QDockWidget(title, main_window);
                dock->setObjectName(title);
-               addDockWidget(Qt::TopDockWidgetArea, dock.get());
+               main_window->addDockWidget(Qt::TopDockWidgetArea, dock);
 
                // Insert a QMainWindow into the dock widget to allow for a tool bar
-               QMainWindow *dock_main = new QMainWindow(dock.get());
+               QMainWindow *dock_main = new QMainWindow(dock);
                dock_main->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
 
-               v = make_shared<pv::view::View>(session, dock_main);
+               shared_ptr<views::TraceView::View> v =
+                       make_shared<views::TraceView::View>(session, dock_main);
                view_docks_[dock] = v;
                session.register_view(v);
 
@@ -195,8 +190,9 @@ shared_ptr<pv::view::View> MainWindow::add_view(const QString &title,
                connect(close_btn, SIGNAL(clicked(bool)),
                        this, SLOT(on_view_close_clicked()));
 
-               if (type == view::TraceView) {
-                       connect(&session, SIGNAL(trigger_event(util::Timestamp)), v.get(),
+               if (type == views::ViewTypeTrace) {
+                       connect(&session, SIGNAL(trigger_event(util::Timestamp)),
+                               qobject_cast<views::ViewBase*>(v.get()),
                                SLOT(trigger_event(util::Timestamp)));
 
                        v->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
@@ -208,8 +204,6 @@ shared_ptr<pv::view::View> MainWindow::add_view(const QString &title,
                                dock_main->addToolBar(main_bar.get());
                                session.set_main_bar(main_bar);
 
-                               connect(main_bar.get(), SIGNAL(new_session()),
-                                       this, SLOT(on_new_session()));
                                connect(main_bar.get(), SIGNAL(new_view(Session*)),
                                        this, SLOT(on_new_view(Session*)));
                        }
@@ -218,46 +212,80 @@ shared_ptr<pv::view::View> MainWindow::add_view(const QString &title,
                        connect(v.get(), SIGNAL(always_zoom_to_fit_changed(bool)),
                                main_bar.get(), SLOT(on_always_zoom_to_fit_changed(bool)));
                }
+
+               return v;
        }
 
-       return v;
+       return nullptr;
 }
 
 shared_ptr<Session> MainWindow::add_session()
 {
-       int id = sessions_.size();
-       QString name = tr("Untitled-%1").arg(id + 1);
+       static int last_session_id = 1;
+       QString name = tr("Untitled-%1").arg(last_session_id++);
 
        shared_ptr<Session> session = make_shared<Session>(device_manager_, name);
 
-       connect(session.get(), SIGNAL(add_view(const QString&, view::ViewType, Session*)),
-               this, SLOT(on_add_view(const QString&, view::ViewType, Session*)));
+       connect(session.get(), SIGNAL(add_view(const QString&, views::ViewType, Session*)),
+               this, SLOT(on_add_view(const QString&, views::ViewType, Session*)));
+       connect(session.get(), SIGNAL(name_changed()),
+               this, SLOT(on_session_name_changed()));
 
        sessions_.push_back(session);
 
-       shared_ptr<view::View> main_view =
-               add_view(name, pv::view::TraceView, *session);
+       QMainWindow *window = new QMainWindow();
+       window->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
+       session_windows_[session] = window;
+       session_selector_.addTab(window, name);
+
+       window->setDockNestingEnabled(true);
+
+       shared_ptr<views::ViewBase> main_view =
+               add_view(name, views::ViewTypeTrace, *session);
 
        return session;
 }
 
 void MainWindow::remove_session(shared_ptr<Session> session)
 {
-       for (shared_ptr<view::View> view : session->views()) {
-               // Find the dock the view is contained in and close it
+       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)
-                               entry.first->close();
+                       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
+                               view_docks_.erase(entry.first);
+                       }
        }
 
+       QMainWindow *window = session_windows_.at(session);
+       session_selector_.removeTab(session_selector_.indexOf(window));
+
+       session_windows_.erase(session);
+
        sessions_.remove_if([&](shared_ptr<Session> s) {
                return s == session; });
+
+       // Update the window title if there is no view left to
+       // generate focus change events
+       if (sessions_.empty())
+               setWindowTitle(WindowTitle);
 }
 
 void MainWindow::setup_ui()
 {
        setObjectName(QString::fromUtf8("MainWindow"));
 
+       setCentralWidget(&session_selector_);
+
        // Set the window icon
        QIcon icon;
        icon.addFile(QString(":/icons/sigrok-logo-notext.png"));
@@ -280,10 +308,35 @@ void MainWindow::setup_ui()
        action_about_->setObjectName(QString::fromUtf8("actionAbout"));
        action_about_->setText(tr("&About..."));
 
-       setDockNestingEnabled(true);
+       // Set up the tab area
+       new_session_button_ = new QToolButton();
+       new_session_button_->setIcon(QIcon::fromTheme("document-new",
+               QIcon(":/icons/document-new.png")));
+       new_session_button_->setAutoRaise(true);
+
+       QHBoxLayout* layout = new QHBoxLayout();
+       layout->setContentsMargins(2, 2, 2, 2);
+       layout->addWidget(new_session_button_);
+
+       QWidget* static_tab_widget_ = new QWidget();
+       static_tab_widget_->setLayout(layout);
+
+       session_selector_.setCornerWidget(static_tab_widget_, Qt::TopLeftCorner);
 
-       // Set the title
-       setWindowTitle(tr("PulseView"));
+       session_selector_.setTabsClosable(true);
+
+       connect(new_session_button_, SIGNAL(clicked(bool)),
+               this, SLOT(on_new_session_clicked()));
+
+       connect(&session_selector_, SIGNAL(tabCloseRequested(int)),
+               this, SLOT(on_tab_close_requested(int)));
+       connect(&session_selector_, SIGNAL(currentChanged(int)),
+               this, SLOT(on_tab_changed(int)));
+
+
+       connect(static_cast<QApplication *>(QCoreApplication::instance()),
+               SIGNAL(focusChanged(QWidget*, QWidget*)),
+               this, SLOT(on_focus_changed()));
 }
 
 void MainWindow::save_ui_settings()
@@ -297,7 +350,7 @@ void MainWindow::save_ui_settings()
        settings.endGroup();
 
        for (shared_ptr<Session> session : sessions_) {
-               // Ignore sessions using the demo device
+               // Ignore sessions using the demo device or no device at all
                if (session->device()) {
                        shared_ptr<devices::HardwareDevice> device =
                                dynamic_pointer_cast< devices::HardwareDevice >
@@ -306,12 +359,12 @@ void MainWindow::save_ui_settings()
                        if (device &&
                                device->hardware_device()->driver()->name() == "demo")
                                continue;
-               }
 
-               settings.beginGroup("Session" + QString::number(id++));
-               settings.remove("");  // Remove all keys in this group
-               session->save_settings(settings);
-               settings.endGroup();
+                       settings.beginGroup("Session" + QString::number(id++));
+                       settings.remove("");  // Remove all keys in this group
+                       session->save_settings(settings);
+                       settings.endGroup();
+               }
        }
 
        settings.setValue("sessions", id);
@@ -342,6 +395,16 @@ void MainWindow::restore_ui_settings()
        }
 }
 
+std::shared_ptr<Session> MainWindow::get_tab_session(int index) const
+{
+       // Find the session that belongs to the tab's main window
+       for (auto entry : session_windows_)
+               if (entry.second == session_selector_.widget(index))
+                       return entry.first;
+
+       return nullptr;
+}
+
 void MainWindow::closeEvent(QCloseEvent *event)
 {
        save_ui_settings();
@@ -364,7 +427,7 @@ bool MainWindow::restoreState(const QByteArray &state, int version)
        return false;
 }
 
-void MainWindow::on_add_view(const QString &title, view::ViewType type,
+void MainWindow::on_add_view(const QString &title, views::ViewType type,
        Session *session)
 {
        // We get a pointer and need a reference
@@ -373,17 +436,74 @@ void MainWindow::on_add_view(const QString &title, view::ViewType type,
                        add_view(title, type, *s);
 }
 
-void MainWindow::on_new_session()
+void MainWindow::on_focus_changed()
+{
+       static shared_ptr<Session> prev_session;
+
+       shared_ptr<views::ViewBase> view = get_active_view();
+
+       if (view) {
+               for (shared_ptr<Session> session : sessions_) {
+                       if (session->has_view(view)) {
+                               if (session != prev_session) {
+                                       // Activate correct tab if necessary
+                                       shared_ptr<Session> tab_session = get_tab_session(
+                                               session_selector_.currentIndex());
+                                       if (tab_session != session)
+                                               session_selector_.setCurrentWidget(
+                                                       session_windows_.at(session));
+
+                                       on_focused_session_changed(session);
+                               }
+
+                               prev_session = session;
+                               break;
+                       }
+               }
+       }
+
+       if (sessions_.empty())
+               setWindowTitle(WindowTitle);
+}
+
+void MainWindow::on_focused_session_changed(shared_ptr<Session> session)
+{
+       setWindowTitle(session->name() + " - " + WindowTitle);
+}
+
+void MainWindow::on_new_session_clicked()
 {
        add_session();
 }
 
+void MainWindow::on_session_name_changed()
+{
+       // Update the corresponding dock widget's name(s)
+       Session *session = qobject_cast<Session*>(QObject::sender());
+       assert(session);
+
+       for (shared_ptr<views::ViewBase> view : session->views()) {
+               // Get the dock that contains the view
+               for (auto entry : view_docks_)
+                       if (entry.second == view) {
+                               entry.first->setObjectName(session->name());
+                               entry.first->setWindowTitle(session->name());
+                       }
+       }
+
+       // Refresh window title if the affected session has focus
+       shared_ptr<views::ViewBase> view = get_active_view();
+
+       if (view && session->has_view(view))
+               setWindowTitle(session->name() + " - " + WindowTitle);
+}
+
 void MainWindow::on_new_view(Session *session)
 {
        // We get a pointer and need a reference
        for (std::shared_ptr<Session> s : sessions_)
                if (s.get() == session)
-                       add_view(session->name(), pv::view::TraceView, *s);
+                       add_view(session->name(), views::ViewTypeTrace, *s);
 }
 
 void MainWindow::on_view_close_clicked()
@@ -400,10 +520,10 @@ void MainWindow::on_view_close_clicked()
        }
 
        // Get the view contained in the dock widget
-       shared_ptr<view::View> view;
+       shared_ptr<views::ViewBase> view;
 
        for (auto entry : view_docks_)
-               if (entry.first.get() == dock)
+               if (entry.first == dock)
                        view = entry.second;
 
        // Deregister the view
@@ -420,16 +540,38 @@ void MainWindow::on_view_close_clicked()
        }
 }
 
+void MainWindow::on_tab_changed(int index)
+{
+       shared_ptr<Session> session = get_tab_session(index);
+
+       if (session)
+               on_focused_session_changed(session);
+}
+
+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)
+               remove_session(session);
+}
+
 void MainWindow::on_actionViewStickyScrolling_triggered()
 {
-       shared_ptr<pv::view::View> view = get_active_view();
+       shared_ptr<views::ViewBase> viewbase = get_active_view();
+       views::TraceView::View* view =
+               qobject_cast<views::TraceView::View*>(viewbase.get());
        if (view)
                view->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
 }
 
 void MainWindow::on_actionViewColouredBg_triggered()
 {
-       shared_ptr<pv::view::View> view = get_active_view();
+       shared_ptr<views::ViewBase> viewbase = get_active_view();
+       views::TraceView::View* view =
+                       qobject_cast<views::TraceView::View*>(viewbase.get());
        if (view)
                view->enable_coloured_bg(action_view_coloured_bg_->isChecked());
 }