|  | Home · All Classes · Main Classes · Annotated · Grouped Classes · Functions |

# treemodel.cpp Example File
itemviews/simpletreemodel/treemodel.cpp

        /****************************************************************************
        **
        ** Copyright (C) 2004-2005 Trolltech AS. All rights reserved.
        **
        ** This file is part of an example program for Qt.
        ** EDITIONS: NOLIMITS
        **
        ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
        ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
        **
        ****************************************************************************/

        /*
            treemodel.cpp

            Provides a simple tree model to show how to create and use hierarchical
            models.
        */

        #include <QtGui>

        #include "treeitem.h"
        #include "treemodel.h"

        TreeModel::TreeModel(const QString &data, QObject *parent)
            : QAbstractItemModel(parent)
        {
            QList<QVariant> rootData;
            rootData << "Title" << "Summary";
            rootItem = new TreeItem(rootData);
            setupModelData(data.split(QString("\n")), rootItem);
        }

        TreeModel::~TreeModel()
        {
            delete rootItem;
        }

        int TreeModel::columnCount(const QModelIndex &parent) const
        {
            if (parent.isValid())
                return static_cast<TreeItem*>(parent.data())->columnCount();
            else
                return rootItem->columnCount();
        }

        QVariant TreeModel::data(const QModelIndex &index, int role) const
        {
            if (!index.isValid())
                return QVariant();

            if (role != DisplayRole)
                return QVariant();

            TreeItem *item = static_cast<TreeItem*>(index.data());

            return item->data(index.column());
        }

        QAbstractItemModel::ItemFlags TreeModel::flags(const QModelIndex &index) const
        {
            if (!index.isValid())
                return ItemIsEnabled;

            return ItemIsEnabled | ItemIsSelectable;
        }

        QVariant TreeModel::headerData(int section, Qt::Orientation orientation,
                                       int role) const
        {
            if (orientation == Qt::Horizontal && role == QAbstractItemModel::DisplayRole)
                return rootItem->data(section);

            return QVariant();
        }

        QModelIndex TreeModel::index(int row, int column, const QModelIndex &parent)
                    const
        {
            TreeItem *parentItem;

            if (!parent.isValid())
                parentItem = rootItem;
            else
                parentItem = static_cast<TreeItem*>(parent.data());

            TreeItem *childItem = parentItem->child(row);
            if (childItem)
                return createIndex(row, column, childItem);
            else
                return QModelIndex();
        }

        QModelIndex TreeModel::parent(const QModelIndex &index) const
        {
            if (!index.isValid())
                return QModelIndex();

            TreeItem *childItem = static_cast<TreeItem*>(index.data());
            TreeItem *parentItem = childItem->parent();

            if (parentItem == rootItem)
                return QModelIndex();

            return createIndex(parentItem->row(), index.column(), parentItem);
        }

        int TreeModel::rowCount(const QModelIndex &parent) const
        {
            TreeItem *parentItem;

            if (!parent.isValid())
                parentItem = rootItem;
            else
                parentItem = static_cast<TreeItem*>(parent.data());

            return parentItem->childCount();
        }

        void TreeModel::setupModelData(const QStringList &lines, TreeItem *parent)
        {
            QList<TreeItem*> parents;
            QList<int> indentations;
            parents << parent;
            indentations << 0;

            int number = 0;

            while (number < lines.count()) {
                int position = 0;
                while (position < lines[number].length()) {
                    if (lines[number].mid(position, 1) != " ")
                        break;
                    position++;
                }

                QString lineData = lines[number].mid(position).trimmed();

                if (!lineData.isEmpty()) {
                    // Read the column data from the rest of the line.
                    QStringList columnStrings = lineData.split("\t", QString::SkipEmptyParts);
                    QList<QVariant> columnData;
                    for (int column = 0; column < columnStrings.count(); ++column)
                        columnData << columnStrings[column];

                    if (position > indentations.last()) {
                        // The last child of the current parent is now the new parent
                        // unless the current parent has no children.

                        if (parents.last()->childCount() > 0) {
                            parents << parents.last()->child(parents.last()->childCount()-1);
                            indentations << position;
                        }
                    } else {
                        while (position < indentations.last() && parents.count() > 0) {
                            parents.pop_back();
                            indentations.pop_back();
                        }
                    }

                    // Append a new item to the current parent's list of children.
                    parents.last()->appendChild(new TreeItem(columnData, parents.last()));
                }

                number++;
            }
        }