|  | Home · Examples

# Editable Tree Model

This example shows how to implement a simple item-based tree model that can be used with other classes the model/view framework.

The model supports editable items, custom headers, and the ability to insert and remove rows and columns. With these features, it is also possible to insert new child items, and this is shown in the supporting example code.

Note: The model only shows the basic principles used when creating an editable, hierarchical model. You may wish to use the ModelTest project to test production models.

## Overview

As described in the Model Subclassing Reference, models must provide implementations for the standard set of model functions: flags(), data(), headerData(), and rowCount(). In addition, hierarchical models, such as this one, need to provide implementations of index() and parent().

An editable model needs to provide implementations of setData() and headerData(), and must return a suitable combination of flags from its flags() function.

Since this example allows the dimensions of the model to be changed, we must also implement insertRows(), insertColumns(), removeRows(), and removeColumns().

## Design

As with the Simple Tree Model example, the model simply acts as a wrapper around a collection of instances of a `TreeItem` class. Each `TreeItem` is designed to hold data for a row of items in a tree view, so it contains a list of values corresponding to the data shown in each column.

Since QTreeView provides a row-oriented view onto a model, it is natural to choose a row-oriented design for data structures that will supply data via a model to this kind of view. Although this makes the tree model less flexible, and possibly less useful for use with more sophisticated views, it makes it less complex to design and easier to implement.

| Relations between internal items When designing a data structure for use with a custom model, it is useful to expose each item's parent via a function like TreeItem::parent() because it will make writing the model's own parent() function easier. Similarly, a function like TreeItem::child() is helpful when implementing the model's index() function. As a result, each `TreeItem` maintains information about its parent and children, making it possible for us to traverse the tree structure. The diagram shows how `TreeItem` instances are connected via their parent() and child() functions. In the example shown, two top-level items, A and B, can be obtained from the root item by calling its child() function, and each of these items return the root node from their parent() functions, though this is only shown for item A.

Each `TreeItem` stores data for each column in the row it represents in its `itemData` private member (a list of QVariant objects). Since there is a one-to-one mapping between each column in the view and each entry in the list, we provide a simple data() function to read entries in the `itemData` list and a setData() function to allow them to be modified. As with other functions in the item, this simplifies the implemention of the model's data() and setData() functions.

We place an item at the root of the tree of items. This root item corresponds to the null model index, QModelIndex(), that is used to represent the parent of a top-level item when handling model indexes. Although the root item does not have a visible representation in any of the standard views, we use its internal list of QVariant objects to store a list of strings that will be passed to views for use as horizontal header titles.

| Accessing data via the model In the case shown in the diagram, the piece of information represented by a can be obtained using the standard model/view API:
    QVariant a = model->index(0, 0, QModelIndex()).data();
Since each items holds pieces of data for each column in a given row, there can be many model indexes that map to the same `TreeItem` object. For example, the information represented by b can be obtained using the following code:
    QVariant b = model->index(1, 0, QModelIndex()).data();
The same underlying `TreeItem` would be accessed to obtain information for the other model indexes in the same row as b.

In the model class, `TreeModel`, we relate `TreeItem` objects to model indexes by passing a pointer for each item when we create its corresponding model index with QAbstractItemModel::createIndex() in our index() and parent() implementations. We can retrieve pointers stored in this way by calling the internalPointer() function on the relevant model index - we create our own getItem() function to do this work for us, and call it from our data() and parent() implementations.

Storing pointers to items is convenient when we control how they are created and destroyed since we can assume that an address obtained from internalPointer() is a valid pointer. However, some models need to handle items that are obtained from other components in a system, and in many cases it is not possible to fully control how items are created or destroyed. In such situations, a pure pointer-based approach needs to be supplemented by safeguards to ensure that the model does not attempt to access items that have been deleted.

Storing information in the underlying data structure Several pieces of data are stored as QVariant objects in the `itemData` member of each `TreeItem` instance The diagram shows how pieces of information, represented by the labels a, b and c in the previous two diagrams, are stored in items A, B and C in the underlying data structure. Note that pieces of information from the same row in the model are all obtained from the same item. Each element in a list corresponds to a piece of information exposed by each column in a given row in the model. |

Since the `TreeModel` implementation has been designed for use with QTreeView, we have added a restriction on the way it uses `TreeItem` instances: each item must expose the same number of columns of data. This makes viewing the model consistent, allowing us to use the root item to determine the number of columns for any given row, and only adds the requirement that we create items containing enough data for the total number of columns. As a result, inserting and removing columns are time-consuming operations because we need to traverse the entire tree to modify every item.

An alternative approach would be to design the `TreeModel` class so that it truncates or expands the list of data in individual `TreeItem` instances as items of data are modified. However, this "lazy" resizing approach would only allow us to insert and remove columns at the end of each row and would not allow columns to be inserted or removed at arbitrary positions in each row.

| Relating items using model indexes As with the Simple Tree Model example, the `TreeModel` needs to be able to take a model index, find the corresponding `TreeItem`, and return model indexes that correspond to its parents and children. In the diagram, we show how the model's parent() implementation obtains the model index corresponding to the parent of an item supplied by the caller, using the items shown in a previous diagram. A pointer to item C is obtained from the corresponding model index using the QModelIndex::internalPointer() function. The pointer was stored internally in the index when it was created. Since the child contains a pointer to its parent, we use its parent() function to obtain a pointer to item B. The parent model index is created using the QAbstractItemModel::createIndex() function, passing the pointer to item B as the internal pointer.

## TreeItem Class Definition

The `TreeItem` class provides simple items that contain several pieces of data, and which can provide information about their parent and child items:

    class TreeItem
    {
    public:
        TreeItem(const QVector<QVariant> &data, TreeItem *parent = 0);
        ~TreeItem();

        TreeItem *child(int number);
        int childCount() const;
        int columnCount() const;
        QVariant data(int column) const;
        bool insertChildren(int position, int count, int columns);
        bool insertColumns(int position, int columns);
        TreeItem *parent();
        bool removeChildren(int position, int count);
        bool removeColumns(int position, int columns);
        int childNumber() const;
        bool setData(int column, const QVariant &value);

    private:
        QList<TreeItem*> childItems;
        QVector<QVariant> itemData;
        TreeItem *parentItem;
    };

We have designed the API to be similar to that provided by QAbstractItemModel by giving each item functions to return the number of columns of information, read and write data, and insert and remove columns. However, we make the relationship between items explicit by providing functions to deal with "children" rather than "rows".

Each item contains a list of pointers to child items, a pointer to its parent item, and a list of QVariant objects that correspond to information held in columns in a given row in the model.

## TreeItem Class Implementation

Each `TreeItem` is constructed with a list of data and an optional parent item:
The following code example is written in c++.

    TreeItem::TreeItem(const QVector<QVariant> &data, TreeItem *parent)
    {
        parentItem = parent;
        itemData = data;
    }

Initially, each item has no children. These are added to the item's internal `childItems` member using the `insertChildren()` function described later.

The destructor ensures that each child added to the item is deleted when the item itself is deleted:
The following code example is written in c++.

    TreeItem::~TreeItem()
    {
        qDeleteAll(childItems);
    }

Since each item stores a pointer to its parent, the `parent()` function is trivial:
The following code example is written in c++.

    TreeItem *TreeItem::parent()
    {
        return parentItem;
    }

Three functions provide information about the children of an item. `child()` returns a specific child from the internal list of children:
The following code example is written in c++.

    TreeItem *TreeItem::child(int number)
    {
        return childItems.value(number);
    }

The `childCount()` function returns the total number of children:
The following code example is written in c++.

    int TreeItem::childCount() const
    {
        return childItems.count();
    }

The `childNumber()` function is used to determine the index of the child in its parent's list of children. It accesses the parent's `childItems` member directly to obtain this information:
The following code example is written in c++.

    int TreeItem::childNumber() const
    {
        if (parentItem)
            return parentItem->childItems.indexOf(const_cast<TreeItem*>(this));

        return 0;
    }

The root item has no parent item; for this item, we return zero to be consistent with the other items.

The `columnCount()` function simply returns the number of elements in the internal `itemData` list of QVariant objects:
The following code example is written in c++.

    int TreeItem::columnCount() const
    {
        return itemData.count();
    }

Data is retrieved using the `data()` function, which accesses the appropriate element in the `itemData` list:
The following code example is written in c++.

    QVariant TreeItem::data(int column) const
    {
        return itemData.value(column);
    }

Data is set using the `setData()` function, which only stores values in the `itemData` list for valid list indexes, corresponding to column values in the model:
The following code example is written in c++.

    bool TreeItem::setData(int column, const QVariant &value)
    {
        if (column < 0 || column >= itemData.size())
            return false;

        itemData[column] = value;
        return true;
    }

To make implementation of the model easier, we return true to indicate whether the data was set successfully, or false if an invalid column

Editable models often need to be resizable, enabling rows and columns to be inserted and removed. The insertion of rows beneath a given model index in the model leads to the insertion of new child items in the corresponding item, handled by the `insertChildren()` function:
The following code example is written in c++.

    bool TreeItem::insertChildren(int position, int count, int columns)
    {
        if (position < 0 || position > childItems.size())
            return false;

        for (int row = 0; row < count; ++row) {
            QVector<QVariant> data(columns);
            TreeItem *item = new TreeItem(data, this);
            childItems.insert(position, item);
        }

        return true;
    }

This ensures that new items are created with the required number of columns and inserted at a valid position in the internal `childItems` list. Items are removed with the `removeChildren()` function:
The following code example is written in c++.

    bool TreeItem::removeChildren(int position, int count)
    {
        if (position < 0 || position + count > childItems.size())
            return false;

        for (int row = 0; row < count; ++row)
            delete childItems.takeAt(position);

        return true;
    }

As discussed above, the functions for inserting and removing columns are used differently to those for inserting and removing child items because they are expected to be called on every item in the tree. We do this by recursively calling this function on each child of the item:
The following code example is written in c++.

    bool TreeItem::insertColumns(int position, int columns)
    {
        if (position < 0 || position > itemData.size())
            return false;

        for (int column = 0; column < columns; ++column)
            itemData.insert(position, QVariant());

        foreach (TreeItem *child, childItems)
            child->insertColumns(position, columns);

        return true;
    }

## TreeModel Class Definition

The `TreeModel` class provides an implementation of the QAbstractItemModel class, exposing the necessary interface for a model that can be edited and resized.

    class TreeModel : public QAbstractItemModel
    {
        Q_OBJECT

    public:
        TreeModel(const QStringList &headers, const QString &data,
                  QObject *parent = 0);
        ~TreeModel();

The constructor and destructor are specific to this model.

        QVariant data(const QModelIndex &index, int role) const;
        QVariant headerData(int section, Qt::Orientation orientation,
                            int role = Qt::DisplayRole) const;

        QModelIndex index(int row, int column,
                          const QModelIndex &parent = QModelIndex()) const;
        QModelIndex parent(const QModelIndex &index) const;

        int rowCount(const QModelIndex &parent = QModelIndex()) const;
        int columnCount(const QModelIndex &parent = QModelIndex()) const;

Read-only tree models only need to provide the above functions. The following public functions provide support for editing and resizing:

        Qt::ItemFlags flags(const QModelIndex &index) const;
        bool setData(const QModelIndex &index, const QVariant &value,
                     int role = Qt::EditRole);
        bool setHeaderData(int section, Qt::Orientation orientation,
                           const QVariant &value, int role = Qt::EditRole);

        bool insertColumns(int position, int columns,
                           const QModelIndex &parent = QModelIndex());
        bool removeColumns(int position, int columns,
                           const QModelIndex &parent = QModelIndex());
        bool insertRows(int position, int rows,
                        const QModelIndex &parent = QModelIndex());
        bool removeRows(int position, int rows,
                        const QModelIndex &parent = QModelIndex());

    private:
        void setupModelData(const QStringList &lines, TreeItem *parent);
        TreeItem *getItem(const QModelIndex &index) const;

        TreeItem *rootItem;
    };

To simplify this example, the data exposed by the model is organized into a data structure by the model's setupModelData() function. Many real world models will not process the raw data at all, but simply work with an existing data structure or library API.

## TreeModel Class Implementation

The constructor creates a root item and initializes it with the header data supplied:
The following code example is written in c++.

    TreeModel::TreeModel(const QStringList &headers, const QString &data,
                         QObject *parent)
        : QAbstractItemModel(parent)
    {
        QVector<QVariant> rootData;
        foreach (QString header, headers)
            rootData << header;

        rootItem = new TreeItem(rootData);
        setupModelData(data.split(QString("\n")), rootItem);
    }

We call the internal setupModelData() function to convert the textual data supplied to a data structure we can use with the model. Other models may be initialized with a ready-made data structure, or use an API to a library that maintains its own data.

The destructor only has to delete the root item; all child items will be recursively deleted by the `TreeItem` destructor.
The following code example is written in c++.

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

Since the model's interface to the other model/view components is based on model indexes, and the internal data structure is item-based, many of the functions implemented by the model need to be able to convert any given model index to its corresponding item. For convenience and consistency, we have defined a `getItem()` function to perform this repetitive task:
The following code example is written in c++.

    TreeItem *TreeModel::getItem(const QModelIndex &index) const
    {
        if (index.isValid()) {
            TreeItem *item = static_cast<TreeItem*>(index.internalPointer());
            if (item) return item;
        }
        return rootItem;
    }

This function assumes that each model index it is passed corresponds to a valid item in memory. If the index is invalid, or its internal pointer does not refer to a valid item, the root item is returned instead.

The model's `rowCount()` implementation is simple: it first uses the `getItem()` function to obtain the relevant item, then returns the number of children it contains:
The following code example is written in c++.

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

        return parentItem->childCount();
    }

By contrast, the `columnCount()` implementation does not need to look for a particular item because all items are defined to have the same number of columns associated with them.
The following code example is written in c++.

    int TreeModel::columnCount(const QModelIndex &parent) const
    {
        return rootItem->columnCount();
    }

As a result, the number of columns can be obtained directly from the root item.

To enable items to be edited and selected, the `flags()` function needs to be implemented so that it returns a combination of flags that includes the Qt::ItemIsEditable and Qt::ItemIsSelectable flags as well as Qt::ItemIsEnabled:
The following code example is written in c++.

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

        return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable;
    }

The model needs to be able to generate model indexes to allow other components to request data and information about its structure. This task is performed by the `index()` function, which is used to obtain model indexes corresponding to children of a given parent item:
The following code example is written in c++.

    QModelIndex TreeModel::index(int row, int column, const QModelIndex &parent) const
    {
        if (parent.isValid() && parent.column() != 0)
            return QModelIndex();

In this model, we only return model indexes for child items if the parent index is invalid (corresponding to the root item) or if it has a zero column number.

We use the custom getItem() function to obtain a `TreeItem` instance that corresponds to the model index supplied, and request its child item that corresponds to the specified row.
The following code example is written in c++.

        TreeItem *parentItem = getItem(parent);

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

Since each item contains information for an entire row of data, we create a model index to uniquely identify it by calling createIndex() it with the row and column numbers and a pointer to the item. In the data() function, we will use the item pointer and column number to access the data associated with the model index; in this model, the row number is not needed to identify data. The `parent()` function supplies model indexes for parents of items by finding the corresponding item for a given model index, using its parent() function to obtain its parent item, then creating a model index to represent the parent. (See the above diagram).
The following code example is written in c++.

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

        TreeItem *childItem = getItem(index);
        TreeItem *parentItem = childItem->parent();

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

        return createIndex(parentItem->childNumber(), 0, parentItem);
    }

Items without parents, including the root item, are handled by returning a null model index. Otherwise, a model index is created and returned as in the index() function, with a suitable row number, but with a zero column number to be consistent with the scheme used in the index() implementation.