Coverage Report - org.seasar.uruma.viewer.GenericTableViewerComparator
 
Classes in this File Line Coverage Branch Coverage Complexity
GenericTableViewerComparator
59%
35/59
40%
12/30
0
GenericTableViewerComparator$1
20%
1/5
N/A
0
 
 1  
 /*
 2  
  * Copyright 2004-2008 the Seasar Foundation and the Others.
 3  
  *
 4  
  * Licensed under the Apache License, Version 2.0 (the "License");
 5  
  * you may not use this file except in compliance with the License.
 6  
  * You may obtain a copy of the License at
 7  
  *
 8  
  *     http://www.apache.org/licenses/LICENSE-2.0
 9  
  *
 10  
  * Unless required by applicable law or agreed to in writing, software
 11  
  * distributed under the License is distributed on an "AS IS" BASIS,
 12  
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
 13  
  * either express or implied. See the License for the specific language
 14  
  * governing permissions and limitations under the License.
 15  
  */
 16  
 package org.seasar.uruma.viewer;
 17  
 
 18  
 import java.util.HashMap;
 19  
 import java.util.Map;
 20  
 
 21  
 import org.eclipse.jface.viewers.IBaseLabelProvider;
 22  
 import org.eclipse.jface.viewers.ITableLabelProvider;
 23  
 import org.eclipse.jface.viewers.TableViewer;
 24  
 import org.eclipse.jface.viewers.Viewer;
 25  
 import org.eclipse.jface.viewers.ViewerComparator;
 26  
 import org.eclipse.swt.SWT;
 27  
 import org.eclipse.swt.events.SelectionEvent;
 28  
 import org.eclipse.swt.events.SelectionListener;
 29  
 import org.eclipse.swt.widgets.Table;
 30  
 import org.eclipse.swt.widgets.TableColumn;
 31  
 
 32  
 /**
 33  
  * {@link TableViewer} 用の汎用ソートクラスです。<br />
 34  
  * 
 35  
  * @author y-komori
 36  
  */
 37  
 public class GenericTableViewerComparator extends ViewerComparator {
 38  
     private Table table;
 39  
 
 40  8
     private Map<TableColumn, Integer> columnMap = new HashMap<TableColumn, Integer>();
 41  
 
 42  
     private TableColumn sortKey;
 43  
 
 44  8
     private SortingState sortingState = SortingState.NONE;
 45  
 
 46  
     /**
 47  
      * {@link GenericTableViewerComparator} を構築します。<br />
 48  
      */
 49  
     public GenericTableViewerComparator() {
 50  8
         super(new StringAndNumberComparator());
 51  8
     }
 52  
 
 53  
     protected void setupColumnMap(final TableViewer viewer) {
 54  8
         this.table = viewer.getTable();
 55  8
         TableColumn[] columns = table.getColumns();
 56  8
         if (columns.length == 0) {
 57  0
             return;
 58  
         }
 59  
 
 60  32
         for (int i = 0; i < columns.length; i++) {
 61  24
             columnMap.put(columns[i], i);
 62  
 
 63  24
             columns[i].addSelectionListener(new SelectionListener() {
 64  
                 public void widgetDefaultSelected(final SelectionEvent e) {
 65  0
                 }
 66  
 
 67  24
                 public void widgetSelected(final SelectionEvent e) {
 68  0
                     setSortKey((TableColumn) e.widget);
 69  0
                     viewer.setInput(viewer.getInput());
 70  0
                 }
 71  
             });
 72  
         }
 73  
 
 74  8
         sortKey = columns[0];
 75  8
     }
 76  
 
 77  
     /*
 78  
      * @see org.eclipse.jface.viewers.ViewerComparator#compare(org.eclipse.jface.viewers.Viewer,
 79  
      *      java.lang.Object, java.lang.Object)
 80  
      */
 81  
     @Override
 82  
     public int compare(final Viewer viewer, final Object e1, final Object e2) {
 83  40
         TableViewer tableViewer = (TableViewer) viewer;
 84  
 
 85  40
         if (this.table == null) {
 86  8
             setupColumnMap(tableViewer);
 87  
         }
 88  
 
 89  40
         int cat1 = category(e1);
 90  40
         int cat2 = category(e2);
 91  
 
 92  40
         if (cat1 != cat2) {
 93  0
             return cat1 - cat2;
 94  
         }
 95  
 
 96  40
         Integer sortColumn = columnMap.get(sortKey);
 97  40
         if (sortColumn != null) {
 98  40
             int result = doCompare(tableViewer, e1, e2, sortColumn.intValue());
 99  
 
 100  40
             if (sortingState == SortingState.ASCENDING) {
 101  0
                 return result;
 102  40
             } else if (sortingState == SortingState.DESCENDING) {
 103  0
                 return result * (-1);
 104  
             } else {
 105  40
                 return 0;
 106  
             }
 107  
         } else {
 108  0
             return 0;
 109  
         }
 110  
     }
 111  
 
 112  
     /**
 113  
      * モデルオブジェクトの比較を行います。<br />
 114  
      * 比較方法をカスタマイズする場合、サブクラスで本メソッドをオーバーライドしてください。<br />
 115  
      * 
 116  
      * @param viewer
 117  
      *            {@link TableViewer} オブジェクト
 118  
      * @param e1
 119  
      *            比較対象1
 120  
      * @param e2
 121  
      *            比較対象2
 122  
      * @param sortColumn
 123  
      *            ソート対象のカラム番号
 124  
      * @return 比較結果
 125  
      */
 126  
     protected int doCompare(final TableViewer viewer, final Object e1,
 127  
             final Object e2, final int sortColumn) {
 128  40
         IBaseLabelProvider baseLabelProvider = viewer.getLabelProvider();
 129  
 
 130  40
         String value1 = "";
 131  40
         String value2 = "";
 132  40
         if (baseLabelProvider instanceof ITableLabelProvider) {
 133  40
             ITableLabelProvider prov = (ITableLabelProvider) baseLabelProvider;
 134  40
             value1 = prov.getColumnText(e1, sortColumn);
 135  40
             value2 = prov.getColumnText(e2, sortColumn);
 136  40
         } else {
 137  0
             value1 = e1.toString();
 138  0
             value2 = e2.toString();
 139  
         }
 140  
 
 141  40
         if (value1 == null) {
 142  0
             value1 = "";
 143  
         }
 144  40
         if (value2 == null) {
 145  0
             value2 = "";
 146  
         }
 147  
 
 148  40
         return getComparator().compare(value1, value2);
 149  
     }
 150  
 
 151  
     /**
 152  
      * ソートキーとなるカラムを設定します。<br />
 153  
      * ソート順は呼び出すたびに反転します。
 154  
      * 
 155  
      * @param tableColumn
 156  
      *            ソートキーとなる {@link TableColumn} オブジェクト
 157  
      */
 158  
     public void setSortKey(final TableColumn tableColumn) {
 159  0
         if ((tableColumn != null) && !tableColumn.isDisposed()) {
 160  0
             sortKey = tableColumn;
 161  
 
 162  0
             if (sortingState == SortingState.DESCENDING) {
 163  0
                 sortingState = SortingState.ASCENDING;
 164  0
                 table.setSortColumn(sortKey);
 165  0
                 table.setSortDirection(SWT.UP);
 166  0
             } else if (sortingState == SortingState.ASCENDING) {
 167  0
                 sortingState = SortingState.NONE;
 168  0
                 table.setSortColumn(null);
 169  0
                 table.setSortDirection(SWT.NONE);
 170  0
             } else if (sortingState == SortingState.NONE) {
 171  0
                 sortingState = SortingState.DESCENDING;
 172  0
                 table.setSortColumn(sortKey);
 173  0
                 table.setSortDirection(SWT.DOWN);
 174  
             }
 175  
         }
 176  0
     }
 177  
 }