Add openpilot tools

tools/cabana/dbc/dbc.cc

@@ -0,0 +1,211 @@
+#include "tools/cabana/dbc/dbc.h"
+
+#include <algorithm>
+
+#include "tools/cabana/util.h"
+
+uint qHash(const MessageId &item) {
+  return qHash(item.source) ^ qHash(item.address);
+}
+
+// cabana::Msg
+
+cabana::Msg::~Msg() {
+  for (auto s : sigs) {
+    delete s;
+  }
+}
+
+cabana::Signal *cabana::Msg::addSignal(const cabana::Signal &sig) {
+  auto s = sigs.emplace_back(new cabana::Signal(sig));
+  update();
+  return s;
+}
+
+cabana::Signal *cabana::Msg::updateSignal(const QString &sig_name, const cabana::Signal &new_sig) {
+  auto s = sig(sig_name);
+  if (s) {
+    *s = new_sig;
+    update();
+  }
+  return s;
+}
+
+void cabana::Msg::removeSignal(const QString &sig_name) {
+  auto it = std::find_if(sigs.begin(), sigs.end(), [&](auto &s) { return s->name == sig_name; });
+  if (it != sigs.end()) {
+    delete *it;
+    sigs.erase(it);
+    update();
+  }
+}
+
+cabana::Msg &cabana::Msg::operator=(const cabana::Msg &other) {
+  address = other.address;
+  name = other.name;
+  size = other.size;
+  comment = other.comment;
+
+  for (auto s : sigs) delete s;
+  sigs.clear();
+  for (auto s : other.sigs) {
+    sigs.push_back(new cabana::Signal(*s));
+  }
+
+  update();
+  return *this;
+}
+
+cabana::Signal *cabana::Msg::sig(const QString &sig_name) const {
+  auto it = std::find_if(sigs.begin(), sigs.end(), [&](auto &s) { return s->name == sig_name; });
+  return it != sigs.end() ? *it : nullptr;
+}
+
+int cabana::Msg::indexOf(const cabana::Signal *sig) const {
+  for (int i = 0; i < sigs.size(); ++i) {
+    if (sigs[i] == sig) return i;
+  }
+  return -1;
+}
+
+QString cabana::Msg::newSignalName() {
+  QString new_name;
+  for (int i = 1; /**/; ++i) {
+    new_name = QString("NEW_SIGNAL_%1").arg(i);
+    if (sig(new_name) == nullptr) break;
+  }
+  return new_name;
+}
+
+void cabana::Msg::update() {
+  if (transmitter.isEmpty()) {
+    transmitter = DEFAULT_NODE_NAME;
+  }
+  mask.assign(size, 0x00);
+  multiplexor = nullptr;
+
+  // sort signals
+  std::sort(sigs.begin(), sigs.end(), [](auto l, auto r) {
+    return std::tie(r->type, l->multiplex_value, l->start_bit, l->name) <
+           std::tie(l->type, r->multiplex_value, r->start_bit, r->name);
+  });
+
+  for (auto sig : sigs) {
+    if (sig->type == cabana::Signal::Type::Multiplexor) {
+      multiplexor = sig;
+    }
+    sig->update();
+
+    // update mask
+    int i = sig->msb / 8;
+    int bits = sig->size;
+    while (i >= 0 && i < size && bits > 0) {
+      int lsb = (int)(sig->lsb / 8) == i ? sig->lsb : i * 8;
+      int msb = (int)(sig->msb / 8) == i ? sig->msb : (i + 1) * 8 - 1;
+
+      int sz = msb - lsb + 1;
+      int shift = (lsb - (i * 8));
+
+      mask[i] |= ((1ULL << sz) - 1) << shift;
+
+      bits -= size;
+      i = sig->is_little_endian ? i - 1 : i + 1;
+    }
+  }
+
+  for (auto sig : sigs) {
+    sig->multiplexor = sig->type == cabana::Signal::Type::Multiplexed ? multiplexor : nullptr;
+    if (!sig->multiplexor) {
+      if (sig->type == cabana::Signal::Type::Multiplexed) {
+        sig->type = cabana::Signal::Type::Normal;
+      }
+      sig->multiplex_value = 0;
+    }
+  }
+}
+
+// cabana::Signal
+
+void cabana::Signal::update() {
+  updateMsbLsb(*this);
+  if (receiver_name.isEmpty()) {
+    receiver_name = DEFAULT_NODE_NAME;
+  }
+
+  float h = 19 * (float)lsb / 64.0;
+  h = fmod(h, 1.0);
+  size_t hash = qHash(name);
+  float s = 0.25 + 0.25 * (float)(hash & 0xff) / 255.0;
+  float v = 0.75 + 0.25 * (float)((hash >> 8) & 0xff) / 255.0;
+
+  color = QColor::fromHsvF(h, s, v);
+  precision = std::max(num_decimals(factor), num_decimals(offset));
+}
+
+QString cabana::Signal::formatValue(double value) const {
+  // Show enum string
+  int64_t raw_value = round((value - offset) / factor);
+  for (const auto &[val, desc] : val_desc) {
+    if (std::abs(raw_value - val) < 1e-6) {
+      return desc;
+    }
+  }
+
+  QString val_str = QString::number(value, 'f', precision);
+  if (!unit.isEmpty()) {
+    val_str += " " + unit;
+  }
+  return val_str;
+}
+
+bool cabana::Signal::getValue(const uint8_t *data, size_t data_size, double *val) const {
+  if (multiplexor && get_raw_value(data, data_size, *multiplexor) != multiplex_value) {
+    return false;
+  }
+  *val = get_raw_value(data, data_size, *this);
+  return true;
+}
+
+bool cabana::Signal::operator==(const cabana::Signal &other) const {
+  return name == other.name && size == other.size &&
+         start_bit == other.start_bit &&
+         msb == other.msb && lsb == other.lsb &&
+         is_signed == other.is_signed && is_little_endian == other.is_little_endian &&
+         factor == other.factor && offset == other.offset &&
+         min == other.min && max == other.max && comment == other.comment && unit == other.unit && val_desc == other.val_desc &&
+         multiplex_value == other.multiplex_value && type == other.type && receiver_name == other.receiver_name;
+}
+
+// helper functions
+
+double get_raw_value(const uint8_t *data, size_t data_size, const cabana::Signal &sig) {
+  int64_t val = 0;
+
+  int i = sig.msb / 8;
+  int bits = sig.size;
+  while (i >= 0 && i < data_size && bits > 0) {
+    int lsb = (int)(sig.lsb / 8) == i ? sig.lsb : i * 8;
+    int msb = (int)(sig.msb / 8) == i ? sig.msb : (i + 1) * 8 - 1;
+    int size = msb - lsb + 1;
+
+    uint64_t d = (data[i] >> (lsb - (i * 8))) & ((1ULL << size) - 1);
+    val |= d << (bits - size);
+
+    bits -= size;
+    i = sig.is_little_endian ? i - 1 : i + 1;
+  }
+  if (sig.is_signed) {
+    val -= ((val >> (sig.size - 1)) & 0x1) ? (1ULL << sig.size) : 0;
+  }
+  return val * sig.factor + sig.offset;
+}
+
+void updateMsbLsb(cabana::Signal &s) {
+  if (s.is_little_endian) {
+    s.lsb = s.start_bit;
+    s.msb = s.start_bit + s.size - 1;
+  } else {
+    s.lsb = flipBitPos(flipBitPos(s.start_bit) + s.size - 1);
+    s.msb = s.start_bit;
+  }
+}

tools/cabana/dbc/dbc.h

@@ -0,0 +1,122 @@
+#pragma once
+
+#include <limits>
+#include <string>
+#include <utility>
+#include <vector>
+
+#include <QColor>
+#include <QList>
+#include <QMetaType>
+#include <QString>
+
+
+const QString UNTITLED = "untitled";
+const QString DEFAULT_NODE_NAME = "XXX";
+
+struct MessageId {
+  uint8_t source = 0;
+  uint32_t address = 0;
+
+  QString toString() const {
+    return QString("%1:%2").arg(source).arg(address, 1, 16);
+  }
+
+  bool operator==(const MessageId &other) const {
+    return source == other.source && address == other.address;
+  }
+
+  bool operator!=(const MessageId &other) const {
+    return !(*this == other);
+  }
+
+  bool operator<(const MessageId &other) const {
+    return std::pair{source, address} < std::pair{other.source, other.address};
+  }
+
+  bool operator>(const MessageId &other) const {
+    return std::pair{source, address} > std::pair{other.source, other.address};
+  }
+};
+
+uint qHash(const MessageId &item);
+Q_DECLARE_METATYPE(MessageId);
+
+template <>
+struct std::hash<MessageId> {
+  std::size_t operator()(const MessageId &k) const noexcept { return qHash(k); }
+};
+
+typedef QList<std::pair<double, QString>> ValueDescription;
+
+namespace cabana {
+
+class Signal {
+public:
+  Signal() = default;
+  Signal(const Signal &other) = default;
+  void update();
+  bool getValue(const uint8_t *data, size_t data_size, double *val) const;
+  QString formatValue(double value) const;
+  bool operator==(const cabana::Signal &other) const;
+  inline bool operator!=(const cabana::Signal &other) const { return !(*this == other); }
+
+  enum class Type {
+    Normal = 0,
+    Multiplexed,
+    Multiplexor
+  };
+
+  Type type = Type::Normal;
+  QString name;
+  int start_bit, msb, lsb, size;
+  double factor = 1.0;
+  double offset = 0;
+  bool is_signed;
+  bool is_little_endian;
+  double min, max;
+  QString unit;
+  QString comment;
+  QString receiver_name;
+  ValueDescription val_desc;
+  int precision = 0;
+  QColor color;
+
+  // Multiplexed
+  int multiplex_value = 0;
+  Signal *multiplexor = nullptr;
+};
+
+class Msg {
+public:
+  Msg() = default;
+  Msg(const Msg &other) { *this = other; }
+  ~Msg();
+  cabana::Signal *addSignal(const cabana::Signal &sig);
+  cabana::Signal *updateSignal(const QString &sig_name, const cabana::Signal &sig);
+  void removeSignal(const QString &sig_name);
+  Msg &operator=(const Msg &other);
+  int indexOf(const cabana::Signal *sig) const;
+  cabana::Signal *sig(const QString &sig_name) const;
+  QString newSignalName();
+  void update();
+  inline const std::vector<cabana::Signal *> &getSignals() const { return sigs; }
+
+  uint32_t address;
+  QString name;
+  uint32_t size;
+  QString comment;
+  QString transmitter;
+  std::vector<cabana::Signal *> sigs;
+
+  std::vector<uint8_t> mask;
+  cabana::Signal *multiplexor = nullptr;
+};
+
+}  // namespace cabana
+
+// Helper functions
+double get_raw_value(const uint8_t *data, size_t data_size, const cabana::Signal &sig);
+void updateMsbLsb(cabana::Signal &s);
+inline int flipBitPos(int start_bit) { return 8 * (start_bit / 8) + 7 - start_bit % 8; }
+inline QString doubleToString(double value) { return QString::number(value, 'g', std::numeric_limits<double>::digits10); }

tools/cabana/dbc/dbcfile.cc

@@ -0,0 +1,240 @@
+#include "tools/cabana/dbc/dbcfile.h"
+
+#include <QFile>
+#include <QFileInfo>
+#include <QRegularExpression>
+#include <QTextStream>
+
+DBCFile::DBCFile(const QString &dbc_file_name) {
+  QFile file(dbc_file_name);
+  if (file.open(QIODevice::ReadOnly)) {
+    name_ = QFileInfo(dbc_file_name).baseName();
+    filename = dbc_file_name;
+    // Remove auto save file extension
+    if (dbc_file_name.endsWith(AUTO_SAVE_EXTENSION)) {
+      filename.chop(AUTO_SAVE_EXTENSION.length());
+    }
+    parse(file.readAll());
+  } else {
+    throw std::runtime_error("Failed to open file.");
+  }
+}
+
+DBCFile::DBCFile(const QString &name, const QString &content) : name_(name), filename("") {
+  // Open from clipboard
+  parse(content);
+}
+
+bool DBCFile::save() {
+  assert(!filename.isEmpty());
+  if (writeContents(filename)) {
+    cleanupAutoSaveFile();
+    return true;
+  }
+  return false;
+}
+
+bool DBCFile::saveAs(const QString &new_filename) {
+  filename = new_filename;
+  return save();
+}
+
+bool DBCFile::autoSave() {
+  return !filename.isEmpty() && writeContents(filename + AUTO_SAVE_EXTENSION);
+}
+
+void DBCFile::cleanupAutoSaveFile() {
+  if (!filename.isEmpty()) {
+    QFile::remove(filename + AUTO_SAVE_EXTENSION);
+  }
+}
+
+bool DBCFile::writeContents(const QString &fn) {
+  QFile file(fn);
+  if (file.open(QIODevice::WriteOnly)) {
+    file.write(generateDBC().toUtf8());
+    return true;
+  }
+  return false;
+}
+
+void DBCFile::updateMsg(const MessageId &id, const QString &name, uint32_t size, const QString &node, const QString &comment) {
+  auto &m = msgs[id.address];
+  m.address = id.address;
+  m.name = name;
+  m.size = size;
+  m.transmitter = node.isEmpty() ? DEFAULT_NODE_NAME : node;
+  m.comment = comment;
+}
+
+cabana::Msg *DBCFile::msg(uint32_t address) {
+  auto it = msgs.find(address);
+  return it != msgs.end() ? &it->second : nullptr;
+}
+
+cabana::Msg *DBCFile::msg(const QString &name) {
+  auto it = std::find_if(msgs.begin(), msgs.end(), [&name](auto &m) { return m.second.name == name; });
+  return it != msgs.end() ? &(it->second) : nullptr;
+}
+
+int DBCFile::signalCount() {
+  return std::accumulate(msgs.cbegin(), msgs.cend(), 0, [](int &n, const auto &m) { return n + m.second.sigs.size(); });
+}
+
+void DBCFile::parse(const QString &content) {
+  static QRegularExpression bo_regexp(R"(^BO_ (\w+) (\w+) *: (\w+) (\w+))");
+  static QRegularExpression sg_regexp(R"(^SG_ (\w+) : (\d+)\|(\d+)@(\d+)([\+|\-]) \(([0-9.+\-eE]+),([0-9.+\-eE]+)\) \[([0-9.+\-eE]+)\|([0-9.+\-eE]+)\] \"(.*)\" (.*))");
+  static QRegularExpression sgm_regexp(R"(^SG_ (\w+) (\w+) *: (\d+)\|(\d+)@(\d+)([\+|\-]) \(([0-9.+\-eE]+),([0-9.+\-eE]+)\) \[([0-9.+\-eE]+)\|([0-9.+\-eE]+)\] \"(.*)\" (.*))");
+  static QRegularExpression msg_comment_regexp(R"(^CM_ BO_ *(\w+) *\"([^"]*)\"\s*;)");
+  static QRegularExpression sg_comment_regexp(R"(^CM_ SG_ *(\w+) *(\w+) *\"([^"]*)\"\s*;)");
+  static QRegularExpression val_regexp(R"(VAL_ (\w+) (\w+) (\s*[-+]?[0-9]+\s+\".+?\"[^;]*))");
+
+  int line_num = 0;
+  QString line;
+  auto dbc_assert = [&line_num, &line, this](bool condition, const QString &msg = "") {
+    if (!condition) throw std::runtime_error(QString("[%1:%2]%3: %4").arg(filename).arg(line_num).arg(msg).arg(line).toStdString());
+  };
+  auto get_sig = [this](uint32_t address, const QString &name) -> cabana::Signal * {
+    auto m = (cabana::Msg *)msg(address);
+    return m ? (cabana::Signal *)m->sig(name) : nullptr;
+  };
+
+  msgs.clear();
+  QTextStream stream((QString *)&content);
+  cabana::Msg *current_msg = nullptr;
+  int multiplexor_cnt = 0;
+  while (!stream.atEnd()) {
+    ++line_num;
+    QString raw_line = stream.readLine();
+    line = raw_line.trimmed();
+    if (line.startsWith("BO_ ")) {
+      multiplexor_cnt = 0;
+      auto match = bo_regexp.match(line);
+      dbc_assert(match.hasMatch());
+      auto address = match.captured(1).toUInt();
+      dbc_assert(msgs.count(address) == 0, QString("Duplicate message address: %1").arg(address));
+      current_msg = &msgs[address];
+      current_msg->address = address;
+      current_msg->name = match.captured(2);
+      current_msg->size = match.captured(3).toULong();
+      current_msg->transmitter = match.captured(4).trimmed();
+    } else if (line.startsWith("SG_ ")) {
+      int offset = 0;
+      auto match = sg_regexp.match(line);
+      if (!match.hasMatch()) {
+        match = sgm_regexp.match(line);
+        offset = 1;
+      }
+      dbc_assert(match.hasMatch());
+      dbc_assert(current_msg, "No Message");
+      auto name = match.captured(1);
+      dbc_assert(current_msg->sig(name) == nullptr, "Duplicate signal name");
+      cabana::Signal s{};
+      if (offset == 1) {
+        auto indicator = match.captured(2);
+        if (indicator == "M") {
+          // Only one signal within a single message can be the multiplexer switch.
+          dbc_assert(++multiplexor_cnt < 2, "Multiple multiplexor");
+          s.type = cabana::Signal::Type::Multiplexor;
+        } else {
+          s.type = cabana::Signal::Type::Multiplexed;
+          s.multiplex_value = indicator.mid(1).toInt();
+        }
+      }
+      s.name = name;
+      s.start_bit = match.captured(offset + 2).toInt();
+      s.size = match.captured(offset + 3).toInt();
+      s.is_little_endian = match.captured(offset + 4).toInt() == 1;
+      s.is_signed = match.captured(offset + 5) == "-";
+      s.factor = match.captured(offset + 6).toDouble();
+      s.offset = match.captured(offset + 7).toDouble();
+      s.min = match.captured(8 + offset).toDouble();
+      s.max = match.captured(9 + offset).toDouble();
+      s.unit = match.captured(10 + offset);
+      s.receiver_name = match.captured(11 + offset).trimmed();
+
+      current_msg->sigs.push_back(new cabana::Signal(s));
+    } else if (line.startsWith("VAL_ ")) {
+      auto match = val_regexp.match(line);
+      dbc_assert(match.hasMatch());
+      if (auto s = get_sig(match.captured(1).toUInt(), match.captured(2))) {
+        QStringList desc_list = match.captured(3).trimmed().split('"');
+        for (int i = 0; i < desc_list.size(); i += 2) {
+          auto val = desc_list[i].trimmed();
+          if (!val.isEmpty() && (i + 1) < desc_list.size()) {
+            auto desc = desc_list[i + 1].trimmed();
+            s->val_desc.push_back({val.toDouble(), desc});
+          }
+        }
+      }
+    } else if (line.startsWith("CM_ BO_")) {
+      if (!line.endsWith("\";")) {
+        int pos = stream.pos() - raw_line.length() - 1;
+        line = content.mid(pos, content.indexOf("\";", pos));
+      }
+      auto match = msg_comment_regexp.match(line);
+      dbc_assert(match.hasMatch());
+      if (auto m = (cabana::Msg *)msg(match.captured(1).toUInt())) {
+        m->comment = match.captured(2).trimmed();
+      }
+    } else if (line.startsWith("CM_ SG_ ")) {
+      if (!line.endsWith("\";")) {
+        int pos = stream.pos() - raw_line.length() - 1;
+        line = content.mid(pos, content.indexOf("\";", pos));
+      }
+      auto match = sg_comment_regexp.match(line);
+      dbc_assert(match.hasMatch());
+      if (auto s = get_sig(match.captured(1).toUInt(), match.captured(2))) {
+        s->comment = match.captured(3).trimmed();
+      }
+    }
+  }
+
+  for (auto &[_, m] : msgs) {
+    m.update();
+  }
+}
+
+QString DBCFile::generateDBC() {
+  QString dbc_string, signal_comment, message_comment, val_desc;
+  for (const auto &[address, m] : msgs) {
+    const QString transmitter = m.transmitter.isEmpty() ? DEFAULT_NODE_NAME : m.transmitter;
+    dbc_string += QString("BO_ %1 %2: %3 %4\n").arg(address).arg(m.name).arg(m.size).arg(transmitter);
+    if (!m.comment.isEmpty()) {
+      message_comment += QString("CM_ BO_ %1 \"%2\";\n").arg(address).arg(m.comment);
+    }
+    for (auto sig : m.getSignals()) {
+      QString multiplexer_indicator;
+      if (sig->type == cabana::Signal::Type::Multiplexor) {
+        multiplexer_indicator = "M ";
+      } else if (sig->type == cabana::Signal::Type::Multiplexed) {
+        multiplexer_indicator = QString("m%1 ").arg(sig->multiplex_value);
+      }
+      dbc_string += QString(" SG_ %1 %2: %3|%4@%5%6 (%7,%8) [%9|%10] \"%11\" %12\n")
+                        .arg(sig->name)
+                        .arg(multiplexer_indicator)
+                        .arg(sig->start_bit)
+                        .arg(sig->size)
+                        .arg(sig->is_little_endian ? '1' : '0')
+                        .arg(sig->is_signed ? '-' : '+')
+                        .arg(doubleToString(sig->factor))
+                        .arg(doubleToString(sig->offset))
+                        .arg(doubleToString(sig->min))
+                        .arg(doubleToString(sig->max))
+                        .arg(sig->unit)
+                        .arg(sig->receiver_name.isEmpty() ? DEFAULT_NODE_NAME : sig->receiver_name);
+      if (!sig->comment.isEmpty()) {
+        signal_comment += QString("CM_ SG_ %1 %2 \"%3\";\n").arg(address).arg(sig->name).arg(sig->comment);
+      }
+      if (!sig->val_desc.isEmpty()) {
+        QStringList text;
+        for (auto &[val, desc] : sig->val_desc) {
+          text << QString("%1 \"%2\"").arg(val).arg(desc);
+        }
+        val_desc += QString("VAL_ %1 %2 %3;\n").arg(address).arg(sig->name).arg(text.join(" "));
+      }
+    }
+    dbc_string += "\n";
+  }
+  return dbc_string + message_comment + signal_comment + val_desc;
+}

tools/cabana/dbc/dbcfile.h

@@ -0,0 +1,41 @@
+#pragma once
+
+#include <map>
+
+#include "tools/cabana/dbc/dbc.h"
+
+const QString AUTO_SAVE_EXTENSION = ".tmp";
+
+class DBCFile {
+public:
+  DBCFile(const QString &dbc_file_name);
+  DBCFile(const QString &name, const QString &content);
+  ~DBCFile() {}
+
+  bool save();
+  bool saveAs(const QString &new_filename);
+  bool autoSave();
+  bool writeContents(const QString &fn);
+  void cleanupAutoSaveFile();
+  QString generateDBC();
+
+  void updateMsg(const MessageId &id, const QString &name, uint32_t size, const QString &node, const QString &comment);
+  inline void removeMsg(const MessageId &id) { msgs.erase(id.address); }
+
+  inline const std::map<uint32_t, cabana::Msg> &getMessages() const { return msgs; }
+  cabana::Msg *msg(uint32_t address);
+  cabana::Msg *msg(const QString &name);
+  inline cabana::Msg *msg(const MessageId &id) { return msg(id.address); }
+
+  int signalCount();
+  inline int msgCount() { return msgs.size(); }
+  inline QString name() { return name_.isEmpty() ? "untitled" : name_; }
+  inline bool isEmpty() { return (signalCount() == 0) && name_.isEmpty(); }
+
+  QString filename;
+
+private:
+  void parse(const QString &content);
+  std::map<uint32_t, cabana::Msg> msgs;
+  QString name_;
+};

tools/cabana/dbc/dbcmanager.cc

@@ -0,0 +1,202 @@
+#include "tools/cabana/dbc/dbcmanager.h"
+
+#include <algorithm>
+#include <numeric>
+
+bool DBCManager::open(const SourceSet &sources, const QString &dbc_file_name, QString *error) {
+  try {
+    auto it = std::find_if(dbc_files.begin(), dbc_files.end(),
+                           [&](auto &f) { return f.second && f.second->filename == dbc_file_name; });
+    auto file = (it != dbc_files.end()) ? it->second : std::make_shared<DBCFile>(dbc_file_name);
+    for (auto s : sources) {
+      dbc_files[s] = file;
+    }
+  } catch (std::exception &e) {
+    if (error) *error = e.what();
+    return false;
+  }
+
+  emit DBCFileChanged();
+  return true;
+}
+
+bool DBCManager::open(const SourceSet &sources, const QString &name, const QString &content, QString *error) {
+  try {
+    auto file = std::make_shared<DBCFile>(name, content);
+    for (auto s : sources) {
+      dbc_files[s] = file;
+    }
+  } catch (std::exception &e) {
+    if (error) *error = e.what();
+    return false;
+  }
+
+  emit DBCFileChanged();
+  return true;
+}
+
+void DBCManager::close(const SourceSet &sources) {
+  for (auto s : sources) {
+    dbc_files[s] = nullptr;
+  }
+  emit DBCFileChanged();
+}
+
+void DBCManager::close(DBCFile *dbc_file) {
+  for (auto &[_, f] : dbc_files) {
+    if (f.get() == dbc_file) f = nullptr;
+  }
+  emit DBCFileChanged();
+}
+
+void DBCManager::closeAll() {
+  dbc_files.clear();
+  emit DBCFileChanged();
+}
+
+void DBCManager::addSignal(const MessageId &id, const cabana::Signal &sig) {
+  if (auto m = msg(id)) {
+    if (auto s = m->addSignal(sig)) {
+      emit signalAdded(id, s);
+      emit maskUpdated();
+    }
+  }
+}
+
+void DBCManager::updateSignal(const MessageId &id, const QString &sig_name, const cabana::Signal &sig) {
+  if (auto m = msg(id)) {
+    if (auto s = m->updateSignal(sig_name, sig)) {
+      emit signalUpdated(s);
+      emit maskUpdated();
+    }
+  }
+}
+
+void DBCManager::removeSignal(const MessageId &id, const QString &sig_name) {
+  if (auto m = msg(id)) {
+    if (auto s = m->sig(sig_name)) {
+      emit signalRemoved(s);
+      m->removeSignal(sig_name);
+      emit maskUpdated();
+    }
+  }
+}
+
+void DBCManager::updateMsg(const MessageId &id, const QString &name, uint32_t size, const QString &node, const QString &comment) {
+  auto dbc_file = findDBCFile(id);
+  assert(dbc_file);  // This should be impossible
+  dbc_file->updateMsg(id, name, size, node, comment);
+  emit msgUpdated(id);
+}
+
+void DBCManager::removeMsg(const MessageId &id) {
+  auto dbc_file = findDBCFile(id);
+  assert(dbc_file);  // This should be impossible
+  dbc_file->removeMsg(id);
+  emit msgRemoved(id);
+  emit maskUpdated();
+}
+
+QString DBCManager::newMsgName(const MessageId &id) {
+  return QString("NEW_MSG_") + QString::number(id.address, 16).toUpper();
+}
+
+QString DBCManager::newSignalName(const MessageId &id) {
+  auto m = msg(id);
+  return m ? m->newSignalName() : "";
+}
+
+const std::vector<uint8_t> &DBCManager::mask(const MessageId &id) {
+  static std::vector<uint8_t> empty_mask;
+  auto m = msg(id);
+  return m ? m->mask : empty_mask;
+}
+
+const std::map<uint32_t, cabana::Msg> &DBCManager::getMessages(uint8_t source) {
+  static std::map<uint32_t, cabana::Msg> empty_msgs;
+  auto dbc_file = findDBCFile(source);
+  return dbc_file ? dbc_file->getMessages() : empty_msgs;
+}
+
+cabana::Msg *DBCManager::msg(const MessageId &id) {
+  auto dbc_file = findDBCFile(id);
+  return dbc_file ? dbc_file->msg(id) : nullptr;
+}
+
+cabana::Msg *DBCManager::msg(uint8_t source, const QString &name) {
+  auto dbc_file = findDBCFile(source);
+  return dbc_file ? dbc_file->msg(name) : nullptr;
+}
+
+QStringList DBCManager::signalNames() {
+  // Used for autocompletion
+  QStringList ret;
+  for (auto &f : allDBCFiles()) {
+    for (auto &[_, m] : f->getMessages()) {
+      for (auto sig : m.getSignals()) {
+        ret << sig->name;
+      }
+    }
+  }
+  ret.sort();
+  ret.removeDuplicates();
+  return ret;
+}
+
+int DBCManager::signalCount(const MessageId &id) {
+  auto m = msg(id);
+  return m ? m->sigs.size() : 0;
+}
+
+int DBCManager::signalCount() {
+  auto files = allDBCFiles();
+  return std::accumulate(files.cbegin(), files.cend(), 0, [](int &n, auto &f) { return n + f->signalCount(); });
+}
+
+int DBCManager::msgCount() {
+  auto files = allDBCFiles();
+  return std::accumulate(files.cbegin(), files.cend(), 0, [](int &n, auto &f) { return n + f->msgCount(); });
+}
+
+int DBCManager::dbcCount() {
+  return allDBCFiles().size();
+}
+
+int DBCManager::nonEmptyDBCCount() {
+  auto files = allDBCFiles();
+  return std::count_if(files.cbegin(), files.cend(), [](auto &f) { return !f->isEmpty(); });
+}
+
+DBCFile *DBCManager::findDBCFile(const uint8_t source) {
+  // Find DBC file that matches id.source, fall back to SOURCE_ALL if no specific DBC is found
+  auto it = dbc_files.count(source) ? dbc_files.find(source) : dbc_files.find(-1);
+  return it != dbc_files.end() ? it->second.get() : nullptr;
+}
+
+std::set<DBCFile *> DBCManager::allDBCFiles() {
+  std::set<DBCFile *> files;
+  for (const auto &[_, f] : dbc_files) {
+    if (f) files.insert(f.get());
+  }
+  return files;
+}
+
+const SourceSet DBCManager::sources(const DBCFile *dbc_file) const {
+  SourceSet sources;
+  for (auto &[s, f] : dbc_files) {
+    if (f.get() == dbc_file) sources.insert(s);
+  }
+  return sources;
+}
+
+QString toString(const SourceSet &ss) {
+  return std::accumulate(ss.cbegin(), ss.cend(), QString(), [](QString str, int source) {
+    if (!str.isEmpty()) str += ", ";
+    return str + (source == -1 ? QStringLiteral("all") : QString::number(source));
+  });
+}
+
+DBCManager *dbc() {
+  static DBCManager dbc_manager(nullptr);
+  return &dbc_manager;
+}

tools/cabana/dbc/dbcmanager.h

@@ -0,0 +1,74 @@
+#pragma once
+
+#include <QObject>
+#include <memory>
+#include <map>
+#include <set>
+#include <vector>
+
+#include "tools/cabana/dbc/dbcfile.h"
+
+typedef std::set<int> SourceSet;
+const SourceSet SOURCE_ALL = {-1};
+const int INVALID_SOURCE = 0xff;
+inline bool operator<(const std::shared_ptr<DBCFile> &l, const std::shared_ptr<DBCFile> &r) { return l.get() < r.get(); }
+
+class DBCManager : public QObject {
+  Q_OBJECT
+
+public:
+  DBCManager(QObject *parent) : QObject(parent) {}
+  ~DBCManager() {}
+  bool open(const SourceSet &sources, const QString &dbc_file_name, QString *error = nullptr);
+  bool open(const SourceSet &sources, const QString &name, const QString &content, QString *error = nullptr);
+  void close(const SourceSet &sources);
+  void close(DBCFile *dbc_file);
+  void closeAll();
+
+  void addSignal(const MessageId &id, const cabana::Signal &sig);
+  void updateSignal(const MessageId &id, const QString &sig_name, const cabana::Signal &sig);
+  void removeSignal(const MessageId &id, const QString &sig_name);
+
+  void updateMsg(const MessageId &id, const QString &name, uint32_t size, const QString &node, const QString &comment);
+  void removeMsg(const MessageId &id);
+
+  QString newMsgName(const MessageId &id);
+  QString newSignalName(const MessageId &id);
+  const std::vector<uint8_t>& mask(const MessageId &id);
+
+  const std::map<uint32_t, cabana::Msg> &getMessages(uint8_t source);
+  cabana::Msg *msg(const MessageId &id);
+  cabana::Msg* msg(uint8_t source, const QString &name);
+
+  QStringList signalNames();
+  int signalCount(const MessageId &id);
+  int signalCount();
+  int msgCount();
+  int dbcCount();
+  int nonEmptyDBCCount();
+
+  const SourceSet sources(const DBCFile *dbc_file) const;
+  DBCFile *findDBCFile(const uint8_t source);
+  inline DBCFile *findDBCFile(const MessageId &id) { return findDBCFile(id.source); }
+  std::set<DBCFile *> allDBCFiles();
+
+signals:
+  void signalAdded(MessageId id, const cabana::Signal *sig);
+  void signalRemoved(const cabana::Signal *sig);
+  void signalUpdated(const cabana::Signal *sig);
+  void msgUpdated(MessageId id);
+  void msgRemoved(MessageId id);
+  void DBCFileChanged();
+  void maskUpdated();
+
+private:
+  std::map<int, std::shared_ptr<DBCFile>> dbc_files;
+};
+
+DBCManager *dbc();
+
+QString toString(const SourceSet &ss);
+inline QString msgName(const MessageId &id) {
+  auto msg = dbc()->msg(id);
+  return msg ? msg->name : UNTITLED;
+}

tools/cabana/dbc/generate_dbc_json.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+import argparse
+import json
+
+from openpilot.selfdrive.car.car_helpers import get_interface_attr
+
+
+def generate_dbc_json() -> str:
+  all_cars_by_brand = get_interface_attr("CAR_INFO")
+  all_dbcs_by_brand = get_interface_attr("DBC")
+  dbc_map = {car: all_dbcs_by_brand[brand][car]['pt'] for brand, cars in all_cars_by_brand.items() for car in cars if car != 'mock'}
+  return json.dumps(dict(sorted(dbc_map.items())), indent=2)
+
+
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser(description="Generate mapping for all car fingerprints to DBC names and outputs json file",
+                                   formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+  parser.add_argument("--out", required=True, help="Generated json filepath")
+  args = parser.parse_args()
+
+  with open(args.out, 'w') as f:
+    f.write(generate_dbc_json())
+  print(f"Generated and written to {args.out}")