SchemaCustomFileModel.cpp

Go to the documentation of this file.

#include "dbe/SchemaCustomFileModel.hpp"
#include "dbe/SchemaKernelWrapper.hpp"
#include "dbe/SchemaStyle.hpp"
#include <vector>
 
#include <QBrush>
#include <QColor>
#include <QSize>
 
dbse::CustomFileModel::CustomFileModel ( QStringList & Headers, QObject * parent )
  : QAbstractTableModel ( parent ),
    HeaderList ( Headers )
{
  setupModel();
}
 
dbse::CustomFileModel::~CustomFileModel()
{
}
 
int dbse::CustomFileModel::rowCount ( const QModelIndex & parent ) const
{
  Q_UNUSED ( parent );
  return Data.size();
}
 
int dbse::CustomFileModel::columnCount ( const QModelIndex & parent ) const
{
  Q_UNUSED ( parent );
  return HeaderList.size();
}
 
// Qt::ItemFlags dbse::CustomFileModel::flags ( const QModelIndex & index ) const
// {
//   // if ( Data.value (index.row() ).value ( 1) == "RW" ) {
//     return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
//   // }
//   // else {
//   //   return Qt::NoItemFlags;
//   // }
// }
 
QVariant dbse::CustomFileModel::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::CustomFileModel::data ( const QModelIndex & index, int role ) const
{
  if ( role == Qt::DisplayRole ) {
    return Data.value(index.row()).value(index.column());
  }
  if (role == Qt::ForegroundRole) {
    if (Data.value(index.row()).value(2).contains("Active")) {
      return QBrush(QColor (SchemaStyle::get_color("foreground", "active_file")));
    }
    if (Data.value(index.row()).value(1) == "RW" ) {
      return QBrush(QColor (SchemaStyle::get_color("foreground", "default")));
    }
    return QBrush(SchemaStyle::get_color("foreground", "readonly"));
  }
  if (role == Qt::BackgroundRole) {
    if (Data.value(index.row()).value(2).contains("Active")) {
      return QBrush(QColor (SchemaStyle::get_color("background", "active_file")));
    }
    if (Data.value(index.row()).value(1) == "RW" ) {
      return QBrush(QColor (SchemaStyle::get_color("background", "default")));
    }
    return QBrush(SchemaStyle::get_color("background", "readonly"));
  }
  if (role == Qt::ToolTipRole) {
    return Data.value(index.row()).value(0);
  }
 
  return QVariant();
}
 
void dbse::CustomFileModel::setupModel()
{
  std::vector<std::string> SchemaFiles;
  KernelWrapper::GetInstance().GetSchemaFiles ( SchemaFiles );
  std::string ActiveSchema = KernelWrapper::GetInstance().GetActiveSchema();
  std::string Modified = KernelWrapper::GetInstance().ModifiedSchemaFiles();
  for ( std::string & FileName : SchemaFiles )
  {
    QStringList Row;
    Row.append ( QString::fromStdString ( FileName ) );
 
    if ( KernelWrapper::GetInstance().IsFileWritable ( FileName ) )
    {
      Row.append ( "RW" );
    }
    else
    {
      Row.append ( "RO" );
    }
 
    std::string Status = "";
    if ( Modified.find( FileName ) != std::string::npos) {
      Status = "Modified";
    }
    if ( FileName == ActiveSchema) {
      if ( !Status.empty() ) {
        Status += "  ";
      }
      Status += "Active";
    }
 
    Row.append ( QString::fromStdString ( Status ) );
    Data.append ( Row );
  }
}
 
QStringList dbse::CustomFileModel::getRowFromIndex ( QModelIndex & index )
{
  if ( !index.isValid() )
  {
    return QStringList();
  }
 
  return Data.at ( index.row() );
}