Source code: com/port80/eclipse/xml/editors/XMLOutlineContentProvider.java

   package com.port80.eclipse.xml.editors;
   
   import org.eclipse.core.resources.ResourcesPlugin;
   import org.eclipse.jface.viewers.ITreeContentProvider;
   import org.eclipse.jface.viewers.LabelProvider;
   import org.eclipse.jface.viewers.TreeViewer;
   import org.eclipse.jface.viewers.Viewer;
   import org.eclipse.swt.graphics.Image;
   
  /**
   * Data model for JavaccOutlinePage.
   * 
   * @author chrisl
   */
  public class XMLOutlineContentProvider extends LabelProvider implements ITreeContentProvider {
  
    ////////////////////////////////////////////////////////////////////////                                
  
    private static final String NAME = "XMLOutlineContentProvider";
    private static final boolean DEBUG = true;
  
    ////////////////////////////////////////////////////////////////////////                                
  
    private TreeFolder fInvisibleRoot;
    private TreeObject fTop;
  
    ////////////////////////////////////////////////////////////////////////                                
  
    /**
     * Constructor for MethodViewContentProvider.
     */
    public XMLOutlineContentProvider() {
      super();
      fInvisibleRoot = new TreeFolder("");
    }
  
    ////////////////////////////////////////////////////////////////////////                                
  
    public void setInput(TreeObject top) {
      fInvisibleRoot=new TreeFolder("");
      if (top != null)
        fInvisibleRoot.addChild(top);
      fTop = top;
    }
  
    // ITreeContentProvider /////////////////////////////////////////////
  
    public Object getParent(Object child) {
      if (child instanceof TreeObject) {
        return ((TreeObject) child).getParent();
      }
      return null;
    }
  
    public Object[] getChildren(Object parent) {
      if (parent instanceof TreeFolder) {
        return ((TreeFolder) parent).getChildren();
      }
      return new Object[0];
    }
  
    public boolean hasChildren(Object parent) {
      if (parent instanceof TreeFolder)
        return ((TreeFolder) parent).hasChildren();
      return false;
    }
  
    // IStructuredContentProvider interface ///////////////////////////
  
    public Object[] getElements(Object parent) {
      if (parent == null || parent == ResourcesPlugin.getWorkspace()) {
        return getChildren(fInvisibleRoot);
      }
      return getChildren(parent);
    }
  
    // IContentProvider interface /////////////////////////////////
  
    public void inputChanged(Viewer v, Object oldInput, Object newInput) {
    }
  
    public void dispose() {
    }
  
    // LabelProvider interface /////////////////////////////////////
  
    public String getText(Object a) {
      return a.toString();
    }
  
    public Image getImage(Object a) {
      return ((TreeObject) a).getImage();
    }
  
    ////////////////////////////////////////////////////////////////////////                                
  
    public TreeFolder getRoot() {
      return fInvisibleRoot;
    }
 
   public TreeObject getTop() {
     return fTop;
   }
 
   public void expandTree(TreeViewer viewer) {
     viewer.expandToLevel(3);
   }
 
   public TreeObject findElement(Object a) {
     return findElement(a, fTop);
   }
 
   /** 
    * Find object 'a' in the given 'subtree' base on comparison of its encasuplated element.
    */
   private TreeObject findElement(Object a, TreeObject subtree) {
     if (a == null)
       return null;
     if (a.equals(subtree.getElement()))
       return subtree;
     if (subtree instanceof TreeFolder) {
       TreeFolder folder = (TreeFolder) subtree;
       TreeObject item;
       for (int i = 0; i < folder.getNumChildren(); ++i) {
         item = findElement(a, folder.getChild(i));
         if (item != null)
           return item;
       }
     }
     return null;
   }
 
   ////////////////////////////////////////////////////////////////////////
 
 }