From b3e82309c9f5f8fa7e1bb61e8fed4d6a0107d449 Mon Sep 17 00:00:00 2001
From: Alain Magloire <alain@qnx.com>
Date: Thu, 2 Sep 2004 01:00:51 +0000
Subject: [PATCH] 2004-09-01 Alain Magloire

	Deal with the fact that elements can come
	from WorkingCopy or TranslationUnit
	* src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java
---
 core/org.eclipse.cdt.ui/ChangeLog             |  6 ++++
 .../ui/cview/CViewElementComparer.java        | 31 +++++++++++--------
 2 files changed, 24 insertions(+), 13 deletions(-)

diff --git a/core/org.eclipse.cdt.ui/ChangeLog b/core/org.eclipse.cdt.ui/ChangeLog
index 29b4f129d97..f6b789593e8 100644
--- a/core/org.eclipse.cdt.ui/ChangeLog
+++ b/core/org.eclipse.cdt.ui/ChangeLog
@@ -1,3 +1,9 @@
+2004-09-01 Alain Magloire
+
+	Deal with the fact that elements can come
+	from WorkingCopy or TranslationUnit
+	* src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java
+
 2004-08-31 Alain Magloire
 
 	Provide a Proxy wrapper for the includeRefence class
diff --git a/core/org.eclipse.cdt.ui/src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java b/core/org.eclipse.cdt.ui/src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java
index 46477072eac..332fa3d1cd8 100644
--- a/core/org.eclipse.cdt.ui/src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java
+++ b/core/org.eclipse.cdt.ui/src/org/eclipse/cdt/internal/ui/cview/CViewElementComparer.java
@@ -31,21 +31,26 @@ public class CViewElementComparer implements IElementComparer {
 		if (c1 == null || c2 == null)
 			return false;
 
-		if (c1 instanceof ITranslationUnit) {
-			ITranslationUnit t1 = (ITranslationUnit)o1;
-			if (t1.isWorkingCopy()) {
-				c1 = ((IWorkingCopy)t1).getOriginalElement();
-			}
-		}
-		if (c2 instanceof ITranslationUnit) {
-			ITranslationUnit t2 = (ITranslationUnit)o2;
-			if (t2.isWorkingCopy()) {
-				c2 = ((IWorkingCopy)t2).getOriginalElement(); 
-			}
-		}
-		if (c1 == null || c2 == null) {
+		// Below is for children of TranslationUnits but we have to make sure
+		// we handle the case that the child comes from the a workingCopy in that
+		// case it should be equal as the original element.
+		ITranslationUnit u1 = (ITranslationUnit)c1.getAncestor(ICElement.C_UNIT);
+		ITranslationUnit u2 = (ITranslationUnit)c2.getAncestor(ICElement.C_UNIT);
+		if (u1 == null || u2 == null) {
 			return false;
 		}
+		
+		if (u1.isWorkingCopy() && u2.isWorkingCopy() || !u1.isWorkingCopy() && !u2.isWorkingCopy()) {
+			return false;
+		}
+		// From here on either c1 or c2 is a working copy.
+		if (u1.isWorkingCopy()) {
+			c1= ((IWorkingCopy)u1).getOriginal(c1);
+		} else if (u2.isWorkingCopy()) {
+			c2= ((IWorkingCopy)u2).getOriginal(c2); 
+		}
+		if (c1 == null || c2 == null)
+			return false;
 		return c1.equals(c2);
 	}