/*
* BioJava development code
*
* This code may be freely distributed and modified under the
* terms of the GNU Lesser General Public Licence. This should
* be distributed with the code. If you do not have a copy,
* see:
*
* http://www.gnu.org/copyleft/lesser.html
*
* Copyright for this code is held jointly by the individual
* authors. These should be listed in @author doc comments.
*
* For more information on the BioJava project and its aims,
* or to join the biojava-l mailing list, visit the home page
* at:
*
* http://www.biojava.org/
*
*/
package org.biojava.nbio.survival.cox;
import org.biojava.nbio.survival.data.WorkSheet;
import java.util.ArrayList;
/**
* The CoxHelper class is provided to start with a tab delimited file in a similar process in R and return the results as a CoxInfo class.
* Given the number of options for adjusting the calculations using weighting, strata, clustering etc the helper class can be used to hide
* the complexity for typical use case.
*
* @author Scooter Willis <willishf at gmail dot com>
*/
public class CoxHelper {
/**
*
* @param datafile The tab delimited file containing survival data and variables. The first column needs to be unique index
* @param timeColumn The column representing the event/censor time
* @param statusColumn The column representing an event=1 and censor=0
* @param weightColumn For case-cohort data sets may require weighting to reflect the entire cohort
* @param strataColumn A column representing strata data
* @param clusterColumn If robost variation calculation is required the cluster column will group samples by the value in this column
* @param variables The variables to be used in the cox regression analysis. For Interactions using variable1:variable2
* @param useStrata Boolean to indicate if strata column should be used
* @param useWeights Boolean to indicate if weight column should be used
* @return
* @throws Exception
*/
public static CoxInfo process(String datafile, String timeColumn, String statusColumn, String weightColumn, String strataColumn, String clusterColumn, ArrayList<String> variables, boolean useStrata, boolean useWeights) throws Exception {
WorkSheet worksheet = WorkSheet.readCSV(datafile, '\t');
return process(worksheet, timeColumn, statusColumn, weightColumn, strataColumn, clusterColumn, variables, useStrata, useWeights);
}
/**
*
* @param worksheet
* @param timeColumn The column representing the event/censor time
* @param statusColumn The column representing an event=1 and censor=0
* @param weightColumn For case-cohort data sets may require weighting to reflect the entire cohort
* @param strataColumn A column representing strata data
* @param clusterColumn If robost variation calculation is required the cluster column will group samples by the value in this column
* @param variables The variables to be used in the cox regression analysis. For Interactions using variable1:variable2
* @param useStrata Boolean to indicate if strata column should be used
* @param useWeights Boolean to indicate if weight column should be used
* @return
*/
public static CoxInfo process(WorkSheet worksheet, String timeColumn, String statusColumn, String weightColumn, String strataColumn, String clusterColumn, ArrayList<String> variables, boolean useStrata, boolean useWeights) {
try {
ArrayList<SurvivalInfo> survivalInfoList = new ArrayList<SurvivalInfo>();
int i = 1;
for (String row : worksheet.getRows()) {
double time = worksheet.getCellDouble(row, timeColumn);
double c = worksheet.getCellDouble(row, statusColumn);
double weight = 1.0;
if (weightColumn != null && weightColumn.length() > 0) {
weight = worksheet.getCellDouble(row, weightColumn);
}
int strata = 0;
if (strataColumn != null && strataColumn.length() > 0) {
strata = worksheet.getCellDouble(row, strataColumn).intValue();
}
int censor = (int) c;
if (weight <= 0) {
// System.out.println("Weight <= 0 Sample=" + row + " weight=" + weight);
i++;
continue;
}
SurvivalInfo si = new SurvivalInfo(time, censor);
si.setOrder(i);
si.setWeight(weight);
si.setStrata(strata);
for (String column : variables) {
if (column.contains(":")) {
continue;
}
String value = worksheet.getCell(row, column);
si.addUnknownDataTypeVariable(column, value);
}
if (clusterColumn != null && clusterColumn.length() > 0) {
String v = worksheet.getCell(row, clusterColumn);
si.setClusterValue(v);
}
survivalInfoList.add(si);
i++;
}
boolean cluster = false;
boolean robust = false;
if (clusterColumn != null && clusterColumn.length() > 0) {
cluster = true;
robust = true;
}
// variables.add("TREAT:AGE");
CoxR cox = new CoxR();
CoxInfo ci = cox.process(variables, survivalInfoList, useStrata, useWeights, robust, cluster);
// System.out.println(ci);
//applying Bob Gray's correction for weighted strata wtexamples.docx
// CoxCC.process(ci, survivalInfoList);
// ci.dump();
// ci.calcSummaryValues();
return ci;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* @param args the command line arguments
*/
public static void main(String[] args) {
// TODO code application logic here
try {
if (true) {
String datafile = "/Users/Scooter/scripps/ngs/DataSets/E2197/misc/ecoglabtransfer/500790/2013.05.10.12.28.58.313/clindasl0228.txt";
ArrayList<String> variables = new ArrayList<String>();
variables.add("nndpos");
variables.add("meno");
// variables.add("er1");
// variables.add("meno:er1");
CoxInfo ci = CoxHelper.process(datafile, "ttr", "recind", "wt", "sstrat", "Seq", variables, false, true);
// ci.dump();
System.out.println(ci);
System.out.println();
CoxCC.process(ci);
ci.dump();
}
} catch (Exception e) {
e.printStackTrace();
}
}
}