annotate tabular_to_fastq.py @ 0:b334cd1095ea draft

Imported from capsule None
author devteam
date Mon, 27 Jan 2014 09:27:43 -0500
parents
children b8cdc0507586
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
1 #Dan Blankenberg
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
2 import sys
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
3
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
4 def main():
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
5 input_filename = sys.argv[1]
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
6 output_filename = sys.argv[2]
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
7 identifier_col = int( sys.argv[3] ) - 1
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
8 sequence_col = int( sys.argv[4] ) - 1
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
9 quality_col = int( sys.argv[5] ) - 1
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
10
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
11 max_col = max( identifier_col, sequence_col, quality_col )
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
12 num_reads = None
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
13 fastq_read = None
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
14 skipped_lines = 0
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
15 out = open( output_filename, 'wb' )
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
16 for num_reads, line in enumerate( open( input_filename ) ):
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
17 fields = line.rstrip( '\n\r' ).split( '\t' )
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
18 if len( fields ) > max_col:
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
19 out.write( "@%s\n%s\n+\n%s\n" % ( fields[identifier_col], fields[sequence_col], fields[quality_col] ) )
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
20 else:
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
21 skipped_lines += 1
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
22
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
23 out.close()
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
24 if num_reads is None:
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
25 print "Input was empty."
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
26 else:
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
27 print "%i tabular lines were written as FASTQ reads. Be sure to use the FASTQ Groomer tool on this output before further analysis." % ( num_reads + 1 - skipped_lines )
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
28
b334cd1095ea Imported from capsule None
devteam
parents:
diff changeset
29 if __name__ == "__main__": main()