X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=classifyseqscommand.h;h=acee70c5cb474dee4e25eea4a6dcf6b5bd55fa06;hb=0caf3fbabaa3ece404f8ce77f4c883dc5b1bf1dc;hp=eef96ffbe5d32f88f9f9552165b299f0247505dc;hpb=4c5e7a20a03ddc6feb49ff9d21fcb4c79bc5508d;p=mothur.git

diff --git a/classifyseqscommand.h b/classifyseqscommand.h
index eef96ff..acee70c 100644
--- a/classifyseqscommand.h
+++ b/classifyseqscommand.h
@@ -10,7 +10,7 @@
  *
  */
 
-#include "mothur.h"
+
 #include "command.hpp"
 #include "classify.h"
 #include "referencedb.h"
@@ -72,16 +72,16 @@ private:
 	string fastaFileName, templateFileName, distanceFileName, namefile, search, method, taxonomyFileName, outputDir, groupfile;
 	int processors, kmerSize, numWanted, cutoff, iters;
 	float match, misMatch, gapOpen, gapExtend;
-	bool abort, probs, save;
+	bool abort, probs, save, flip;
 	
-	int driver(linePair*, string, string, string);
+	int driver(linePair*, string, string, string, string);
 	void appendTaxFiles(string, string);
-	int createProcesses(string, string, string); 
+	int createProcesses(string, string, string, string); 
 	string addUnclassifieds(string, int);
 	
 	int MPIReadNamesFile(string);
 	#ifdef USE_MPI
-	int driverMPI(int, int, MPI_File&, MPI_File&, MPI_File&, vector<unsigned long long>&);
+	int driverMPI(int, int, MPI_File&, MPI_File&, MPI_File&, MPI_File&, vector<unsigned long long>&);
 	#endif
 };
 
@@ -93,16 +93,17 @@ struct classifyData {
 	string taxFName; 
 	string tempTFName; 
 	string filename;
-	string search, taxonomyFileName, templateFileName, method;
+	string search, taxonomyFileName, templateFileName, method, accnos;
 	unsigned long long start;
 	unsigned long long end;
 	MothurOut* m;
 	float match, misMatch, gapOpen, gapExtend;
 	int count, kmerSize, threadID, cutoff, iters, numWanted;
-	bool probs;
+	bool probs, flip;
 	 
 	classifyData(){}
-	classifyData(bool p, string me, string te, string tx, string a, string r, string f, string se, int ks, int i, int numW, MothurOut* mout, unsigned long long st, unsigned long long en, float ma, float misMa, float gapO, float gapE, int cut, int tid) {
+	classifyData(string acc, bool p, string me, string te, string tx, string a, string r, string f, string se, int ks, int i, int numW, MothurOut* mout, unsigned long long st, unsigned long long en, float ma, float misMa, float gapO, float gapE, int cut, int tid, bool fli) {
+		accnos = acc;
 		taxonomyFileName = tx;
 		templateFileName = te;
 		taxFName = a;
@@ -124,11 +125,12 @@ struct classifyData {
 		threadID = tid;
 		probs = p;
 		count = 0;
+		flip = fli;
 	}
 };
 
 /**************************************************************************************************/
-#if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+#if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
 #else
 static DWORD WINAPI MyClassThreadFunction(LPVOID lpParam){ 
 	classifyData* pDataArray;
@@ -141,6 +143,9 @@ static DWORD WINAPI MyClassThreadFunction(LPVOID lpParam){
 		ofstream outTaxSimple;
 		pDataArray->m->openOutputFile(pDataArray->tempTFName, outTaxSimple);
 		
+		ofstream outAcc;
+		pDataArray->m->openOutputFile(pDataArray->accnos, outAcc);
+		
 		ifstream inFASTA;
 		pDataArray->m->openInputFile(pDataArray->filename, inFASTA);
 		
@@ -157,12 +162,12 @@ static DWORD WINAPI MyClassThreadFunction(LPVOID lpParam){
 		
 		//make classify
 		Classify* myclassify;
-		if(pDataArray->method == "bayesian"){	myclassify = new Bayesian("saved", "saved", pDataArray->search, pDataArray->kmerSize, pDataArray->cutoff, pDataArray->iters, pDataArray->threadID);		}
-		else if(pDataArray->method == "knn"){	myclassify = new Knn("saved", "saved", pDataArray->search, pDataArray->kmerSize, pDataArray->gapOpen, pDataArray->gapExtend, pDataArray->match, pDataArray->misMatch, pDataArray->numWanted, pDataArray->threadID);				}
+		if(pDataArray->method == "bayesian"){	myclassify = new Bayesian(pDataArray->taxonomyFileName, pDataArray->templateFileName, pDataArray->search, pDataArray->kmerSize, pDataArray->cutoff, pDataArray->iters, pDataArray->threadID, pDataArray->flip);		}
+		else if(pDataArray->method == "knn"){	myclassify = new Knn(pDataArray->taxonomyFileName, pDataArray->templateFileName, pDataArray->search, pDataArray->kmerSize, pDataArray->gapOpen, pDataArray->gapExtend, pDataArray->match, pDataArray->misMatch, pDataArray->numWanted, pDataArray->threadID);				}
 		else {
 			pDataArray->m->mothurOut(pDataArray->search + " is not a valid method option. I will run the command using bayesian.");
 			pDataArray->m->mothurOutEndLine();
-			myclassify = new Bayesian(pDataArray->taxonomyFileName, pDataArray->templateFileName, pDataArray->search, pDataArray->kmerSize, pDataArray->cutoff, pDataArray->iters, pDataArray->threadID);	
+			myclassify = new Bayesian(pDataArray->taxonomyFileName, pDataArray->templateFileName, pDataArray->search, pDataArray->kmerSize, pDataArray->cutoff, pDataArray->iters, pDataArray->threadID, pDataArray->flip);	
 		}
 		
 		if (pDataArray->m->control_pressed) { delete myclassify; return 0; }
@@ -180,16 +185,19 @@ static DWORD WINAPI MyClassThreadFunction(LPVOID lpParam){
 				
 				if (pDataArray->m->control_pressed) { delete candidateSeq; return 0; }
 				
-				if (taxonomy != "bad seq") {
-					//output confidence scores or not
-					if (pDataArray->probs) {
-						outTax << candidateSeq->getName() << '\t' << taxonomy << endl;
-					}else{
-						outTax << candidateSeq->getName() << '\t' << myclassify->getSimpleTax() << endl;
-					}
-					
-					outTaxSimple << candidateSeq->getName() << '\t' << myclassify->getSimpleTax() << endl;
+				if (taxonomy == "unknown;") { pDataArray->m->mothurOut("[WARNING]: " + candidateSeq->getName() + " could not be classified. You can use the remove.lineage command with taxon=unknown; to remove such sequences."); pDataArray->m->mothurOutEndLine(); }
+
+				//output confidence scores or not
+				if (pDataArray->probs) {
+					outTax << candidateSeq->getName() << '\t' << taxonomy << endl;
+				}else{
+					outTax << candidateSeq->getName() << '\t' << myclassify->getSimpleTax() << endl;
 				}
+					
+				outTaxSimple << candidateSeq->getName() << '\t' << myclassify->getSimpleTax() << endl;
+					
+				if (myclassify->getFlipped()) { outAcc << candidateSeq->getName() << endl; }
+				
 				count++;
 			}
 			delete candidateSeq;