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 */
021package org.biojava.nbio.survival.cox.comparators;
022
023import org.biojava.nbio.survival.cox.SurvivalInfo;
024
025import java.io.Serializable;
026import java.util.Comparator;
027
028/**
029 *
030 * @author Scooter Willis <willishf at gmail dot com>
031 */
032public class SurvivalInfoValueComparator implements Comparator<SurvivalInfo>, Serializable {
033        private static final long serialVersionUID = 1;
034
035        String variable = "";
036
037        /**
038         *
039         * @param variable
040         */
041        public SurvivalInfoValueComparator(String variable){
042                this.variable = variable;
043        }
044
045        @Override
046        public int compare(SurvivalInfo t, SurvivalInfo t1) {
047                double v = t.getContinuousVariable(variable);
048                double v1 = t1.getContinuousVariable(variable);
049                if(v < v1)
050                        return -1;
051                else if(v > v1)
052                        return 1;
053                else
054                        return 0;
055        }
056
057}