From 0167673957075286cf4f19b99fbbcdab29083ca9 Mon Sep 17 00:00:00 2001
From: John Camelon <jcamelon@ca.ibm.com>
Date: Mon, 7 Feb 2005 14:11:24 +0000
Subject: [PATCH] Patch for Devin Steffler re: DOMAST View.

I finally figured out how to refresh the TreeViewer in a manner that will let me determine programmatically what tree nodes to expand.
So now after a refresh, if a treenode has the same offset (via getOffset) and toString() then it is expanded.
---
 .../eclipse/cdt/ui/tests/DOMAST/DOMAST.java   | 25 ++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/core/org.eclipse.cdt.ui.tests/src/org/eclipse/cdt/ui/tests/DOMAST/DOMAST.java b/core/org.eclipse.cdt.ui.tests/src/org/eclipse/cdt/ui/tests/DOMAST/DOMAST.java
index e850a7a5e25..fff570f6ec2 100644
--- a/core/org.eclipse.cdt.ui.tests/src/org/eclipse/cdt/ui/tests/DOMAST/DOMAST.java
+++ b/core/org.eclipse.cdt.ui.tests/src/org/eclipse/cdt/ui/tests/DOMAST/DOMAST.java
@@ -79,6 +79,8 @@ import org.eclipse.swt.SWT;
 import org.eclipse.swt.graphics.Image;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Menu;
+import org.eclipse.swt.widgets.Tree;
+import org.eclipse.swt.widgets.TreeItem;
 import org.eclipse.ui.IActionBars;
 import org.eclipse.ui.IEditorPart;
 import org.eclipse.ui.IEditorReference;
@@ -471,14 +473,31 @@ public class DOMAST extends ViewPart {
      loadActiveEditorAction.setImageDescriptor(DOMASTPluginImages.DESC_DEFAULT);
    	
      refreshAction = new Action() {
+     	private void expandTreeIfNecessary(TreeItem[] tree, Object[] expanded) {
+     		for( int i=0; i<tree.length; i++) {
+     			for( int j=0; j<expanded.length; j++) {
+	     			if (expanded[j] instanceof TreeObject &&
+	     					tree[i].getData() instanceof TreeObject &&
+	     					((TreeObject)expanded[j]).toString().equals(((TreeObject)tree[i].getData()).toString()) && 
+	     					((TreeObject)expanded[j]).getOffset() == (((TreeObject)tree[i].getData()).getOffset())) {
+	     				tree[i].setExpanded(true);
+	     				viewer.refresh();
+	     				expandTreeIfNecessary(tree[i].getItems(), expanded);
+	     			}
+     			}
+     		}
+     	}
+     	
          public void run() {
-            // TODO take a snapshot of the tree expansion
-         	
+            // take a snapshot of the tree expansion
+         	Object[] expanded = viewer.getExpandedElements();
          	
          	// set the new content provider
          	setContentProvider(new ViewContentProvider(file));
             
-         	// TODO set the expansion of the view based on the original snapshot (educated guess)
+         	// set the expansion of the view based on the original snapshot (educated guess)
+         	Tree tree = viewer.getTree();
+         	expandTreeIfNecessary(tree.getItems(), expanded);
          	
          }
       };