X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=sequence.cpp;h=38e9b6230838424dd136562111f25967f6822099;hb=c27e902ede71ea4d20af65752ef04459c101611b;hp=10a964848bd24f81a2f7658b0f1711427fbe3897;hpb=2ad3477dcd6a01628712b32e767353f917f2a95a;p=mothur.git

diff --git a/sequence.cpp b/sequence.cpp
index 10a9648..38e9b62 100644
--- a/sequence.cpp
+++ b/sequence.cpp
@@ -10,54 +10,310 @@
 #include "sequence.hpp"
 
 /***********************************************************************/
-
 Sequence::Sequence(){
+	m = MothurOut::getInstance();
 	initialize();
 }
-
 /***********************************************************************/
-
 Sequence::Sequence(string newName, string sequence) {
+	try {
+		m = MothurOut::getInstance();
+		initialize();	
+		name = newName;
+		
+		//setUnaligned removes any gap characters for us
+		setUnaligned(sequence);
+		setAligned(sequence);
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}			
+}
+/***********************************************************************/
+Sequence::Sequence(string newName, string sequence, string justUnAligned) {
+	try {
+		m = MothurOut::getInstance();
+		initialize();	
+		name = newName;
+		
+		//setUnaligned removes any gap characters for us
+		setUnaligned(sequence);
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}			
+}
 
-	initialize();	
-	name = newName;
-	
-	//setUnaligned removes any gap characters for us
-	setUnaligned(sequence);
-	setAligned(sequence);
+//********************************************************************************************************************
+//this function will jump over commented out sequences, but if the last sequence in a file is commented out it makes a blank seq
+Sequence::Sequence(istringstream& fastaString){
+	try {
+		m = MothurOut::getInstance();
 	
+		initialize();
+		fastaString >> name;
+		
+		if (name.length() != 0) { 
+		
+			name = name.substr(1);
+			string sequence;
+		
+			//read comments
+			while ((name[0] == '#') && fastaString) { 
+				while (!fastaString.eof())	{	char c = fastaString.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+				sequence = getCommentString(fastaString);
+				
+				if (fastaString) {  
+					fastaString >> name;  
+					name = name.substr(1);	
+				}else { 
+					name = "";
+					break;
+				}
+			}
+			
+			while (!fastaString.eof())	{	char c = fastaString.get();  if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+			
+			sequence = getSequenceString(fastaString);		
+			setAligned(sequence);	
+			//setUnaligned removes any gap characters for us						
+			setUnaligned(sequence);		
+		}else{ m->mothurOut("Error in reading your fastafile, at position " + toString(fastaString.tellg()) + ". Blank name."); m->mothurOutEndLine(); }
+
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}								
 }
 //********************************************************************************************************************
+//this function will jump over commented out sequences, but if the last sequence in a file is commented out it makes a blank seq
+Sequence::Sequence(istringstream& fastaString, string JustUnaligned){
+	try {
+		m = MothurOut::getInstance();
+	
+		initialize();
+		fastaString >> name;
+		
+		if (name.length() != 0) { 
+		
+			name = name.substr(1);
+			string sequence;
+		
+			//read comments
+			while ((name[0] == '#') && fastaString) { 
+				while (!fastaString.eof())	{	char c = fastaString.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+				sequence = getCommentString(fastaString);
+				
+				if (fastaString) {  
+					fastaString >> name;  
+					name = name.substr(1);	
+				}else { 
+					name = "";
+					break;
+				}
+			}
+			
+			while (!fastaString.eof())	{	char c = fastaString.get();  if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+			
+			sequence = getSequenceString(fastaString);		
+			
+			//setUnaligned removes any gap characters for us						
+			setUnaligned(sequence);		
+		}else{ m->mothurOut("Error in reading your fastafile, at position " + toString(fastaString.tellg()) + ". Blank name."); m->mothurOutEndLine(); }
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}								
+}
 
+
+//********************************************************************************************************************
+//this function will jump over commented out sequences, but if the last sequence in a file is commented out it makes a blank seq
 Sequence::Sequence(ifstream& fastaFile){
+	try {
+		m = MothurOut::getInstance();
+		initialize();
+		fastaFile >> name;
+
+		if (name.length() != 0) { 
+		
+			name = name.substr(1); 
+			
+			string sequence;
+		
+			//read comments
+			while ((name[0] == '#') && fastaFile) { 
+				while (!fastaFile.eof())	{	char c = fastaFile.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+				sequence = getCommentString(fastaFile);
+				
+				if (fastaFile) {  
+					fastaFile >> name;  
+					name = name.substr(1);	
+				}else { 
+					name = "";
+					break;
+				}
+			}
+			
+			//read real sequence
+			while (!fastaFile.eof())	{	char c = fastaFile.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+			
+			sequence = getSequenceString(fastaFile);		
+		
+			setAligned(sequence);	
+			//setUnaligned removes any gap characters for us						
+			setUnaligned(sequence);	
+		}else{ m->mothurOut("Error in reading your fastafile, at position " + toString(fastaFile.tellg()) + ". Blank name."); m->mothurOutEndLine(); }
+		
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}							
+}
+//********************************************************************************************************************
+//this function will jump over commented out sequences, but if the last sequence in a file is commented out it makes a blank seq
+Sequence::Sequence(ifstream& fastaFile, string JustUnaligned){
+	try {
+		m = MothurOut::getInstance();
+		initialize();
+		fastaFile >> name;
+		
+		if (name.length() != 0) { 
+			name = name.substr(1);
+			string sequence;
+			
+			//read comments
+			while ((name[0] == '#') && fastaFile) { 
+				while (!fastaFile.eof())	{	char c = fastaFile.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+				sequence = getCommentString(fastaFile);
+				
+				if (fastaFile) {  
+					fastaFile >> name;  
+					name = name.substr(1);	
+				}else { 
+					name = "";
+					break;
+				}
+			}
+			
+			//read real sequence
+			while (!fastaFile.eof())	{	char c = fastaFile.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+			
+			sequence = getSequenceString(fastaFile);		
+			
+			//setUnaligned removes any gap characters for us						
+			setUnaligned(sequence);	
+		}else{ m->mothurOut("Error in reading your fastafile, at position " + toString(fastaFile.tellg()) + ". Blank name."); m->mothurOutEndLine(); }
 
-	initialize();
-	fastaFile >> name;
-	name = name.substr(1);
-	
-	while (!fastaFile.eof())	{	char c = fastaFile.get(); if (c == 10 || c == 13){	break;	}	} // get rest of line if there's any crap there
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}							
+}
 
-	char letter;
-	string sequence;
+//********************************************************************************************************************
+string Sequence::getSequenceString(ifstream& fastaFile) {
+	try {
+		char letter;
+		string sequence = "";	
+		
+		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;
+			}
+		}
+		
+		return sequence;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "getSequenceString");
+		exit(1);
+	}
+}
+//********************************************************************************************************************
+//comment can contain '>' so we need to account for that
+string Sequence::getCommentString(ifstream& fastaFile) {
+	try {
+		char letter;
+		string sequence = "";
+		
+		while(fastaFile){
+			letter=fastaFile.get();
+			if((letter == '\r') || (letter == '\n')){  
+				gobble(fastaFile);  //in case its a \r\n situation
+				break;
+			}
+		}
+		
+		return sequence;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "getCommentString");
+		exit(1);
+	}
+}
+//********************************************************************************************************************
+string Sequence::getSequenceString(istringstream& fastaFile) {
+	try {
+		char letter;
+		string sequence = "";	
+		
+		while(!fastaFile.eof()){
+			letter= fastaFile.get();
 	
-	while(fastaFile){
-		letter= fastaFile.get();
-		if(letter == '>'){
-			fastaFile.putback(letter);
-			break;
+			if(letter == '>'){
+				fastaFile.putback(letter);
+				break;
+			}
+			else if(isprint(letter)){
+				letter = toupper(letter);
+				if(letter == 'U'){letter = 'T';}
+				sequence += letter;
+			}
 		}
-		else if(isprint(letter)){
-			letter = toupper(letter);
-			if(letter == 'U'){letter = 'T';}
-			sequence += letter;
+		
+		return sequence;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "getSequenceString");
+		exit(1);
+	}
+}
+//********************************************************************************************************************
+//comment can contain '>' so we need to account for that
+string Sequence::getCommentString(istringstream& fastaFile) {
+	try {
+		char letter;
+		string sequence = "";
+		
+		while(fastaFile){
+			letter=fastaFile.get();
+			if((letter == '\r') || (letter == '\n')){  
+				gobble(fastaFile);  //in case its a \r\n situation
+				break;
+			}
 		}
+		
+		return sequence;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "getCommentString");
+		exit(1);
 	}
-
-	setAligned(sequence);	
-	//setUnaligned removes any gap characters for us						
-	setUnaligned(sequence);								
 }
-
 //********************************************************************************************************************
 
 void Sequence::initialize(){
@@ -109,6 +365,7 @@ 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();
+	setUnaligned(sequence);	
 
 	if(aligned[0] == '-'){
 		for(int i=0;i<alignmentLength;i++){
@@ -165,7 +422,8 @@ string Sequence::getName(){
 //********************************************************************************************************************
 
 string Sequence::getAligned(){
-	return aligned;
+	if(isAligned == 0)	{ return unaligned; }
+	else				{  return aligned;  }
 }
 
 //********************************************************************************************************************
@@ -222,6 +480,18 @@ int Sequence::getAmbigBases(){
 
 //********************************************************************************************************************
 
+void Sequence::removeAmbigBases(){
+	
+	for(int j=0;j<alignmentLength;j++){
+		if(aligned[j] != 'A' && aligned[j] != 'T' && aligned[j] != 'G' && aligned[j] != 'C'){
+			aligned[j] = '-';
+		}
+	}
+	setUnaligned(aligned);
+}
+	
+//********************************************************************************************************************
+
 int Sequence::getLongHomoPolymer(){
 	if(longHomoPolymer == -1){
 		longHomoPolymer = 1;
@@ -291,7 +561,7 @@ void Sequence::reverseComplement(){
 		else						{	temp += 'N';	}
 	}
 	unaligned = temp;
+	aligned = temp;
 	
 }
-
-//********************************************************************************************************************
+/**************************************************************************************************/