annotate revcompl.py @ 15:0e221dbd17b2 default tip

Uploaded
author xuebing
date Sat, 31 Mar 2012 08:53:06 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
15
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
1 import sys
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
2
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
3 compldna = {'A':'T',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
4 'C':'G',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
5 'G':'C',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
6 'T':'A',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
7 'U':'A',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
8 'M':'K',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
9 'K':'M',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
10 'W':'W',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
11 'S':'S',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
12 'R':'Y',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
13 'Y':'R',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
14 'N':'N'}
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
15 complrna = {'A':'U',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
16 'C':'G',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
17 'G':'C',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
18 'T':'A',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
19 'U':'A',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
20 'M':'K',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
21 'K':'M',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
22 'W':'W',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
23 'S':'S',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
24 'R':'Y',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
25 'Y':'R',
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
26 'N':'N'}
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
27 def complement(seq,compl):
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
28 complseq = [compl[base] for base in seq]
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
29 return complseq
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
30
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
31 def reverse_complement(seq,compl):
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
32 seq = list(seq)
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
33 seq.reverse()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
34 return ''.join(complement(seq,compl))
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
35
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
36 def readFastaFile(infile,outfile,compl):
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
37
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
38 fin = open(infile)
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
39 out = open(outfile,'w')
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
40
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
41 currSeq=''
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
42 currSeqname=None
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
43 for line in fin:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
44 if '>' in line:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
45 if currSeqname !=None:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
46 out.write(currSeqname+reverse_complement(currSeq,compl)+'\n')
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
47 currSeqname=None
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
48 currSeq=''
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
49 currSeqname=line
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
50 else:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
51 currSeq=currSeq+line.strip().upper()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
52
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
53 if currSeqname!=None:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
54 out.write(currSeqname+reverse_complement(currSeq,compl)+'\n')
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
55
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
56 fin.close()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
57 out.close()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
58
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
59 def readrawseq(infile,outfile,compl):
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
60 '''
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
61 each line is a sequence
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
62 '''
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
63 fin = open(infile)
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
64 out = open(outfile,'w')
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
65 for line in fin:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
66 out.write(reverse_complement(line.strip().upper(),compl)+'\n')
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
67 fin.close()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
68 out.close()
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
69
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
70 def main():
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
71 seqfile = sys.argv[1]
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
72 outfile = sys.argv[2]
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
73 fasta = sys.argv[3]
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
74 rna = sys.argv[4]
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
75 if rna == 'rna':
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
76 compl = complrna
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
77 else:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
78 compl = compldna
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
79 if fasta == 'fasta':
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
80 readFastaFile(seqfile,outfile,compl)
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
81 else:
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
82 readrawseq(seqfile,outfile,compl)
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
83
0e221dbd17b2 Uploaded
xuebing
parents:
diff changeset
84 main()