001/*
002 *                    BioJava development code
003 *
004 * This code may be freely distributed and modified under the
005 * terms of the GNU Lesser General Public Licence.  This should
006 * be distributed with the code.  If you do not have a copy,
007 * see:
008 *
009 *      http://www.gnu.org/copyleft/lesser.html
010 *
011 * Copyright for this code is held jointly by the individual
012 * authors.  These should be listed in @author doc comments.
013 *
014 * For more information on the BioJava project and its aims,
015 * or to join the biojava-l mailing list, visit the home page
016 * at:
017 *
018 *      http://www.biojava.org/
019 *
020 * Created on Mar 30, 2010
021 * Author: Andreas Prlic
022 *
023 */
024
025package org.biojava.nbio.structure.align.gui;
026
027import javax.swing.table.TableModel;
028import javax.swing.table.TableRowSorter;
029import java.util.Comparator;
030
031public class MyTableRowSorter extends TableRowSorter<TableModel>
032{
033
034        public MyTableRowSorter(TableModel tm)
035        {
036                super(tm);
037        }
038
039        @Override
040        public Comparator<?> getComparator(int column)
041        {
042
043                return new MyComparator(column);
044
045        }
046
047}
048
049class MyComparator implements Comparator<String> {
050
051        int column;
052        public MyComparator(int column){
053                this.column = column;
054        }
055
056        @Override
057        public int compare(String s1, String s2){
058                if (( column >= 2 && column <= 4)|| (column==9)){
059                        return compare(Float.parseFloat(s1), Float.parseFloat(s2));
060                } else if  (  column > 4 && column < 10) {
061                        return compare(Integer.parseInt(s1), Integer.parseInt(s2));
062                } else
063                        return s1.compareTo(s2);
064        }
065
066        public int compare(Float f1, Float f2){
067                return f1.compareTo(f2);
068        }
069        public int compare(Integer o1, Integer o2)
070        {
071                return o1.compareTo(o2);
072        }
073}