X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=binsequencecommand.cpp;h=0c867e35f730cc8e72c66153b14c23b5f54ce6f3;hb=6ede3bf5c0a9eedb23f24577a97da81ab3e1f7df;hp=2117daff4f2558e26bcf0008666f87bee0180f4d;hpb=0caf3fbabaa3ece404f8ce77f4c883dc5b1bf1dc;p=mothur.git

diff --git a/binsequencecommand.cpp b/binsequencecommand.cpp
index 2117daf..0c867e3 100644
--- a/binsequencecommand.cpp
+++ b/binsequencecommand.cpp
@@ -50,6 +50,26 @@ string BinSeqCommand::getHelpString(){
 	}
 }
 //**********************************************************************************************************************
+string BinSeqCommand::getOutputFileNameTag(string type, string inputName=""){	
+	try {
+        string outputFileName = "";
+		map<string, vector<string> >::iterator it;
+        
+        //is this a type this command creates
+        it = outputTypes.find(type);
+        if (it == outputTypes.end()) {  m->mothurOut("[ERROR]: this command doesn't create a " + type + " output file.\n"); }
+        else {
+            if (type == "fasta") {  outputFileName =  "fasta"; }
+            else { m->mothurOut("[ERROR]: No definition for type " + type + " output file tag.\n"); m->control_pressed = true;  }
+        }
+        return outputFileName;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "BinSeqCommand", "getOutputFileNameTag");
+		exit(1);
+	}
+}
+//**********************************************************************************************************************
 BinSeqCommand::BinSeqCommand(){	
 	try {
 		abort = true; calledHelp = true; 
@@ -342,79 +362,48 @@ void BinSeqCommand::readNamesFile() {
 //return 1 if error, 0 otherwise
 int BinSeqCommand::process(ListVector* list) {
 	try {
-				string binnames, name, sequence;
-				
-				string outputFileName = outputDir + m->getRootName(m->getSimpleName(listfile)) + list->getLabel() + ".fasta";
-				m->openOutputFile(outputFileName, out);
-				
-				//save to output list of output file names
-				outputNames.push_back(outputFileName);  outputTypes["fasta"].push_back(outputFileName);
-
-				m->mothurOut(list->getLabel()); m->mothurOutEndLine();
-				
-				//for each bin in the list vector
-				for (int i = 0; i < list->size(); i++) {
-					
-					if (m->control_pressed) {  return 1; }
-					
-					binnames = list->get(i);
-					while (binnames.find_first_of(',') != -1) { 
-						name = binnames.substr(0,binnames.find_first_of(','));
-						binnames = binnames.substr(binnames.find_first_of(',')+1, binnames.length());
-						
-						//do work for that name
-						sequence = fasta->getSequence(name);
-						if (sequence != "not found") {
-							//if you don't have groups
-							if (groupfile == "") {
-								name = name + "\t" + toString(i+1);
-								out << ">" << name << endl;
-								out << sequence << endl;
-							}else {//if you do have groups
-								string group = groupMap->getGroup(name);
-								if (group == "not found") {  
-									m->mothurOut(name + " is missing from your group file. Please correct. ");  m->mothurOutEndLine();
-									return 1;
-								}else{
-									name = name + "\t" + group + "\t" + toString(i+1);
-									out << ">" << name << endl;
-									out << sequence << endl;
-								}
-							}
-						}else { 
-							m->mothurOut(name + " is missing from your fasta or name file. Please correct. "); m->mothurOutEndLine();
-							return 1;
-						}
-						
-					}
-					
-					//get last name
-					sequence = fasta->getSequence(binnames);
-					if (sequence != "not found") {
-						//if you don't have groups
-						if (groupfile == "") {
-							binnames = binnames + "\t" + toString(i+1);
-							out << ">" << binnames << endl;
-							out << sequence << endl;
-						}else {//if you do have groups
-							string group = groupMap->getGroup(binnames);
-							if (group == "not found") {  
-								m->mothurOut(binnames + " is missing from your group file. Please correct. "); m->mothurOutEndLine();
-								return 1;
-							}else{
-								binnames = binnames + "\t" + group + "\t" + toString(i+1);
-								out << ">" << binnames << endl;
-								out << sequence << endl;
-							}
-						}
-					}else { 
-						m->mothurOut(binnames + " is missing from your fasta or name file. Please correct. "); m->mothurOutEndLine();
-						return 1;
-					}
-				}
-					
-				out.close();
-				return 0;
+        string outputFileName = outputDir + m->getRootName(m->getSimpleName(listfile)) + list->getLabel() + getOutputFileNameTag("fasta");
+        m->openOutputFile(outputFileName, out);
+        outputNames.push_back(outputFileName);  outputTypes["fasta"].push_back(outputFileName);
+        
+        m->mothurOut(list->getLabel()); m->mothurOutEndLine();
+        
+        //for each bin in the list vector
+        for (int i = 0; i < list->size(); i++) {
+            
+            if (m->control_pressed) {  return 1; }
+            
+            string binnames = list->get(i);
+            vector<string> names;
+            m->splitAtComma(binnames, names);
+            for (int j = 0; j < names.size(); j++) {
+                string name = names[j];
+                
+                //do work for that name
+                string sequence = fasta->getSequence(name);
+                if (sequence != "not found") {
+                    //if you don't have groups
+                    if (groupfile == "") {
+                        name = name + "\t" + toString(i+1);
+                        out << ">" << name << endl;
+                        out << sequence << endl;
+                    }else {//if you do have groups
+                        string group = groupMap->getGroup(name);
+                        if (group == "not found") {  
+                            m->mothurOut(name + " is missing from your group file. Please correct. ");  m->mothurOutEndLine();
+                            return 1;
+                        }else{
+                            name = name + "\t" + group + "\t" + toString(i+1);
+                            out << ">" << name << endl;
+                            out << sequence << endl;
+                        }
+                    }
+                }else { m->mothurOut(name + " is missing from your fasta or name file. Please correct. "); m->mothurOutEndLine(); return 1; }
+            }
+        }
+        
+        out.close();
+        return 0;
 
 	}
 	catch(exception& e) {