set address color to AddressColor for import and export tree

1 files changed, 24 insertions, 14 deletions

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();
 }