Sandboxie/SandboxiePlus/SandMan/Models/ResMonModel.cpp

#include "stdafx.h"
#include "ResMonModel.h"
#include "../MiscHelpers/Common/Common.h"

CResMonModel::CResMonModel(QObject *parent)
:CListItemModel(parent)
{
}

CResMonModel::~CResMonModel()
{
}

void CResMonModel::Sync(const QList<CResLogEntryPtr>& List, QSet<quint64> PIDs)
{
	QList<SListNode*> New;
	QHash<QVariant, SListNode*> Old = m_Map;

	foreach (const CResLogEntryPtr& pEntry, List)
	{
		QVariant ID = pEntry->GetUID();

		if (!PIDs.isEmpty() && !PIDs.contains(pEntry->GetProcessId()))
			continue;

		int Row = -1;
		QHash<QVariant, SListNode*>::iterator I = Old.find(ID);
		SResLogNode* pNode = I != Old.end() ? static_cast<SResLogNode*>(I.value()) : NULL;
		if(!pNode)
		{
			pNode = static_cast<SResLogNode*>(MkNode(ID));
			pNode->Values.resize(columnCount());
			pNode->pEntry = pEntry;
			New.append(pNode);
		}
		else
		{
			I.value() = NULL;
			Row = GetRow(pNode);
		}

		int Col = 0;
		bool State = false;
		int Changed = 0;

		/*int RowColor = CTaskExplorer::eNone;
		if (pGDI->IsMarkedForRemoval() && CTaskExplorer::UseListColor(CTaskExplorer::eToBeRemoved))		RowColor = CTaskExplorer::eToBeRemoved;
		else if (pGDI->IsNewlyCreated() && CTaskExplorer::UseListColor(CTaskExplorer::eAdded))			RowColor = CTaskExplorer::eAdded;

		if (pNode->iColor != RowColor) {
			pNode->iColor = RowColor;
			pNode->Color = CTaskExplorer::GetListColor(RowColor);
			Changed = 2;
		}*/

		for(int section = 0; section < columnCount(); section++)
		{
			if (!m_Columns.contains(section))
				continue; // ignore columns which are hidden

			QVariant Value;
			switch(section)
			{
				case eProcess:			Value = pEntry->GetProcessId(); break;
				case eTimeStamp:		Value = pEntry->GetTimeStamp(); break;
				case eType:				Value = pEntry->GetType(); break;
				case eValue:			Value = pEntry->GetValue(); break;
			}

			SResLogNode::SValue& ColValue = pNode->Values[section];

			if (ColValue.Raw != Value)
			{
				if(Changed == 0)
					Changed = 1;
				ColValue.Raw = Value;

				switch (section)
				{
					case eProcess:			ColValue.Formated = QString::number(pEntry->GetProcessId()); break;
					case eTimeStamp:		ColValue.Formated = pEntry->GetTimeStamp().toString("hh:mm:ss.zzz"); break;
					//case eType:			ColValue.Formated = ; break;
					//case eValue:			ColValue.Formated = ; break;
				}
			}

			if(State != (Changed != 0))
			{
				if(State && Row != -1)
					emit dataChanged(createIndex(Row, Col), createIndex(Row, section-1));
				State = (Changed != 0);
				Col = section;
			}
			if(Changed == 1)
				Changed = 0;
		}
		if(State && Row != -1)
			emit dataChanged(createIndex(Row, Col, pNode), createIndex(Row, columnCount()-1, pNode));

	}

	CListItemModel::Sync(New, Old);
}

CResLogEntryPtr CResMonModel::GetEntry(const QModelIndex &index) const
{
	if (!index.isValid())
        return CResLogEntryPtr();

	SResLogNode* pNode = static_cast<SResLogNode*>(index.internalPointer());
	return pNode->pEntry;
}

int CResMonModel::columnCount(const QModelIndex &parent) const
{
	return eCount;
}

QVariant CResMonModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
	{
		switch(section)
		{
			case eProcess:			return tr("Process");
			case eTimeStamp:		return tr("Time Stamp");
			case eType:				return tr("Type");
			case eValue:			return tr("Value");
		}
	}
    return QVariant();
}