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, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include <cassert>
  22
  23 #ifdef ENABLE_DECODE
  24 #include <libsigrokdecode/libsigrokdecode.h>
  25 #endif
  26
  27 #include <algorithm>
  28 #include <iterator>
  29
  30 #include <QAction>
  31 #include <QApplication>
  32 #include <QCloseEvent>
  33 #include <QDockWidget>
  34 #include <QHBoxLayout>
  35 #include <QMessageBox>
  36 #include <QSettings>
  37 #include <QWidget>
  38
  39 #include "mainwindow.hpp"
  40
  41 #include "devicemanager.hpp"
  42 #include "util.hpp"
  43 #include "devices/hardwaredevice.hpp"
  44 #include "dialogs/about.hpp"
  45 #include "toolbars/mainbar.hpp"
  46 #include "view/view.hpp"
  47
  48 #include <stdint.h>
  49 #include <stdarg.h>
  50 #include <libsigrokcxx/libsigrokcxx.hpp>
  51
  52 using std::dynamic_pointer_cast;
  53 using std::list;
  54 using std::make_shared;
  55 using std::map;
  56 using std::shared_ptr;
  57 using std::string;
  58
  59 namespace pv {
  60
  61 namespace view {
  62 class ViewItem;
  63 }
  64
  65 using toolbars::MainBar;
  66
  67 const QString MainWindow::WindowTitle = tr("PulseView");
  68
  69 MainWindow::MainWindow(DeviceManager &device_manager,
  70         string open_file_name, string open_file_format,
  71         QWidget *parent) :
  72         QMainWindow(parent),
  73         device_manager_(device_manager),
  74         session_selector_(this),
  75         session_state_mapper_(this),
  76         action_view_sticky_scrolling_(new QAction(this)),
  77         action_view_coloured_bg_(new QAction(this)),
  78         action_about_(new QAction(this)),
  79         icon_red_(":/icons/status-red.svg"),
  80         icon_green_(":/icons/status-green.svg"),
  81         icon_grey_(":/icons/status-grey.svg")
  82 {
  83         qRegisterMetaType<util::Timestamp>("util::Timestamp");
  84
  85         setup_ui();
  86         restore_ui_settings();
  87
  88         if (!open_file_name.empty()) {
  89                 shared_ptr<Session> session = add_session();
  90                 session->main_bar()->load_init_file(open_file_name, open_file_format);
  91         }
  92
  93         // Add empty default session if there aren't any sessions
  94         if (sessions_.size() == 0) {
  95                 shared_ptr<Session> session = add_session();
  96
  97                 map<string, string> dev_info;
  98                 shared_ptr<devices::HardwareDevice> other_device, demo_device;
  99
 100                 // Use any available device that's not demo
 101                 for (shared_ptr<devices::HardwareDevice> dev : device_manager_.devices()) {
 102                         if (dev->hardware_device()->driver()->name() == "demo") {
 103                                 demo_device = dev;
 104                         } else {
 105                                 other_device = dev;
 106                         }
 107                 }
 108
 109                 // ...and if there isn't any, just use demo then
 110                 session->main_bar()->select_device(other_device ?
 111                         other_device : demo_device);
 112         }
 113 }
 114
 115 MainWindow::~MainWindow()
 116 {
 117         while (!sessions_.empty())
 118                 remove_session(sessions_.front());
 119 }
 120
 121 QAction* MainWindow::action_view_sticky_scrolling() const
 122 {
 123         return action_view_sticky_scrolling_;
 124 }
 125
 126 QAction* MainWindow::action_view_coloured_bg() const
 127 {
 128         return action_view_coloured_bg_;
 129 }
 130
 131 QAction* MainWindow::action_about() const
 132 {
 133         return action_about_;
 134 }
 135
 136 shared_ptr<views::ViewBase> MainWindow::get_active_view() const
 137 {
 138         // If there's only one view, use it...
 139         if (view_docks_.size() == 1)
 140                 return view_docks_.begin()->second;
 141
 142         // ...otherwise find the dock widget the widget with focus is contained in
 143         QObject *w = QApplication::focusWidget();
 144         QDockWidget *dock = 0;
 145
 146         while (w) {
 147             dock = qobject_cast<QDockWidget*>(w);
 148             if (dock)
 149                 break;
 150             w = w->parent();
 151         }
 152
 153         // Get the view contained in the dock widget
 154         for (auto entry : view_docks_)
 155                 if (entry.first == dock)
 156                         return entry.second;
 157
 158         return nullptr;
 159 }
 160
 161 shared_ptr<views::ViewBase> MainWindow::add_view(const QString &title,
 162         views::ViewType type, Session &session)
 163 {
 164         QMainWindow *main_window;
 165         for (auto entry : session_windows_)
 166                 if (entry.first.get() == &session)
 167                         main_window = entry.second;
 168
 169         assert(main_window);
 170
 171         if (type == views::ViewTypeTrace) {
 172                 QDockWidget* dock = new QDockWidget(title, main_window);
 173                 dock->setObjectName(title);
 174                 main_window->addDockWidget(Qt::TopDockWidgetArea, dock);
 175
 176                 // Insert a QMainWindow into the dock widget to allow for a tool bar
 177                 QMainWindow *dock_main = new QMainWindow(dock);
 178                 dock_main->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
 179
 180                 shared_ptr<views::TraceView::View> v =
 181                         make_shared<views::TraceView::View>(session, dock_main);
 182                 view_docks_[dock] = v;
 183                 session.register_view(v);
 184
 185                 dock_main->setCentralWidget(v.get());
 186                 dock->setWidget(dock_main);
 187
 188                 dock->setFeatures(QDockWidget::DockWidgetMovable |
 189                         QDockWidget::DockWidgetFloatable | QDockWidget::DockWidgetClosable);
 190
 191                 QAbstractButton *close_btn =
 192                         dock->findChildren<QAbstractButton*>
 193                                 ("qt_dockwidget_closebutton").front();
 194
 195                 connect(close_btn, SIGNAL(clicked(bool)),
 196                         this, SLOT(on_view_close_clicked()));
 197
 198                 if (type == views::ViewTypeTrace) {
 199                         connect(&session, SIGNAL(trigger_event(util::Timestamp)),
 200                                 qobject_cast<views::ViewBase*>(v.get()),
 201                                 SLOT(trigger_event(util::Timestamp)));
 202
 203                         v->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
 204                         v->enable_coloured_bg(action_view_coloured_bg_->isChecked());
 205
 206                         shared_ptr<MainBar> main_bar = session.main_bar();
 207                         if (!main_bar) {
 208                                 main_bar = make_shared<MainBar>(session, *this);
 209                                 dock_main->addToolBar(main_bar.get());
 210                                 session.set_main_bar(main_bar);
 211
 212                                 connect(main_bar.get(), SIGNAL(new_view(Session*)),
 213                                         this, SLOT(on_new_view(Session*)));
 214                         }
 215                         main_bar->action_view_show_cursors()->setChecked(v->cursors_shown());
 216
 217                         connect(v.get(), SIGNAL(always_zoom_to_fit_changed(bool)),
 218                                 main_bar.get(), SLOT(on_always_zoom_to_fit_changed(bool)));
 219                 }
 220
 221                 return v;
 222         }
 223
 224         return nullptr;
 225 }
 226
 227 shared_ptr<Session> MainWindow::add_session()
 228 {
 229         static int last_session_id = 1;
 230         QString name = tr("Untitled-%1").arg(last_session_id++);
 231
 232         shared_ptr<Session> session = make_shared<Session>(device_manager_, name);
 233
 234         connect(session.get(), SIGNAL(add_view(const QString&, views::ViewType, Session*)),
 235                 this, SLOT(on_add_view(const QString&, views::ViewType, Session*)));
 236         connect(session.get(), SIGNAL(name_changed()),
 237                 this, SLOT(on_session_name_changed()));
 238         session_state_mapper_.setMapping(session.get(), session.get());
 239         connect(session.get(), SIGNAL(capture_state_changed(int)),
 240                 &session_state_mapper_, SLOT(map()));
 241
 242         sessions_.push_back(session);
 243
 244         QMainWindow *window = new QMainWindow();
 245         window->setWindowFlags(Qt::Widget);  // Remove Qt::Window flag
 246         session_windows_[session] = window;
 247
 248         int index = session_selector_.addTab(window, name);
 249         session_selector_.setCurrentIndex(index);
 250         last_focused_session_ = session;
 251
 252         window->setDockNestingEnabled(true);
 253
 254         shared_ptr<views::ViewBase> main_view =
 255                 add_view(name, views::ViewTypeTrace, *session);
 256
 257         return session;
 258 }
 259
 260 void MainWindow::remove_session(shared_ptr<Session> session)
 261 {
 262         int h = new_session_button_->height();
 263
 264         for (shared_ptr<views::ViewBase> view : session->views()) {
 265                 // Find the dock the view is contained in and remove it
 266                 for (auto entry : view_docks_)
 267                         if (entry.second == view) {
 268                                 // Remove the view from the session
 269                                 session->deregister_view(view);
 270
 271                                 // Remove the view from its parent; otherwise, Qt will
 272                                 // call deleteLater() on it, which causes a double free
 273                                 // since the shared_ptr in view_docks_ doesn't know
 274                                 // that Qt keeps a pointer to the view around
 275                                 entry.second->setParent(0);
 276
 277                                 // Remove this entry from the container
 278                                 view_docks_.erase(entry.first);
 279                         }
 280         }
 281
 282         QMainWindow *window = session_windows_.at(session);
 283         session_selector_.removeTab(session_selector_.indexOf(window));
 284
 285         session_windows_.erase(session);
 286
 287         if (last_focused_session_ == session)
 288                 last_focused_session_.reset();
 289
 290         sessions_.remove_if([&](shared_ptr<Session> s) {
 291                 return s == session; });
 292
 293         if (sessions_.empty()) {
 294                 // When there are no more tabs, the height of the QTabWidget
 295                 // drops to zero. We must prevent this to keep the static
 296                 // widgets visible
 297                 for (QWidget *w : static_tab_widget_->findChildren<QWidget*>())
 298                         w->setMinimumHeight(h);
 299
 300                 int margin = static_tab_widget_->layout()->contentsMargins().bottom();
 301                 static_tab_widget_->setMinimumHeight(h + 2 * margin);
 302                 session_selector_.setMinimumHeight(h + 2 * margin);
 303
 304                 // Update the window title if there is no view left to
 305                 // generate focus change events
 306                 setWindowTitle(WindowTitle);
 307         }
 308 }
 309
 310 void MainWindow::setup_ui()
 311 {
 312         setObjectName(QString::fromUtf8("MainWindow"));
 313
 314         setCentralWidget(&session_selector_);
 315
 316         // Set the window icon
 317         QIcon icon;
 318         icon.addFile(QString(":/icons/sigrok-logo-notext.png"));
 319         setWindowIcon(icon);
 320
 321         action_view_sticky_scrolling_->setCheckable(true);
 322         action_view_sticky_scrolling_->setChecked(true);
 323         action_view_sticky_scrolling_->setShortcut(QKeySequence(Qt::Key_S));
 324         action_view_sticky_scrolling_->setObjectName(
 325                 QString::fromUtf8("actionViewStickyScrolling"));
 326         action_view_sticky_scrolling_->setText(tr("&Sticky Scrolling"));
 327
 328         action_view_coloured_bg_->setCheckable(true);
 329         action_view_coloured_bg_->setChecked(true);
 330         action_view_coloured_bg_->setShortcut(QKeySequence(Qt::Key_B));
 331         action_view_coloured_bg_->setObjectName(
 332                 QString::fromUtf8("actionViewColouredBg"));
 333         action_view_coloured_bg_->setText(tr("Use &coloured backgrounds"));
 334
 335         action_about_->setObjectName(QString::fromUtf8("actionAbout"));
 336         action_about_->setText(tr("&About..."));
 337
 338         // Set up the tab area
 339         new_session_button_ = new QToolButton();
 340         new_session_button_->setIcon(QIcon::fromTheme("document-new",
 341                 QIcon(":/icons/document-new.png")));
 342         new_session_button_->setAutoRaise(true);
 343
 344         run_stop_button_ = new QToolButton();
 345         run_stop_button_->setAutoRaise(true);
 346         run_stop_button_->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
 347         run_stop_button_->setShortcut(QKeySequence(Qt::Key_Space));
 348
 349         QFrame *separator = new QFrame();
 350         separator->setFrameStyle(QFrame::VLine | QFrame::Raised);
 351
 352         QHBoxLayout* layout = new QHBoxLayout();
 353         layout->setContentsMargins(2, 2, 2, 2);
 354         layout->addWidget(new_session_button_);
 355         layout->addWidget(separator);
 356         layout->addWidget(run_stop_button_);
 357
 358         static_tab_widget_ = new QWidget();
 359         static_tab_widget_->setLayout(layout);
 360
 361         session_selector_.setCornerWidget(static_tab_widget_, Qt::TopLeftCorner);
 362         session_selector_.setTabsClosable(true);
 363
 364         connect(new_session_button_, SIGNAL(clicked(bool)),
 365                 this, SLOT(on_new_session_clicked()));
 366         connect(run_stop_button_, SIGNAL(clicked(bool)),
 367                 this, SLOT(on_run_stop_clicked()));
 368         connect(&session_state_mapper_, SIGNAL(mapped(QObject*)),
 369                 this, SLOT(on_capture_state_changed(QObject*)));
 370
 371         connect(&session_selector_, SIGNAL(tabCloseRequested(int)),
 372                 this, SLOT(on_tab_close_requested(int)));
 373         connect(&session_selector_, SIGNAL(currentChanged(int)),
 374                 this, SLOT(on_tab_changed(int)));
 375
 376
 377         connect(static_cast<QApplication *>(QCoreApplication::instance()),
 378                 SIGNAL(focusChanged(QWidget*, QWidget*)),
 379                 this, SLOT(on_focus_changed()));
 380 }
 381
 382 void MainWindow::save_ui_settings()
 383 {
 384         QSettings settings;
 385         int id = 0;
 386
 387         settings.beginGroup("MainWindow");
 388         settings.setValue("state", saveState());
 389         settings.setValue("geometry", saveGeometry());
 390         settings.endGroup();
 391
 392         for (shared_ptr<Session> session : sessions_) {
 393                 // Ignore sessions using the demo device or no device at all
 394                 if (session->device()) {
 395                         shared_ptr<devices::HardwareDevice> device =
 396                                 dynamic_pointer_cast< devices::HardwareDevice >
 397                                 (session->device());
 398
 399                         if (device &&
 400                                 device->hardware_device()->driver()->name() == "demo")
 401                                 continue;
 402
 403                         settings.beginGroup("Session" + QString::number(id++));
 404                         settings.remove("");  // Remove all keys in this group
 405                         session->save_settings(settings);
 406                         settings.endGroup();
 407                 }
 408         }
 409
 410         settings.setValue("sessions", id);
 411 }
 412
 413 void MainWindow::restore_ui_settings()
 414 {
 415         QSettings settings;
 416         int i, session_count;
 417
 418         settings.beginGroup("MainWindow");
 419
 420         if (settings.contains("geometry")) {
 421                 restoreGeometry(settings.value("geometry").toByteArray());
 422                 restoreState(settings.value("state").toByteArray());
 423         } else
 424                 resize(1000, 720);
 425
 426         settings.endGroup();
 427
 428         session_count = settings.value("sessions", 0).toInt();
 429
 430         for (i = 0; i < session_count; i++) {
 431                 settings.beginGroup("Session" + QString::number(i));
 432                 shared_ptr<Session> session = add_session();
 433                 session->restore_settings(settings);
 434                 settings.endGroup();
 435         }
 436 }
 437
 438 std::shared_ptr<Session> MainWindow::get_tab_session(int index) const
 439 {
 440         // Find the session that belongs to the tab's main window
 441         for (auto entry : session_windows_)
 442                 if (entry.second == session_selector_.widget(index))
 443                         return entry.first;
 444
 445         return nullptr;
 446 }
 447
 448 void MainWindow::closeEvent(QCloseEvent *event)
 449 {
 450         save_ui_settings();
 451         event->accept();
 452 }
 453
 454 QMenu* MainWindow::createPopupMenu()
 455 {
 456         return nullptr;
 457 }
 458
 459 bool MainWindow::restoreState(const QByteArray &state, int version)
 460 {
 461         (void)state;
 462         (void)version;
 463
 464         // Do nothing. We don't want Qt to handle this, or else it
 465         // will try to restore all the dock widgets and create havoc.
 466
 467         return false;
 468 }
 469
 470 void MainWindow::session_error(const QString text, const QString info_text)
 471 {
 472         QMetaObject::invokeMethod(this, "show_session_error",
 473                 Qt::QueuedConnection, Q_ARG(QString, text),
 474                 Q_ARG(QString, info_text));
 475 }
 476
 477 void MainWindow::show_session_error(const QString text, const QString info_text)
 478 {
 479         QMessageBox msg(this);
 480         msg.setText(text);
 481         msg.setInformativeText(info_text);
 482         msg.setStandardButtons(QMessageBox::Ok);
 483         msg.setIcon(QMessageBox::Warning);
 484         msg.exec();
 485 }
 486
 487 void MainWindow::on_add_view(const QString &title, views::ViewType type,
 488         Session *session)
 489 {
 490         // We get a pointer and need a reference
 491         for (std::shared_ptr<Session> s : sessions_)
 492                 if (s.get() == session)
 493                         add_view(title, type, *s);
 494 }
 495
 496 void MainWindow::on_focus_changed()
 497 {
 498         shared_ptr<views::ViewBase> view = get_active_view();
 499
 500         if (view) {
 501                 for (shared_ptr<Session> session : sessions_) {
 502                         if (session->has_view(view)) {
 503                                 if (session != last_focused_session_) {
 504                                         // Activate correct tab if necessary
 505                                         shared_ptr<Session> tab_session = get_tab_session(
 506                                                 session_selector_.currentIndex());
 507                                         if (tab_session != session)
 508                                                 session_selector_.setCurrentWidget(
 509                                                         session_windows_.at(session));
 510
 511                                         on_focused_session_changed(session);
 512                                 }
 513
 514                                 break;
 515                         }
 516                 }
 517         }
 518
 519         if (sessions_.empty())
 520                 setWindowTitle(WindowTitle);
 521 }
 522
 523 void MainWindow::on_focused_session_changed(shared_ptr<Session> session)
 524 {
 525         last_focused_session_ = session;
 526
 527         setWindowTitle(session->name() + " - " + WindowTitle);
 528
 529         // Update the state of the run/stop button, too
 530         on_capture_state_changed(session.get());
 531 }
 532
 533 void MainWindow::on_new_session_clicked()
 534 {
 535         add_session();
 536 }
 537
 538 void MainWindow::on_run_stop_clicked()
 539 {
 540         shared_ptr<Session> session = last_focused_session_;
 541
 542         if (!session)
 543                 return;
 544
 545         switch (session->get_capture_state()) {
 546         case Session::Stopped:
 547                 session->start_capture([&](QString message) {
 548                         session_error("Capture failed", message); });
 549                 break;
 550         case Session::AwaitingTrigger:
 551         case Session::Running:
 552                 session->stop_capture();
 553                 break;
 554         }
 555 }
 556
 557 void MainWindow::on_session_name_changed()
 558 {
 559         // Update the corresponding dock widget's name(s)
 560         Session *session = qobject_cast<Session*>(QObject::sender());
 561         assert(session);
 562
 563         for (shared_ptr<views::ViewBase> view : session->views()) {
 564                 // Get the dock that contains the view
 565                 for (auto entry : view_docks_)
 566                         if (entry.second == view) {
 567                                 entry.first->setObjectName(session->name());
 568                                 entry.first->setWindowTitle(session->name());
 569                         }
 570         }
 571
 572         // Refresh window title if the affected session has focus
 573         if (session == last_focused_session_.get())
 574                 setWindowTitle(session->name() + " - " + WindowTitle);
 575 }
 576
 577 void MainWindow::on_capture_state_changed(QObject *obj)
 578 {
 579         Session *caller = qobject_cast<Session*>(obj);
 580
 581         // Ignore if caller is not the currently focused session
 582         // unless there is only one session
 583         if ((sessions_.size() > 1) && (caller != last_focused_session_.get()))
 584                 return;
 585
 586         int state = caller->get_capture_state();
 587
 588         const QIcon *icons[] = {&icon_grey_, &icon_red_, &icon_green_};
 589         run_stop_button_->setIcon(*icons[state]);
 590         run_stop_button_->setText((state == pv::Session::Stopped) ?
 591                 tr("Run") : tr("Stop"));
 592 }
 593
 594 void MainWindow::on_new_view(Session *session)
 595 {
 596         // We get a pointer and need a reference
 597         for (std::shared_ptr<Session> s : sessions_)
 598                 if (s.get() == session)
 599                         add_view(session->name(), views::ViewTypeTrace, *s);
 600 }
 601
 602 void MainWindow::on_view_close_clicked()
 603 {
 604         // Find the dock widget that contains the close button that was clicked
 605         QObject *w = QObject::sender();
 606         QDockWidget *dock = 0;
 607
 608         while (w) {
 609             dock = qobject_cast<QDockWidget*>(w);
 610             if (dock)
 611                 break;
 612             w = w->parent();
 613         }
 614
 615         // Get the view contained in the dock widget
 616         shared_ptr<views::ViewBase> view;
 617
 618         for (auto entry : view_docks_)
 619                 if (entry.first == dock)
 620                         view = entry.second;
 621
 622         // Deregister the view
 623         for (shared_ptr<Session> session : sessions_) {
 624                 if (!session->has_view(view))
 625                         continue;
 626
 627                 // Also destroy the entire session if its main view is closing
 628                 if (view == session->main_view()) {
 629                         remove_session(session);
 630                         break;
 631                 } else
 632                         session->deregister_view(view);
 633         }
 634 }
 635
 636 void MainWindow::on_tab_changed(int index)
 637 {
 638         shared_ptr<Session> session = get_tab_session(index);
 639
 640         if (session)
 641                 on_focused_session_changed(session);
 642 }
 643
 644 void MainWindow::on_tab_close_requested(int index)
 645 {
 646         // TODO Ask user if this is intended in case data is unsaved
 647
 648         shared_ptr<Session> session = get_tab_session(index);
 649
 650         if (session)
 651                 remove_session(session);
 652 }
 653
 654 void MainWindow::on_actionViewStickyScrolling_triggered()
 655 {
 656         shared_ptr<views::ViewBase> viewbase = get_active_view();
 657         views::TraceView::View* view =
 658                 qobject_cast<views::TraceView::View*>(viewbase.get());
 659         if (view)
 660                 view->enable_sticky_scrolling(action_view_sticky_scrolling_->isChecked());
 661 }
 662
 663 void MainWindow::on_actionViewColouredBg_triggered()
 664 {
 665         shared_ptr<views::ViewBase> viewbase = get_active_view();
 666         views::TraceView::View* view =
 667                         qobject_cast<views::TraceView::View*>(viewbase.get());
 668         if (view)
 669                 view->enable_coloured_bg(action_view_coloured_bg_->isChecked());
 670 }
 671
 672 void MainWindow::on_actionAbout_triggered()
 673 {
 674         dialogs::About dlg(device_manager_.context(), this);
 675         dlg.exec();
 676 }
 677
 678 } // namespace pv