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

**

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

** All rights reserved.

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

**

** This file is part of the documentation 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$

**

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

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

**

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

** All rights reserved.

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

**

** This file is part of an example program for Qt.

** EDITIONS: NOLIMITS

**

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

#include <QtGui>

#include "dragdropmodel.h"

DragDropModel::DragDropModel(const QStringList &strings, QObject *parent)

    : TreeModel(strings, parent)

{

}

bool DragDropModel::dropMimeData(const QMimeData *data,

    Qt::DropAction action, int row, int column, const QModelIndex &parent)

{

    if (action == Qt::IgnoreAction)

        return true;

    if (!data->hasFormat("text/plain"))

        return false;

    int beginRow;

    if (row != -1)

        beginRow = row;

    else if (parent.isValid())

        beginRow = 0;

    else

        beginRow = rowCount(QModelIndex());

    QByteArray encodedData = data->data("text/plain");

    QDataStream stream(&encodedData, QIODevice::ReadOnly);

    QHash<qint64, QMap<int,QHash<int,QString> > > newItems;

    while (!stream.atEnd()) {

        qint64 id;

        int row;

        int column;

        QString text;

        stream >> id >> row >> column >> text;

        newItems[id][row][column] = text;

    }

    int rows = newItems.count();

    insertRows(beginRow, rows, parent);

    QMap<int,QHash<int,QString> > childItems;

    foreach (childItems, newItems.values()) {

        QHash<int,QString> rowItems;

        foreach (rowItems, childItems.values()) {

            foreach (int column, rowItems.keys()) {

                QModelIndex idx = index(beginRow, column, parent);

                setData(idx, rowItems[column]);

            }

            ++beginRow;

        }

    }

    return true;

}

Qt::ItemFlags DragDropModel::flags(const QModelIndex &index) const

{

    Qt::ItemFlags defaultFlags = TreeModel::flags(index);

    if (index.isValid())

        return Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled | defaultFlags;

    else

        return Qt::ItemIsDropEnabled | defaultFlags;

}

QMimeData *DragDropModel::mimeData(const QModelIndexList &indexes) const

{

    QMimeData *mimeData = new QMimeData();

    QByteArray encodedData;

    QDataStream stream(&encodedData, QIODevice::WriteOnly);

    foreach (QModelIndex index, indexes) {

        if (index.isValid()) {

            QString text = data(index, Qt::DisplayRole).toString();

            stream << index.internalId() << index.row() << index.column() << text;

        }

    }

    mimeData->setData("text/plain", encodedData);

    return mimeData;

}

QStringList DragDropModel::mimeTypes() const

{

    QStringList types;

    types << "text/plain";

    return types;

}

Qt::DropActions DragDropModel::supportedDropActions() const

{

    return Qt::CopyAction | Qt::MoveAction;

}