1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
|
package stream;
import java.util.ArrayList;
import fileIO.ByteFile;
import fileIO.FileFormat;
import shared.Shared;
public class FastqReadInputStream extends ReadInputStream {
public static void main(String[] args){
FastqReadInputStream fris=new FastqReadInputStream(args[0], true);
Read r=fris.next();
System.out.println(r.toText(false));
}
public FastqReadInputStream(String fname, boolean allowSubprocess_){
this(FileFormat.testInput(fname, FileFormat.FASTQ, null, allowSubprocess_, false));
}
public FastqReadInputStream(FileFormat ff){
if(verbose){System.err.println("FastqReadInputStream("+ff+")");}
flag=(Shared.AMINO_IN ? Read.AAMASK : 0);
stdin=ff.stdio();
if(!ff.fastq()){
System.err.println("Warning: Did not find expected fastq file extension for filename "+ff.name());
}
if(FASTQ.PARSE_CUSTOM){
try {
String s[]=ff.name().split("_");
// maxSnps=toNumber(s[3]);
// maxInss=toNumber(s[4]);
// maxDels=toNumber(s[5]);
// maxSubs=toNumber(s[6]);
// s=s[8].split("\\.");
//
// s=s[0].split("-");
// if(s.length!=8 && s.length!=9){
// if(Shared.WINDOWS){System.err.println("Note: Filename indicates non-synthetic data, but FASTQ.PARSE_CUSTOM="+FASTQ.PARSE_CUSTOM);}
// }
// minChrom=Gene.toChromosome(s[0]);
// maxChrom=Gene.toChromosome(s[1]);
} catch (Exception e) {
// TODO Auto-generated catch block
// e.printStackTrace();
// if(Shared.WINDOWS){System.err.println("Note: Filename indicates non-synthetic data, but FASTQ.PARSE_CUSTOM="+FASTQ.PARSE_CUSTOM);}
}
}
// interleaved=false;
interleaved=ff.interleaved();//(ff.stdio()) ? FASTQ.FORCE_INTERLEAVED : FASTQ.isInterleaved(ff.name(), false);
tf=ByteFile.makeByteFile(ff);
// assert(false) : interleaved;
}
@Override
public void start() {
// if(cris!=null){cris.start();}
}
@Override
public boolean hasMore() {
if(buffer==null || next>=buffer.size()){
if(tf.isOpen()){
fillBuffer();
}else{
assert(generated>0) : "Was the file empty?";
}
}
return (buffer!=null && next<buffer.size());
}
@Override
public Read next() {
if(!hasMore()){return null;}
Read r=buffer.set(next, null);
next++;
consumed++;
return r;
}
@Override
public synchronized ArrayList<Read> nextList() {
if(next!=0){throw new RuntimeException("'next' should not be used when doing blockwise access.");}
if(buffer==null || next>=buffer.size()){fillBuffer();}
ArrayList<Read> list=buffer;
buffer=null;
if(list!=null && list.size()==0){list=null;}
consumed+=(list==null ? 0 : list.size());
return list;
}
private synchronized void fillBuffer(){
assert(buffer==null || next>=buffer.size());
buffer=null;
next=0;
buffer=FASTQ.toReadList(tf, BUF_LEN, nextReadID, interleaved, flag);
int bsize=(buffer==null ? 0 : buffer.size());
nextReadID+=bsize;
if(bsize<BUF_LEN){tf.close();}
generated+=bsize;
if(buffer==null){
if(!errorState){
errorState=true;
System.err.println("Null buffer in FastqReadInputStream.");
}
}
}
@Override
public boolean close(){
if(verbose){System.err.println("Closing "+this.getClass().getName()+" for "+tf.name()+"; errorState="+errorState);}
errorState|=tf.close();
if(verbose){System.err.println("Closed "+this.getClass().getName()+" for "+tf.name()+"; errorState="+errorState);}
return errorState;
}
@Override
public synchronized void restart() {
generated=0;
consumed=0;
next=0;
nextReadID=0;
buffer=null;
tf.reset();
}
@Override
public boolean paired() {return interleaved;}
@Override
public String fname(){return tf.name();}
/** Return true if this stream has detected an error */
@Override
public boolean errorState(){return errorState || FASTQ.errorState();}
private ArrayList<Read> buffer=null;
private int next=0;
private final ByteFile tf;
private final boolean interleaved;
private final int flag;
private final int BUF_LEN=Shared.bufferLen();;
private final long MAX_DATA=Shared.bufferData(); //TODO - lot of work for unlikely case of super-long fastq reads. Must be disabled for paired-ends.
public long generated=0;
public long consumed=0;
private long nextReadID=0;
public final boolean stdin;
public static boolean verbose=false;
}
|