Mercurial > repos > sanbi-uwc > qc
annotate qc.py @ 0:2aa014ad54bc draft
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
author | sanbi-uwc |
---|---|
date | Tue, 30 Mar 2021 15:00:18 +0000 |
parents | |
children | 878f18249f76 |
rev | line source |
---|---|
0
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
1 #!/usr/bin/env python3 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
2 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
3 from Bio import SeqIO |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
4 import csv |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
5 import subprocess |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
6 import pandas as pd |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
7 import matplotlib.pyplot as plt |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
8 import shlex |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
9 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
10 """ |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
11 This script can incorporate as many QC checks as required |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
12 as long as it outputs a csv file containing a final column |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
13 headed with 'qc_pass' and rows for each sample indcating |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
14 'TRUE' if the overall QC check has passed or 'FALSE' if not. |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
15 """ |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
16 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
17 def make_qc_plot(depth_pos, n_density, samplename, window=200): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
18 depth_df = pd.DataFrame( { 'position' : [pos[1] for pos in depth_pos], 'depth' : [dep[2] for dep in depth_pos] } ) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
19 depth_df['depth_moving_average'] = depth_df.iloc[:,1].rolling(window=window).mean() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
20 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
21 n_df = pd.DataFrame( { 'position' : [pos[0] for pos in n_density], 'n_density' : [dens[1] for dens in n_density] } ) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
22 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
23 fig, ax1 = plt.subplots() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
24 ax2 = ax1.twinx() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
25 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
26 ax1.set_xlabel('Position') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
27 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
28 ax1.set_ylabel('Depth', color = 'g') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
29 ax1.set_ylim(top=10**5, bottom=1) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
30 ax1.set_yscale('log') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
31 ax1.plot(depth_df['depth_moving_average'], color = 'g') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
32 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
33 ax2.set_ylabel('N density', color = 'r') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
34 ax2.plot(n_df['n_density'], color = 'r') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
35 ax2.set_ylim(top=1) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
36 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
37 plt.title(samplename) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
38 plt.savefig(samplename + '.depth.png') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
39 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
40 def read_depth_file(bamfile): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
41 p = subprocess.Popen(['samtools', 'depth', '-a', '-d', '0', bamfile], |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
42 stdout=subprocess.PIPE) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
43 out, err = p.communicate() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
44 counter = 0 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
45 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
46 pos_depth = [] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
47 for ln in out.decode('utf-8').split("\n"): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
48 if ln: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
49 pos_depth.append(ln.split("\t")) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
50 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
51 return pos_depth |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
52 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
53 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
54 def get_covered_pos(pos_depth, min_depth): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
55 counter = 0 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
56 for contig, pos,depth in pos_depth: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
57 if int(depth) >= min_depth: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
58 counter = counter + 1 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
59 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
60 return counter |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
61 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
62 def get_N_positions(fasta): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
63 n_pos = [i for i, letter in enumerate(fasta.seq.lower()) if letter == 'n'] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
64 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
65 return n_pos |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
66 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
67 def get_pct_N_bases(fasta): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
68 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
69 count_N = len(get_N_positions(fasta)) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
70 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
71 pct_N_bases = count_N / len(fasta.seq) * 100 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
72 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
73 return pct_N_bases |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
74 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
75 def get_largest_N_gap(fasta): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
76 n_pos = get_N_positions(fasta) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
77 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
78 n_pos = [0] + n_pos + [len(fasta.seq)] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
79 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
80 n_gaps = [j-i for i, j in zip(n_pos[:-1], n_pos[1:])] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
81 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
82 return sorted(n_gaps)[-1] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
83 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
84 def get_ref_length(ref): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
85 record = SeqIO.read(ref, "fasta") |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
86 return len(record.seq) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
87 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
88 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
89 def sliding_window_N_density(sequence, window=10): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
90 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
91 sliding_window_n_density = [] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
92 for i in range(0, len(sequence.seq), 1): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
93 window_mid = i + ( window / 2) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
94 window_seq = sequence.seq[i:i+window] |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
95 n_count = window_seq.lower().count('n') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
96 n_density = n_count / window |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
97 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
98 sliding_window_n_density.append( [ window_mid, n_density ] ) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
99 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
100 return sliding_window_n_density |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
101 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
102 def get_num_reads(bamfile): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
103 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
104 st_filter = '0x900' |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
105 command = 'samtools view -c -F{} {}'.format(st_filter, bamfile) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
106 what = shlex.split(command) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
107 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
108 return subprocess.check_output(what).decode().strip() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
109 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
110 def go(args): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
111 if args.illumina: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
112 depth = 10 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
113 elif args.nanopore: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
114 depth = 20 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
115 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
116 ## Depth calcs |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
117 ref_length = get_ref_length(args.ref) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
118 depth_pos = read_depth_file(args.bam) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
119 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
120 depth_covered_bases = get_covered_pos(depth_pos, depth) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
121 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
122 pct_covered_bases = depth_covered_bases / ref_length * 100 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
123 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
124 ## Number of aligned reads calculaton |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
125 num_reads = get_num_reads(args.bam) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
126 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
127 # Unknown base calcs |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
128 fasta = SeqIO.read(args.fasta, "fasta") |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
129 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
130 pct_N_bases = 0 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
131 largest_N_gap = 0 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
132 qc_pass = "FALSE" |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
133 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
134 if len(fasta.seq) != 0: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
135 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
136 pct_N_bases = get_pct_N_bases(fasta) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
137 largest_N_gap = get_largest_N_gap(fasta) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
138 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
139 # QC PASS / FAIL |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
140 if largest_N_gap >= 10000 or pct_N_bases < 50.0: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
141 qc_pass = "TRUE" |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
142 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
143 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
144 qc_line = { 'sample_name' : args.sample, |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
145 'pct_N_bases' : "{:.2f}".format(pct_N_bases), |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
146 'pct_covered_bases' : "{:.2f}".format(pct_covered_bases), |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
147 'longest_no_N_run' : largest_N_gap, |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
148 'num_aligned_reads' : num_reads, |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
149 'fasta': args.fasta, |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
150 'bam' : args.bam, |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
151 'qc_pass' : qc_pass} |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
152 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
153 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
154 with open(args.outfile, 'w') as csvfile: |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
155 header = qc_line.keys() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
156 writer = csv.DictWriter(csvfile, fieldnames=header) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
157 writer.writeheader() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
158 writer.writerow(qc_line) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
159 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
160 N_density = sliding_window_N_density(fasta) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
161 make_qc_plot(depth_pos, N_density, args.sample) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
162 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
163 def main(): |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
164 import argparse |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
165 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
166 parser = argparse.ArgumentParser() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
167 group = parser.add_mutually_exclusive_group(required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
168 group.add_argument('--nanopore', action='store_true') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
169 group.add_argument('--illumina', action='store_true') |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
170 parser.add_argument('--outfile', required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
171 parser.add_argument('--sample', required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
172 parser.add_argument('--ref', required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
173 parser.add_argument('--bam', required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
174 parser.add_argument('--fasta', required=True) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
175 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
176 args = parser.parse_args() |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
177 go(args) |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
178 |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
179 if __name__ == "__main__": |
2aa014ad54bc
"planemo upload for repository https://github.com/SANBI-SA/tools-sanbi-uwc/tree/master/tools/qc commit 54450d56a66bd4b01a17c8ec8aa3e649e3e4749f-dirty"
sanbi-uwc
parents:
diff
changeset
|
180 main() |