annotate BEDTools-Version-2.14.3/src/coverageBed/coverageMain.cpp @ 0:dfcd8b6c1bda

Uploaded
author aaronquinlan
date Thu, 03 Nov 2011 10:25:04 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
1 /*****************************************************************************
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
2 coverageMain.cpp
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
3
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
4 (c) 2009 - Aaron Quinlan
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
5 Hall Laboratory
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
6 Department of Biochemistry and Molecular Genetics
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
7 University of Virginia
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
8 aaronquinlan@gmail.com
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
9
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
10 Licenced under the GNU General Public License 2.0 license.
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
11 ******************************************************************************/
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
12 #include "coverageBed.h"
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
13 #include "version.h"
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
14
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
15 using namespace std;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
16
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
17 // define the version
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
18 #define PROGRAM_NAME "coverageBed"
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
19
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
20 // define our parameter checking macro
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
21 #define PARAMETER_CHECK(param, paramLen, actualLen) (strncmp(argv[i], param, min(actualLen, paramLen))== 0) && (actualLen == paramLen)
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
22
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
23 // function declarations
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
24 void ShowHelp(void);
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
25
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
26 int main(int argc, char* argv[]) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
27
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
28 // our configuration variables
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
29 bool showHelp = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
30
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
31 // input files
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
32 string bedAFile;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
33 string bedBFile;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
34
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
35 // parm flags
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
36 bool sameStrand = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
37 bool diffStrand = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
38 bool writeHistogram = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
39 bool eachBase = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
40 bool obeySplits = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
41 bool bamInput = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
42 bool haveBedA = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
43 bool haveBedB = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
44 bool countsOnly = false;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
45
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
46 // check to see if we should print out some help
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
47 if(argc <= 1) showHelp = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
48
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
49 for(int i = 1; i < argc; i++) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
50 int parameterLength = (int)strlen(argv[i]);
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
51
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
52 if((PARAMETER_CHECK("-h", 2, parameterLength)) ||
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
53 (PARAMETER_CHECK("--help", 5, parameterLength))) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
54 showHelp = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
55 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
56 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
57
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
58 if(showHelp) ShowHelp();
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
59
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
60 // do some parsing (all of these parameters require 2 strings)
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
61 for(int i = 1; i < argc; i++) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
62
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
63 int parameterLength = (int)strlen(argv[i]);
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
64
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
65 if(PARAMETER_CHECK("-a", 2, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
66 if ((i+1) < argc) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
67 haveBedA = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
68 bedAFile = argv[i + 1];
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
69 i++;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
70 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
71 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
72 else if(PARAMETER_CHECK("-abam", 5, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
73 if ((i+1) < argc) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
74 haveBedA = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
75 bamInput = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
76 bedAFile = argv[i + 1];
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
77 i++;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
78 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
79 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
80 else if(PARAMETER_CHECK("-b", 2, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
81 if ((i+1) < argc) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
82 haveBedB = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
83 bedBFile = argv[i + 1];
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
84 i++;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
85 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
86 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
87 else if (PARAMETER_CHECK("-s", 2, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
88 sameStrand = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
89 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
90 else if (PARAMETER_CHECK("-S", 2, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
91 diffStrand = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
92 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
93 else if (PARAMETER_CHECK("-hist", 5, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
94 writeHistogram = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
95 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
96 else if(PARAMETER_CHECK("-d", 2, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
97 eachBase = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
98 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
99 else if (PARAMETER_CHECK("-split", 6, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
100 obeySplits = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
101 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
102 else if (PARAMETER_CHECK("-counts", 7, parameterLength)) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
103 countsOnly = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
104 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
105 else {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
106 cerr << endl << "*****ERROR: Unrecognized parameter: " << argv[i] << " *****" << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
107 showHelp = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
108 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
109 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
110
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
111 // make sure we have both input files
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
112 if (!haveBedA || !haveBedB) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
113 cerr << endl << "*****" << endl << "*****ERROR: Need -a and -b files. " << endl << "*****" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
114 showHelp = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
115 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
116
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
117 if (sameStrand && diffStrand) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
118 cerr << endl << "*****" << endl << "*****ERROR: Request either -s OR -S, not both." << endl << "*****" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
119 showHelp = true;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
120 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
121
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
122 if (!showHelp) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
123 BedCoverage *bg = new BedCoverage(bedAFile, bedBFile, sameStrand, diffStrand,
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
124 writeHistogram, bamInput, obeySplits, eachBase, countsOnly);
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
125 delete bg;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
126 return 0;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
127 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
128 else {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
129 ShowHelp();
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
130 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
131 }
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
132
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
133 void ShowHelp(void) {
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
134
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
135 cerr << endl << "Program: " << PROGRAM_NAME << " (v" << VERSION << ")" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
136
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
137 cerr << "Author: Aaron Quinlan (aaronquinlan@gmail.com)" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
138
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
139 cerr << "Summary: Returns the depth and breadth of coverage of features from A" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
140 cerr << "\t on the intervals in B." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
141
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
142 cerr << "Usage: " << PROGRAM_NAME << " [OPTIONS] -a <bed/gff/vcf> -b <bed/gff/vcf>" << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
143
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
144 cerr << "Options: " << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
145
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
146 cerr << "\t-abam\t" << "The A input file is in BAM format." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
147
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
148 cerr << "\t-s\t" << "Require same strandedness. That is, only counts hits in A that" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
149 cerr << "\t\toverlap B on the _same_ strand." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
150 cerr << "\t\t- By default, overlaps are counted without respect to strand." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
151
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
152 cerr << "\t-S\t" << "Require different strandedness. That is, only report hits in A that" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
153 cerr << "\t\toverlap B on the _opposite_ strand." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
154 cerr << "\t\t- By default, overlaps are counted without respect to strand." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
155
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
156 cerr << "\t-hist\t" << "Report a histogram of coverage for each feature in B" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
157 cerr << "\t\tas well as a summary histogram for _all_ features in B." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
158 cerr << "\t\tOutput (tab delimited) after each feature in B:" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
159 cerr << "\t\t 1) depth\n\t\t 2) # bases at depth\n\t\t 3) size of B\n\t\t 4) % of B at depth" << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
160
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
161 cerr << "\t-d\t" << "Report the depth at each position in each B feature." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
162 cerr << "\t\tPositions reported are one based. Each position" << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
163 cerr << "\t\tand depth follow the complete B feature." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
164
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
165 cerr << "\t-counts\t" << "Only report the count of overlaps, don't compute fraction, etc." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
166
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
167 cerr << "\t-split\t" << "Treat \"split\" BAM or BED12 entries as distinct BED intervals." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
168 cerr << "\t\twhen computing coverage." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
169 cerr << "\t\tFor BAM files, this uses the CIGAR \"N\" and \"D\" operations " << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
170 cerr << "\t\tto infer the blocks for computing coverage." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
171 cerr << "\t\tFor BED12 files, this uses the BlockCount, BlockStarts," << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
172 cerr << "\t\tand BlockEnds fields (i.e., columns 10,11,12)." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
173
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
174 cerr << "Default Output: " << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
175 cerr << "\t" << " After each entry in B, reports: " << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
176 cerr << "\t 1) The number of features in A that overlapped the B interval." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
177 cerr << "\t 2) The number of bases in B that had non-zero coverage." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
178 cerr << "\t 3) The length of the entry in B." << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
179 cerr << "\t 4) The fraction of bases in B that had non-zero coverage." << endl << endl;
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
180
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
181 exit(1);
dfcd8b6c1bda Uploaded
aaronquinlan
parents:
diff changeset
182 }