1 /****************************************************************************

     2 **

     3 ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

     4 ** All rights reserved.

     5 ** Contact: Nokia Corporation (qt-info@nokia.com)

     6 **

     7 ** This file is part of the QtSql module of the Qt Toolkit.

     8 **

     9 ** $QT_BEGIN_LICENSE:LGPL$

    10 ** No Commercial Usage

    11 ** This file contains pre-release code and may not be distributed.

    12 ** You may use this file in accordance with the terms and conditions

    13 ** contained in the Technology Preview License Agreement accompanying

    14 ** this package.

    15 **

    16 ** GNU Lesser General Public License Usage

    17 ** Alternatively, this file may be used under the terms of the GNU Lesser

    18 ** General Public License version 2.1 as published by the Free Software

    19 ** Foundation and appearing in the file LICENSE.LGPL included in the

    20 ** packaging of this file.  Please review the following information to

    21 ** ensure the GNU Lesser General Public License version 2.1 requirements

    22 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

    23 **

    24 ** In addition, as a special exception, Nokia gives you certain additional

    25 ** rights.  These rights are described in the Nokia Qt LGPL Exception

    26 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

    27 **

    28 ** If you have questions regarding the use of this file, please contact

    29 ** Nokia at qt-info@nokia.com.

    30 **

    31 **

    32 **

    33 **

    34 **

    35 **

    36 **

    37 **

    38 ** $QT_END_LICENSE$

    39 **

    40 ****************************************************************************/

    41 

    42 #ifndef QSQLRELATIONALDELEGATE_H

    43 #define QSQLRELATIONALDELEGATE_H

    44 

    45 #ifdef QT_GUI_LIB

    46 

    47 #include <QtGui/qitemdelegate.h>

    48 #include <QtGui/qlistview.h>

    49 #include <QtGui/qcombobox.h>

    50 #include <QtSql/qsqlrelationaltablemodel.h>

    51 

    52 QT_BEGIN_HEADER

    53 

    54 QT_BEGIN_NAMESPACE

    55 

    56 QT_MODULE(Sql)

    57 

    58 class QSqlRelationalDelegate: public QItemDelegate

    59 {

    60 public:

    61 

    62 explicit QSqlRelationalDelegate(QObject *parent = 0)

    63     : QItemDelegate(parent)

    64 {}

    65 

    66 ~QSqlRelationalDelegate()

    67 {}

    68 

    69 QWidget *createEditor(QWidget *parent,

    70                       const QStyleOptionViewItem &option,

    71                       const QModelIndex &index) const

    72 {

    73     const QSqlRelationalTableModel *sqlModel = qobject_cast<const QSqlRelationalTableModel *>(index.model());

    74     QSqlTableModel *childModel = sqlModel ? sqlModel->relationModel(index.column()) : 0;

    75     if (!childModel)

    76         return QItemDelegate::createEditor(parent, option, index);

    77 

    78     QComboBox *combo = new QComboBox(parent);

    79     combo->setModel(childModel);

    80     combo->setModelColumn(childModel->fieldIndex(sqlModel->relation(index.column()).displayColumn()));

    81     combo->installEventFilter(const_cast<QSqlRelationalDelegate *>(this));

    82 

    83     return combo;

    84 }

    85 

    86 void setEditorData(QWidget *editor, const QModelIndex &index) const

    87 {

    88     const QSqlRelationalTableModel *sqlModel = qobject_cast<const QSqlRelationalTableModel *>(index.model());

    89     QComboBox *combo = qobject_cast<QComboBox *>(editor);

    90     if (!sqlModel || !combo) {

    91         QItemDelegate::setEditorData(editor, index);

    92         return;

    93     }

    94     combo->setCurrentIndex(combo->findText(sqlModel->data(index).toString()));

    95 }

    96 

    97 void setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const

    98 {

    99     if (!index.isValid())

   100         return;

   101 

   102     QSqlRelationalTableModel *sqlModel = qobject_cast<QSqlRelationalTableModel *>(model);

   103     QSqlTableModel *childModel = sqlModel ? sqlModel->relationModel(index.column()) : 0;

   104     QComboBox *combo = qobject_cast<QComboBox *>(editor);

   105     if (!sqlModel || !childModel || !combo) {

   106         QItemDelegate::setModelData(editor, model, index);

   107         return;

   108     }

   109 

   110     int currentItem = combo->currentIndex();

   111     int childColIndex = childModel->fieldIndex(sqlModel->relation(index.column()).displayColumn());

   112     int childEditIndex = childModel->fieldIndex(sqlModel->relation(index.column()).indexColumn());

   113     sqlModel->setData(index,

   114             childModel->data(childModel->index(currentItem, childColIndex), Qt::DisplayRole),

   115             Qt::DisplayRole);

   116     sqlModel->setData(index,

   117             childModel->data(childModel->index(currentItem, childEditIndex), Qt::EditRole),

   118             Qt::EditRole);

   119 }

   120 

   121 };

   122 

   123 QT_END_NAMESPACE

   124 

   125 QT_END_HEADER

   126 

   127 #endif // QT_GUI_LIB

   128 

   129 #endif // QSQLRELATIONALDELEGATE_H