annotate bedClean.py @ 12:2f4ea569f048

Uploaded
author xuebing
date Sat, 10 Mar 2012 08:10:44 -0500
parents b7f1d9f8f3bc
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
11
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
1 import sys
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
2
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
3 def readChrSize(filename):
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
4 f = open(filename)
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
5 chrSize = {}
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
6 for line in f:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
7 chrom,size = line.strip().split()
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
8 chrSize[chrom]=int(size)
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
9 f.close()
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
10 return chrSize
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
11
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
12 def cleanFile(filename,chrSize,outfile):
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
13 f = open(filename)
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
14 out = open(outfile,'w')
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
15 i = 0
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
16 for line in f:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
17 i = i + 1
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
18 flds = line.strip().split('\t')
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
19 if len(flds) < 3:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
20 print 'line',i,'incomplete line:\n',line
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
21 elif chrSize.has_key(flds[0]):
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
22 if int(flds[1]) > int(flds[2]):
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
23 tmp = flds[1]
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
24 flds[1] = flds[2]
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
25 flds[2] = tmp
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
26 if int( flds[1]) < 0 or int(flds[2]) <0:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
27 print 'line',i,'negative coordinates:\n',line
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
28 elif int(flds[2]) > chrSize[flds[0]]:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
29 print 'line',i,'end larger than chr size:\n',line
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
30 else:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
31 out.write('\t'.join(flds)+'\n')
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
32 else:
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
33 print 'line',i,'chromosome',flds[0],'not found!\n',line
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
34 f.close()
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
35 out.close()
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
36
b7f1d9f8f3bc Uploaded
xuebing
parents:
diff changeset
37 cleanFile(sys.argv[1],readChrSize(sys.argv[2]),sys.argv[3])