openpilot_comma/tools/cabana/streamselector.cc

#include "tools/cabana/streamselector.h"

#include <QDialogButtonBox>
#include <QFileDialog>
#include <QFormLayout>
#include <QLabel>
#include <QPushButton>

#include "tools/cabana/streams/devicestream.h"
#include "tools/cabana/streams/pandastream.h"
#include "tools/cabana/streams/replaystream.h"

StreamSelector::StreamSelector(AbstractStream **stream, QWidget *parent) : QDialog(parent) {
  setWindowTitle(tr("Open stream"));
  QVBoxLayout *main_layout = new QVBoxLayout(this);

  QWidget *w = new QWidget(this);
  QVBoxLayout *layout = new QVBoxLayout(w);
  tab = new QTabWidget(this);
  tab->setTabBarAutoHide(true);
  layout->addWidget(tab);

  QHBoxLayout *dbc_layout = new QHBoxLayout();
  dbc_file = new QLineEdit(this);
  dbc_file->setReadOnly(true);
  dbc_file->setPlaceholderText(tr("Choose a dbc file to open"));
  QPushButton *file_btn = new QPushButton(tr("Browse..."));
  dbc_layout->addWidget(new QLabel(tr("dbc File")));
  dbc_layout->addWidget(dbc_file);
  dbc_layout->addWidget(file_btn);
  layout->addLayout(dbc_layout);

  QFrame *line = new QFrame(this);
  line->setFrameStyle(QFrame::HLine | QFrame::Sunken);
  layout->addWidget(line);

  main_layout->addWidget(w);
  auto btn_box = new QDialogButtonBox(QDialogButtonBox::Open | QDialogButtonBox::Cancel);
  main_layout->addWidget(btn_box);

  addStreamWidget(ReplayStream::widget(stream));
  addStreamWidget(PandaStream::widget(stream));
  addStreamWidget(DeviceStream::widget(stream));

  QObject::connect(btn_box, &QDialogButtonBox::rejected, this, &QDialog::reject);
  QObject::connect(btn_box, &QDialogButtonBox::accepted, [=]() {
    btn_box->button(QDialogButtonBox::Open)->setEnabled(false);
    w->setEnabled(false);
    if (((AbstractOpenStreamWidget *)tab->currentWidget())->open()) {
      accept();
    } else {
      btn_box->button(QDialogButtonBox::Open)->setEnabled(true);
      w->setEnabled(true);
    }
  });
  QObject::connect(file_btn, &QPushButton::clicked, [this]() {
    QString fn = QFileDialog::getOpenFileName(this, tr("Open File"), settings.last_dir, "DBC (*.dbc)");
    if (!fn.isEmpty()) {
      dbc_file->setText(fn);
      settings.last_dir = QFileInfo(fn).absolutePath();
    }
  });
}

void StreamSelector::addStreamWidget(AbstractOpenStreamWidget *w) {
  tab->addTab(w, w->title());
}