Add openpilot tools

tools/cabana/signalview.h

@@ -0,0 +1,149 @@
+#pragma once
+
+#include <memory>
+#include <set>
+
+#include <QAbstractItemModel>
+#include <QLabel>
+#include <QLineEdit>
+#include <QSlider>
+#include <QStyledItemDelegate>
+#include <QTableWidget>
+#include <QTreeView>
+
+#include "tools/cabana/chart/chartswidget.h"
+#include "tools/cabana/chart/sparkline.h"
+
+class SignalModel : public QAbstractItemModel {
+  Q_OBJECT
+public:
+  struct Item {
+    enum Type {Root, Sig, Name, Size, Node, Endian, Signed, Offset, Factor, SignalType, MultiplexValue, ExtraInfo, Unit, Comment, Min, Max, Desc };
+    ~Item() { qDeleteAll(children); }
+    inline int row() { return parent->children.indexOf(this); }
+
+    Type type = Type::Root;
+    Item *parent = nullptr;
+    QList<Item *> children;
+
+    const cabana::Signal *sig = nullptr;
+    QString title;
+    bool highlight = false;
+    bool extra_expanded = false;
+    QString sig_val = "-";
+    Sparkline sparkline;
+  };
+
+  SignalModel(QObject *parent);
+  int rowCount(const QModelIndex &parent = QModelIndex()) const override;
+  int columnCount(const QModelIndex &parent = QModelIndex()) const override { return 2; }
+  QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
+  QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override;
+  QModelIndex parent(const QModelIndex &index) const override;
+  Qt::ItemFlags flags(const QModelIndex &index) const override;
+  bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
+  void setMessage(const MessageId &id);
+  void setFilter(const QString &txt);
+  bool saveSignal(const cabana::Signal *origin_s, cabana::Signal &s);
+  Item *getItem(const QModelIndex &index) const;
+  int signalRow(const cabana::Signal *sig) const;
+  void showExtraInfo(const QModelIndex &index);
+
+private:
+  void insertItem(SignalModel::Item *parent_item, int pos, const cabana::Signal *sig);
+  void handleSignalAdded(MessageId id, const cabana::Signal *sig);
+  void handleSignalUpdated(const cabana::Signal *sig);
+  void handleSignalRemoved(const cabana::Signal *sig);
+  void handleMsgChanged(MessageId id);
+  void refresh();
+
+  MessageId msg_id;
+  QString filter_str;
+  std::unique_ptr<Item> root;
+  friend class SignalView;
+  friend class SignalItemDelegate;
+};
+
+class ValueDescriptionDlg : public QDialog {
+public:
+  ValueDescriptionDlg(const ValueDescription &descriptions, QWidget *parent);
+  ValueDescription val_desc;
+
+private:
+  struct Delegate : public QStyledItemDelegate {
+    Delegate(QWidget *parent) : QStyledItemDelegate(parent) {}
+    QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index) const override;
+  };
+
+  void save();
+  QTableWidget *table;
+};
+
+class SignalItemDelegate : public QStyledItemDelegate {
+public:
+  SignalItemDelegate(QObject *parent);
+  void paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const override;
+  QSize sizeHint(const QStyleOptionViewItem &option, const QModelIndex &index) const override;
+  QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index) const override;
+  void updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option, const QModelIndex &index) const override;
+  void setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const override;
+
+  QValidator *name_validator, *double_validator, *node_validator;
+  QFont label_font, minmax_font;
+  const int color_label_width = 18;
+  mutable QSize button_size;
+  mutable QHash<QString, int> width_cache;
+};
+
+class SignalView : public QFrame {
+  Q_OBJECT
+
+public:
+  SignalView(ChartsWidget *charts, QWidget *parent);
+  void setMessage(const MessageId &id);
+  void signalHovered(const cabana::Signal *sig);
+  void updateChartState();
+  void selectSignal(const cabana::Signal *sig, bool expand = false);
+  void rowClicked(const QModelIndex &index);
+  SignalModel *model = nullptr;
+
+signals:
+  void highlight(const cabana::Signal *sig);
+  void showChart(const MessageId &id, const cabana::Signal *sig, bool show, bool merge);
+
+private:
+  void rowsChanged();
+  void resizeEvent(QResizeEvent* event) override;
+  void updateToolBar();
+  void setSparklineRange(int value);
+  void handleSignalAdded(MessageId id, const cabana::Signal *sig);
+  void handleSignalUpdated(const cabana::Signal *sig);
+  void updateState(const std::set<MessageId> *msgs = nullptr);
+
+  struct TreeView : public QTreeView {
+    TreeView(QWidget *parent) : QTreeView(parent) {}
+    void rowsInserted(const QModelIndex &parent, int start, int end) override {
+      ((SignalView *)parentWidget())->rowsChanged();
+      // update widget geometries in QTreeView::rowsInserted
+      QTreeView::rowsInserted(parent, start, end);
+    }
+    void dataChanged(const QModelIndex &topLeft, const QModelIndex &bottomRight, const QVector<int> &roles = QVector<int>()) override {
+      // Bypass the slow call to QTreeView::dataChanged.
+      QAbstractItemView::dataChanged(topLeft, bottomRight, roles);
+    }
+    void leaveEvent(QEvent *event) override {
+      emit static_cast<SignalView *>(parentWidget())->highlight(nullptr);
+      QTreeView::leaveEvent(event);
+    }
+  };
+  int max_value_width = 0;
+  int value_column_width = 0;
+  TreeView *tree;
+  QLabel *sparkline_label;
+  QSlider *sparkline_range_slider;
+  QLineEdit *filter_edit;
+  ChartsWidget *charts;
+  QLabel *signal_count_lb;
+  SignalItemDelegate *delegate;
+  friend SignalItemDelegate;
+};