Using Utils.stream and Utils.validate

Author: davidbenjamin

src/main/java/org/broadinstitute/hellbender/tools/HaplotypeCallerSpark.java

@@ -46,7 +46,6 @@
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
-import java.util.stream.StreamSupport;
 
 /**
  * Call germline SNPs and indels via local re-assembly of haplotypes
@@ -179,15 +178,10 @@ private static FlatMapFunction<Iterator<Tuple2<AssemblyRegion, SimpleInterval>>,
             //HaplotypeCallerEngine isn't serializable but is expensive to instantiate, so construct and reuse one for every partition
             final ReferenceMultiSourceAdapter referenceReader = new ReferenceMultiSourceAdapter(referenceBroadcast.getValue(), authHolder);
             final HaplotypeCallerEngine hcEngine = new HaplotypeCallerEngine(hcArgsBroadcast.value(), header, referenceReader);
-            return iteratorToStream(regionAndIntervals).flatMap(regionToVariants(hcEngine)).iterator();
+            return Utils.stream(regionAndIntervals).flatMap(regionToVariants(hcEngine)).iterator();
         };
     }
 
-    private static <T> Stream<T> iteratorToStream(Iterator<T> iterator) {
-        Iterable<T> regionsIterable = () -> iterator;
-        return StreamSupport.stream(regionsIterable.spliterator(), false);
-    }
-
     private static Function<Tuple2<AssemblyRegion, SimpleInterval>, Stream<? extends VariantContext>> regionToVariants(HaplotypeCallerEngine hcEngine) {
         return regionAndInterval -> {
             final List<VariantContext> variantContexts = hcEngine.callRegion(regionAndInterval._1(), new FeatureContext());
@@ -259,7 +253,7 @@ private static FlatMapFunction<Iterator<Shard<GATKRead>>, Tuple2<AssemblyRegion,
             final ReferenceMultiSourceAdapter referenceSource = new ReferenceMultiSourceAdapter(referenceMultiSource, authHolder);
             final HaplotypeCallerEngine hcEngine = new HaplotypeCallerEngine(hcArgsBroadcast.value(), header, referenceSource);
 
-            return iteratorToStream(shards).flatMap(shardToRegion(assemblyArgs, header, referenceSource, hcEngine)).iterator();
+            return Utils.stream(shards).flatMap(shardToRegion(assemblyArgs, header, referenceSource, hcEngine)).iterator();
         };
     }
 
@@ -280,8 +274,7 @@ private static Function<Shard<GATKRead>, Stream<? extends Tuple2<AssemblyRegion,
                     assemblyArgs.assemblyRegionPadding, assemblyArgs.activeProbThreshold,
                     assemblyArgs.maxProbPropagationDistance);
 
-            return StreamSupport.stream(assemblyRegions.spliterator(), false)
-                    .map(a -> new Tuple2<>(a, shard.getInterval()));
+            return Utils.stream(assemblyRegions).map(a -> new Tuple2<>(a, shard.getInterval()));
         };
     }
 

src/main/java/org/broadinstitute/hellbender/tools/exome/CombineReadCounts.java

@@ -10,6 +10,7 @@
 import org.broadinstitute.hellbender.exceptions.GATKException;
 import org.broadinstitute.hellbender.exceptions.UserException;
 import org.broadinstitute.hellbender.utils.SimpleInterval;
+import org.broadinstitute.hellbender.utils.Utils;
 import org.broadinstitute.hellbender.utils.tsv.DataLine;
 import org.broadinstitute.hellbender.utils.tsv.TableColumnCollection;
 import org.broadinstitute.hellbender.utils.tsv.TableReader;
@@ -508,9 +509,7 @@ private void composeCountColumnNamesAndSourceIndexMapping() {
             for (final TableReader<ReadCountRecord> reader : readers) {
                 unsortedCountColumnNames.addAll(readCountColumnNames(reader.columns()));
             }
-            if (unsortedCountColumnNames.isEmpty()) {
-                throw new IllegalStateException("there must be at least one count column");
-            }
+            Utils.validate(!unsortedCountColumnNames.isEmpty(), "there must be at least one count column");
             countColumnSourceIndexMap = IntStream.range(0, unsortedCountColumnNames.size()).boxed()
                     .sorted(Comparator.comparing(unsortedCountColumnNames::get))
                     .mapToInt(Integer::intValue).toArray();

src/main/java/org/broadinstitute/hellbender/tools/exome/ReadCountCollectionUtils.java

@@ -156,9 +156,7 @@ public static TableWriter<ReadCountRecord> writerWithIntervals(final Writer writ
             @Override
             protected void composeLine(final ReadCountRecord record, final DataLine dataLine) {
                 final SimpleInterval interval = record.getTarget().getInterval();
-                if (interval == null) {
-                    throw new IllegalStateException("invalid combination of targets with and without intervals defined");
-                }
+                Utils.validate(interval != null, "invalid combination of targets with and without intervals defined");
                 dataLine.append(interval.getContig())
                         .append(interval.getStart())
                         .append(interval.getEnd())

src/main/java/org/broadinstitute/hellbender/tools/exome/Target.java

@@ -98,25 +98,19 @@ public String getName() {
      // Three methods to implement Locatable
     @Override
     public String getContig() {
-        if (interval == null) {
-            throw new IllegalStateException("the target does not have an interval assigned");
-        }
+        Utils.validate(interval != null, "the target does not have an interval assigned");
         return interval.getContig();
     }
 
     @Override
     public int getStart() {
-        if (interval == null) {
-            throw new IllegalStateException("the target does not have an interval assigned");
-        }
+        Utils.validate(interval != null, "the target does not have an interval assigned");
         return interval.getStart();
     }
 
     @Override
     public int getEnd() {
-        if (interval == null) {
-            throw new IllegalStateException("the target does not have an interval assigned");
-        }
+        Utils.validate(interval != null, "the target does not have an interval assigned");
         return interval.getEnd();
     }
 

src/main/java/org/broadinstitute/hellbender/tools/exome/eval/EvaluateCopyNumberTriStateCalls.java

@@ -35,7 +35,6 @@
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
-import java.util.stream.StreamSupport;
 
 /**
  * Tool to evaluate the output of {@link XHMMSegmentCaller}.
@@ -686,8 +685,7 @@ private CopyNumberTriStateAllele copyNumberToTrueAllele(final int cn) {
      * @return never {@code null}, potentially immutably
      */
     private List<VariantContext> variantQueryToList(final VCFFileReader reader, final Locatable interval) {
-        return StreamSupport.stream(Spliterators.spliteratorUnknownSize(reader.query(interval.getContig(),
-                interval.getStart(), interval.getEnd()), Spliterator.NONNULL),false)
+        return Utils.stream(reader.query(interval.getContig(), interval.getStart(), interval.getEnd()))
                 .sorted(IntervalUtils.LEXICOGRAPHICAL_ORDER_COMPARATOR)
                 .collect(Collectors.toList());
     }

src/main/java/org/broadinstitute/hellbender/tools/exome/eval/VariantEvaluationContext.java

@@ -69,9 +69,8 @@ private double[] getAlleleDoubleArrayFromAlternativeAlleleArrayAttribute(final S
         final double[] alternativeAllelesFrequencies = GATKProtectedVariantContextUtils.getAttributeAsDoubleArray(this, key,
                 () -> { return new double[alleles.size() - 1]; }, missingValue);
         final double[] result = new double[alleles.size()];
-        if (alternativeAllelesFrequencies.length != alleles.size() - 1) {
-            throw new IllegalStateException(String.format("We expect the %s Info annotation to contain an array of %d elements (alt. allele count)", key, alleles.size() - 1));
-        }
+        Utils.validate(alternativeAllelesFrequencies.length == alleles.size() - 1,
+                () -> String.format("We expect the %s Info annotation to contain an array of %d elements (alt. allele count)", key, alleles.size() - 1));
         System.arraycopy(alternativeAllelesFrequencies, 0, result, 1, alternativeAllelesFrequencies.length);
         final double nonRefSum = MathUtils.sum(result);
         Utils.validateArg(nonRefSum <= 1.0, () -> String.format("The sum of element on annotation %s cannot greater than 1.0: %g", key, nonRefSum));

src/main/java/org/broadinstitute/hellbender/tools/exome/eval/VariantEvaluationContextBuilder.java

@@ -81,9 +81,7 @@ public VariantEvaluationContextBuilder alleles(final String... alleleStrings) {
     @Override
     public List<Allele> getAlleles() {
         final List<Allele> result = super.getAlleles();
-        if (result == null) {
-            throw new IllegalStateException("you must set the alleles before calling getAlleles");
-        }
+        Utils.validate(result != null, "you must set the alleles before calling getAlleles");
         return result;
     }
 

src/main/java/org/broadinstitute/hellbender/tools/exome/sexgenotyper/SexGenotypeData.java

@@ -1,6 +1,7 @@
 package org.broadinstitute.hellbender.tools.exome.sexgenotyper;
 
 import org.broadinstitute.hellbender.exceptions.UserException;
+import org.broadinstitute.hellbender.utils.Utils;
 import org.broadinstitute.hellbender.utils.param.ParamUtils;
 
 import javax.annotation.Nonnull;
@@ -64,11 +65,8 @@ public SexGenotypeData(@Nonnull final String sampleName,
      * @return set of genotype string identifiers
      */
     public Set<String> getSexGenotypesSet() {
-        if (logLikelihoodMap != null) {
-            return logLikelihoodMap.keySet();
-        } else {
-            throw new IllegalStateException("Genotyping inference data is not available");
-        }
+        Utils.validate(logLikelihoodMap != null, "Genotyping inference data is not available");
+        return logLikelihoodMap.keySet();
     }
 
     /**
@@ -77,11 +75,8 @@ public Set<String> getSexGenotypesSet() {
      * @return log likelihood
      */
     public double getLogLikelihoodPerGenotype(@Nonnull final String genotype) {
-        if (logLikelihoodMap != null) {
-            return logLikelihoodMap.get(genotype);
-        } else {
-            throw new IllegalStateException("Genotyping inference data is not available");
-        }
+        Utils.validate(logLikelihoodMap != null, "Genotyping inference data is not available");
+        return logLikelihoodMap.get(genotype);
     }
 
     /**

src/main/java/org/broadinstitute/hellbender/tools/exome/sexgenotyper/SexGenotypeDataCollection.java

@@ -83,9 +83,7 @@ public void add(@Nonnull final SexGenotypeData sexGenotypeData) {
      * @param dataWriter an instance of {@link Writer}
      */
     public void write(@Nonnull final Writer dataWriter) {
-        if (sexGenotypeDataList.isEmpty()) {
-            throw new IllegalStateException("The sex genotype data collection is empty");
-        }
+        Utils.validate(!sexGenotypeDataList.isEmpty(), "The sex genotype data collection is empty");
         /* check if extended genotyping information is available; first, check for nulls */
         boolean extended = true;
         if (sexGenotypeDataList.stream().filter(dat -> !dat.hasExtendedGenotypingInfo()).count() > 0) {

src/main/java/org/broadinstitute/hellbender/tools/walkers/ReferenceConfidenceVariantContextMerger.java

@@ -171,9 +171,7 @@ static List<Allele> remapAlleles(final VariantContext vc, final Allele refAllele
         final Allele vcRef = vc.getReference();
         final byte[] refBases = refAllele.getBases();
         final int extraBaseCount = refBases.length - vcRef.getBases().length;
-        if (extraBaseCount < 0) {
-            throw new IllegalStateException("the wrong reference was selected");
-        }
+        Utils.validate(extraBaseCount >= 0, "the wrong reference was selected");
 
         final List<Allele> result = new ArrayList<>(vc.getNAlleles());
         result.add(refAllele);