forked from samtools/samtools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
coverage.c
685 lines (602 loc) · 27.3 KB
/
coverage.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
/* coverage.c -- samtools coverage subcommand
Copyright (C) 2018,2019 Florian Breitwieser
Portions copyright (C) 2019-2021 Genome Research Ltd.
Author: Florian P Breitwieser <[email protected]>
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
DEALINGS IN THE SOFTWARE. */
/* This program calculates coverage from multiple BAMs
* simultaneously, to achieve random access and to use the BED interface.
* To compile this program separately, you may:
*
* gcc -g -O2 -Wall -o bamcov -D_MAIN_BAMCOV coverage.c -lhts -lz
*/
// C headers
#include <config.h>
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h> // variadic functions
#include <limits.h> // INT_MAX
#include <math.h> // round
#include <stdbool.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#ifdef _WIN32
#include <windows.h>
#else
#include <sys/ioctl.h>
#endif
#include "htslib/sam.h"
#include "htslib/hts.h"
#include "samtools.h"
#include "sam_opts.h"
const char *VERSION = "0.1";
typedef struct { // auxiliary data structure to hold stats on coverage
unsigned long long n_covered_bases;
unsigned long long summed_coverage;
unsigned long long summed_baseQ;
unsigned long long summed_mapQ;
unsigned int n_reads;
unsigned int n_selected_reads;
bool covered;
hts_pos_t beg;
hts_pos_t end;
int64_t bin_width;
} stats_aux_t;
typedef struct { // auxiliary data structure to hold a BAM file
samFile *fp; // file handle
sam_hdr_t *hdr; // file header
hts_itr_t *iter; // iterator to a region - NULL for us by default
int min_mapQ; // mapQ filter
int min_len; // length filter
int fail_flags;
int required_flags;
stats_aux_t *stats;
} bam_aux_t;
#if __STDC_VERSION__ >= 199901L
#define VERTICAL_LINE "\u2502" // BOX DRAWINGS LIGHT VERTICAL
// UTF8 specifies block characters in eights going from \u2581 (lower one eight block) to \u2588 (full block)
// https://en.wikipedia.org/wiki/Block_Elements
// LOWER ONE EIGHTH BLOCK … FULL BLOCK
static const char *const BLOCK_CHARS8[8] = {"\u2581", "\u2582", "\u2583", "\u2584", "\u2585", "\u2586", "\u2587", "\u2588"};
// In some terminals / with some fonts not all UTF8 block characters are supported (e.g. Putty). Use only half and full block for those
static const char *const BLOCK_CHARS2[2] = {".", ":"};
#else
// Fall back to explicit UTF-8 encodings of the same characters
#define VERTICAL_LINE "\xE2\x94\x82"
static const char *const BLOCK_CHARS8[8] = {
"\xE2\x96\x81", "\xE2\x96\x82", "\xE2\x96\x83", "\xE2\x96\x84",
"\xE2\x96\x85", "\xE2\x96\x86", "\xE2\x96\x87", "\xE2\x96\x88" };
static const char *const BLOCK_CHARS2[2] = {".", ":"};
#endif
// in bam_plcmd.c
int read_file_list(const char *file_list, int *n, char **argv[]);
static int usage() {
fprintf(stdout, "Usage: samtools coverage [options] in1.bam [in2.bam [...]]\n\n"
"Input options:\n"
" -b, --bam-list FILE list of input BAM filenames, one per line\n"
" -l, --min-read-len INT ignore reads shorter than INT bp [0]\n"
" -q, --min-MQ INT mapping quality threshold [0]\n"
" -Q, --min-BQ INT base quality threshold [0]\n"
" --rf <int|str> required flags: skip reads with mask bits unset []\n"
" --ff <int|str> filter flags: skip reads with mask bits set \n"
" [UNMAP,SECONDARY,QCFAIL,DUP]\n"
" -d, --depth INT maximum allowed coverage depth [1000000].\n"
" If 0, depth is set to the maximum integer value,\n"
" effectively removing any depth limit.\n"
"Output options:\n"
" -m, --histogram show histogram instead of tabular output\n"
" -A, --ascii show only ASCII characters in histogram\n"
" -o, --output FILE write output to FILE [stdout]\n"
" -H, --no-header don't print a header in tabular mode\n"
" -w, --n-bins INT number of bins in histogram [terminal width - 40]\n"
" -r, --region REG show specified region. Format: chr:start-end. \n"
" -h, --help help (this page)\n");
fprintf(stdout, "\nGeneric options:\n");
sam_global_opt_help(stdout, "-.--.--.");
fprintf(stdout,
"\nSee manpage for additional details.\n"
" rname Reference name / chromosome\n"
" startpos Start position\n"
" endpos End position (or sequence length)\n"
" numreads Number reads aligned to the region (after filtering)\n"
" covbases Number of covered bases with depth >= 1\n"
" coverage Proportion of covered bases [0..1]\n"
" meandepth Mean depth of coverage\n"
" meanbaseq Mean baseQ in covered region\n"
" meanmapq Mean mapQ of selected reads\n"
);
return EXIT_SUCCESS;
}
static char* center_text(char *text, char *buf, int width) {
int len = strlen(text);
assert(len <= width);
int padding = (width - len) / 2;
int padding_ex = (width - len) % 2;
if (padding >= 1)
sprintf(buf, " %*s%*s", len+padding, text, padding-1+padding_ex, " ");
else
sprintf(buf, "%s", text);
return buf;
}
static char* readable_bps(double base_pairs, char *buf) {
const char* units[] = {"", "K", "M", "G", "T"};
int i = 0;
while (base_pairs >= 1000 && i < (sizeof(units)/sizeof(units[0]) - 1)) {
base_pairs /= 1000;
i++;
}
sprintf(buf, "%.*f%s", i, base_pairs, units[i]);
return buf;
}
// read one alignment from one BAM file
static int read_bam(void *data, bam1_t *b) {
bam_aux_t *aux = (bam_aux_t*)data; // data in fact is a pointer to an auxiliary structure
int nref = sam_hdr_nref(aux->hdr);
int ret;
while (1) {
if((ret = aux->iter? sam_itr_next(aux->fp, aux->iter, b) : sam_read1(aux->fp, aux->hdr, b)) < 0) break;
if (b->core.tid >= 0 && b->core.tid < nref)
aux->stats[b->core.tid].n_reads++;
if ( aux->fail_flags && (b->core.flag & aux->fail_flags) ) continue;
if ( aux->required_flags && !(b->core.flag & aux->required_flags) ) continue;
if ( b->core.qual < aux->min_mapQ ) continue;
if ( aux->min_len && bam_cigar2qlen(b->core.n_cigar, bam_get_cigar(b)) < aux->min_len ) continue;
if (b->core.tid >= 0 && b->core.tid < nref) {
aux->stats[b->core.tid].n_selected_reads++;
aux->stats[b->core.tid].summed_mapQ += b->core.qual;
}
break;
}
return ret;
}
void print_tabular_line(FILE *file_out, const sam_hdr_t *h, const stats_aux_t *stats, int tid) {
fputs(sam_hdr_tid2name(h, tid), file_out);
double region_len = (double) stats[tid].end - stats[tid].beg;
fprintf(file_out, "\t%"PRId64"\t%"PRId64"\t%u\t%llu\t%g\t%g\t%.3g\t%.3g\n",
stats[tid].beg+1,
stats[tid].end,
stats[tid].n_selected_reads,
stats[tid].n_covered_bases,
100.0 * stats[tid].n_covered_bases / region_len,
stats[tid].summed_coverage / region_len,
stats[tid].summed_coverage > 0? stats[tid].summed_baseQ/(double) stats[tid].summed_coverage : 0,
stats[tid].n_selected_reads > 0? stats[tid].summed_mapQ/(double) stats[tid].n_selected_reads : 0
);
}
void print_hist(FILE *file_out, const sam_hdr_t *h, const stats_aux_t *stats, int tid, const uint32_t *hist,
const int hist_size, const bool full_utf) {
int i, col;
bool show_percentiles = false;
const int n_rows = 10;
const char * const * BLOCK_CHARS = full_utf? BLOCK_CHARS8 : BLOCK_CHARS2;
const int blockchar_len = full_utf? 8 : 2;
double region_len = stats[tid].end - stats[tid].beg;
// Calculate histogram that contains percent covered
double hist_data[hist_size];
double max_val = 0.0;
for (i = 0; i < hist_size; ++i) {
hist_data[i] = 100 * hist[i] / (double) stats[tid].bin_width;
if (hist_data[i] > max_val) max_val = hist_data[i];
}
char buf[30];
fprintf(file_out, "%s (%sbp)\n", sam_hdr_tid2name(h, tid), readable_bps(sam_hdr_tid2len(h, tid), buf));
double row_bin_size = max_val / (double) n_rows;
for (i = n_rows-1; i >= 0; --i) {
double current_bin = row_bin_size * i;
if (show_percentiles) {
fprintf(file_out, ">%3i%% ", i*10);
} else {
fprintf(file_out, ">%7.2f%% ", current_bin);
}
fprintf(file_out, full_utf ? VERTICAL_LINE : "|");
for (col = 0; col < hist_size; ++col) {
// get the difference in eights, or halfs when full UTF8 is not supported
int cur_val_diff = round(blockchar_len * (hist_data[col] - current_bin) / row_bin_size) - 1;
if (cur_val_diff < 0) {
fputc(' ', file_out);
} else {
if (cur_val_diff >= blockchar_len)
cur_val_diff = blockchar_len - 1;
fprintf(file_out, "%s", BLOCK_CHARS[cur_val_diff]);
}
}
fprintf(file_out, full_utf ? VERTICAL_LINE : "|");
fputc(' ', file_out);
switch (i) {
case 9: fprintf(file_out, "Number of reads: %i", stats[tid].n_selected_reads); break;
case 8: if (stats[tid].n_reads - stats[tid].n_selected_reads > 0) fprintf(file_out, " (%i filtered)", stats[tid].n_reads - stats[tid].n_selected_reads); break;
case 7: fprintf(file_out, "Covered bases: %sbp", readable_bps(stats[tid].n_covered_bases, buf)); break;
case 6: fprintf(file_out, "Percent covered: %.4g%%",
100.0 * stats[tid].n_covered_bases / region_len); break;
case 5: fprintf(file_out, "Mean coverage: %.3gx",
stats[tid].summed_coverage / region_len); break;
case 4: fprintf(file_out, "Mean baseQ: %.3g",
stats[tid].summed_baseQ/(double) stats[tid].summed_coverage); break;
case 3: fprintf(file_out, "Mean mapQ: %.3g",
stats[tid].summed_mapQ/(double) stats[tid].n_selected_reads); break;
case 1: fprintf(file_out, "Histo bin width: %sbp",
readable_bps(stats[tid].bin_width, buf)); break;
case 0: fprintf(file_out, "Histo max bin: %.5g%%", max_val); break;
};
fputc('\n', file_out);
}
// print x axis. Could be made pretty for widths that are not divisible
// by 10 by variable spacing of the labels, instead of placing a label every 10 characters
char buf2[50];
fprintf(file_out, " %s", center_text(readable_bps(stats[tid].beg + 1, buf), buf2, 10));
int rest;
for (rest = 10; rest < 10*(hist_size/10); rest += 10) {
fprintf(file_out, "%s", center_text(readable_bps(stats[tid].beg + stats[tid].bin_width*rest, buf), buf2, 10));
}
int last_padding = hist_size%10;
fprintf(file_out, "%*s%s", last_padding, " ", center_text(readable_bps(stats[tid].end, buf), buf2, 10));
fprintf(file_out, "\n");
}
int main_coverage(int argc, char *argv[]) {
int status = EXIT_SUCCESS;
int ret, tid = -1, old_tid = -1, pos, i, j;
int max_depth = 1000000;
int opt_min_baseQ = 0;
int opt_min_mapQ = 0;
int opt_min_len = 0;
int opt_n_bins = 50;
bool opt_full_width = true;
char *opt_output_file = NULL;
bam_aux_t **data = NULL;
bam_mplp_t mplp = NULL;
const bam_pileup1_t **plp = NULL;
uint32_t *hist = NULL;
stats_aux_t *stats = NULL;
char *opt_reg = 0; // specified region
char *opt_file_list = NULL;
int n_bam_files = 0;
char **fn = NULL;
int fail_flags = (BAM_FUNMAP | BAM_FSECONDARY | BAM_FQCFAIL | BAM_FDUP); // Default fail flags
int required_flags = 0;
int *n_plp = NULL;
sam_hdr_t *h = NULL; // BAM header of the 1st input
bool opt_print_header = true;
bool opt_print_tabular = true;
bool opt_print_histogram = false;
bool opt_full_utf = true;
FILE *file_out = stdout;
sam_global_args ga = SAM_GLOBAL_ARGS_INIT;
static const struct option lopts[] = {
SAM_OPT_GLOBAL_OPTIONS('-', 0, '-', '-', 0, '-'),
{"rf", required_argument, NULL, 1}, // require flag
{"ff", required_argument, NULL, 2}, // filter flag
{"incl-flags", required_argument, NULL, 1}, // require flag
{"excl-flags", required_argument, NULL, 2}, // filter flag
{"bam-list", required_argument, NULL, 'b'},
{"min-read-len", required_argument, NULL, 'l'},
{"min-MQ", required_argument, NULL, 'q'},
{"min-mq", required_argument, NULL, 'q'},
{"min-BQ", required_argument, NULL, 'Q'},
{"min-bq", required_argument, NULL, 'Q'},
{"histogram", no_argument, NULL, 'm'},
{"ascii", no_argument, NULL, 'A'},
{"output", required_argument, NULL, 'o'},
{"no-header", no_argument, NULL, 'H'},
{"n-bins", required_argument, NULL, 'w'},
{"region", required_argument, NULL, 'r'},
{"help", no_argument, NULL, 'h'},
{"depth", required_argument, NULL, 'd'},
{ NULL, 0, NULL, 0 }
};
// parse the command line
int c;
opterr = 0;
while ((c = getopt_long(argc, argv, "Ao:l:q:Q:hHw:r:b:md:", lopts, NULL)) != -1) {
switch (c) {
case 1:
if ((required_flags = bam_str2flag(optarg)) < 0) {
fprintf(stderr,"Could not parse --rf %s\n", optarg); return EXIT_FAILURE;
}; break;
case 2:
if ((fail_flags = bam_str2flag(optarg)) < 0) {
fprintf(stderr,"Could not parse --ff %s\n", optarg); return EXIT_FAILURE;
}; break;
case 'o': opt_output_file = optarg; opt_full_width = false; break;
case 'l': opt_min_len = atoi(optarg); break;
case 'q': opt_min_mapQ = atoi(optarg); break;
case 'Q': opt_min_baseQ = atoi(optarg); break;
case 'd': max_depth = atoi(optarg); break; // maximum coverage depth
case 'w': opt_n_bins = atoi(optarg); opt_full_width = false;
opt_print_histogram = true; opt_print_tabular = false;
break;
case 'r': opt_reg = optarg; break; // parsing a region requires a BAM header (strdup unnecessary)
case 'b': opt_file_list = optarg; break;
case 'm': opt_print_histogram = true; opt_print_tabular = false; break;
case 'A': opt_full_utf = false;
opt_print_histogram = true; opt_print_tabular = false;
break;
case 'H': opt_print_header = false; break;
case 'h': return usage();
default: if (parse_sam_global_opt(c, optarg, lopts, &ga) == 0) break;
/* else fall-through */
case '?':
if (optopt != '?') { // '-?' appeared on command line
if (optopt) { // Bad short option
print_error("coverage", "invalid option -- '%c'", optopt);
} else { // Bad long option
// Do our best. There is no good solution to finding
// out what the bad option was.
// See, e.g. https://stackoverflow.com/questions/2723888/where-does-getopt-long-store-an-unrecognized-option
if (optind > 0 && strncmp(argv[optind - 1], "--", 2) == 0) {
print_error("coverage", "unrecognised option '%s'",
argv[optind - 1]);
}
}
}
return usage();
}
}
if (optind == argc && !opt_file_list)
return usage();
// output file provided by user
if (opt_output_file != NULL && strcmp(opt_output_file,"-")!=0) {
file_out = fopen( opt_output_file, "w" );
if (file_out == NULL) {
print_error_errno("coverage", "Cannot open \"%s\" for writing.", opt_output_file);
return EXIT_FAILURE;
}
}
if (opt_n_bins <= 0 || opt_full_width) {
// get number of columns of terminal
const char* env_columns = getenv("COLUMNS");
int columns = 0;
if (env_columns == NULL) {
#ifdef _WIN32
CONSOLE_SCREEN_BUFFER_INFO csbi;
if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &csbi)) {
columns = csbi.srWindow.Right - csbi.srWindow.Left + 1;
}
#elif defined TIOCGWINSZ
struct winsize w;
if (ioctl(2, TIOCGWINSZ, &w) == 0)
columns = w.ws_col;
#endif
} else {
columns = atoi(env_columns); // atoi(NULL) returns 0
}
if (columns > 60) {
opt_n_bins = columns - 40;
} else {
opt_n_bins = 40;
}
}
// setvbuf(file_out, NULL, _IONBF, 0); //turn off buffering
// Open all BAM files
if (opt_file_list) {
// Read file names from opt_file_list into argv, and record the number of files in n_bam_files
if (read_file_list(opt_file_list, &n_bam_files, &fn)) {
print_error_errno("coverage", "Cannot open file list \"%s\".", opt_file_list);
return EXIT_FAILURE;
}
argv = fn;
optind = 0;
} else {
n_bam_files = argc - optind; // the number of BAMs on the command line
}
data = (bam_aux_t **)calloc(n_bam_files, sizeof(bam_aux_t*)); // data[i] for the i-th BAM file
if (!data) {
print_error_errno("coverage", "Failed to allocate memory");
status = EXIT_FAILURE;
goto coverage_end;
}
for (i = 0; i < n_bam_files; ++i) {
int rf;
data[i] = (bam_aux_t *) calloc(1, sizeof(bam_aux_t));
if (!data[i]) {
print_error_errno("coverage", "Failed to allocate memory");
status = EXIT_FAILURE;
goto coverage_end;
}
data[i]->fp = sam_open_format(argv[optind+i], "r", &ga.in); // open BAM
if (data[i]->fp == NULL) {
print_error_errno("coverage", "Could not open \"%s\"", argv[optind+i]);
status = EXIT_FAILURE;
goto coverage_end;
}
rf = SAM_FLAG | SAM_RNAME | SAM_POS | SAM_MAPQ | SAM_CIGAR | SAM_SEQ;
if (opt_min_baseQ) rf |= SAM_QUAL;
// Set CRAM options on file handle - returns 0 on success
if (hts_set_opt(data[i]->fp, CRAM_OPT_REQUIRED_FIELDS, rf)) {
print_error("coverage", "Failed to set CRAM_OPT_REQUIRED_FIELDS value");
status = EXIT_FAILURE;
goto coverage_end;
}
if (hts_set_opt(data[i]->fp, CRAM_OPT_DECODE_MD, 0)) {
print_error("coverage", "Failed to set CRAM_OPT_DECODE_MD value");
status = EXIT_FAILURE;
goto coverage_end;
}
data[i]->min_mapQ = opt_min_mapQ; // set the mapQ filter
data[i]->min_len = opt_min_len; // set the qlen filter
data[i]->hdr = sam_hdr_read(data[i]->fp); // read the BAM header
data[i]->fail_flags = fail_flags;
data[i]->required_flags = required_flags;
if (data[i]->hdr == NULL) {
print_error_errno("coverage", "Could not read header for \"%s\"", argv[optind+i]);
status = EXIT_FAILURE;
goto coverage_end;
}
// Lookup region if specified
if (opt_reg) { // if a region is specified
hts_idx_t *idx = sam_index_load(data[i]->fp, argv[optind+i]); // load the index
if (idx == NULL) {
print_error_errno("coverage", "Failed to load index for \"%s\"", argv[optind+i]);
status = EXIT_FAILURE;
goto coverage_end;
}
data[i]->iter = sam_itr_querys(idx, data[i]->hdr, opt_reg); // set the iterator
hts_idx_destroy(idx); // the index is not needed any more; free the memory
if (data[i]->iter == NULL) {
print_error("coverage", "Failed to parse region \"%s\". Check the region format or region name presence in the file \"%s\"", opt_reg, argv[optind+i]);
status = EXIT_FAILURE;
goto coverage_end;
}
}
}
if (opt_print_tabular && opt_print_header)
fputs("#rname\tstartpos\tendpos\tnumreads\tcovbases\tcoverage\tmeandepth\tmeanbaseq\tmeanmapq\n", file_out);
h = data[0]->hdr; // easy access to the header of the 1st BAM
int n_targets = sam_hdr_nref(h);
stats = calloc(n_targets, sizeof(stats_aux_t));
if (!stats) {
print_error_errno("coverage", "Failed to allocate memory");
status = EXIT_FAILURE;
goto coverage_end;
}
int64_t n_bins = opt_n_bins;
if (opt_reg) {
stats_aux_t *s = stats + data[0]->iter->tid;
s->beg = data[0]->iter->beg; // and to the parsed region coordinates
s->end = data[0]->iter->end;
if (s->end == HTS_POS_MAX) {
s->end = sam_hdr_tid2len(h, data[0]->iter->tid);
}
if (opt_n_bins > s->end - s->beg) {
n_bins = s->end - s->beg;
}
s->bin_width = (s->end-s->beg) / (n_bins > 0 ? n_bins : 1);
}
for (i=0; i<n_bam_files; i++)
data[i]->stats = stats;
int64_t current_bin = 0;
// the core multi-pileup loop
mplp = bam_mplp_init(n_bam_files, read_bam, (void**)data); // initialization
if (max_depth > 0)
bam_mplp_set_maxcnt(mplp, max_depth); // set maximum coverage depth
else if (!max_depth)
bam_mplp_set_maxcnt(mplp, INT_MAX);
// Extra info for histogram and coverage counting
hist = (uint32_t*) calloc(opt_n_bins, sizeof(uint32_t));
n_plp = (int*) calloc(n_bam_files, sizeof(int*)); // n_plp[i] is the number of covering reads from the i-th BAM
plp = (const bam_pileup1_t**) calloc(n_bam_files, sizeof(bam_pileup1_t*)); // plp[i] points to the array of covering reads (internal in mplp)
if (!hist || !n_plp || !plp) {
print_error_errno("coverage", "Failed to allocate memory");
status = EXIT_FAILURE;
goto coverage_end;
}
while ((ret=bam_mplp_auto(mplp, &tid, &pos, n_plp, plp)) > 0) { // come to the next covered position
if (tid != old_tid) { // Next target sequence
if (old_tid >= 0) {
if (opt_print_histogram) {
print_hist(file_out, h, stats, old_tid, hist, n_bins, opt_full_utf);
fputc('\n', file_out);
} else if (opt_print_tabular) {
print_tabular_line(file_out, h, stats, old_tid);
}
if (opt_print_histogram)
memset(hist, 0, n_bins*sizeof(uint32_t));
}
stats[tid].covered = true;
if (!opt_reg)
stats[tid].end = sam_hdr_tid2len(h, tid);
if (opt_print_histogram) {
n_bins = opt_n_bins > stats[tid].end-stats[tid].beg? stats[tid].end-stats[tid].beg : opt_n_bins;
stats[tid].bin_width = (stats[tid].end-stats[tid].beg) / n_bins;
}
old_tid = tid;
}
if (pos < stats[tid].beg || pos >= stats[tid].end) continue; // out of range; skip
if (tid >= n_targets) continue; // diff number of @SQ lines per file?
if (opt_print_histogram) {
current_bin = (pos - stats[tid].beg) / stats[tid].bin_width;
}
bool count_base = false;
for (i = 0; i < n_bam_files; ++i) { // base level filters have to go here
int depth_at_pos = n_plp[i];
for (j = 0; j < n_plp[i]; ++j) {
const bam_pileup1_t *p = plp[i] + j; // DON'T modify plp[][] unless you really know
if (p->is_del || p->is_refskip) --depth_at_pos; // having dels or refskips at tid:pos
else if (p->qpos < p->b->core.l_qseq &&
bam_get_qual(p->b)[p->qpos] < opt_min_baseQ) --depth_at_pos; // low base quality
else
stats[tid].summed_baseQ += bam_get_qual(p->b)[p->qpos];
}
if (depth_at_pos > 0) {
count_base = true;
stats[tid].summed_coverage += depth_at_pos;
}
// hist[current_bin] += depth_at_pos; // Add counts to the histogram here to have one based on coverage
//fprintf(file_out, "\t%d", n_plp[i] - m); // this the depth to output
}
if (count_base) {
stats[tid].n_covered_bases++;
if (opt_print_histogram && current_bin < n_bins)
++(hist[current_bin]); // Histogram based on breadth of coverage
}
}
if (tid == -1 && opt_reg && *opt_reg != '*')
// Region specified but no data covering it.
tid = data[0]->iter->tid;
if (tid < n_targets && tid >=0) {
if (opt_print_histogram) {
print_hist(file_out, h, stats, tid, hist, n_bins, opt_full_utf);
} else if (opt_print_tabular) {
print_tabular_line(file_out, h, stats, tid);
}
}
if (!opt_reg && opt_print_tabular) {
for (i = 0; i < n_targets; ++i) {
if (!stats[i].covered) {
stats[i].end = sam_hdr_tid2len(h, i);
print_tabular_line(file_out, h, stats, i);
}
}
}
if (ret < 0) status = EXIT_FAILURE;
coverage_end:
if (n_plp) free(n_plp);
if (plp) free(plp);
if (mplp) bam_mplp_destroy(mplp);
if (hist) free(hist);
if (stats) free(stats);
// Close files and free data structures
if (!(file_out == stdout || fclose(file_out) == 0)) {
if (status == EXIT_SUCCESS) {
print_error_errno("coverage", "error on closing \"%s\"",
(opt_output_file && strcmp(opt_output_file, "-") != 0?
opt_output_file : "stdout"));
status = EXIT_FAILURE;
}
}
if (data) {
for (i = 0; i < n_bam_files && data[i]; ++i) {
sam_hdr_destroy(data[i]->hdr);
if (data[i]->fp) sam_close(data[i]->fp);
hts_itr_destroy(data[i]->iter);
free(data[i]);
}
free(data);
}
if (opt_file_list && fn) {
for (i = 0; i < n_bam_files; ++i)
free(fn[i]);
free(fn);
}
sam_global_args_free(&ga);
return status;
}
#ifdef _MAIN_BAMCOV
int main(int argc, char *argv[]) {
return main_coverage(argc, argv);
}
#endif