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.CoxInfo;
024import org.biojava.nbio.survival.cox.CoxVariables;
025
026import java.io.Serializable;
027import java.util.Comparator;
028
029/**
030 *
031 * @author Scooter Willis <willishf at gmail dot com>
032 */
033public class MeanModelComparator implements Comparator<CoxVariables>, Serializable {
034    private static final long serialVersionUID = 1;
035
036        String variable = "";
037
038        /**
039         *
040         * @param variable
041         */
042        public MeanModelComparator(String variable) {
043          this.variable = variable;
044        }
045
046        @Override
047        public int compare(CoxVariables coxVariables1, CoxVariables coxVariables2) {
048                CoxInfo ci1LTMean = coxVariables1.getCoxInfo("<MEAN");
049                CoxInfo ci1GTMean = coxVariables1.getCoxInfo(">MEAN");
050
051                if(ci1LTMean == null || ci1GTMean == null)
052                        return 0;
053
054                double c1LTpvalue = ci1LTMean.getCoefficient(variable).getPvalue();
055                double c1GTpvalue = ci1GTMean.getCoefficient(variable).getPvalue();
056
057                double c1ratio = Math.min(c1LTpvalue, c1GTpvalue) / Math.max(c1LTpvalue, c1GTpvalue);
058
059                CoxInfo ci2LTMean = coxVariables2.getCoxInfo("<MEAN");
060                CoxInfo ci2GTMean = coxVariables2.getCoxInfo(">MEAN");
061
062                double c2LTpvalue = ci2LTMean.getCoefficient(variable).getPvalue();
063                double c2GTpvalue = ci2GTMean.getCoefficient(variable).getPvalue();
064
065           double c2ratio = Math.min(c2LTpvalue, c2GTpvalue) / Math.max(c2LTpvalue, c2GTpvalue);
066
067                if (c1ratio > c2ratio) {
068                        return 1;
069                } else if (c1ratio < c2ratio) {
070                        return -1;
071                } else {
072                        return 0;
073                }
074                //ascending order
075                // return coxVariables1.compareTo(coxVariables2);
076        }
077}