2 * This file is part of the PulseView project.
4 * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
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.
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.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
24 #include <libsigrokdecode/libsigrokdecode.h>
31 #include <QApplication>
32 #include <QCloseEvent>
33 #include <QDockWidget>
34 #include <QHBoxLayout>
38 #include "mainwindow.hpp"
40 #include "devicemanager.hpp"
42 #include "devices/hardwaredevice.hpp"
43 #include "dialogs/about.hpp"
44 #include "toolbars/mainbar.hpp"
45 #include "view/view.hpp"
49 #include <libsigrokcxx/libsigrokcxx.hpp>
51 using std::dynamic_pointer_cast;
53 using std::make_shared;
55 using std::shared_ptr;
64 using toolbars::MainBar;
66 const QString MainWindow::WindowTitle = tr("PulseView");
68 MainWindow::MainWindow(DeviceManager &device_manager,
69 string open_file_name, string open_file_format,
72 device_manager_(device_manager),
73 session_selector_(this),
74 action_view_sticky_scrolling_(new QAction(this)),
75 action_view_coloured_bg_(new QAction(this)),
76 action_about_(new QAction(this))
78 qRegisterMetaType<util::Timestamp>("util::Timestamp");
81 restore_ui_settings();
83 if (!open_file_name.empty()) {
84 shared_ptr<Session> session = add_session();
85 session->main_bar()->load_init_file(open_file_name, open_file_format);
88 // Add empty default session if there aren't any sessions
89 if (sessions_.size() == 0) {
90 shared_ptr<Session> session = add_session();
92 map<string, string> dev_info;
93 shared_ptr<devices::HardwareDevice> other_device, demo_device;
95 // Use any available device that's not demo
96 for (shared_ptr<devices::HardwareDevice> dev : device_manager_.devices()) {
97 if (dev->hardware_device()->driver()->name() == "demo") {
104 // ...and if there isn't any, just use demo then
105 session->main_bar()->select_device(other_device ?
106 other_device : demo_device);
110 MainWindow::~MainWindow()
112 while (!sessions_.empty())
113 remove_session(sessions_.front());
116 QAction* MainWindow::action_view_sticky_scrolling() const
118 return action_view_sticky_scrolling_;
121 QAction* MainWindow::action_view_coloured_bg() const
123 return action_view_coloured_bg_;
126 QAction* MainWindow::action_about() const
128 return action_about_;
131 shared_ptr<views::ViewBase> MainWindow::get_active_view() const
133 // If there's only one view, use it...
134 if (view_docks_.size() == 1)
135 return view_docks_.begin()->second;
137 // ...otherwise find the dock widget the widget with focus is contained in
138 QObject *w = QApplication::focusWidget();
139 QDockWidget *dock = 0;
142 dock = qobject_cast<QDockWidget*>(w);
148 // Get the view contained in the dock widget
149 for (auto entry : view_docks_)
150 if (entry.first == dock)
156 shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
157 views::ViewType type, Session &session)
159 QMainWindow *main_window;
160 for (auto entry : session_windows_)
161 if (entry.first.get() == &session)
162 main_window = entry.second;
166 if (type == views::ViewTypeTrace) {
167 QDockWidget* dock = new QDockWidget(title, main_window);
168 dock->setObjectName(title);
169 main_window->addDockWidget(Qt::TopDockWidgetArea, dock);
171 // Insert a QMainWindow into the dock widget to allow for a tool bar
172 QMainWindow *dock_main = new QMainWindow(dock);
173 dock_main->setWindowFlags(Qt::Widget); // Remove Qt::Window flag
175 shared_ptr<views::TraceView::View> v =
176 make_shared<views::TraceView::View>(session, dock_main);
177 view_docks_[dock] = v;
178 session.register_view(v);
180 dock_main->setCentralWidget(v.get());
181 dock->setWidget(dock_main);
183 dock->setFeatures(QDockWidget::DockWidgetMovable |
184 QDockWidget::DockWidgetFloatable | QDockWidget::DockWidgetClosable);
186 QAbstractButton *close_btn =
187 dock->findChildren<QAbstractButton*>
188 ("qt_dockwidget_closebutton").front();
190 connect(close_btn, SIGNAL(clicked(bool)),
191 this, SLOT(on_view_close_clicked()));
193 if (type == views::ViewTypeTrace) {
194 connect(&session, SIGNAL(trigger_event(util::Timestamp)),
195 qobject_cast<views::ViewBase*>(v.get()),
196 SLOT(trigger_event(util::Timestamp)));
198 v->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
199 v->enable_coloured_bg(action_view_coloured_bg_->isChecked());
201 shared_ptr<MainBar> main_bar = session.main_bar();
203 main_bar = make_shared<MainBar>(session, *this);
204 dock_main->addToolBar(main_bar.get());
205 session.set_main_bar(main_bar);
207 connect(main_bar.get(), SIGNAL(new_view(Session*)),
208 this, SLOT(on_new_view(Session*)));
210 main_bar->action_view_show_cursors()->setChecked(v->cursors_shown());
212 connect(v.get(), SIGNAL(always_zoom_to_fit_changed(bool)),
213 main_bar.get(), SLOT(on_always_zoom_to_fit_changed(bool)));
222 shared_ptr<Session> MainWindow::add_session()
224 static int last_session_id = 1;
225 QString name = tr("Untitled-%1").arg(last_session_id++);
227 shared_ptr<Session> session = make_shared<Session>(device_manager_, name);
229 connect(session.get(), SIGNAL(add_view(const QString&, views::ViewType, Session*)),
230 this, SLOT(on_add_view(const QString&, views::ViewType, Session*)));
231 connect(session.get(), SIGNAL(name_changed()),
232 this, SLOT(on_session_name_changed()));
234 sessions_.push_back(session);
236 QMainWindow *window = new QMainWindow();
237 window->setWindowFlags(Qt::Widget); // Remove Qt::Window flag
238 session_windows_[session] = window;
239 session_selector_.addTab(window, name);
241 window->setDockNestingEnabled(true);
243 shared_ptr<views::ViewBase> main_view =
244 add_view(name, views::ViewTypeTrace, *session);
249 void MainWindow::remove_session(shared_ptr<Session> session)
251 int h = new_session_button_->height();
253 for (shared_ptr<views::ViewBase> view : session->views()) {
254 // Find the dock the view is contained in and remove it
255 for (auto entry : view_docks_)
256 if (entry.second == view) {
257 // Remove the view from the session
258 session->deregister_view(view);
260 // Remove the view from its parent; otherwise, Qt will
261 // call deleteLater() on it, which causes a double free
262 // since the shared_ptr in view_docks_ doesn't know
263 // that Qt keeps a pointer to the view around
264 entry.second->setParent(0);
266 // Remove this entry from the container
267 view_docks_.erase(entry.first);
271 QMainWindow *window = session_windows_.at(session);
272 session_selector_.removeTab(session_selector_.indexOf(window));
274 session_windows_.erase(session);
276 sessions_.remove_if([&](shared_ptr<Session> s) {
277 return s == session; });
279 if (sessions_.empty()) {
280 // When there are no more tabs, the height of the QTabWidget
281 // drops to zero. We must prevent this to keep the static
283 for (QWidget *w : static_tab_widget_->findChildren<QWidget*>())
284 w->setMinimumHeight(h);
286 int margin = static_tab_widget_->layout()->contentsMargins().bottom();
287 static_tab_widget_->setMinimumHeight(h + 2 * margin);
288 session_selector_.setMinimumHeight(h + 2 * margin);
290 // Update the window title if there is no view left to
291 // generate focus change events
292 setWindowTitle(WindowTitle);
296 void MainWindow::setup_ui()
298 setObjectName(QString::fromUtf8("MainWindow"));
300 setCentralWidget(&session_selector_);
302 // Set the window icon
304 icon.addFile(QString(":/icons/sigrok-logo-notext.png"));
307 action_view_sticky_scrolling_->setCheckable(true);
308 action_view_sticky_scrolling_->setChecked(true);
309 action_view_sticky_scrolling_->setShortcut(QKeySequence(Qt::Key_S));
310 action_view_sticky_scrolling_->setObjectName(
311 QString::fromUtf8("actionViewStickyScrolling"));
312 action_view_sticky_scrolling_->setText(tr("&Sticky Scrolling"));
314 action_view_coloured_bg_->setCheckable(true);
315 action_view_coloured_bg_->setChecked(true);
316 action_view_coloured_bg_->setShortcut(QKeySequence(Qt::Key_B));
317 action_view_coloured_bg_->setObjectName(
318 QString::fromUtf8("actionViewColouredBg"));
319 action_view_coloured_bg_->setText(tr("Use &coloured backgrounds"));
321 action_about_->setObjectName(QString::fromUtf8("actionAbout"));
322 action_about_->setText(tr("&About..."));
324 // Set up the tab area
325 new_session_button_ = new QToolButton();
326 new_session_button_->setIcon(QIcon::fromTheme("document-new",
327 QIcon(":/icons/document-new.png")));
328 new_session_button_->setAutoRaise(true);
330 QHBoxLayout* layout = new QHBoxLayout();
331 layout->setContentsMargins(2, 2, 2, 2);
332 layout->addWidget(new_session_button_);
334 static_tab_widget_ = new QWidget();
335 static_tab_widget_->setLayout(layout);
337 session_selector_.setCornerWidget(static_tab_widget_, Qt::TopLeftCorner);
338 session_selector_.setTabsClosable(true);
340 connect(new_session_button_, SIGNAL(clicked(bool)),
341 this, SLOT(on_new_session_clicked()));
343 connect(&session_selector_, SIGNAL(tabCloseRequested(int)),
344 this, SLOT(on_tab_close_requested(int)));
345 connect(&session_selector_, SIGNAL(currentChanged(int)),
346 this, SLOT(on_tab_changed(int)));
349 connect(static_cast<QApplication *>(QCoreApplication::instance()),
350 SIGNAL(focusChanged(QWidget*, QWidget*)),
351 this, SLOT(on_focus_changed()));
354 void MainWindow::save_ui_settings()
359 settings.beginGroup("MainWindow");
360 settings.setValue("state", saveState());
361 settings.setValue("geometry", saveGeometry());
364 for (shared_ptr<Session> session : sessions_) {
365 // Ignore sessions using the demo device or no device at all
366 if (session->device()) {
367 shared_ptr<devices::HardwareDevice> device =
368 dynamic_pointer_cast< devices::HardwareDevice >
372 device->hardware_device()->driver()->name() == "demo")
375 settings.beginGroup("Session" + QString::number(id++));
376 settings.remove(""); // Remove all keys in this group
377 session->save_settings(settings);
382 settings.setValue("sessions", id);
385 void MainWindow::restore_ui_settings()
388 int i, session_count;
390 settings.beginGroup("MainWindow");
392 if (settings.contains("geometry")) {
393 restoreGeometry(settings.value("geometry").toByteArray());
394 restoreState(settings.value("state").toByteArray());
400 session_count = settings.value("sessions", 0).toInt();
402 for (i = 0; i < session_count; i++) {
403 settings.beginGroup("Session" + QString::number(i));
404 shared_ptr<Session> session = add_session();
405 session->restore_settings(settings);
410 std::shared_ptr<Session> MainWindow::get_tab_session(int index) const
412 // Find the session that belongs to the tab's main window
413 for (auto entry : session_windows_)
414 if (entry.second == session_selector_.widget(index))
420 void MainWindow::closeEvent(QCloseEvent *event)
426 QMenu* MainWindow::createPopupMenu()
431 bool MainWindow::restoreState(const QByteArray &state, int version)
436 // Do nothing. We don't want Qt to handle this, or else it
437 // will try to restore all the dock widgets and create havoc.
442 void MainWindow::on_add_view(const QString &title, views::ViewType type,
445 // We get a pointer and need a reference
446 for (std::shared_ptr<Session> s : sessions_)
447 if (s.get() == session)
448 add_view(title, type, *s);
451 void MainWindow::on_focus_changed()
453 static shared_ptr<Session> prev_session;
455 shared_ptr<views::ViewBase> view = get_active_view();
458 for (shared_ptr<Session> session : sessions_) {
459 if (session->has_view(view)) {
460 if (session != prev_session) {
461 // Activate correct tab if necessary
462 shared_ptr<Session> tab_session = get_tab_session(
463 session_selector_.currentIndex());
464 if (tab_session != session)
465 session_selector_.setCurrentWidget(
466 session_windows_.at(session));
468 on_focused_session_changed(session);
471 prev_session = session;
477 if (sessions_.empty())
478 setWindowTitle(WindowTitle);
481 void MainWindow::on_focused_session_changed(shared_ptr<Session> session)
483 setWindowTitle(session->name() + " - " + WindowTitle);
486 void MainWindow::on_new_session_clicked()
491 void MainWindow::on_session_name_changed()
493 // Update the corresponding dock widget's name(s)
494 Session *session = qobject_cast<Session*>(QObject::sender());
497 for (shared_ptr<views::ViewBase> view : session->views()) {
498 // Get the dock that contains the view
499 for (auto entry : view_docks_)
500 if (entry.second == view) {
501 entry.first->setObjectName(session->name());
502 entry.first->setWindowTitle(session->name());
506 // Refresh window title if the affected session has focus
507 shared_ptr<views::ViewBase> view = get_active_view();
509 if (view && session->has_view(view))
510 setWindowTitle(session->name() + " - " + WindowTitle);
513 void MainWindow::on_new_view(Session *session)
515 // We get a pointer and need a reference
516 for (std::shared_ptr<Session> s : sessions_)
517 if (s.get() == session)
518 add_view(session->name(), views::ViewTypeTrace, *s);
521 void MainWindow::on_view_close_clicked()
523 // Find the dock widget that contains the close button that was clicked
524 QObject *w = QObject::sender();
525 QDockWidget *dock = 0;
528 dock = qobject_cast<QDockWidget*>(w);
534 // Get the view contained in the dock widget
535 shared_ptr<views::ViewBase> view;
537 for (auto entry : view_docks_)
538 if (entry.first == dock)
541 // Deregister the view
542 for (shared_ptr<Session> session : sessions_) {
543 if (!session->has_view(view))
546 // Also destroy the entire session if its main view is closing
547 if (view == session->main_view()) {
548 remove_session(session);
551 session->deregister_view(view);
555 void MainWindow::on_tab_changed(int index)
557 shared_ptr<Session> session = get_tab_session(index);
560 on_focused_session_changed(session);
563 void MainWindow::on_tab_close_requested(int index)
565 // TODO Ask user if this is intended in case data is unsaved
567 shared_ptr<Session> session = get_tab_session(index);
570 remove_session(session);
573 void MainWindow::on_actionViewStickyScrolling_triggered()
575 shared_ptr<views::ViewBase> viewbase = get_active_view();
576 views::TraceView::View* view =
577 qobject_cast<views::TraceView::View*>(viewbase.get());
579 view->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
582 void MainWindow::on_actionViewColouredBg_triggered()
584 shared_ptr<views::ViewBase> viewbase = get_active_view();
585 views::TraceView::View* view =
586 qobject_cast<views::TraceView::View*>(viewbase.get());
588 view->enable_coloured_bg(action_view_coloured_bg_->isChecked());
591 void MainWindow::on_actionAbout_triggered()
593 dialogs::About dlg(device_manager_.context(), this);