zzh
/
gatk-3.8
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of ssh://nickel.broadinstitute.org/humgen/gsa-scr1/gsa-engineering/git/unstable

This commit is contained in:
Ryan Poplin 2012-03-27 23:09:18 -04:00
parent 60c9ada5d8 12aa72f200
commit 7f211c66fa
1 changed files with 23 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
public/java/src/org/broadinstitute/sting/utils/variantcontext/GenotypeLikelihoods.java
View File

@ -27,6 +27,7 @@ package org.broadinstitute.sting.utils.variantcontext;
import org.broad.tribble.TribbleException; import org.broad.tribble.TribbleException;
import org.broadinstitute.sting.utils.MathUtils; import org.broadinstitute.sting.utils.MathUtils;
import org.broadinstitute.sting.utils.codecs.vcf.VCFConstants; import org.broadinstitute.sting.utils.codecs.vcf.VCFConstants;
import org.broadinstitute.sting.utils.exceptions.ReviewedStingException;
import java.util.EnumMap; import java.util.EnumMap;
@ -219,35 +220,38 @@ public class GenotypeLikelihoods {
} }
} }
/**
* The maximum number of alleles that we can represent as genotype likelihoods
*/
final static int MAX_ALLELES_THAT_CAN_BE_GENOTYPED = 500;
/* /*
* a cache of the PL index to the 2 alleles it represents over all possible numbers of alternate alleles * a cache of the PL index to the 2 alleles it represents over all possible numbers of alternate alleles
*/ */
private static GenotypeLikelihoodsAllelePair[] PLIndexToAlleleIndex; private final static GenotypeLikelihoodsAllelePair[] PLIndexToAlleleIndex = calculatePLcache(MAX_ALLELES_THAT_CAN_BE_GENOTYPED); // start with data for 10 alternate alleles
static {
calculatePLcache(65); // start with data for 10 alternate alleles
}
private static void calculatePLcache(final int minIndex) { private static GenotypeLikelihoodsAllelePair[] calculatePLcache(final int altAlleles) {
// how many alternate alleles do we need to calculate for? final int numLikelihoods = calculateNumLikelihoods(altAlleles);
int altAlleles = 0; final GenotypeLikelihoodsAllelePair[] cache = new GenotypeLikelihoodsAllelePair[numLikelihoods];
int numLikelihoods = 1;
while ( numLikelihoods <= minIndex ) {
altAlleles++;
numLikelihoods += altAlleles + 1;
}
PLIndexToAlleleIndex = new GenotypeLikelihoodsAllelePair[numLikelihoods];
// for all possible combinations of 2 alleles // for all possible combinations of 2 alleles
for ( int allele1 = 0; allele1 <= altAlleles; allele1++ ) { for ( int allele1 = 0; allele1 <= altAlleles; allele1++ ) {
for ( int allele2 = allele1; allele2 <= altAlleles; allele2++ ) { for ( int allele2 = allele1; allele2 <= altAlleles; allele2++ ) {
PLIndexToAlleleIndex[calculatePLindex(allele1, allele2)] = new GenotypeLikelihoodsAllelePair(allele1, allele2); cache[calculatePLindex(allele1, allele2)] = new GenotypeLikelihoodsAllelePair(allele1, allele2);
} }
} }
}
// a bit of sanity checking
for ( int i = 0; i < cache.length; i++ ) {
if ( cache[i] == null )
throw new ReviewedStingException("BUG: cache entry " + i + " is unexpected null");
}
return cache;
}
// how many likelihoods are associated with the given number of alternate alleles? // how many likelihoods are associated with the given number of alternate alleles?
public static int calculateNumLikelihoods(int numAltAlleles) { public static int calculateNumLikelihoods(final int numAltAlleles) {
int numLikelihoods = 1; int numLikelihoods = 1;
for ( int i = 1; i <= numAltAlleles; i++ ) for ( int i = 1; i <= numAltAlleles; i++ )
numLikelihoods += i + 1; numLikelihoods += i + 1;
@ -269,9 +273,8 @@ public class GenotypeLikelihoods {
*/ */
public static GenotypeLikelihoodsAllelePair getAllelePair(final int PLindex) { public static GenotypeLikelihoodsAllelePair getAllelePair(final int PLindex) {
// make sure that we've cached enough data // make sure that we've cached enough data
// TODO -- this is not thread-safe!
if ( PLindex >= PLIndexToAlleleIndex.length ) if ( PLindex >= PLIndexToAlleleIndex.length )
calculatePLcache(PLindex); throw new ReviewedStingException("GATK limitation: cannot genotype more than " + MAX_ALLELES_THAT_CAN_BE_GENOTYPED + " alleles");
return PLIndexToAlleleIndex[PLindex]; return PLIndexToAlleleIndex[PLindex];
} }
@ -292,13 +295,7 @@ public class GenotypeLikelihoods {
* @return the allele index pair * @return the allele index pair
*/ */
public static GenotypeLikelihoodsAllelePair getAllelePairUsingDeprecatedOrdering(final int PLindex) { public static GenotypeLikelihoodsAllelePair getAllelePairUsingDeprecatedOrdering(final int PLindex) {
final int convertedIndex = PLindexConversion[PLindex]; return getAllelePair(PLindexConversion[PLindex]);
// make sure that we've cached enough data
if ( convertedIndex >= PLIndexToAlleleIndex.length )
calculatePLcache(convertedIndex);
return PLIndexToAlleleIndex[convertedIndex];
} }
/** /**