Mercurial > repos > yating-l > rename_tracks

annotate rename_tracks.py @ 1:75d9357a0a53 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
author yating-l
date Mon, 25 Jun 2018 13:12:47 -0400
parents dd6d3b009659
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
1 # -*- coding: utf8 -*-
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
2
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
3 """
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
4 Rename the custom evidence tracks so that the tracks use the same sequence names as the renamed reference
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
5 """
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
6 import sys
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
7 import csv
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
8 import subprocess
1
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
9 import tempfile
0
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
10
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
11 def rename_interval(inputFile, nameDict, renamedFile):
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
12 writer = open(renamedFile, 'w')
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
13 with open(inputFile, 'r') as f:
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
14 lines = f.readlines()
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
15 for l in lines:
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
16 if not l.startswith("#"):
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
17 scaffold_name = l.split()[0]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
18 if scaffold_name in nameDict:
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
19 l = l.replace(scaffold_name, nameDict[scaffold_name])
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
20 writer.write(l)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
21 writer.close()
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
22
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
23 def rename_bam(inputFile, nameDict, renamedFile):
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
24 header = subprocess.Popen(['samtools', 'view', '-H', inputFile], stdout=subprocess.PIPE)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
25 array_call = ['sed']
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
26 for k,v in nameDict.items():
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
27 substitute = "s/%s/%s/" % (str(k), str(v))
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
28 array_call.append('-e')
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
29 array_call.append(substitute)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
30 reheader = subprocess.Popen(array_call, stdin=header.stdout, stdout=subprocess.PIPE)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
31 out = open(renamedFile, 'w')
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
32 subprocess.Popen(['samtools', 'reheader', '-', inputFile], stdin=reheader.stdout, stdout=out)
1
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
33
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
34 def rename_bigwig(inputFile, renamedReference, nameDict, renamedFile):
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
35 bedGraphFile = tempfile.NamedTemporaryFile(bufsize=0)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
36 chrom_sizes = tempfile.NamedTemporaryFile(bufsize=0)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
37 sorted_bedGraphFile = tempfile.NamedTemporaryFile(bufsize=0)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
38 renamed_sorted_bedGraphFile = tempfile.NamedTemporaryFile(bufsize=0)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
39
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
40 subprocess.call(['bigWigToBedGraph', inputFile, bedGraphFile.name])
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
41 subprocess.call(['faSize', '-detailed', '-tab', renamedReference], stdout=chrom_sizes)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
42 subprocess.call(['sort', '-k1,1', '-k2,2n', bedGraphFile.name], stdout=sorted_bedGraphFile)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
43 rename_interval(sorted_bedGraphFile.name, nameDict, renamed_sorted_bedGraphFile.name)
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
44 subprocess.call(['bedGraphToBigWig', renamed_sorted_bedGraphFile.name, chrom_sizes.name, renamedFile])
0
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
45
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
46 def getNameDict(nameMapping):
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
47 nameDict = {}
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
48 with open(nameMapping, 'r') as f:
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
49 reader = csv.reader(f)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
50 for row in reader:
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
51 nameDict[row[0]] = row[1]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
52 return nameDict
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
53
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
54 def main():
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
55 inputFile = sys.argv[1]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
56 nameMapping = sys.argv[2]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
57 inputFormat = sys.argv[3]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
58 outputfile = sys.argv[4]
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
59 nameDict = getNameDict(nameMapping)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
60 if inputFormat == "bed" or inputFormat == "gff3" or inputFormat == "gtf":
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
61 rename_interval(inputFile, nameDict, outputfile)
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
62 elif inputFormat == "bam":
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
63 rename_bam(inputFile, nameDict, outputfile)
1
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
64 elif inputFormat == "bigwig":
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
65 renamedReference = sys.argv[5]
75d9357a0a53 planemo upload commit 58057cff13206fae66c938464c052f416047c1b3-dirty
yating-l
parents: 0
diff changeset
66 rename_bigwig(inputFile, renamedReference, nameDict, outputfile)
0
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
67
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
68 if __name__ == "__main__":
dd6d3b009659 planemo upload
yating-l
parents:
diff changeset
69 main()