openpilot_comma/tools/cabana/dbc/dbcfile.h

#pragma once

#include <map>
#include <QTextStream>

#include "tools/cabana/dbc/dbc.h"

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 writeContents(const QString &fn);
  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); }
  cabana::Signal *signal(uint32_t address, const QString &name);

  inline QString name() const { return name_.isEmpty() ? "untitled" : name_; }
  inline bool isEmpty() const { return msgs.empty() && name_.isEmpty(); }

  QString filename;

private:
  void parse(const QString &content);
  cabana::Msg *parseBO(const QString &line);
  void parseSG(const QString &line, cabana::Msg *current_msg, int &multiplexor_cnt);
  void parseCM_BO(const QString &line, const QString &content, const QString &raw_line, const QTextStream &stream);
  void parseCM_SG(const QString &line, const QString &content, const QString &raw_line, const QTextStream &stream);
  void parseVAL(const QString &line);

  QString header;
  std::map<uint32_t, cabana::Msg> msgs;
  QString name_;
};