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>); //pdiffs, bdiffs, ldiffs, sdiffs, primers, barcodes
  33                 ~TrimOligos();
  34
  35                 int stripBarcode(Sequence&, int&);
  36                 int stripBarcode(Sequence&, QualityScores&, int&);
  37         int stripBarcode(Sequence&, Sequence&, QualityScores&, QualityScores&, int&);
  38         int stripBarcode(Sequence&, Sequence&, int&);
  39
  40                 int stripForward(Sequence&, int&);
  41                 int stripForward(Sequence&, QualityScores&, int&, bool);
  42         int stripForward(Sequence&, Sequence&, QualityScores&, QualityScores&, int&);
  43         int stripForward(Sequence&, Sequence&, int&);
  44
  45                 bool stripReverse(Sequence&);
  46                 bool stripReverse(Sequence&, QualityScores&);
  47
  48         bool stripLinker(Sequence&);
  49         bool stripLinker(Sequence&, QualityScores&);
  50
  51         bool stripSpacer(Sequence&);
  52         bool stripSpacer(Sequence&, QualityScores&);
  53
  54
  55         private:
  56                 int pdiffs, bdiffs, ldiffs, sdiffs;
  57
  58                 map<string, int> barcodes;
  59                 map<string, int> primers;
  60                 vector<string> revPrimer;
  61         vector<string> linker;
  62         vector<string> spacer;
  63         map<string, vector<int> > ifbarcodes;
  64         map<string, vector<int> > ifprimers;
  65         map<string, vector<int> > irbarcodes;
  66         map<string, vector<int> > irprimers;
  67         map<int, oligosPair> ipbarcodes;
  68         map<int, oligosPair> ipprimers;
  69
  70         int maxFBarcodeLength, maxRBarcodeLength, maxFPrimerLength, maxRPrimerLength, maxLinkerLength, maxSpacerLength;
  71
  72                 MothurOut* m;
  73
  74                 bool compareDNASeq(string, string);
  75                 int countDiffs(string, string);
  76 };
  77
  78 #endif
  79