LibrePCB

/*
 * LibrePCB - Professional EDA for everyone!
 * Copyright (C) 2013 LibrePCB Developers, see AUTHORS.md for contributors.
 * https://librepcb.org/
 *
 * 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/>.
 */

/*******************************************************************************
 *  Includes
 ******************************************************************************/
#include "comboboxdelegate.h"

#include <librepcb/core/utils/toolbox.h>

#include <QtCore>

/*******************************************************************************
 *  Namespace
 ******************************************************************************/
namespace librepcb {
namespace editor {

/*******************************************************************************
 *  Class ComboBoxDelegate::Items
 ******************************************************************************/

void ComboBoxDelegate::Items::sort() noexcept {
  Toolbox::sortNumeric(
      *this,
      [](const QCollator& cmp, const Item& lhs, const Item& rhs) {
        return cmp(lhs.text, rhs.text);
      },
      Qt::CaseInsensitive, false);
}

/*******************************************************************************
 *  Constructors / Destructor
 ******************************************************************************/

ComboBoxDelegate::ComboBoxDelegate(bool editable, QObject* parent) noexcept
  : QStyledItemDelegate(parent), mEditable(editable) {
}

ComboBoxDelegate::~ComboBoxDelegate() noexcept {
}

/*******************************************************************************
 *  Inherited from QStyledItemDelegate
 ******************************************************************************/

QWidget* ComboBoxDelegate::createEditor(QWidget* parent,
                                        const QStyleOptionViewItem& option,
                                        const QModelIndex& index) const {
  Q_UNUSED(option);
  Q_UNUSED(index);
  QComboBox* cbx = new QComboBox(parent);
  cbx->setFrame(false);
  cbx->setSizePolicy(QSizePolicy::Ignored, QSizePolicy::Ignored);
  cbx->setEditable(mEditable);
  cbx->setAutoFillBackground(true);  // Fix transparency on macOS.
  Items items = index.data(Qt::UserRole).value<Items>();
  foreach (const auto& item, items) {
    cbx->addItem(item.icon, item.text, item.data);
  }
  return cbx;
}

void ComboBoxDelegate::setEditorData(QWidget* editor,
                                     const QModelIndex& index) const {
  QComboBox* cbx = static_cast<QComboBox*>(editor);
  QVariant data = index.data(Qt::EditRole);
  int i = cbx->findData(data, Qt::UserRole);
  if ((i >= 0) || (!mEditable)) {
    cbx->setCurrentIndex(i);
  } else {
    cbx->setCurrentText(data.toString());
  }
}

void ComboBoxDelegate::setModelData(QWidget* editor, QAbstractItemModel* model,
                                    const QModelIndex& index) const {
  QComboBox* cbx = static_cast<QComboBox*>(editor);
  if (mEditable) {
    model->setData(index, cbx->currentText(), Qt::EditRole);
  } else {
    model->setData(index, cbx->currentData(Qt::UserRole), Qt::EditRole);
  }
}

void ComboBoxDelegate::updateEditorGeometry(QWidget* editor,
                                            const QStyleOptionViewItem& option,
                                            const QModelIndex& index) const {
  Q_UNUSED(index);
  editor->setGeometry(option.rect);
}

/*******************************************************************************
 *  End of File
 ******************************************************************************/

}  // namespace editor
}  // namespace librepcb