From 6afd790abeccc98e4319e392ed1f8d8dd78baa8f Mon Sep 17 00:00:00 2001
From: op2786 <op2786@proton.me>
Date: Mon, 31 Oct 2022 09:03:15 +0100
Subject: set address color to AddressColor for import and export tree

---
 examples/triage/exports.cpp | 32 ++++++++++++++++++++++----------
 examples/triage/imports.cpp | 38 ++++++++++++++++++++++++--------------
 2 files changed, 46 insertions(+), 24 deletions(-)

(limited to 'examples')

diff --git a/examples/triage/exports.cpp b/examples/triage/exports.cpp
index c4d2e678..452401f2 100644
--- a/examples/triage/exports.cpp
+++ b/examples/triage/exports.cpp
@@ -83,16 +83,28 @@ int GenericExportsModel::rowCount(const QModelIndex& parent) const
 
 QVariant GenericExportsModel::data(const QModelIndex& index, int role) const
 {
-	if (role != Qt::DisplayRole)
-		return QVariant();
-	if (index.row() >= (int)m_entries.size())
-		return QVariant();
-	if (index.column() == AddressColumn)
-		return QString("0x") + QString::number(m_entries[index.row()]->GetAddress(), 16);
-	if (index.column() == NameColumn)
-		return QString::fromStdString(m_entries[index.row()]->GetFullName());
-	if (index.column() == OrdinalColumn)
-		return QString::number(m_entries[index.row()]->GetOrdinal());
+	switch (role)
+	{
+	case Qt::DisplayRole:
+		if (role != Qt::DisplayRole)
+			return QVariant();
+		if (index.row() >= (int)m_entries.size())
+			return QVariant();
+		if (index.column() == AddressColumn)
+			return QString("0x") + QString::number(m_entries[index.row()]->GetAddress(), 16);
+		if (index.column() == NameColumn)
+			return QString::fromStdString(m_entries[index.row()]->GetFullName());
+		if (index.column() == OrdinalColumn)
+			return QString::number(m_entries[index.row()]->GetOrdinal());
+		break;
+	case Qt::ForegroundRole:
+		if (index.column() == AddressColumn)
+			return getThemeColor(AddressColor);
+		break;
+	default:
+		break;
+	}
+
 	return QVariant();
 }
 
diff --git a/examples/triage/imports.cpp b/examples/triage/imports.cpp
index b96457f8..e492bdd4 100644
--- a/examples/triage/imports.cpp
+++ b/examples/triage/imports.cpp
@@ -52,20 +52,30 @@ int GenericImportsModel::rowCount(const QModelIndex& parent) const
 
 QVariant GenericImportsModel::data(const QModelIndex& index, int role) const
 {
-	if (role != Qt::DisplayRole)
-		return QVariant();
-	if (index.row() >= (int)m_entries.size())
-		return QVariant();
-	if (index.column() == 0)
-		return QString("0x") + QString::number(m_entries[index.row()]->GetAddress(), 16);
-	if (index.column() == m_nameCol)
-		return QString::fromStdString(m_entries[index.row()]->GetFullName());
-	if (index.column() == m_moduleCol)
-		return getNamespace(m_entries[index.row()]);
-	if (index.column() == m_ordinalCol)
-		return QString::number(m_entries[index.row()]->GetOrdinal());
-	if (index.column() == m_typeLibCol)
-		return getLibrarySource(m_entries[index.row()]);
+	switch (role)
+	{
+	case Qt::DisplayRole:
+		if (index.row() >= (int)m_entries.size())
+			return QVariant();
+		if (index.column() == 0)
+			return QString("0x") + QString::number(m_entries[index.row()]->GetAddress(), 16);
+		if (index.column() == m_nameCol)
+			return QString::fromStdString(m_entries[index.row()]->GetFullName());
+		if (index.column() == m_moduleCol)
+			return getNamespace(m_entries[index.row()]);
+		if (index.column() == m_ordinalCol)
+			return QString::number(m_entries[index.row()]->GetOrdinal());
+		if (index.column() == m_typeLibCol)
+			return getLibrarySource(m_entries[index.row()]);
+		break;
+	case Qt::ForegroundRole:
+		if (index.column() == 0)
+			return getThemeColor(AddressColor);
+		break;
+	default:
+		break;
+	}
+
 	return QVariant();
 }
 
-- 
cgit v1.3.1