X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.utils.ui.workbench%2Fsrc%2Forg%2Fsimantics%2Futils%2Fui%2Fworkbench%2Fui%2FTableColumnSorter.java;fp=bundles%2Forg.simantics.utils.ui.workbench%2Fsrc%2Forg%2Fsimantics%2Futils%2Fui%2Fworkbench%2Fui%2FTableColumnSorter.java;h=2ce76b50144c25e59f4cce5e28d99acb77a6be4b;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.utils.ui.workbench/src/org/simantics/utils/ui/workbench/ui/TableColumnSorter.java b/bundles/org.simantics.utils.ui.workbench/src/org/simantics/utils/ui/workbench/ui/TableColumnSorter.java
new file mode 100644
index 000000000..2ce76b501
--- /dev/null
+++ b/bundles/org.simantics.utils.ui.workbench/src/org/simantics/utils/ui/workbench/ui/TableColumnSorter.java
@@ -0,0 +1,236 @@
+/*******************************************************************************
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * 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:
+ *     VTT Technical Research Centre of Finland - initial API and implementation
+ *******************************************************************************/
+package org.simantics.utils.ui.workbench.ui;
+
+import java.util.Comparator;
+import java.util.HashMap;
+import java.util.Map;
+
+import org.eclipse.jface.viewers.IBaseLabelProvider;
+import org.eclipse.jface.viewers.ITableLabelProvider;
+import org.eclipse.jface.viewers.TableViewer;
+import org.eclipse.jface.viewers.Viewer;
+import org.eclipse.jface.viewers.ViewerSorter;
+import org.eclipse.swt.events.SelectionAdapter;
+import org.eclipse.swt.events.SelectionEvent;
+import org.eclipse.swt.widgets.TableColumn;
+import org.eclipse.ui.IMemento;
+
+/**
+ * This class adds column sorting functionality to SWT Table widget.
+ * <p>
+ * Attach this Sorter after you have created all columns.
+ * <p>
+ * Usage: TableColumnSorter.attachTableColumnSorter(myTable);
+ * 
+ * @author Toni Kalajainen
+ */
+public class TableColumnSorter {
+
+    /** The table */
+    protected final TableViewer viewer;
+    
+    /** the sorter */
+    protected final ColumnSorter sorter;
+    
+    /** column specific comparators*/
+    protected Map<Integer, Comparator<?>> columnComparators = 
+        new HashMap<Integer, Comparator<?>>();    
+    
+    public static TableColumnSorter attachTableColumnSorter(TableViewer viewer) {
+        return new TableColumnSorter(viewer);
+    }
+    
+    public static void unattachTableColumnSorter(TableViewer viewer) {
+        viewer.setSorter(null);
+    }
+    
+    public void setColumnComparator(int index, Comparator<?> comparator)
+    {
+        columnComparators.put(index, comparator);
+        viewer.refresh();
+    }
+    
+    private final static String PRI_ASC = "PriAsc";
+    private final static String SEC_ASC = "SecAsc";
+    private final static String PRI_IND = "PriInd";
+    private final static String SEC_IND = "SecInd";
+    
+    public void saveState(String id, IMemento memento) {
+        memento.putInteger(id+PRI_ASC, sorter.isAscending()?1:0);
+        memento.putInteger(id+SEC_ASC, sorter.isSecondaryAscending()?1:0);        
+        memento.putInteger(id+PRI_IND, sorter.getIndex());
+        memento.putInteger(id+SEC_IND, sorter.getSecondaryIndex());
+    }
+
+    public void restoreState(String id, IMemento memento) {
+        if (!hasState(id, memento)) return;
+            
+        sorter.setAscending( memento.getInteger(id+PRI_ASC)==1 );
+        sorter.setSecondaryAscending( memento.getInteger(id+SEC_ASC)==1 );
+        sorter.setIndex( memento.getInteger(id+PRI_IND) );
+        sorter.setSecondaryIndex( memento.getInteger(id+SEC_IND) );
+        
+        viewer.refresh();
+    }
+    
+    public boolean hasState(String id, IMemento memento) {
+        return (memento.getInteger(id+PRI_ASC)!=null) && 
+               (memento.getInteger(id+SEC_ASC)!=null) && 
+               (memento.getInteger(id+PRI_IND)!=null) && 
+               (memento.getInteger(id+SEC_IND)!=null); 
+    }
+    
+    public void setColumnAscending(int index)
+    {
+        sorter.setIndex(index);
+        sorter.setAscending(true);
+        viewer.refresh();
+    }    
+    
+    private TableColumnSorter(TableViewer viewer) {
+        this.viewer = viewer;
+        this.sorter = new ColumnSorter();
+        
+        // Attach columns
+        TableColumn columns[] = viewer.getTable().getColumns(); 
+        for(int i=0; i<columns.length; i++) {
+            TableColumn column = columns[i];
+            column.setData("index", new Integer(i));
+            column.addSelectionListener(new SelectionAdapter() {
+                public void widgetSelected(SelectionEvent e) {                    
+                    int index = (Integer) e.widget.getData("index");
+                    if (index == TableColumnSorter.this.sorter.getIndex()) {
+                        // Reverse order
+                        TableColumnSorter.this.sorter.setAscending( !TableColumnSorter.this.sorter.isAscending() );
+                    } else {
+                        TableColumnSorter.this.sorter.setSecondaryIndex( TableColumnSorter.this.sorter.getIndex() );
+                        TableColumnSorter.this.sorter.setSecondaryAscending( TableColumnSorter.this.sorter.isAscending() );
+                        TableColumnSorter.this.sorter.setIndex(index);
+                        TableColumnSorter.this.sorter.setAscending(true);
+                    }
+                    TableColumnSorter.this.viewer.refresh();
+                }
+            });
+        }
+        viewer.setSorter(sorter);
+    }
+    
+    
+    class ColumnSorter extends ViewerSorter {
+        /** Sort direction */
+        private boolean ascending = true;
+        /** Secondary Sort direction */
+        private boolean secondaryAscending = true;
+        /** Sort column */
+        private int columnIndex = 0;
+        /** Secondary sort column */
+        private int secondaryColumnIndex = -1;
+        /** case sensitive */
+        private boolean caseSensitive = false;
+        
+        public void setAscending(boolean ascending) {
+            this.ascending = ascending;
+        }
+                
+        public boolean isAscending() {
+            return ascending;
+        }        
+        
+        public void setSecondaryAscending(boolean ascending) {
+            this.secondaryAscending = ascending;
+        }
+                
+        public boolean isSecondaryAscending() {
+            return secondaryAscending;
+        }        
+        
+        public void setIndex(int index) {
+            this.columnIndex = index;
+        }
+        
+        public int getIndex() {
+            return columnIndex;
+        }
+        
+        public void setSecondaryIndex(int index) {
+            this.secondaryColumnIndex = index;
+        }
+        
+        public int getSecondaryIndex() {
+            return secondaryColumnIndex;
+        }
+
+        @SuppressWarnings("unchecked")
+        private int compare(int columnIndex, String text1, String text2, Object o1, Object o2)
+        {
+            @SuppressWarnings("rawtypes")
+            Comparator c = columnComparators.get(columnIndex);
+            if (c==null || o1==null || o2==null)
+                return text1.compareTo(text2);
+            return c.compare(o1, o2);
+        }
+        
+        public int compare(Viewer viewer, Object e1, Object e2) {
+            int result = 0;
+            
+            TableViewer v = (TableViewer) viewer;            
+            IBaseLabelProvider blp = v.getLabelProvider();
+            if (!(blp instanceof ITableLabelProvider)) {
+                return super.compare(viewer, e1, e2);
+            }
+            ITableLabelProvider tlp = (ITableLabelProvider) blp;
+            
+            // Primary sort
+            String text1 = tlp.getColumnText(e1, columnIndex);
+            String text2 = tlp.getColumnText(e2, columnIndex);
+            if (text1==null) text1="";
+            if (text2==null) text2="";            
+            if (!caseSensitive) {
+                text1 = text1.toLowerCase();
+                text2 = text2.toLowerCase();
+            }            
+            result = compare(columnIndex, text1, text2, e1, e2);            
+            if (!ascending) return -result;
+            
+            // secondary sort
+            if (result==0 && (secondaryColumnIndex>=0)) {
+                text1 = tlp.getColumnText(e1, secondaryColumnIndex);
+                text2 = tlp.getColumnText(e2, secondaryColumnIndex);
+                if (text1==null) text1="";
+                if (text2==null) text2="";            
+                if (!caseSensitive) {
+                    text1 = text1.toLowerCase();
+                    text2 = text2.toLowerCase();
+                }            
+                result = compare(secondaryColumnIndex, text1, text2, e1, e2);
+                
+                if (!secondaryAscending) return -result;
+            }
+            
+            return result;
+        }
+
+        public boolean isCaseSensitive() {
+            return caseSensitive;
+        }
+
+        public void setCaseSensitive(boolean caseSensitive) {
+            this.caseSensitive = caseSensitive;
+        }
+    }
+
+    public ColumnSorter getSorter() {
+        return sorter;
+    }
+
+}