org.eclipse.egerrit.ui/src/org/eclipse/egerrit/internal/ui/table/model/ReviewTableSorter.java

/*******************************************************************************
 * Copyright (c) 2015 Ericsson
 *
 * 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
 *
 * Contributors:
 *   Jacques Bouthillier - Initial Implementation of the view sorter
 ******************************************************************************/
package org.eclipse.egerrit.internal.ui.table.model;

import org.eclipse.egerrit.internal.model.impl.StringToFileInfoImpl;
import org.eclipse.egerrit.internal.ui.compare.GerritDiffNode;
import org.eclipse.egerrit.internal.ui.table.provider.FileInfoCompareCellLabelProvider;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.TableViewer;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.jface.viewers.ViewerComparator;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableColumn;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeColumn;

/**
 * This class implements the EGerrit UI view column sorter.
 *
 * @since 1.0
 */
public class ReviewTableSorter extends ViewerComparator {
	// The target column
	private int fColumnIndex = 0;

	public ReviewTableSorter(int columnIndex) {
		super();
		this.fColumnIndex = columnIndex;
	}

	public int getColumnSorter() {
		return fColumnIndex;
	}

	@Override
	public int compare(Viewer viewer, Object item1, Object item2) {
		int sortDirection = SWT.NONE;
		if (viewer instanceof TableViewer) {
			sortDirection = ((TableViewer) viewer).getTable().getSortDirection();
		} else if (viewer instanceof TreeViewer) {
			sortDirection = ((TreeViewer) viewer).getTree().getSortDirection();
		}
		int result = 0;
		if (viewer instanceof TableViewer) {
			if (fColumnIndex == 0) {
				result = getComparator().compare(
						Boolean.toString(((StringToFileInfoImpl) item1).getValue().isReviewed()),
						Boolean.toString(((StringToFileInfoImpl) item2).getValue().isReviewed()));
			} else {
				String l1 = ((ITableLabelProvider) ((TableViewer) viewer).getLabelProvider()).getColumnText(item1,
						fColumnIndex);
				String l2 = ((ITableLabelProvider) ((TableViewer) viewer).getLabelProvider()).getColumnText(item2,
						fColumnIndex);
				result = getComparator().compare(l1, l2);
			}
		} else {
			result = defaultCompare(viewer, item1, item2);
		}
		return sortDirection == SWT.UP ? -result : result;
	}

	private int defaultCompare(Viewer aViewer, Object aE1, Object aE2) {
		if (aViewer instanceof TreeViewer) {
			if ((fColumnIndex == 0) && (aE1 instanceof GerritDiffNode && aE2 instanceof GerritDiffNode)) {
				return getComparator().compare(Boolean.toString(((GerritDiffNode) aE1).getFileInfo().isReviewed()),
						Boolean.toString(((GerritDiffNode) aE2).getFileInfo().isReviewed()));
			}
			TreeViewer tv = (TreeViewer) aViewer;
			FileInfoCompareCellLabelProvider provider = (FileInfoCompareCellLabelProvider) tv
					.getLabelProvider(fColumnIndex);
			return getComparator().compare(provider.getLabel(aE1, fColumnIndex), provider.getLabel(aE2, fColumnIndex));
		}
		return 0;
	}

	// ------------------------------------------------------------------------
	// Static methods
	// ------------------------------------------------------------------------

	/**
	 * Bind a sorter to each table column
	 *
	 * @param aTableViewer
	 */
	public static void bind(final TableViewer aTableViewer) {
		for (int i = 0; i < aTableViewer.getTable().getColumnCount(); i++) {
			final int columnNum = i;
			TableColumn column = aTableViewer.getTable().getColumn(i);
			column.addSelectionListener(new SelectionAdapter() {
				@Override
				public void widgetSelected(final SelectionEvent e) {
					ReviewTableSorter sorter = new ReviewTableSorter(columnNum);
					Table table = aTableViewer.getTable();
					TableColumn currentColumn = (TableColumn) e.widget;
					table.setSortColumn(currentColumn);
					table.setSortDirection(table.getSortDirection() == SWT.UP ? SWT.DOWN : SWT.UP);
					aTableViewer.setComparator(sorter);
				}
			});
		}
	}

	/**
	 * Bind a sorter to each tree column
	 *
	 * @param aTreeViewer
	 */
	public static void bind(final TreeViewer aTreeViewer) {
		for (int i = 0; i < aTreeViewer.getTree().getColumnCount(); i++) {
			final int columnNum = i;
			TreeColumn column = aTreeViewer.getTree().getColumn(i);
			column.addSelectionListener(new SelectionAdapter() {
				@Override
				public void widgetSelected(final SelectionEvent e) {
					ReviewTableSorter sorter = new ReviewTableSorter(columnNum);
					Tree table = aTreeViewer.getTree();
					TreeColumn currentColumn = (TreeColumn) e.widget;
					table.setSortColumn(currentColumn);
					table.setSortDirection(table.getSortDirection() == SWT.UP ? SWT.DOWN : SWT.UP);
					aTreeViewer.setComparator(sorter);
				}
			});
		}
	}

}