Sandboxie/SandboxiePlus/MiscHelpers/Common/TreeItemModel.h

#pragma once
#include "TreeViewEx.h"

#include "../mischelpers_global.h"

#if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
bool MISCHELPERS_EXPORT operator < (const QVariant& l, const QVariant& r);
#endif

class MISCHELPERS_EXPORT CTreeItemModel : public QAbstractItemModelEx
{
    Q_OBJECT

public:
    CTreeItemModel(QObject *parent = 0);
	virtual ~CTreeItemModel();

	void			SetUseIcons(bool bUseIcons)		{ m_bUseIcons = bUseIcons; }
	static void		SetDarkMode(bool bDark)			{ m_DarkMode = bDark;}
	static bool		GetDarkMode()					{ return m_DarkMode;}

	//void			CountItems();
	QModelIndex		FindIndex(const QVariant& ID);
	void			RemoveIndex(const QModelIndex &index);
	int				Count() const					{ return m_Map.count(); }

	QVariant		GetItemID(const QModelIndex& index) const;

	QVariant		Data(const QModelIndex &index, int role, int section) const;

	// derived functions
    virtual QVariant		data(const QModelIndex &index, int role) const;
	virtual bool			setData(const QModelIndex &index, const QVariant &value, int role);
    virtual Qt::ItemFlags	flags(const QModelIndex &index) const;
    virtual QModelIndex		index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
    virtual QModelIndex		parent(const QModelIndex &index) const;
    virtual int				rowCount(const QModelIndex &parent = QModelIndex()) const;
	virtual int				columnCount(const QModelIndex &parent = QModelIndex()) const = 0;
	virtual QVariant		headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const = 0;

public slots:
	virtual void			Clear();

signals:
	void			CheckChanged(const QVariant& ID, bool State);
	void			ToolTipCallback(const QVariant& ID, QString& ToolTip) const;
	void			Updated();

protected:
	struct STreeNode
	{
		STreeNode(CTreeItemModel* pModel, const QVariant& Id) {
			ID = Id;
			Parent = NULL;
			Row = 0;
			//AllChildren = 0;

			Virtual = false;

			IsBold = false;
			IsGray = false;

			Model = pModel;
			Model->m_Nodes.insert(this);
		}
		virtual ~STreeNode(){
			Model->m_Nodes.remove(this);
		}

		QVariant			ID;

		STreeNode*			Parent;
		int					Row;
		QList<QVariant>		Path;
		QList<STreeNode*>	Children;
		//int				AllChildren;
		//QMap<QVariant, int>	Aux;
		bool				Virtual;

		QVariant			Icon;
		bool				IsBold;
		bool				IsGray;
		QColor				Color;
		struct SValue
		{
			QVariant Raw;
			QVariant SortKey;
			QVariant Formatted;
		};
		QVector<SValue>		Values;
		CTreeItemModel*		Model;
	};

	virtual QVariant	NodeData(STreeNode* pNode, int role, int section) const;

	virtual STreeNode*	MkNode(const QVariant& Id) = 0;
	virtual void		FreeNode(STreeNode* pNode) { 
		foreach(STreeNode* pSubNode, pNode->Children)
			FreeNode(pSubNode);
		delete pNode; 
	}
	virtual STreeNode*	MkVirtualNode(const QVariant& Id, STreeNode* pParent);

	void			Sync(QMap<QList<QVariant>, QList<STreeNode*> >& New, QHash<QVariant, STreeNode*>& Old, QList<QModelIndex>* pNewBranches = NULL);
	void			Purge(STreeNode* pParent, const QModelIndex &parent, QHash<QVariant, STreeNode*>& Old);
	void			Fill(STreeNode* pParent, /*const QModelIndex &parent,*/ const QList<QVariant>& Paths, int PathsIndex, const QList<STreeNode*>& New, QList<QModelIndex>* pNewBranches);
	QModelIndex		Find(STreeNode* pParent, STreeNode* pNode);
	//int				CountItems(STreeNode* pRoot);

	virtual QVariant GetDefaultIcon() const { return QVariant(); }

	STreeNode*							m_Root;
	QHash<QVariant, STreeNode*>			m_Map;
	QSet<STreeNode*>					m_Nodes;
	bool								m_bUseIcons;

	static bool							m_DarkMode;
};

class MISCHELPERS_EXPORT CSimpleTreeModel : public CTreeItemModel
{
	Q_OBJECT

public:
	CSimpleTreeModel(QObject *parent = 0);
	virtual ~CSimpleTreeModel();
	
	void			SetTree(bool bTree)				{ m_bTree = bTree; }
	bool			IsTree() const					{ return m_bTree; }

	void					Sync(const QMap<QVariant, QVariantMap>& List);

	void					AddColumn(const QString& Name, const QString& Key) { m_ColumnKeys.append(qMakePair(Name, Key)); }

	virtual int				columnCount(const QModelIndex &parent = QModelIndex()) const;
    virtual QVariant		headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;

protected:
	virtual STreeNode*		MkNode(const QVariant& Id) { return new STreeNode(this, Id); }
	virtual void			FreeNode(STreeNode* pNode) { delete pNode; }

	QList<QVariant>			MakePath(const QVariantMap& Cur, const QMap<QVariant, QVariantMap>& List);
	bool					TestPath(const QList<QVariant>& Path, const QVariantMap& Cur, const QMap<QVariant, QVariantMap>& List, int Index = 0);

	bool								m_bTree;
	QList<QPair<QString, QString>> m_ColumnKeys;
};