X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=sequence.cpp;h=19adf796b5d1d26f3a681144c6c801cdb4b4f63d;hb=438aa88dbc092d9c1c80ec3fa20c8e47f97101c4;hp=7bc8b051f2f7465ee0dd7c030ffd31ca0569ac2c;hpb=bfbc55964f1977da72c2cea984288a427d370a59;p=mothur.git

diff --git a/sequence.cpp b/sequence.cpp
index 7bc8b05..19adf79 100644
--- a/sequence.cpp
+++ b/sequence.cpp
@@ -7,67 +7,227 @@
  *
  */
 
-using namespace std;
-
 #include "sequence.hpp"
 
-//********************************************************************************************************************
-
-Sequence::Sequence(){}
+/***********************************************************************/
 
-//********************************************************************************************************************
+Sequence::Sequence(){
+	m = MothurOut::getInstance();
+	initialize();
+}
 
-Sequence::Sequence(ifstream& fastaFile){
-	
-	string accession;
-	fastaFile >> accession;
-	setName(accession);
+/***********************************************************************/
 
-	char letter;
-	string sequence;
+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);
+	}			
+}
+//********************************************************************************************************************
+//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();
 	
-	while(fastaFile && letter != '>'){
+		initialize();
+		fastaString >> name;
+		name = name.substr(1);
+		string sequence;
 	
-		letter = fastaFile.get();
+		//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
 		
-		if(isalpha(letter)){
+		sequence = getSequenceString(fastaString);		
+		setAligned(sequence);	
+		//setUnaligned removes any gap characters for us						
+		setUnaligned(sequence);		
+	}
+	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;
+		name = name.substr(1);
+		string sequence;
 		
-			sequence += letter;
+		//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);	
 	}
-	fastaFile.putback(letter);
-
-	if(sequence.find_first_of('-') != string::npos){
-		setAligned(sequence);
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "Sequence");
+		exit(1);
+	}							
+}
+//********************************************************************************************************************
+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();
+	
+			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(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);
 	}
-	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 +236,48 @@ 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();
+	setUnaligned(sequence);	
+
+	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 +288,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 +330,178 @@ 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;
+	
+}
+#ifdef USE_MPI	
+//********************************************************************************************************************
+int Sequence::MPISend(int receiver) {
+	try {
+		//send name - string
+		int length = name.length();
+		char buf[name.length()];
+		strcpy(buf, name.c_str()); 
+		
+		MPI_Send(&length, 1, MPI_INT, receiver, 2001, MPI_COMM_WORLD); 
+
+		MPI_Send(&buf, length, MPI_CHAR, receiver, 2001, MPI_COMM_WORLD);
+	
+		//send aligned - string
+		length = aligned.length();
+		char buf2[aligned.length()];
+		strcpy(buf2, aligned.c_str()); 
+	
+		MPI_Send(&length, 1, MPI_INT, receiver, 2001, MPI_COMM_WORLD); 
+	
+		MPI_Send(&buf2, length, MPI_CHAR, receiver, 2001, MPI_COMM_WORLD);
+	
+		return 0;
+
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "MPISend");
+		exit(1);
+	}
+}
+/**************************************************************************************************/
+int Sequence::MPIRecv(int sender) {
+	try {
+		MPI_Status status;
+	
+		//receive name - string
+		int length;
+		MPI_Recv(&length, 1, MPI_INT, sender, 2001, MPI_COMM_WORLD, &status);
+	
+		char buf[length];
+		MPI_Recv(&buf, length, MPI_CHAR, sender, 2001, MPI_COMM_WORLD, &status);
+		name = buf;
+		
+		//receive aligned - string
+		MPI_Recv(&length, 1, MPI_INT, sender, 2001, MPI_COMM_WORLD, &status);
+	
+		char buf2[length];
+		MPI_Recv(&buf2, length, MPI_CHAR, sender, 2001, MPI_COMM_WORLD, &status);
+		aligned = buf2;
+		
+		setAligned(aligned);
+		
+		return 0;
+
+	}
+	catch(exception& e) {
+		m->errorOut(e, "Sequence", "MPIRecv");
+		exit(1);
+	}
+}
+#endif
+/**************************************************************************************************/