Merge pull request #614 from broadinstitute/rp_fix_GenotypeGVCF_VCF_headers

GenotypeGVCF was pulling the headers from all input rods including DBsnp...
This commit is contained in:
Eric Banks 2014-04-25 15:36:35 -04:00
commit 385fe5fb56
2 changed files with 7 additions and 7 deletions

View File

@ -154,8 +154,12 @@ public class GenotypeGVCFs extends RodWalker<VariantContext, VariantContextWrite
// create the annotation engine
annotationEngine = new VariantAnnotatorEngine(Arrays.asList("none"), annotationsToUse, Collections.<String>emptyList(), this, getToolkit());
// collect the actual rod bindings into a list for use later
for ( final RodBindingCollection<VariantContext> variantCollection : variantCollections )
variants.addAll(variantCollection.getRodBindings());
// take care of the VCF headers
final Map<String, VCFHeader> vcfRods = GATKVCFUtils.getVCFHeadersFromRods(getToolkit());
final Map<String, VCFHeader> vcfRods = GATKVCFUtils.getVCFHeadersFromRods(getToolkit(), variants);
final Set<VCFHeaderLine> headerLines = VCFUtils.smartMergeHeaders(vcfRods.values(), true);
headerLines.addAll(annotationEngine.getVCFAnnotationDescriptions());
VCFStandardHeaderLines.addStandardInfoLines(headerLines, true, VCFConstants.MLE_ALLELE_COUNT_KEY, VCFConstants.MLE_ALLELE_FREQUENCY_KEY);
@ -167,11 +171,7 @@ public class GenotypeGVCFs extends RodWalker<VariantContext, VariantContextWrite
vcfWriter.writeHeader(vcfHeader);
// create the genotyping engine
genotypingEngine = new UnifiedGenotypingEngine(getToolkit(), new UnifiedArgumentCollection(),samples);
// collect the actual rod bindings into a list for use later
for ( final RodBindingCollection<VariantContext> variantCollection : variantCollections )
variants.addAll(variantCollection.getRodBindings());
genotypingEngine = new UnifiedGenotypingEngine(getToolkit(), new UnifiedArgumentCollection(), samples);
}
public VariantContext map(final RefMetaDataTracker tracker, final ReferenceContext ref, final AlignmentContext context) {

View File

@ -102,7 +102,7 @@ public class GenotypeGVCFsIntegrationTest extends WalkerTest {
" -V:sample3 " + privateTestDir + "combine.single.sample.pipeline.3.vcf" +
" -L 20:10,000,000-11,000,000 --dbsnp " + b37dbSNP132, b37KGReference),
1,
Arrays.asList("91061171c480901df8f776090527c182"));
Arrays.asList("dc13542c71fbb4846507b0cdcc4827be"));
executeTest("combineSingleSamplePipelineGVCF_addDbsnp", spec);
}