TYTreeStateHelper.cpp

Go to the documentation of this file.

/*
 * Copyright (C) <2012-2024> <EDF-DTG> <FRANCE>
 * This file is part of Code_TYMPAN (R).
 * Code_TYMPAN (R) is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * Code_TYMPAN (R) is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License along
 * with Code_TYMPAN (R). If not, see <https://www.gnu.org/licenses/>.
 */
 
#include "TYTreeStateHelper.h"
#include <QScrollBar>
#include <QVariant>
 
void TYTreeStateHelper::capture(QTreeWidget* tree, TYTreeViewState& outState)
{
    outState.expansion.clear();
 
    // 1) Sauvegarde expansions
    for (int i = 0; i < tree->topLevelItemCount(); ++i)
    {
        saveExpansionRec(tree->topLevelItem(i), outState.expansion);
    }
 
    // 2) Sauvegarde scroll
    if (tree->verticalScrollBar())
        outState.vScroll = tree->verticalScrollBar()->value();
    else
        outState.vScroll = 0;
 
    if (tree->horizontalScrollBar())
        outState.hScroll = tree->horizontalScrollBar()->value();
    else
        outState.hScroll = 0;
 
    // >>> Points d’arrêt utiles :
    // qDebug() << "[CAPTURE] vScroll=" << outState.vScroll << "hScroll=" << outState.hScroll;
}
 
void TYTreeStateHelper::apply(QTreeWidget* tree, const TYTreeViewState& state)
{
    // 1) Restaure expansions
    for (int i = 0; i < tree->topLevelItemCount(); ++i)
    {
        restoreExpansionRec(tree->topLevelItem(i), state.expansion);
    }
 
    // 2) Restaure scroll (après expansions pour que les barres soient dimensionnées)
    if (tree->verticalScrollBar())
        tree->verticalScrollBar()->setValue(state.vScroll);
 
    if (tree->horizontalScrollBar())
        tree->horizontalScrollBar()->setValue(state.hScroll);
 
    // >>> Points d’arrêt utiles :
    // qDebug() << "[APPLY] vScroll->" << state.vScroll << "hScroll->" << state.hScroll;
}
 
void TYTreeStateHelper::saveExpansionRec(QTreeWidgetItem* item, QHash<QString, bool>& map)
{
    const QString key = makeKey(item);
    map.insert(key, item->isExpanded());
 
    // >>> Breakpoint utile :
    // qDebug() << "[SAVE EXP]" << key << "expanded=" << item->isExpanded();
 
    const int n = item->childCount();
    for (int i = 0; i < n; ++i)
    {
        saveExpansionRec(item->child(i), map);
    }
}
 
void TYTreeStateHelper::restoreExpansionRec(QTreeWidgetItem* item, const QHash<QString, bool>& map)
{
    const QString key = makeKey(item);
    const auto it = map.constFind(key);
    if (it != map.constEnd())
    {
        item->setExpanded(it.value());
        // >>> Breakpoint utile :
        // qDebug() << "[RESTORE EXP] found" << key << "expanded=" << it.value();
    }
    else
    {
        // >>> En debug, voir pourquoi cette clé manque :
        // qDebug() << "[RESTORE EXP] missing key for" << key;
    }
 
    const int n = item->childCount();
    for (int i = 0; i < n; ++i)
    {
        restoreExpansionRec(item->child(i), map);
    }
}
 
QString TYTreeStateHelper::makeKey(QTreeWidgetItem* item)
{
    const QVariant v = item->data(0, Qt::UserRole);
    if (v.isValid())
    {
        const QString id = v.toString();
        if (!id.isEmpty())
            return id; // Clé stable prioritaire
    }
 
    // Fallback : chemin textuel
    QStringList path;
    for (QTreeWidgetItem* it = item; it; it = it->parent())
        path.prepend(it->text(0));
    return path.join(u"/");
}