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

**

** 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 <QtGui>

#include <QtXml>

#include "domitem.h"

#include "dommodel.h"

//! [0]

DomModel::DomModel(QDomDocument document, QObject *parent)

    : QAbstractItemModel(parent), domDocument(document)

{

    rootItem = new DomItem(domDocument, 0);

}

//! [0]

//! [1]

DomModel::~DomModel()

{

    delete rootItem;

}

//! [1]

//! [2]

int DomModel::columnCount(const QModelIndex &/*parent*/) const

{

    return 3;

}

//! [2]

//! [3]

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

{

    if (!index.isValid())

        return QVariant();

    if (role != Qt::DisplayRole)

        return QVariant();

    DomItem *item = static_cast<DomItem*>(index.internalPointer());

    QDomNode node = item->node();

//! [3] //! [4]

    QStringList attributes;

    QDomNamedNodeMap attributeMap = node.attributes();

    switch (index.column()) {

        case 0:

            return node.nodeName();

        case 1:

            for (int i = 0; i < attributeMap.count(); ++i) {

                QDomNode attribute = attributeMap.item(i);

                attributes << attribute.nodeName() + "=\""

                              +attribute.nodeValue() + "\"";

            }

            return attributes.join(" ");

        case 2:

            return node.nodeValue().split("\n").join(" ");

        default:

            return QVariant();

    }

}

//! [4]

//! [5]

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

{

    if (!index.isValid())

        return 0;

    return Qt::ItemIsEnabled | Qt::ItemIsSelectable;

}

//! [5]

//! [6]

QVariant DomModel::headerData(int section, Qt::Orientation orientation,

                              int role) const

{

    if (orientation == Qt::Horizontal && role == Qt::DisplayRole) {

        switch (section) {

            case 0:

                return tr("Name");

            case 1:

                return tr("Attributes");

            case 2:

                return tr("Value");

            default:

                return QVariant();

        }

    }

    return QVariant();

}

//! [6]

//! [7]

QModelIndex DomModel::index(int row, int column, const QModelIndex &parent)

            const

{

    if (!hasIndex(row, column, parent))

        return QModelIndex();

    DomItem *parentItem;

    if (!parent.isValid())

        parentItem = rootItem;

    else

        parentItem = static_cast<DomItem*>(parent.internalPointer());

//! [7]

//! [8]

    DomItem *childItem = parentItem->child(row);

    if (childItem)

        return createIndex(row, column, childItem);

    else

        return QModelIndex();

}

//! [8]

//! [9]

QModelIndex DomModel::parent(const QModelIndex &child) const

{

    if (!child.isValid())

        return QModelIndex();

    DomItem *childItem = static_cast<DomItem*>(child.internalPointer());

    DomItem *parentItem = childItem->parent();

    if (!parentItem || parentItem == rootItem)

        return QModelIndex();

    return createIndex(parentItem->row(), 0, parentItem);

}

//! [9]

//! [10]

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

{

    if (parent.column() > 0)

        return 0;

    DomItem *parentItem;

    if (!parent.isValid())

        parentItem = rootItem;

    else

        parentItem = static_cast<DomItem*>(parent.internalPointer());

    return parentItem->node().childNodes().count();

}

//! [10]