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

 **

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

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

 **

 ** This file is part of the examples of the Qt Toolkit.

 **

 ** $QT_BEGIN_LICENSE:LGPL$

 ** Commercial Usage

 ** Licensees holding valid Qt Commercial licenses may use this file in

 ** accordance with the Qt Commercial License Agreement provided with the

 ** Software or, alternatively, in accordance with the terms contained in

 ** a written agreement between you and Nokia.

 **

 ** 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.0, included in the file LGPL_EXCEPTION.txt in this

 ** package.

 **

 ** GNU General Public License Usage

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

 ** General Public License version 3.0 as published by the Free Software

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

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

 ** ensure the GNU General Public License version 3.0 requirements will be

 ** met: http://www.gnu.org/copyleft/gpl.html.

 **

 ** If you are unsure which license is appropriate for your use, please

 ** contact the sales department at http://www.qtsoftware.com/contact.

 ** $QT_END_LICENSE$

 **

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

 #include <QtGui>

 #include <QtXml>

 #include "domitem.h"

 #include "dommodel.h"

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

     : QAbstractItemModel(parent), domDocument(document)

 {

     rootItem = new DomItem(domDocument, 0);

 }

 DomModel::~DomModel()

 {

     delete rootItem;

 }

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

 {

     return 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();

     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();

     }

 }

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

 {

     if (!index.isValid())

         return 0;

     return Qt::ItemIsEnabled | Qt::ItemIsSelectable;

 }

 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();

 }

 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());

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

     if (childItem)

         return createIndex(row, column, childItem);

     else

         return QModelIndex();

 }

 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);

 }

 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();

 }