SW implementation is command line argument in FilterAlignmentArtifacts

broadinstitute · Oct 26, 2021 · 2d2a47b · 2d2a47b
1 parent db971f0
commit 2d2a47b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...g/broadinstitute/hellbender/tools/walkers/realignmentfilter/FilterAlignmentArtifacts.java b/...g/broadinstitute/hellbender/tools/walkers/realignmentfilter/FilterAlignmentArtifacts.java
@@ -112,7 +112,6 @@ public class FilterAlignmentArtifacts extends MultiVariantWalkerGroupedOnStart {
     public static final int DEFAULT_MAX_GROUPED_SPAN = 10_000;
     private static final int MIN_UNITIG_LENGTH = 30;
     private static final int ASSEMBLY_PADDING = 50;
-    private static final SmithWatermanAligner ALIGNER = SmithWatermanAligner.getAligner(SmithWatermanAligner.Implementation.FASTEST_AVAILABLE);
 
     @Argument(fullName = StandardArgumentDefinitions.OUTPUT_LONG_NAME, shortName = StandardArgumentDefinitions.OUTPUT_SHORT_NAME,
             doc="The output filtered VCF file", optional=false)
@@ -136,10 +135,15 @@ public class FilterAlignmentArtifacts extends MultiVariantWalkerGroupedOnStart {
     @Argument(fullName= AssemblyBasedCallerArgumentCollection.BAM_OUTPUT_LONG_NAME, shortName= AssemblyBasedCallerArgumentCollection.BAM_OUTPUT_SHORT_NAME, doc="File to which assembled haplotypes should be written", optional = true)
     public String bamOutputPath = null;
 
+    @Advanced
+    @Argument(fullName = AssemblyBasedCallerArgumentCollection.SMITH_WATERMAN_LONG_NAME, doc = "Which Smith-Waterman implementation to use, generally FASTEST_AVAILABLE is the right choice", optional = true)
+    public SmithWatermanAligner.Implementation smithWatermanImplementation = SmithWatermanAligner.Implementation.JAVA;
+
 
     @ArgumentCollection
     protected RealignmentArgumentCollection realignmentArgumentCollection = new RealignmentArgumentCollection();
 
+    private SmithWatermanAligner smithWatermanAligner;
     private VariantContextWriter vcfWriter;
     private RealignmentEngine realignmentEngine;
     private SAMFileHeader bamHeader;
@@ -174,6 +178,7 @@ protected int defaultMaxGroupedSpan() {
 
     @Override
     public void onTraversalStart() {
+        smithWatermanAligner = SmithWatermanAligner.getAligner(smithWatermanImplementation);
         realignmentEngine = new RealignmentEngine(realignmentArgumentCollection);
         vcfWriter = createVCFWriter(outputVcf);
 
@@ -211,15 +216,15 @@ public void apply(List<VariantContext> variantContexts, ReferenceContext referen
 
 
             // TODO: give this tool M2 Assembler args to allow override default M2ArgumentCollection?
-            final AssemblyResultSet assemblyResult = AssemblyBasedCallerUtils.assembleReads(assemblyRegion, Collections.emptyList(), MTAC, bamHeader, samplesList, logger, referenceReader, assemblyEngine, ALIGNER, false);
+            final AssemblyResultSet assemblyResult = AssemblyBasedCallerUtils.assembleReads(assemblyRegion, Collections.emptyList(), MTAC, bamHeader, samplesList, logger, referenceReader, assemblyEngine, smithWatermanAligner, false);
             final AssemblyRegion regionForGenotyping = assemblyResult.getRegionForGenotyping();
 
             final Map<String,List<GATKRead>> reads = AssemblyBasedCallerUtils.splitReadsBySample(samplesList, bamHeader, regionForGenotyping.getReads());
 
             final AlleleLikelihoods<GATKRead, Haplotype> readLikelihoods = likelihoodCalculationEngine.computeReadLikelihoods(assemblyResult,samplesList,reads);
             readLikelihoods.switchToNaturalLog();
             final SWParameters readToHaplotypeSWParameters = MTAC.getReadToHaplotypeSWParameters();
-            final Map<GATKRead,GATKRead> readRealignments = AssemblyBasedCallerUtils.realignReadsToTheirBestHaplotype(readLikelihoods, assemblyResult.getReferenceHaplotype(), assemblyResult.getPaddedReferenceLoc(), ALIGNER, readToHaplotypeSWParameters);
+            final Map<GATKRead,GATKRead> readRealignments = AssemblyBasedCallerUtils.realignReadsToTheirBestHaplotype(readLikelihoods, assemblyResult.getReferenceHaplotype(), assemblyResult.getPaddedReferenceLoc(), smithWatermanAligner, readToHaplotypeSWParameters);
             readLikelihoods.changeEvidence(readRealignments);
             writeBamOutput(assemblyResult, readLikelihoods, new HashSet<>(readLikelihoods.alleles()), regionForGenotyping.getSpan());