/****************************************************************************

**

** All rights reserved.

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

**

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

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

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

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

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** 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 as published by the Free Software

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

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

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

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

**

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

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

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

**

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

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

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#ifndef QSQLRELATIONALDELEGATE_H

#define QSQLRELATIONALDELEGATE_H

#ifdef QT_GUI_LIB

#include <QtGui/qitemdelegate.h>

#include <QtGui/qlistview.h>

#include <QtGui/qcombobox.h>

#include <QtSql/qsqlrelationaltablemodel.h>

QT_BEGIN_HEADER

QT_BEGIN_NAMESPACE

QT_MODULE(Sql)

class QSqlRelationalDelegate: public QItemDelegate

{

public:

explicit QSqlRelationalDelegate(QObject *parent = 0)

    : QItemDelegate(parent)

{}

~QSqlRelationalDelegate()

{}

QWidget *createEditor(QWidget *parent,

                      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(parent, option, index);

    QComboBox *combo = new QComboBox(parent);

    combo->setModel(childModel);

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

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

    return combo;

}

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

{

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

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

    if (!sqlModel || !combo) {

        QItemDelegate::setEditorData(editor, index);

        return;

    }

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

}

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

QT_END_HEADER

#endif // QT_GUI_LIB

#endif // QSQLRELATIONALDELEGATE_H