net.refractions.udig.issues.internal.view.IssuesSorter.java Source code

Introduction

Here is the source code for net.refractions.udig.issues.internal.view.IssuesSorter.java

Source

/* uDig - User Friendly Desktop Internet GIS client
 * http://udig.refractions.net
 * (C) 2004, Refractions Research Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 */
package net.refractions.udig.issues.internal.view;

import net.refractions.udig.issues.Column;
import net.refractions.udig.issues.IIssue;
import net.refractions.udig.issues.IIssuesViewSorter;

import org.eclipse.jface.viewers.ContentViewer;
import org.eclipse.jface.viewers.IBaseLabelProvider;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.jface.viewers.ViewerSorter;

/**
 * Sort the viewer by one of the columns
 * 
 * @author jones
 * @since 1.0.0
 */
public class IssuesSorter implements IIssuesViewSorter {
    public int compare(Viewer viewer, ViewerSorter defaultSorter, Column selectedColumn, boolean direction,
            Object e1, Object e2) {
        if (!(e1 instanceof IIssue))
            return defaultSorter.compare(viewer, e1, e2);

        IIssue issue1 = (IIssue) e1;
        IIssue issue2 = (IIssue) e2;

        if (!direction) {
            issue1 = (IIssue) e2;
            issue2 = (IIssue) e1;
        }

        if (viewer instanceof ContentViewer) {
            ContentViewer tviewer = (ContentViewer) viewer;
            IBaseLabelProvider provider = tviewer.getLabelProvider();
            if (provider instanceof ITableLabelProvider) {
                ITableLabelProvider tableProvider = (ITableLabelProvider) provider;
                String text1 = tableProvider.getColumnText(issue1, IssuesView.columnToIndex(selectedColumn));
                String text2 = tableProvider.getColumnText(issue2, IssuesView.columnToIndex(selectedColumn));
                if (text1 != null && text2 != null) {
                    return defaultSorter.compare(viewer, text1, text2);
                }

                switch (selectedColumn) {
                case PRIORITY:
                    return defaultSorter.compare(null, ((IIssue) issue1).getPriority().ordinal(),
                            ((IIssue) issue2).getPriority().ordinal());
                case RESOLUTION:
                    return defaultSorter.compare(null, ((IIssue) issue1).getResolution().ordinal(),
                            ((IIssue) issue2).getResolution().ordinal());

                }
            }
        }

        return defaultSorter.compare(viewer, issue1, issue2);
    }

    public String getExtensionID() {
        return null;
    }

    public int category(ViewerSorter defaultSorter, Object element) {
        return defaultSorter.category(element);
    }

    public boolean isSorterProperty(ViewerSorter defaultSorter, Object element, String property) {
        return defaultSorter.isSorterProperty(element, property);
    }
}