Mercurial > repos > artbio > small_rna_signatures

annotate overlapping_reads.py @ 3:c0e72bd28453 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit bc672cf9d5fe9c1f7eaf02abdc906cf2c1763668
author artbio
date Sat, 25 Feb 2023 09:54:18 +0000
parents ca2b04cdbf4d
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
1 import argparse
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
2 from collections import defaultdict
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
3
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
4 import pysam
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
5
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
6
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
7 def Parser():
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
8 the_parser = argparse.ArgumentParser()
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
9 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
10 '--input', action="store", type=str, help="bam alignment file")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
11 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
12 '--minquery', type=int,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
13 help="Minimum readsize of query reads (nt) - must be an integer")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
14 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
15 '--maxquery', type=int,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
16 help="Maximum readsize of query reads (nt) - must be an integer")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
17 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
18 '--mintarget', type=int,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
19 help="Minimum readsize of target reads (nt) - must be an integer")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
20 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
21 '--maxtarget', type=int,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
22 help="Maximum readsize of target reads (nt) - must be an integer")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
23 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
24 '--overlap', type=int,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
25 help="Overlap analyzed (nt) - must be an integer")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
26 the_parser.add_argument(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
27 '--output', action="store", type=str,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
28 help="Pairable sequences")
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
29 args = the_parser.parse_args()
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
30 return args
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
31
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
32
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
33 class Map:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
34
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
35 def __init__(self, bam_file, output, minquery=23, maxquery=29,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
36 mintarget=23, maxtarget=29, overlap=10):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
37 self.bam_object = pysam.AlignmentFile(bam_file, 'rb')
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
38 self.output = output
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
39 self.query_range = range(minquery, maxquery + 1)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
40 self.target_range = range(mintarget, maxtarget + 1)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
41 self.overlap = overlap
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
42 self.chromosomes = dict(zip(self.bam_object.references,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
43 self.bam_object.lengths))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
44 self.alignement_dic = self.index_alignments(self.bam_object)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
45 self.all_query_positions = self.query_positions(self.bam_object,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
46 overlap=self.overlap)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
47 self.readdic = self.make_readdic(self.bam_object)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
48 self.pairing()
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
49
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
50 def make_readdic(self, bam_object):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
51 readdic = defaultdict(int)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
52 for read in bam_object.fetch():
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
53 readdic[read.query_sequence] += 1
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
54 return readdic
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
55
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
56 def index_alignments(self, bam_object):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
57 '''
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
58 dic[(chrom, pos, polarity)]: [readseq1, readseq2, ...]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
59 the list value is further converted in set
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
60 '''
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
61 dic = defaultdict(list)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
62 for chrom in self.chromosomes:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
63 for read in bam_object.fetch(chrom):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
64 if read.is_reverse:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
65 coord = read.reference_end-1
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
66 pol = 'R'
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
67 else:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
68 coord = read.reference_start
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
69 pol = 'F'
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
70 dic[(chrom, coord, pol)].append(read.query_sequence)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
71 for key in dic:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
72 dic[key] = set(dic[key])
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
73 return dic
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
74
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
75 def query_positions(self, bam_object, overlap):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
76 all_query_positions = defaultdict(list)
1
ca2b04cdbf4d "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 8691f2575cacd71c971338658198a1324e5f9370"
artbio
parents: 0
diff changeset
77 for genomicKey in list(self.alignement_dic):
0
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
78 chrom, coord, pol = genomicKey
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
79 if pol == 'F' and len(self.alignement_dic[(chrom,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
80 coord+overlap-1,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
81 'R')]) > 0:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
82 all_query_positions[chrom].append(coord)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
83 for chrom in all_query_positions:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
84 all_query_positions[chrom] = sorted(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
85 list(set(all_query_positions[chrom])))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
86 return all_query_positions
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
87
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
88 def countpairs(self, uppers, lowers):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
89 query_range = self.query_range
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
90 target_range = self.target_range
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
91 uppers = [seq for seq in uppers if (len(seq) in query_range or len(seq)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
92 in target_range)]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
93 uppers_expanded = []
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
94 for seq in uppers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
95 expand = [seq for i in range(self.readdic[seq])]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
96 uppers_expanded.extend(expand)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
97 uppers = uppers_expanded
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
98 lowers = [seq for seq in lowers if (len(seq) in query_range or len(seq)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
99 in target_range)]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
100 lowers_expanded = []
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
101 for seq in lowers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
102 expand = [seq for i in range(self.readdic[seq])]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
103 lowers_expanded.extend(expand)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
104 lowers = lowers_expanded
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
105 paired = []
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
106 for upread in uppers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
107 for downread in lowers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
108 if (len(upread) in query_range and len(downread) in
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
109 target_range) or (len(upread) in target_range and
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
110 len(downread) in query_range):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
111 paired.append(upread)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
112 lowers.remove(downread)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
113 break
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
114 return len(paired)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
115
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
116 def pairing(self):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
117 F = open(self.output, 'w')
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
118 query_range = self.query_range
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
119 target_range = self.target_range
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
120 overlap = self.overlap
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
121 stringresult = []
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
122 header_template = '>%s|coord=%s|strand %s|size=%s|nreads=%s\n%s\n'
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
123 total_pairs = 0
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
124 for chrom in sorted(self.chromosomes):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
125 number_pairs = 0
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
126 for pos in self.all_query_positions[chrom]:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
127 stringbuffer = []
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
128 uppers = self.alignement_dic[chrom, pos, 'F']
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
129 lowers = self.alignement_dic[chrom, pos+overlap-1, 'R']
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
130 number_pairs += self.countpairs(uppers, lowers)
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
131 total_pairs += number_pairs
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
132 if uppers and lowers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
133 for upread in uppers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
134 for downread in lowers:
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
135 if (len(upread) in query_range and len(downread) in
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
136 target_range) or (len(upread) in target_range
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
137 and len(downread) in
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
138 query_range):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
139 stringbuffer.append(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
140 header_template %
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
141 (chrom, pos+1, '+', len(upread),
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
142 self.readdic[upread], upread))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
143 stringbuffer.append(
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
144 header_template %
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
145 (chrom, pos+overlap-len(downread)+1, '-',
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
146 len(downread), self.readdic[downread],
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
147 self.revcomp(downread)))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
148 stringresult.extend(sorted(set(stringbuffer)))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
149 F.write(''.join(stringresult))
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
150
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
151 def revcomp(self, sequence):
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
152 antidict = {"A": "T", "T": "A", "G": "C", "C": "G", "N": "N"}
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
153 revseq = sequence[::-1]
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
154 return "".join([antidict[i] for i in revseq])
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
155
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
156
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
157 if __name__ == "__main__":
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
158 args = Parser()
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
159 mapobj = Map(args.input, args.output, args.minquery, args.maxquery,
3e0ea204d09e "planemo upload for repository https://github.com/ARTbio/tools-artbio/tree/master/tools/small_rna_signatures commit 4e84386b619a7941f23d175d7fc86aba7990ac36"
artbio
parents:
diff changeset
160 args.mintarget, args.maxtarget, args.overlap)