zzh
/
gatk-3.8
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactored to clean it up a bit 

git-svn-id: file:///humgen/gsa-scr1/gsa-engineering/svn_contents/trunk@495 348d0f76-0448-11de-a6fe-93d51630548a
This commit is contained in:
ebanks 2009-04-22 19:18:46 +00:00
parent 1bf4d040d8
commit d99d67d51c
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
java/src/org/broadinstitute/sting/playground/gatk/walkers/ReadFilterWalker.java
View File

@ -15,18 +15,20 @@ public class ReadFilterWalker extends ReadWalker<Integer,Integer> {
@Argument(fullName="output_file", shortName="O",doc="SAM or BAM file to write filtered reads into (will be overwritten if exists)",required=true ) public String output;
@Argument(fullName="max_read_length",doc="Discard reads with length greater than the specified value",required=false) public Integer max_len;
private SAMFileWriter writer = null;
public void initialize() {
SAMFileHeader header = getToolkit().getSamReader().getFileHeader();
writer = new SAMFileWriterFactory().makeSAMOrBAMWriter(header, header.getSortOrder() != SAMFileHeader.SortOrder.unsorted, new File(output));
}
@Override
public boolean filter(LocusContext context, SAMRecord read) {
if ( read.getReadLength() > max_len ) return false;
return true;
return read.getReadLength() <= max_len;
}
@Override
public Integer map(LocusContext context, SAMRecord read) {
if ( writer == null ) writer = new SAMFileWriterFactory().makeSAMOrBAMWriter(read.getHeader(), read.getHeader().getSortOrder() != SAMFileHeader.SortOrder.unsorted, new File(output));
writer.addAlignment(read);
return 1;
}
@ -42,8 +44,8 @@ public class ReadFilterWalker extends ReadWalker<Integer,Integer> {
}
public void onTraversalDone(Integer nReads) {
super.onTraversalDone(nReads);
out.println(nReads +" reads passed the filter and were written into output file "+output);
writer.close();
out.println(nReads +" reads passed the filter and were written into output file "+output);
}
}