File indexing completed on 2024-04-28 04:21:03

 // SPDX-FileCopyrightText: 2003-2020 The KPhotoAlbum Development Team
 // SPDX-FileCopyrightText: 2021 Johannes Zarl-Zierl <johannes@zarl-zierl.at>
 // SPDX-FileCopyrightText: 2022 Johannes Zarl-Zierl <johannes@zarl-zierl.at>
 //
 // SPDX-License-Identifier: GPL-2.0-or-later
 
 #include "TreeView.h"
 
 #include <kpaexif/Info.h>
 
 #include <kpabase/StringSet.h>
 
 #include <qmap.h>
 #include <qstringlist.h>
 
 using Utilities::StringSet;
 
 Exif::TreeView::TreeView(const QString &title, QWidget *parent)
     : QTreeWidget(parent)
 {
     setHeaderLabel(title);
     reload();
     connect(this, &TreeView::itemClicked, this, &TreeView::toggleChildren);
 }
 
 void Exif::TreeView::toggleChildren(QTreeWidgetItem *parent)
 {
     if (!parent)
         return;
 
     bool on = parent->checkState(0) == Qt::Checked;
     for (int index = 0; index < parent->childCount(); ++index) {
         parent->child(index)->setCheckState(0, on ? Qt::Checked : Qt::Unchecked);
         toggleChildren(parent->child(index));
     }
 }
 
 StringSet Exif::TreeView::selected()
 {
     StringSet result;
     for (QTreeWidgetItemIterator it(this); *it; ++it) {
         if ((*it)->checkState(0) == Qt::Checked)
             result.insert((*it)->text(1));
     }
     return result;
 }
 
 void Exif::TreeView::setSelectedExif(const StringSet &selected)
 {
     for (QTreeWidgetItemIterator it(this); *it; ++it) {
         bool on = selected.contains((*it)->text(1));
         (*it)->setCheckState(0, on ? Qt::Checked : Qt::Unchecked);
     }
 }
 
 void Exif::TreeView::reload()
 {
     clear();
     setRootIsDecorated(true);
     const auto availableKeys = Exif::Info::instance()->availableKeys();
     QStringList keys(availableKeys.begin(), availableKeys.end());
     keys.sort();
 
     QMap<QString, QTreeWidgetItem *> tree;
 
     for (QStringList::const_iterator keysIt = keys.constBegin(); keysIt != keys.constEnd(); ++keysIt) {
         const QStringList subKeys = (*keysIt).split(QLatin1String("."));
         QTreeWidgetItem *parent = nullptr;
         QString path;
         for (const QString &subKey : subKeys) {
             if (!path.isEmpty())
                 path += QString::fromLatin1(".");
             path += subKey;
             if (tree.contains(path))
                 parent = tree[path];
             else {
                 if (parent == nullptr)
                     parent = new QTreeWidgetItem(this, QStringList(subKey));
                 else
                     parent = new QTreeWidgetItem(parent, QStringList(subKey));
                 parent->setText(1, path); // This is simply to make the implementation of selected easier.
                 parent->setFlags(Qt::ItemIsUserCheckable | Qt::ItemIsEnabled);
                 parent->setCheckState(0, Qt::Unchecked);
                 tree.insert(path, parent);
             }
         }
     }
 
     if (QTreeWidgetItem *item = topLevelItem(0))
         item->setExpanded(true);
 }
 
 // vi:expandtab:tabstop=4 shiftwidth=4:
 
 #include "moc_TreeView.cpp"