]> git.donarmstrong.com Git - mothur.git/blob - trimoligos.h
Merge remote-tracking branch 'mothur/master'
[mothur.git] / trimoligos.h
1 #ifndef TRIMOLIGOS_H
2 #define TRIMOLIGOS_H
3
4 /*
5  *  trimoligos.h
6  *  Mothur
7  *
8  *  Created by westcott on 9/1/11.
9  *  Copyright 2011 Schloss Lab. All rights reserved.
10  *
11  */
12
13 #include "mothur.h"
14 #include "mothurout.h"
15 #include "sequence.hpp"
16 #include "qualityscores.h"
17
18 struct oligosPair {
19         string forward;
20         string reverse;
21         
22         oligosPair() { forward = ""; reverse = "";  }
23         oligosPair(string f, string r) : forward(f), reverse(r) {}
24         ~oligosPair() {}
25 };
26
27 class TrimOligos {
28         
29         public:
30         TrimOligos(int,int, map<string, int>, map<string, int>, vector<string>); //pdiffs, bdiffs, primers, barcodes, revPrimers
31         TrimOligos(int,int, int, int, map<string, int>, map<string, int>, vector<string>, vector<string>, vector<string>); //pdiffs, bdiffs, ldiffs, sdiffs, primers, barcodes, revPrimers, linker, spacer
32         TrimOligos(int,int, int, int, map<int, oligosPair>, map<int, oligosPair>, vector<string>, vector<string>); //pdiffs, bdiffs, ldiffs, sdiffs, primers, barcodes, linker, spacer
33                 ~TrimOligos();
34         
35                 int stripBarcode(Sequence&, int&);      
36                 int stripBarcode(Sequence&, QualityScores&, int&);
37         int stripBarcode(Sequence&, Sequence&, QualityScores&, QualityScores&, int&);
38         
39                 int stripForward(Sequence&, int&);
40                 int stripForward(Sequence&, QualityScores&, int&, bool);
41         int stripForward(Sequence&, Sequence&, QualityScores&, QualityScores&, int&);
42         
43                 bool stripReverse(Sequence&);
44                 bool stripReverse(Sequence&, QualityScores&);
45     
46         bool stripLinker(Sequence&);
47         bool stripLinker(Sequence&, QualityScores&);
48     
49         bool stripSpacer(Sequence&);
50         bool stripSpacer(Sequence&, QualityScores&);
51                                 
52         
53         private:
54                 int pdiffs, bdiffs, ldiffs, sdiffs;
55         
56                 map<string, int> barcodes;
57                 map<string, int> primers;
58                 vector<string> revPrimer;
59         vector<string> linker;
60         vector<string> spacer;
61         map<int, oligosPair> ibarcodes;
62         map<int, oligosPair> iprimers;
63         
64                 MothurOut* m;
65         
66                 bool compareDNASeq(string, string);                             
67                 int countDiffs(string, string);                 
68 };
69
70 #endif
71