org.nmdp.ngs.tools.ValidateInterpretation Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of ngs-tools Show documentation
Show all versions of ngs-tools Show documentation
Next generation sequencing (NGS/HTS) command line tools.
/*
ngs-tools Next generation sequencing (NGS/HTS) command line tools.
Copyright (c) 2014 National Marrow Donor Program (NMDP)
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published
by the Free Software Foundation; either version 3 of the License, or (at
your option) any later version.
This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; with out even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
License for more details.
You should have received a copy of the GNU Lesser General Public License
along with this library; if not, write to the Free Software Foundation,
Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
> http://www.gnu.org/licenses/lgpl.html
*/
package org.nmdp.ngs.tools;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static org.dishevelled.compress.Readers.reader;
import static org.dishevelled.compress.Writers.writer;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
import com.google.common.base.Splitter;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.ListMultimap;
import org.dishevelled.commandline.ArgumentList;
import org.dishevelled.commandline.CommandLine;
import org.dishevelled.commandline.CommandLineParseException;
import org.dishevelled.commandline.CommandLineParser;
import org.dishevelled.commandline.Switch;
import org.dishevelled.commandline.Usage;
import org.dishevelled.commandline.argument.FileArgument;
import org.dishevelled.commandline.argument.IntegerArgument;
/**
* Validate interpretation.
*/
public final class ValidateInterpretation implements Callable {
private final File expectedFile;
private final File observedFile;
private final File outputFile;
private final int resolution;
private final boolean printSummary;
static final int DEFAULT_RESOLUTION = 2;
private static final String USAGE = "ngs-validate-interpretation -e expected.txt -b observed.txt [args]";
/**
* Validate interpretation.
*
* @param expectedFile expected file, must not be null
* @param observedFile observed file, must not be null
* @param outputFile output file, if any
* @param resolution minimum fields of resolution, must be in the range [1..4]
* @param printSummary print summary report
*/
public ValidateInterpretation(final File expectedFile, final File observedFile, final File outputFile, final int resolution, final boolean printSummary) {
checkNotNull(expectedFile);
checkNotNull(observedFile);
checkArgument(resolution > 0 && resolution < 5, "resolution must be in the range [1..4]");
this.expectedFile = expectedFile;
this.observedFile = observedFile;
this.outputFile = outputFile;
this.resolution = resolution;
this.printSummary = printSummary;
}
@Override
public Integer call() throws Exception {
PrintWriter writer = null;
int passes = 0;
int failures = 0;
try {
writer = writer(outputFile);
ListMultimap expected = readExpected(expectedFile);
ListMultimap observed = readObserved(observedFile);
for (String sample : expected.keySet()) {
List alleles = expected.get(sample);
List interpretations = observed.get(sample);
for (String expectedAllele : alleles) {
boolean result = false;
for (String interpretation : interpretations) {
List interpretedAlleles = Splitter.onPattern("[/|]+").splitToList(interpretation);
List found = new ArrayList();
for (String interpretedAllele : interpretedAlleles) {
if (matchByField(expectedAllele, interpretedAllele) >= resolution) {
found.add(interpretedAllele);
}
}
if (!found.isEmpty()) {
result = true;
}
}
if (result) {
passes++;
}
else {
failures++;
}
if (!printSummary) {
writer.println((result ? "PASS" : "FAIL") + "\t" + sample + "\t" + expectedAllele);
}
}
}
if (printSummary) {
writer.println("PASS\t" + passes);
writer.println("FAIL\t" + failures);
}
return 0;
}
finally {
try {
writer.close();
}
catch (Exception e) {
// ignore
}
}
}
static ListMultimap readExpected(final File expectedFile) throws IOException {
BufferedReader reader = null;
ListMultimap expected = ArrayListMultimap.create();
try {
reader = reader(expectedFile);
int lineNumber = 1;
while (reader.ready()) {
String line = reader.readLine();
if (line == null) {
break;
}
List tokens = Splitter.onPattern("\\s+").splitToList(line);
if (tokens.size() != 6) {
throw new IOException("invalid expected file format at line " + lineNumber);
}
String sample = tokens.get(0);
String locus = tokens.get(1);
String regionsFile = tokens.get(2);
String zygosity = tokens.get(3);
String firstAllele = tokens.get(4);
String secondAllele = tokens.get(5);
expected.put(sample, firstAllele);
expected.put(sample, secondAllele);
lineNumber++;
}
}
finally {
try {
reader.close();
}
catch (Exception e) {
// ignore
}
}
return expected;
}
static ListMultimap readObserved(final File observedFile) throws IOException {
BufferedReader reader = null;
ListMultimap observed = ArrayListMultimap.create();
try {
reader = reader(observedFile);
int lineNumber = 1;
while (reader.ready()) {
String line = reader.readLine();
if (line == null) {
break;
}
List tokens = Splitter.onPattern("\\s+").splitToList(line);
if (tokens.size() != 2) {
throw new IOException("invalid observed file format at line " + lineNumber);
}
String sample = tokens.get(0);
String interpretation = tokens.get(1);
observed.put(sample, interpretation);
lineNumber++;
}
}
finally {
try {
reader.close();
}
catch (Exception e) {
// ignore
}
}
return observed;
}
static int matchByField(final String firstAllele, final String secondAllele) {
checkNotNull(firstAllele);
checkNotNull(secondAllele);
List firstAlleleParts = Splitter.on(":").splitToList(firstAllele);
List secondAlleleParts = Splitter.on(":").splitToList(secondAllele);
int smallest = firstAlleleParts.size() < secondAlleleParts.size() ? firstAlleleParts.size() : secondAlleleParts.size();
for (int i = 0; i < smallest; i++) {
if (!firstAlleleParts.get(i).equals(secondAlleleParts.get(i))) {
return i;
}
}
return smallest;
}
/**
* Main.
*
* @param args command line args
*/
public static void main(final String[] args) {
Switch about = new Switch("a", "about", "display about message");
Switch help = new Switch("h", "help", "display help message");
FileArgument expectedFile = new FileArgument("e", "expected-file", "expected interpretation file", true);
FileArgument observedFile = new FileArgument("b", "observed-file", "observed interpretation file", true);
FileArgument outputFile = new FileArgument("o", "output-file", "output file, default stdout", false);
IntegerArgument resolution = new IntegerArgument("r", "resolution", "resolution, must be in the range [1..4], default " + DEFAULT_RESOLUTION, false);
Switch printSummary = new Switch("s", "summary", "print summary");
ArgumentList arguments = new ArgumentList(about, help, expectedFile, observedFile, outputFile, resolution, printSummary);
CommandLine commandLine = new CommandLine(args);
ValidateInterpretation validateInterpretation = null;
try
{
CommandLineParser.parse(commandLine, arguments);
if (about.wasFound()) {
About.about(System.out);
System.exit(0);
}
if (help.wasFound()) {
Usage.usage(USAGE, null, commandLine, arguments, System.out);
System.exit(0);
}
validateInterpretation = new ValidateInterpretation(expectedFile.getValue(), observedFile.getValue(), outputFile.getValue(), resolution.getValue(DEFAULT_RESOLUTION), printSummary.wasFound());
}
catch (CommandLineParseException e) {
if (about.wasFound()) {
About.about(System.out);
System.exit(0);
}
if (help.wasFound()) {
Usage.usage(USAGE, null, commandLine, arguments, System.out);
System.exit(0);
}
Usage.usage(USAGE, e, commandLine, arguments, System.err);
System.exit(-1);
}
try {
System.exit(validateInterpretation.call());
}
catch (Exception e) {
e.printStackTrace();
System.exit(1);
}
}
}