TreeItemManager.java

Go to the documentation of this file.

package edu.rice.cs.hpc.viewer.util;

import java.util.Stack;

import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;

import edu.rice.cs.hpc.data.experiment.scope.Scope;

public class TreeItemManager {

    private Stack<ElementDescriptor> stackElement;
    
    public TreeItemManager() {
        this.stackElement = new Stack<ElementDescriptor>();
    }
    
    public void saveContext(TreeViewer treeViewer) {
        final Tree tree = treeViewer.getTree();
        if (!tree.isDisposed())
        {
            ElementDescriptor objDesc = this.getTopItem(tree);
            if (objDesc != null)
                this.stackElement.push(objDesc);
        }
    }
    
    public void restoreContext(TreeViewer treeViewer) {
        if (this.stackElement.size()>0) {
            ElementDescriptor objDesc = this.stackElement.pop();
            Tree tree = treeViewer.getTree();
            if (!tree.isDisposed())
            {
                Utilities.insertTopRow(treeViewer, objDesc.imgElement, objDesc.sTextElement);
            }
        }
    }
    
    private ElementDescriptor getTopItem(Tree tree) {
        if (tree.getItemCount() == 0)
            return null;
        
        TreeItem itemTop = tree.getItem(0);
        Image imgItem = itemTop.getImage(0);
        String []sText= null;
        if(itemTop.getData() instanceof Scope) {
            // the table has been zoomed-out
        } else {
            // the table is in original form or flattened or zoom-in
            Object o = itemTop.getData();
            if(o != null) {
                Object []arrObj = (Object []) o;
                if(arrObj[0] instanceof String) {
                    sText = (String[]) itemTop.getData(); 
                }
            }
        }
        return new ElementDescriptor(itemTop, imgItem, sText);
    }

    private class ElementDescriptor {
        Image imgElement;
        String []sTextElement;
        
        
        ElementDescriptor (TreeItem item, Image img, String []sTexts) {
            this.imgElement = img;
            this.sTextElement = sTexts;
        }
    }

}