Sandboxie/SandboxiePlus/MiscHelpers/Common/TreeWidgetEx.h

#pragma once

#include "../mischelpers_global.h"
#include "TreeViewEx.h"

class MISCHELPERS_EXPORT QTreeWidgetEx: public QTreeWidget
{
	Q_OBJECT
public:
	QTreeWidgetEx(QWidget *parent = 0) : QTreeWidget(parent) 
	{
		setUniformRowHeights(true);

		m_AutoFitMax = 0;
		m_ColumnReset = 1;

		header()->setContextMenuPolicy(Qt::CustomContextMenu);
		connect(header(), SIGNAL(customContextMenuRequested( const QPoint& )), this, SLOT(OnMenuRequested(const QPoint &)));

		m_pMenu = new QMenu(this);

        // Important: if something is shown/hidden we need a new size
		connect(this->model(), SIGNAL(rowsInserted(const QModelIndex &, int, int)), this, SLOT(OnExpandCollapsed()));
		connect(this->model(), SIGNAL(rowsRemoved(const QModelIndex &, int, int)), this, SLOT(OnExpandCollapsed()));
		//connect(this->header(), SIGNAL(geometriesChanged()), this, SLOT(OnExpandCollapsed())); // fixme
        connect(this, SIGNAL(expanded(const QModelIndex &)), this, SLOT(OnExpandCollapsed()));
        connect(this, SIGNAL(collapsed(const QModelIndex &)), this, SLOT(OnExpandCollapsed()));
	}

	static void AddSubItem(QTreeWidgetItem* pRoot, const QString& Key, const QString& Value)
	{
		QTreeWidgetItem* pItem = new QTreeWidgetItem(QStringList(Key));
		pItem->setText(1, Value);
		pRoot->addChild(pItem);
	}

	// Use this function to make the widget autosize to show all items up to this value
	void setAutoFitMax(int AutoFitMax)
	{
		m_AutoFitMax = AutoFitMax;
	}

	void setColumnReset(int iMode)
	{
		m_ColumnReset = iMode;
	}

	void setColumnFixed(int column, bool fixed) 
	{
		if (fixed)
			m_FixedColumns.insert(column);
		else
			m_FixedColumns.remove(column);
	}

	bool isColumnFixed(int column) const
	{
		return m_FixedColumns.contains(column);
	}

    QSize sizeHint() const {return m_AutoFitMax ? MySize() : QTreeWidget::sizeHint(); };
    QSize minimumSizeHint() const { return m_AutoFitMax ? MySize() : QTreeWidget::sizeHint(); };

signals:
	void ResetColumns();

public slots:
	void OnResetColumns()
	{
		for (int i = 0; i < columnCount(); i++) {
			setColumnHidden(i, false);
			QTreeView::resizeColumnToContents(i);
		}
	}

private slots:
	void OnMenuRequested(const QPoint &point)
	{
		if(m_pMenu->actions().isEmpty())
		{
			QTreeWidgetItem* pHeader = headerItem();
			for(int i=0; i < columnCount(); i++)
			{
				QString Label = pHeader->text(i);
				if(Label.isEmpty() || m_FixedColumns.contains(i))
					continue;
				QAction* pAction = new QAction(Label, m_pMenu);
				pAction->setCheckable(true);
				connect(pAction, SIGNAL(triggered()), this, SLOT(OnMenu()));
				m_pMenu->addAction(pAction);
				m_Columns[pAction] = i;
			}

			if (m_ColumnReset)
			{
				m_pMenu->addSeparator();
				QAction* pAction = m_pMenu->addAction(QTreeViewEx::m_ResetColumns);
				if(m_ColumnReset == 1)
					connect(pAction, SIGNAL(triggered()), this, SLOT(OnResetColumns()));
				else
					connect(pAction, SIGNAL(triggered()), this, SIGNAL(ResetColumns()));
			}
		}

		for(QMap<QAction*, int>::iterator I = m_Columns.begin(); I != m_Columns.end(); I++)
			I.key()->setChecked(!isColumnHidden(I.value()));

		m_pMenu->popup(QCursor::pos());	
	}

	void OnMenu()
	{
		QAction* pAction = (QAction*)sender();
		int Column = m_Columns.value(pAction, -1);
		setColumnHidden(Column, !pAction->isChecked());
		QTimer::singleShot(10, this, SLOT(OnExpandCollapsed()));
	}

	void OnExpandCollapsed() 
	{
		if (m_AutoFitMax) 
			updateGeometry();
	}

protected:
    QSize MySize() const
    {   //QSize tst(sizeHintForColumn(0) + 2 * frameWidth(), sizeHintForRow(0) + 2 * frameWidth());

        int neededHight= 2 * frameWidth()+ this->header()->height();

        QAbstractItemModel* m = this->model();

		QModelIndex root = this->rootIndex();
        //if(this->rootIsDecorated())
        neededHight += recursiveHeightHint(root,m);
        
		if (this->horizontalScrollBar()->isVisible())
			neededHight += this->horizontalScrollBar()->height();

		if (neededHight > m_AutoFitMax)
			neededHight = m_AutoFitMax;

        QSize temp = QTreeView::sizeHint();
        temp.setHeight(neededHight);
        return QSize(1,neededHight);
    }

    // we need the size of all visible items -> isExpanded
    // the root item is usually shown as a non-Valid index -> !i.isValid()
    int recursiveHeightHint(QModelIndex i,QAbstractItemModel* m) const
    {
        int temp=sizeHintForIndex(i).height() + 1;
        if(this->isExpanded(i) || !i.isValid())
        {
			if(m->hasChildren(i))
			{
				int numRows = m->rowCount(i);
				for(int count =0;count<numRows;count++)
					temp+=recursiveHeightHint(m->index(count,0,i),m);
			}
        }
        return temp;
    }

	QMenu*				m_pMenu;
	QMap<QAction*, int>	m_Columns;
	int					m_AutoFitMax;
	QSet<int>			m_FixedColumns;
	int					m_ColumnReset;
};