lcd-image-converter/classes/data/canvasmodproxy.cpp

/*
 * LCD Image Converter. Converts images and fonts for embedded applications.
 * Copyright (C) 2019 riuson
 * mailto: riuson@gmail.com
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/
 */

#include "canvasmodproxy.h"

#include <QColor>

#include "bitmaphelper.h"
#include "canvasmodinfo.h"
#include "imagesmodel.h"

namespace Data
{
namespace Models
{

CanvasModProxy::CanvasModProxy(QMap<QString, CanvasModInfo*>* canvasMods, QObject* parent)
    : QSortFilterProxyModel(parent), mCanvasMods(canvasMods)
{
}

QVariant CanvasModProxy::headerData(int section, Qt::Orientation orientation, int role) const
{
  QVariant result = this->sourceModel()->headerData(section, orientation, role);

  if (role == Qt::DisplayRole) {
    int sourceColumns = this->sourceModel()->columnCount();

    if (orientation == Qt::Horizontal) {
      if (section == sourceColumns + 0) {
        result = tr("Original size");
      } else if (section == sourceColumns + 1) {
        result = tr("New size");
      }
    }
  }

  return result;
}

int CanvasModProxy::columnCount(const QModelIndex& parent) const
{
  if (this->sourceModel() == nullptr) {
    return 0;
  }

  int sourceColumns = this->sourceModel()->columnCount(parent);
  return sourceColumns + 2;
}

QVariant CanvasModProxy::data(const QModelIndex& index, int role) const
{
  QVariant result = this->sourceModel()->data(index, role);

  if (!index.isValid()) {
    return result;
  }

  int sourceColumns = this->sourceModel()->columnCount();
  int columnIndex = index.column();

  switch (role) {
    case Qt::DecorationRole: {
      if (columnIndex == 1) {
        QImage imageSource = result.value<QImage>();
        QColor backgroundColor = Parsing::Conversion::BitmapHelper::detectBackgroundColor(&imageSource);

        QVariant varKey = this->sourceModel()->data(index, Data::Models::ImagesModel::KeyRole);
        QString key = varKey.toString();

        const Data::CanvasModInfo* modInfo = this->mCanvasMods->value(key);
        Data::CanvasModInfo::Mods mods;

        if (modInfo != nullptr) {
          mods = modInfo->summary();
        } else {
          mods.reset();
        }

        QImage imageScaled = Parsing::Conversion::BitmapHelper::crop(&imageSource, mods.left, mods.top, mods.right,
                                                                     mods.bottom, backgroundColor);
        result = imageScaled;
      }

      break;
    }

    case Qt::SizeHintRole: {
      if (columnIndex == 1) {
        QSize size = result.toSize();

        QVariant varKey = this->sourceModel()->data(index, Data::Models::ImagesModel::KeyRole);
        QString key = varKey.toString();

        const Data::CanvasModInfo* modInfo = this->mCanvasMods->value(key);
        Data::CanvasModInfo::Mods mods;

        if (modInfo != nullptr) {
          mods = modInfo->summary();
        } else {
          mods.reset();
        }

        size.rwidth() += mods.left + mods.right;
        size.rheight() += mods.top + mods.bottom;

        result = size;
      }

      break;
    }

    case Qt::DisplayRole: {
      if (columnIndex == sourceColumns + 0) {
        QVariant varSize = this->sourceModel()->data(index, Data::Models::ImagesModel::ImageSizeRole);
        QSize size = varSize.toSize();
        result = QString("%1x%2").arg(size.width()).arg(size.height());
      } else if (columnIndex == sourceColumns + 1) {
        QVariant varKey = this->sourceModel()->data(index, Data::Models::ImagesModel::KeyRole);
        QString key = varKey.toString();

        const Data::CanvasModInfo* modInfo = this->mCanvasMods->value(key);
        Data::CanvasModInfo::Mods mods;

        if (modInfo != nullptr) {
          mods = modInfo->summary();
        } else {
          mods.reset();
        }

        QVariant varSize = this->sourceModel()->data(index, Data::Models::ImagesModel::ImageSizeRole);
        QSize size = varSize.toSize();

        size.setWidth(size.width() + mods.left + mods.right);
        size.setHeight(size.height() + mods.top + mods.bottom);
        result = QString("%1x%2").arg(size.width()).arg(size.height());
      }

      break;
    }

    default:
      break;
  }

  return result;
}

QModelIndex CanvasModProxy::index(int row, int column, const QModelIndex& parent) const
{
  Q_UNUSED(parent)
  return this->createIndex(row, column);
}

QModelIndex CanvasModProxy::parent(const QModelIndex& index) const
{
  Q_UNUSED(index)
  return QModelIndex();
}

QModelIndex CanvasModProxy::mapFromSource(const QModelIndex& sourceIndex) const
{
  return this->index(sourceIndex.row(), sourceIndex.column(), sourceIndex.parent());
}

QModelIndex CanvasModProxy::mapToSource(const QModelIndex& proxyIndex) const
{
  if (sourceModel() && proxyIndex.isValid()) {
    return sourceModel()->index(proxyIndex.row(), proxyIndex.column(), proxyIndex.parent());
  } else {
    return QModelIndex();
  }
}

} // namespace Models
} // namespace Data