]> git.donarmstrong.com Git - mothur.git/blobdiff - sharedrabundvector.cpp
1.18.1
[mothur.git] / sharedrabundvector.cpp
index 114de90e7b9ee2490f3f47379487a02ba62d5bb1..b41c5fafa40255b1535526f79c2ce1c5dc637a43 100644 (file)
@@ -14,8 +14,7 @@
 
 
 /***********************************************************************/
-
-SharedRAbundVector::SharedRAbundVector() : DataVector(), maxRank(0), numBins(0), numSeqs(0) {globaldata = GlobalData::getInstance();}
+SharedRAbundVector::SharedRAbundVector() : DataVector(), maxRank(0), numBins(0), numSeqs(0) {} 
 /***********************************************************************/
 
 SharedRAbundVector::~SharedRAbundVector() {
@@ -26,7 +25,6 @@ SharedRAbundVector::~SharedRAbundVector() {
 /***********************************************************************/
 
 SharedRAbundVector::SharedRAbundVector(int n) : DataVector(), maxRank(0), numBins(n), numSeqs(0) {
-               globaldata = GlobalData::getInstance();
                individual newGuy;
                //initialize data
                for (int i=0; i< n; i++) {
@@ -61,20 +59,21 @@ SharedRAbundVector::SharedRAbundVector(string id, vector<individual> rav) : Data
 //reads a shared file
 SharedRAbundVector::SharedRAbundVector(ifstream& f) : DataVector(), maxRank(0), numBins(0), numSeqs(0) {
        try {
-               globaldata = GlobalData::getInstance();
-               
-               if (globaldata->gGroupmap == NULL) {  groupmap = new GroupMap(); }
-               
+               m->namesOfGroups.clear();
+                               
                int num, inputData, count;
                count = 0;  
                string holdLabel, nextLabel, groupN;
                individual newguy;
                
-               for (int i = 0; i < lookup.size(); i++) {  delete lookup[i];  }
-               lookup.clear();
+               for (int i = 0; i < lookup.size(); i++) {  delete lookup[i]; lookup[i] = NULL; }  lookup.clear();
+               
+               if (m->saveNextLabel == "") {  f >> label;  }
+               else { label = m->saveNextLabel; }
                
                //read in first row since you know there is at least 1 group.
-               f >> label >> groupN >> num;
+               f >> groupN >> num;
+
                holdLabel = label;
                
                //add new vector to lookup
@@ -83,11 +82,7 @@ SharedRAbundVector::SharedRAbundVector(ifstream& f) : DataVector(), maxRank(0),
                lookup[0]->setLabel(label);
                lookup[0]->setGroup(groupN);
                
-               if (globaldata->gGroupmap == NULL) { 
-                       //save group in groupmap
-                       groupmap->namesOfGroups.push_back(groupN);
-                       groupmap->groupIndex[groupN] = 0;
-               }
+               m->namesOfGroups.push_back(groupN);
                
                //fill vector.  data = first sharedrabund in file
                for(int i=0;i<num;i++){
@@ -95,26 +90,21 @@ SharedRAbundVector::SharedRAbundVector(ifstream& f) : DataVector(), maxRank(0),
                        
                        lookup[0]->push_back(inputData, groupN); //abundance, bin, group
                        push_back(inputData, groupN);
-                       numSeqs += inputData;
-                       numBins++;
+                       //numSeqs += inputData;
+                       //numBins++;
                        if (inputData > maxRank) { maxRank = inputData; }
-                       
                }
                
-               gobble(f);
-               
-               if (f.eof() != true) { f >> nextLabel; }
+               m->gobble(f);
                
+               if (!(f.eof())) { f >> nextLabel; }
+       
                //read the rest of the groups info in
                while ((nextLabel == holdLabel) && (f.eof() != true)) {
                        f >> groupN >> num;
                        count++;
                        
-                       if (globaldata->gGroupmap == NULL) { 
-                               //save group in groupmap
-                               groupmap->namesOfGroups.push_back(groupN);
-                               groupmap->groupIndex[groupN] = count;
-                       }
+                       m->namesOfGroups.push_back(groupN);
                        
                        //add new vector to lookup
                        temp = new SharedRAbundVector();
@@ -128,15 +118,11 @@ SharedRAbundVector::SharedRAbundVector(ifstream& f) : DataVector(), maxRank(0),
                                lookup[count]->push_back(inputData, groupN); //abundance, bin, group
                        }
                        
-                       gobble(f);
+                       m->gobble(f);
                                
                        if (f.eof() != true) { f >> nextLabel; }
                }
-               
-               //put file pointer back since you are now at a new distance label
-               for (int i = 0; i < nextLabel.length(); i++) { f.unget();  }
-       
-               if (globaldata->gGroupmap == NULL) { globaldata->gGroupmap = groupmap;  }
+                       m->saveNextLabel = nextLabel;
                
        }
        catch(exception& e) {
@@ -202,6 +188,16 @@ vector <individual> SharedRAbundVector::getData(){
 }
 /***********************************************************************/
 
+void SharedRAbundVector::clear(){
+       numBins = 0;
+       maxRank = 0;
+       numSeqs = 0;
+       data.clear();
+       for (int i = 0; i < lookup.size(); i++) {  delete lookup[i]; lookup[i] = NULL; }
+       lookup.clear();
+}
+/***********************************************************************/
+
 void SharedRAbundVector::push_back(int binSize, string groupName){
        try {
                individual newGuy;
@@ -358,18 +354,22 @@ vector<SharedRAbundVector*> SharedRAbundVector::getSharedRAbundVectors(){
                SharedUtil* util;
                util = new SharedUtil();
                
-               util->setGroups(globaldata->Groups, globaldata->gGroupmap->namesOfGroups);
-
+               util->setGroups(m->Groups, m->namesOfGroups);
+               
+               bool remove = false;
                for (int i = 0; i < lookup.size(); i++) {
                        //if this sharedrabund is not from a group the user wants then delete it.
-                       if (util->isValidGroup(lookup[i]->getGroup(), globaldata->Groups) == false) { 
-                               delete lookup[i]; 
+                       if (util->isValidGroup(lookup[i]->getGroup(), m->Groups) == false) { 
+                               remove = true;
+                               delete lookup[i]; lookup[i] = NULL;
                                lookup.erase(lookup.begin()+i); 
                                i--; 
                        }
                }
                
                delete util;
+               
+               if (remove) { eliminateZeroOTUS(lookup); }
        
                return lookup;
        }
@@ -378,6 +378,81 @@ vector<SharedRAbundVector*> SharedRAbundVector::getSharedRAbundVectors(){
                exit(1);
        }
 }
+//**********************************************************************************************************************
+int SharedRAbundVector::eliminateZeroOTUS(vector<SharedRAbundVector*>& thislookup) {
+               try {
+                       
+                       vector<SharedRAbundVector*> newLookup;
+                       for (int i = 0; i < thislookup.size(); i++) {
+                               SharedRAbundVector* temp = new SharedRAbundVector();
+                               temp->setLabel(thislookup[i]->getLabel());
+                               temp->setGroup(thislookup[i]->getGroup());
+                               newLookup.push_back(temp);
+                       }
+                       
+                       //for each bin
+                       for (int i = 0; i < thislookup[0]->getNumBins(); i++) {
+                               if (m->control_pressed) { for (int j = 0; j < newLookup.size(); j++) {  delete newLookup[j];  } return 0; }
+                               
+                               //look at each sharedRabund and make sure they are not all zero
+                               bool allZero = true;
+                               for (int j = 0; j < thislookup.size(); j++) {
+                                       if (thislookup[j]->getAbundance(i) != 0) { allZero = false;  break;  }
+                               }
+                               
+                               //if they are not all zero add this bin
+                               if (!allZero) {
+                                       for (int j = 0; j < thislookup.size(); j++) {
+                                               newLookup[j]->push_back(thislookup[j]->getAbundance(i), thislookup[j]->getGroup());
+                                       }
+                               }
+                       }
+                       
+                       for (int j = 0; j < thislookup.size(); j++) {  delete thislookup[j];  }
+                       
+                       thislookup = newLookup;
+                       
+                       return 0;
+                       
+               }
+               catch(exception& e) {
+                       m->errorOut(e, "SharedRAbundVector", "eliminateZeroOTUS");
+                       exit(1);
+               }
+       }
+       
+/***********************************************************************/
+vector<SharedRAbundFloatVector*> SharedRAbundVector::getSharedRAbundFloatVectors(vector<SharedRAbundVector*> thislookup){
+       try {
+               vector<SharedRAbundFloatVector*> newLookupFloat;        
+               for (int i = 0; i < lookup.size(); i++) {
+                       SharedRAbundFloatVector* temp = new SharedRAbundFloatVector();
+                       temp->setLabel(thislookup[i]->getLabel());
+                       temp->setGroup(thislookup[i]->getGroup());
+                       newLookupFloat.push_back(temp);
+               }
+               
+               for (int i = 0; i < thislookup.size(); i++) {
+                       
+                       for (int j = 0; j < thislookup[i]->getNumBins(); j++) {
+                               
+                               if (m->control_pressed) { return newLookupFloat; }
+                               
+                               int abund = thislookup[i]->getAbundance(j);
+                               
+                               float relabund = abund / (float) thislookup[i]->getNumSeqs();
+                               
+                               newLookupFloat[i]->push_back(relabund, thislookup[i]->getGroup());
+                       }
+               }
+               
+               return newLookupFloat;
+       }
+       catch(exception& e) {
+               m->errorOut(e, "SharedRAbundVector", "getSharedRAbundVectors");
+               exit(1);
+       }
+}
 /***********************************************************************/
 
 RAbundVector SharedRAbundVector::getRAbundVector() {
@@ -483,15 +558,15 @@ SharedOrderVector SharedRAbundVector::getSharedOrderVector() {
 OrderVector SharedRAbundVector::getOrderVector(map<string,int>* nameMap = NULL) {
        try {
                OrderVector ov;
-       
-               for(int i=0;i<data.size();i++){
+               for(int i=0;i<numBins;i++){
                        for(int j=0;j<data[i].abundance;j++){
                                ov.push_back(i);
                        }
                }
                random_shuffle(ov.begin(), ov.end());
-
+               
                ov.setLabel(label);     
+
                return ov;
        }
        catch(exception& e) {