movies.ui.views.sort.TreeViewerColumnSorter.java Source code

Introduction

Here is the source code for movies.ui.views.sort.TreeViewerColumnSorter.java

Source

/**
 * <copyright> Copyright (c) 2008-2009 Jonas Helming, Maximilian Koegel. All rights reserved. This program and the
 * accompanying materials are made available under the terms of the Eclipse Public License v1.0 which accompanies this
 * distribution, and is available at http://www.eclipse.org/legal/epl-v10.html </copyright>
 */

package movies.ui.views.sort;

import org.eclipse.emf.ecp.common.ColumnViewerSorter;
import org.eclipse.jface.viewers.ColumnLabelProvider;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.TreeViewerColumn;
import org.eclipse.jface.viewers.ViewerColumn;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;

/**
 * This is a sorter class used to sort TreeViewers. Note this class is currently only used by StatusView and its
 * category method is tailored to handle WorkPackages and Annotations. It may be changed to an abstract class without
 * any implementation of category method.
 * 
 * @author hodaie
 */
public class TreeViewerColumnSorter extends ColumnViewerSorter {

    private TreeViewerColumn column;

    /**
     * Constructor.
     * 
     * @param viewer TreeViewer to be sorted
     * @param column TreeViewerColumn based on which the TreeViewer is sorted
     * @param columnLabelProvider LabelProvider used to sort the contents
     */
    public TreeViewerColumnSorter(TreeViewer viewer, TreeViewerColumn column,
            ColumnLabelProvider columnLabelProvider) {
        super(viewer, columnLabelProvider);
        this.column = column;

        column.getColumn().addSelectionListener(new SelectionAdapter() {

            @Override
            public void widgetSelected(SelectionEvent e) {
                if (getViewer().getComparator() != null) {
                    if (getViewer().getComparator() == TreeViewerColumnSorter.this) {
                        if (getDirection() == ASC) {
                            setSorter(TreeViewerColumnSorter.this, DESC);
                        } else {
                            setSorter(TreeViewerColumnSorter.this, ASC);
                        }
                    } else {
                        setSorter(TreeViewerColumnSorter.this, ASC);
                    }
                } else {
                    setSorter(TreeViewerColumnSorter.this, ASC);
                }
            }
        });
    }

    /**
     * This is used to set the right direction arrow at column header and refresh the viewer.
     * 
     * @param sorter sorter
     * @param direction direction
     */
    @Override
    protected void setSorter(ColumnViewerSorter sorter, int direction) {

        super.setSorter(sorter, direction);

        column.getColumn().getParent().setSortColumn(column.getColumn());
        if (direction == ASC) {
            column.getColumn().getParent().setSortDirection(SWT.DOWN);
        } else {
            column.getColumn().getParent().setSortDirection(SWT.UP);
        }

    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.emf.ecp.common.ColumnViewerSorter#getViewerColumn()
     */
    public ViewerColumn getViewerColumn() {
        return column;
    }

}