Как восстановить последнее расширенное состояние QTreeView?

Что у меня:

  • QTreeView класс с данными таблицы
  • И подключенная модель QAbstractTableModel

Вопрос: как сохранить расширенное состояние элементов? Готовы ли некоторые из них?

PS: Я знаю, что я могу сделать этот код сам, но у меня мало времени, и это не главная проблема нашего проекта, но все же нам это нужно, потому что приложение содержит много таких таблицы, и каждый раз, когда расширяются элементы дерева, возникает раздраженный процесс...

ОТВЕТЫ

Ответ 1

Во-первых, благодаря Razi для persistentIndexList и isExpanded.

Во-вторых, вот код, который работает для меня просто отлично: -)

файл dialog.h:

class Dialog : public QDialog
{
    Q_OBJECT;

    TreeModel *model;
    TreeView *view;

public:
    Dialog(QWidget *parent = 0);
    ~Dialog(void);

    void reload(void);

protected:
    void createGUI(void);
    void closeEvent(QCloseEvent *);
    void saveState(void);
    void restoreState(void);
};

файл dialog.cpp:

Dialog::Dialog(QWidget *parent)
{
    createGUI();
    reload();
}

Dialog::~Dialog(void) {};

void Dialog::reload(void)
{
    restoreState();
}

void Dialog::createGUI(void)
{
    QFile file(":/Resources/default.txt");
    file.open(QIODevice::ReadOnly);
    model = new TreeModel(file.readAll());
    file.close();

    view = new TreeView(this);
    view->setModel(model);

    QVBoxLayout *mainVLayout = new QVBoxLayout;
    mainVLayout->addWidget(view);

    setLayout(mainVLayout);
}

void Dialog::closeEvent(QCloseEvent *event_)
{
    saveState();
}

void Dialog::saveState(void)
{
    QStringList List;

    // prepare list
    // PS: getPersistentIndexList() function is a simple `return this->persistentIndexList()` from TreeModel model class
    foreach (QModelIndex index, model->getPersistentIndexList())
    {
        if (view->isExpanded(index))
        {
            List << index.data(Qt::DisplayRole).toString();
        }
    }

    // save list
    QSettings settings("settings.ini", QSettings::IniFormat);
    settings.beginGroup("MainWindow");
    settings.setValue("ExpandedItems", QVariant::fromValue(List));
    settings.endGroup();
}

void Dialog::restoreState(void)
{
    QStringList List;

    // get list
    QSettings settings("settings.ini", QSettings::IniFormat);
    settings.beginGroup("MainWindow");
    List = settings.value("ExpandedItems").toStringList();
    settings.endGroup();

    foreach (QString item, List)
    {
        // search `item` text in model
        QModelIndexList Items = model->match(model->index(0, 0), Qt::DisplayRole, QVariant::fromValue(item));
        if (!Items.isEmpty())
        {
            // Information: with this code, expands ONLY first level in QTreeView
            view->setExpanded(Items.first(), true);
        }
    }
}

Иметь хороший день!)

PS: этот пример основан на коде C:\Qt\4.6.3\examples\itemviews\simpletreemodel.

Ответ 2

Благодаря Razi и mosg я смог получить эту работу. Я заставил его восстановить расширенное состояние рекурсивно, поэтому я решил поделиться этой частью.

void applyExpandState_sub(QStringList& expandedItems,
                          QTreeView* treeView,
                          QAbstractItemModel* model,
                          QModelIndex startIndex)
{
    foreach (QString item, expandedItems) 
    {
        QModelIndexList matches = model->match( startIndex, Qt::UserRole, item );
        foreach (QModelIndex index, matches) 
        {
            treeView->setExpanded( index, true );
            applyExpandState_sub(expandedItems, 
                                 treeView,
                                 model,
                                 model->index( 0, 0, index ) );
        }
    }
}

Затем используйте как:

void myclass::applyExpandState() 
{
    m_treeView->setUpdatesEnabled(false);

    applyExpandState_sub( m_expandedItems,
                          m_treeView,
                          m_model,
                          m_model->index( 0, 0, QModelIndex() ) );

    m_treeView->setUpdatesEnabled(true);
}

Я использую Qt:: UserRole здесь, потому что несколько элементов в моей модели могут иметь одинаковое отображаемое имя, которое испортило бы восстановление состояния расширения, поэтому UserRole предоставляет уникальный идентификатор для каждого элемента, чтобы избежать этой проблемы.

Ответ 3

Эти две функции, используя цикл, должны сделать это для вас:

QModelIndexList QAbstractItemModel::persistentIndexList () const
bool isExpanded ( const QModelIndex & index ) const

Ответ 4

Вот общий подход, который должен работать с любым виджетом на основе QTreeView, который использует некоторую систему идентификаторов для идентификации элементов (я предполагаю, что идентификатор является int, который хранится внутри Qt::UserRole):

void MyWidget::saveExpandedState()
{
    for(int row = 0; row < tree_view_->model()->rowCount(); ++row)
        saveExpandedOnLevel(tree_view_->model()->index(row,0));
}

void Widget::restoreExpandedState()
{
    tree_view_->setUpdatesEnabled(false);

    for(int row = 0; row < tree_view_->model()->rowCount(); ++row)
        restoreExpandedOnLevel(tree_view_->model()->index(row,0));

    tree_view_->setUpdatesEnabled(true);
}

void MyWidget::saveExpandedOnLevel(const QModelIndex& index)
{
    if(tree_view_->isExpanded(index)) {
        if(index.isValid())
            expanded_ids_.insert(index.data(Qt::UserRole).toInt());
        for(int row = 0; row < tree_view_->model()->rowCount(index); ++row)
            saveExpandedOnLevel(index.child(row,0));
    }
}

void MyWidget::restoreExpandedOnLevel(const QModelIndex& index)
{
    if(expanded_ids_.contains(index.data(Qt::UserRole).toInt())) {
        tree_view_->setExpanded(index, true);
        for(int row = 0; row < tree_view_->model()->rowCount(index); ++row)
            restoreExpandedOnLevel(index.child(row,0));
    }
}

Вместо MyWidget::saveExpandedState() и MyWidget::saveExpandedState() можно также напрямую вызвать MyWidget::saveExpandedOnLevel(tree_view_->rootIndex()) и MyWidget::restoreExpandedOnLevel(tree_view_->rootIndex()). Я использовал эту реализацию только потому, что цикл for будет вызван в любом случае, а MyWidget::saveExpandedState() и MyWidget::saveExpandedState() выглядят более чистыми с моими конструкциями SIGNAL и SLOT.

Ответ 5

Я переработал решение iforce2d:

 void ApplyExpandState(QStringList & nodes,
                       QTreeView * view,
                       QAbstractItemModel * model,
                       const QModelIndex startIndex,
                       QString path)
{
    path+=QString::number(startIndex.row()) + QString::number(startIndex.column());
    for(int i(0); i < model->rowCount(startIndex); ++i)
    {
        QModelIndex nextIndex = model->index(i, 0, startIndex);
        QString nextPath = path + QString::number(nextIndex.row()) + QString::number(nextIndex.column());
        if(!nodes.contains(nextPath))
            continue;
        ApplyExpandState(nodes, view, model, model->index(i, 0, startIndex), path);
    }
    if(nodes.contains(path))
        view->setExpanded( startIndex.sibling(startIndex.row(), 0), true );
}

void StoreExpandState(QStringList & nodes,
                      QTreeView * view,
                      QAbstractItemModel * model,
                      const QModelIndex startIndex,
                      QString path)
{
    path+=QString::number(startIndex.row()) + QString::number(startIndex.column());
    for(int i(0); i < model->rowCount(startIndex); ++i)
    {
        if(!view->isExpanded(model->index(i, 0, startIndex)))
            continue;
        StoreExpandState(nodes, view, model, model->index(i, 0, startIndex), path);
    }

    if(view->isExpanded(startIndex))
        nodes << path;
}

Таким образом, нет необходимости сопоставлять данные. Очевидно, что для этого подхода к работе дерево должно оставаться относительно неизменным. Если вы каким-то образом измените порядок элементов дерева - он расширит неправильные узлы.

Ответ 6

При a QFileSystemModel вы не можете использовать persistentIndexList().

Вот моя работа. Он работает очень хорошо, даже если я сам так говорю. Я не проверял, что произойдет, если у вас медленная файловая система загрузки, или если вы удалите файл или путь.

// scrolling code connection in constructor
model = new QFileSystemModel();

QObject::connect(ui->treeView, &QTreeView::expanded, [=](const QModelIndex &index)
{
    ui->treeView->scrollTo(index, QAbstractItemView::PositionAtTop);//PositionAtCenter);
});

// save state, probably in your closeEvent()
QSettings s;
s.setValue("header_state",ui->treeView->header()->saveState());
s.setValue("header_geometry",ui->treeView->header()->saveGeometry());

if(ui->treeView->currentIndex().isValid())
{
    QFileInfo info = model->fileInfo(ui->treeView->currentIndex());
    QString filename = info.absoluteFilePath();
    s.setValue("last_directory",filename);
}

// restore state, probably in your showEvent()
QSettings s;
ui->treeView->header()->restoreState(s.value("header_state").toByteArray());
ui->treeView->header()->restoreGeometry(s.value("header_geometry").toByteArray());
QTimer::singleShot(1000, [=]() {
    QSettings s;
    QString filename = s.value("last_directory").toString();
    QModelIndex index = model->index(filename);
    if(index.isValid())
    {
        ui->treeView->expand(index);
        ui->treeView->setCurrentIndex(index);

        ui->treeView->scrollTo(index, QAbstractItemView::PositionAtCenter);
        qDebug() << "Expanded" << filename;
    }
    else
        qDebug() << "Invalid index" << filename;
} );

Надеюсь, что это поможет кому-то.

Ответ 7

Вот версия, которая не полагается на узлы, имеющие уникальный Qt::UserRole или Qt::DisplayRole - он просто сериализует весь QModelIndex

Заголовок:

#pragma once
#include <QTreeView>

class TreeView : public QTreeView
{
    Q_OBJECT
public:
    using QTreeView::QTreeView;

    QStringList saveExpandedState(const QModelIndexList&) const;
    void        restoreExpandedState(const QStringList&);
};

Источник:

#include "tree_view.h"
#include <QAbstractItemModel>

namespace
{
    std::string toString(const QModelIndex& index)
    {
        std::string parent = index.parent().isValid() ? toString(index.parent()) : "X";

        char buf[512];
        sprintf(buf, "%d:%d[%s]", index.row(), index.column(), parent.c_str());
        return buf;
    }

    QModelIndex fromString(const std::string& string, QAbstractItemModel& model)
    {
        int row, column;
        char parent_str[512];
        sscanf(string.c_str(), "%d:%d[%s]", &row, &column, parent_str);

        QModelIndex parent = *parent_str == 'X' ? QModelIndex() : fromString(parent_str, model);

        return model.index(row, column, parent);
    }
}

QStringList TreeView::saveExpandedState(const QModelIndexList& indices) const
{
    QStringList list;
    for (const QModelIndex& index : indices)
    {
        if (isExpanded(index))
        {
            list << QString::fromStdString(toString(index));
        }
    }
    return list;
}

void TreeView::restoreExpandedState(const QStringList& list)
{
    setUpdatesEnabled(false);

    for (const QString& string : list)
    {
        QModelIndex index = fromString(string.toStdString(), *model());
        setExpanded(index, true);
    }

    setUpdatesEnabled(true);
};