X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=cooccurrencecommand.cpp;h=ef8c9eddce64e93092df8ed1031b37684f09fe38;hb=79a7d3273749b08d4f9f8dfe350c964ff0c4351e;hp=aa553b4903c10cecc582bd33dab6a130f25e4787;hpb=40b3324ee4fcb80efb3459e78efbc3140d5e739a;p=mothur.git

diff --git a/cooccurrencecommand.cpp b/cooccurrencecommand.cpp
index aa553b4..ef8c9ed 100644
--- a/cooccurrencecommand.cpp
+++ b/cooccurrencecommand.cpp
@@ -51,6 +51,26 @@ string CooccurrenceCommand::getHelpString(){
 	}
 }
 //**********************************************************************************************************************
+string CooccurrenceCommand::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 == "summary") {  outputFileName =  "cooccurence.summary"; }
+            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, "CooccurrenceCommand", "getOutputFileNameTag");
+		exit(1);
+	}
+}
+//**********************************************************************************************************************
 CooccurrenceCommand::CooccurrenceCommand(){	
 	try {
 		abort = true; calledHelp = true; 
@@ -176,7 +196,7 @@ int CooccurrenceCommand::execute(){
 		set<string> userLabels = labels;
 
         ofstream out;
-		string outputFileName = outputDir + m->getRootName(m->getSimpleName(sharedfile)) + "cooccurence.summary";
+		string outputFileName = outputDir + m->getRootName(m->getSimpleName(sharedfile)) + getOutputFileNameTag("summary");
         m->openOutputFile(outputFileName, out);
         outputNames.push_back(outputFileName);  outputTypes["summary"].push_back(outputFileName);
         out.setf(ios::fixed, ios::floatfield); out.setf(ios::showpoint);
@@ -272,6 +292,12 @@ int CooccurrenceCommand::execute(){
 int CooccurrenceCommand::getCooccurrence(vector<SharedRAbundVector*>& thisLookUp, ofstream& out){
     try {
         int numOTUS = thisLookUp[0]->getNumBins();
+        
+        if(numOTUS < 2) {
+            m->mothurOut("Not enough OTUs for co-occurrence analysis, skipping"); m->mothurOutEndLine();
+            return 0;
+        }
+        
         vector< vector<int> > co_matrix; co_matrix.resize(thisLookUp[0]->getNumBins());
         for (int i = 0; i < thisLookUp[0]->getNumBins(); i++) { co_matrix[i].resize((thisLookUp.size()), 0); }
         vector<int> columntotal; columntotal.resize(thisLookUp.size(), 0);
@@ -295,11 +321,11 @@ int CooccurrenceCommand::getCooccurrence(vector<SharedRAbundVector*>& thisLookUp
         int nrows = numOTUS;//rows of inital matrix
         int ncols = thisLookUp.size();//groups
         double initscore = 0.0;
-        
+       
         vector<double> stats;
-        double probabilityMatrix[ncols * nrows];
+        vector<double> probabilityMatrix; probabilityMatrix.resize(ncols * nrows, 0);
         vector<vector<int> > nullmatrix(nrows, vector<int>(ncols, 0));
-        
+       
         TrialSwap2 trial;
         
         int n = accumulate( columntotal.begin(), columntotal.end(), 0 );
@@ -543,7 +569,7 @@ int CooccurrenceCommand::getCooccurrence(vector<SharedRAbundVector*>& thisLookUp
         
         m->mothurOut("zscore: " + toString(zscore)); m->mothurOutEndLine();
         m->mothurOut("standard deviation: " + toString(sd)); m->mothurOutEndLine();
-        out << metric << '\t' << thisLookUp[0]->getLabel() << '\t' << nullMean << '\t' << zscore '\t' << sd << endl;
+        out << metric << '\t' << thisLookUp[0]->getLabel() << '\t' << nullMean << '\t' << zscore << '\t' << sd << endl;
         
         return 0;
     }