[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.ui / src / org / eclipse / linuxtools / tmf / ui / widgets / timegraph / TimeGraphCombo.java

/*******************************************************************************\r
 * Copyright (c) 2012 Ericsson\r
 *\r
 * All rights reserved. This program and the accompanying materials are\r
 * made available under the terms of the Eclipse Public License v1.0 which\r
 * accompanies this distribution, and is available at\r
 * http://www.eclipse.org/legal/epl-v10.html\r
 *\r
 * Contributors:\r
 *   Patrick Tasse - Initial API and implementation\r
 *******************************************************************************/\r
\r
package org.eclipse.linuxtools.tmf.ui.widgets.timegraph;\r
\r
import java.util.ArrayList;\r
import java.util.Arrays;\r
import java.util.HashMap;\r
\r
import org.eclipse.jface.viewers.ILabelProviderListener;\r
import org.eclipse.jface.viewers.ISelectionChangedListener;\r
import org.eclipse.jface.viewers.IStructuredSelection;\r
import org.eclipse.jface.viewers.ITableLabelProvider;\r
import org.eclipse.jface.viewers.ITreeContentProvider;\r
import org.eclipse.jface.viewers.ITreeViewerListener;\r
import org.eclipse.jface.viewers.SelectionChangedEvent;\r
import org.eclipse.jface.viewers.StructuredSelection;\r
import org.eclipse.jface.viewers.TreeExpansionEvent;\r
import org.eclipse.jface.viewers.TreeViewer;\r
import org.eclipse.jface.viewers.Viewer;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.model.ITimeGraphEntry;\r
import org.eclipse.swt.SWT;\r
import org.eclipse.swt.custom.SashForm;\r
import org.eclipse.swt.events.ControlAdapter;\r
import org.eclipse.swt.events.ControlEvent;\r
import org.eclipse.swt.events.MouseEvent;\r
import org.eclipse.swt.events.MouseTrackAdapter;\r
import org.eclipse.swt.events.MouseWheelListener;\r
import org.eclipse.swt.events.PaintEvent;\r
import org.eclipse.swt.events.PaintListener;\r
import org.eclipse.swt.events.SelectionAdapter;\r
import org.eclipse.swt.events.SelectionEvent;\r
import org.eclipse.swt.graphics.Image;\r
import org.eclipse.swt.graphics.Point;\r
import org.eclipse.swt.layout.FillLayout;\r
import org.eclipse.swt.widgets.Composite;\r
import org.eclipse.swt.widgets.Display;\r
import org.eclipse.swt.widgets.Event;\r
import org.eclipse.swt.widgets.Listener;\r
import org.eclipse.swt.widgets.Slider;\r
import org.eclipse.swt.widgets.Tree;\r
import org.eclipse.swt.widgets.TreeColumn;\r
import org.eclipse.swt.widgets.TreeItem;\r
\r
/**\r
 * Time graph "combo" view (with the list/tree on the left and the gantt chart\r
 * on the right)\r
 *\r
 * @version 1.0\r
 * @author Patrick Tasse\r
 */\r
public class TimeGraphCombo extends Composite {\r
\r
    // ------------------------------------------------------------------------\r
    // Constants\r
    // ------------------------------------------------------------------------\r
\r
    private static final Object FILLER = new Object();\r
\r
    // ------------------------------------------------------------------------\r
    // Fields\r
    // ------------------------------------------------------------------------\r
\r
    // The tree viewer\r
    private TreeViewer fTreeViewer;\r
\r
    // The time viewer\r
    private TimeGraphViewer fTimeGraphViewer;\r
\r
    // The selection listener map\r
    private final HashMap<ITimeGraphSelectionListener, SelectionListenerWrapper> fSelectionListenerMap = new HashMap<ITimeGraphSelectionListener, SelectionListenerWrapper>();\r
\r
    // Flag to block the tree selection changed listener when triggered by the time graph combo\r
    private boolean fInhibitTreeSelection = false;\r
\r
    // Number of filler rows used by the tree content provider\r
    private int fNumFillerRows;\r
\r
    // Calculated item height for Linux workaround\r
    private int fLinuxItemHeight = 0;\r
\r
    // ------------------------------------------------------------------------\r
    // Classes\r
    // ------------------------------------------------------------------------\r
\r
    /**\r
     * The TreeContentProviderWrapper is used to insert filler items after\r
     * the elements of the tree's real content provider.\r
     */\r
    private class TreeContentProviderWrapper implements ITreeContentProvider {\r
        private final ITreeContentProvider contentProvider;\r
\r
        public TreeContentProviderWrapper(ITreeContentProvider contentProvider) {\r
            this.contentProvider = contentProvider;\r
        }\r
\r
        @Override\r
        public void dispose() {\r
            contentProvider.dispose();\r
        }\r
\r
        @Override\r
        public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {\r
            contentProvider.inputChanged(viewer, oldInput, newInput);\r
        }\r
\r
        @Override\r
        public Object[] getElements(Object inputElement) {\r
            Object[] elements = contentProvider.getElements(inputElement);\r
            // add filler elements to ensure alignment with time analysis viewer\r
            Object[] oElements = Arrays.copyOf(elements, elements.length + fNumFillerRows, new Object[0].getClass());\r
            for (int i = 0; i < fNumFillerRows; i++) {\r
                oElements[elements.length + i] = FILLER;\r
            }\r
            return oElements;\r
        }\r
\r
        @Override\r
        public Object[] getChildren(Object parentElement) {\r
            if (parentElement instanceof ITimeGraphEntry) {\r
                return contentProvider.getChildren(parentElement);\r
            } else {\r
                return new Object[0];\r
            }\r
        }\r
\r
        @Override\r
        public Object getParent(Object element) {\r
            if (element instanceof ITimeGraphEntry) {\r
                return contentProvider.getParent(element);\r
            } else {\r
                return null;\r
            }\r
        }\r
\r
        @Override\r
        public boolean hasChildren(Object element) {\r
            if (element instanceof ITimeGraphEntry) {\r
                return contentProvider.hasChildren(element);\r
            } else {\r
                return false;\r
            }\r
        }\r
    }\r
\r
    /**\r
     * The TreeLabelProviderWrapper is used to intercept the filler items\r
     * from the calls to the tree's real label provider.\r
     */\r
    private class TreeLabelProviderWrapper implements ITableLabelProvider {\r
        private final ITableLabelProvider labelProvider;\r
\r
        public TreeLabelProviderWrapper(ITableLabelProvider labelProvider) {\r
            this.labelProvider = labelProvider;\r
        }\r
\r
        @Override\r
        public void addListener(ILabelProviderListener listener) {\r
            labelProvider.addListener(listener);\r
        }\r
\r
        @Override\r
        public void dispose() {\r
            labelProvider.dispose();\r
        }\r
\r
        @Override\r
        public boolean isLabelProperty(Object element, String property) {\r
            if (element instanceof ITimeGraphEntry) {\r
                return labelProvider.isLabelProperty(element, property);\r
            } else {\r
                return false;\r
            }\r
        }\r
\r
        @Override\r
        public void removeListener(ILabelProviderListener listener) {\r
            labelProvider.removeListener(listener);\r
        }\r
\r
        @Override\r
        public Image getColumnImage(Object element, int columnIndex) {\r
            if (element instanceof ITimeGraphEntry) {\r
                return labelProvider.getColumnImage(element, columnIndex);\r
            } else {\r
                return null;\r
            }\r
        }\r
\r
        @Override\r
        public String getColumnText(Object element, int columnIndex) {\r
            if (element instanceof ITimeGraphEntry) {\r
                return labelProvider.getColumnText(element, columnIndex);\r
            } else {\r
                return null;\r
            }\r
        }\r
\r
    }\r
\r
    /**\r
     * The SelectionListenerWrapper is used to intercept the filler items from\r
     * the time graph combo's real selection listener, and to prevent double\r
     * notifications from being sent when selection changes in both tree and\r
     * time graph at the same time.\r
     */\r
    private class SelectionListenerWrapper implements ISelectionChangedListener, ITimeGraphSelectionListener {\r
        private final ITimeGraphSelectionListener listener;\r
        private ITimeGraphEntry selection = null;\r
\r
        public SelectionListenerWrapper(ITimeGraphSelectionListener listener) {\r
            this.listener = listener;\r
        }\r
\r
        @Override\r
        public void selectionChanged(SelectionChangedEvent event) {\r
            if (fInhibitTreeSelection) {\r
                return;\r
            }\r
            Object element = ((IStructuredSelection) event.getSelection()).getFirstElement();\r
            if (element instanceof ITimeGraphEntry) {\r
                ITimeGraphEntry entry = (ITimeGraphEntry) element;\r
                if (entry != selection) {\r
                    selection = entry;\r
                    listener.selectionChanged(new TimeGraphSelectionEvent(event.getSource(), selection));\r
                }\r
            }\r
        }\r
\r
        @Override\r
        public void selectionChanged(TimeGraphSelectionEvent event) {\r
            ITimeGraphEntry entry = event.getSelection();\r
            if (entry != selection) {\r
                selection = entry;\r
                listener.selectionChanged(new TimeGraphSelectionEvent(event.getSource(), selection));\r
            }\r
        }\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Constructors\r
    // ------------------------------------------------------------------------\r
\r
    /**\r
     * Constructs a new instance of this class given its parent\r
     * and a style value describing its behavior and appearance.\r
     *\r
     * @param parent a widget which will be the parent of the new instance (cannot be null)\r
     * @param style the style of widget to construct\r
     */\r
    public TimeGraphCombo(Composite parent, int style) {\r
        super(parent, style);\r
        setLayout(new FillLayout());\r
\r
        final SashForm sash = new SashForm(this, SWT.NONE);\r
\r
        fTreeViewer = new TreeViewer(sash, SWT.FULL_SELECTION | SWT.H_SCROLL);\r
        final Tree tree = fTreeViewer.getTree();\r
        tree.setHeaderVisible(true);\r
        tree.setLinesVisible(true);\r
\r
        fTimeGraphViewer = new TimeGraphViewer(sash, SWT.NONE);\r
        fTimeGraphViewer.setItemHeight(getItemHeight(tree));\r
        fTimeGraphViewer.setHeaderHeight(tree.getHeaderHeight());\r
        fTimeGraphViewer.setBorderWidth(tree.getBorderWidth());\r
        fTimeGraphViewer.setNameWidthPref(0);\r
\r
        // Feature in Windows. The tree vertical bar reappears when\r
        // the control is resized so we need to hide it again.\r
        // Bug in Linux. The tree header height is 0 in constructor,\r
        // so we need to reset it later when the control is resized.\r
        tree.addControlListener(new ControlAdapter() {\r
            int depth = 0;\r
            @Override\r
            public void controlResized(ControlEvent e) {\r
                if (depth == 0) {\r
                    depth++;\r
                    tree.getVerticalBar().setEnabled(false);\r
                    // this can trigger controlResized recursively\r
                    tree.getVerticalBar().setVisible(false);\r
                    depth--;\r
                }\r
                fTimeGraphViewer.setHeaderHeight(tree.getHeaderHeight());\r
            }\r
        });\r
\r
        // ensure synchronization of expanded items between tree and time graph\r
        fTreeViewer.addTreeListener(new ITreeViewerListener() {\r
            @Override\r
            public void treeCollapsed(TreeExpansionEvent event) {\r
                fTimeGraphViewer.setExpandedState((ITimeGraphEntry) event.getElement(), false);\r
            }\r
\r
            @Override\r
            public void treeExpanded(TreeExpansionEvent event) {\r
                fTimeGraphViewer.setExpandedState((ITimeGraphEntry) event.getElement(), true);\r
            }\r
        });\r
\r
        // ensure synchronization of expanded items between tree and time graph\r
        fTimeGraphViewer.addTreeListener(new ITimeGraphTreeListener() {\r
            @Override\r
            public void treeCollapsed(TimeGraphTreeExpansionEvent event) {\r
                fTreeViewer.setExpandedState(event.getEntry(), false);\r
            }\r
\r
            @Override\r
            public void treeExpanded(TimeGraphTreeExpansionEvent event) {\r
                fTreeViewer.setExpandedState(event.getEntry(), true);\r
            }\r
        });\r
\r
        // prevent mouse button from selecting a filler tree item\r
        tree.addListener(SWT.MouseDown, new Listener() {\r
            @Override\r
            public void handleEvent(Event event) {\r
                TreeItem treeItem = tree.getItem(new Point(event.x, event.y));\r
                if (treeItem == null || treeItem.getData() == FILLER) {\r
                    event.doit = false;\r
                    ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                    if (treeItems.size() == 0) {\r
                        fTreeViewer.setSelection(new StructuredSelection());\r
                        fTimeGraphViewer.setSelection(null);\r
                        return;\r
                    }\r
                    // this prevents from scrolling up when selecting\r
                    // the partially visible tree item at the bottom\r
                    tree.select(treeItems.get(treeItems.size() - 1));\r
                    fTreeViewer.setSelection(new StructuredSelection());\r
                    fTimeGraphViewer.setSelection(null);\r
                }\r
            }\r
        });\r
\r
        // prevent mouse wheel from scrolling down into filler tree items\r
        tree.addListener(SWT.MouseWheel, new Listener() {\r
            @Override\r
            public void handleEvent(Event event) {\r
                event.doit = false;\r
                Slider scrollBar = fTimeGraphViewer.getVerticalBar();\r
                fTimeGraphViewer.setTopIndex(scrollBar.getSelection() - event.count);\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    return;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
            }\r
        });\r
\r
        // prevent key stroke from selecting a filler tree item\r
        tree.addListener(SWT.KeyDown, new Listener() {\r
            @Override\r
            public void handleEvent(Event event) {\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    fTreeViewer.setSelection(new StructuredSelection());\r
                    event.doit = false;\r
                    return;\r
                }\r
                if (event.keyCode == SWT.ARROW_DOWN) {\r
                    int index = Math.min(fTimeGraphViewer.getSelectionIndex() + 1, treeItems.size() - 1);\r
                    fTimeGraphViewer.setSelection((ITimeGraphEntry) treeItems.get(index).getData());\r
                    event.doit = false;\r
                } else if (event.keyCode == SWT.PAGE_DOWN) {\r
                    int height = tree.getSize().y - tree.getHeaderHeight() - tree.getHorizontalBar().getSize().y;\r
                    int countPerPage = height / getItemHeight(tree);\r
                    int index = Math.min(fTimeGraphViewer.getSelectionIndex() + countPerPage - 1, treeItems.size() - 1);\r
                    fTimeGraphViewer.setSelection((ITimeGraphEntry) treeItems.get(index).getData());\r
                    event.doit = false;\r
                } else if (event.keyCode == SWT.END) {\r
                    fTimeGraphViewer.setSelection((ITimeGraphEntry) treeItems.get(treeItems.size() - 1).getData());\r
                    event.doit = false;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
                if (fTimeGraphViewer.getSelectionIndex() >= 0) {\r
                    fTreeViewer.setSelection(new StructuredSelection(fTimeGraphViewer.getSelection()));\r
                } else {\r
                    fTreeViewer.setSelection(new StructuredSelection());\r
                }\r
            }\r
        });\r
\r
        // ensure alignment of top item between tree and time graph\r
        fTimeGraphViewer.getTimeGraphControl().addControlListener(new ControlAdapter() {\r
            @Override\r
            public void controlResized(ControlEvent e) {\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    return;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
            }\r
        });\r
\r
        // ensure synchronization of selected item between tree and time graph\r
        fTreeViewer.addSelectionChangedListener(new ISelectionChangedListener() {\r
            @Override\r
            public void selectionChanged(SelectionChangedEvent event) {\r
                if (fInhibitTreeSelection) {\r
                    return;\r
                }\r
                if (event.getSelection() instanceof IStructuredSelection) {\r
                    Object selection = ((IStructuredSelection) event.getSelection()).getFirstElement();\r
                    if (selection instanceof ITimeGraphEntry) {\r
                        fTimeGraphViewer.setSelection((ITimeGraphEntry) selection);\r
                    }\r
                    ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                    if (treeItems.size() == 0) {\r
                        return;\r
                    }\r
                    TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                    tree.setTopItem(treeItem);\r
                }\r
            }\r
        });\r
\r
        // ensure synchronization of selected item between tree and time graph\r
        fTimeGraphViewer.addSelectionListener(new ITimeGraphSelectionListener() {\r
            @Override\r
            public void selectionChanged(TimeGraphSelectionEvent event) {\r
                ITimeGraphEntry entry = fTimeGraphViewer.getSelection();\r
                fInhibitTreeSelection = true; // block the tree selection changed listener\r
                if (entry != null) {\r
                    StructuredSelection selection = new StructuredSelection(entry);\r
                    fTreeViewer.setSelection(selection);\r
                } else {\r
                    fTreeViewer.setSelection(new StructuredSelection());\r
                }\r
                fInhibitTreeSelection = false;\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    return;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
            }\r
        });\r
\r
        // ensure alignment of top item between tree and time graph\r
        fTimeGraphViewer.getVerticalBar().addSelectionListener(new SelectionAdapter() {\r
            @Override\r
            public void widgetSelected(SelectionEvent e) {\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    return;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
            }\r
        });\r
\r
        // ensure alignment of top item between tree and time graph\r
        fTimeGraphViewer.getTimeGraphControl().addMouseWheelListener(new MouseWheelListener() {\r
            @Override\r
            public void mouseScrolled(MouseEvent e) {\r
                ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
                if (treeItems.size() == 0) {\r
                    return;\r
                }\r
                TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
                tree.setTopItem(treeItem);\r
            }\r
        });\r
\r
        // ensure the tree has focus control when mouse is over it if the time graph had control\r
        fTreeViewer.getControl().addMouseTrackListener(new MouseTrackAdapter() {\r
            @Override\r
            public void mouseEnter(MouseEvent e) {\r
                if (fTimeGraphViewer.getTimeGraphControl().isFocusControl()) {\r
                    fTreeViewer.getControl().setFocus();\r
                }\r
            }\r
        });\r
\r
        // ensure the time graph has focus control when mouse is over it if the tree had control\r
        fTimeGraphViewer.getTimeGraphControl().addMouseTrackListener(new MouseTrackAdapter() {\r
            @Override\r
            public void mouseEnter(MouseEvent e) {\r
                if (fTreeViewer.getControl().isFocusControl()) {\r
                    fTimeGraphViewer.getTimeGraphControl().setFocus();\r
                }\r
            }\r
        });\r
        fTimeGraphViewer.getTimeGraphScale().addMouseTrackListener(new MouseTrackAdapter() {\r
            @Override\r
            public void mouseEnter(MouseEvent e) {\r
                if (fTreeViewer.getControl().isFocusControl()) {\r
                    fTimeGraphViewer.getTimeGraphControl().setFocus();\r
                }\r
            }\r
        });\r
\r
        // The filler rows are required to ensure alignment when the tree does not have a\r
        // visible horizontal scroll bar. The tree does not allow its top item to be set\r
        // to a value that would cause blank space to be drawn at the bottom of the tree.\r
        fNumFillerRows = Display.getDefault().getBounds().height / getItemHeight(tree);\r
\r
        sash.setWeights(new int[] { 1, 1 });\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Accessors\r
    // ------------------------------------------------------------------------\r
\r
    /**\r
     * Returns this time graph combo's tree viewer.\r
     *\r
     * @return the tree viewer\r
     */\r
    public TreeViewer getTreeViewer() {\r
        return fTreeViewer;\r
    }\r
\r
    /**\r
     * Returns this time graph combo's time graph viewer.\r
     *\r
     * @return the time graph viewer\r
     */\r
    public TimeGraphViewer getTimeGraphViewer() {\r
        return fTimeGraphViewer;\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Control\r
    // ------------------------------------------------------------------------\r
\r
    /* (non-Javadoc)\r
     * @see org.eclipse.swt.widgets.Control#redraw()\r
     */\r
    @Override\r
    public void redraw() {\r
        fTimeGraphViewer.getControl().redraw();\r
        super.redraw();\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Operations\r
    // ------------------------------------------------------------------------\r
\r
    /**\r
     * Sets the tree content provider used by this time graph combo.\r
     *\r
     * @param contentProvider the tree content provider\r
     */\r
    public void setTreeContentProvider(ITreeContentProvider contentProvider) {\r
        fTreeViewer.setContentProvider(new TreeContentProviderWrapper(contentProvider));\r
    }\r
\r
    /**\r
     * Sets the tree label provider used by this time graph combo.\r
     *\r
     * @param treeLabelProvider the tree label provider\r
     */\r
    public void setTreeLabelProvider(ITableLabelProvider labelProvider) {\r
        fTreeViewer.setLabelProvider(new TreeLabelProviderWrapper(labelProvider));\r
    }\r
\r
    /**\r
     * Sets the tree columns for this time graph combo.\r
     *\r
     * @param columnNames the tree column names\r
     */\r
    public void setTreeColumns(String[] columnNames) {\r
        final Tree tree = fTreeViewer.getTree();\r
        for (String columnName : columnNames) {\r
            TreeColumn column = new TreeColumn(tree, SWT.LEFT);\r
            column.setText(columnName);\r
            column.pack();\r
        }\r
    }\r
\r
\r
    /**\r
     * Sets the time graph provider used by this time graph combo.\r
     *\r
     * @param timeGraphProvider the time graph provider\r
     */\r
    public void setTimeGraphProvider(ITimeGraphPresentationProvider timeGraphProvider) {\r
        fTimeGraphViewer.setTimeGraphProvider(timeGraphProvider);\r
    }\r
\r
    /**\r
     * Sets or clears the input for this time graph combo.\r
     * The input array should only contain top-level elements.\r
     *\r
     * @param input the input of this time graph combo, or <code>null</code> if none\r
     */\r
    public void setInput(ITimeGraphEntry[] input) {\r
        fInhibitTreeSelection = true;\r
        fTreeViewer.setInput(input);\r
        fInhibitTreeSelection = false;\r
        fTreeViewer.expandAll();\r
        fTreeViewer.getTree().getVerticalBar().setEnabled(false);\r
        fTreeViewer.getTree().getVerticalBar().setVisible(false);\r
        fTimeGraphViewer.setItemHeight(getItemHeight(fTreeViewer.getTree()));\r
        fTimeGraphViewer.setInput(input);\r
    }\r
\r
    /**\r
     * Refreshes this time graph completely with information freshly obtained from its model.\r
     */\r
    public void refresh() {\r
        fTreeViewer.refresh();\r
        fTimeGraphViewer.refresh();\r
    }\r
\r
    /**\r
     * Adds a listener for selection changes in this time graph combo.\r
     *\r
     * @param listener a selection listener\r
     */\r
    public void addSelectionListener(ITimeGraphSelectionListener listener) {\r
        SelectionListenerWrapper listenerWrapper = new SelectionListenerWrapper(listener);\r
        fTreeViewer.addSelectionChangedListener(listenerWrapper);\r
        fSelectionListenerMap.put(listener, listenerWrapper);\r
        fTimeGraphViewer.addSelectionListener(listenerWrapper);\r
    }\r
\r
    /**\r
     * Removes the given selection listener from this time graph combo.\r
     *\r
     * @param listener a selection changed listener\r
     */\r
    public void removeSelectionListener(ITimeGraphSelectionListener listener) {\r
        SelectionListenerWrapper listenerWrapper = fSelectionListenerMap.remove(listener);\r
        fTreeViewer.removeSelectionChangedListener(listenerWrapper);\r
        fTimeGraphViewer.removeSelectionListener(listenerWrapper);\r
    }\r
\r
    /**\r
     * Sets the current selection for this time graph combo.\r
     *\r
     * @param selection the new selection\r
     */\r
    public void setSelection(ITimeGraphEntry selection) {\r
        fTimeGraphViewer.setSelection(selection);\r
        fInhibitTreeSelection = true; // block the tree selection changed listener\r
        if (selection != null) {\r
            StructuredSelection structuredSelection = new StructuredSelection(selection);\r
            fTreeViewer.setSelection(structuredSelection);\r
        } else {\r
            fTreeViewer.setSelection(new StructuredSelection());\r
        }\r
        fInhibitTreeSelection = false;\r
        ArrayList<TreeItem> treeItems = getVisibleExpandedItems(fTreeViewer.getTree());\r
        if (treeItems.size() == 0) {\r
            return;\r
        }\r
        TreeItem treeItem = treeItems.get(fTimeGraphViewer.getTopIndex());\r
        fTreeViewer.getTree().setTopItem(treeItem);\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Internal\r
    // ------------------------------------------------------------------------\r
\r
    private ArrayList<TreeItem> getVisibleExpandedItems(Tree tree) {\r
        ArrayList<TreeItem> items = new ArrayList<TreeItem>();\r
        for (TreeItem item : tree.getItems()) {\r
            if (item.getData() == FILLER) {\r
                break;\r
            }\r
            items.add(item);\r
            if (item.getExpanded()) {\r
                items.addAll(getVisibleExpandedItems(item));\r
            }\r
        }\r
        return items;\r
    }\r
\r
    private ArrayList<TreeItem> getVisibleExpandedItems(TreeItem treeItem) {\r
        ArrayList<TreeItem> items = new ArrayList<TreeItem>();\r
        for (TreeItem item : treeItem.getItems()) {\r
            items.add(item);\r
            if (item.getExpanded()) {\r
                items.addAll(getVisibleExpandedItems(item));\r
            }\r
        }\r
        return items;\r
    }\r
\r
    private int getItemHeight(final Tree tree) {\r
        /*\r
         * Bug in Linux.  The method getItemHeight doesn't always return the correct value.\r
         */\r
        if (fLinuxItemHeight >= 0 && System.getProperty("os.name").contains("Linux")) { //$NON-NLS-1$ //$NON-NLS-2$\r
            if (fLinuxItemHeight != 0) {\r
                return fLinuxItemHeight;\r
            }\r
            ArrayList<TreeItem> treeItems = getVisibleExpandedItems(tree);\r
            if (treeItems.size() > 1) {\r
                final TreeItem treeItem0 = treeItems.get(0);\r
                final TreeItem treeItem1 = treeItems.get(1);\r
                PaintListener paintListener = new PaintListener() {\r
                    @Override\r
                    public void paintControl(PaintEvent e) {\r
                        tree.removePaintListener(this);\r
                        int y0 = treeItem0.getBounds().y;\r
                        int y1 = treeItem1.getBounds().y;\r
                        int itemHeight = y1 - y0;\r
                        if (itemHeight > 0) {\r
                            fLinuxItemHeight = itemHeight;\r
                            fTimeGraphViewer.setItemHeight(itemHeight);\r
                        }\r
                    }\r
                };\r
                tree.addPaintListener(paintListener);\r
            }\r
        } else {\r
            fLinuxItemHeight = -1; // Not Linux, don't perform os.name check anymore\r
        }\r
        return tree.getItemHeight();\r
    }\r
\r
}\r