1 // For license of this file, see <project-root-folder>/LICENSE.md.
2
3 #include "gui/reusable/treeviewcolumnsmenu.h"
4
5 #include <QHeaderView>
6
TreeViewColumnsMenu(QHeaderView * parent)7 TreeViewColumnsMenu::TreeViewColumnsMenu(QHeaderView* parent) : NonClosableMenu(parent) {
8 connect(this, &TreeViewColumnsMenu::aboutToShow, this, &TreeViewColumnsMenu::prepareMenu);
9 }
10
prepareMenu()11 void TreeViewColumnsMenu::prepareMenu() {
12 QHeaderView* header_view = header();
13
14 for (int i = 0; i < header_view->count(); i++) {
15 QAction* act = addAction(header_view->model()->headerData(i, Qt::Horizontal, Qt::EditRole).toString());
16
17 act->setData(i);
18 act->setCheckable(true);
19 act->setChecked(!header_view->isSectionHidden(i));
20
21 connect(act, &QAction::toggled, this, &TreeViewColumnsMenu::actionTriggered);
22 }
23 }
24
actionTriggered(bool toggle)25 void TreeViewColumnsMenu::actionTriggered(bool toggle) {
26 auto* send_act = qobject_cast<QAction*>(sender());
27
28 header()->setSectionHidden(send_act->data().toInt(), !send_act->isChecked());
29
30 Q_UNUSED(toggle)
31 }
32
header()33 QHeaderView* TreeViewColumnsMenu::header() {
34 return qobject_cast<QHeaderView*>(parent());
35 }
36