File indexing completed on 2024-05-19 05:21:55

 /*
  * Copyright (C) 2007 by Mathias Soeken <msoeken@tzi.de>
  * Copyright (C) 2019  Alexander Potashev <aspotashev@gmail.com>
  *
  *   This program 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 2 of the License, or
  *   (at your option) any later version.
  *
  *   This program 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 this program; if not, write to the
  *      Free Software Foundation, Inc.
  *      51 Franklin Street, Fifth Floor
  *      Boston, MA  02110-1301  USA.
  *
  */
 
 #include "treeviewheadercontextmenu.h"
 
 #include <QAction>
 #include <QHeaderView>
 #include <QTreeView>
 
 #include <KLocalizedString>
 
 #include "ktt_debug.h"
 
 TreeViewHeaderContextMenu::TreeViewHeaderContextMenu(QObject *parent, QTreeView *widget, QVector<int> &&excludedColumns)
     : QObject(parent)
     , m_widget(widget)
     , m_contextMenu(nullptr)
     , m_excludedColumns(excludedColumns)
 {
     m_widget->header()->setContextMenuPolicy(Qt::CustomContextMenu);
     connect(m_widget->header(),
             &QHeaderView::customContextMenuRequested,
             this,
             &TreeViewHeaderContextMenu::slotCustomContextMenuRequested);
 
     m_contextMenu = new QMenu(m_widget);
     m_contextMenu->addSection(i18nc("@title:menu", "Columns"));
     connect(m_contextMenu, &QMenu::triggered, this, &TreeViewHeaderContextMenu::slotTriggered);
     connect(m_contextMenu, &QMenu::aboutToShow, this, &TreeViewHeaderContextMenu::slotAboutToShow);
     updateActions();
 }
 
 TreeViewHeaderContextMenu::~TreeViewHeaderContextMenu()
 {
     qDeleteAll(m_actions);
 }
 
 void TreeViewHeaderContextMenu::slotCustomContextMenuRequested(const QPoint &pos)
 {
     qCDebug(KTT_LOG) << "Entering function";
     if (m_widget && m_contextMenu) {
         m_contextMenu->exec(m_widget->mapToGlobal(pos));
     }
 }
 
 void TreeViewHeaderContextMenu::updateActions()
 {
     qCDebug(KTT_LOG) << "Entering function";
     if (m_widget) {
         for (QAction *action : m_actions) {
             m_contextMenu->removeAction(action);
         }
 
         m_actionColumnMapping.clear();
         qDeleteAll(m_actions);
         m_actions.clear();
 
         for (int c = 0; c < m_widget->model()->columnCount(); ++c) {
             if (m_excludedColumns.contains(c)) {
                 continue;
             }
 
             auto *action = new QAction(this);
             updateAction(action, c);
             m_actions.append(action);
 
             m_contextMenu->addAction(action);
             m_actionColumnMapping[action] = c;
         }
     }
 }
 
 void TreeViewHeaderContextMenu::slotTriggered(QAction *action)
 {
     qCDebug(KTT_LOG) << "Entering function";
     if (m_widget && action) {
         int column = m_actionColumnMapping[action];
         bool hidden = m_widget->isColumnHidden(column);
         m_widget->setColumnHidden(column, !hidden);
         updateAction(action, column);
         Q_EMIT columnToggled(column);
     }
 }
 
 void TreeViewHeaderContextMenu::slotAboutToShow()
 {
     qCDebug(KTT_LOG) << "Entering function";
     for (QAction *action : m_actions) {
         updateAction(action, m_actionColumnMapping[action]);
     }
 }
 
 void TreeViewHeaderContextMenu::updateAction(QAction *action, int column)
 {
     action->setCheckable(true);
     action->setChecked(!m_widget->isColumnHidden(column));
     action->setText(m_widget->model()->headerData(column, Qt::Horizontal).toString());
 }
 
 #include "moc_treeviewheadercontextmenu.cpp"