| /**************************************************************************** |
| ** |
| ** Copyright (C) 2015 The Qt Company Ltd. |
| ** Contact: http://www.qt.io/licensing/ |
| ** |
| ** This file is part of the QtSql module of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:LGPL21$ |
| ** Commercial License Usage |
| ** Licensees holding valid commercial Qt licenses may use this file in |
| ** accordance with the commercial license agreement provided with the |
| ** Software or, alternatively, in accordance with the terms contained in |
| ** a written agreement between you and The Qt Company. For licensing terms |
| ** and conditions see http://www.qt.io/terms-conditions. For further |
| ** information use the contact form at http://www.qt.io/contact-us. |
| ** |
| ** GNU Lesser General Public License Usage |
| ** Alternatively, this file may be used under the terms of the GNU Lesser |
| ** General Public License version 2.1 or version 3 as published by the Free |
| ** Software Foundation and appearing in the file LICENSE.LGPLv21 and |
| ** LICENSE.LGPLv3 included in the packaging of this file. Please review the |
| ** following information to ensure the GNU Lesser General Public License |
| ** requirements will be met: https://www.gnu.org/licenses/lgpl.html and |
| ** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. |
| ** |
| ** As a special exception, The Qt Company gives you certain additional |
| ** rights. These rights are described in The Qt Company LGPL Exception |
| ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package. |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| #ifndef QSQLRELATIONALDELEGATE_H |
| #define QSQLRELATIONALDELEGATE_H |
| |
| #ifdef QT_WIDGETS_LIB |
| |
| #include <QtWidgets/qitemdelegate.h> |
| #include <QtWidgets/qlistview.h> |
| #include <QtWidgets/qcombobox.h> |
| #include <QtSql/qsqlrelationaltablemodel.h> |
| |
| QT_BEGIN_NAMESPACE |
| |
| |
| class QSqlRelationalDelegate: public QItemDelegate |
| { |
| public: |
| |
| explicit QSqlRelationalDelegate(QObject *aParent = 0) |
| : QItemDelegate(aParent) |
| {} |
| |
| ~QSqlRelationalDelegate() |
| {} |
| |
| QWidget *createEditor(QWidget *aParent, |
| const QStyleOptionViewItem &option, |
| const QModelIndex &index) const |
| { |
| const QSqlRelationalTableModel *sqlModel = qobject_cast<const QSqlRelationalTableModel *>(index.model()); |
| QSqlTableModel *childModel = sqlModel ? sqlModel->relationModel(index.column()) : 0; |
| if (!childModel) |
| return QItemDelegate::createEditor(aParent, option, index); |
| |
| QComboBox *combo = new QComboBox(aParent); |
| combo->setModel(childModel); |
| combo->setModelColumn(childModel->fieldIndex(sqlModel->relation(index.column()).displayColumn())); |
| combo->installEventFilter(const_cast<QSqlRelationalDelegate *>(this)); |
| |
| return combo; |
| } |
| |
| void setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const |
| { |
| if (!index.isValid()) |
| return; |
| |
| QSqlRelationalTableModel *sqlModel = qobject_cast<QSqlRelationalTableModel *>(model); |
| QSqlTableModel *childModel = sqlModel ? sqlModel->relationModel(index.column()) : 0; |
| QComboBox *combo = qobject_cast<QComboBox *>(editor); |
| if (!sqlModel || !childModel || !combo) { |
| QItemDelegate::setModelData(editor, model, index); |
| return; |
| } |
| |
| int currentItem = combo->currentIndex(); |
| int childColIndex = childModel->fieldIndex(sqlModel->relation(index.column()).displayColumn()); |
| int childEditIndex = childModel->fieldIndex(sqlModel->relation(index.column()).indexColumn()); |
| sqlModel->setData(index, |
| childModel->data(childModel->index(currentItem, childColIndex), Qt::DisplayRole), |
| Qt::DisplayRole); |
| sqlModel->setData(index, |
| childModel->data(childModel->index(currentItem, childEditIndex), Qt::EditRole), |
| Qt::EditRole); |
| } |
| |
| }; |
| |
| QT_END_NAMESPACE |
| |
| #endif // QT_WIDGETS_LIB |
| |
| #endif // QSQLRELATIONALDELEGATE_H |