]> git.donarmstrong.com Git - mothur.git/blobdiff - sequence.cpp
continued work on chimeras and fixed bug in trim.seqs and reverse.seqs that was due...
[mothur.git] / sequence.cpp
index 7bc8b051f2f7465ee0dd7c030ffd31ca0569ac2c..b2c2b53d0b4078e6e296312662dcdfe0ffe5fbec 100644 (file)
@@ -7,67 +7,79 @@
  *
  */
 
-using namespace std;
-
 #include "sequence.hpp"
 
-//********************************************************************************************************************
+/***********************************************************************/
+
+Sequence::Sequence(){
+       initialize();
+}
 
-Sequence::Sequence(){}
+/***********************************************************************/
 
+Sequence::Sequence(string newName, string sequence) {
+
+       initialize();   
+       name = newName;
+       
+       //setUnaligned removes any gap characters for us
+       setUnaligned(sequence);
+       setAligned(sequence);
+       
+}
 //********************************************************************************************************************
 
 Sequence::Sequence(ifstream& fastaFile){
+
+       initialize();
+       fastaFile >> name;
+       name = name.substr(1);
        
-       string accession;
-       fastaFile >> accession;
-       setName(accession);
+       while (!fastaFile.eof())        {       char c = fastaFile.get(); if (c == 10 || c == 13){      break;  }       } // get rest of line if there's any crap there
 
        char letter;
        string sequence;
        
-       while(fastaFile && letter != '>'){
-       
-               letter = fastaFile.get();
-               
-               if(isalpha(letter)){
-               
+       while(fastaFile){
+               letter= fastaFile.get();
+               if(letter == '>'){
+                       fastaFile.putback(letter);
+                       break;
+               }
+               else if(isprint(letter)){
+                       letter = toupper(letter);
+                       if(letter == 'U'){letter = 'T';}
                        sequence += letter;
-                       
                }
-               
        }
-       fastaFile.putback(letter);
 
-       if(sequence.find_first_of('-') != string::npos){
-               setAligned(sequence);
-       }
-       setUnaligned(sequence);
+       setAligned(sequence);   
+       //setUnaligned removes any gap characters for us                                                
+       setUnaligned(sequence);                                                         
 }
 
-
 //********************************************************************************************************************
 
-string Sequence::convert2ints(){
+void Sequence::initialize(){
        
-       if(unaligned == "")     {       /* need to throw an error */    }
+       name = "";
+       unaligned = "";
+       aligned = "";
+       pairwise = "";
        
-       string processed;
+       numBases = 0;
+       alignmentLength = 0;
+       isAligned = 0;
+       startPos = -1;
+       endPos = -1;
+       longHomoPolymer = -1;
+       ambigBases = -1;
        
-       for(int i=0;i<unaligned.length();i++){
-               if(toupper(unaligned[i]) == 'A')                        {       processed += '0';       }
-               else if(toupper(unaligned[i]) == 'C')   {       processed += '1';       }
-               else if(toupper(unaligned[i]) == 'G')   {       processed += '2';       }
-               else if(toupper(unaligned[i]) == 'T')   {       processed += '3';       }
-               else if(toupper(unaligned[i]) == 'U')   {       processed += '3';       }
-               else                                                                    {       processed += '4';       }
-       }
-       return processed;
-}
+}      
 
 //********************************************************************************************************************
 
-void Sequence::setName(string seqName){
+void Sequence::setName(string seqName) {
        if(seqName[0] == '>')   {       name = seqName.substr(1);       }
        else                                    {       name = seqName;                         }
 }
@@ -76,23 +88,47 @@ void Sequence::setName(string seqName){
 
 void Sequence::setUnaligned(string sequence){
        
-       if(sequence.find_first_of('-') != string::npos){
+       if(sequence.find_first_of('.') != string::npos || sequence.find_first_of('-') != string::npos) {
                string temp = "";
-               for(int j=0;j<sequence.length();j++){
+               for(int j=0;j<sequence.length();j++) {
                        if(isalpha(sequence[j]))        {       temp += sequence[j];    }
                }
                unaligned = temp;
        }
-       else{
+       else {
                unaligned = sequence;
        }
+       numBases = unaligned.length();
        
 }
 
 //********************************************************************************************************************
 
 void Sequence::setAligned(string sequence){
+       
+       //if the alignment starts or ends with a gap, replace it with a period to indicate missing data
        aligned = sequence;
+       alignmentLength = aligned.length();
+
+       if(aligned[0] == '-'){
+               for(int i=0;i<alignmentLength;i++){
+                       if(aligned[i] == '-'){
+                               aligned[i] = '.';
+                       }
+                       else{
+                               break;
+                       }
+               }
+               for(int i=alignmentLength-1;i>=0;i--){
+                       if(aligned[i] == '-'){
+                               aligned[i] = '.';
+                       }
+                       else{
+                               break;
+                       }
+               }
+       }
+       isAligned = 1;  
 }
 
 //********************************************************************************************************************
@@ -103,7 +139,26 @@ void Sequence::setPairwise(string sequence){
 
 //********************************************************************************************************************
 
-string Sequence::getSeqName(){
+string Sequence::convert2ints() {
+       
+       if(unaligned == "")     {       /* need to throw an error */    }
+       
+       string processed;
+       
+       for(int i=0;i<unaligned.length();i++) {
+               if(toupper(unaligned[i]) == 'A')                {       processed += '0';       }
+               else if(toupper(unaligned[i]) == 'C')   {       processed += '1';       }
+               else if(toupper(unaligned[i]) == 'G')   {       processed += '2';       }
+               else if(toupper(unaligned[i]) == 'T')   {       processed += '3';       }
+               else if(toupper(unaligned[i]) == 'U')   {       processed += '3';       }
+               else                                                                    {       processed += '4';       }
+       }
+       return processed;
+}
+
+//********************************************************************************************************************
+
+string Sequence::getName(){
        return name;
 }
 
@@ -126,3 +181,118 @@ string Sequence::getUnaligned(){
 }
 
 //********************************************************************************************************************
+
+int Sequence::getNumBases(){
+       return numBases;
+}
+
+//********************************************************************************************************************
+
+void Sequence::printSequence(ostream& out){
+
+       out << ">" << name << endl;
+       if(isAligned){
+               out << aligned << endl;
+       }
+       else{
+               out << unaligned << endl;
+       }
+}
+
+//********************************************************************************************************************
+
+int Sequence::getAlignLength(){
+       return alignmentLength;
+}
+
+//********************************************************************************************************************
+
+int Sequence::getAmbigBases(){
+       if(ambigBases == -1){
+               ambigBases = 0;
+               for(int j=0;j<numBases;j++){
+                       if(unaligned[j] != 'A' && unaligned[j] != 'T' && unaligned[j] != 'G' && unaligned[j] != 'C'){
+                               ambigBases++;
+                       }
+               }
+       }       
+       
+       return ambigBases;
+}
+
+//********************************************************************************************************************
+
+int Sequence::getLongHomoPolymer(){
+       if(longHomoPolymer == -1){
+               longHomoPolymer = 1;
+               int homoPolymer = 1;
+               for(int j=1;j<numBases;j++){
+                       if(unaligned[j] == unaligned[j-1]){
+                               homoPolymer++;
+                       }
+                       else{
+                               if(homoPolymer > longHomoPolymer){      longHomoPolymer = homoPolymer;  }
+                               homoPolymer = 1;
+                       }
+               }
+               if(homoPolymer > longHomoPolymer){      longHomoPolymer = homoPolymer;  }
+       }
+       return longHomoPolymer;
+}
+
+//********************************************************************************************************************
+
+int Sequence::getStartPos(){
+       if(endPos == -1){
+               for(int j = 0; j < alignmentLength; j++) {
+                       if(aligned[j] != '.'){
+                               startPos = j + 1;
+                               break;
+                       }
+               }
+       }
+       if(isAligned == 0){     startPos = 1;   }
+
+       return startPos;
+}
+
+//********************************************************************************************************************
+
+int Sequence::getEndPos(){
+       if(endPos == -1){
+               for(int j=alignmentLength-1;j>=0;j--){
+                       if(aligned[j] != '.'){
+                               endPos = j + 1;
+                               break;
+                       }
+               }
+       }
+       if(isAligned == 0){     endPos = numBases;      }
+       
+       return endPos;
+}
+
+//********************************************************************************************************************
+
+bool Sequence::getIsAligned(){
+       return isAligned;
+}
+
+//********************************************************************************************************************
+
+void Sequence::reverseComplement(){
+
+       string temp;
+       for(int i=numBases-1;i>=0;i--){
+               if(unaligned[i] == 'A')         {       temp += 'T';    }
+               else if(unaligned[i] == 'T'){   temp += 'A';    }
+               else if(unaligned[i] == 'G'){   temp += 'C';    }
+               else if(unaligned[i] == 'C'){   temp += 'G';    }
+               else                                            {       temp += 'N';    }
+       }
+       unaligned = temp;
+       aligned = temp;
+       
+}
+
+//********************************************************************************************************************