This is a walker that lets the user generate the bed file for declaring variants true positives or false positives. For use with the IGV crowd sourcing project.

git-svn-id: file:///humgen/gsa-scr1/gsa-engineering/svn_contents/trunk@5440 348d0f76-0448-11de-a6fe-93d51630548a
This commit is contained in:
corin 2011-03-14 19:56:16 +00:00
parent f75ad0dee3
commit 6d09cdd4bc
1 changed files with 126 additions and 0 deletions

View File

@ -0,0 +1,126 @@
/*
* Copyright (c) 2010, The Broad Institute
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use,
* copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
package org.broadinstitute.sting.playground.gatk.walkers.validation;
import com.sun.tools.corba.se.idl.constExpr.Not;
import com.sun.tools.internal.xjc.reader.xmlschema.bindinfo.BIConversion;
import org.broad.tribble.Feature;
import org.broad.tribble.util.variantcontext.Genotype;
import org.broad.tribble.util.variantcontext.VariantContext;
import org.broad.tribble.vcf.*;
import org.broadinstitute.sting.gatk.contexts.AlignmentContext;
import org.broadinstitute.sting.gatk.contexts.ReferenceContext;
import org.broadinstitute.sting.gatk.contexts.variantcontext.VariantContextUtils;
import org.broadinstitute.sting.gatk.datasources.rmd.ReferenceOrderedDataSource;
import org.broadinstitute.sting.gatk.refdata.RefMetaDataTracker;
import org.broadinstitute.sting.gatk.walkers.*;
import org.broadinstitute.sting.commandline.Argument;
import org.broadinstitute.sting.commandline.Output;
import org.broadinstitute.sting.utils.GenomeLoc;
import org.broadinstitute.sting.utils.SampleUtils;
import org.broadinstitute.sting.utils.bed.BedParser;
import org.broadinstitute.sting.utils.vcf.VCFUtils;
import java.io.PrintStream;
import java.io.Writer;
import java.util.*;
/**
* Declares the validity of variants in a vcf as either true or false. For use with the IGV crowd-sourcing bed generation
*/
public class DeclareValidityWalker extends RodWalker<Integer, Integer>{
@Output(doc = "Write bed to this file instead of standard out")
public PrintStream out;
@Argument(fullName = "DeclareFalse", shortName = "isFP",
doc = "Argument to create a bed file of with loci annotated as false positive", required = false)
boolean isFP = false;
@Argument(fullName = "Note", shortName = "N", doc = "Annotation to be included in FP field", required = false)
String Note =".";
@Argument(fullName = "Source", shortName = "s", doc = "Source of annotation", required = false)
String Source = ".";
@Override
public Integer reduceInit() {
return 0; //To change body of implemented methods use File | Settings | File Templates.
}
public class ValidityDeclaration{
protected String contig;
protected long start;
protected long stop;
protected boolean TPorFP;
protected String Note;
protected String Source;
protected String Build;
protected String user;
public ValidityDeclaration(GenomeLoc Loc, Boolean TPorFP, String Note, String Source){ //Constructor expects 1 based
this.contig=Loc.getContig();
this.start=Loc.getStart()-1;
this.stop=Loc.getStop()-1;
this.TPorFP =TPorFP;
this.Note = Note;
this.Source = Source;
this.user = System.getenv("USER");
//is there a better way to get the build than this?:
if (this.contig.contains("chr")) {this.Build="hg18";}
else {this.Build="hg19";}
}
public String toString() {
return String.format("%s\t%d\t%d\t%b positive\t%s\t%s\t%s\t%s\n", contig, start, stop, TPorFP, user, Build, Note, Source);
}
}
/**
*
* @param tracker the meta-data tracker
* @param ref the reference base
* @param context the context for the given locus
* @return 1 if the locus was successfully processed, 0 if otherwise
*/
public Integer map(RefMetaDataTracker tracker, ReferenceContext ref, AlignmentContext context){
if ( tracker == null )
return 0;
Boolean tpOrFp = !isFP;
ValidityDeclaration bedLine = new ValidityDeclaration(ref.getLocus(), tpOrFp, Note, Source);
out.append(bedLine.toString());
return 1;
}
public Integer reduce(Integer counter, Integer sum) {
return counter+sum;
}
}