66 lines
1.6 KiB
Java
66 lines
1.6 KiB
Java
package org.broadinstitute.sting.oneoffprojects.multisamplecaller;
|
|
|
|
import java.lang.*;
|
|
import java.util.*;
|
|
import java.io.*;
|
|
|
|
import org.broadinstitute.sting.utils.*;
|
|
|
|
public class ConfusionMatrix
|
|
{
|
|
private double[][] ILLUMINA;
|
|
private double[][] solid;
|
|
private double[][] LS454;
|
|
|
|
public ConfusionMatrix(String file_name)
|
|
{
|
|
//System.out.println("DBG: ConfusionMatrix constructor! (" + file_name + ")");
|
|
|
|
ILLUMINA = new double[4][4];
|
|
solid = new double[4][4];
|
|
LS454 = new double[4][4];
|
|
|
|
try
|
|
{
|
|
Scanner sc = new Scanner(new File(file_name));
|
|
while (sc.hasNext())
|
|
{
|
|
String platform = sc.next();
|
|
char read = sc.next().charAt(0);
|
|
char ref = sc.next().charAt(0);
|
|
double p = sc.nextDouble();
|
|
|
|
int read_i = BaseUtils.simpleBaseToBaseIndex(read);
|
|
int ref_i = BaseUtils.simpleBaseToBaseIndex(ref);
|
|
|
|
if (platform.equals("ILLUMINA")) { ILLUMINA[read_i][ref_i] = p; }
|
|
if (platform.equals("solid")) { solid[read_i][ref_i] = p; }
|
|
if (platform.equals("LS454")) { LS454[read_i][ref_i] = p; }
|
|
|
|
//System.out.println("DBG: " + key + " " + p);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
e.printStackTrace();
|
|
System.exit(-1);
|
|
}
|
|
|
|
}
|
|
|
|
double lookup(String platform, char read, char truth)
|
|
{
|
|
int read_i = BaseUtils.simpleBaseToBaseIndex(read);
|
|
int truth_i = BaseUtils.simpleBaseToBaseIndex(truth);
|
|
|
|
double d = 0;
|
|
|
|
if (platform.equals("ILLUMINA")) { d = ILLUMINA[read_i][truth_i]; }
|
|
else if (platform.equals("solid")) { d = solid[read_i][truth_i]; }
|
|
else if (platform.equals("LS454")) { d = LS454[read_i][truth_i]; }
|
|
else { assert(false); }
|
|
|
|
return d;
|
|
}
|
|
}
|