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

**

** 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 examples 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$

**

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

#include "tablemodel.h"

//! [0]

TableModel::TableModel(QObject *parent)

    : QAbstractTableModel(parent)

{

}

TableModel::TableModel(QList< QPair<QString, QString> > pairs, QObject *parent)

    : QAbstractTableModel(parent)

{

    listOfPairs=pairs;

}

//! [0]

//! [1]

int TableModel::rowCount(const QModelIndex &parent) const

{

    Q_UNUSED(parent);

    return listOfPairs.size();

}

int TableModel::columnCount(const QModelIndex &parent) const

{

    Q_UNUSED(parent);

    return 2;

}

//! [1]

//! [2]

QVariant TableModel::data(const QModelIndex &index, int role) const

{

    if (!index.isValid())

        return QVariant();

    if (index.row() >= listOfPairs.size() || index.row() < 0)

        return QVariant();

    if (role == Qt::DisplayRole) {

        QPair<QString, QString> pair = listOfPairs.at(index.row());

        if (index.column() == 0)

            return pair.first;

        else if (index.column() == 1)

            return pair.second;

    }

    return QVariant();

}

//! [2]

//! [3]

QVariant TableModel::headerData(int section, Qt::Orientation orientation, int role) const

{

    if (role != Qt::DisplayRole)

        return QVariant();

    if (orientation == Qt::Horizontal) {

        switch (section) {

            case 0:

                return tr("Name");

            case 1:

                return tr("Address");

            default:

                return QVariant();

        }

    }

    return QVariant();

}

//! [3]

//! [4]

bool TableModel::insertRows(int position, int rows, const QModelIndex &index)

{

    Q_UNUSED(index);

    beginInsertRows(QModelIndex(), position, position+rows-1);

    for (int row=0; row < rows; row++) {

        QPair<QString, QString> pair(" ", " ");

        listOfPairs.insert(position, pair);

    }

    endInsertRows();

    return true;

}

//! [4]

//! [5]

bool TableModel::removeRows(int position, int rows, const QModelIndex &index)

{

    Q_UNUSED(index);    

    beginRemoveRows(QModelIndex(), position, position+rows-1);

    for (int row=0; row < rows; ++row) {

        listOfPairs.removeAt(position);

    }

    endRemoveRows();

    return true;

}

//! [5]

//! [6]

bool TableModel::setData(const QModelIndex &index, const QVariant &value, int role)

{

	if (index.isValid() && role == Qt::EditRole) {

		int row = index.row();

		QPair<QString, QString> p = listOfPairs.value(row);

		if (index.column() == 0)

			p.first = value.toString();

		else if (index.column() == 1)

			p.second = value.toString();

        else

            return false;

        listOfPairs.replace(row, p);

		emit(dataChanged(index, index));

        return true;

	}

	return false;

}

//! [6]

//! [7]

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

{

    if (!index.isValid())

        return Qt::ItemIsEnabled;

    return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;

}

//! [7]

//! [8]

QList< QPair<QString, QString> > TableModel::getList()

{

    return listOfPairs;

}

//! [8]