SchemaCustomTableModel.cpp

Go to the documentation of this file.

// DUNE DAQ modification notice:
// This file has been modified from the original ATLAS dbe source for the DUNE DAQ project.
// Fork baseline commit: dbe-02-12-17 (2022-05-12).
// Renamed since fork: yes (from src/SchemaEditor/SchemaCustomTableModel.cpp to apps/SchemaEditor/SchemaCustomTableModel.cpp).
 
#include "dbe/SchemaCustomTableModel.hpp"
#include "dbe/SchemaKernelWrapper.hpp"
#include "dbe/SchemaStyle.hpp"
#include <QIODevice>
#include <QDataStream>
 
using namespace dunedaq::oks;
 
dbse::CustomTableModel::CustomTableModel ( QStringList Headers, QObject * parent )
  : QAbstractTableModel ( parent ),
    HeaderList ( Headers )
{
  setupModel();
}
 
dbse::CustomTableModel::~CustomTableModel()
{
}
 
int dbse::CustomTableModel::rowCount ( const QModelIndex & parent ) const
{
  Q_UNUSED ( parent );
  return m_data.size();
}
 
int dbse::CustomTableModel::columnCount ( const QModelIndex & parent ) const
{
  Q_UNUSED ( parent );
  return HeaderList.size();
}
 
Qt::ItemFlags dbse::CustomTableModel::flags ( const QModelIndex & index ) const
{
  Q_UNUSED ( index );
  return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
}
 
QVariant dbse::CustomTableModel::headerData ( int section, Qt::Orientation orientation,
                                              int role ) const
{
  if ( role != Qt::DisplayRole )
  {
    return QVariant();
  }
 
  if ( orientation == Qt::Horizontal )
  {
    return HeaderList.at ( section );
  }
 
  return QVariant();
}
 
QVariant dbse::CustomTableModel::data ( const QModelIndex & index, int role ) const
{
  if ( role == Qt::DisplayRole )
  {
    return m_data.value ( index.row() ).value ( index.column() );
  }
  if ( role == Qt::ToolTipRole )
  {
    return m_tooltips.value ( index.row() ).value ( index.column() );
  }
  if (role == Qt::ForegroundRole) {
    return m_brushes.at(index.row());
  }
  if (role == Qt::BackgroundRole) {
    return m_backgrounds.at(index.row());
  }
 
  return QVariant();
}
 
QStringList dbse::CustomTableModel::getRowFromIndex ( QModelIndex & index )
{
  if ( !index.isValid() )
  {
    return QStringList();
  }
 
  return m_data.at ( index.row() );
}
 
void dbse::CustomTableModel::setupModel()
{
  std::vector<OksClass *> ClassList;
  KernelWrapper::GetInstance().GetClassList ( ClassList );
 
  for ( unsigned int i = 0; i < ClassList.size(); ++i )
  {
    QList<QString> Row;
    OksClass * Class = ClassList.at ( i );
    Row.append ( QString ( Class->get_name().c_str() ) );
    m_data.append ( Row );
    m_tooltips.append (QStringList {QString ( Class->get_description().c_str() )});
 
    auto fn = Class->get_file()->get_full_file_name();
    if (!KernelWrapper::GetInstance().IsFileWritable (fn)) {
      m_brushes.emplace_back(QBrush(SchemaStyle::get_color("foreground", "readonly")));
      m_backgrounds.emplace_back(SchemaStyle::get_color("background", "readonly"));
    }
    else if (fn == KernelWrapper::GetInstance().GetActiveSchema()) {
      m_brushes.emplace_back(QBrush(SchemaStyle::get_color("foreground", "active_file")));
      m_backgrounds.emplace_back(SchemaStyle::get_color("background", "active_file"));
    }
    else {
      m_brushes.emplace_back(QBrush(SchemaStyle::get_color("foreground", "default")));
      m_backgrounds.emplace_back(SchemaStyle::get_color("background", "default"));
    }
  }
}
 
QStringList dbse::CustomTableModel::mimeTypes() const
{
  QStringList types;
  types << "application/vnd.text.list";
  return types;
}
 
QMimeData * dbse::CustomTableModel::mimeData ( const QModelIndexList & indexes ) const
{
  QMimeData * mimeData = new QMimeData();
  QByteArray encodedData;
 
  QDataStream stream ( &encodedData, QIODevice::WriteOnly );
 
  foreach ( QModelIndex index, indexes )
  {
    if ( index.isValid() && index.column() == 0 )
    {
      QString ClassName = data ( index, Qt::DisplayRole ).toString();
      stream << ClassName;
    }
  }
 
  mimeData->setData ( "application/vnd.text.list", encodedData );
  return mimeData;
}