]> git.donarmstrong.com Git - mothur.git/blobdiff - clusterclassic.cpp
fixed bug with dist.shared subsampling. added mode parameter to dist.shared so...
[mothur.git] / clusterclassic.cpp
index 41c1647bccf6196efb54b31333a7d5298feaaeed..2d1b9a6b781960a824fddafc03b904e8039179e1 100644 (file)
@@ -11,7 +11,7 @@
 #include "progress.hpp"
 
 /***********************************************************************/
-ClusterClassic::ClusterClassic(float c, string f) : method(f), smallDist(1e6), nseqs(0) {
+ClusterClassic::ClusterClassic(float c, string f, bool s) : method(f), smallDist(1e6), nseqs(0), sim(s) {
        try {
                mapWanted = false;  //set to true by mgcluster to speed up overlap merge
        
@@ -19,7 +19,10 @@ ClusterClassic::ClusterClassic(float c, string f) : method(f), smallDist(1e6), n
                cutoff = c;
                aboveCutoff = cutoff + 10000.0;
                m = MothurOut::getInstance();
-               globaldata = GlobalData::getInstance();
+        if(method == "furthest")        {   tag = "fn";   }
+        else if (method == "average")   {   tag = "an";   }
+        else if (method == "weighted")  {   tag = "wn";   }        
+        else if (method == "nearest")   {   tag = "nn";   }
        }
        catch(exception& e) {
                m->errorOut(e, "ClusterClassic", "ClusterClassic");
@@ -37,7 +40,12 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                ifstream fileHandle;
                m->openInputFile(filename, fileHandle);
                
-               fileHandle >> nseqs >> name;
+        string numTest;
+               fileHandle >> numTest >> name;
+        
+        if (!m->isContainingOnlyDigits(numTest)) { m->mothurOut("[ERROR]: expected a number and got " + numTest + ", quitting."); m->mothurOutEndLine(); exit(1); }
+        else { convert(numTest, nseqs); }
+
 
                matrixNames.push_back(name);
 
@@ -102,7 +110,7 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                                                                                fileHandle >> distance;
                                                        
                                                                                if (distance == -1) { distance = 1000000; }
-                                                                               else if (globaldata->sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
+                                                                               else if (sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
                                                                
                                                                                //if(distance < cutoff){
                                                                                        dMatrix[i][j] = distance;
@@ -124,7 +132,7 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                                                                                if (m->control_pressed) { delete reading; fileHandle.close(); return 0;  }
                                
                                                                                if (distance == -1) { distance = 1000000; }
-                                                                               else if (globaldata->sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
+                                                                               else if (sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
                                                                                
                                                                                //if(distance < cutoff){
                                                                                        if (distance < smallDist) { smallDist = distance; }
@@ -162,7 +170,7 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                                                                                if (m->control_pressed) {  fileHandle.close();  delete reading; return 0; }
                                                                                
                                                                                if (distance == -1) { distance = 1000000; }
-                                                                               else if (globaldata->sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
+                                                                               else if (sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.
                                                                                
                                                                                if(j < i){
                                                                                        if (distance < smallDist) { smallDist = distance; }
@@ -185,7 +193,7 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                                                                                if (m->control_pressed) {  fileHandle.close();  delete reading; return 0; }
                                                                                
                                                                           if (distance == -1) { distance = 1000000; }
-                                                                               else if (globaldata->sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.                                                        
+                                                                               else if (sim) { distance = 1.0 - distance;  }  //user has entered a sim matrix that we need to convert.                                                        
                                                                                
                                                                                if(j < i){
                                                                                        if (distance < smallDist) { smallDist = distance; }
@@ -215,6 +223,8 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
                rabund = new RAbundVector(list->getRAbundVector());
                
                fileHandle.close();
+               
+               return 0;
        }
        catch(exception& e) {
                m->errorOut(e, "ClusterClassic", "readPhylipFile");
@@ -226,7 +236,7 @@ int ClusterClassic::readPhylipFile(string filename, NameAssignment* nameMap) {
 //sets smallCol and smallRow, returns distance
 double ClusterClassic::getSmallCell() {
        try {
-               
+                       
                smallDist = aboveCutoff;
                smallRow = 1;
                smallCol = 0;
@@ -279,6 +289,10 @@ void ClusterClassic::clusterBins(){
 
                rabund->set(smallRow, rabund->get(smallRow)+rabund->get(smallCol));     
                rabund->set(smallCol, 0);       
+               /*for (int i = smallCol+1; i < rabund->size(); i++) {
+                       rabund->set((i-1), rabund->get(i));
+               }
+               rabund->resize((rabund->size()-1));*/
                rabund->setLabel(toString(smallDist));
 
        //      cout << '\t' << rabund->get(smallRow) << '\t' << rabund->get(smallCol) << endl;
@@ -296,6 +310,10 @@ void ClusterClassic::clusterNames(){
                
                list->set(smallRow, list->get(smallRow)+','+list->get(smallCol));
                list->set(smallCol, "");        
+               /*for (int i = smallCol+1; i < list->size(); i++) {
+                       list->set((i-1), list->get(i));
+               }
+               list->resize((list->size()-1));*/
                list->setLabel(toString(smallDist));
        
        //      cout << '\t' << list->get(smallRow) << '\t' << list->get(smallCol) << endl;
@@ -308,36 +326,21 @@ void ClusterClassic::clusterNames(){
 /***********************************************************************/
 void ClusterClassic::update(double& cutOFF){
        try {
-//cout << "before update " << endl;
 //print();             
                getSmallCell();
                
                int r, c;
                r = smallRow; c = smallCol;
-               //because we only store lt, we need to make sure we grab the right location
-               //if (smallRow < smallCol) { c = smallRow; r = smallCol; } //smallRow is really our column value
-               //else { r = smallRow; c = smallCol; } //smallRow is the row value
-               
-               //reset rows smallest distance
-               //rowSmallDists[r].dist = aboveCutoff; rowSmallDists[r].row = 0; rowSmallDists[r].col = 0;
-               //rowSmallDists[c].dist = aboveCutoff; rowSmallDists[c].row = 0; rowSmallDists[c].col = 0;
-               
-               //if your rows smallest distance is from smallRow or smallCol, reset
-               //for(int i=0;i<nseqs;i++){
-                       //if ((rowSmallDists[i].row == r) || (rowSmallDists[i].row == c) || (rowSmallDists[i].col == r) || (rowSmallDists[i].col == c)) {
-                       //      rowSmallDists[i].dist = aboveCutoff; rowSmallDists[i].row = 0; rowSmallDists[i].col = 0;
-                       //}
-               //}
-               
+                               
                for(int i=0;i<nseqs;i++){
                        if(i != r && i != c){
                                double distRow, distCol, newDist;
                                if (i > r) { distRow = dMatrix[i][r]; }
                                else { distRow =  dMatrix[r][i]; }
-                               
+
                                if (i > c) { distCol = dMatrix[i][c]; dMatrix[i][c] = aboveCutoff; } //like removeCell
                                else { distCol =  dMatrix[c][i]; dMatrix[c][i] = aboveCutoff; }
-                                       
+                               
                                if(method == "furthest"){
                                        newDist = max(distRow, distCol);
                                }
@@ -352,39 +355,33 @@ void ClusterClassic::update(double& cutOFF){
                                else if (method == "nearest"){
                                        newDist = min(distRow, distCol);
                                }
-                               
+                               //cout << "newDist = " << newDist << endl;      
                                if (i > r) { dMatrix[i][r] = newDist; }
                                else { dMatrix[r][i] = newDist; }
                                
-                               //if (newDist < rowSmallDists[i].dist) {  rowSmallDists[i].dist = newDist; rowSmallDists[i].col = r; rowSmallDists[i].row = i;  }
                        }
-                       //cout << "rowsmall = " << i << '\t' << rowSmallDists[i].dist << endl;  
                }
                        
                clusterBins();
                clusterNames();
-       
-               //find new small for 2 rows we just merged
-               //colDist temp(0,0,100.0);
-               //rowSmallDists[r] = temp;
-
-               //for (int i = 0; i < dMatrix[r].size(); i++) {
-               //      if (dMatrix[r][i] < rowSmallDists[r].dist) { rowSmallDists[r].dist = dMatrix[r][i]; rowSmallDists[r].col = r; rowSmallDists[r].row = i; }
-               //}
-               //for (int i = dMatrix[r].size()+1; i < dMatrix.size(); i++) {
-               //      if (dMatrix[i][dMatrix[r].size()] < rowSmallDists[r].dist) { rowSmallDists[r].dist = dMatrix[i][dMatrix[r].size()]; rowSmallDists[r].col = r; rowSmallDists[r].row = i; }
-               //}
                
-               //rowSmallDists[c] = temp;
-               //for (int i = 0; i < dMatrix[c].size(); i++) {
-               //      if (dMatrix[c][i] < rowSmallDists[c].dist) { rowSmallDists[c].dist = dMatrix[c][i]; rowSmallDists[c].col = c; rowSmallDists[c].row = i; }
-               //}
-               //for (int i = dMatrix[c].size()+1; i < dMatrix.size(); i++) {
-               //      if (dMatrix[i][dMatrix[c].size()] < rowSmallDists[c].dist) { rowSmallDists[c].dist = dMatrix[i][dMatrix[c].size()]; rowSmallDists[c].col = c; rowSmallDists[c].row = i; }
-               //}
+               //resize each row
+               /*for(int i=0;i<nseqs;i++){
+                       for(int j=c+1;j<dMatrix[i].size();j++){
+                               dMatrix[i][j-1]=dMatrix[i][j];
+                       }
+               }                       
                
-               //cout << "after update " << endl;
-               //print();
+               //resize each col
+               for(int i=c+1;i<nseqs;i++){
+                       for(int j=0;j < dMatrix[i-1].size();j++){
+                               dMatrix[i-1][j]=dMatrix[i][j];
+                       }
+               }       
+               
+               nseqs--;
+               dMatrix.pop_back();*/
+
        }
        catch(exception& e) {
                m->errorOut(e, "ClusterClassic", "update");
@@ -446,11 +443,11 @@ void ClusterClassic::print() {
 try {
                //update location of seqs in smallRow since they move to smallCol now
                for (int i = 0; i < dMatrix.size(); i++) {
-                       cout << "row = " << i << '\t';
+                       m->mothurOut("row = " + toString(i) + "\t");
                        for (int j = 0; j < dMatrix[i].size(); j++) {
-                               cout << dMatrix[i][j] << '\t';
+                               m->mothurOut(toString(dMatrix[i][j]) + "\t");
                        }
-                       cout << endl;
+                       m->mothurOutEndLine();
                }
        }
        catch(exception& e) {