qxrdmaskstackmodel.cpp

Go to the documentation of this file.

#include "qxrdmaskstackmodel.h"
#include "qxrdmaskstack.h"
#include "qcepmaskdata.h"

QxrdMaskStackModel::QxrdMaskStackModel(QxrdMaskStackPtr masks) :
    m_MaskStack(masks)
{
  connect(masks, &QxrdMaskStack::maskChanged, this, &QxrdMaskStackModel::onMaskChanged);
}

int QxrdMaskStackModel::rowCount (const QModelIndex & parent) const
{
  if (parent.isValid()) {
    return 0;
  }

  return m_MaskStack->count();
}

int QxrdMaskStackModel::columnCount(const QModelIndex & parent) const
{
  if (parent.isValid()) {
    return 0;
  }

  return NumColumns;
}

QVariant QxrdMaskStackModel::data (const QModelIndex & index, int role) const
{
  if (index.row() < 0 || index.row() >= m_MaskStack->count()) {
    return QVariant();
  }

  QcepMaskDataPtr p = m_MaskStack->at(index.row());

  if (p) {
    int col = index.column();

    if (columnCount()==1) {
      if (col == 0) {
        if (role == Qt::DecorationRole) {
          return p->thumbnailImage();
        } else if (role == Qt::CheckStateRole) {
          return (p->get_Used()?Qt::Checked:Qt::Unchecked);
        } else if (role == Qt::DisplayRole || role == Qt::EditRole) {
          //        return tr("%1 : %2").arg(m_MaskStack->stackLevelName(index.row())).arg(p->get_Title());
          return p->get_Title();
        }
      }
    } else {
      if (col == ThumbnailColumn) {
        if (role == Qt::DecorationRole) {
          return p->thumbnailImage();
        } else if (role == Qt::SizeHintRole) {
          return p->thumbnailImageSize();
        }
      } else if (col == VisibilityColumn) {
        if (role == Qt::CheckStateRole) {
          return (p->get_Used()?Qt::Checked:Qt::Unchecked);
        } else if (role == Qt::SizeHintRole) {
          return 30;
        }
      } else if (col == TitleColumn) {
        if (role == Qt::DisplayRole || role == Qt::EditRole) {
          return p->get_Title();
        } else if (role==Qt::SizeHintRole) {
          return 120;
        }
      }
    }
  }

  return QVariant();
}

QVariant QxrdMaskStackModel::headerData ( int section, Qt::Orientation orientation, int role) const
{
  if (columnCount() == 1) {
    return QVariant();
  } else {
    if (orientation==Qt::Horizontal && role==Qt::DisplayRole) {
      switch (section) {
      case VisibilityColumn:
        return "Vis";

      case ThumbnailColumn:
        return "";

      case TitleColumn:
        return "Name";
      }
    } else if (orientation==Qt::Vertical && role==Qt::DisplayRole) {
      return QxrdMaskStack::stackLevelName(section);
    }
  }

  return QVariant();
}

Qt::ItemFlags QxrdMaskStackModel::flags (const QModelIndex & index) const
{
  if (!index.isValid()) {
    return QAbstractItemModel::flags(index) | Qt::ItemIsDropEnabled;
  }

  return QAbstractItemModel::flags(index) | Qt::ItemIsSelectable |Qt::ItemIsEditable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled | Qt::ItemIsUserCheckable;
}

bool QxrdMaskStackModel::insertRows ( int row, int count, const QModelIndex & /*parent*/)
{
  printf("QxrdMaskStackModel::insertRows(%d,%d,...)\n", row, count);

  return false;
}

bool QxrdMaskStackModel::removeRows ( int row, int count, const QModelIndex & /*parent*/)
{
  printf("QxrdMaskStackModel::removeRows(%d,%d,...)\n", row, count);

  return false;
}

bool QxrdMaskStackModel::setData ( const QModelIndex & index, const QVariant & value, int role)
{
  if (columnCount() == 1) {
    if (index.column() == 0) {
      if ((index.row() >= 0) && (index.row() < m_MaskStack->count())) {
        QcepMaskDataPtr p = m_MaskStack->at(index.row());

        if ((role == Qt::EditRole || role == Qt::DisplayRole)) {
          if (p) {
            p->set_Title(value.toString());
          }

          emit dataChanged(index, index);

          return true;
        }

        if (role == Qt::CheckStateRole) {
          if (p) {
            p->set_Used(!(p->get_Used()));
          }

          emit dataChanged(index, index);

          return true;
        }
      }
    }
  } else {
    if ((index.row() >= 0) && (index.row() < m_MaskStack->count())) {
      QcepMaskDataPtr p = m_MaskStack->at(index.row());

      if (index.column() == TitleColumn && (role == Qt::EditRole || role == Qt::DisplayRole)) {
        if (p) {
          p->set_Title(value.toString());
        }

        emit dataChanged(index, index);

        return true;
      }

      if (index.column() == VisibilityColumn && (role == Qt::CheckStateRole)) {
        if (p) {
          p->set_Used(!(p->get_Used()));
        }

        emit dataChanged(index, index);

        return true;
      }
    }
  }

  return false;
}

//void QxrdMaskStackModel::sort ( int column, Qt::SortOrder order)
//{
//}

Qt::DropActions QxrdMaskStackModel::supportedDropActions () const
{
  return QAbstractItemModel::supportedDropActions() | Qt::MoveAction;
}

QxrdMaskStackPtr QxrdMaskStackModel::maskStack()
{
  return m_MaskStack;
}

void QxrdMaskStackModel::setMaskStack(QxrdMaskStackPtr masks)
{
  emit beginResetModel();

  m_MaskStack = masks;

  emit endResetModel();
}

void QxrdMaskStackModel::onMaskChanged()
{
  emit beginResetModel();

  emit endResetModel();
}