diff --git a/java/src/org/broadinstitute/sting/oneoffprojects/walkers/CalibrateGenotypeLikelihoods.java b/java/src/org/broadinstitute/sting/oneoffprojects/walkers/CalibrateGenotypeLikelihoods.java index 4f0a6287b..eacc7498e 100644 --- a/java/src/org/broadinstitute/sting/oneoffprojects/walkers/CalibrateGenotypeLikelihoods.java +++ b/java/src/org/broadinstitute/sting/oneoffprojects/walkers/CalibrateGenotypeLikelihoods.java @@ -44,6 +44,7 @@ import org.broadinstitute.sting.gatk.refdata.RefMetaDataTracker; import org.broadinstitute.sting.gatk.walkers.*; import org.broadinstitute.sting.gatk.walkers.genotyper.*; import org.broadinstitute.sting.utils.*; +import org.broadinstitute.sting.utils.collections.NestedHashMap; import org.broadinstitute.sting.utils.exceptions.ReviewedStingException; import org.broadinstitute.sting.utils.exceptions.UserException; import org.broadinstitute.sting.utils.vcf.VCFUtils; @@ -94,10 +95,10 @@ public class CalibrateGenotypeLikelihoods extends RodWalker { - // todo -- add event type (SNP, indel) String rgID, sample; GenotypeLikelihoods pl; - Genotype.Type compType; + VariantContext.Type siteType; + Genotype.Type genotypeType; @Override public int compareTo(Datum o) { @@ -106,11 +107,12 @@ public class CalibrateGenotypeLikelihoods extends RodWalker pGNames = Arrays.asList("QofAAGivenD", "QofABGivenD", "QofBBGivenD"); - List fields = Arrays.asList("sample", "rg", "pls", "comp", "pGGivenDType", "pGGivenD"); + List fields = Arrays.asList("sample", "rg", "siteType", "pls", "comp", "pGGivenDType", "pGGivenD"); out.println(Utils.join("\t", fields)); double[] counts = new double[]{1, 1, 1}; - for ( Datum d : data.values ) { counts[d.compType.ordinal()-1]++; } + for ( Datum d : data.values ) { counts[d.genotypeType.ordinal()-1]++; } double sum = MathUtils.sum(counts); logger.info(String.format("Types %s %s %s", Genotype.Type.values()[1], Genotype.Type.values()[2], Genotype.Type.values()[3])); logger.info(String.format("Counts %.0f %.0f %.0f %.0f", counts[0], counts[1], counts[2], sum)); @@ -257,8 +260,8 @@ public class CalibrateGenotypeLikelihoods extends RodWalker 1 ) { // tons of 1s, and not interesting - out.printf("%s\t%s\t%s\t%s\t%s\t%d%n", - d.sample, d.rgID, d.pl.getAsString(), d.compType.toString(), + out.printf("%s\t%s\t%s\t%s\t%s\t%s\t%d%n", + d.sample, d.rgID, d.siteType, d.pl.getAsString(), d.genotypeType.toString(), pGNames.get(i), q); } }